feat(api): add inference errors

Author: stainless-app[bot]

.stats.yml

@@ -1,4 +1,4 @@
 configured_endpoints: 175
-openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/digitalocean%2Fgradient-cb3bf9b21459cad24410206c27a32fd31ef6cf86711700597549dbbd0d634002.yml
-openapi_spec_hash: 6a9149a81ba15e7c5c5c1f4d77daad92
+openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/digitalocean%2Fgradient-f07d74847e620dfa26d8df40ea4680814af9bba381b3a57a7b6ed76ad49d85f8.yml
+openapi_spec_hash: e3553dc2abf2afd4368b736bcc32a289
 config_hash: bad49c3bf949d5168ec3896bedff253a

src/gradient/resources/agents/agents.py

@@ -185,12 +185,14 @@ def create(
         description: str | Omit = omit,
         instruction: str | Omit = omit,
         knowledge_base_uuid: SequenceNotStr[str] | Omit = omit,
+        model_provider_key_uuid: str | Omit = omit,
         model_uuid: str | Omit = omit,
         name: str | Omit = omit,
         openai_key_uuid: str | Omit = omit,
         project_id: str | Omit = omit,
         region: str | Omit = omit,
         tags: SequenceNotStr[str] | Omit = omit,
+        workspace_uuid: str | Omit = omit,
         # 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,
@@ -227,6 +229,8 @@ def create(
 
           tags: Agent tag to organize related resources
 
+          workspace_uuid: Identifier for the workspace
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -245,12 +249,14 @@ def create(
                     "description": description,
                     "instruction": instruction,
                     "knowledge_base_uuid": knowledge_base_uuid,
+                    "model_provider_key_uuid": model_provider_key_uuid,
                     "model_uuid": model_uuid,
                     "name": name,
                     "openai_key_uuid": openai_key_uuid,
                     "project_id": project_id,
                     "region": region,
                     "tags": tags,
+                    "workspace_uuid": workspace_uuid,
                 },
                 agent_create_params.AgentCreateParams,
             ),
@@ -302,12 +308,14 @@ def update(
         path_uuid: str,
         *,
         agent_log_insights_enabled: bool | Omit = omit,
+        allowed_domains: SequenceNotStr[str] | Omit = omit,
         anthropic_key_uuid: str | Omit = omit,
         conversation_logs_enabled: bool | Omit = omit,
         description: str | Omit = omit,
         instruction: str | Omit = omit,
         k: int | Omit = omit,
         max_tokens: int | Omit = omit,
+        model_provider_key_uuid: str | Omit = omit,
         model_uuid: str | Omit = omit,
         name: str | Omit = omit,
         openai_key_uuid: str | Omit = omit,
@@ -331,6 +339,9 @@ def update(
         response body is a JSON object containing the agent.
 
         Args:
+          allowed_domains: Optional list of allowed domains for the chatbot - Must use fully qualified
+              domain name (FQDN) such as https://example.com
+
           anthropic_key_uuid: Optional anthropic key uuid for use with anthropic models
 
           conversation_logs_enabled: Optional update of conversation logs enabled
@@ -348,6 +359,8 @@ def update(
               or output, set as a number between 1 and 512. This determines the length of each
               response.
 
+          model_provider_key_uuid: Optional Model Provider uuid for use with provider models
+
           model_uuid: Identifier for the foundation model.
 
           name: Agent name
@@ -391,12 +404,14 @@ def update(
             body=maybe_transform(
                 {
                     "agent_log_insights_enabled": agent_log_insights_enabled,
+                    "allowed_domains": allowed_domains,
                     "anthropic_key_uuid": anthropic_key_uuid,
                     "conversation_logs_enabled": conversation_logs_enabled,
                     "description": description,
                     "instruction": instruction,
                     "k": k,
                     "max_tokens": max_tokens,
+                    "model_provider_key_uuid": model_provider_key_uuid,
                     "model_uuid": model_uuid,
                     "name": name,
                     "openai_key_uuid": openai_key_uuid,
@@ -767,12 +782,14 @@ async def create(
         description: str | Omit = omit,
         instruction: str | Omit = omit,
         knowledge_base_uuid: SequenceNotStr[str] | Omit = omit,
+        model_provider_key_uuid: str | Omit = omit,
         model_uuid: str | Omit = omit,
         name: str | Omit = omit,
         openai_key_uuid: str | Omit = omit,
         project_id: str | Omit = omit,
         region: str | Omit = omit,
         tags: SequenceNotStr[str] | Omit = omit,
+        workspace_uuid: str | Omit = omit,
         # 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,
@@ -809,6 +826,8 @@ async def create(
 
           tags: Agent tag to organize related resources
 
+          workspace_uuid: Identifier for the workspace
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -827,12 +846,14 @@ async def create(
                     "description": description,
                     "instruction": instruction,
                     "knowledge_base_uuid": knowledge_base_uuid,
+                    "model_provider_key_uuid": model_provider_key_uuid,
                     "model_uuid": model_uuid,
                     "name": name,
                     "openai_key_uuid": openai_key_uuid,
                     "project_id": project_id,
                     "region": region,
                     "tags": tags,
+                    "workspace_uuid": workspace_uuid,
                 },
                 agent_create_params.AgentCreateParams,
             ),
@@ -884,12 +905,14 @@ async def update(
         path_uuid: str,
         *,
         agent_log_insights_enabled: bool | Omit = omit,
+        allowed_domains: SequenceNotStr[str] | Omit = omit,
         anthropic_key_uuid: str | Omit = omit,
         conversation_logs_enabled: bool | Omit = omit,
         description: str | Omit = omit,
         instruction: str | Omit = omit,
         k: int | Omit = omit,
         max_tokens: int | Omit = omit,
+        model_provider_key_uuid: str | Omit = omit,
         model_uuid: str | Omit = omit,
         name: str | Omit = omit,
         openai_key_uuid: str | Omit = omit,
@@ -913,6 +936,9 @@ async def update(
         response body is a JSON object containing the agent.
 
         Args:
+          allowed_domains: Optional list of allowed domains for the chatbot - Must use fully qualified
+              domain name (FQDN) such as https://example.com
+
           anthropic_key_uuid: Optional anthropic key uuid for use with anthropic models
 
           conversation_logs_enabled: Optional update of conversation logs enabled
@@ -930,6 +956,8 @@ async def update(
               or output, set as a number between 1 and 512. This determines the length of each
               response.
 
+          model_provider_key_uuid: Optional Model Provider uuid for use with provider models
+
           model_uuid: Identifier for the foundation model.
 
           name: Agent name
@@ -973,12 +1001,14 @@ async def update(
             body=await async_maybe_transform(
                 {
                     "agent_log_insights_enabled": agent_log_insights_enabled,
+                    "allowed_domains": allowed_domains,
                     "anthropic_key_uuid": anthropic_key_uuid,
                     "conversation_logs_enabled": conversation_logs_enabled,
                     "description": description,
                     "instruction": instruction,
                     "k": k,
                     "max_tokens": max_tokens,
+                    "model_provider_key_uuid": model_provider_key_uuid,
                     "model_uuid": model_uuid,
                     "name": name,
                     "openai_key_uuid": openai_key_uuid,

src/gradient/types/agent_create_params.py

@@ -28,6 +28,8 @@ class AgentCreateParams(TypedDict, total=False):
     knowledge_base_uuid: SequenceNotStr[str]
     """Ids of the knowledge base(s) to attach to the agent"""
 
+    model_provider_key_uuid: str
+
     model_uuid: str
     """Identifier for the foundation model."""
 
@@ -45,3 +47,6 @@ class AgentCreateParams(TypedDict, total=False):
 
     tags: SequenceNotStr[str]
     """Agent tag to organize related resources"""
+
+    workspace_uuid: str
+    """Identifier for the workspace"""

src/gradient/types/agent_list_response.py

@@ -24,6 +24,8 @@
 
 
 class AgentChatbot(BaseModel):
+    allowed_domains: Optional[List[str]] = None
+
     button_background_color: Optional[str] = None
 
     logo: Optional[str] = None
@@ -61,6 +63,7 @@ class AgentDeployment(BaseModel):
             "STATUS_UNDEPLOYING",
             "STATUS_UNDEPLOYMENT_FAILED",
             "STATUS_DELETED",
+            "STATUS_BUILDING",
         ]
     ] = None
 

src/gradient/types/agent_update_params.py

@@ -14,6 +14,12 @@
 class AgentUpdateParams(TypedDict, total=False):
     agent_log_insights_enabled: bool
 
+    allowed_domains: SequenceNotStr[str]
+    """
+    Optional list of allowed domains for the chatbot - Must use fully qualified
+    domain name (FQDN) such as https://example.com
+    """
+
     anthropic_key_uuid: str
     """Optional anthropic key uuid for use with anthropic models"""
 
@@ -41,6 +47,9 @@ class AgentUpdateParams(TypedDict, total=False):
     response.
     """
 
+    model_provider_key_uuid: str
+    """Optional Model Provider uuid for use with provider models"""
+
     model_uuid: str
     """Identifier for the foundation model."""
 

src/gradient/types/api_agent.py

@@ -6,6 +6,8 @@
 from datetime import datetime
 from typing_extensions import Literal
 
+from pydantic import Field as FieldInfo
+
 from .._models import BaseModel
 from .api_agent_model import APIAgentModel
 from .api_knowledge_base import APIKnowledgeBase
@@ -24,6 +26,7 @@
     "Function",
     "Guardrail",
     "LoggingConfig",
+    "ModelProviderKey",
     "Template",
     "TemplateGuardrail",
 ]
@@ -35,6 +38,8 @@ class APIKey(BaseModel):
 
 
 class Chatbot(BaseModel):
+    allowed_domains: Optional[List[str]] = None
+
     button_background_color: Optional[str] = None
 
     logo: Optional[str] = None
@@ -72,6 +77,7 @@ class Deployment(BaseModel):
             "STATUS_UNDEPLOYING",
             "STATUS_UNDEPLOYMENT_FAILED",
             "STATUS_DELETED",
+            "STATUS_BUILDING",
         ]
     ] = None
 
@@ -186,6 +192,33 @@ class LoggingConfig(BaseModel):
     """Name of the log stream"""
 
 
+class ModelProviderKey(BaseModel):
+    api_key_uuid: Optional[str] = None
+    """API key ID"""
+
+    created_at: Optional[datetime] = None
+    """Key creation date"""
+
+    created_by: Optional[str] = None
+    """Created by user id from DO"""
+
+    deleted_at: Optional[datetime] = None
+    """Key deleted date"""
+
+    models: Optional[List[APIAgentModel]] = None
+    """Models supported by the openAI api key"""
+
+    name: Optional[str] = None
+    """Name of the key"""
+
+    provider: Optional[Literal["MODEL_PROVIDER_DIGITALOCEAN", "MODEL_PROVIDER_ANTHROPIC", "MODEL_PROVIDER_OPENAI"]] = (
+        None
+    )
+
+    updated_at: Optional[datetime] = None
+    """Key last updated date"""
+
+
 class TemplateGuardrail(BaseModel):
     priority: Optional[int] = None
     """Priority of the guardrail"""
@@ -311,6 +344,8 @@ class APIAgent(BaseModel):
     model: Optional[APIAgentModel] = None
     """Description of a Model"""
 
+    api_model_provider_key: Optional[ModelProviderKey] = FieldInfo(alias="model_provider_key", default=None)
+
     name: Optional[str] = None
     """Agent name"""
 
@@ -372,6 +407,11 @@ class APIAgent(BaseModel):
     version_hash: Optional[str] = None
     """The latest version of the agent"""
 
+    vpc_egress_ips: Optional[List[str]] = None
+    """VPC Egress IPs"""
+
+    vpc_uuid: Optional[str] = None
+
     workspace: Optional["APIWorkspace"] = None
 
 

src/gradient/types/knowledge_base_create_params.py

@@ -11,7 +11,7 @@
 from .knowledge_bases.api_file_upload_data_source_param import APIFileUploadDataSourceParam
 from .knowledge_bases.api_web_crawler_data_source_param import APIWebCrawlerDataSourceParam
 
-__all__ = ["KnowledgeBaseCreateParams", "Datasource", "DatasourceDropboxDataSource"]
+__all__ = ["KnowledgeBaseCreateParams", "Datasource", "DatasourceDropboxDataSource", "DatasourceGoogleDriveDataSource"]
 
 
 class KnowledgeBaseCreateParams(TypedDict, total=False):
@@ -63,6 +63,17 @@ class DatasourceDropboxDataSource(TypedDict, total=False):
     """
 
 
+class DatasourceGoogleDriveDataSource(TypedDict, total=False):
+    folder_id: str
+
+    refresh_token: str
+    """Refresh token.
+
+    you can obrain a refresh token by following the oauth2 flow. see
+    /v2/gen-ai/oauth2/google/tokens for reference.
+    """
+
+
 class Datasource(TypedDict, total=False):
     aws_data_source: AwsDataSourceParam
     """AWS S3 Data Source"""
@@ -79,6 +90,9 @@ class Datasource(TypedDict, total=False):
     file_upload_data_source: APIFileUploadDataSourceParam
     """File to upload as data source for knowledge base."""
 
+    google_drive_data_source: DatasourceGoogleDriveDataSource
+    """Google Drive Data Source"""
+
     item_path: str
 
     spaces_data_source: APISpacesDataSourceParam

src/gradient/types/knowledge_bases/api_indexed_data_source.py

@@ -48,6 +48,7 @@ class APIIndexedDataSource(BaseModel):
             "DATA_SOURCE_STATUS_PARTIALLY_UPDATED",
             "DATA_SOURCE_STATUS_NOT_UPDATED",
             "DATA_SOURCE_STATUS_FAILED",
+            "DATA_SOURCE_STATUS_CANCELLED",
         ]
     ] = None