agenta 0.32.0a2__py3-none-any.whl → 0.33.0__py3-none-any.whl

agenta/client/backend/testsets/client.py

@@ -108,6 +108,7 @@ class TestsetsClient:
     def import_testset(
         self,
         *,
+        authorization: typing.Optional[str] = None,
         endpoint: typing.Optional[str] = OMIT,
         testset_name: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
@@ -124,6 +125,8 @@ class TestsetsClient:
 
         Parameters
         ----------
+        authorization : typing.Optional[str]
+
         endpoint : typing.Optional[str]
 
         testset_name : typing.Optional[str]
@@ -149,6 +152,9 @@ class TestsetsClient:
         _response = self._client_wrapper.httpx_client.request(
             "testsets/endpoint",
             method="POST",
+            params={
+                "authorization": authorization,
+            },
             json={
                 "endpoint": endpoint,
                 "testset_name": testset_name,
@@ -187,11 +193,9 @@ class TestsetsClient:
         Get all testsets.
 
         Returns:
-
         - A list of testset objects.
 
         Raises:
-
         - `HTTPException` with status code 404 if no testsets are found.
 
         Parameters
@@ -386,88 +390,6 @@ class TestsetsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def deprecating_create_testset(
-        self,
-        app_id: str,
-        *,
-        name: str,
-        csvdata: typing.Sequence[typing.Dict[str, typing.Optional[typing.Any]]],
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> TestSetSimpleResponse:
-        """
-        Create a testset with given name, save the testset to Postgres.
-
-        Args:
-        name (str): name of the test set.
-        testset (Dict[str, str]): test set data.
-
-        Returns:
-        str: The id of the test set created.
-
-        Parameters
-        ----------
-        app_id : str
-
-        name : str
-
-        csvdata : typing.Sequence[typing.Dict[str, typing.Optional[typing.Any]]]
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        TestSetSimpleResponse
-            Successful Response
-
-        Examples
-        --------
-        from agenta import AgentaApi
-
-        client = AgentaApi(
-            api_key="YOUR_API_KEY",
-            base_url="https://yourhost.com/path/to/api",
-        )
-        client.testsets.deprecating_create_testset(
-            app_id="app_id",
-            name="name",
-            csvdata=[{"key": "value"}],
-        )
-        """
-        _response = self._client_wrapper.httpx_client.request(
-            f"testsets/{jsonable_encoder(app_id)}",
-            method="POST",
-            json={
-                "name": name,
-                "csvdata": csvdata,
-            },
-            request_options=request_options,
-            omit=OMIT,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    TestSetSimpleResponse,
-                    parse_obj_as(
-                        type_=TestSetSimpleResponse,  # type: ignore
-                        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)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
-
     def get_single_testset(
         self,
         testset_id: str,
@@ -475,10 +397,10 @@ class TestsetsClient:
         request_options: typing.Optional[RequestOptions] = None,
     ) -> typing.Optional[typing.Any]:
         """
-        Fetch a specific testset in a MongoDB collection using its _id.
+        Fetch a specific testset in Postgres.
 
         Args:
-            testset_id (str): The _id of the testset to fetch.
+            testset_id (str): The id of the testset to fetch.
 
         Returns:
             The requested testset if found, else an HTTPException.
@@ -717,6 +639,7 @@ class AsyncTestsetsClient:
     async def import_testset(
         self,
         *,
+        authorization: typing.Optional[str] = None,
         endpoint: typing.Optional[str] = OMIT,
         testset_name: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
@@ -733,6 +656,8 @@ class AsyncTestsetsClient:
 
         Parameters
         ----------
+        authorization : typing.Optional[str]
+
         endpoint : typing.Optional[str]
 
         testset_name : typing.Optional[str]
@@ -766,6 +691,9 @@ class AsyncTestsetsClient:
         _response = await self._client_wrapper.httpx_client.request(
             "testsets/endpoint",
             method="POST",
+            params={
+                "authorization": authorization,
+            },
             json={
                 "endpoint": endpoint,
                 "testset_name": testset_name,
@@ -804,11 +732,9 @@ class AsyncTestsetsClient:
         Get all testsets.
 
         Returns:
-
         - A list of testset objects.
 
         Raises:
-
         - `HTTPException` with status code 404 if no testsets are found.
 
         Parameters
@@ -1027,96 +953,6 @@ class AsyncTestsetsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    async def deprecating_create_testset(
-        self,
-        app_id: str,
-        *,
-        name: str,
-        csvdata: typing.Sequence[typing.Dict[str, typing.Optional[typing.Any]]],
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> TestSetSimpleResponse:
-        """
-        Create a testset with given name, save the testset to Postgres.
-
-        Args:
-        name (str): name of the test set.
-        testset (Dict[str, str]): test set data.
-
-        Returns:
-        str: The id of the test set created.
-
-        Parameters
-        ----------
-        app_id : str
-
-        name : str
-
-        csvdata : typing.Sequence[typing.Dict[str, typing.Optional[typing.Any]]]
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        TestSetSimpleResponse
-            Successful Response
-
-        Examples
-        --------
-        import asyncio
-
-        from agenta import AsyncAgentaApi
-
-        client = AsyncAgentaApi(
-            api_key="YOUR_API_KEY",
-            base_url="https://yourhost.com/path/to/api",
-        )
-
-
-        async def main() -> None:
-            await client.testsets.deprecating_create_testset(
-                app_id="app_id",
-                name="name",
-                csvdata=[{"key": "value"}],
-            )
-
-
-        asyncio.run(main())
-        """
-        _response = await self._client_wrapper.httpx_client.request(
-            f"testsets/{jsonable_encoder(app_id)}",
-            method="POST",
-            json={
-                "name": name,
-                "csvdata": csvdata,
-            },
-            request_options=request_options,
-            omit=OMIT,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    TestSetSimpleResponse,
-                    parse_obj_as(
-                        type_=TestSetSimpleResponse,  # type: ignore
-                        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)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
-
     async def get_single_testset(
         self,
         testset_id: str,
@@ -1124,10 +960,10 @@ class AsyncTestsetsClient:
         request_options: typing.Optional[RequestOptions] = None,
     ) -> typing.Optional[typing.Any]:
         """
-        Fetch a specific testset in a MongoDB collection using its _id.
+        Fetch a specific testset in Postgres.
 
         Args:
-            testset_id (str): The _id of the testset to fetch.
+            testset_id (str): The id of the testset to fetch.
 
         Returns:
             The requested testset if found, else an HTTPException.

agenta/client/backend/types/__init__.py

@@ -1,5 +1,6 @@
 # This file was auto-generated by Fern from our API Definition.
 
+from .account_response import AccountResponse
 from .agenta_node_dto import AgentaNodeDto
 from .agenta_node_dto_nodes_value import AgentaNodeDtoNodesValue
 from .agenta_nodes_response import AgentaNodesResponse
@@ -22,8 +23,6 @@ from .config_dto import ConfigDto
 from .config_response_model import ConfigResponseModel
 from .correct_answer import CorrectAnswer
 from .create_app_output import CreateAppOutput
-from .create_span import CreateSpan
-from .create_trace_response import CreateTraceResponse
 from .delete_evaluation import DeleteEvaluation
 from .docker_env_vars import DockerEnvVars
 from .environment_output import EnvironmentOutput
@@ -53,16 +52,18 @@ from .image import Image
 from .invite_request import InviteRequest
 from .legacy_analytics_response import LegacyAnalyticsResponse
 from .legacy_data_point import LegacyDataPoint
+from .legacy_scope_request import LegacyScopeRequest
+from .legacy_scopes_response import LegacyScopesResponse
+from .legacy_user_request import LegacyUserRequest
+from .legacy_user_response import LegacyUserResponse
 from .lifecycle_dto import LifecycleDto
 from .link_dto import LinkDto
 from .list_api_keys_response import ListApiKeysResponse
 from .llm_run_rate_limit import LlmRunRateLimit
-from .llm_tokens import LlmTokens
 from .metrics_dto import MetricsDto
 from .new_testset import NewTestset
 from .node_dto import NodeDto
 from .node_type import NodeType
-
 from .o_tel_context_dto import OTelContextDto
 from .o_tel_event_dto import OTelEventDto
 from .o_tel_extra_dto import OTelExtraDto
@@ -72,28 +73,31 @@ 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_membership_request import OrganizationMembershipRequest
 from .organization_output import OrganizationOutput
-from .outputs import Outputs
+from .organization_request import OrganizationRequest
 from .parent_dto import ParentDto
 from .permission import Permission
+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
 from .result import Result
+from .role import Role
 from .root_dto import RootDto
+from .scopes_response_model import ScopesResponseModel
 from .score import Score
 from .secret_dto import SecretDto
 from .secret_kind import SecretKind
 from .secret_response_dto import SecretResponseDto
 from .simple_evaluation_output import SimpleEvaluationOutput
-from .span import Span
-from .span_detail import SpanDetail
 from .span_dto import SpanDto
 from .span_dto_nodes_value import SpanDtoNodesValue
-from .span_status_code import SpanStatusCode
-from .span_variant import SpanVariant
 from .status_code import StatusCode
 from .status_dto import StatusDto
 from .template import Template
@@ -101,23 +105,25 @@ from .template_image_info import TemplateImageInfo
 from .test_set_output_response import TestSetOutputResponse
 from .test_set_simple_response import TestSetSimpleResponse
 from .time_dto import TimeDto
-from .trace_detail import TraceDetail
 from .tree_dto import TreeDto
 from .tree_type import TreeType
 from .update_app_output import UpdateAppOutput
 from .uri import Uri
+from .user_request import UserRequest
 from .validation_error import ValidationError
 from .validation_error_loc_item import ValidationErrorLocItem
 from .variant_action import VariantAction
 from .variant_action_enum import VariantActionEnum
-from .with_pagination import WithPagination
 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
 
 __all__ = [
+    "AccountResponse",
     "AgentaNodeDto",
     "AgentaNodeDtoNodesValue",
     "AgentaNodesResponse",
@@ -140,8 +146,6 @@ __all__ = [
     "ConfigResponseModel",
     "CorrectAnswer",
     "CreateAppOutput",
-    "CreateSpan",
-    "CreateTraceResponse",
     "DeleteEvaluation",
     "DockerEnvVars",
     "EnvironmentOutput",
@@ -171,13 +175,15 @@ __all__ = [
     "InviteRequest",
     "LegacyAnalyticsResponse",
     "LegacyDataPoint",
+    "LegacyScopeRequest",
+    "LegacyScopesResponse",
+    "LegacyUserRequest",
+    "LegacyUserResponse",
     "LifecycleDto",
     "LinkDto",
     "ListApiKeysResponse",
     "LlmRunRateLimit",
-    "LlmTokens",
     "MetricsDto",
-    "NewHumanEvaluation",
     "NewTestset",
     "NodeDto",
     "NodeType",
@@ -190,28 +196,31 @@ __all__ = [
     "OTelSpansResponse",
     "OTelStatusCode",
     "Organization",
+    "OrganizationMembershipRequest",
     "OrganizationOutput",
-    "Outputs",
+    "OrganizationRequest",
     "ParentDto",
     "Permission",
+    "ProjectMembershipRequest",
+    "ProjectRequest",
+    "ProjectScope",
     "ProjectsResponse",
     "ProviderKeyDto",
     "ProviderKind",
+    "Reference",
     "ReferenceDto",
     "ReferenceRequestModel",
     "Result",
+    "Role",
     "RootDto",
+    "ScopesResponseModel",
     "Score",
     "SecretDto",
     "SecretKind",
     "SecretResponseDto",
     "SimpleEvaluationOutput",
-    "Span",
-    "SpanDetail",
     "SpanDto",
     "SpanDtoNodesValue",
-    "SpanStatusCode",
-    "SpanVariant",
     "StatusCode",
     "StatusDto",
     "Template",
@@ -219,18 +228,19 @@ __all__ = [
     "TestSetOutputResponse",
     "TestSetSimpleResponse",
     "TimeDto",
-    "TraceDetail",
     "TreeDto",
     "TreeType",
     "UpdateAppOutput",
     "Uri",
+    "UserRequest",
     "ValidationError",
     "ValidationErrorLocItem",
     "VariantAction",
     "VariantActionEnum",
-    "WithPagination",
     "WorkspaceMemberResponse",
+    "WorkspaceMembershipRequest",
     "WorkspacePermission",
+    "WorkspaceRequest",
     "WorkspaceResponse",
     "WorkspaceRole",
     "WorkspaceRoleResponse",

agenta/client/backend/types/account_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 .legacy_user_response import LegacyUserResponse
+from .legacy_scopes_response import LegacyScopesResponse
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class AccountResponse(UniversalBaseModel):
+    user: typing.Optional[LegacyUserResponse] = None
+    scopes: typing.Optional[typing.List[LegacyScopesResponse]] = 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/app_variant_revision.py

@@ -1,13 +1,14 @@
 # This file was auto-generated by Fern from our API Definition.
 
 from ..core.pydantic_utilities import UniversalBaseModel
+import typing
 from .config_db import ConfigDb
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
-import typing
 import pydantic
 
 
 class AppVariantRevision(UniversalBaseModel):
+    id: typing.Optional[str] = None
     revision: int
     modified_by: str
     config: ConfigDb

agenta/client/backend/types/{create_trace_response.py → legacy_scope_request.py}

@@ -1,14 +1,13 @@
 # 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 typing
 import pydantic
 
 
-class CreateTraceResponse(UniversalBaseModel):
-    message: str
-    data: typing.Dict[str, typing.Optional[typing.Any]]
+class LegacyScopeRequest(UniversalBaseModel):
+    name: str
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(

agenta/client/backend/types/legacy_scopes_response.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 LegacyScopesResponse(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: typing.Optional[str] = None
+    project_name: typing.Optional[str] = None
+    is_demo: typing.Optional[bool] = None
+    user_role: typing.Optional[str] = None
+    credentials: 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/{span_variant.py → legacy_user_request.py}

@@ -1,15 +1,14 @@
 # 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 typing
 import pydantic
 
 
-class SpanVariant(UniversalBaseModel):
-    variant_id: typing.Optional[str] = None
-    variant_name: typing.Optional[str] = None
-    revision: typing.Optional[int] = None
+class LegacyUserRequest(UniversalBaseModel):
+    name: str
+    email: str
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(

agenta/client/backend/types/{llm_tokens.py → legacy_user_response.py}

@@ -6,10 +6,8 @@ from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import pydantic
 
 
-class LlmTokens(UniversalBaseModel):
-    prompt_tokens: typing.Optional[int] = None
-    completion_tokens: typing.Optional[int] = None
-    total_tokens: typing.Optional[int] = None
+class LegacyUserResponse(UniversalBaseModel):
+    id: typing.Optional[str] = None
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(

agenta/client/backend/types/{with_pagination.py → organization_membership_request.py}

@@ -2,17 +2,16 @@
 
 from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-import typing_extensions
-from ..core.serialization import FieldMetadata
+from .reference import Reference
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import pydantic
 
 
-class WithPagination(UniversalBaseModel):
-    data: typing.List[typing.Optional[typing.Any]]
-    total: int
-    page: int
-    page_size: typing_extensions.Annotated[int, FieldMetadata(alias="pageSize")]
+class OrganizationMembershipRequest(UniversalBaseModel):
+    role: typing.Literal["owner"] = "owner"
+    is_demo: bool
+    user_ref: Reference
+    organization_ref: Reference
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(

agenta/client/backend/types/organization_request.py

@@ -0,0 +1,23 @@
+# 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 OrganizationRequest(UniversalBaseModel):
+    name: str
+    description: str
+    is_paying: bool
+
+    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

@@ -14,6 +14,10 @@ Permission = typing.Union[
         "modify_variant_configurations",
         "delete_application_variant",
         "run_service",
+        "create_secret",
+        "view_secret",
+        "update_secret",
+        "delete_secret",
         "view_app_environment_deployment",
         "edit_app_environment_deployment",
         "create_app_environment_deployment",

agenta/client/backend/types/project_membership_request.py

@@ -0,0 +1,26 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+from .role import Role
+from .reference import Reference
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import typing
+import pydantic
+
+
+class ProjectMembershipRequest(UniversalBaseModel):
+    role: Role
+    is_demo: bool
+    user_ref: Reference
+    project_ref: Reference
+
+    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