agenta 0.48.8__py3-none-any.whl → 0.48.10__py3-none-any.whl

agenta/client/backend/tracing/types/__init__.py

@@ -0,0 +1,8 @@
+# This file was auto-generated by Fern from our API Definition.
+
+# isort: skip_file
+
+from .fetch_trace_request_trace_id import FetchTraceRequestTraceId
+from .remove_trace_request_trace_id import RemoveTraceRequestTraceId
+
+__all__ = ["FetchTraceRequestTraceId", "RemoveTraceRequestTraceId"]

agenta/client/backend/{types/variant_action_enum.py → tracing/types/fetch_trace_request_trace_id.py}

@@ -2,4 +2,4 @@
 
 import typing
 
-VariantActionEnum = typing.Union[typing.Literal["START", "STOP"], typing.Any]
+FetchTraceRequestTraceId = typing.Union[str, int]

agenta/client/backend/tracing/types/remove_trace_request_trace_id.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+RemoveTraceRequestTraceId = typing.Union[str, int]

agenta/client/backend/types/__init__.py

@@ -1,8 +1,17 @@
 # This file was auto-generated by Fern from our API Definition.
 
+# isort: skip_file
+
+# Import Pydantic schema fix to prevent recursion errors
+# try:
+#     from .__pydantic_init__ import SafeGenerateJsonSchema, disable_schema_generation
+# except ImportError:
+#     pass  # Fallback gracefully if the file doesn't exist
+
+
+from .account_request import AccountRequest
 from .account_response import AccountResponse
 from .agenta_node_dto import AgentaNodeDto
-from .agenta_node_dto_nodes_value import AgentaNodeDtoNodesValue
 from .agenta_nodes_response import AgentaNodesResponse
 from .agenta_root_dto import AgentaRootDto
 from .agenta_roots_response import AgentaRootsResponse
@@ -11,10 +20,25 @@ from .agenta_trees_response import AgentaTreesResponse
 from .aggregated_result import AggregatedResult
 from .aggregated_result_evaluator_config import AggregatedResultEvaluatorConfig
 from .analytics_response import AnalyticsResponse
+from .annotation import Annotation
+from .annotation_create import AnnotationCreate
+from .annotation_edit import AnnotationEdit
+from .annotation_kind import AnnotationKind
+from .annotation_link import AnnotationLink
+from .annotation_link_response import AnnotationLinkResponse
+from .annotation_query import AnnotationQuery
+from .annotation_query_request import AnnotationQueryRequest
+from .annotation_reference import AnnotationReference
+from .annotation_references import AnnotationReferences
+from .annotation_response import AnnotationResponse
+from .annotation_source import AnnotationSource
+from .annotations_response import AnnotationsResponse
 from .app import App
 from .app_variant_response import AppVariantResponse
 from .app_variant_revision import AppVariantRevision
+from .artifact import Artifact
 from .base_output import BaseOutput
+from .body_fetch_workflow_revision import BodyFetchWorkflowRevision
 from .body_import_testset import BodyImportTestset
 from .bucket_dto import BucketDto
 from .collect_status_response import CollectStatusResponse
@@ -29,7 +53,6 @@ from .custom_provider_kind import CustomProviderKind
 from .custom_provider_settings_dto import CustomProviderSettingsDto
 from .data import Data
 from .delete_evaluation import DeleteEvaluation
-from .docker_env_vars import DockerEnvVars
 from .environment_output import EnvironmentOutput
 from .environment_output_extended import EnvironmentOutputExtended
 from .environment_revision import EnvironmentRevision
@@ -43,40 +66,72 @@ from .evaluation_status_enum import EvaluationStatusEnum
 from .evaluation_type import EvaluationType
 from .evaluator import Evaluator
 from .evaluator_config import EvaluatorConfig
+from .evaluator_flags import EvaluatorFlags
 from .evaluator_mapping_output_interface import EvaluatorMappingOutputInterface
 from .evaluator_output_interface import EvaluatorOutputInterface
+from .evaluator_query import EvaluatorQuery
+from .evaluator_query_request import EvaluatorQueryRequest
+from .evaluator_request import EvaluatorRequest
+from .evaluator_response import EvaluatorResponse
+from .evaluators_response import EvaluatorsResponse
 from .exception_dto import ExceptionDto
+from .focus import Focus
+from .format import Format
+from .full_json_input import FullJsonInput
+from .full_json_output import FullJsonOutput
 from .get_config_response import GetConfigResponse
-from .header_dto import HeaderDto
+from .header import Header
 from .http_validation_error import HttpValidationError
 from .human_evaluation import HumanEvaluation
 from .human_evaluation_scenario import HumanEvaluationScenario
 from .human_evaluation_scenario_input import HumanEvaluationScenarioInput
 from .human_evaluation_scenario_output import HumanEvaluationScenarioOutput
-from .image import Image
 from .invite_request import InviteRequest
 from .legacy_analytics_response import LegacyAnalyticsResponse
 from .legacy_data_point import LegacyDataPoint
+from .legacy_evaluator import LegacyEvaluator
 from .legacy_scope_request import LegacyScopeRequest
 from .legacy_scopes_response import LegacyScopesResponse
+from .legacy_subscription_request import LegacySubscriptionRequest
 from .legacy_user_request import LegacyUserRequest
 from .legacy_user_response import LegacyUserResponse
 from .lifecycle_dto import LifecycleDto
 from .link_dto import LinkDto
 from .list_api_keys_response import ListApiKeysResponse
 from .llm_run_rate_limit import LlmRunRateLimit
+from .meta_request import MetaRequest
 from .metrics_dto import MetricsDto
 from .new_testset import NewTestset
 from .node_dto import NodeDto
 from .node_type import NodeType
 from .o_tel_context_dto import OTelContextDto
+from .o_tel_event import OTelEvent
 from .o_tel_event_dto import OTelEventDto
 from .o_tel_extra_dto import OTelExtraDto
+from .o_tel_flat_span import OTelFlatSpan
+from .o_tel_flat_span_input_end_time import OTelFlatSpanInputEndTime
+from .o_tel_flat_span_input_start_time import OTelFlatSpanInputStartTime
+from .o_tel_flat_span_output_end_time import OTelFlatSpanOutputEndTime
+from .o_tel_flat_span_output_start_time import OTelFlatSpanOutputStartTime
+from .o_tel_link import OTelLink
 from .o_tel_link_dto import OTelLinkDto
+from .o_tel_links_response import OTelLinksResponse
+from .o_tel_span import OTelSpan
 from .o_tel_span_dto import OTelSpanDto
+from .o_tel_span_input_end_time import OTelSpanInputEndTime
+from .o_tel_span_input_spans_value import OTelSpanInputSpansValue
+from .o_tel_span_input_start_time import OTelSpanInputStartTime
 from .o_tel_span_kind import OTelSpanKind
-from .o_tel_spans_response import OTelSpansResponse
+from .o_tel_span_output_end_time import OTelSpanOutputEndTime
+from .o_tel_span_output_spans_value import OTelSpanOutputSpansValue
+from .o_tel_span_output_start_time import OTelSpanOutputStartTime
+from .o_tel_spans_tree import OTelSpansTree
+from .o_tel_spans_tree_input_spans_value import OTelSpansTreeInputSpansValue
+from .o_tel_spans_tree_output_spans_value import OTelSpansTreeOutputSpansValue
 from .o_tel_status_code import OTelStatusCode
+from .o_tel_tracing_data_response import OTelTracingDataResponse
+from .o_tel_tracing_request import OTelTracingRequest
+from .o_tel_tracing_response import OTelTracingResponse
 from .organization import Organization
 from .organization_details import OrganizationDetails
 from .organization_membership_request import OrganizationMembershipRequest
@@ -84,6 +139,7 @@ from .organization_output import OrganizationOutput
 from .organization_request import OrganizationRequest
 from .parent_dto import ParentDto
 from .permission import Permission
+from .plan import Plan
 from .project_membership_request import ProjectMembershipRequest
 from .project_request import ProjectRequest
 from .project_scope import ProjectScope
@@ -100,40 +156,52 @@ from .secret_dto import SecretDto
 from .secret_kind import SecretKind
 from .secret_response_dto import SecretResponseDto
 from .simple_evaluation_output import SimpleEvaluationOutput
-from .span_dto import SpanDto
-from .span_dto_nodes_value import SpanDtoNodesValue
 from .standard_provider_dto import StandardProviderDto
 from .standard_provider_kind import StandardProviderKind
 from .standard_provider_settings_dto import StandardProviderSettingsDto
 from .status_code import StatusCode
 from .status_dto import StatusDto
-from .template import Template
-from .template_image_info import TemplateImageInfo
 from .test_set_output_response import TestSetOutputResponse
 from .test_set_simple_response import TestSetSimpleResponse
+from .testcase_response import TestcaseResponse
+from .testset import Testset
+from .testset_request import TestsetRequest
+from .testset_response import TestsetResponse
+from .testsets_response import TestsetsResponse
 from .time_dto import TimeDto
+from .timestamp import Timestamp
 from .tree_dto import TreeDto
 from .tree_type import TreeType
 from .update_app_output import UpdateAppOutput
-from .uri import Uri
 from .user_request import UserRequest
 from .validation_error import ValidationError
 from .validation_error_loc_item import ValidationErrorLocItem
-from .variant_action import VariantAction
-from .variant_action_enum import VariantActionEnum
+from .workflow_artifact import WorkflowArtifact
+from .workflow_data import WorkflowData
+from .workflow_flags import WorkflowFlags
+from .workflow_request import WorkflowRequest
+from .workflow_response import WorkflowResponse
+from .workflow_revision import WorkflowRevision
+from .workflow_revision_request import WorkflowRevisionRequest
+from .workflow_revision_response import WorkflowRevisionResponse
+from .workflow_revisions_response import WorkflowRevisionsResponse
+from .workflow_variant import WorkflowVariant
+from .workflow_variant_request import WorkflowVariantRequest
+from .workflow_variant_response import WorkflowVariantResponse
+from .workflow_variants_response import WorkflowVariantsResponse
+from .workflows_response import WorkflowsResponse
 from .workspace import Workspace
 from .workspace_member_response import WorkspaceMemberResponse
 from .workspace_membership_request import WorkspaceMembershipRequest
 from .workspace_permission import WorkspacePermission
 from .workspace_request import WorkspaceRequest
 from .workspace_response import WorkspaceResponse
-from .workspace_role_response import WorkspaceRoleResponse
 from .workspace_role import WorkspaceRole
 
 __all__ = [
+    "AccountRequest",
     "AccountResponse",
     "AgentaNodeDto",
-    "AgentaNodeDtoNodesValue",
     "AgentaNodesResponse",
     "AgentaRootDto",
     "AgentaRootsResponse",
@@ -142,10 +210,25 @@ __all__ = [
     "AggregatedResult",
     "AggregatedResultEvaluatorConfig",
     "AnalyticsResponse",
+    "Annotation",
+    "AnnotationCreate",
+    "AnnotationEdit",
+    "AnnotationKind",
+    "AnnotationLink",
+    "AnnotationLinkResponse",
+    "AnnotationQuery",
+    "AnnotationQueryRequest",
+    "AnnotationReference",
+    "AnnotationReferences",
+    "AnnotationResponse",
+    "AnnotationSource",
+    "AnnotationsResponse",
     "App",
     "AppVariantResponse",
     "AppVariantRevision",
+    "Artifact",
     "BaseOutput",
+    "BodyFetchWorkflowRevision",
     "BodyImportTestset",
     "BucketDto",
     "CollectStatusResponse",
@@ -160,7 +243,6 @@ __all__ = [
     "CustomProviderSettingsDto",
     "Data",
     "DeleteEvaluation",
-    "DockerEnvVars",
     "EnvironmentOutput",
     "EnvironmentOutputExtended",
     "EnvironmentRevision",
@@ -174,40 +256,72 @@ __all__ = [
     "EvaluationType",
     "Evaluator",
     "EvaluatorConfig",
+    "EvaluatorFlags",
     "EvaluatorMappingOutputInterface",
     "EvaluatorOutputInterface",
+    "EvaluatorQuery",
+    "EvaluatorQueryRequest",
+    "EvaluatorRequest",
+    "EvaluatorResponse",
+    "EvaluatorsResponse",
     "ExceptionDto",
+    "Focus",
+    "Format",
+    "FullJsonInput",
+    "FullJsonOutput",
     "GetConfigResponse",
-    "HeaderDto",
+    "Header",
     "HttpValidationError",
     "HumanEvaluation",
     "HumanEvaluationScenario",
     "HumanEvaluationScenarioInput",
     "HumanEvaluationScenarioOutput",
-    "Image",
     "InviteRequest",
     "LegacyAnalyticsResponse",
     "LegacyDataPoint",
+    "LegacyEvaluator",
     "LegacyScopeRequest",
     "LegacyScopesResponse",
+    "LegacySubscriptionRequest",
     "LegacyUserRequest",
     "LegacyUserResponse",
     "LifecycleDto",
     "LinkDto",
     "ListApiKeysResponse",
     "LlmRunRateLimit",
+    "MetaRequest",
     "MetricsDto",
     "NewTestset",
     "NodeDto",
     "NodeType",
     "OTelContextDto",
+    "OTelEvent",
     "OTelEventDto",
     "OTelExtraDto",
+    "OTelFlatSpan",
+    "OTelFlatSpanInputEndTime",
+    "OTelFlatSpanInputStartTime",
+    "OTelFlatSpanOutputEndTime",
+    "OTelFlatSpanOutputStartTime",
+    "OTelLink",
     "OTelLinkDto",
+    "OTelLinksResponse",
+    "OTelSpan",
     "OTelSpanDto",
+    "OTelSpanInputEndTime",
+    "OTelSpanInputSpansValue",
+    "OTelSpanInputStartTime",
     "OTelSpanKind",
-    "OTelSpansResponse",
+    "OTelSpanOutputEndTime",
+    "OTelSpanOutputSpansValue",
+    "OTelSpanOutputStartTime",
+    "OTelSpansTree",
+    "OTelSpansTreeInputSpansValue",
+    "OTelSpansTreeOutputSpansValue",
     "OTelStatusCode",
+    "OTelTracingDataResponse",
+    "OTelTracingRequest",
+    "OTelTracingResponse",
     "Organization",
     "OrganizationDetails",
     "OrganizationMembershipRequest",
@@ -215,6 +329,7 @@ __all__ = [
     "OrganizationRequest",
     "ParentDto",
     "Permission",
+    "Plan",
     "ProjectMembershipRequest",
     "ProjectRequest",
     "ProjectScope",
@@ -231,33 +346,45 @@ __all__ = [
     "SecretKind",
     "SecretResponseDto",
     "SimpleEvaluationOutput",
-    "SpanDto",
-    "SpanDtoNodesValue",
     "StandardProviderDto",
     "StandardProviderKind",
     "StandardProviderSettingsDto",
     "StatusCode",
     "StatusDto",
-    "Template",
-    "TemplateImageInfo",
     "TestSetOutputResponse",
     "TestSetSimpleResponse",
+    "TestcaseResponse",
+    "Testset",
+    "TestsetRequest",
+    "TestsetResponse",
+    "TestsetsResponse",
     "TimeDto",
+    "Timestamp",
     "TreeDto",
     "TreeType",
     "UpdateAppOutput",
-    "Uri",
     "UserRequest",
     "ValidationError",
     "ValidationErrorLocItem",
-    "VariantAction",
-    "VariantActionEnum",
+    "WorkflowArtifact",
+    "WorkflowData",
+    "WorkflowFlags",
+    "WorkflowRequest",
+    "WorkflowResponse",
+    "WorkflowRevision",
+    "WorkflowRevisionRequest",
+    "WorkflowRevisionResponse",
+    "WorkflowRevisionsResponse",
+    "WorkflowVariant",
+    "WorkflowVariantRequest",
+    "WorkflowVariantResponse",
+    "WorkflowVariantsResponse",
+    "WorkflowsResponse",
     "Workspace",
     "WorkspaceMemberResponse",
     "WorkspaceMembershipRequest",
     "WorkspacePermission",
     "WorkspaceRequest",
     "WorkspaceResponse",
-    "WorkspaceRoleResponse",
     "WorkspaceRole",
 ]

agenta/client/backend/types/account_request.py

@@ -0,0 +1,24 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .legacy_scope_request import LegacyScopeRequest
+from .legacy_subscription_request import LegacySubscriptionRequest
+from .legacy_user_request import LegacyUserRequest
+
+
+class AccountRequest(UniversalBaseModel):
+    user: typing.Optional[LegacyUserRequest] = None
+    scope: typing.Optional[LegacyScopeRequest] = None
+    subscription: typing.Optional[LegacySubscriptionRequest] = None
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

agenta/client/backend/types/account_response.py

@@ -1,11 +1,11 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-from .legacy_user_response import LegacyUserResponse
-from .legacy_scopes_response import LegacyScopesResponse
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .legacy_scopes_response import LegacyScopesResponse
+from .legacy_user_response import LegacyUserResponse
 
 
 class AccountResponse(UniversalBaseModel):
@@ -13,9 +13,7 @@ class AccountResponse(UniversalBaseModel):
     scopes: typing.Optional[typing.List[LegacyScopesResponse]] = None
 
     if IS_PYDANTIC_V2:
-        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
-            extra="allow", frozen=True
-        )  # type: ignore # Pydantic v2
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
     else:
 
         class Config:

agenta/client/backend/types/agenta_node_dto.py

@@ -1,23 +1,25 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .agenta_node_dto_nodes_value import AgentaNodeDtoNodesValue
+from .exception_dto import ExceptionDto
 from .lifecycle_dto import LifecycleDto
-from .root_dto import RootDto
-from .tree_dto import TreeDto
+from .link_dto import LinkDto
 from .node_dto import NodeDto
+from .o_tel_extra_dto import OTelExtraDto
 from .parent_dto import ParentDto
-from .time_dto import TimeDto
+from .root_dto import RootDto
 from .status_dto import StatusDto
-from .exception_dto import ExceptionDto
-from .link_dto import LinkDto
-from .o_tel_extra_dto import OTelExtraDto
-from .agenta_node_dto_nodes_value import AgentaNodeDtoNodesValue
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
-import pydantic
+from .time_dto import TimeDto
+from .tree_dto import TreeDto
 
 
 class AgentaNodeDto(UniversalBaseModel):
+    trace_id: str
+    span_id: str
     lifecycle: typing.Optional[LifecycleDto] = None
     root: RootDto
     tree: TreeDto
@@ -37,9 +39,7 @@ class AgentaNodeDto(UniversalBaseModel):
     ] = None
 
     if IS_PYDANTIC_V2:
-        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
-            extra="allow", frozen=True
-        )  # type: ignore # Pydantic v2
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
     else:
 
         class Config:

agenta/client/backend/types/agenta_node_dto_nodes_value.py

@@ -1,6 +1,7 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
+
 from .span_dto import SpanDto
 
 AgentaNodeDtoNodesValue = typing.Union[SpanDto, typing.List[SpanDto]]

agenta/client/backend/types/agenta_nodes_response.py

@@ -1,25 +1,31 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-from .span_dto import SpanDto
 import typing
-from .agenta_node_dto import AgentaNodeDto
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import (
+    IS_PYDANTIC_V2,
+    UniversalBaseModel,
+    update_forward_refs,
+)
+from .agenta_node_dto import AgentaNodeDto
 
 
 class AgentaNodesResponse(UniversalBaseModel):
-    nodes: typing.List[AgentaNodeDto]
+    nodes: typing.Optional[typing.List[AgentaNodeDto]] = None
     version: str
     count: typing.Optional[int] = None
 
     if IS_PYDANTIC_V2:
-        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
-            extra="allow", frozen=True
-        )  # type: ignore # Pydantic v2
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
     else:
 
         class Config:
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
+
+
+from .span_dto import SpanDto  # noqa: E402, F401, I001
+
+update_forward_refs(AgentaNodesResponse)

agenta/client/backend/types/agenta_root_dto.py

@@ -1,12 +1,17 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-from .span_dto import SpanDto
-from .root_dto import RootDto
+from __future__ import annotations
+
 import typing
-from .agenta_tree_dto import AgentaTreeDto
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import (
+    IS_PYDANTIC_V2,
+    UniversalBaseModel,
+    update_forward_refs,
+)
+from .agenta_tree_dto import AgentaTreeDto
+from .root_dto import RootDto
 
 
 class AgentaRootDto(UniversalBaseModel):
@@ -14,12 +19,15 @@ class AgentaRootDto(UniversalBaseModel):
     trees: typing.List[AgentaTreeDto]
 
     if IS_PYDANTIC_V2:
-        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
-            extra="allow", frozen=True
-        )  # type: ignore # Pydantic v2
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
     else:
 
         class Config:
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
+
+
+from .span_dto import SpanDto  # noqa: E402, F401, I001
+
+update_forward_refs(AgentaRootDto)

agenta/client/backend/types/agenta_roots_response.py

@@ -1,25 +1,33 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-from .span_dto import SpanDto
+from __future__ import annotations
+
 import typing
-from .agenta_root_dto import AgentaRootDto
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import (
+    IS_PYDANTIC_V2,
+    UniversalBaseModel,
+    update_forward_refs,
+)
+from .agenta_root_dto import AgentaRootDto
 
 
 class AgentaRootsResponse(UniversalBaseModel):
-    roots: typing.List[AgentaRootDto]
+    roots: typing.Optional[typing.List[AgentaRootDto]] = None
     version: str
     count: typing.Optional[int] = None
 
     if IS_PYDANTIC_V2:
-        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
-            extra="allow", frozen=True
-        )  # type: ignore # Pydantic v2
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
     else:
 
         class Config:
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
+
+
+from .span_dto import SpanDto  # noqa: E402, F401, I001
+
+update_forward_refs(AgentaRootsResponse)

agenta/client/backend/types/agenta_tree_dto.py

@@ -1,12 +1,17 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-from .span_dto import SpanDto
-from .tree_dto import TreeDto
+from __future__ import annotations
+
 import typing
-from .agenta_node_dto import AgentaNodeDto
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import (
+    IS_PYDANTIC_V2,
+    UniversalBaseModel,
+    update_forward_refs,
+)
+from .agenta_node_dto import AgentaNodeDto
+from .tree_dto import TreeDto
 
 
 class AgentaTreeDto(UniversalBaseModel):
@@ -14,12 +19,15 @@ class AgentaTreeDto(UniversalBaseModel):
     nodes: typing.List[AgentaNodeDto]
 
     if IS_PYDANTIC_V2:
-        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
-            extra="allow", frozen=True
-        )  # type: ignore # Pydantic v2
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
     else:
 
         class Config:
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
+
+
+from .span_dto import SpanDto  # noqa: E402, F401, I001
+
+update_forward_refs(AgentaTreeDto)