c63a5cfe-b235-4fbe-8bbb-82a9e02a482a-python 0.1.0a4__py3-none-any.whl → 0.1.0a5__py3-none-any.whl

{gradientai/resources/providers → do_gradientai/resources/model_providers}/anthropic/keys.py

@@ -15,13 +15,18 @@ from ...._response import (
     async_to_streamed_response_wrapper,
 )
 from ...._base_client import make_request_options
-from ....types.providers.anthropic import key_list_params, key_create_params, key_update_params, key_list_agents_params
-from ....types.providers.anthropic.key_list_response import KeyListResponse
-from ....types.providers.anthropic.key_create_response import KeyCreateResponse
-from ....types.providers.anthropic.key_delete_response import KeyDeleteResponse
-from ....types.providers.anthropic.key_update_response import KeyUpdateResponse
-from ....types.providers.anthropic.key_retrieve_response import KeyRetrieveResponse
-from ....types.providers.anthropic.key_list_agents_response import KeyListAgentsResponse
+from ....types.model_providers.anthropic import (
+    key_list_params,
+    key_create_params,
+    key_update_params,
+    key_list_agents_params,
+)
+from ....types.model_providers.anthropic.key_list_response import KeyListResponse
+from ....types.model_providers.anthropic.key_create_response import KeyCreateResponse
+from ....types.model_providers.anthropic.key_delete_response import KeyDeleteResponse
+from ....types.model_providers.anthropic.key_update_response import KeyUpdateResponse
+from ....types.model_providers.anthropic.key_retrieve_response import KeyRetrieveResponse
+from ....types.model_providers.anthropic.key_list_agents_response import KeyListAgentsResponse
 
 __all__ = ["KeysResource", "AsyncKeysResource"]
 

gradientai/resources/providers/providers.py → do_gradientai/resources/model_providers/model_providers.py

@@ -21,10 +21,10 @@ from .anthropic.anthropic import (
     AsyncAnthropicResourceWithStreamingResponse,
 )
 
-__all__ = ["ProvidersResource", "AsyncProvidersResource"]
+__all__ = ["ModelProvidersResource", "AsyncModelProvidersResource"]
 
 
-class ProvidersResource(SyncAPIResource):
+class ModelProvidersResource(SyncAPIResource):
     @cached_property
     def anthropic(self) -> AnthropicResource:
         return AnthropicResource(self._client)
@@ -34,26 +34,26 @@ class ProvidersResource(SyncAPIResource):
         return OpenAIResource(self._client)
 
     @cached_property
-    def with_raw_response(self) -> ProvidersResourceWithRawResponse:
+    def with_raw_response(self) -> ModelProvidersResourceWithRawResponse:
         """
         This property can be used as a prefix for any HTTP method call to return
         the raw response object instead of the parsed content.
 
         For more information, see https://www.github.com/digitalocean/gradientai-python#accessing-raw-response-data-eg-headers
         """
-        return ProvidersResourceWithRawResponse(self)
+        return ModelProvidersResourceWithRawResponse(self)
 
     @cached_property
-    def with_streaming_response(self) -> ProvidersResourceWithStreamingResponse:
+    def with_streaming_response(self) -> ModelProvidersResourceWithStreamingResponse:
         """
         An alternative to `.with_raw_response` that doesn't eagerly read the response body.
 
         For more information, see https://www.github.com/digitalocean/gradientai-python#with_streaming_response
         """
-        return ProvidersResourceWithStreamingResponse(self)
+        return ModelProvidersResourceWithStreamingResponse(self)
 
 
-class AsyncProvidersResource(AsyncAPIResource):
+class AsyncModelProvidersResource(AsyncAPIResource):
     @cached_property
     def anthropic(self) -> AsyncAnthropicResource:
         return AsyncAnthropicResource(self._client)
@@ -63,72 +63,72 @@ class AsyncProvidersResource(AsyncAPIResource):
         return AsyncOpenAIResource(self._client)
 
     @cached_property
-    def with_raw_response(self) -> AsyncProvidersResourceWithRawResponse:
+    def with_raw_response(self) -> AsyncModelProvidersResourceWithRawResponse:
         """
         This property can be used as a prefix for any HTTP method call to return
         the raw response object instead of the parsed content.
 
         For more information, see https://www.github.com/digitalocean/gradientai-python#accessing-raw-response-data-eg-headers
         """
-        return AsyncProvidersResourceWithRawResponse(self)
+        return AsyncModelProvidersResourceWithRawResponse(self)
 
     @cached_property
-    def with_streaming_response(self) -> AsyncProvidersResourceWithStreamingResponse:
+    def with_streaming_response(self) -> AsyncModelProvidersResourceWithStreamingResponse:
         """
         An alternative to `.with_raw_response` that doesn't eagerly read the response body.
 
         For more information, see https://www.github.com/digitalocean/gradientai-python#with_streaming_response
         """
-        return AsyncProvidersResourceWithStreamingResponse(self)
+        return AsyncModelProvidersResourceWithStreamingResponse(self)
 
 
-class ProvidersResourceWithRawResponse:
-    def __init__(self, providers: ProvidersResource) -> None:
-        self._providers = providers
+class ModelProvidersResourceWithRawResponse:
+    def __init__(self, model_providers: ModelProvidersResource) -> None:
+        self._model_providers = model_providers
 
     @cached_property
     def anthropic(self) -> AnthropicResourceWithRawResponse:
-        return AnthropicResourceWithRawResponse(self._providers.anthropic)
+        return AnthropicResourceWithRawResponse(self._model_providers.anthropic)
 
     @cached_property
     def openai(self) -> OpenAIResourceWithRawResponse:
-        return OpenAIResourceWithRawResponse(self._providers.openai)
+        return OpenAIResourceWithRawResponse(self._model_providers.openai)
 
 
-class AsyncProvidersResourceWithRawResponse:
-    def __init__(self, providers: AsyncProvidersResource) -> None:
-        self._providers = providers
+class AsyncModelProvidersResourceWithRawResponse:
+    def __init__(self, model_providers: AsyncModelProvidersResource) -> None:
+        self._model_providers = model_providers
 
     @cached_property
     def anthropic(self) -> AsyncAnthropicResourceWithRawResponse:
-        return AsyncAnthropicResourceWithRawResponse(self._providers.anthropic)
+        return AsyncAnthropicResourceWithRawResponse(self._model_providers.anthropic)
 
     @cached_property
     def openai(self) -> AsyncOpenAIResourceWithRawResponse:
-        return AsyncOpenAIResourceWithRawResponse(self._providers.openai)
+        return AsyncOpenAIResourceWithRawResponse(self._model_providers.openai)
 
 
-class ProvidersResourceWithStreamingResponse:
-    def __init__(self, providers: ProvidersResource) -> None:
-        self._providers = providers
+class ModelProvidersResourceWithStreamingResponse:
+    def __init__(self, model_providers: ModelProvidersResource) -> None:
+        self._model_providers = model_providers
 
     @cached_property
     def anthropic(self) -> AnthropicResourceWithStreamingResponse:
-        return AnthropicResourceWithStreamingResponse(self._providers.anthropic)
+        return AnthropicResourceWithStreamingResponse(self._model_providers.anthropic)
 
     @cached_property
     def openai(self) -> OpenAIResourceWithStreamingResponse:
-        return OpenAIResourceWithStreamingResponse(self._providers.openai)
+        return OpenAIResourceWithStreamingResponse(self._model_providers.openai)
 
 
-class AsyncProvidersResourceWithStreamingResponse:
-    def __init__(self, providers: AsyncProvidersResource) -> None:
-        self._providers = providers
+class AsyncModelProvidersResourceWithStreamingResponse:
+    def __init__(self, model_providers: AsyncModelProvidersResource) -> None:
+        self._model_providers = model_providers
 
     @cached_property
     def anthropic(self) -> AsyncAnthropicResourceWithStreamingResponse:
-        return AsyncAnthropicResourceWithStreamingResponse(self._providers.anthropic)
+        return AsyncAnthropicResourceWithStreamingResponse(self._model_providers.anthropic)
 
     @cached_property
     def openai(self) -> AsyncOpenAIResourceWithStreamingResponse:
-        return AsyncOpenAIResourceWithStreamingResponse(self._providers.openai)
+        return AsyncOpenAIResourceWithStreamingResponse(self._model_providers.openai)

{gradientai/resources/providers → do_gradientai/resources/model_providers}/openai/keys.py

@@ -15,13 +15,18 @@ from ...._response import (
     async_to_streamed_response_wrapper,
 )
 from ...._base_client import make_request_options
-from ....types.providers.openai import key_list_params, key_create_params, key_update_params, key_retrieve_agents_params
-from ....types.providers.openai.key_list_response import KeyListResponse
-from ....types.providers.openai.key_create_response import KeyCreateResponse
-from ....types.providers.openai.key_delete_response import KeyDeleteResponse
-from ....types.providers.openai.key_update_response import KeyUpdateResponse
-from ....types.providers.openai.key_retrieve_response import KeyRetrieveResponse
-from ....types.providers.openai.key_retrieve_agents_response import KeyRetrieveAgentsResponse
+from ....types.model_providers.openai import (
+    key_list_params,
+    key_create_params,
+    key_update_params,
+    key_retrieve_agents_params,
+)
+from ....types.model_providers.openai.key_list_response import KeyListResponse
+from ....types.model_providers.openai.key_create_response import KeyCreateResponse
+from ....types.model_providers.openai.key_delete_response import KeyDeleteResponse
+from ....types.model_providers.openai.key_update_response import KeyUpdateResponse
+from ....types.model_providers.openai.key_retrieve_response import KeyRetrieveResponse
+from ....types.model_providers.openai.key_retrieve_agents_response import KeyRetrieveAgentsResponse
 
 __all__ = ["KeysResource", "AsyncKeysResource"]
 

{gradientai/resources/inference → do_gradientai/resources}/models.py

@@ -4,18 +4,18 @@ from __future__ import annotations
 
 import httpx
 
-from ..._types import NOT_GIVEN, Body, Query, Headers, NotGiven
-from ..._compat import cached_property
-from ..._resource import SyncAPIResource, AsyncAPIResource
-from ..._response import (
+from .._types import NOT_GIVEN, Body, Query, Headers, NotGiven
+from .._compat import cached_property
+from .._resource import SyncAPIResource, AsyncAPIResource
+from .._response import (
     to_raw_response_wrapper,
     to_streamed_response_wrapper,
     async_to_raw_response_wrapper,
     async_to_streamed_response_wrapper,
 )
-from ..._base_client import make_request_options
-from ...types.inference.model import Model
-from ...types.inference.model_list_response import ModelListResponse
+from ..types.model import Model
+from .._base_client import make_request_options
+from ..types.model_list_response import ModelListResponse
 
 __all__ = ["ModelsResource", "AsyncModelsResource"]
 

do_gradientai/resources/regions.py

@@ -0,0 +1,195 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+import httpx
+
+from ..types import region_list_params
+from .._types import NOT_GIVEN, Body, Query, Headers, NotGiven
+from .._utils import maybe_transform, async_maybe_transform
+from .._compat import cached_property
+from .._resource import SyncAPIResource, AsyncAPIResource
+from .._response import (
+    to_raw_response_wrapper,
+    to_streamed_response_wrapper,
+    async_to_raw_response_wrapper,
+    async_to_streamed_response_wrapper,
+)
+from .._base_client import make_request_options
+from ..types.region_list_response import RegionListResponse
+
+__all__ = ["RegionsResource", "AsyncRegionsResource"]
+
+
+class RegionsResource(SyncAPIResource):
+    @cached_property
+    def with_raw_response(self) -> RegionsResourceWithRawResponse:
+        """
+        This property can be used as a prefix for any HTTP method call to return
+        the raw response object instead of the parsed content.
+
+        For more information, see https://www.github.com/digitalocean/gradientai-python#accessing-raw-response-data-eg-headers
+        """
+        return RegionsResourceWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> RegionsResourceWithStreamingResponse:
+        """
+        An alternative to `.with_raw_response` that doesn't eagerly read the response body.
+
+        For more information, see https://www.github.com/digitalocean/gradientai-python#with_streaming_response
+        """
+        return RegionsResourceWithStreamingResponse(self)
+
+    def list(
+        self,
+        *,
+        serves_batch: bool | NotGiven = NOT_GIVEN,
+        serves_inference: bool | NotGiven = NOT_GIVEN,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> RegionListResponse:
+        """
+        To list all datacenter regions, send a GET request to `/v2/gen-ai/regions`.
+
+        Args:
+          serves_batch: include datacenters that are capable of running batch jobs.
+
+          serves_inference: include datacenters that serve inference.
+
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        return self._get(
+            "/v2/gen-ai/regions"
+            if self._client._base_url_overridden
+            else "https://api.digitalocean.com/v2/gen-ai/regions",
+            options=make_request_options(
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                query=maybe_transform(
+                    {
+                        "serves_batch": serves_batch,
+                        "serves_inference": serves_inference,
+                    },
+                    region_list_params.RegionListParams,
+                ),
+            ),
+            cast_to=RegionListResponse,
+        )
+
+
+class AsyncRegionsResource(AsyncAPIResource):
+    @cached_property
+    def with_raw_response(self) -> AsyncRegionsResourceWithRawResponse:
+        """
+        This property can be used as a prefix for any HTTP method call to return
+        the raw response object instead of the parsed content.
+
+        For more information, see https://www.github.com/digitalocean/gradientai-python#accessing-raw-response-data-eg-headers
+        """
+        return AsyncRegionsResourceWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> AsyncRegionsResourceWithStreamingResponse:
+        """
+        An alternative to `.with_raw_response` that doesn't eagerly read the response body.
+
+        For more information, see https://www.github.com/digitalocean/gradientai-python#with_streaming_response
+        """
+        return AsyncRegionsResourceWithStreamingResponse(self)
+
+    async def list(
+        self,
+        *,
+        serves_batch: bool | NotGiven = NOT_GIVEN,
+        serves_inference: bool | NotGiven = NOT_GIVEN,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> RegionListResponse:
+        """
+        To list all datacenter regions, send a GET request to `/v2/gen-ai/regions`.
+
+        Args:
+          serves_batch: include datacenters that are capable of running batch jobs.
+
+          serves_inference: include datacenters that serve inference.
+
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        return await self._get(
+            "/v2/gen-ai/regions"
+            if self._client._base_url_overridden
+            else "https://api.digitalocean.com/v2/gen-ai/regions",
+            options=make_request_options(
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                query=await async_maybe_transform(
+                    {
+                        "serves_batch": serves_batch,
+                        "serves_inference": serves_inference,
+                    },
+                    region_list_params.RegionListParams,
+                ),
+            ),
+            cast_to=RegionListResponse,
+        )
+
+
+class RegionsResourceWithRawResponse:
+    def __init__(self, regions: RegionsResource) -> None:
+        self._regions = regions
+
+        self.list = to_raw_response_wrapper(
+            regions.list,
+        )
+
+
+class AsyncRegionsResourceWithRawResponse:
+    def __init__(self, regions: AsyncRegionsResource) -> None:
+        self._regions = regions
+
+        self.list = async_to_raw_response_wrapper(
+            regions.list,
+        )
+
+
+class RegionsResourceWithStreamingResponse:
+    def __init__(self, regions: RegionsResource) -> None:
+        self._regions = regions
+
+        self.list = to_streamed_response_wrapper(
+            regions.list,
+        )
+
+
+class AsyncRegionsResourceWithStreamingResponse:
+    def __init__(self, regions: AsyncRegionsResource) -> None:
+        self._regions = regions
+
+        self.list = async_to_streamed_response_wrapper(
+            regions.list,
+        )

{gradientai → do_gradientai}/types/__init__.py

@@ -2,15 +2,14 @@
 
 from __future__ import annotations
 
+from .model import Model as Model
+from .shared import APIMeta as APIMeta, APILinks as APILinks
 from .api_agent import APIAgent as APIAgent
-from .api_model import APIModel as APIModel
 from .api_agreement import APIAgreement as APIAgreement
 from .api_workspace import APIWorkspace as APIWorkspace
 from .api_agent_model import APIAgentModel as APIAgentModel
-from .api_indexing_job import APIIndexingJob as APIIndexingJob
 from .agent_list_params import AgentListParams as AgentListParams
 from .api_model_version import APIModelVersion as APIModelVersion
-from .model_list_params import ModelListParams as ModelListParams
 from .api_knowledge_base import APIKnowledgeBase as APIKnowledgeBase
 from .region_list_params import RegionListParams as RegionListParams
 from .agent_create_params import AgentCreateParams as AgentCreateParams
@@ -22,32 +21,18 @@ from .region_list_response import RegionListResponse as RegionListResponse
 from .agent_create_response import AgentCreateResponse as AgentCreateResponse
 from .agent_delete_response import AgentDeleteResponse as AgentDeleteResponse
 from .agent_update_response import AgentUpdateResponse as AgentUpdateResponse
-from .api_evaluation_metric import APIEvaluationMetric as APIEvaluationMetric
 from .api_agent_api_key_info import APIAgentAPIKeyInfo as APIAgentAPIKeyInfo
 from .agent_retrieve_response import AgentRetrieveResponse as AgentRetrieveResponse
 from .api_openai_api_key_info import APIOpenAIAPIKeyInfo as APIOpenAIAPIKeyInfo
-from .indexing_job_list_params import IndexingJobListParams as IndexingJobListParams
 from .api_deployment_visibility import APIDeploymentVisibility as APIDeploymentVisibility
 from .agent_update_status_params import AgentUpdateStatusParams as AgentUpdateStatusParams
 from .api_anthropic_api_key_info import APIAnthropicAPIKeyInfo as APIAnthropicAPIKeyInfo
-from .indexing_job_create_params import IndexingJobCreateParams as IndexingJobCreateParams
-from .indexing_job_list_response import IndexingJobListResponse as IndexingJobListResponse
 from .knowledge_base_list_params import KnowledgeBaseListParams as KnowledgeBaseListParams
 from .agent_update_status_response import AgentUpdateStatusResponse as AgentUpdateStatusResponse
-from .indexing_job_create_response import IndexingJobCreateResponse as IndexingJobCreateResponse
 from .knowledge_base_create_params import KnowledgeBaseCreateParams as KnowledgeBaseCreateParams
 from .knowledge_base_list_response import KnowledgeBaseListResponse as KnowledgeBaseListResponse
 from .knowledge_base_update_params import KnowledgeBaseUpdateParams as KnowledgeBaseUpdateParams
-from .indexing_job_retrieve_response import IndexingJobRetrieveResponse as IndexingJobRetrieveResponse
 from .knowledge_base_create_response import KnowledgeBaseCreateResponse as KnowledgeBaseCreateResponse
 from .knowledge_base_delete_response import KnowledgeBaseDeleteResponse as KnowledgeBaseDeleteResponse
 from .knowledge_base_update_response import KnowledgeBaseUpdateResponse as KnowledgeBaseUpdateResponse
 from .knowledge_base_retrieve_response import KnowledgeBaseRetrieveResponse as KnowledgeBaseRetrieveResponse
-from .indexing_job_update_cancel_params import IndexingJobUpdateCancelParams as IndexingJobUpdateCancelParams
-from .indexing_job_update_cancel_response import IndexingJobUpdateCancelResponse as IndexingJobUpdateCancelResponse
-from .region_list_evaluation_metrics_response import (
-    RegionListEvaluationMetricsResponse as RegionListEvaluationMetricsResponse,
-)
-from .indexing_job_retrieve_data_sources_response import (
-    IndexingJobRetrieveDataSourcesResponse as IndexingJobRetrieveDataSourcesResponse,
-)

{gradientai → do_gradientai}/types/agent_list_response.py

@@ -5,9 +5,9 @@ from datetime import datetime
 from typing_extensions import Literal
 
 from .._models import BaseModel
-from .agents.api_meta import APIMeta
 from .api_agent_model import APIAgentModel
-from .agents.api_links import APILinks
+from .shared.api_meta import APIMeta
+from .shared.api_links import APILinks
 from .api_knowledge_base import APIKnowledgeBase
 from .api_retrieval_method import APIRetrievalMethod
 from .api_deployment_visibility import APIDeploymentVisibility

do_gradientai/types/agents/__init__.py

@@ -0,0 +1,67 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from .api_star_metric import APIStarMetric as APIStarMetric
+from .route_add_params import RouteAddParams as RouteAddParams
+from .api_evaluation_run import APIEvaluationRun as APIEvaluationRun
+from .route_add_response import RouteAddResponse as RouteAddResponse
+from .api_key_list_params import APIKeyListParams as APIKeyListParams
+from .route_update_params import RouteUpdateParams as RouteUpdateParams
+from .route_view_response import RouteViewResponse as RouteViewResponse
+from .version_list_params import VersionListParams as VersionListParams
+from .api_evaluation_metric import APIEvaluationMetric as APIEvaluationMetric
+from .api_evaluation_prompt import APIEvaluationPrompt as APIEvaluationPrompt
+from .api_key_create_params import APIKeyCreateParams as APIKeyCreateParams
+from .api_key_list_response import APIKeyListResponse as APIKeyListResponse
+from .api_key_update_params import APIKeyUpdateParams as APIKeyUpdateParams
+from .api_star_metric_param import APIStarMetricParam as APIStarMetricParam
+from .route_delete_response import RouteDeleteResponse as RouteDeleteResponse
+from .route_update_response import RouteUpdateResponse as RouteUpdateResponse
+from .version_list_response import VersionListResponse as VersionListResponse
+from .version_update_params import VersionUpdateParams as VersionUpdateParams
+from .function_create_params import FunctionCreateParams as FunctionCreateParams
+from .function_update_params import FunctionUpdateParams as FunctionUpdateParams
+from .api_key_create_response import APIKeyCreateResponse as APIKeyCreateResponse
+from .api_key_delete_response import APIKeyDeleteResponse as APIKeyDeleteResponse
+from .api_key_update_response import APIKeyUpdateResponse as APIKeyUpdateResponse
+from .version_update_response import VersionUpdateResponse as VersionUpdateResponse
+from .api_evaluation_test_case import APIEvaluationTestCase as APIEvaluationTestCase
+from .function_create_response import FunctionCreateResponse as FunctionCreateResponse
+from .function_delete_response import FunctionDeleteResponse as FunctionDeleteResponse
+from .function_update_response import FunctionUpdateResponse as FunctionUpdateResponse
+from .api_key_regenerate_response import APIKeyRegenerateResponse as APIKeyRegenerateResponse
+from .api_evaluation_metric_result import APIEvaluationMetricResult as APIEvaluationMetricResult
+from .evaluation_run_create_params import EvaluationRunCreateParams as EvaluationRunCreateParams
+from .api_link_knowledge_base_output import APILinkKnowledgeBaseOutput as APILinkKnowledgeBaseOutput
+from .evaluation_run_create_response import EvaluationRunCreateResponse as EvaluationRunCreateResponse
+from .knowledge_base_detach_response import KnowledgeBaseDetachResponse as KnowledgeBaseDetachResponse
+from .evaluation_metric_list_response import EvaluationMetricListResponse as EvaluationMetricListResponse
+from .evaluation_dataset_create_params import EvaluationDatasetCreateParams as EvaluationDatasetCreateParams
+from .evaluation_run_retrieve_response import EvaluationRunRetrieveResponse as EvaluationRunRetrieveResponse
+from .evaluation_dataset_create_response import EvaluationDatasetCreateResponse as EvaluationDatasetCreateResponse
+from .evaluation_test_case_create_params import EvaluationTestCaseCreateParams as EvaluationTestCaseCreateParams
+from .evaluation_test_case_list_response import EvaluationTestCaseListResponse as EvaluationTestCaseListResponse
+from .evaluation_test_case_update_params import EvaluationTestCaseUpdateParams as EvaluationTestCaseUpdateParams
+from .evaluation_run_list_results_response import EvaluationRunListResultsResponse as EvaluationRunListResultsResponse
+from .evaluation_test_case_create_response import EvaluationTestCaseCreateResponse as EvaluationTestCaseCreateResponse
+from .evaluation_test_case_retrieve_params import EvaluationTestCaseRetrieveParams as EvaluationTestCaseRetrieveParams
+from .evaluation_test_case_update_response import EvaluationTestCaseUpdateResponse as EvaluationTestCaseUpdateResponse
+from .evaluation_test_case_retrieve_response import (
+    EvaluationTestCaseRetrieveResponse as EvaluationTestCaseRetrieveResponse,
+)
+from .evaluation_run_retrieve_results_response import (
+    EvaluationRunRetrieveResultsResponse as EvaluationRunRetrieveResultsResponse,
+)
+from .evaluation_test_case_list_evaluation_runs_params import (
+    EvaluationTestCaseListEvaluationRunsParams as EvaluationTestCaseListEvaluationRunsParams,
+)
+from .evaluation_test_case_list_evaluation_runs_response import (
+    EvaluationTestCaseListEvaluationRunsResponse as EvaluationTestCaseListEvaluationRunsResponse,
+)
+from .evaluation_dataset_create_file_upload_presigned_urls_params import (
+    EvaluationDatasetCreateFileUploadPresignedURLsParams as EvaluationDatasetCreateFileUploadPresignedURLsParams,
+)
+from .evaluation_dataset_create_file_upload_presigned_urls_response import (
+    EvaluationDatasetCreateFileUploadPresignedURLsResponse as EvaluationDatasetCreateFileUploadPresignedURLsResponse,
+)

{gradientai/types → do_gradientai/types/agents}/api_evaluation_metric.py

@@ -3,7 +3,7 @@
 from typing import Optional
 from typing_extensions import Literal
 
-from .._models import BaseModel
+from ..._models import BaseModel
 
 __all__ = ["APIEvaluationMetric"]
 

{gradientai/types/regions/evaluation_runs → do_gradientai/types/agents}/api_evaluation_metric_result.py

@@ -2,7 +2,7 @@
 
 from typing import Optional
 
-from ...._models import BaseModel
+from ..._models import BaseModel
 
 __all__ = ["APIEvaluationMetricResult"]
 

gradientai/types/regions/evaluation_runs/api_prompt.py → do_gradientai/types/agents/api_evaluation_prompt.py

@@ -2,10 +2,10 @@
 
 from typing import List, Optional
 
-from ...._models import BaseModel
+from ..._models import BaseModel
 from .api_evaluation_metric_result import APIEvaluationMetricResult
 
-__all__ = ["APIPrompt", "PromptChunk"]
+__all__ = ["APIEvaluationPrompt", "PromptChunk"]
 
 
 class PromptChunk(BaseModel):
@@ -25,7 +25,7 @@ class PromptChunk(BaseModel):
     """Text content of the chunk."""
 
 
-class APIPrompt(BaseModel):
+class APIEvaluationPrompt(BaseModel):
     ground_truth: Optional[str] = None
     """The ground truth for the prompt."""
 

{gradientai/types/regions/evaluation_runs → do_gradientai/types/agents}/api_evaluation_run.py

@@ -4,18 +4,30 @@ from typing import List, Optional
 from datetime import datetime
 from typing_extensions import Literal
 
-from ...._models import BaseModel
+from ..._models import BaseModel
 from .api_evaluation_metric_result import APIEvaluationMetricResult
 
 __all__ = ["APIEvaluationRun"]
 
 
 class APIEvaluationRun(BaseModel):
+    agent_deleted: Optional[bool] = None
+
+    agent_name: Optional[str] = None
+
     agent_uuid: Optional[str] = None
     """Agent UUID."""
 
     agent_version_hash: Optional[str] = None
 
+    agent_workspace_uuid: Optional[str] = None
+
+    created_by_user_email: Optional[str] = None
+
+    created_by_user_id: Optional[str] = None
+
+    error_description: Optional[str] = None
+
     evaluation_run_uuid: Optional[str] = None
     """Evaluation run UUID."""
 

{gradientai/types/regions → do_gradientai/types/agents}/api_evaluation_test_case.py

@@ -5,7 +5,7 @@ from datetime import datetime
 
 from ..._models import BaseModel
 from .api_star_metric import APIStarMetric
-from ..api_evaluation_metric import APIEvaluationMetric
+from .api_evaluation_metric import APIEvaluationMetric
 
 __all__ = ["APIEvaluationTestCase"]
 

{gradientai → do_gradientai}/types/agents/api_key_list_response.py

@@ -2,9 +2,9 @@
 
 from typing import List, Optional
 
-from .api_meta import APIMeta
 from ..._models import BaseModel
-from .api_links import APILinks
+from ..shared.api_meta import APIMeta
+from ..shared.api_links import APILinks
 from ..api_agent_api_key_info import APIAgentAPIKeyInfo
 
 __all__ = ["APIKeyListResponse"]