agenta 0.24.1__py3-none-any.whl → 0.24.2__py3-none-any.whl

agenta/client/backend/types/create_span.py

@@ -1,61 +1,44 @@
 # 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 .outputs import Outputs
+import datetime as dt
 from .llm_tokens import LlmTokens
-from .span_kind import SpanKind
-from .span_status_code import SpanStatusCode
-
-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 CreateSpan(pydantic.BaseModel):
+class CreateSpan(UniversalBaseModel):
     id: str
-    app_id: typing.Optional[str]
-    variant_id: typing.Optional[str]
-    variant_name: typing.Optional[str]
-    inputs: typing.Optional[typing.Dict[str, typing.Any]]
-    internals: typing.Optional[typing.Dict[str, typing.Any]]
-    outputs: typing.Optional[
-        typing.Union[typing.Dict[str, typing.Any], typing.List[str]]
-    ]
-    config: typing.Optional[typing.Dict[str, typing.Any]]
-    environment: typing.Optional[str]
-    tags: typing.Optional[typing.List[str]]
-    token_consumption: typing.Optional[int]
+    app_id: str
+    variant_id: typing.Optional[str] = None
+    variant_name: typing.Optional[str] = None
+    inputs: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = None
+    internals: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = None
+    outputs: typing.Optional[Outputs] = None
+    config: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = None
+    environment: typing.Optional[str] = None
+    tags: typing.Optional[typing.List[str]] = None
+    token_consumption: typing.Optional[int] = None
     name: str
-    parent_span_id: typing.Optional[str]
-    attributes: typing.Optional[typing.Dict[str, typing.Any]]
+    parent_span_id: typing.Optional[str] = None
+    attributes: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = None
     spankind: str
     status: str
-    user: typing.Optional[str]
+    user: typing.Optional[str] = None
     start_time: dt.datetime
-    end_time: typing.Optional[dt.datetime]
-    tokens: typing.Optional[LlmTokens]
-    cost: typing.Optional[float]
-
-    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 = False
-        smart_union = True
-        json_encoders = {dt.datetime: serialize_datetime}
+    end_time: dt.datetime
+    tokens: typing.Optional[LlmTokens] = None
+    cost: typing.Optional[float] = None
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
+            extra="allow", frozen=False
+        )  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = False
+            smart_union = True
+            extra = pydantic.Extra.allow

agenta/client/backend/types/create_trace_response.py

@@ -1,37 +1,22 @@
 # 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.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 CreateTraceResponse(pydantic.BaseModel):
+class CreateTraceResponse(UniversalBaseModel):
     message: str
-    data: typing.Dict[str, 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}
+    data: typing.Dict[str, typing.Optional[typing.Any]]
+
+    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/docker_env_vars.py

@@ -1,36 +1,21 @@
 # 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.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 DockerEnvVars(pydantic.BaseModel):
+class DockerEnvVars(UniversalBaseModel):
     env_vars: typing.Dict[str, 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)
+    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/environment_output.py

@@ -1,43 +1,28 @@
 # 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.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 EnvironmentOutput(pydantic.BaseModel):
+class EnvironmentOutput(UniversalBaseModel):
     name: str
     app_id: str
-    deployed_app_variant_id: typing.Optional[str]
-    deployed_variant_name: typing.Optional[str]
-    deployed_app_variant_revision_id: typing.Optional[str]
-    revision: typing.Optional[int]
-    organization_id: typing.Optional[str]
-    workspace_id: 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}
+    deployed_app_variant_id: typing.Optional[str] = None
+    deployed_variant_name: typing.Optional[str] = None
+    deployed_app_variant_revision_id: typing.Optional[str] = None
+    revision: typing.Optional[int] = None
+    organization_id: typing.Optional[str] = None
+    workspace_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/environment_output_extended.py

@@ -1,45 +1,30 @@
 # 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 .environment_revision import EnvironmentRevision
-
-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 EnvironmentOutputExtended(pydantic.BaseModel):
+class EnvironmentOutputExtended(UniversalBaseModel):
     name: str
     app_id: str
-    deployed_app_variant_id: typing.Optional[str]
-    deployed_variant_name: typing.Optional[str]
-    deployed_app_variant_revision_id: typing.Optional[str]
-    revision: typing.Optional[int]
+    deployed_app_variant_id: typing.Optional[str] = None
+    deployed_variant_name: typing.Optional[str] = None
+    deployed_app_variant_revision_id: typing.Optional[str] = None
+    revision: typing.Optional[int] = None
     revisions: typing.List[EnvironmentRevision]
-    organization_id: typing.Optional[str]
-    workspace_id: 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}
+    organization_id: typing.Optional[str] = None
+    workspace_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/environment_revision.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.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 EnvironmentRevision(pydantic.BaseModel):
+class EnvironmentRevision(UniversalBaseModel):
     id: str
     revision: int
     modified_by: str
-    deployed_app_variant_revision: typing.Optional[str]
-    deployment: typing.Optional[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
-        json_encoders = {dt.datetime: serialize_datetime}
+    deployed_app_variant_revision: typing.Optional[str] = None
+    deployment: typing.Optional[str] = None
+    created_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

agenta/client/backend/types/error.py

@@ -1,37 +1,22 @@
 # 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.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 Error(pydantic.BaseModel):
+class Error(UniversalBaseModel):
     message: str
-    stacktrace: 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}
+    stacktrace: 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/evaluation.py

@@ -1,19 +1,15 @@
 # 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 .aggregated_result import AggregatedResult
 from .result import Result
-
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
+from .aggregated_result import AggregatedResult
+import datetime as dt
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
 
 
-class Evaluation(pydantic.BaseModel):
+class Evaluation(UniversalBaseModel):
     id: str
     app_id: str
     user_id: str
@@ -22,33 +18,23 @@ class Evaluation(pydantic.BaseModel):
     variant_names: typing.List[str]
     variant_revision_ids: typing.List[str]
     revisions: typing.List[str]
-    testset_id: typing.Optional[str]
-    testset_name: typing.Optional[str]
+    testset_id: typing.Optional[str] = None
+    testset_name: typing.Optional[str] = None
     status: Result
     aggregated_results: typing.List[AggregatedResult]
-    average_cost: typing.Optional[Result]
-    total_cost: typing.Optional[Result]
-    average_latency: typing.Optional[Result]
+    average_cost: typing.Optional[Result] = None
+    total_cost: typing.Optional[Result] = None
+    average_latency: typing.Optional[Result] = None
     created_at: dt.datetime
     updated_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)
+    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/evaluation_scenario.py

@@ -1,47 +1,32 @@
 # 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_scenario_input import EvaluationScenarioInput
 from .evaluation_scenario_output import EvaluationScenarioOutput
+from .correct_answer import CorrectAnswer
 from .evaluation_scenario_result import EvaluationScenarioResult
-
-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 EvaluationScenario(pydantic.BaseModel):
-    id: typing.Optional[str]
+class EvaluationScenario(UniversalBaseModel):
+    id: typing.Optional[str] = None
     evaluation_id: str
     inputs: typing.List[EvaluationScenarioInput]
     outputs: typing.List[EvaluationScenarioOutput]
-    evaluation: typing.Optional[str]
-    correct_answer: typing.Optional[str]
-    is_pinned: typing.Optional[bool]
-    note: typing.Optional[str]
+    correct_answers: typing.Optional[typing.List[CorrectAnswer]] = None
+    is_pinned: typing.Optional[bool] = None
+    note: typing.Optional[str] = None
     results: typing.List[EvaluationScenarioResult]
 
-    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/evaluation_scenario_input.py

@@ -1,38 +1,23 @@
 # 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.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 EvaluationScenarioInput(pydantic.BaseModel):
+class EvaluationScenarioInput(UniversalBaseModel):
     name: str
     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.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