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

agenta/client/backend/types/metrics_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 ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class MetricsDto(UniversalBaseModel):
+    count: typing.Optional[int] = None
+    duration: typing.Optional[float] = None
+    cost: typing.Optional[float] = None
+    tokens: 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/new_human_evaluation.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 .evaluation_type import EvaluationType
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class NewHumanEvaluation(UniversalBaseModel):
+    app_id: str
+    variant_ids: typing.List[str]
+    evaluation_type: EvaluationType
+    inputs: typing.List[str]
+    testset_id: 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/new_testset.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 NewTestset(pydantic.BaseModel):
+class NewTestset(UniversalBaseModel):
     name: str
-    csvdata: typing.List[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)
-
-    class Config:
-        frozen = True
-        smart_union = True
-        json_encoders = {dt.datetime: serialize_datetime}
+    csvdata: typing.List[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/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,
+]

agenta/client/backend/types/organization.py

@@ -1,42 +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.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 Organization(pydantic.BaseModel):
-    id: typing.Optional[str]
+class Organization(UniversalBaseModel):
+    id: str
     name: str
-    description: typing.Optional[str]
-    type: typing.Optional[str]
+    description: str
+    type: typing.Optional[str] = None
     owner: str
-    members: typing.Optional[typing.List[str]]
-    invitations: typing.Optional[typing.List[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}
+    workspaces: typing.Optional[typing.List[str]] = None
+    members: typing.Optional[typing.List[str]] = None
+    invitations: typing.Optional[typing.List[typing.Optional[typing.Any]]] = None
+    is_paying: typing.Optional[bool] = 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/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/parent_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 ParentDto(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/permission.py

@@ -0,0 +1,41 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+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",
+        "run_service",
+        "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/projects_response.py

@@ -0,0 +1,28 @@
+# 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 ProjectsResponse(UniversalBaseModel):
+    organization_id: typing.Optional[str] = None
+    organization_name: typing.Optional[str] = None
+    workspace_id: typing.Optional[str] = None
+    workspace_name: typing.Optional[str] = None
+    project_id: str
+    project_name: str
+    user_role: typing.Optional[str] = None
+    is_demo: typing.Optional[bool] = 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/provider_key_dto.py

@@ -0,0 +1,23 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+from .provider_kind import ProviderKind
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import typing
+import pydantic
+
+
+class ProviderKeyDto(UniversalBaseModel):
+    provider: ProviderKind
+    key: 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/provider_kind.py

@@ -0,0 +1,21 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+ProviderKind = typing.Union[
+    typing.Literal[
+        "openai",
+        "cohere",
+        "anyscale",
+        "deepinfra",
+        "alephalpha",
+        "groq",
+        "mistral",
+        "anthropic",
+        "perplexityai",
+        "togetherai",
+        "openrouter",
+        "gemini",
+    ],
+    typing.Any,
+]

agenta/client/backend/types/reference_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 ReferenceDto(UniversalBaseModel):
+    slug: typing.Optional[str] = None
+    version: typing.Optional[int] = None
+    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/reference_request_model.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 ReferenceRequestModel(UniversalBaseModel):
+    slug: typing.Optional[str] = None
+    version: typing.Optional[int] = None
+    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