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

{digitalocean_genai_sdk/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"]
 
@@ -33,7 +38,7 @@ class KeysResource(SyncAPIResource):
         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/genai-python#accessing-raw-response-data-eg-headers
+        For more information, see https://www.github.com/digitalocean/gradientai-python#accessing-raw-response-data-eg-headers
         """
         return KeysResourceWithRawResponse(self)
 
@@ -42,7 +47,7 @@ class KeysResource(SyncAPIResource):
         """
         An alternative to `.with_raw_response` that doesn't eagerly read the response body.
 
-        For more information, see https://www.github.com/digitalocean/genai-python#with_streaming_response
+        For more information, see https://www.github.com/digitalocean/gradientai-python#with_streaming_response
         """
         return KeysResourceWithStreamingResponse(self)
 
@@ -71,7 +76,9 @@ class KeysResource(SyncAPIResource):
           timeout: Override the client-level default timeout for this request, in seconds
         """
         return self._post(
-            "/v2/genai/openai/keys",
+            "/v2/gen-ai/openai/keys"
+            if self._client._base_url_overridden
+            else "https://api.digitalocean.com/v2/gen-ai/openai/keys",
             body=maybe_transform(
                 {
                     "api_key": api_key,
@@ -112,7 +119,9 @@ class KeysResource(SyncAPIResource):
         if not api_key_uuid:
             raise ValueError(f"Expected a non-empty value for `api_key_uuid` but received {api_key_uuid!r}")
         return self._get(
-            f"/v2/genai/openai/keys/{api_key_uuid}",
+            f"/v2/gen-ai/openai/keys/{api_key_uuid}"
+            if self._client._base_url_overridden
+            else f"https://api.digitalocean.com/v2/gen-ai/openai/keys/{api_key_uuid}",
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
@@ -149,7 +158,9 @@ class KeysResource(SyncAPIResource):
         if not path_api_key_uuid:
             raise ValueError(f"Expected a non-empty value for `path_api_key_uuid` but received {path_api_key_uuid!r}")
         return self._put(
-            f"/v2/genai/openai/keys/{path_api_key_uuid}",
+            f"/v2/gen-ai/openai/keys/{path_api_key_uuid}"
+            if self._client._base_url_overridden
+            else f"https://api.digitalocean.com/v2/gen-ai/openai/keys/{path_api_key_uuid}",
             body=maybe_transform(
                 {
                     "api_key": api_key,
@@ -193,7 +204,9 @@ class KeysResource(SyncAPIResource):
           timeout: Override the client-level default timeout for this request, in seconds
         """
         return self._get(
-            "/v2/genai/openai/keys",
+            "/v2/gen-ai/openai/keys"
+            if self._client._base_url_overridden
+            else "https://api.digitalocean.com/v2/gen-ai/openai/keys",
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -237,7 +250,9 @@ class KeysResource(SyncAPIResource):
         if not api_key_uuid:
             raise ValueError(f"Expected a non-empty value for `api_key_uuid` but received {api_key_uuid!r}")
         return self._delete(
-            f"/v2/genai/openai/keys/{api_key_uuid}",
+            f"/v2/gen-ai/openai/keys/{api_key_uuid}"
+            if self._client._base_url_overridden
+            else f"https://api.digitalocean.com/v2/gen-ai/openai/keys/{api_key_uuid}",
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
@@ -276,7 +291,9 @@ class KeysResource(SyncAPIResource):
         if not uuid:
             raise ValueError(f"Expected a non-empty value for `uuid` but received {uuid!r}")
         return self._get(
-            f"/v2/genai/openai/keys/{uuid}/agents",
+            f"/v2/gen-ai/openai/keys/{uuid}/agents"
+            if self._client._base_url_overridden
+            else f"https://api.digitalocean.com/v2/gen-ai/openai/keys/{uuid}/agents",
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -301,7 +318,7 @@ class AsyncKeysResource(AsyncAPIResource):
         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/genai-python#accessing-raw-response-data-eg-headers
+        For more information, see https://www.github.com/digitalocean/gradientai-python#accessing-raw-response-data-eg-headers
         """
         return AsyncKeysResourceWithRawResponse(self)
 
@@ -310,7 +327,7 @@ class AsyncKeysResource(AsyncAPIResource):
         """
         An alternative to `.with_raw_response` that doesn't eagerly read the response body.
 
-        For more information, see https://www.github.com/digitalocean/genai-python#with_streaming_response
+        For more information, see https://www.github.com/digitalocean/gradientai-python#with_streaming_response
         """
         return AsyncKeysResourceWithStreamingResponse(self)
 
@@ -339,7 +356,9 @@ class AsyncKeysResource(AsyncAPIResource):
           timeout: Override the client-level default timeout for this request, in seconds
         """
         return await self._post(
-            "/v2/genai/openai/keys",
+            "/v2/gen-ai/openai/keys"
+            if self._client._base_url_overridden
+            else "https://api.digitalocean.com/v2/gen-ai/openai/keys",
             body=await async_maybe_transform(
                 {
                     "api_key": api_key,
@@ -380,7 +399,9 @@ class AsyncKeysResource(AsyncAPIResource):
         if not api_key_uuid:
             raise ValueError(f"Expected a non-empty value for `api_key_uuid` but received {api_key_uuid!r}")
         return await self._get(
-            f"/v2/genai/openai/keys/{api_key_uuid}",
+            f"/v2/gen-ai/openai/keys/{api_key_uuid}"
+            if self._client._base_url_overridden
+            else f"https://api.digitalocean.com/v2/gen-ai/openai/keys/{api_key_uuid}",
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
@@ -417,7 +438,9 @@ class AsyncKeysResource(AsyncAPIResource):
         if not path_api_key_uuid:
             raise ValueError(f"Expected a non-empty value for `path_api_key_uuid` but received {path_api_key_uuid!r}")
         return await self._put(
-            f"/v2/genai/openai/keys/{path_api_key_uuid}",
+            f"/v2/gen-ai/openai/keys/{path_api_key_uuid}"
+            if self._client._base_url_overridden
+            else f"https://api.digitalocean.com/v2/gen-ai/openai/keys/{path_api_key_uuid}",
             body=await async_maybe_transform(
                 {
                     "api_key": api_key,
@@ -461,7 +484,9 @@ class AsyncKeysResource(AsyncAPIResource):
           timeout: Override the client-level default timeout for this request, in seconds
         """
         return await self._get(
-            "/v2/genai/openai/keys",
+            "/v2/gen-ai/openai/keys"
+            if self._client._base_url_overridden
+            else "https://api.digitalocean.com/v2/gen-ai/openai/keys",
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -505,7 +530,9 @@ class AsyncKeysResource(AsyncAPIResource):
         if not api_key_uuid:
             raise ValueError(f"Expected a non-empty value for `api_key_uuid` but received {api_key_uuid!r}")
         return await self._delete(
-            f"/v2/genai/openai/keys/{api_key_uuid}",
+            f"/v2/gen-ai/openai/keys/{api_key_uuid}"
+            if self._client._base_url_overridden
+            else f"https://api.digitalocean.com/v2/gen-ai/openai/keys/{api_key_uuid}",
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
@@ -544,7 +571,9 @@ class AsyncKeysResource(AsyncAPIResource):
         if not uuid:
             raise ValueError(f"Expected a non-empty value for `uuid` but received {uuid!r}")
         return await self._get(
-            f"/v2/genai/openai/keys/{uuid}/agents",
+            f"/v2/gen-ai/openai/keys/{uuid}/agents"
+            if self._client._base_url_overridden
+            else f"https://api.digitalocean.com/v2/gen-ai/openai/keys/{uuid}/agents",
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,

{digitalocean_genai_sdk/resources/providers → do_gradientai/resources/model_providers}/openai/openai.py

@@ -27,7 +27,7 @@ class OpenAIResource(SyncAPIResource):
         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/genai-python#accessing-raw-response-data-eg-headers
+        For more information, see https://www.github.com/digitalocean/gradientai-python#accessing-raw-response-data-eg-headers
         """
         return OpenAIResourceWithRawResponse(self)
 
@@ -36,7 +36,7 @@ class OpenAIResource(SyncAPIResource):
         """
         An alternative to `.with_raw_response` that doesn't eagerly read the response body.
 
-        For more information, see https://www.github.com/digitalocean/genai-python#with_streaming_response
+        For more information, see https://www.github.com/digitalocean/gradientai-python#with_streaming_response
         """
         return OpenAIResourceWithStreamingResponse(self)
 
@@ -52,7 +52,7 @@ class AsyncOpenAIResource(AsyncAPIResource):
         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/genai-python#accessing-raw-response-data-eg-headers
+        For more information, see https://www.github.com/digitalocean/gradientai-python#accessing-raw-response-data-eg-headers
         """
         return AsyncOpenAIResourceWithRawResponse(self)
 
@@ -61,7 +61,7 @@ class AsyncOpenAIResource(AsyncAPIResource):
         """
         An alternative to `.with_raw_response` that doesn't eagerly read the response body.
 
-        For more information, see https://www.github.com/digitalocean/genai-python#with_streaming_response
+        For more information, see https://www.github.com/digitalocean/gradientai-python#with_streaming_response
         """
         return AsyncOpenAIResourceWithStreamingResponse(self)
 

{digitalocean_genai_sdk → do_gradientai}/resources/models.py

@@ -27,7 +27,7 @@ class ModelsResource(SyncAPIResource):
         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/genai-python#accessing-raw-response-data-eg-headers
+        For more information, see https://www.github.com/digitalocean/gradientai-python#accessing-raw-response-data-eg-headers
         """
         return ModelsResourceWithRawResponse(self)
 
@@ -36,7 +36,7 @@ class ModelsResource(SyncAPIResource):
         """
         An alternative to `.with_raw_response` that doesn't eagerly read the response body.
 
-        For more information, see https://www.github.com/digitalocean/genai-python#with_streaming_response
+        For more information, see https://www.github.com/digitalocean/gradientai-python#with_streaming_response
         """
         return ModelsResourceWithStreamingResponse(self)
 
@@ -67,7 +67,9 @@ class ModelsResource(SyncAPIResource):
         if not model:
             raise ValueError(f"Expected a non-empty value for `model` but received {model!r}")
         return self._get(
-            f"/models/{model}",
+            f"/models/{model}"
+            if self._client._base_url_overridden
+            else f"https://inference.do-ai.run/v1/models/{model}",
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
@@ -89,7 +91,7 @@ class ModelsResource(SyncAPIResource):
         one such as the owner and availability.
         """
         return self._get(
-            "/models",
+            "/models" if self._client._base_url_overridden else "https://inference.do-ai.run/v1/models",
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
@@ -104,7 +106,7 @@ class AsyncModelsResource(AsyncAPIResource):
         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/genai-python#accessing-raw-response-data-eg-headers
+        For more information, see https://www.github.com/digitalocean/gradientai-python#accessing-raw-response-data-eg-headers
         """
         return AsyncModelsResourceWithRawResponse(self)
 
@@ -113,7 +115,7 @@ class AsyncModelsResource(AsyncAPIResource):
         """
         An alternative to `.with_raw_response` that doesn't eagerly read the response body.
 
-        For more information, see https://www.github.com/digitalocean/genai-python#with_streaming_response
+        For more information, see https://www.github.com/digitalocean/gradientai-python#with_streaming_response
         """
         return AsyncModelsResourceWithStreamingResponse(self)
 
@@ -144,7 +146,9 @@ class AsyncModelsResource(AsyncAPIResource):
         if not model:
             raise ValueError(f"Expected a non-empty value for `model` but received {model!r}")
         return await self._get(
-            f"/models/{model}",
+            f"/models/{model}"
+            if self._client._base_url_overridden
+            else f"https://inference.do-ai.run/v1/models/{model}",
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
@@ -166,7 +170,7 @@ class AsyncModelsResource(AsyncAPIResource):
         one such as the owner and availability.
         """
         return await self._get(
-            "/models",
+            "/models" if self._client._base_url_overridden else "https://inference.do-ai.run/v1/models",
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),

{digitalocean_genai_sdk → do_gradientai}/resources/regions.py

@@ -28,7 +28,7 @@ class RegionsResource(SyncAPIResource):
         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/genai-python#accessing-raw-response-data-eg-headers
+        For more information, see https://www.github.com/digitalocean/gradientai-python#accessing-raw-response-data-eg-headers
         """
         return RegionsResourceWithRawResponse(self)
 
@@ -37,7 +37,7 @@ class RegionsResource(SyncAPIResource):
         """
         An alternative to `.with_raw_response` that doesn't eagerly read the response body.
 
-        For more information, see https://www.github.com/digitalocean/genai-python#with_streaming_response
+        For more information, see https://www.github.com/digitalocean/gradientai-python#with_streaming_response
         """
         return RegionsResourceWithStreamingResponse(self)
 
@@ -70,7 +70,9 @@ class RegionsResource(SyncAPIResource):
           timeout: Override the client-level default timeout for this request, in seconds
         """
         return self._get(
-            "/v2/genai/regions",
+            "/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,
@@ -95,7 +97,7 @@ class AsyncRegionsResource(AsyncAPIResource):
         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/genai-python#accessing-raw-response-data-eg-headers
+        For more information, see https://www.github.com/digitalocean/gradientai-python#accessing-raw-response-data-eg-headers
         """
         return AsyncRegionsResourceWithRawResponse(self)
 
@@ -104,7 +106,7 @@ class AsyncRegionsResource(AsyncAPIResource):
         """
         An alternative to `.with_raw_response` that doesn't eagerly read the response body.
 
-        For more information, see https://www.github.com/digitalocean/genai-python#with_streaming_response
+        For more information, see https://www.github.com/digitalocean/gradientai-python#with_streaming_response
         """
         return AsyncRegionsResourceWithStreamingResponse(self)
 
@@ -137,7 +139,9 @@ class AsyncRegionsResource(AsyncAPIResource):
           timeout: Override the client-level default timeout for this request, in seconds
         """
         return await self._get(
-            "/v2/genai/regions",
+            "/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,

{digitalocean_genai_sdk → do_gradientai}/types/__init__.py

@@ -3,10 +3,11 @@
 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_indexing_job import APIIndexingJob as APIIndexingJob
+from .api_workspace import APIWorkspace as APIWorkspace
+from .api_agent_model import APIAgentModel as APIAgentModel
 from .agent_list_params import AgentListParams as AgentListParams
 from .api_model_version import APIModelVersion as APIModelVersion
 from .api_knowledge_base import APIKnowledgeBase as APIKnowledgeBase
@@ -14,44 +15,24 @@ 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_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
-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 .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 .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 → 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 .api_model import APIModel
-from .agents.api_meta import APIMeta
-from .agents.api_links import APILinks
+from .api_agent_model import APIAgentModel
+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
@@ -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
 

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,
+)

do_gradientai/types/agents/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

do_gradientai/types/agents/api_evaluation_metric_result.py

@@ -0,0 +1,17 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Optional
+
+from ..._models import BaseModel
+
+__all__ = ["APIEvaluationMetricResult"]
+
+
+class APIEvaluationMetricResult(BaseModel):
+    metric_name: Optional[str] = None
+
+    number_value: Optional[float] = None
+    """The value of the metric as a number."""
+
+    string_value: Optional[str] = None
+    """The value of the metric as a string."""

do_gradientai/types/agents/api_evaluation_prompt.py

@@ -0,0 +1,42 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import List, Optional
+
+from ..._models import BaseModel
+from .api_evaluation_metric_result import APIEvaluationMetricResult
+
+__all__ = ["APIEvaluationPrompt", "PromptChunk"]
+
+
+class PromptChunk(BaseModel):
+    chunk_usage_pct: Optional[float] = None
+    """The usage percentage of the chunk."""
+
+    chunk_used: Optional[bool] = None
+    """Indicates if the chunk was used in the prompt."""
+
+    index_uuid: Optional[str] = None
+    """The index uuid (Knowledge Base) of the chunk."""
+
+    source_name: Optional[str] = None
+    """The source name for the chunk, e.g., the file name or document title."""
+
+    text: Optional[str] = None
+    """Text content of the chunk."""
+
+
+class APIEvaluationPrompt(BaseModel):
+    ground_truth: Optional[str] = None
+    """The ground truth for the prompt."""
+
+    input: Optional[str] = None
+
+    output: Optional[str] = None
+
+    prompt_chunks: Optional[List[PromptChunk]] = None
+    """The list of prompt chunks."""
+
+    prompt_id: Optional[int] = None
+
+    prompt_level_metric_results: Optional[List[APIEvaluationMetricResult]] = None
+    """The metric results for the prompt."""