agenta 0.48.9__py3-none-any.whl → 0.49.0__py3-none-any.whl

agenta/client/backend/__init__.py

@@ -0,0 +1,461 @@
+# This file was auto-generated by Fern from our API Definition.
+
+# isort: skip_file
+
+from .types import (
+    AccountRequest,
+    AccountResponse,
+    AgentaNodeDto,
+    AgentaNodesResponse,
+    AgentaRootDto,
+    AgentaRootsResponse,
+    AgentaTreeDto,
+    AgentaTreesResponse,
+    AggregatedResult,
+    AggregatedResultEvaluatorConfig,
+    AnalyticsResponse,
+    Annotation,
+    AnnotationCreate,
+    AnnotationEdit,
+    AnnotationKind,
+    AnnotationLink,
+    AnnotationLinkResponse,
+    AnnotationQuery,
+    AnnotationQueryRequest,
+    AnnotationReference,
+    AnnotationReferences,
+    AnnotationResponse,
+    AnnotationSource,
+    AnnotationsResponse,
+    App,
+    AppVariantResponse,
+    AppVariantRevision,
+    Artifact,
+    BaseOutput,
+    BodyFetchWorkflowRevision,
+    BodyImportTestset,
+    BucketDto,
+    CollectStatusResponse,
+    ConfigDb,
+    ConfigDto,
+    ConfigResponseModel,
+    CorrectAnswer,
+    CreateAppOutput,
+    CustomModelSettingsDto,
+    CustomProviderDto,
+    CustomProviderKind,
+    CustomProviderSettingsDto,
+    Data,
+    DeleteEvaluation,
+    EnvironmentOutput,
+    EnvironmentOutputExtended,
+    EnvironmentRevision,
+    Error,
+    Evaluation,
+    EvaluationScenario,
+    EvaluationScenarioInput,
+    EvaluationScenarioOutput,
+    EvaluationScenarioResult,
+    EvaluationStatusEnum,
+    EvaluationType,
+    Evaluator,
+    EvaluatorConfig,
+    EvaluatorFlags,
+    EvaluatorMappingOutputInterface,
+    EvaluatorOutputInterface,
+    EvaluatorQuery,
+    EvaluatorQueryRequest,
+    EvaluatorRequest,
+    EvaluatorResponse,
+    EvaluatorsResponse,
+    ExceptionDto,
+    Focus,
+    Format,
+    FullJsonInput,
+    FullJsonOutput,
+    GetConfigResponse,
+    Header,
+    HttpValidationError,
+    HumanEvaluation,
+    HumanEvaluationScenario,
+    HumanEvaluationScenarioInput,
+    HumanEvaluationScenarioOutput,
+    InviteRequest,
+    LegacyAnalyticsResponse,
+    LegacyDataPoint,
+    LegacyEvaluator,
+    LegacyScopeRequest,
+    LegacyScopesResponse,
+    LegacySubscriptionRequest,
+    LegacyUserRequest,
+    LegacyUserResponse,
+    LifecycleDto,
+    LinkDto,
+    ListApiKeysResponse,
+    LlmRunRateLimit,
+    MetaRequest,
+    MetricsDto,
+    NewTestset,
+    NodeDto,
+    NodeType,
+    OTelContextDto,
+    OTelEvent,
+    OTelEventDto,
+    OTelExtraDto,
+    OTelFlatSpan,
+    OTelFlatSpanInputEndTime,
+    OTelFlatSpanInputStartTime,
+    OTelFlatSpanOutputEndTime,
+    OTelFlatSpanOutputStartTime,
+    OTelLink,
+    OTelLinkDto,
+    OTelLinksResponse,
+    OTelSpan,
+    OTelSpanDto,
+    OTelSpanInputEndTime,
+    OTelSpanInputSpansValue,
+    OTelSpanInputStartTime,
+    OTelSpanKind,
+    OTelSpanOutputEndTime,
+    OTelSpanOutputSpansValue,
+    OTelSpanOutputStartTime,
+    OTelSpansTree,
+    OTelSpansTreeInputSpansValue,
+    OTelSpansTreeOutputSpansValue,
+    OTelStatusCode,
+    OTelTracingDataResponse,
+    OTelTracingRequest,
+    OTelTracingResponse,
+    Organization,
+    OrganizationDetails,
+    OrganizationMembershipRequest,
+    OrganizationOutput,
+    OrganizationRequest,
+    ParentDto,
+    Permission,
+    Plan,
+    ProjectMembershipRequest,
+    ProjectRequest,
+    ProjectScope,
+    ProjectsResponse,
+    Reference,
+    ReferenceDto,
+    ReferenceRequestModel,
+    Result,
+    Role,
+    RootDto,
+    ScopesResponseModel,
+    Score,
+    SecretDto,
+    SecretKind,
+    SecretResponseDto,
+    SimpleEvaluationOutput,
+    StandardProviderDto,
+    StandardProviderKind,
+    StandardProviderSettingsDto,
+    StatusCode,
+    StatusDto,
+    TestSetOutputResponse,
+    TestSetSimpleResponse,
+    TestcaseResponse,
+    Testset,
+    TestsetRequest,
+    TestsetResponse,
+    TestsetsResponse,
+    TimeDto,
+    Timestamp,
+    TreeDto,
+    TreeType,
+    UpdateAppOutput,
+    UserRequest,
+    ValidationError,
+    ValidationErrorLocItem,
+    WorkflowArtifact,
+    WorkflowData,
+    WorkflowFlags,
+    WorkflowRequest,
+    WorkflowResponse,
+    WorkflowRevision,
+    WorkflowRevisionRequest,
+    WorkflowRevisionResponse,
+    WorkflowRevisionsResponse,
+    WorkflowVariant,
+    WorkflowVariantRequest,
+    WorkflowVariantResponse,
+    WorkflowVariantsResponse,
+    WorkflowsResponse,
+    Workspace,
+    WorkspaceMemberResponse,
+    WorkspaceMembershipRequest,
+    WorkspacePermission,
+    WorkspaceRequest,
+    WorkspaceResponse,
+    WorkspaceRole,
+)
+from .errors import UnprocessableEntityError
+from . import (
+    access_control,
+    admin,
+    api_keys,
+    apps,
+    bases,
+    billing,
+    configs,
+    containers,
+    environments,
+    evals,
+    evaluations,
+    evaluators,
+    human_evaluations,
+    observability,
+    organization,
+    scopes,
+    testsets,
+    tracing,
+    variants,
+    vault,
+    workflows,
+    workspace,
+)
+from .client import AgentaApi, AsyncAgentaApi
+from .environment import AgentaApiEnvironment
+from .observability import (
+    FetchTraceByIdRequestTraceId,
+    FetchTraceByIdResponse,
+    QueryAnalyticsResponse,
+    QueryTracesResponse,
+)
+from .testsets import (
+    CreateTestsetFromFileRequestFileType,
+    FetchTestsetToFileRequestFileType,
+    UpdateTestsetFromFileRequestFileType,
+)
+from .tracing import FetchTraceRequestTraceId, RemoveTraceRequestTraceId
+from .variants import AddVariantFromBaseAndConfigResponse
+
+__all__ = [
+    "AccountRequest",
+    "AccountResponse",
+    "AddVariantFromBaseAndConfigResponse",
+    "AgentaApi",
+    "AgentaApiEnvironment",
+    "AgentaNodeDto",
+    "AgentaNodesResponse",
+    "AgentaRootDto",
+    "AgentaRootsResponse",
+    "AgentaTreeDto",
+    "AgentaTreesResponse",
+    "AggregatedResult",
+    "AggregatedResultEvaluatorConfig",
+    "AnalyticsResponse",
+    "Annotation",
+    "AnnotationCreate",
+    "AnnotationEdit",
+    "AnnotationKind",
+    "AnnotationLink",
+    "AnnotationLinkResponse",
+    "AnnotationQuery",
+    "AnnotationQueryRequest",
+    "AnnotationReference",
+    "AnnotationReferences",
+    "AnnotationResponse",
+    "AnnotationSource",
+    "AnnotationsResponse",
+    "App",
+    "AppVariantResponse",
+    "AppVariantRevision",
+    "Artifact",
+    "AsyncAgentaApi",
+    "BaseOutput",
+    "BodyFetchWorkflowRevision",
+    "BodyImportTestset",
+    "BucketDto",
+    "CollectStatusResponse",
+    "ConfigDb",
+    "ConfigDto",
+    "ConfigResponseModel",
+    "CorrectAnswer",
+    "CreateAppOutput",
+    "CreateTestsetFromFileRequestFileType",
+    "CustomModelSettingsDto",
+    "CustomProviderDto",
+    "CustomProviderKind",
+    "CustomProviderSettingsDto",
+    "Data",
+    "DeleteEvaluation",
+    "EnvironmentOutput",
+    "EnvironmentOutputExtended",
+    "EnvironmentRevision",
+    "Error",
+    "Evaluation",
+    "EvaluationScenario",
+    "EvaluationScenarioInput",
+    "EvaluationScenarioOutput",
+    "EvaluationScenarioResult",
+    "EvaluationStatusEnum",
+    "EvaluationType",
+    "Evaluator",
+    "EvaluatorConfig",
+    "EvaluatorFlags",
+    "EvaluatorMappingOutputInterface",
+    "EvaluatorOutputInterface",
+    "EvaluatorQuery",
+    "EvaluatorQueryRequest",
+    "EvaluatorRequest",
+    "EvaluatorResponse",
+    "EvaluatorsResponse",
+    "ExceptionDto",
+    "FetchTestsetToFileRequestFileType",
+    "FetchTraceByIdRequestTraceId",
+    "FetchTraceByIdResponse",
+    "FetchTraceRequestTraceId",
+    "Focus",
+    "Format",
+    "FullJsonInput",
+    "FullJsonOutput",
+    "GetConfigResponse",
+    "Header",
+    "HttpValidationError",
+    "HumanEvaluation",
+    "HumanEvaluationScenario",
+    "HumanEvaluationScenarioInput",
+    "HumanEvaluationScenarioOutput",
+    "InviteRequest",
+    "LegacyAnalyticsResponse",
+    "LegacyDataPoint",
+    "LegacyEvaluator",
+    "LegacyScopeRequest",
+    "LegacyScopesResponse",
+    "LegacySubscriptionRequest",
+    "LegacyUserRequest",
+    "LegacyUserResponse",
+    "LifecycleDto",
+    "LinkDto",
+    "ListApiKeysResponse",
+    "LlmRunRateLimit",
+    "MetaRequest",
+    "MetricsDto",
+    "NewTestset",
+    "NodeDto",
+    "NodeType",
+    "OTelContextDto",
+    "OTelEvent",
+    "OTelEventDto",
+    "OTelExtraDto",
+    "OTelFlatSpan",
+    "OTelFlatSpanInputEndTime",
+    "OTelFlatSpanInputStartTime",
+    "OTelFlatSpanOutputEndTime",
+    "OTelFlatSpanOutputStartTime",
+    "OTelLink",
+    "OTelLinkDto",
+    "OTelLinksResponse",
+    "OTelSpan",
+    "OTelSpanDto",
+    "OTelSpanInputEndTime",
+    "OTelSpanInputSpansValue",
+    "OTelSpanInputStartTime",
+    "OTelSpanKind",
+    "OTelSpanOutputEndTime",
+    "OTelSpanOutputSpansValue",
+    "OTelSpanOutputStartTime",
+    "OTelSpansTree",
+    "OTelSpansTreeInputSpansValue",
+    "OTelSpansTreeOutputSpansValue",
+    "OTelStatusCode",
+    "OTelTracingDataResponse",
+    "OTelTracingRequest",
+    "OTelTracingResponse",
+    "Organization",
+    "OrganizationDetails",
+    "OrganizationMembershipRequest",
+    "OrganizationOutput",
+    "OrganizationRequest",
+    "ParentDto",
+    "Permission",
+    "Plan",
+    "ProjectMembershipRequest",
+    "ProjectRequest",
+    "ProjectScope",
+    "ProjectsResponse",
+    "QueryAnalyticsResponse",
+    "QueryTracesResponse",
+    "Reference",
+    "ReferenceDto",
+    "ReferenceRequestModel",
+    "RemoveTraceRequestTraceId",
+    "Result",
+    "Role",
+    "RootDto",
+    "ScopesResponseModel",
+    "Score",
+    "SecretDto",
+    "SecretKind",
+    "SecretResponseDto",
+    "SimpleEvaluationOutput",
+    "StandardProviderDto",
+    "StandardProviderKind",
+    "StandardProviderSettingsDto",
+    "StatusCode",
+    "StatusDto",
+    "TestSetOutputResponse",
+    "TestSetSimpleResponse",
+    "TestcaseResponse",
+    "Testset",
+    "TestsetRequest",
+    "TestsetResponse",
+    "TestsetsResponse",
+    "TimeDto",
+    "Timestamp",
+    "TreeDto",
+    "TreeType",
+    "UnprocessableEntityError",
+    "UpdateAppOutput",
+    "UpdateTestsetFromFileRequestFileType",
+    "UserRequest",
+    "ValidationError",
+    "ValidationErrorLocItem",
+    "WorkflowArtifact",
+    "WorkflowData",
+    "WorkflowFlags",
+    "WorkflowRequest",
+    "WorkflowResponse",
+    "WorkflowRevision",
+    "WorkflowRevisionRequest",
+    "WorkflowRevisionResponse",
+    "WorkflowRevisionsResponse",
+    "WorkflowVariant",
+    "WorkflowVariantRequest",
+    "WorkflowVariantResponse",
+    "WorkflowVariantsResponse",
+    "WorkflowsResponse",
+    "Workspace",
+    "WorkspaceMemberResponse",
+    "WorkspaceMembershipRequest",
+    "WorkspacePermission",
+    "WorkspaceRequest",
+    "WorkspaceResponse",
+    "WorkspaceRole",
+    "access_control",
+    "admin",
+    "api_keys",
+    "apps",
+    "bases",
+    "billing",
+    "configs",
+    "containers",
+    "environments",
+    "evals",
+    "evaluations",
+    "evaluators",
+    "human_evaluations",
+    "observability",
+    "organization",
+    "scopes",
+    "testsets",
+    "tracing",
+    "variants",
+    "vault",
+    "workflows",
+    "workspace",
+]

agenta/client/backend/access_control/__init__.py

@@ -1 +1,3 @@
 # This file was auto-generated by Fern from our API Definition.
+
+# isort: skip_file

agenta/client/backend/access_control/client.py

@@ -1,24 +1,33 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.client_wrapper import SyncClientWrapper
 import typing
+
+from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ..core.request_options import RequestOptions
-from ..core.pydantic_utilities import parse_obj_as
-from ..errors.unprocessable_entity_error import UnprocessableEntityError
-from ..types.http_validation_error import HttpValidationError
-from json.decoder import JSONDecodeError
-from ..core.api_error import ApiError
-from ..core.client_wrapper import AsyncClientWrapper
+from .raw_client import AsyncRawAccessControlClient, RawAccessControlClient
 
 
 class AccessControlClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
-        self._client_wrapper = client_wrapper
+        self._raw_client = RawAccessControlClient(client_wrapper=client_wrapper)
+
+    @property
+    def with_raw_response(self) -> RawAccessControlClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        RawAccessControlClient
+        """
+        return self._raw_client
 
     def verify_permissions(
         self,
         *,
         action: typing.Optional[str] = None,
+        scope_type: typing.Optional[str] = None,
+        scope_id: typing.Optional[str] = None,
         resource_type: typing.Optional[str] = None,
         resource_id: typing.Optional[str] = None,
         request_options: typing.Optional[RequestOptions] = None,
@@ -28,6 +37,10 @@ class AccessControlClient:
         ----------
         action : typing.Optional[str]
 
+        scope_type : typing.Optional[str]
+
+        scope_id : typing.Optional[str]
+
         resource_type : typing.Optional[str]
 
         resource_id : typing.Optional[str]
@@ -46,53 +59,41 @@ class AccessControlClient:
 
         client = AgentaApi(
             api_key="YOUR_API_KEY",
-            base_url="https://yourhost.com/path/to/api",
         )
         client.access_control.verify_permissions()
         """
-        _response = self._client_wrapper.httpx_client.request(
-            "permissions/verify",
-            method="GET",
-            params={
-                "action": action,
-                "resource_type": resource_type,
-                "resource_id": resource_id,
-            },
+        _response = self._raw_client.verify_permissions(
+            action=action,
+            scope_type=scope_type,
+            scope_id=scope_id,
+            resource_type=resource_type,
+            resource_id=resource_id,
             request_options=request_options,
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    typing.Optional[typing.Any],
-                    parse_obj_as(
-                        type_=typing.Optional[typing.Any],  # 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)
+        return _response.data
 
 
 class AsyncAccessControlClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
-        self._client_wrapper = client_wrapper
+        self._raw_client = AsyncRawAccessControlClient(client_wrapper=client_wrapper)
+
+    @property
+    def with_raw_response(self) -> AsyncRawAccessControlClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        AsyncRawAccessControlClient
+        """
+        return self._raw_client
 
     async def verify_permissions(
         self,
         *,
         action: typing.Optional[str] = None,
+        scope_type: typing.Optional[str] = None,
+        scope_id: typing.Optional[str] = None,
         resource_type: typing.Optional[str] = None,
         resource_id: typing.Optional[str] = None,
         request_options: typing.Optional[RequestOptions] = None,
@@ -102,6 +103,10 @@ class AsyncAccessControlClient:
         ----------
         action : typing.Optional[str]
 
+        scope_type : typing.Optional[str]
+
+        scope_id : typing.Optional[str]
+
         resource_type : typing.Optional[str]
 
         resource_id : typing.Optional[str]
@@ -122,7 +127,6 @@ class AsyncAccessControlClient:
 
         client = AsyncAgentaApi(
             api_key="YOUR_API_KEY",
-            base_url="https://yourhost.com/path/to/api",
         )
 
 
@@ -132,36 +136,12 @@ class AsyncAccessControlClient:
 
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            "permissions/verify",
-            method="GET",
-            params={
-                "action": action,
-                "resource_type": resource_type,
-                "resource_id": resource_id,
-            },
+        _response = await self._raw_client.verify_permissions(
+            action=action,
+            scope_type=scope_type,
+            scope_id=scope_id,
+            resource_type=resource_type,
+            resource_id=resource_id,
             request_options=request_options,
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    typing.Optional[typing.Any],
-                    parse_obj_as(
-                        type_=typing.Optional[typing.Any],  # 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)
+        return _response.data