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

gradientai/resources/regions/regions.py

@@ -0,0 +1,352 @@
+# 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 .evaluation_datasets import (
+    EvaluationDatasetsResource,
+    AsyncEvaluationDatasetsResource,
+    EvaluationDatasetsResourceWithRawResponse,
+    AsyncEvaluationDatasetsResourceWithRawResponse,
+    EvaluationDatasetsResourceWithStreamingResponse,
+    AsyncEvaluationDatasetsResourceWithStreamingResponse,
+)
+from .evaluation_test_cases import (
+    EvaluationTestCasesResource,
+    AsyncEvaluationTestCasesResource,
+    EvaluationTestCasesResourceWithRawResponse,
+    AsyncEvaluationTestCasesResourceWithRawResponse,
+    EvaluationTestCasesResourceWithStreamingResponse,
+    AsyncEvaluationTestCasesResourceWithStreamingResponse,
+)
+from ...types.region_list_response import RegionListResponse
+from .evaluation_runs.evaluation_runs import (
+    EvaluationRunsResource,
+    AsyncEvaluationRunsResource,
+    EvaluationRunsResourceWithRawResponse,
+    AsyncEvaluationRunsResourceWithRawResponse,
+    EvaluationRunsResourceWithStreamingResponse,
+    AsyncEvaluationRunsResourceWithStreamingResponse,
+)
+from ...types.region_list_evaluation_metrics_response import RegionListEvaluationMetricsResponse
+
+__all__ = ["RegionsResource", "AsyncRegionsResource"]
+
+
+class RegionsResource(SyncAPIResource):
+    @cached_property
+    def evaluation_runs(self) -> EvaluationRunsResource:
+        return EvaluationRunsResource(self._client)
+
+    @cached_property
+    def evaluation_test_cases(self) -> EvaluationTestCasesResource:
+        return EvaluationTestCasesResource(self._client)
+
+    @cached_property
+    def evaluation_datasets(self) -> EvaluationDatasetsResource:
+        return EvaluationDatasetsResource(self._client)
+
+    @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,
+        )
+
+    def list_evaluation_metrics(
+        self,
+        *,
+        # 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,
+    ) -> RegionListEvaluationMetricsResponse:
+        """
+        To list all evaluation metrics, send a GET request to
+        `/v2/gen-ai/evaluation_metrics`.
+        """
+        return self._get(
+            "/v2/gen-ai/evaluation_metrics"
+            if self._client._base_url_overridden
+            else "https://api.digitalocean.com/v2/gen-ai/evaluation_metrics",
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=RegionListEvaluationMetricsResponse,
+        )
+
+
+class AsyncRegionsResource(AsyncAPIResource):
+    @cached_property
+    def evaluation_runs(self) -> AsyncEvaluationRunsResource:
+        return AsyncEvaluationRunsResource(self._client)
+
+    @cached_property
+    def evaluation_test_cases(self) -> AsyncEvaluationTestCasesResource:
+        return AsyncEvaluationTestCasesResource(self._client)
+
+    @cached_property
+    def evaluation_datasets(self) -> AsyncEvaluationDatasetsResource:
+        return AsyncEvaluationDatasetsResource(self._client)
+
+    @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,
+        )
+
+    async def list_evaluation_metrics(
+        self,
+        *,
+        # 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,
+    ) -> RegionListEvaluationMetricsResponse:
+        """
+        To list all evaluation metrics, send a GET request to
+        `/v2/gen-ai/evaluation_metrics`.
+        """
+        return await self._get(
+            "/v2/gen-ai/evaluation_metrics"
+            if self._client._base_url_overridden
+            else "https://api.digitalocean.com/v2/gen-ai/evaluation_metrics",
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=RegionListEvaluationMetricsResponse,
+        )
+
+
+class RegionsResourceWithRawResponse:
+    def __init__(self, regions: RegionsResource) -> None:
+        self._regions = regions
+
+        self.list = to_raw_response_wrapper(
+            regions.list,
+        )
+        self.list_evaluation_metrics = to_raw_response_wrapper(
+            regions.list_evaluation_metrics,
+        )
+
+    @cached_property
+    def evaluation_runs(self) -> EvaluationRunsResourceWithRawResponse:
+        return EvaluationRunsResourceWithRawResponse(self._regions.evaluation_runs)
+
+    @cached_property
+    def evaluation_test_cases(self) -> EvaluationTestCasesResourceWithRawResponse:
+        return EvaluationTestCasesResourceWithRawResponse(self._regions.evaluation_test_cases)
+
+    @cached_property
+    def evaluation_datasets(self) -> EvaluationDatasetsResourceWithRawResponse:
+        return EvaluationDatasetsResourceWithRawResponse(self._regions.evaluation_datasets)
+
+
+class AsyncRegionsResourceWithRawResponse:
+    def __init__(self, regions: AsyncRegionsResource) -> None:
+        self._regions = regions
+
+        self.list = async_to_raw_response_wrapper(
+            regions.list,
+        )
+        self.list_evaluation_metrics = async_to_raw_response_wrapper(
+            regions.list_evaluation_metrics,
+        )
+
+    @cached_property
+    def evaluation_runs(self) -> AsyncEvaluationRunsResourceWithRawResponse:
+        return AsyncEvaluationRunsResourceWithRawResponse(self._regions.evaluation_runs)
+
+    @cached_property
+    def evaluation_test_cases(self) -> AsyncEvaluationTestCasesResourceWithRawResponse:
+        return AsyncEvaluationTestCasesResourceWithRawResponse(self._regions.evaluation_test_cases)
+
+    @cached_property
+    def evaluation_datasets(self) -> AsyncEvaluationDatasetsResourceWithRawResponse:
+        return AsyncEvaluationDatasetsResourceWithRawResponse(self._regions.evaluation_datasets)
+
+
+class RegionsResourceWithStreamingResponse:
+    def __init__(self, regions: RegionsResource) -> None:
+        self._regions = regions
+
+        self.list = to_streamed_response_wrapper(
+            regions.list,
+        )
+        self.list_evaluation_metrics = to_streamed_response_wrapper(
+            regions.list_evaluation_metrics,
+        )
+
+    @cached_property
+    def evaluation_runs(self) -> EvaluationRunsResourceWithStreamingResponse:
+        return EvaluationRunsResourceWithStreamingResponse(self._regions.evaluation_runs)
+
+    @cached_property
+    def evaluation_test_cases(self) -> EvaluationTestCasesResourceWithStreamingResponse:
+        return EvaluationTestCasesResourceWithStreamingResponse(self._regions.evaluation_test_cases)
+
+    @cached_property
+    def evaluation_datasets(self) -> EvaluationDatasetsResourceWithStreamingResponse:
+        return EvaluationDatasetsResourceWithStreamingResponse(self._regions.evaluation_datasets)
+
+
+class AsyncRegionsResourceWithStreamingResponse:
+    def __init__(self, regions: AsyncRegionsResource) -> None:
+        self._regions = regions
+
+        self.list = async_to_streamed_response_wrapper(
+            regions.list,
+        )
+        self.list_evaluation_metrics = async_to_streamed_response_wrapper(
+            regions.list_evaluation_metrics,
+        )
+
+    @cached_property
+    def evaluation_runs(self) -> AsyncEvaluationRunsResourceWithStreamingResponse:
+        return AsyncEvaluationRunsResourceWithStreamingResponse(self._regions.evaluation_runs)
+
+    @cached_property
+    def evaluation_test_cases(self) -> AsyncEvaluationTestCasesResourceWithStreamingResponse:
+        return AsyncEvaluationTestCasesResourceWithStreamingResponse(self._regions.evaluation_test_cases)
+
+    @cached_property
+    def evaluation_datasets(self) -> AsyncEvaluationDatasetsResourceWithStreamingResponse:
+        return AsyncEvaluationDatasetsResourceWithStreamingResponse(self._regions.evaluation_datasets)

{digitalocean_genai_sdk → gradientai}/types/__init__.py

@@ -2,33 +2,32 @@
 
 from __future__ import annotations
 
-from .model import Model as Model
 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
 from .agent_list_response import AgentListResponse as AgentListResponse
 from .agent_update_params import AgentUpdateParams as AgentUpdateParams
-from .api_key_list_params import APIKeyListParams as APIKeyListParams
 from .model_list_response import ModelListResponse as ModelListResponse
 from .api_retrieval_method import APIRetrievalMethod as APIRetrievalMethod
 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_key_list_response import APIKeyListResponse as APIKeyListResponse
+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 .embedding_create_params import EmbeddingCreateParams as EmbeddingCreateParams
 from .indexing_job_list_params import IndexingJobListParams as IndexingJobListParams
 from .api_deployment_visibility import APIDeploymentVisibility as APIDeploymentVisibility
-from .embedding_create_response import EmbeddingCreateResponse as EmbeddingCreateResponse
 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
@@ -39,19 +38,16 @@ from .indexing_job_create_response import IndexingJobCreateResponse as IndexingJ
 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 .chat_completion_token_logprob import ChatCompletionTokenLogprob as ChatCompletionTokenLogprob
-from .chat_create_completion_params import ChatCreateCompletionParams as ChatCreateCompletionParams
 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 .chat_create_completion_response import ChatCreateCompletionResponse as ChatCreateCompletionResponse
 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,
 )
-from .chat_completion_request_message_content_part_text_param import (
-    ChatCompletionRequestMessageContentPartTextParam as ChatCompletionRequestMessageContentPartTextParam,
-)

{digitalocean_genai_sdk → gradientai}/types/agent_list_response.py

@@ -5,8 +5,8 @@ from datetime import datetime
 from typing_extensions import Literal
 
 from .._models import BaseModel
-from .api_model import APIModel
 from .agents.api_meta import APIMeta
+from .api_agent_model import APIAgentModel
 from .agents.api_links import APILinks
 from .api_knowledge_base import APIKnowledgeBase
 from .api_retrieval_method import APIRetrievalMethod
@@ -92,7 +92,7 @@ class AgentTemplate(BaseModel):
 
     max_tokens: Optional[int] = None
 
-    model: Optional[APIModel] = None
+    model: Optional[APIAgentModel] = None
 
     name: Optional[str] = None
 
@@ -143,7 +143,7 @@ class Agent(BaseModel):
     response.
     """
 
-    model: Optional[APIModel] = None
+    model: Optional[APIAgentModel] = None
 
     name: Optional[str] = None
 

{digitalocean_genai_sdk → gradientai}/types/api_agent.py

@@ -7,7 +7,7 @@ from datetime import datetime
 from typing_extensions import Literal
 
 from .._models import BaseModel
-from .api_model import APIModel
+from .api_agent_model import APIAgentModel
 from .api_knowledge_base import APIKnowledgeBase
 from .api_retrieval_method import APIRetrievalMethod
 from .api_agent_api_key_info import APIAgentAPIKeyInfo
@@ -162,7 +162,7 @@ class Template(BaseModel):
 
     max_tokens: Optional[int] = None
 
-    model: Optional[APIModel] = None
+    model: Optional[APIAgentModel] = None
 
     name: Optional[str] = None
 
@@ -222,7 +222,7 @@ class APIAgent(BaseModel):
 
     max_tokens: Optional[int] = None
 
-    model: Optional[APIModel] = None
+    model: Optional[APIAgentModel] = None
 
     name: Optional[str] = None
 
@@ -261,3 +261,8 @@ class APIAgent(BaseModel):
     user_id: Optional[str] = None
 
     uuid: Optional[str] = None
+
+    workspace: Optional["APIWorkspace"] = None
+
+
+from .api_workspace import APIWorkspace

digitalocean_genai_sdk/types/api_model.py → gradientai/types/api_agent_model.py

@@ -8,10 +8,10 @@ from .._models import BaseModel
 from .api_agreement import APIAgreement
 from .api_model_version import APIModelVersion
 
-__all__ = ["APIModel"]
+__all__ = ["APIAgentModel"]
 
 
-class APIModel(BaseModel):
+class APIAgentModel(BaseModel):
     agreement: Optional[APIAgreement] = None
 
     created_at: Optional[datetime] = None

gradientai/types/api_evaluation_metric.py

@@ -0,0 +1,24 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Optional
+from typing_extensions import Literal
+
+from .._models import BaseModel
+
+__all__ = ["APIEvaluationMetric"]
+
+
+class APIEvaluationMetric(BaseModel):
+    description: Optional[str] = None
+
+    metric_name: Optional[str] = None
+
+    metric_type: Optional[
+        Literal["METRIC_TYPE_UNSPECIFIED", "METRIC_TYPE_GENERAL_QUALITY", "METRIC_TYPE_RAG_AND_TOOL"]
+    ] = None
+
+    metric_uuid: Optional[str] = None
+
+    metric_value_type: Optional[
+        Literal["METRIC_VALUE_TYPE_UNSPECIFIED", "METRIC_VALUE_TYPE_NUMBER", "METRIC_VALUE_TYPE_STRING"]
+    ] = None

digitalocean_genai_sdk/types/api_key_list_response.py → gradientai/types/api_model.py

@@ -1,18 +1,16 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
-from typing import List, Optional
+from typing import Optional
 from datetime import datetime
 
 from .._models import BaseModel
 from .api_agreement import APIAgreement
-from .agents.api_meta import APIMeta
-from .agents.api_links import APILinks
 from .api_model_version import APIModelVersion
 
-__all__ = ["APIKeyListResponse", "Model"]
+__all__ = ["APIModel"]
 
 
-class Model(BaseModel):
+class APIModel(BaseModel):
     agreement: Optional[APIAgreement] = None
 
     created_at: Optional[datetime] = None
@@ -32,11 +30,3 @@ class Model(BaseModel):
     uuid: Optional[str] = None
 
     version: Optional[APIModelVersion] = None
-
-
-class APIKeyListResponse(BaseModel):
-    links: Optional[APILinks] = None
-
-    meta: Optional[APIMeta] = None
-
-    models: Optional[List[Model]] = None

{digitalocean_genai_sdk → gradientai}/types/api_openai_api_key_info.py

@@ -4,7 +4,7 @@ from typing import List, Optional
 from datetime import datetime
 
 from .._models import BaseModel
-from .api_model import APIModel
+from .api_agent_model import APIAgentModel
 
 __all__ = ["APIOpenAIAPIKeyInfo"]
 
@@ -16,7 +16,7 @@ class APIOpenAIAPIKeyInfo(BaseModel):
 
     deleted_at: Optional[datetime] = None
 
-    models: Optional[List[APIModel]] = None
+    models: Optional[List[APIAgentModel]] = None
 
     name: Optional[str] = None
 

gradientai/types/api_workspace.py

@@ -0,0 +1,36 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import List, Optional
+from datetime import datetime
+
+from .._models import BaseModel
+from .regions.api_evaluation_test_case import APIEvaluationTestCase
+
+__all__ = ["APIWorkspace"]
+
+
+class APIWorkspace(BaseModel):
+    agents: Optional[List["APIAgent"]] = None
+
+    created_at: Optional[datetime] = None
+
+    created_by: Optional[str] = None
+
+    created_by_email: Optional[str] = None
+
+    deleted_at: Optional[datetime] = None
+
+    description: Optional[str] = None
+
+    evaluation_test_cases: Optional[List[APIEvaluationTestCase]] = None
+
+    name: Optional[str] = None
+
+    updated_at: Optional[datetime] = None
+
+    uuid: Optional[str] = None
+
+
+from .api_agent import APIAgent

gradientai/types/chat/__init__.py

@@ -0,0 +1,7 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from .completion_create_params import CompletionCreateParams as CompletionCreateParams
+from .completion_create_response import CompletionCreateResponse as CompletionCreateResponse
+from .chat_completion_token_logprob import ChatCompletionTokenLogprob as ChatCompletionTokenLogprob

{digitalocean_genai_sdk/types → gradientai/types/chat}/chat_completion_token_logprob.py

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