agenta 0.27.7a0__py3-none-any.whl → 0.27.7a2__py3-none-any.whl

agenta/client/backend/types/__init__.py

@@ -1,5 +1,12 @@
 # This file was auto-generated by Fern from our API Definition.
 
+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
+from .agenta_tree_dto import AgentaTreeDto
+from .agenta_trees_response import AgentaTreesResponse
 from .aggregated_result import AggregatedResult
 from .aggregated_result_evaluator_config import AggregatedResultEvaluatorConfig
 from .app import App
@@ -7,6 +14,7 @@ from .app_variant_response import AppVariantResponse
 from .app_variant_revision import AppVariantRevision
 from .base_output import BaseOutput
 from .body_import_testset import BodyImportTestset
+from .collect_status_response import CollectStatusResponse
 from .config_db import ConfigDb
 from .config_dto import ConfigDto
 from .config_response_model import ConfigResponseModel
@@ -31,6 +39,7 @@ from .evaluator import Evaluator
 from .evaluator_config import EvaluatorConfig
 from .evaluator_mapping_output_interface import EvaluatorMappingOutputInterface
 from .evaluator_output_interface import EvaluatorOutputInterface
+from .exception_dto import ExceptionDto
 from .get_config_response import GetConfigResponse
 from .http_validation_error import HttpValidationError
 from .human_evaluation import HumanEvaluation
@@ -42,30 +51,50 @@ from .human_evaluation_update import HumanEvaluationUpdate
 from .image import Image
 from .invite_request import InviteRequest
 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 .llm_tokens import LlmTokens
 from .lm_providers_enum import LmProvidersEnum
 from .new_human_evaluation import NewHumanEvaluation
 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_dto import OTelEventDto
+from .o_tel_extra_dto import OTelExtraDto
+from .o_tel_link_dto import OTelLinkDto
+from .o_tel_span_dto import OTelSpanDto
+from .o_tel_span_kind import OTelSpanKind
+from .o_tel_spans_response import OTelSpansResponse
+from .o_tel_status_code import OTelStatusCode
 from .organization import Organization
 from .organization_output import OrganizationOutput
 from .outputs import Outputs
+from .parent_dto import ParentDto
 from .permission import Permission
 from .reference_dto import ReferenceDto
 from .reference_request_model import ReferenceRequestModel
 from .result import Result
+from .root_dto import RootDto
 from .score import Score
 from .simple_evaluation_output import SimpleEvaluationOutput
 from .span import Span
 from .span_detail import SpanDetail
+from .span_dto import SpanDto
+from .span_dto_nodes_value import SpanDtoNodesValue
 from .span_status_code import SpanStatusCode
 from .span_variant import SpanVariant
+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 .time_dto import TimeDto
 from .trace_detail import TraceDetail
+from .tree_dto import TreeDto
+from .tree_type import TreeType
 from .update_app_output import UpdateAppOutput
 from .uri import Uri
 from .validation_error import ValidationError
@@ -80,6 +109,13 @@ from .workspace_role import WorkspaceRole
 from .workspace_role_response import WorkspaceRoleResponse
 
 __all__ = [
+    "AgentaNodeDto",
+    "AgentaNodeDtoNodesValue",
+    "AgentaNodesResponse",
+    "AgentaRootDto",
+    "AgentaRootsResponse",
+    "AgentaTreeDto",
+    "AgentaTreesResponse",
     "AggregatedResult",
     "AggregatedResultEvaluatorConfig",
     "App",
@@ -87,6 +123,7 @@ __all__ = [
     "AppVariantRevision",
     "BaseOutput",
     "BodyImportTestset",
+    "CollectStatusResponse",
     "ConfigDb",
     "ConfigDto",
     "ConfigResponseModel",
@@ -111,6 +148,7 @@ __all__ = [
     "EvaluatorConfig",
     "EvaluatorMappingOutputInterface",
     "EvaluatorOutputInterface",
+    "ExceptionDto",
     "GetConfigResponse",
     "HttpValidationError",
     "HumanEvaluation",
@@ -122,30 +160,50 @@ __all__ = [
     "Image",
     "InviteRequest",
     "LifecycleDto",
+    "LinkDto",
     "ListApiKeysResponse",
     "LlmRunRateLimit",
     "LlmTokens",
     "LmProvidersEnum",
     "NewHumanEvaluation",
     "NewTestset",
+    "NodeDto",
+    "NodeType",
+    "OTelContextDto",
+    "OTelEventDto",
+    "OTelExtraDto",
+    "OTelLinkDto",
+    "OTelSpanDto",
+    "OTelSpanKind",
+    "OTelSpansResponse",
+    "OTelStatusCode",
     "Organization",
     "OrganizationOutput",
     "Outputs",
+    "ParentDto",
     "Permission",
     "ReferenceDto",
     "ReferenceRequestModel",
     "Result",
+    "RootDto",
     "Score",
     "SimpleEvaluationOutput",
     "Span",
     "SpanDetail",
+    "SpanDto",
+    "SpanDtoNodesValue",
     "SpanStatusCode",
     "SpanVariant",
+    "StatusCode",
+    "StatusDto",
     "Template",
     "TemplateImageInfo",
     "TestSetOutputResponse",
     "TestSetSimpleResponse",
+    "TimeDto",
     "TraceDetail",
+    "TreeDto",
+    "TreeType",
     "UpdateAppOutput",
     "Uri",
     "ValidationError",

agenta/client/backend/types/agenta_node_dto.py

@@ -0,0 +1,48 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+from .lifecycle_dto import LifecycleDto
+from .root_dto import RootDto
+from .tree_dto import TreeDto
+from .node_dto import NodeDto
+from .parent_dto import ParentDto
+from .time_dto import TimeDto
+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
+
+
+class AgentaNodeDto(UniversalBaseModel):
+    lifecycle: typing.Optional[LifecycleDto] = None
+    root: RootDto
+    tree: TreeDto
+    node: NodeDto
+    parent: typing.Optional[ParentDto] = None
+    time: TimeDto
+    status: StatusDto
+    exception: typing.Optional[ExceptionDto] = None
+    data: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = None
+    metrics: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = None
+    meta: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = None
+    refs: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = None
+    links: typing.Optional[typing.List[LinkDto]] = None
+    otel: typing.Optional[OTelExtraDto] = None
+    nodes: typing.Optional[
+        typing.Dict[str, typing.Optional[AgentaNodeDtoNodesValue]]
+    ] = 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/agenta_node_dto_nodes_value.py

@@ -0,0 +1,6 @@
+# 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

@@ -0,0 +1,30 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from __future__ import annotations
+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 update_forward_refs
+
+
+class AgentaNodesResponse(UniversalBaseModel):
+    nodes: typing.List[AgentaNodeDto]
+    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
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow
+
+
+update_forward_refs(SpanDto, AgentaNodesResponse=AgentaNodesResponse)

agenta/client/backend/types/agenta_root_dto.py

@@ -0,0 +1,30 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from __future__ import annotations
+from ..core.pydantic_utilities import UniversalBaseModel
+from .span_dto import SpanDto
+from .root_dto import RootDto
+import typing
+from .agenta_tree_dto import AgentaTreeDto
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+from ..core.pydantic_utilities import update_forward_refs
+
+
+class AgentaRootDto(UniversalBaseModel):
+    root: RootDto
+    trees: typing.List[AgentaTreeDto]
+
+    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
+
+
+update_forward_refs(SpanDto, AgentaRootDto=AgentaRootDto)

agenta/client/backend/types/agenta_roots_response.py

@@ -0,0 +1,30 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from __future__ import annotations
+from ..core.pydantic_utilities import UniversalBaseModel
+from .span_dto import SpanDto
+import typing
+from .agenta_root_dto import AgentaRootDto
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+from ..core.pydantic_utilities import update_forward_refs
+
+
+class AgentaRootsResponse(UniversalBaseModel):
+    roots: typing.List[AgentaRootDto]
+    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
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow
+
+
+update_forward_refs(SpanDto, AgentaRootsResponse=AgentaRootsResponse)

agenta/client/backend/types/agenta_tree_dto.py

@@ -0,0 +1,30 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from __future__ import annotations
+from ..core.pydantic_utilities import UniversalBaseModel
+from .span_dto import SpanDto
+from .tree_dto import TreeDto
+import typing
+from .agenta_node_dto import AgentaNodeDto
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+from ..core.pydantic_utilities import update_forward_refs
+
+
+class AgentaTreeDto(UniversalBaseModel):
+    tree: TreeDto
+    nodes: typing.List[AgentaNodeDto]
+
+    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
+
+
+update_forward_refs(SpanDto, AgentaTreeDto=AgentaTreeDto)

agenta/client/backend/types/agenta_trees_response.py

@@ -0,0 +1,30 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from __future__ import annotations
+from ..core.pydantic_utilities import UniversalBaseModel
+from .span_dto import SpanDto
+import typing
+from .agenta_tree_dto import AgentaTreeDto
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+from ..core.pydantic_utilities import update_forward_refs
+
+
+class AgentaTreesResponse(UniversalBaseModel):
+    trees: typing.List[AgentaTreeDto]
+    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
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow
+
+
+update_forward_refs(SpanDto, AgentaTreesResponse=AgentaTreesResponse)

agenta/client/backend/types/collect_status_response.py

@@ -0,0 +1,22 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import typing
+import pydantic
+
+
+class CollectStatusResponse(UniversalBaseModel):
+    version: str
+    status: str
+
+    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/exception_dto.py

@@ -0,0 +1,26 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import datetime as dt
+import typing
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class ExceptionDto(UniversalBaseModel):
+    timestamp: dt.datetime
+    type: str
+    message: typing.Optional[str] = None
+    stacktrace: typing.Optional[str] = None
+    attributes: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = 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/link_dto.py

@@ -0,0 +1,24 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+from .tree_type import TreeType
+import typing
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class LinkDto(UniversalBaseModel):
+    type: TreeType = "invocation"
+    id: str
+    tree_id: typing.Optional[str] = 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/node_dto.py

@@ -0,0 +1,24 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+from .node_type import NodeType
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class NodeDto(UniversalBaseModel):
+    id: str
+    name: str
+    type: typing.Optional[NodeType] = 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/node_type.py

@@ -0,0 +1,19 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+NodeType = typing.Union[
+    typing.Literal[
+        "agent",
+        "workflow",
+        "chain",
+        "task",
+        "tool",
+        "embedding",
+        "query",
+        "completion",
+        "chat",
+        "rerank",
+    ],
+    typing.Any,
+]

agenta/client/backend/types/o_tel_context_dto.py

@@ -0,0 +1,22 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import typing
+import pydantic
+
+
+class OTelContextDto(UniversalBaseModel):
+    trace_id: str
+    span_id: str
+
+    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/o_tel_event_dto.py

@@ -0,0 +1,23 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class OTelEventDto(UniversalBaseModel):
+    name: str
+    timestamp: str
+    attributes: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = 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/o_tel_extra_dto.py

@@ -0,0 +1,26 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+from .o_tel_event_dto import OTelEventDto
+from .o_tel_link_dto import OTelLinkDto
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class OTelExtraDto(UniversalBaseModel):
+    kind: typing.Optional[str] = None
+    attributes: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = None
+    events: typing.Optional[typing.List[OTelEventDto]] = None
+    links: typing.Optional[typing.List[OTelLinkDto]] = 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/o_tel_link_dto.py

@@ -0,0 +1,23 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+from .o_tel_context_dto import OTelContextDto
+import typing
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class OTelLinkDto(UniversalBaseModel):
+    context: OTelContextDto
+    attributes: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = 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/o_tel_span_dto.py

@@ -0,0 +1,37 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+from .o_tel_context_dto import OTelContextDto
+import typing
+from .o_tel_span_kind import OTelSpanKind
+import datetime as dt
+from .o_tel_status_code import OTelStatusCode
+from .o_tel_event_dto import OTelEventDto
+from .o_tel_link_dto import OTelLinkDto
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class OTelSpanDto(UniversalBaseModel):
+    context: OTelContextDto
+    name: str
+    kind: typing.Optional[OTelSpanKind] = None
+    start_time: dt.datetime
+    end_time: dt.datetime
+    status_code: typing.Optional[OTelStatusCode] = None
+    status_message: typing.Optional[str] = None
+    attributes: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = None
+    events: typing.Optional[typing.List[OTelEventDto]] = None
+    parent: typing.Optional[OTelContextDto] = None
+    links: typing.Optional[typing.List[OTelLinkDto]] = 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/o_tel_span_kind.py

@@ -0,0 +1,15 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+OTelSpanKind = typing.Union[
+    typing.Literal[
+        "SPAN_KIND_UNSPECIFIED",
+        "SPAN_KIND_INTERNAL",
+        "SPAN_KIND_SERVER",
+        "SPAN_KIND_CLIENT",
+        "SPAN_KIND_PRODUCER",
+        "SPAN_KIND_CONSUMER",
+    ],
+    typing.Any,
+]

agenta/client/backend/types/o_tel_spans_response.py

@@ -0,0 +1,24 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+from .o_tel_span_dto import OTelSpanDto
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class OTelSpansResponse(UniversalBaseModel):
+    version: str
+    count: typing.Optional[int] = None
+    spans: typing.List[OTelSpanDto]
+
+    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/o_tel_status_code.py

@@ -0,0 +1,8 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+OTelStatusCode = typing.Union[
+    typing.Literal["STATUS_CODE_OK", "STATUS_CODE_ERROR", "STATUS_CODE_UNSET"],
+    typing.Any,
+]