agenta 0.12.2__py3-none-any.whl → 0.32.0a1__py3-none-any.whl

agenta/client/backend/types/result.py

@@ -1,37 +1,24 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import datetime as dt
+from ..core.pydantic_utilities import UniversalBaseModel
 import typing
+from .error import Error
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
 
-from ..core.datetime_utils import serialize_datetime
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
-
-
-class Result(pydantic.BaseModel):
+class Result(UniversalBaseModel):
     type: str
-    value: typing.Optional[typing.Any]
-
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {
-            "by_alias": True,
-            "exclude_unset": True,
-            **kwargs,
-        }
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults: typing.Any = {
-            "by_alias": True,
-            "exclude_unset": True,
-            **kwargs,
-        }
-        return super().dict(**kwargs_with_defaults)
-
-    class Config:
-        frozen = True
-        smart_union = True
-        json_encoders = {dt.datetime: serialize_datetime}
+    value: typing.Optional[typing.Optional[typing.Any]] = None
+    error: typing.Optional[Error] = 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/root_dto.py

@@ -0,0 +1,21 @@
+# 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 RootDto(UniversalBaseModel):
+    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/{human_evaluation_scenario_score.py → score.py}

@@ -2,4 +2,4 @@
 
 import typing
 
-HumanEvaluationScenarioScore = typing.Union[str, int]
+Score = typing.Union[str, int]

agenta/client/backend/types/secret_dto.py

@@ -0,0 +1,24 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+from .secret_kind import SecretKind
+from .provider_key_dto import ProviderKeyDto
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import typing
+import pydantic
+
+
+class SecretDto(UniversalBaseModel):
+    kind: SecretKind = "provider_key"
+    data: ProviderKeyDto
+
+    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/{human_evaluation_scenario_update_score.py → secret_kind.py}

@@ -2,4 +2,4 @@
 
 import typing
 
-HumanEvaluationScenarioUpdateScore = typing.Union[str, int]
+SecretKind = typing.Literal["provider_key"]

agenta/client/backend/types/secret_response_dto.py

@@ -0,0 +1,27 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+from .header_dto import HeaderDto
+from .secret_dto import SecretDto
+from .lifecycle_dto import LifecycleDto
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class SecretResponseDto(UniversalBaseModel):
+    header: typing.Optional[HeaderDto] = None
+    secret: SecretDto
+    id: str
+    lifecycle: typing.Optional[LifecycleDto] = 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/simple_evaluation_output.py

@@ -1,41 +1,26 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import datetime as dt
+from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-
-from ..core.datetime_utils import serialize_datetime
 from .evaluation_type import EvaluationType
-
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
 
 
-class SimpleEvaluationOutput(pydantic.BaseModel):
+class SimpleEvaluationOutput(UniversalBaseModel):
     id: str
     variant_ids: typing.List[str]
     app_id: str
     status: str
     evaluation_type: EvaluationType
 
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {
-            "by_alias": True,
-            "exclude_unset": True,
-            **kwargs,
-        }
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults: typing.Any = {
-            "by_alias": True,
-            "exclude_unset": True,
-            **kwargs,
-        }
-        return super().dict(**kwargs_with_defaults)
+    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
-        json_encoders = {dt.datetime: serialize_datetime}
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

agenta/client/backend/types/span.py

@@ -1,52 +1,42 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import datetime as dt
+from __future__ import annotations
+from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-
-from ..core.datetime_utils import serialize_datetime
-
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
-
-
-class Span(pydantic.BaseModel):
-    parent_span_id: typing.Optional[str]
-    meta: typing.Optional[typing.Dict[str, typing.Any]]
-    event_name: str
-    event_type: typing.Optional[str]
-    start_time: dt.datetime
-    duration: typing.Optional[int]
-    status: str
-    end_time: dt.datetime
-    inputs: typing.Optional[typing.List[str]]
-    outputs: typing.Optional[typing.List[str]]
-    prompt_template: typing.Optional[str]
-    tokens_input: typing.Optional[int]
-    tokens_output: typing.Optional[int]
-    token_total: typing.Optional[int]
-    cost: typing.Optional[float]
-    tags: typing.Optional[typing.List[str]]
-    span_id: str
-
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {
-            "by_alias": True,
-            "exclude_unset": True,
-            **kwargs,
-        }
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults: typing.Any = {
-            "by_alias": True,
-            "exclude_unset": True,
-            **kwargs,
-        }
-        return super().dict(**kwargs_with_defaults)
-
-    class Config:
-        frozen = True
-        smart_union = True
-        json_encoders = {dt.datetime: serialize_datetime}
+import datetime as dt
+from .span_variant import SpanVariant
+from .span_status_code import SpanStatusCode
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+from ..core.pydantic_utilities import update_forward_refs
+
+
+class Span(UniversalBaseModel):
+    id: str
+    name: str
+    project_id: typing.Optional[str] = None
+    parent_span_id: typing.Optional[str] = None
+    created_at: dt.datetime
+    variant: SpanVariant
+    environment: typing.Optional[str] = None
+    spankind: str
+    status: SpanStatusCode
+    metadata: typing.Dict[str, typing.Optional[typing.Any]]
+    trace_id: str
+    user_id: typing.Optional[str] = None
+    content: typing.Dict[str, typing.Optional[typing.Any]]
+    children: typing.Optional[typing.List["Span"]] = 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(Span)

agenta/client/backend/types/span_detail.py

@@ -0,0 +1,44 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from __future__ import annotations
+from ..core.pydantic_utilities import UniversalBaseModel
+from .span import Span
+import typing
+import datetime as dt
+from .span_variant import SpanVariant
+from .span_status_code import SpanStatusCode
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+from ..core.pydantic_utilities import update_forward_refs
+
+
+class SpanDetail(UniversalBaseModel):
+    id: str
+    name: str
+    project_id: typing.Optional[str] = None
+    parent_span_id: typing.Optional[str] = None
+    created_at: dt.datetime
+    variant: SpanVariant
+    environment: typing.Optional[str] = None
+    spankind: str
+    status: SpanStatusCode
+    metadata: typing.Dict[str, typing.Optional[typing.Any]]
+    trace_id: str
+    user_id: typing.Optional[str] = None
+    content: typing.Dict[str, typing.Optional[typing.Any]]
+    children: typing.Optional[typing.List[Span]] = None
+    config: 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
+
+
+update_forward_refs(Span, SpanDetail=SpanDetail)

agenta/client/backend/types/span_dto.py

@@ -0,0 +1,54 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from __future__ import annotations
+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 ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+from ..core.pydantic_utilities import update_forward_refs
+
+
+class SpanDto(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["SpanDtoNodesValue"]]
+    ] = 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
+
+
+from .span_dto_nodes_value import SpanDtoNodesValue  # noqa: E402
+
+update_forward_refs(SpanDto)

agenta/client/backend/types/span_dto_nodes_value.py

@@ -0,0 +1,9 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from __future__ import annotations
+import typing
+import typing
+
+if typing.TYPE_CHECKING:
+    from .span_dto import SpanDto
+SpanDtoNodesValue = typing.Union["SpanDto", typing.List["SpanDto"]]

agenta/client/backend/types/span_status_code.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+SpanStatusCode = typing.Union[typing.Literal["UNSET", "OK", "ERROR"], typing.Any]

agenta/client/backend/types/span_variant.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 SpanVariant(UniversalBaseModel):
+    variant_id: typing.Optional[str] = None
+    variant_name: typing.Optional[str] = None
+    revision: 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

agenta/client/backend/types/status_code.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+StatusCode = typing.Union[typing.Literal["UNSET", "OK", "ERROR"], typing.Any]

agenta/client/backend/types/status_dto.py

@@ -0,0 +1,23 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+from .status_code import StatusCode
+import typing
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class StatusDto(UniversalBaseModel):
+    code: StatusCode
+    message: 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/template.py

@@ -1,38 +1,23 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import datetime as dt
-import typing
-
-from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import UniversalBaseModel
 from .template_image_info import TemplateImageInfo
-
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import typing
+import pydantic
 
 
-class Template(pydantic.BaseModel):
+class Template(UniversalBaseModel):
     id: str
     image: TemplateImageInfo
 
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {
-            "by_alias": True,
-            "exclude_unset": True,
-            **kwargs,
-        }
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults: typing.Any = {
-            "by_alias": True,
-            "exclude_unset": True,
-            **kwargs,
-        }
-        return super().dict(**kwargs_with_defaults)
+    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
-        json_encoders = {dt.datetime: serialize_datetime}
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

agenta/client/backend/types/template_image_info.py

@@ -1,43 +1,29 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import datetime as dt
+from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-
-from ..core.datetime_utils import serialize_datetime
-
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
+import datetime as dt
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
 
 
-class TemplateImageInfo(pydantic.BaseModel):
+class TemplateImageInfo(UniversalBaseModel):
     name: str
-    size: typing.Optional[int]
-    digest: typing.Optional[str]
+    size: typing.Optional[int] = None
+    digest: typing.Optional[str] = None
     title: str
     description: str
-    last_pushed: typing.Optional[dt.datetime]
-    repo_name: typing.Optional[str]
-    template_uri: typing.Optional[str]
-
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {
-            "by_alias": True,
-            "exclude_unset": True,
-            **kwargs,
-        }
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults: typing.Any = {
-            "by_alias": True,
-            "exclude_unset": True,
-            **kwargs,
-        }
-        return super().dict(**kwargs_with_defaults)
-
-    class Config:
-        frozen = True
-        smart_union = True
-        json_encoders = {dt.datetime: serialize_datetime}
+    last_pushed: typing.Optional[dt.datetime] = None
+    repo_name: typing.Optional[str] = None
+    template_uri: 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/test_set_output_response.py

@@ -1,39 +1,26 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import datetime as dt
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing_extensions
+from ..core.serialization import FieldMetadata
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import typing
+import pydantic
 
-from ..core.datetime_utils import serialize_datetime
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
-
-
-class TestSetOutputResponse(pydantic.BaseModel):
-    id: str = pydantic.Field(alias="_id")
+class TestSetOutputResponse(UniversalBaseModel):
+    id: typing_extensions.Annotated[str, FieldMetadata(alias="_id")]
     name: str
-    created_at: dt.datetime
-
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {
-            "by_alias": True,
-            "exclude_unset": True,
-            **kwargs,
-        }
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults: typing.Any = {
-            "by_alias": True,
-            "exclude_unset": True,
-            **kwargs,
-        }
-        return super().dict(**kwargs_with_defaults)
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-        json_encoders = {dt.datetime: serialize_datetime}
+    created_at: str
+    updated_at: 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