agenta 0.36.4__py3-none-any.whl → 0.37.0__py3-none-any.whl

agenta/client/backend/testsets/__init__.py

@@ -0,0 +1 @@
+# This file was auto-generated by Fern from our API Definition.

agenta/client/{types → backend/types}/__init__.py

@@ -23,6 +23,11 @@ from .config_dto import ConfigDto
 from .config_response_model import ConfigResponseModel
 from .correct_answer import CorrectAnswer
 from .create_app_output import CreateAppOutput
+from .custom_model_settings_dto import CustomModelSettingsDto
+from .custom_provider_dto import CustomProviderDto
+from .custom_provider_kind import CustomProviderKind
+from .custom_provider_settings_dto import CustomProviderSettingsDto
+from .data import Data
 from .delete_evaluation import DeleteEvaluation
 from .docker_env_vars import DockerEnvVars
 from .environment_output import EnvironmentOutput
@@ -73,6 +78,7 @@ 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_details import OrganizationDetails
 from .organization_membership_request import OrganizationMembershipRequest
 from .organization_output import OrganizationOutput
 from .organization_request import OrganizationRequest
@@ -82,8 +88,6 @@ from .project_membership_request import ProjectMembershipRequest
 from .project_request import ProjectRequest
 from .project_scope import ProjectScope
 from .projects_response import ProjectsResponse
-from .provider_key_dto import ProviderKeyDto
-from .provider_kind import ProviderKind
 from .reference import Reference
 from .reference_dto import ReferenceDto
 from .reference_request_model import ReferenceRequestModel
@@ -98,6 +102,9 @@ from .secret_response_dto import SecretResponseDto
 from .simple_evaluation_output import SimpleEvaluationOutput
 from .span_dto import SpanDto
 from .span_dto_nodes_value import SpanDtoNodesValue
+from .standard_provider_dto import StandardProviderDto
+from .standard_provider_kind import StandardProviderKind
+from .standard_provider_settings_dto import StandardProviderSettingsDto
 from .status_code import StatusCode
 from .status_dto import StatusDto
 from .template import Template
@@ -114,13 +121,14 @@ from .validation_error import ValidationError
 from .validation_error_loc_item import ValidationErrorLocItem
 from .variant_action import VariantAction
 from .variant_action_enum import VariantActionEnum
+from .workspace import Workspace
 from .workspace_member_response import WorkspaceMemberResponse
 from .workspace_membership_request import WorkspaceMembershipRequest
 from .workspace_permission import WorkspacePermission
 from .workspace_request import WorkspaceRequest
 from .workspace_response import WorkspaceResponse
-from .workspace_role import WorkspaceRole
 from .workspace_role_response import WorkspaceRoleResponse
+from .workspace_role import WorkspaceRole
 
 __all__ = [
     "AccountResponse",
@@ -146,6 +154,11 @@ __all__ = [
     "ConfigResponseModel",
     "CorrectAnswer",
     "CreateAppOutput",
+    "CustomModelSettingsDto",
+    "CustomProviderDto",
+    "CustomProviderKind",
+    "CustomProviderSettingsDto",
+    "Data",
     "DeleteEvaluation",
     "DockerEnvVars",
     "EnvironmentOutput",
@@ -196,6 +209,7 @@ __all__ = [
     "OTelSpansResponse",
     "OTelStatusCode",
     "Organization",
+    "OrganizationDetails",
     "OrganizationMembershipRequest",
     "OrganizationOutput",
     "OrganizationRequest",
@@ -205,8 +219,6 @@ __all__ = [
     "ProjectRequest",
     "ProjectScope",
     "ProjectsResponse",
-    "ProviderKeyDto",
-    "ProviderKind",
     "Reference",
     "ReferenceDto",
     "ReferenceRequestModel",
@@ -221,6 +233,9 @@ __all__ = [
     "SimpleEvaluationOutput",
     "SpanDto",
     "SpanDtoNodesValue",
+    "StandardProviderDto",
+    "StandardProviderKind",
+    "StandardProviderSettingsDto",
     "StatusCode",
     "StatusDto",
     "Template",
@@ -237,11 +252,12 @@ __all__ = [
     "ValidationErrorLocItem",
     "VariantAction",
     "VariantActionEnum",
+    "Workspace",
     "WorkspaceMemberResponse",
     "WorkspaceMembershipRequest",
     "WorkspacePermission",
     "WorkspaceRequest",
     "WorkspaceResponse",
-    "WorkspaceRole",
     "WorkspaceRoleResponse",
+    "WorkspaceRole",
 ]

agenta/client/{types → backend/types}/agenta_nodes_response.py

@@ -1,13 +1,11 @@
 # 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):
@@ -25,6 +23,3 @@ class AgentaNodesResponse(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
-
-
-update_forward_refs(SpanDto, AgentaNodesResponse=AgentaNodesResponse)

agenta/client/{types → backend/types}/agenta_root_dto.py

@@ -1,6 +1,5 @@
 # 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
@@ -8,7 +7,6 @@ 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):
@@ -25,6 +23,3 @@ class AgentaRootDto(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
-
-
-update_forward_refs(SpanDto, AgentaRootDto=AgentaRootDto)

agenta/client/{types → backend/types}/agenta_roots_response.py

@@ -1,13 +1,11 @@
 # 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):
@@ -25,6 +23,3 @@ class AgentaRootsResponse(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
-
-
-update_forward_refs(SpanDto, AgentaRootsResponse=AgentaRootsResponse)

agenta/client/{types → backend/types}/agenta_tree_dto.py

@@ -1,6 +1,5 @@
 # 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
@@ -8,7 +7,6 @@ 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):
@@ -25,6 +23,3 @@ class AgentaTreeDto(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
-
-
-update_forward_refs(SpanDto, AgentaTreeDto=AgentaTreeDto)

agenta/client/{types → backend/types}/agenta_trees_response.py

@@ -1,13 +1,11 @@
 # 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):
@@ -25,6 +23,3 @@ class AgentaTreesResponse(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
-
-
-update_forward_refs(SpanDto, AgentaTreesResponse=AgentaTreesResponse)

agenta/client/backend/types/custom_model_settings_dto.py

@@ -0,0 +1,22 @@
+# 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 CustomModelSettingsDto(UniversalBaseModel):
+    slug: str
+    extras: 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/custom_provider_dto.py

@@ -0,0 +1,29 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+from .custom_provider_kind import CustomProviderKind
+from .custom_provider_settings_dto import CustomProviderSettingsDto
+import typing
+from .custom_model_settings_dto import CustomModelSettingsDto
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class CustomProviderDto(UniversalBaseModel):
+    kind: CustomProviderKind
+    provider: CustomProviderSettingsDto
+    models: typing.List[CustomModelSettingsDto]
+
+    provider_slug: typing.Optional[str] = None
+    model_keys: typing.Optional[typing.List[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/custom_provider_kind.py

@@ -0,0 +1,26 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+CustomProviderKind = typing.Union[
+    typing.Literal[
+        "openai",
+        "cohere",
+        "anyscale",
+        "deepinfra",
+        "alephalpha",
+        "groq",
+        "mistralai",
+        "anthropic",
+        "perplexityai",
+        "togetherai",
+        "openrouter",
+        "gemini",
+        "custom",
+        "azure",
+        "bedrock",
+        "sagemaker",
+        "vertex",
+    ],
+    typing.Any,
+]

agenta/client/backend/types/custom_provider_settings_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 CustomProviderSettingsDto(UniversalBaseModel):
+    url: typing.Optional[str] = None
+    version: typing.Optional[str] = None
+    key: typing.Optional[str] = None
+    extras: 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/data.py

@@ -0,0 +1,7 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+from .standard_provider_dto import StandardProviderDto
+from .custom_provider_dto import CustomProviderDto
+
+Data = typing.Union[StandardProviderDto, CustomProviderDto]

agenta/client/{types → backend/types}/organization.py

@@ -9,13 +9,10 @@ import pydantic
 class Organization(UniversalBaseModel):
     id: str
     name: str
+    owner: str
     description: str
     type: typing.Optional[str] = None
-    owner: str
     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(

agenta/client/backend/types/organization_details.py

@@ -0,0 +1,29 @@
+# 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 OrganizationDetails(UniversalBaseModel):
+    id: str
+    name: str
+    owner: str
+    description: str
+    type: typing.Optional[str] = None
+    workspaces: typing.Optional[typing.List[str]] = None
+    default_workspace: 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/{types → backend/types}/organization_membership_request.py

@@ -1,14 +1,15 @@
 # This file was auto-generated by Fern from our API Definition.
 
 from ..core.pydantic_utilities import UniversalBaseModel
-import typing
+from .role import Role
 from .reference import Reference
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import typing
 import pydantic
 
 
 class OrganizationMembershipRequest(UniversalBaseModel):
-    role: typing.Literal["owner"] = "owner"
+    role: Role
     is_demo: bool
     user_ref: Reference
     organization_ref: Reference

agenta/client/{types → backend/types}/permission.py

@@ -40,6 +40,7 @@ Permission = typing.Union[
         "edit_organization",
         "delete_organization",
         "add_new_user_to_organization",
+        "reset_password",
     ],
     typing.Any,
 ]

agenta/client/backend/types/provider_key_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 StandardProviderSettingsDto(UniversalBaseModel):
+    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
+
+StandardProviderKind = typing.Union[
+    typing.Literal[
+        "openai",
+        "cohere",
+        "anyscale",
+        "deepinfra",
+        "alephalpha",
+        "groq",
+        "mistralai",
+        "anthropic",
+        "perplexityai",
+        "togetherai",
+        "openrouter",
+        "gemini",
+    ],
+    typing.Any,
+]

agenta/client/backend/types/score.py

@@ -0,0 +1,3 @@
+# This file was auto-generated by Fern from our API Definition.
+
+Score = int

agenta/client/{types → backend/types}/secret_dto.py

@@ -2,15 +2,15 @@
 
 from ..core.pydantic_utilities import UniversalBaseModel
 from .secret_kind import SecretKind
-from .provider_key_dto import ProviderKeyDto
+from .data import Data
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import typing
 import pydantic
 
 
 class SecretDto(UniversalBaseModel):
-    kind: SecretKind = "provider_key"
-    data: ProviderKeyDto
+    kind: SecretKind
+    data: Data
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(

agenta/client/backend/types/secret_kind.py

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

agenta/client/backend/types/standard_provider_dto.py

@@ -0,0 +1,24 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+from .standard_provider_kind import StandardProviderKind
+from .standard_provider_settings_dto import StandardProviderSettingsDto
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import typing
+import pydantic
+
+
+class StandardProviderDto(UniversalBaseModel):
+    kind: StandardProviderKind
+    provider: StandardProviderSettingsDto
+
+    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/standard_provider_kind.py

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

agenta/client/backend/types/standard_provider_settings_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 StandardProviderSettingsDto(UniversalBaseModel):
+    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/workspace.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 Workspace(UniversalBaseModel):
+    id: typing.Optional[str] = None
+    name: str
+    description: typing.Optional[str] = None
+    type: 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/{variants → backend/variants}/client.py

@@ -47,7 +47,6 @@ class VariantsClient:
 
         Args:
             payload (AddVariantFromBasePayload): Payload containing base variant ID, new variant name, and parameters.
-            stoken_session (SessionContainer, optional): Session container. Defaults to result of verify_session().
 
         Raises:
             HTTPException: Raised if the variant could not be added or accessed.
@@ -198,17 +197,16 @@ class VariantsClient:
         """
         Start a variant of an app.
 
-        Args:
-            variant_id (str): The ID of the variant to start.
-            action (VariantAction): The action to perform on the variant (start).
-            env_vars (Optional[DockerEnvVars], optional): The environment variables to inject to the Docker container. Defaults to None.
-            stoken_session (SessionContainer, optional): The session container. Defaults to Depends(verify_session()).
-
-        Returns:
-            URI: The URL of the started variant.
+            Args:
+                variant_id (str): The ID of the variant to start.
+                action (VariantAction): The action to perform on the variant (start).
+                env_vars (Optional[DockerEnvVars], optional): The environment variables to inject to the Docker container. Defaults to None.
+        =
+            Returns:
+                URI: The URL of the started variant.
 
-        Raises:
-            HTTPException: If the app container cannot be started.
+            Raises:
+                HTTPException: If the app container cannot be started.
 
         Parameters
         ----------
@@ -364,7 +362,6 @@ class VariantsClient:
         Args:
             variant_id (str): The ID of the app variant to update.
             payload (UpdateVariantParameterPayload): The payload containing the updated parameters.
-            stoken_session (SessionContainer, optional): The session container. Defaults to Depends(verify_session()).
 
         Raises:
             HTTPException: If there is an error while trying to update the app variant.
@@ -1451,7 +1448,6 @@ class AsyncVariantsClient:
 
         Args:
             payload (AddVariantFromBasePayload): Payload containing base variant ID, new variant name, and parameters.
-            stoken_session (SessionContainer, optional): Session container. Defaults to result of verify_session().
 
         Raises:
             HTTPException: Raised if the variant could not be added or accessed.
@@ -1618,17 +1614,16 @@ class AsyncVariantsClient:
         """
         Start a variant of an app.
 
-        Args:
-            variant_id (str): The ID of the variant to start.
-            action (VariantAction): The action to perform on the variant (start).
-            env_vars (Optional[DockerEnvVars], optional): The environment variables to inject to the Docker container. Defaults to None.
-            stoken_session (SessionContainer, optional): The session container. Defaults to Depends(verify_session()).
-
-        Returns:
-            URI: The URL of the started variant.
+            Args:
+                variant_id (str): The ID of the variant to start.
+                action (VariantAction): The action to perform on the variant (start).
+                env_vars (Optional[DockerEnvVars], optional): The environment variables to inject to the Docker container. Defaults to None.
+        =
+            Returns:
+                URI: The URL of the started variant.
 
-        Raises:
-            HTTPException: If the app container cannot be started.
+            Raises:
+                HTTPException: If the app container cannot be started.
 
         Parameters
         ----------
@@ -1800,7 +1795,6 @@ class AsyncVariantsClient:
         Args:
             variant_id (str): The ID of the app variant to update.
             payload (UpdateVariantParameterPayload): The payload containing the updated parameters.
-            stoken_session (SessionContainer, optional): The session container. Defaults to Depends(verify_session()).
 
         Raises:
             HTTPException: If there is an error while trying to update the app variant.

agenta/client/backend/vault/__init__.py

@@ -0,0 +1 @@
+# This file was auto-generated by Fern from our API Definition.