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

do_gradientai/types/agents/evaluation_test_case_list_response.py

@@ -0,0 +1,12 @@
+# 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_test_case import APIEvaluationTestCase
+
+__all__ = ["EvaluationTestCaseListResponse"]
+
+
+class EvaluationTestCaseListResponse(BaseModel):
+    evaluation_test_cases: Optional[List[APIEvaluationTestCase]] = None

do_gradientai/types/agents/evaluation_test_case_retrieve_params.py

@@ -0,0 +1,12 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing_extensions import TypedDict
+
+__all__ = ["EvaluationTestCaseRetrieveParams"]
+
+
+class EvaluationTestCaseRetrieveParams(TypedDict, total=False):
+    evaluation_test_case_version: int
+    """Version of the test case."""

do_gradientai/types/agents/evaluation_test_case_retrieve_response.py

@@ -0,0 +1,12 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Optional
+
+from ..._models import BaseModel
+from .api_evaluation_test_case import APIEvaluationTestCase
+
+__all__ = ["EvaluationTestCaseRetrieveResponse"]
+
+
+class EvaluationTestCaseRetrieveResponse(BaseModel):
+    evaluation_test_case: Optional[APIEvaluationTestCase] = None

do_gradientai/types/agents/evaluation_test_case_update_params.py

@@ -0,0 +1,32 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import List
+from typing_extensions import Annotated, TypedDict
+
+from ..._utils import PropertyInfo
+from .api_star_metric_param import APIStarMetricParam
+
+__all__ = ["EvaluationTestCaseUpdateParams", "Metrics"]
+
+
+class EvaluationTestCaseUpdateParams(TypedDict, total=False):
+    dataset_uuid: str
+    """Dataset against which the test‑case is executed."""
+
+    description: str
+    """Description of the test case."""
+
+    metrics: Metrics
+
+    name: str
+    """Name of the test case."""
+
+    star_metric: APIStarMetricParam
+
+    body_test_case_uuid: Annotated[str, PropertyInfo(alias="test_case_uuid")]
+
+
+class Metrics(TypedDict, total=False):
+    metric_uuids: List[str]

do_gradientai/types/agents/evaluation_test_case_update_response.py

@@ -0,0 +1,14 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Optional
+
+from ..._models import BaseModel
+
+__all__ = ["EvaluationTestCaseUpdateResponse"]
+
+
+class EvaluationTestCaseUpdateResponse(BaseModel):
+    test_case_uuid: Optional[str] = None
+
+    version: Optional[int] = None
+    """The new verson of the test case."""

digitalocean_genai_sdk/types/agents/child_agent_add_params.py → do_gradientai/types/agents/route_add_params.py

@@ -6,10 +6,10 @@ from typing_extensions import Required, Annotated, TypedDict
 
 from ..._utils import PropertyInfo
 
-__all__ = ["ChildAgentAddParams"]
+__all__ = ["RouteAddParams"]
 
 
-class ChildAgentAddParams(TypedDict, total=False):
+class RouteAddParams(TypedDict, total=False):
     path_parent_agent_uuid: Required[Annotated[str, PropertyInfo(alias="parent_agent_uuid")]]
 
     body_child_agent_uuid: Annotated[str, PropertyInfo(alias="child_agent_uuid")]

digitalocean_genai_sdk/types/agents/child_agent_add_response.py → do_gradientai/types/agents/route_add_response.py

@@ -4,10 +4,10 @@ from typing import Optional
 
 from ..._models import BaseModel
 
-__all__ = ["ChildAgentAddResponse"]
+__all__ = ["RouteAddResponse"]
 
 
-class ChildAgentAddResponse(BaseModel):
+class RouteAddResponse(BaseModel):
     child_agent_uuid: Optional[str] = None
 
     parent_agent_uuid: Optional[str] = None

digitalocean_genai_sdk/types/agents/child_agent_delete_response.py → do_gradientai/types/agents/route_delete_response.py

@@ -4,10 +4,10 @@ from typing import Optional
 
 from ..._models import BaseModel
 
-__all__ = ["ChildAgentDeleteResponse"]
+__all__ = ["RouteDeleteResponse"]
 
 
-class ChildAgentDeleteResponse(BaseModel):
+class RouteDeleteResponse(BaseModel):
     child_agent_uuid: Optional[str] = None
 
     parent_agent_uuid: Optional[str] = None

digitalocean_genai_sdk/types/agents/child_agent_update_params.py → do_gradientai/types/agents/route_update_params.py

@@ -6,10 +6,10 @@ from typing_extensions import Required, Annotated, TypedDict
 
 from ..._utils import PropertyInfo
 
-__all__ = ["ChildAgentUpdateParams"]
+__all__ = ["RouteUpdateParams"]
 
 
-class ChildAgentUpdateParams(TypedDict, total=False):
+class RouteUpdateParams(TypedDict, total=False):
     path_parent_agent_uuid: Required[Annotated[str, PropertyInfo(alias="parent_agent_uuid")]]
 
     body_child_agent_uuid: Annotated[str, PropertyInfo(alias="child_agent_uuid")]

digitalocean_genai_sdk/types/agents/child_agent_update_response.py → do_gradientai/types/agents/route_update_response.py

@@ -4,10 +4,10 @@ from typing import Optional
 
 from ..._models import BaseModel
 
-__all__ = ["ChildAgentUpdateResponse"]
+__all__ = ["RouteUpdateResponse"]
 
 
-class ChildAgentUpdateResponse(BaseModel):
+class RouteUpdateResponse(BaseModel):
     child_agent_uuid: Optional[str] = None
 
     parent_agent_uuid: Optional[str] = None

digitalocean_genai_sdk/types/agents/child_agent_view_response.py → do_gradientai/types/agents/route_view_response.py

@@ -6,10 +6,10 @@ from typing import List, Optional
 
 from ..._models import BaseModel
 
-__all__ = ["ChildAgentViewResponse"]
+__all__ = ["RouteViewResponse"]
 
 
-class ChildAgentViewResponse(BaseModel):
+class RouteViewResponse(BaseModel):
     children: Optional[List["APIAgent"]] = None
 
 

{digitalocean_genai_sdk → do_gradientai}/types/agents/version_list_response.py

@@ -5,9 +5,9 @@ from datetime import datetime
 
 from pydantic import Field as FieldInfo
 
-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_retrieval_method import APIRetrievalMethod
 
 __all__ = [

{digitalocean_genai_sdk → do_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 → do_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

{digitalocean_genai_sdk → do_gradientai}/types/api_knowledge_base.py

@@ -4,7 +4,7 @@ from typing import List, Optional
 from datetime import datetime
 
 from .._models import BaseModel
-from .api_indexing_job import APIIndexingJob
+from .knowledge_bases.api_indexing_job import APIIndexingJob
 
 __all__ = ["APIKnowledgeBase"]
 

{digitalocean_genai_sdk → do_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
 

do_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 .agents.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

do_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 → do_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"]
 

digitalocean_genai_sdk/types/chat_create_completion_params.py → do_gradientai/types/chat/completion_create_params.py

@@ -5,22 +5,18 @@ from __future__ import annotations
 from typing import Dict, List, Union, Iterable, Optional
 from typing_extensions import Literal, Required, TypeAlias, TypedDict
 
-from .chat_completion_request_message_content_part_text_param import ChatCompletionRequestMessageContentPartTextParam
-
 __all__ = [
-    "ChatCreateCompletionParams",
+    "CompletionCreateParams",
     "Message",
     "MessageChatCompletionRequestSystemMessage",
     "MessageChatCompletionRequestDeveloperMessage",
     "MessageChatCompletionRequestUserMessage",
     "MessageChatCompletionRequestAssistantMessage",
-    "MessageChatCompletionRequestAssistantMessageContentArrayOfContentPart",
-    "MessageChatCompletionRequestAssistantMessageContentArrayOfContentPartChatCompletionRequestMessageContentPartRefusal",
     "StreamOptions",
 ]
 
 
-class ChatCreateCompletionParams(TypedDict, total=False):
+class CompletionCreateParams(TypedDict, total=False):
     messages: Required[Iterable[Message]]
     """A list of messages comprising the conversation so far."""
 
@@ -137,7 +133,7 @@ class ChatCreateCompletionParams(TypedDict, total=False):
 
 
 class MessageChatCompletionRequestSystemMessage(TypedDict, total=False):
-    content: Required[Union[str, Iterable[ChatCompletionRequestMessageContentPartTextParam]]]
+    content: Required[Union[str, List[str]]]
     """The contents of the system message."""
 
     role: Required[Literal["system"]]
@@ -145,7 +141,7 @@ class MessageChatCompletionRequestSystemMessage(TypedDict, total=False):
 
 
 class MessageChatCompletionRequestDeveloperMessage(TypedDict, total=False):
-    content: Required[Union[str, Iterable[ChatCompletionRequestMessageContentPartTextParam]]]
+    content: Required[Union[str, List[str]]]
     """The contents of the developer message."""
 
     role: Required[Literal["developer"]]
@@ -153,39 +149,20 @@ class MessageChatCompletionRequestDeveloperMessage(TypedDict, total=False):
 
 
 class MessageChatCompletionRequestUserMessage(TypedDict, total=False):
-    content: Required[Union[str, Iterable[ChatCompletionRequestMessageContentPartTextParam]]]
+    content: Required[Union[str, List[str]]]
     """The contents of the user message."""
 
     role: Required[Literal["user"]]
     """The role of the messages author, in this case `user`."""
 
 
-class MessageChatCompletionRequestAssistantMessageContentArrayOfContentPartChatCompletionRequestMessageContentPartRefusal(
-    TypedDict, total=False
-):
-    refusal: Required[str]
-    """The refusal message generated by the model."""
-
-    type: Required[Literal["refusal"]]
-    """The type of the content part."""
-
-
-MessageChatCompletionRequestAssistantMessageContentArrayOfContentPart: TypeAlias = Union[
-    ChatCompletionRequestMessageContentPartTextParam,
-    MessageChatCompletionRequestAssistantMessageContentArrayOfContentPartChatCompletionRequestMessageContentPartRefusal,
-]
-
-
 class MessageChatCompletionRequestAssistantMessage(TypedDict, total=False):
     role: Required[Literal["assistant"]]
     """The role of the messages author, in this case `assistant`."""
 
-    content: Union[str, Iterable[MessageChatCompletionRequestAssistantMessageContentArrayOfContentPart], None]
+    content: Union[str, List[str], None]
     """The contents of the assistant message."""
 
-    refusal: Optional[str]
-    """The refusal message by the assistant."""
-
 
 Message: TypeAlias = Union[
     MessageChatCompletionRequestSystemMessage,

digitalocean_genai_sdk/types/chat_create_completion_response.py → do_gradientai/types/chat/completion_create_response.py

@@ -3,10 +3,10 @@
 from typing import List, Optional
 from typing_extensions import Literal
 
-from .._models import BaseModel
+from ..._models import BaseModel
 from .chat_completion_token_logprob import ChatCompletionTokenLogprob
 
-__all__ = ["ChatCreateCompletionResponse", "Choice", "ChoiceLogprobs", "ChoiceMessage", "Usage"]
+__all__ = ["CompletionCreateResponse", "Choice", "ChoiceLogprobs", "ChoiceMessage", "Usage"]
 
 
 class ChoiceLogprobs(BaseModel):
@@ -58,7 +58,7 @@ class Usage(BaseModel):
     """Total number of tokens used in the request (prompt + completion)."""
 
 
-class ChatCreateCompletionResponse(BaseModel):
+class CompletionCreateResponse(BaseModel):
     id: str
     """A unique identifier for the chat completion."""
 

{digitalocean_genai_sdk/types/api_keys → do_gradientai/types/inference}/api_key_list_response.py

@@ -3,8 +3,8 @@
 from typing import List, Optional
 
 from ..._models import BaseModel
-from ..agents.api_meta import APIMeta
-from ..agents.api_links import APILinks
+from ..shared.api_meta import APIMeta
+from ..shared.api_links import APILinks
 from .api_model_api_key_info import APIModelAPIKeyInfo
 
 __all__ = ["APIKeyListResponse"]

{digitalocean_genai_sdk → do_gradientai}/types/knowledge_base_create_params.py

@@ -5,6 +5,7 @@ from __future__ import annotations
 from typing import List, Iterable
 from typing_extensions import TypedDict
 
+from .knowledge_bases.aws_data_source_param import AwsDataSourceParam
 from .knowledge_bases.api_spaces_data_source_param import APISpacesDataSourceParam
 from .knowledge_bases.api_file_upload_data_source_param import APIFileUploadDataSourceParam
 from .knowledge_bases.api_web_crawler_data_source_param import APIWebCrawlerDataSourceParam
@@ -50,6 +51,8 @@ class KnowledgeBaseCreateParams(TypedDict, total=False):
 
 
 class Datasource(TypedDict, total=False):
+    aws_data_source: AwsDataSourceParam
+
     bucket_name: str
 
     bucket_region: str

{digitalocean_genai_sdk → do_gradientai}/types/knowledge_base_list_response.py

@@ -3,8 +3,8 @@
 from typing import List, Optional
 
 from .._models import BaseModel
-from .agents.api_meta import APIMeta
-from .agents.api_links import APILinks
+from .shared.api_meta import APIMeta
+from .shared.api_links import APILinks
 from .api_knowledge_base import APIKnowledgeBase
 
 __all__ = ["KnowledgeBaseListResponse"]

{digitalocean_genai_sdk → do_gradientai}/types/knowledge_bases/__init__.py

@@ -2,15 +2,28 @@
 
 from __future__ import annotations
 
+from .api_indexing_job import APIIndexingJob as APIIndexingJob
+from .aws_data_source_param import AwsDataSourceParam as AwsDataSourceParam
 from .api_spaces_data_source import APISpacesDataSource as APISpacesDataSource
+from .api_indexed_data_source import APIIndexedDataSource as APIIndexedDataSource
 from .data_source_list_params import DataSourceListParams as DataSourceListParams
+from .indexing_job_list_params import IndexingJobListParams as IndexingJobListParams
 from .data_source_create_params import DataSourceCreateParams as DataSourceCreateParams
 from .data_source_list_response import DataSourceListResponse as DataSourceListResponse
+from .indexing_job_create_params import IndexingJobCreateParams as IndexingJobCreateParams
+from .indexing_job_list_response import IndexingJobListResponse as IndexingJobListResponse
 from .api_file_upload_data_source import APIFileUploadDataSource as APIFileUploadDataSource
 from .api_web_crawler_data_source import APIWebCrawlerDataSource as APIWebCrawlerDataSource
 from .data_source_create_response import DataSourceCreateResponse as DataSourceCreateResponse
 from .data_source_delete_response import DataSourceDeleteResponse as DataSourceDeleteResponse
 from .api_spaces_data_source_param import APISpacesDataSourceParam as APISpacesDataSourceParam
+from .indexing_job_create_response import IndexingJobCreateResponse as IndexingJobCreateResponse
 from .api_knowledge_base_data_source import APIKnowledgeBaseDataSource as APIKnowledgeBaseDataSource
+from .indexing_job_retrieve_response import IndexingJobRetrieveResponse as IndexingJobRetrieveResponse
 from .api_file_upload_data_source_param import APIFileUploadDataSourceParam as APIFileUploadDataSourceParam
 from .api_web_crawler_data_source_param import APIWebCrawlerDataSourceParam as APIWebCrawlerDataSourceParam
+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,
+)

digitalocean_genai_sdk/types/indexing_job_retrieve_data_sources_response.py → do_gradientai/types/knowledge_bases/api_indexed_data_source.py

@@ -1,15 +1,15 @@
 # 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 typing_extensions import Literal
 
-from .._models import BaseModel
+from ..._models import BaseModel
 
-__all__ = ["IndexingJobRetrieveDataSourcesResponse", "IndexedDataSource"]
+__all__ = ["APIIndexedDataSource"]
 
 
-class IndexedDataSource(BaseModel):
+class APIIndexedDataSource(BaseModel):
     completed_at: Optional[datetime] = None
 
     data_source_uuid: Optional[str] = None
@@ -46,7 +46,3 @@ class IndexedDataSource(BaseModel):
     total_bytes_indexed: Optional[str] = None
 
     total_file_count: Optional[str] = None
-
-
-class IndexingJobRetrieveDataSourcesResponse(BaseModel):
-    indexed_data_sources: Optional[List[IndexedDataSource]] = None

{digitalocean_genai_sdk/types → do_gradientai/types/knowledge_bases}/api_indexing_job.py

@@ -4,7 +4,7 @@ from typing import List, Optional
 from datetime import datetime
 from typing_extensions import Literal
 
-from .._models import BaseModel
+from ..._models import BaseModel
 
 __all__ = ["APIIndexingJob"]
 
@@ -34,6 +34,18 @@ class APIIndexingJob(BaseModel):
 
     started_at: Optional[datetime] = None
 
+    status: Optional[
+        Literal[
+            "INDEX_JOB_STATUS_UNKNOWN",
+            "INDEX_JOB_STATUS_PARTIAL",
+            "INDEX_JOB_STATUS_IN_PROGRESS",
+            "INDEX_JOB_STATUS_COMPLETED",
+            "INDEX_JOB_STATUS_FAILED",
+            "INDEX_JOB_STATUS_NO_CHANGES",
+            "INDEX_JOB_STATUS_PENDING",
+        ]
+    ] = None
+
     tokens: Optional[int] = None
 
     total_datasources: Optional[int] = None

{digitalocean_genai_sdk → do_gradientai}/types/knowledge_bases/api_knowledge_base_data_source.py

@@ -4,15 +4,26 @@ from typing import Optional
 from datetime import datetime
 
 from ..._models import BaseModel
-from ..api_indexing_job import APIIndexingJob
+from .api_indexing_job import APIIndexingJob
 from .api_spaces_data_source import APISpacesDataSource
+from .api_indexed_data_source import APIIndexedDataSource
 from .api_file_upload_data_source import APIFileUploadDataSource
 from .api_web_crawler_data_source import APIWebCrawlerDataSource
 
-__all__ = ["APIKnowledgeBaseDataSource"]
+__all__ = ["APIKnowledgeBaseDataSource", "AwsDataSource"]
+
+
+class AwsDataSource(BaseModel):
+    bucket_name: Optional[str] = None
+
+    item_path: Optional[str] = None
+
+    region: Optional[str] = None
 
 
 class APIKnowledgeBaseDataSource(BaseModel):
+    aws_data_source: Optional[AwsDataSource] = None
+
     bucket_name: Optional[str] = None
 
     created_at: Optional[datetime] = None
@@ -22,6 +33,8 @@ class APIKnowledgeBaseDataSource(BaseModel):
 
     item_path: Optional[str] = None
 
+    last_datasource_indexing_job: Optional[APIIndexedDataSource] = None
+
     last_indexing_job: Optional[APIIndexingJob] = None
 
     region: Optional[str] = None

do_gradientai/types/knowledge_bases/aws_data_source_param.py

@@ -0,0 +1,19 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing_extensions import TypedDict
+
+__all__ = ["AwsDataSourceParam"]
+
+
+class AwsDataSourceParam(TypedDict, total=False):
+    bucket_name: str
+
+    item_path: str
+
+    key_id: str
+
+    region: str
+
+    secret_key: str

{digitalocean_genai_sdk → do_gradientai}/types/knowledge_bases/data_source_create_params.py

@@ -5,29 +5,18 @@ from __future__ import annotations
 from typing_extensions import Annotated, TypedDict
 
 from ..._utils import PropertyInfo
+from .aws_data_source_param import AwsDataSourceParam
 from .api_spaces_data_source_param import APISpacesDataSourceParam
 from .api_web_crawler_data_source_param import APIWebCrawlerDataSourceParam
 
-__all__ = ["DataSourceCreateParams", "AwsDataSource"]
+__all__ = ["DataSourceCreateParams"]
 
 
 class DataSourceCreateParams(TypedDict, total=False):
-    aws_data_source: AwsDataSource
+    aws_data_source: AwsDataSourceParam
 
     body_knowledge_base_uuid: Annotated[str, PropertyInfo(alias="knowledge_base_uuid")]
 
     spaces_data_source: APISpacesDataSourceParam
 
     web_crawler_data_source: APIWebCrawlerDataSourceParam
-
-
-class AwsDataSource(TypedDict, total=False):
-    bucket_name: str
-
-    item_path: str
-
-    key_id: str
-
-    region: str
-
-    secret_key: str

{digitalocean_genai_sdk → do_gradientai}/types/knowledge_bases/data_source_list_response.py

@@ -3,8 +3,8 @@
 from typing import List, Optional
 
 from ..._models import BaseModel
-from ..agents.api_meta import APIMeta
-from ..agents.api_links import APILinks
+from ..shared.api_meta import APIMeta
+from ..shared.api_links import APILinks
 from .api_knowledge_base_data_source import APIKnowledgeBaseDataSource
 
 __all__ = ["DataSourceListResponse"]