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

agenta/client/backend/types/organization_output.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
+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 OrganizationOutput(pydantic.BaseModel):
+class OrganizationOutput(UniversalBaseModel):
     id: str
     name: 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/outputs.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+Outputs = typing.Union[typing.Dict[str, typing.Optional[typing.Any]], typing.List[str]]

agenta/client/backend/types/permission.py

@@ -1,141 +1,40 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import enum
 import typing
 
-T_Result = typing.TypeVar("T_Result")
-
-
-class Permission(str, enum.Enum):
-    """
-    An enumeration.
-    """
-
-    READ_SYSTEM = "read_system"
-    VIEW_APPLICATION = "view_application"
-    EDIT_APPLICATION = "edit_application"
-    CREATE_APPLICATION = "create_application"
-    DELETE_APPLICATION = "delete_application"
-    CREATE_APP_VARIANT = "create_app_variant"
-    DELETE_APP_VARIANT = "delete_app_variant"
-    MODIFY_VARIANT_CONFIGURATIONS = "modify_variant_configurations"
-    DELETE_APPLICATION_VARIANT = "delete_application_variant"
-    VIEW_APP_ENVIRONMENT_DEPLOYMENT = "view_app_environment_deployment"
-    EDIT_APP_ENVIRONMENT_DEPLOYMENT = "edit_app_environment_deployment"
-    CREATE_APP_ENVIRONMENT_DEPLOYMENT = "create_app_environment_deployment"
-    VIEW_TESTSET = "view_testset"
-    EDIT_TESTSET = "edit_testset"
-    CREATE_TESTSET = "create_testset"
-    DELETE_TESTSET = "delete_testset"
-    VIEW_EVALUATION = "view_evaluation"
-    RUN_EVALUATIONS = "run_evaluations"
-    EDIT_EVALUATION = "edit_evaluation"
-    CREATE_EVALUATION = "create_evaluation"
-    DELETE_EVALUATION = "delete_evaluation"
-    DEPLOY_APPLICATION = "deploy_application"
-    VIEW_WORKSPACE = "view_workspace"
-    EDIT_WORKSPACE = "edit_workspace"
-    CREATE_WORKSPACE = "create_workspace"
-    DELETE_WORKSPACE = "delete_workspace"
-    MODIFY_USER_ROLES = "modify_user_roles"
-    ADD_NEW_USER_TO_WORKSPACE = "add_new_user_to_workspace"
-    EDIT_ORGANIZATION = "edit_organization"
-    DELETE_ORGANIZATION = "delete_organization"
-    ADD_NEW_USER_TO_ORGANIZATION = "add_new_user_to_organization"
-
-    def visit(
-        self,
-        read_system: typing.Callable[[], T_Result],
-        view_application: typing.Callable[[], T_Result],
-        edit_application: typing.Callable[[], T_Result],
-        create_application: typing.Callable[[], T_Result],
-        delete_application: typing.Callable[[], T_Result],
-        create_app_variant: typing.Callable[[], T_Result],
-        delete_app_variant: typing.Callable[[], T_Result],
-        modify_variant_configurations: typing.Callable[[], T_Result],
-        delete_application_variant: typing.Callable[[], T_Result],
-        view_app_environment_deployment: typing.Callable[[], T_Result],
-        edit_app_environment_deployment: typing.Callable[[], T_Result],
-        create_app_environment_deployment: typing.Callable[[], T_Result],
-        view_testset: typing.Callable[[], T_Result],
-        edit_testset: typing.Callable[[], T_Result],
-        create_testset: typing.Callable[[], T_Result],
-        delete_testset: typing.Callable[[], T_Result],
-        view_evaluation: typing.Callable[[], T_Result],
-        run_evaluations: typing.Callable[[], T_Result],
-        edit_evaluation: typing.Callable[[], T_Result],
-        create_evaluation: typing.Callable[[], T_Result],
-        delete_evaluation: typing.Callable[[], T_Result],
-        deploy_application: typing.Callable[[], T_Result],
-        view_workspace: typing.Callable[[], T_Result],
-        edit_workspace: typing.Callable[[], T_Result],
-        create_workspace: typing.Callable[[], T_Result],
-        delete_workspace: typing.Callable[[], T_Result],
-        modify_user_roles: typing.Callable[[], T_Result],
-        add_new_user_to_workspace: typing.Callable[[], T_Result],
-        edit_organization: typing.Callable[[], T_Result],
-        delete_organization: typing.Callable[[], T_Result],
-        add_new_user_to_organization: typing.Callable[[], T_Result],
-    ) -> T_Result:
-        if self is Permission.READ_SYSTEM:
-            return read_system()
-        if self is Permission.VIEW_APPLICATION:
-            return view_application()
-        if self is Permission.EDIT_APPLICATION:
-            return edit_application()
-        if self is Permission.CREATE_APPLICATION:
-            return create_application()
-        if self is Permission.DELETE_APPLICATION:
-            return delete_application()
-        if self is Permission.CREATE_APP_VARIANT:
-            return create_app_variant()
-        if self is Permission.DELETE_APP_VARIANT:
-            return delete_app_variant()
-        if self is Permission.MODIFY_VARIANT_CONFIGURATIONS:
-            return modify_variant_configurations()
-        if self is Permission.DELETE_APPLICATION_VARIANT:
-            return delete_application_variant()
-        if self is Permission.VIEW_APP_ENVIRONMENT_DEPLOYMENT:
-            return view_app_environment_deployment()
-        if self is Permission.EDIT_APP_ENVIRONMENT_DEPLOYMENT:
-            return edit_app_environment_deployment()
-        if self is Permission.CREATE_APP_ENVIRONMENT_DEPLOYMENT:
-            return create_app_environment_deployment()
-        if self is Permission.VIEW_TESTSET:
-            return view_testset()
-        if self is Permission.EDIT_TESTSET:
-            return edit_testset()
-        if self is Permission.CREATE_TESTSET:
-            return create_testset()
-        if self is Permission.DELETE_TESTSET:
-            return delete_testset()
-        if self is Permission.VIEW_EVALUATION:
-            return view_evaluation()
-        if self is Permission.RUN_EVALUATIONS:
-            return run_evaluations()
-        if self is Permission.EDIT_EVALUATION:
-            return edit_evaluation()
-        if self is Permission.CREATE_EVALUATION:
-            return create_evaluation()
-        if self is Permission.DELETE_EVALUATION:
-            return delete_evaluation()
-        if self is Permission.DEPLOY_APPLICATION:
-            return deploy_application()
-        if self is Permission.VIEW_WORKSPACE:
-            return view_workspace()
-        if self is Permission.EDIT_WORKSPACE:
-            return edit_workspace()
-        if self is Permission.CREATE_WORKSPACE:
-            return create_workspace()
-        if self is Permission.DELETE_WORKSPACE:
-            return delete_workspace()
-        if self is Permission.MODIFY_USER_ROLES:
-            return modify_user_roles()
-        if self is Permission.ADD_NEW_USER_TO_WORKSPACE:
-            return add_new_user_to_workspace()
-        if self is Permission.EDIT_ORGANIZATION:
-            return edit_organization()
-        if self is Permission.DELETE_ORGANIZATION:
-            return delete_organization()
-        if self is Permission.ADD_NEW_USER_TO_ORGANIZATION:
-            return add_new_user_to_organization()
+Permission = typing.Union[
+    typing.Literal[
+        "read_system",
+        "view_application",
+        "edit_application",
+        "create_application",
+        "delete_application",
+        "create_app_variant",
+        "delete_app_variant",
+        "modify_variant_configurations",
+        "delete_application_variant",
+        "view_app_environment_deployment",
+        "edit_app_environment_deployment",
+        "create_app_environment_deployment",
+        "view_testset",
+        "edit_testset",
+        "create_testset",
+        "delete_testset",
+        "view_evaluation",
+        "run_evaluations",
+        "edit_evaluation",
+        "create_evaluation",
+        "delete_evaluation",
+        "deploy_application",
+        "view_workspace",
+        "edit_workspace",
+        "create_workspace",
+        "delete_workspace",
+        "modify_user_roles",
+        "add_new_user_to_workspace",
+        "edit_organization",
+        "delete_organization",
+        "add_new_user_to_organization",
+    ],
+    typing.Any,
+]

agenta/client/backend/types/result.py

@@ -1,39 +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 ..core.datetime_utils import serialize_datetime
 from .error import Error
-
-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 Result(pydantic.BaseModel):
+class Result(UniversalBaseModel):
     type: str
-    value: typing.Optional[typing.Any]
-    error: typing.Optional[Error]
-
-    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/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,56 +1,41 @@
 # This file was auto-generated by Fern from our API Definition.
 
 from __future__ import annotations
-
-import datetime as dt
+from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-
-from ..core.datetime_utils import serialize_datetime
-from .span_kind import SpanKind
-from .span_status_code import SpanStatusCode
+import datetime as dt
 from .span_variant import SpanVariant
-
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
+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(pydantic.BaseModel):
+class Span(UniversalBaseModel):
     id: str
     name: str
-    parent_span_id: typing.Optional[str]
+    parent_span_id: typing.Optional[str] = None
     created_at: dt.datetime
     variant: SpanVariant
-    environment: typing.Optional[str]
+    environment: typing.Optional[str] = None
     spankind: str
     status: SpanStatusCode
-    metadata: typing.Dict[str, typing.Any]
+    metadata: typing.Dict[str, typing.Optional[typing.Any]]
     trace_id: str
-    user_id: typing.Optional[str]
-    content: typing.Dict[str, typing.Any]
-    children: typing.Optional[typing.List[Span]]
-
-    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)
+    user_id: typing.Optional[str] = None
+    content: typing.Dict[str, typing.Optional[typing.Any]]
+    children: typing.Optional[typing.List["Span"]] = None
 
-    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
 
 
-Span.update_forward_refs()
+update_forward_refs(Span)

agenta/client/backend/types/span_detail.py

@@ -1,52 +1,38 @@
 # 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 .span import Span
-from .span_kind import SpanKind
-from .span_status_code import SpanStatusCode
+import datetime as dt
 from .span_variant import SpanVariant
-
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
+from .span_status_code import SpanStatusCode
+from .span import Span
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
 
 
-class SpanDetail(pydantic.BaseModel):
+class SpanDetail(UniversalBaseModel):
     id: str
     name: str
-    parent_span_id: typing.Optional[str]
+    parent_span_id: typing.Optional[str] = None
     created_at: dt.datetime
     variant: SpanVariant
-    environment: typing.Optional[str]
+    environment: typing.Optional[str] = None
     spankind: str
     status: SpanStatusCode
-    metadata: typing.Dict[str, typing.Any]
-    user_id: typing.Optional[str]
-    children: typing.Optional[typing.List[Span]]
-    content: typing.Dict[str, typing.Any]
-    config: typing.Optional[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}
+    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

agenta/client/backend/types/span_status_code.py

@@ -1,29 +1,5 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import enum
 import typing
 
-T_Result = typing.TypeVar("T_Result")
-
-
-class SpanStatusCode(str, enum.Enum):
-    """
-    An enumeration.
-    """
-
-    UNSET = "UNSET"
-    OK = "OK"
-    ERROR = "ERROR"
-
-    def visit(
-        self,
-        unset: typing.Callable[[], T_Result],
-        ok: typing.Callable[[], T_Result],
-        error: typing.Callable[[], T_Result],
-    ) -> T_Result:
-        if self is SpanStatusCode.UNSET:
-            return unset()
-        if self is SpanStatusCode.OK:
-            return ok()
-        if self is SpanStatusCode.ERROR:
-            return error()
+SpanStatusCode = typing.Union[typing.Literal["UNSET", "OK", "ERROR"], typing.Any]

agenta/client/backend/types/span_variant.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 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 SpanVariant(pydantic.BaseModel):
-    variant_id: str
-    variant_name: str
-    revision: int
-
-    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}
+        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