agenta 0.30.0__py3-none-any.whl → 0.30.0a2__py3-none-any.whl

agenta/client/backend/__init__.py

@@ -10,13 +10,11 @@ from .types import (
     AgentaTreesResponse,
     AggregatedResult,
     AggregatedResultEvaluatorConfig,
-    AnalyticsResponse,
     App,
     AppVariantResponse,
     AppVariantRevision,
     BaseOutput,
     BodyImportTestset,
-    BucketDto,
     CollectStatusResponse,
     ConfigDb,
     ConfigDto,
@@ -44,7 +42,6 @@ from .types import (
     EvaluatorOutputInterface,
     ExceptionDto,
     GetConfigResponse,
-    HeaderDto,
     HttpValidationError,
     HumanEvaluation,
     HumanEvaluationScenario,
@@ -54,14 +51,12 @@ from .types import (
     HumanEvaluationUpdate,
     Image,
     InviteRequest,
-    LegacyAnalyticsResponse,
-    LegacyDataPoint,
     LifecycleDto,
     LinkDto,
     ListApiKeysResponse,
     LlmRunRateLimit,
     LlmTokens,
-    MetricsDto,
+    LmProvidersEnum,
     NewHumanEvaluation,
     NewTestset,
     NodeDto,
@@ -79,17 +74,11 @@ from .types import (
     Outputs,
     ParentDto,
     Permission,
-    ProjectsResponse,
-    ProviderKeyDto,
-    ProviderKind,
     ReferenceDto,
     ReferenceRequestModel,
     Result,
     RootDto,
     Score,
-    SecretDto,
-    SecretKind,
-    SecretResponseDto,
     SimpleEvaluationOutput,
     Span,
     SpanDetail,
@@ -122,7 +111,6 @@ from .types import (
 )
 from .errors import UnprocessableEntityError
 from . import (
-    access_control,
     apps,
     bases,
     configs,
@@ -132,14 +120,12 @@ from . import (
     evaluators,
     observability,
     observability_v_1,
-    scopes,
     testsets,
     variants,
-    vault,
 )
 from .client import AgentaApi, AsyncAgentaApi
 from .containers import ContainerTemplatesResponse
-from .observability_v_1 import Format, QueryAnalyticsResponse, QueryTracesResponse
+from .observability_v_1 import Format, QueryTracesResponse
 from .variants import AddVariantFromBaseAndConfigResponse
 
 __all__ = [
@@ -154,14 +140,12 @@ __all__ = [
     "AgentaTreesResponse",
     "AggregatedResult",
     "AggregatedResultEvaluatorConfig",
-    "AnalyticsResponse",
     "App",
     "AppVariantResponse",
     "AppVariantRevision",
     "AsyncAgentaApi",
     "BaseOutput",
     "BodyImportTestset",
-    "BucketDto",
     "CollectStatusResponse",
     "ConfigDb",
     "ConfigDto",
@@ -191,7 +175,6 @@ __all__ = [
     "ExceptionDto",
     "Format",
     "GetConfigResponse",
-    "HeaderDto",
     "HttpValidationError",
     "HumanEvaluation",
     "HumanEvaluationScenario",
@@ -201,14 +184,12 @@ __all__ = [
     "HumanEvaluationUpdate",
     "Image",
     "InviteRequest",
-    "LegacyAnalyticsResponse",
-    "LegacyDataPoint",
     "LifecycleDto",
     "LinkDto",
     "ListApiKeysResponse",
     "LlmRunRateLimit",
     "LlmTokens",
-    "MetricsDto",
+    "LmProvidersEnum",
     "NewHumanEvaluation",
     "NewTestset",
     "NodeDto",
@@ -226,19 +207,12 @@ __all__ = [
     "Outputs",
     "ParentDto",
     "Permission",
-    "ProjectsResponse",
-    "ProviderKeyDto",
-    "ProviderKind",
-    "QueryAnalyticsResponse",
     "QueryTracesResponse",
     "ReferenceDto",
     "ReferenceRequestModel",
     "Result",
     "RootDto",
     "Score",
-    "SecretDto",
-    "SecretKind",
-    "SecretResponseDto",
     "SimpleEvaluationOutput",
     "Span",
     "SpanDetail",
@@ -269,7 +243,6 @@ __all__ = [
     "WorkspaceResponse",
     "WorkspaceRole",
     "WorkspaceRoleResponse",
-    "access_control",
     "apps",
     "bases",
     "configs",
@@ -279,8 +252,6 @@ __all__ = [
     "evaluators",
     "observability",
     "observability_v_1",
-    "scopes",
     "testsets",
     "variants",
-    "vault",
 ]

agenta/client/backend/apps/client.py

@@ -216,9 +216,7 @@ class AppsClient:
         _response = self._client_wrapper.httpx_client.request(
             "apps",
             method="GET",
-            params={
-                "app_name": app_name,
-            },
+            params={"app_name": app_name},
             request_options=request_options,
         )
         try:
@@ -250,15 +248,13 @@ class AppsClient:
         *,
         app_name: str,
         project_id: typing.Optional[str] = OMIT,
-        workspace_id: typing.Optional[str] = OMIT,
-        organization_id: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> CreateAppOutput:
         """
-        Create a new app for a user or organization.
+        Create a new app for a user.
 
         Args:
-        payload (CreateApp): The payload containing the app name and organization ID (optional).
+        payload (CreateApp): The payload containing the app name.
         stoken_session (SessionContainer): The session container containing the user's session token.
 
         Returns:
@@ -273,10 +269,6 @@ class AppsClient:
 
         project_id : typing.Optional[str]
 
-        workspace_id : typing.Optional[str]
-
-        organization_id : typing.Optional[str]
-
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -303,11 +295,6 @@ class AppsClient:
             json={
                 "app_name": app_name,
                 "project_id": project_id,
-                "workspace_id": workspace_id,
-                "organization_id": organization_id,
-            },
-            headers={
-                "content-type": "application/json",
             },
             request_options=request_options,
             omit=OMIT,
@@ -406,7 +393,7 @@ class AppsClient:
         request_options: typing.Optional[RequestOptions] = None,
     ) -> UpdateAppOutput:
         """
-        Update an app for a user or organization.
+        Update an app for a user.
 
         Args:
         app_id (str): The ID of the app.
@@ -452,9 +439,6 @@ class AppsClient:
             json={
                 "app_name": app_name,
             },
-            headers={
-                "content-type": "application/json",
-            },
             request_options=request_options,
             omit=OMIT,
         )
@@ -554,9 +538,6 @@ class AppsClient:
                 "base_name": base_name,
                 "config_name": config_name,
             },
-            headers={
-                "content-type": "application/json",
-            },
             request_options=request_options,
             omit=OMIT,
         )
@@ -591,8 +572,6 @@ class AppsClient:
         template_id: str,
         env_vars: typing.Dict[str, str],
         project_id: typing.Optional[str] = OMIT,
-        workspace_id: typing.Optional[str] = OMIT,
-        organization_id: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> AppVariantResponse:
         """
@@ -618,10 +597,6 @@ class AppsClient:
 
         project_id : typing.Optional[str]
 
-        workspace_id : typing.Optional[str]
-
-        organization_id : typing.Optional[str]
-
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -651,12 +626,7 @@ class AppsClient:
                 "app_name": app_name,
                 "template_id": template_id,
                 "project_id": project_id,
-                "workspace_id": workspace_id,
                 "env_vars": env_vars,
-                "organization_id": organization_id,
-            },
-            headers={
-                "content-type": "application/json",
             },
             request_options=request_options,
             omit=OMIT,
@@ -782,7 +752,7 @@ class AppsClient:
             base_url="https://yourhost.com/path/to/api",
         )
         client.apps.environment_revisions(
-            app_id="app_id",
+            app_id="string",
             environment_name={"key": "value"},
         )
         """
@@ -1035,9 +1005,7 @@ class AsyncAppsClient:
         _response = await self._client_wrapper.httpx_client.request(
             "apps",
             method="GET",
-            params={
-                "app_name": app_name,
-            },
+            params={"app_name": app_name},
             request_options=request_options,
         )
         try:
@@ -1069,15 +1037,13 @@ class AsyncAppsClient:
         *,
         app_name: str,
         project_id: typing.Optional[str] = OMIT,
-        workspace_id: typing.Optional[str] = OMIT,
-        organization_id: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> CreateAppOutput:
         """
-        Create a new app for a user or organization.
+        Create a new app for a user.
 
         Args:
-        payload (CreateApp): The payload containing the app name and organization ID (optional).
+        payload (CreateApp): The payload containing the app name.
         stoken_session (SessionContainer): The session container containing the user's session token.
 
         Returns:
@@ -1092,10 +1058,6 @@ class AsyncAppsClient:
 
         project_id : typing.Optional[str]
 
-        workspace_id : typing.Optional[str]
-
-        organization_id : typing.Optional[str]
-
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -1130,11 +1092,6 @@ class AsyncAppsClient:
             json={
                 "app_name": app_name,
                 "project_id": project_id,
-                "workspace_id": workspace_id,
-                "organization_id": organization_id,
-            },
-            headers={
-                "content-type": "application/json",
             },
             request_options=request_options,
             omit=OMIT,
@@ -1241,7 +1198,7 @@ class AsyncAppsClient:
         request_options: typing.Optional[RequestOptions] = None,
     ) -> UpdateAppOutput:
         """
-        Update an app for a user or organization.
+        Update an app for a user.
 
         Args:
         app_id (str): The ID of the app.
@@ -1295,9 +1252,6 @@ class AsyncAppsClient:
             json={
                 "app_name": app_name,
             },
-            headers={
-                "content-type": "application/json",
-            },
             request_options=request_options,
             omit=OMIT,
         )
@@ -1405,9 +1359,6 @@ class AsyncAppsClient:
                 "base_name": base_name,
                 "config_name": config_name,
             },
-            headers={
-                "content-type": "application/json",
-            },
             request_options=request_options,
             omit=OMIT,
         )
@@ -1442,8 +1393,6 @@ class AsyncAppsClient:
         template_id: str,
         env_vars: typing.Dict[str, str],
         project_id: typing.Optional[str] = OMIT,
-        workspace_id: typing.Optional[str] = OMIT,
-        organization_id: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> AppVariantResponse:
         """
@@ -1469,10 +1418,6 @@ class AsyncAppsClient:
 
         project_id : typing.Optional[str]
 
-        workspace_id : typing.Optional[str]
-
-        organization_id : typing.Optional[str]
-
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -1510,12 +1455,7 @@ class AsyncAppsClient:
                 "app_name": app_name,
                 "template_id": template_id,
                 "project_id": project_id,
-                "workspace_id": workspace_id,
                 "env_vars": env_vars,
-                "organization_id": organization_id,
-            },
-            headers={
-                "content-type": "application/json",
             },
             request_options=request_options,
             omit=OMIT,
@@ -1654,7 +1594,7 @@ class AsyncAppsClient:
 
         async def main() -> None:
             await client.apps.environment_revisions(
-                app_id="app_id",
+                app_id="string",
                 environment_name={"key": "value"},
             )
 

agenta/client/backend/client.py

@@ -4,9 +4,6 @@ import typing
 import httpx
 from .core.client_wrapper import SyncClientWrapper
 from .observability.client import ObservabilityClient
-from .vault.client import VaultClient
-from .access_control.client import AccessControlClient
-from .scopes.client import ScopesClient
 from .apps.client import AppsClient
 from .variants.client import VariantsClient
 from .evaluations.client import EvaluationsClient
@@ -22,9 +19,9 @@ from .types.list_api_keys_response import ListApiKeysResponse
 from .core.pydantic_utilities import parse_obj_as
 from json.decoder import JSONDecodeError
 from .core.api_error import ApiError
-from .core.jsonable_encoder import jsonable_encoder
 from .errors.unprocessable_entity_error import UnprocessableEntityError
 from .types.http_validation_error import HttpValidationError
+from .core.jsonable_encoder import jsonable_encoder
 from .types.organization import Organization
 from .types.organization_output import OrganizationOutput
 from .types.invite_request import InviteRequest
@@ -35,9 +32,6 @@ from .types.workspace_role_response import WorkspaceRoleResponse
 from .types.permission import Permission
 from .core.client_wrapper import AsyncClientWrapper
 from .observability.client import AsyncObservabilityClient
-from .vault.client import AsyncVaultClient
-from .access_control.client import AsyncAccessControlClient
-from .scopes.client import AsyncScopesClient
 from .apps.client import AsyncAppsClient
 from .variants.client import AsyncVariantsClient
 from .evaluations.client import AsyncEvaluationsClient
@@ -107,9 +101,6 @@ class AgentaApi:
             timeout=_defaulted_timeout,
         )
         self.observability = ObservabilityClient(client_wrapper=self._client_wrapper)
-        self.vault = VaultClient(client_wrapper=self._client_wrapper)
-        self.access_control = AccessControlClient(client_wrapper=self._client_wrapper)
-        self.scopes = ScopesClient(client_wrapper=self._client_wrapper)
         self.apps = AppsClient(client_wrapper=self._client_wrapper)
         self.variants = VariantsClient(client_wrapper=self._client_wrapper)
         self.evaluations = EvaluationsClient(client_wrapper=self._client_wrapper)
@@ -175,7 +166,10 @@ class AgentaApi:
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
     def create_api_key(
-        self, *, request_options: typing.Optional[RequestOptions] = None
+        self,
+        *,
+        workspace_id: str,
+        request_options: typing.Optional[RequestOptions] = None,
     ) -> str:
         """
         Creates an API key for a user.
@@ -188,6 +182,8 @@ class AgentaApi:
 
         Parameters
         ----------
+        workspace_id : str
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -204,11 +200,16 @@ class AgentaApi:
             api_key="YOUR_API_KEY",
             base_url="https://yourhost.com/path/to/api",
         )
-        client.create_api_key()
+        client.create_api_key(
+            workspace_id="workspace_id",
+        )
         """
         _response = self._client_wrapper.httpx_client.request(
             "keys",
             method="POST",
+            params={
+                "workspace_id": workspace_id,
+            },
             request_options=request_options,
         )
         try:
@@ -220,6 +221,16 @@ class AgentaApi:
                         object_=_response.json(),
                     ),
                 )
+            if _response.status_code == 422:
+                raise UnprocessableEntityError(
+                    typing.cast(
+                        HttpValidationError,
+                        parse_obj_as(
+                            type_=HttpValidationError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    )
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -465,9 +476,6 @@ class AgentaApi:
                 "description": description,
                 "type": type,
             },
-            headers={
-                "content-type": "application/json",
-            },
             request_options=request_options,
             omit=OMIT,
         )
@@ -652,9 +660,6 @@ class AgentaApi:
                 "description": description,
                 "updated_at": updated_at,
             },
-            headers={
-                "content-type": "application/json",
-            },
             request_options=request_options,
             omit=OMIT,
         )
@@ -816,9 +821,6 @@ class AgentaApi:
             json={
                 "email": email,
             },
-            headers={
-                "content-type": "application/json",
-            },
             request_options=request_options,
             omit=OMIT,
         )
@@ -908,9 +910,6 @@ class AgentaApi:
             json={
                 "token": token,
             },
-            headers={
-                "content-type": "application/json",
-            },
             request_options=request_options,
             omit=OMIT,
         )
@@ -987,9 +986,6 @@ class AgentaApi:
                 "description": description,
                 "type": type,
             },
-            headers={
-                "content-type": "application/json",
-            },
             request_options=request_options,
             omit=OMIT,
         )
@@ -1069,9 +1065,6 @@ class AgentaApi:
                 "description": description,
                 "updated_at": updated_at,
             },
-            headers={
-                "content-type": "application/json",
-            },
             request_options=request_options,
             omit=OMIT,
         )
@@ -1269,9 +1262,6 @@ class AgentaApi:
                 "organization_id": organization_id,
                 "role": role,
             },
-            headers={
-                "content-type": "application/json",
-            },
             request_options=request_options,
             omit=OMIT,
         )
@@ -1642,11 +1632,6 @@ class AsyncAgentaApi:
         self.observability = AsyncObservabilityClient(
             client_wrapper=self._client_wrapper
         )
-        self.vault = AsyncVaultClient(client_wrapper=self._client_wrapper)
-        self.access_control = AsyncAccessControlClient(
-            client_wrapper=self._client_wrapper
-        )
-        self.scopes = AsyncScopesClient(client_wrapper=self._client_wrapper)
         self.apps = AsyncAppsClient(client_wrapper=self._client_wrapper)
         self.variants = AsyncVariantsClient(client_wrapper=self._client_wrapper)
         self.evaluations = AsyncEvaluationsClient(client_wrapper=self._client_wrapper)
@@ -1720,7 +1705,10 @@ class AsyncAgentaApi:
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
     async def create_api_key(
-        self, *, request_options: typing.Optional[RequestOptions] = None
+        self,
+        *,
+        workspace_id: str,
+        request_options: typing.Optional[RequestOptions] = None,
     ) -> str:
         """
         Creates an API key for a user.
@@ -1733,6 +1721,8 @@ class AsyncAgentaApi:
 
         Parameters
         ----------
+        workspace_id : str
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -1754,7 +1744,9 @@ class AsyncAgentaApi:
 
 
         async def main() -> None:
-            await client.create_api_key()
+            await client.create_api_key(
+                workspace_id="workspace_id",
+            )
 
 
         asyncio.run(main())
@@ -1762,6 +1754,9 @@ class AsyncAgentaApi:
         _response = await self._client_wrapper.httpx_client.request(
             "keys",
             method="POST",
+            params={
+                "workspace_id": workspace_id,
+            },
             request_options=request_options,
         )
         try:
@@ -1773,6 +1768,16 @@ class AsyncAgentaApi:
                         object_=_response.json(),
                     ),
                 )
+            if _response.status_code == 422:
+                raise UnprocessableEntityError(
+                    typing.cast(
+                        HttpValidationError,
+                        parse_obj_as(
+                            type_=HttpValidationError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    )
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -2050,9 +2055,6 @@ class AsyncAgentaApi:
                 "description": description,
                 "type": type,
             },
-            headers={
-                "content-type": "application/json",
-            },
             request_options=request_options,
             omit=OMIT,
         )
@@ -2261,9 +2263,6 @@ class AsyncAgentaApi:
                 "description": description,
                 "updated_at": updated_at,
             },
-            headers={
-                "content-type": "application/json",
-            },
             request_options=request_options,
             omit=OMIT,
         )
@@ -2441,9 +2440,6 @@ class AsyncAgentaApi:
             json={
                 "email": email,
             },
-            headers={
-                "content-type": "application/json",
-            },
             request_options=request_options,
             omit=OMIT,
         )
@@ -2541,9 +2537,6 @@ class AsyncAgentaApi:
             json={
                 "token": token,
             },
-            headers={
-                "content-type": "application/json",
-            },
             request_options=request_options,
             omit=OMIT,
         )
@@ -2628,9 +2621,6 @@ class AsyncAgentaApi:
                 "description": description,
                 "type": type,
             },
-            headers={
-                "content-type": "application/json",
-            },
             request_options=request_options,
             omit=OMIT,
         )
@@ -2718,9 +2708,6 @@ class AsyncAgentaApi:
                 "description": description,
                 "updated_at": updated_at,
             },
-            headers={
-                "content-type": "application/json",
-            },
             request_options=request_options,
             omit=OMIT,
         )
@@ -2942,9 +2929,6 @@ class AsyncAgentaApi:
                 "organization_id": organization_id,
                 "role": role,
             },
-            headers={
-                "content-type": "application/json",
-            },
             request_options=request_options,
             omit=OMIT,
         )

agenta/client/backend/configs/client.py

@@ -143,9 +143,6 @@ class ConfigsClient:
                 "parameters": parameters,
                 "overwrite": overwrite,
             },
-            headers={
-                "content-type": "application/json",
-            },
             request_options=request_options,
             omit=OMIT,
         )
@@ -437,9 +434,6 @@ class AsyncConfigsClient:
                 "parameters": parameters,
                 "overwrite": overwrite,
             },
-            headers={
-                "content-type": "application/json",
-            },
             request_options=request_options,
             omit=OMIT,
         )