diff --git a/.release-please-manifest.json b/.release-please-manifest.json index ae66b92e..d25059a8 100644 --- a/.release-please-manifest.json +++ b/.release-please-manifest.json @@ -1,3 +1,3 @@ { - ".": "3.0.0-beta.1" + ".": "3.0.0-beta.2" } \ No newline at end of file diff --git a/.stats.yml b/.stats.yml index 8c069c9b..9dda6e1d 100644 --- a/.stats.yml +++ b/.stats.yml @@ -1,4 +1,4 @@ configured_endpoints: 170 -openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/digitalocean%2Fgradient-015417b36365dfcb32166e67379c38de8bf5127c33dff646097a819a7b4dc588.yml -openapi_spec_hash: d7d811c13cc79f15d82fe680cf425859 -config_hash: 77ddef130940a6ad8ea6c6f66aee8757 +openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/digitalocean%2Fgradient-9aca3802735e1375125412aa28ac36bf2175144b8218610a73d2e7f775694dff.yml +openapi_spec_hash: e29d14e3e4679fcf22b3e760e49931b1 +config_hash: de89a9c8fde0120577d2aca8be4ae027 diff --git a/CHANGELOG.md b/CHANGELOG.md index c1c37c72..594ea379 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -1,5 +1,13 @@ # Changelog +## 3.0.0-beta.2 (2025-08-04) + +Full Changelog: [v3.0.0-beta.1...v3.0.0-beta.2](https://github.com/digitalocean/gradient-python/compare/v3.0.0-beta.1...v3.0.0-beta.2) + +### Features + +* **api:** collected updates 8/4 ([90ff9f2](https://github.com/digitalocean/gradient-python/commit/90ff9f227aa00805deb270e8e1de0ea9b56e3b4e)) + ## 3.0.0-beta.1 (2025-07-31) Full Changelog: [v0.1.0-beta.4...v3.0.0-beta.1](https://github.com/digitalocean/gradient-python/compare/v0.1.0-beta.4...v3.0.0-beta.1) diff --git a/pyproject.toml b/pyproject.toml index 6628f6fa..b21336d9 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [project] name = "gradient" -version = "3.0.0-beta.1" +version = "3.0.0-beta.2" description = "The official Python library for the Gradient API" dynamic = ["readme"] license = "Apache-2.0" diff --git a/src/gradient/_client.py b/src/gradient/_client.py index 9693f622..c25e3976 100644 --- a/src/gradient/_client.py +++ b/src/gradient/_client.py @@ -118,7 +118,7 @@ def __init__( base_url = os.environ.get("GRADIENT_BASE_URL") self._base_url_overridden = base_url is not None if base_url is None: - base_url = f"https://api.digitalocean.com/" + base_url = f"https://api.digitalocean.com" super().__init__( version=__version__, @@ -386,7 +386,7 @@ def __init__( base_url = os.environ.get("GRADIENT_BASE_URL") self._base_url_overridden = base_url is not None if base_url is None: - base_url = f"https://api.digitalocean.com/" + base_url = f"https://api.digitalocean.com" super().__init__( version=__version__, diff --git a/src/gradient/_version.py b/src/gradient/_version.py index f0d3c820..5bf5b229 100644 --- a/src/gradient/_version.py +++ b/src/gradient/_version.py @@ -1,4 +1,4 @@ # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details. __title__ = "gradient" -__version__ = "3.0.0-beta.1" # x-release-please-version +__version__ = "3.0.0-beta.2" # x-release-please-version diff --git a/src/gradient/resources/agents/agents.py b/src/gradient/resources/agents/agents.py index cff147c9..67f7f4ae 100644 --- a/src/gradient/resources/agents/agents.py +++ b/src/gradient/resources/agents/agents.py @@ -300,6 +300,7 @@ def update( self, path_uuid: str, *, + agent_log_insights_enabled: bool | NotGiven = NOT_GIVEN, anthropic_key_uuid: str | NotGiven = NOT_GIVEN, conversation_logs_enabled: bool | NotGiven = NOT_GIVEN, description: str | NotGiven = NOT_GIVEN, @@ -388,6 +389,7 @@ def update( else f"https://api.digitalocean.com/v2/gen-ai/agents/{path_uuid}", body=maybe_transform( { + "agent_log_insights_enabled": agent_log_insights_enabled, "anthropic_key_uuid": anthropic_key_uuid, "conversation_logs_enabled": conversation_logs_enabled, "description": description, @@ -741,6 +743,7 @@ async def update( self, path_uuid: str, *, + agent_log_insights_enabled: bool | NotGiven = NOT_GIVEN, anthropic_key_uuid: str | NotGiven = NOT_GIVEN, conversation_logs_enabled: bool | NotGiven = NOT_GIVEN, description: str | NotGiven = NOT_GIVEN, @@ -829,6 +832,7 @@ async def update( else f"https://api.digitalocean.com/v2/gen-ai/agents/{path_uuid}", body=await async_maybe_transform( { + "agent_log_insights_enabled": agent_log_insights_enabled, "anthropic_key_uuid": anthropic_key_uuid, "conversation_logs_enabled": conversation_logs_enabled, "description": description, diff --git a/src/gradient/types/agent_update_params.py b/src/gradient/types/agent_update_params.py index 5d2b5597..c26bf833 100644 --- a/src/gradient/types/agent_update_params.py +++ b/src/gradient/types/agent_update_params.py @@ -12,6 +12,8 @@ class AgentUpdateParams(TypedDict, total=False): + agent_log_insights_enabled: bool + anthropic_key_uuid: str """Optional anthropic key uuid for use with anthropic models""" diff --git a/src/gradient/types/api_agent.py b/src/gradient/types/api_agent.py index 4be22aa5..abfbe828 100644 --- a/src/gradient/types/api_agent.py +++ b/src/gradient/types/api_agent.py @@ -173,6 +173,12 @@ class LoggingConfig(BaseModel): galileo_project_name: Optional[str] = None """Name of the Galileo project""" + insights_enabled: Optional[bool] = None + """Whether insights are enabled""" + + insights_enabled_at: Optional[datetime] = None + """Timestamp when insights were enabled""" + log_stream_id: Optional[str] = None """Identifier for the log stream""" diff --git a/src/gradient/types/api_model.py b/src/gradient/types/api_model.py index 7c530ee2..e7f99bc1 100644 --- a/src/gradient/types/api_model.py +++ b/src/gradient/types/api_model.py @@ -11,6 +11,9 @@ class APIModel(BaseModel): + id: Optional[str] = None + """Human-readable model identifier""" + agreement: Optional[APIAgreement] = None """Agreement Description""" @@ -21,7 +24,7 @@ class APIModel(BaseModel): """True if it is a foundational model provided by do""" name: Optional[str] = None - """Name of the model""" + """Display name of the model""" parent_uuid: Optional[str] = None """Unique id of the model, this model is based on""" diff --git a/src/gradient/types/knowledge_base_create_params.py b/src/gradient/types/knowledge_base_create_params.py index 9ecd777d..5c0df9a6 100644 --- a/src/gradient/types/knowledge_base_create_params.py +++ b/src/gradient/types/knowledge_base_create_params.py @@ -10,7 +10,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"] +__all__ = ["KnowledgeBaseCreateParams", "Datasource", "DatasourceDropboxDataSource"] class KnowledgeBaseCreateParams(TypedDict, total=False): @@ -51,6 +51,17 @@ class KnowledgeBaseCreateParams(TypedDict, total=False): """The VPC to deploy the knowledge base database in""" +class DatasourceDropboxDataSource(TypedDict, total=False): + folder: str + + refresh_token: str + """Refresh token. + + you can obrain a refresh token by following the oauth2 flow. see + /v2/gen-ai/oauth2/dropbox/tokens for reference. + """ + + class Datasource(TypedDict, total=False): aws_data_source: AwsDataSourceParam """AWS S3 Data Source""" @@ -61,6 +72,9 @@ class Datasource(TypedDict, total=False): bucket_region: str """Deprecated, moved to data_source_details""" + dropbox_data_source: DatasourceDropboxDataSource + """Dropbox Data Source""" + file_upload_data_source: APIFileUploadDataSourceParam """File to upload as data source for knowledge base.""" diff --git a/src/gradient/types/knowledge_bases/api_indexing_job.py b/src/gradient/types/knowledge_bases/api_indexing_job.py index 240fd709..312e465c 100644 --- a/src/gradient/types/knowledge_bases/api_indexing_job.py +++ b/src/gradient/types/knowledge_bases/api_indexing_job.py @@ -55,6 +55,15 @@ class APIIndexingJob(BaseModel): total_datasources: Optional[int] = None """Number of datasources being indexed""" + total_items_failed: Optional[str] = None + """Total Items Failed""" + + total_items_indexed: Optional[str] = None + """Total Items Indexed""" + + total_items_skipped: Optional[str] = None + """Total Items Skipped""" + updated_at: Optional[datetime] = None """Last modified""" diff --git a/src/gradient/types/knowledge_bases/api_knowledge_base_data_source.py b/src/gradient/types/knowledge_bases/api_knowledge_base_data_source.py index a4d695d2..ed370eb5 100644 --- a/src/gradient/types/knowledge_bases/api_knowledge_base_data_source.py +++ b/src/gradient/types/knowledge_bases/api_knowledge_base_data_source.py @@ -10,7 +10,7 @@ from .api_file_upload_data_source import APIFileUploadDataSource from .api_web_crawler_data_source import APIWebCrawlerDataSource -__all__ = ["APIKnowledgeBaseDataSource", "AwsDataSource"] +__all__ = ["APIKnowledgeBaseDataSource", "AwsDataSource", "DropboxDataSource"] class AwsDataSource(BaseModel): @@ -23,6 +23,10 @@ class AwsDataSource(BaseModel): """Region of bucket""" +class DropboxDataSource(BaseModel): + folder: Optional[str] = None + + class APIKnowledgeBaseDataSource(BaseModel): aws_data_source: Optional[AwsDataSource] = None """AWS S3 Data Source for Display""" @@ -33,6 +37,9 @@ class APIKnowledgeBaseDataSource(BaseModel): created_at: Optional[datetime] = None """Creation date / time""" + dropbox_data_source: Optional[DropboxDataSource] = None + """Dropbox Data Source for Display""" + file_upload_data_source: Optional[APIFileUploadDataSource] = None """File to upload as data source for knowledge base.""" diff --git a/tests/api_resources/agents/evaluation_metrics/test_workspaces.py b/tests/api_resources/agents/evaluation_metrics/test_workspaces.py index 608406bf..f326c1e3 100644 --- a/tests/api_resources/agents/evaluation_metrics/test_workspaces.py +++ b/tests/api_resources/agents/evaluation_metrics/test_workspaces.py @@ -35,8 +35,8 @@ def test_method_create(self, client: Gradient) -> None: def test_method_create_with_all_params(self, client: Gradient) -> None: workspace = client.agents.evaluation_metrics.workspaces.create( agent_uuids=["example string"], - description='"example string"', - name='"example name"', + description="example string", + name="example name", ) assert_matches_type(WorkspaceCreateResponse, workspace, path=["response"]) @@ -117,9 +117,9 @@ def test_method_update(self, client: Gradient) -> None: def test_method_update_with_all_params(self, client: Gradient) -> None: workspace = client.agents.evaluation_metrics.workspaces.update( path_workspace_uuid='"123e4567-e89b-12d3-a456-426614174000"', - description='"example string"', - name='"example name"', - body_workspace_uuid='"123e4567-e89b-12d3-a456-426614174000"', + description="example string", + name="example name", + body_workspace_uuid="123e4567-e89b-12d3-a456-426614174000", ) assert_matches_type(WorkspaceUpdateResponse, workspace, path=["response"]) @@ -286,8 +286,8 @@ async def test_method_create(self, async_client: AsyncGradient) -> None: async def test_method_create_with_all_params(self, async_client: AsyncGradient) -> None: workspace = await async_client.agents.evaluation_metrics.workspaces.create( agent_uuids=["example string"], - description='"example string"', - name='"example name"', + description="example string", + name="example name", ) assert_matches_type(WorkspaceCreateResponse, workspace, path=["response"]) @@ -368,9 +368,9 @@ async def test_method_update(self, async_client: AsyncGradient) -> None: async def test_method_update_with_all_params(self, async_client: AsyncGradient) -> None: workspace = await async_client.agents.evaluation_metrics.workspaces.update( path_workspace_uuid='"123e4567-e89b-12d3-a456-426614174000"', - description='"example string"', - name='"example name"', - body_workspace_uuid='"123e4567-e89b-12d3-a456-426614174000"', + description="example string", + name="example name", + body_workspace_uuid="123e4567-e89b-12d3-a456-426614174000", ) assert_matches_type(WorkspaceUpdateResponse, workspace, path=["response"]) diff --git a/tests/api_resources/agents/evaluation_metrics/workspaces/test_agents.py b/tests/api_resources/agents/evaluation_metrics/workspaces/test_agents.py index b70f9c58..2d63ff65 100644 --- a/tests/api_resources/agents/evaluation_metrics/workspaces/test_agents.py +++ b/tests/api_resources/agents/evaluation_metrics/workspaces/test_agents.py @@ -87,7 +87,7 @@ def test_method_move_with_all_params(self, client: Gradient) -> None: agent = client.agents.evaluation_metrics.workspaces.agents.move( path_workspace_uuid='"123e4567-e89b-12d3-a456-426614174000"', agent_uuids=["example string"], - body_workspace_uuid='"123e4567-e89b-12d3-a456-426614174000"', + body_workspace_uuid="123e4567-e89b-12d3-a456-426614174000", ) assert_matches_type(AgentMoveResponse, agent, path=["response"]) @@ -198,7 +198,7 @@ async def test_method_move_with_all_params(self, async_client: AsyncGradient) -> agent = await async_client.agents.evaluation_metrics.workspaces.agents.move( path_workspace_uuid='"123e4567-e89b-12d3-a456-426614174000"', agent_uuids=["example string"], - body_workspace_uuid='"123e4567-e89b-12d3-a456-426614174000"', + body_workspace_uuid="123e4567-e89b-12d3-a456-426614174000", ) assert_matches_type(AgentMoveResponse, agent, path=["response"]) diff --git a/tests/api_resources/agents/test_evaluation_datasets.py b/tests/api_resources/agents/test_evaluation_datasets.py index f60c4720..3978ebdd 100644 --- a/tests/api_resources/agents/test_evaluation_datasets.py +++ b/tests/api_resources/agents/test_evaluation_datasets.py @@ -31,11 +31,11 @@ def test_method_create(self, client: Gradient) -> None: def test_method_create_with_all_params(self, client: Gradient) -> None: evaluation_dataset = client.agents.evaluation_datasets.create( file_upload_dataset={ - "original_file_name": '"example name"', - "size_in_bytes": '"12345"', - "stored_object_key": '"example string"', + "original_file_name": "example name", + "size_in_bytes": "12345", + "stored_object_key": "example string", }, - name='"example name"', + name="example name", ) assert_matches_type(EvaluationDatasetCreateResponse, evaluation_dataset, path=["response"]) @@ -75,7 +75,7 @@ def test_method_create_file_upload_presigned_urls_with_all_params(self, client: evaluation_dataset = client.agents.evaluation_datasets.create_file_upload_presigned_urls( files=[ { - "file_name": '"example name"', + "file_name": "example name", "file_size": "file_size", } ], @@ -127,11 +127,11 @@ async def test_method_create(self, async_client: AsyncGradient) -> None: async def test_method_create_with_all_params(self, async_client: AsyncGradient) -> None: evaluation_dataset = await async_client.agents.evaluation_datasets.create( file_upload_dataset={ - "original_file_name": '"example name"', - "size_in_bytes": '"12345"', - "stored_object_key": '"example string"', + "original_file_name": "example name", + "size_in_bytes": "12345", + "stored_object_key": "example string", }, - name='"example name"', + name="example name", ) assert_matches_type(EvaluationDatasetCreateResponse, evaluation_dataset, path=["response"]) @@ -171,7 +171,7 @@ async def test_method_create_file_upload_presigned_urls_with_all_params(self, as evaluation_dataset = await async_client.agents.evaluation_datasets.create_file_upload_presigned_urls( files=[ { - "file_name": '"example name"', + "file_name": "example name", "file_size": "file_size", } ], diff --git a/tests/api_resources/agents/test_evaluation_test_cases.py b/tests/api_resources/agents/test_evaluation_test_cases.py index 2860aa2c..b1d92580 100644 --- a/tests/api_resources/agents/test_evaluation_test_cases.py +++ b/tests/api_resources/agents/test_evaluation_test_cases.py @@ -33,17 +33,17 @@ def test_method_create(self, client: Gradient) -> None: @parametrize def test_method_create_with_all_params(self, client: Gradient) -> None: evaluation_test_case = client.agents.evaluation_test_cases.create( - dataset_uuid='"123e4567-e89b-12d3-a456-426614174000"', - description='"example string"', + dataset_uuid="123e4567-e89b-12d3-a456-426614174000", + description="example string", metrics=["example string"], - name='"example name"', + name="example name", star_metric={ - "metric_uuid": '"123e4567-e89b-12d3-a456-426614174000"', - "name": '"example name"', + "metric_uuid": "123e4567-e89b-12d3-a456-426614174000", + "name": "example name", "success_threshold": 123, "success_threshold_pct": 123, }, - workspace_uuid='"123e4567-e89b-12d3-a456-426614174000"', + workspace_uuid="123e4567-e89b-12d3-a456-426614174000", ) assert_matches_type(EvaluationTestCaseCreateResponse, evaluation_test_case, path=["response"]) @@ -133,17 +133,17 @@ def test_method_update(self, client: Gradient) -> None: def test_method_update_with_all_params(self, client: Gradient) -> None: evaluation_test_case = client.agents.evaluation_test_cases.update( path_test_case_uuid='"123e4567-e89b-12d3-a456-426614174000"', - dataset_uuid='"123e4567-e89b-12d3-a456-426614174000"', - description='"example string"', + dataset_uuid="123e4567-e89b-12d3-a456-426614174000", + description="example string", metrics={"metric_uuids": ["example string"]}, - name='"example name"', + name="example name", star_metric={ - "metric_uuid": '"123e4567-e89b-12d3-a456-426614174000"', - "name": '"example name"', + "metric_uuid": "123e4567-e89b-12d3-a456-426614174000", + "name": "example name", "success_threshold": 123, "success_threshold_pct": 123, }, - body_test_case_uuid='"123e4567-e89b-12d3-a456-426614174000"', + body_test_case_uuid="123e4567-e89b-12d3-a456-426614174000", ) assert_matches_type(EvaluationTestCaseUpdateResponse, evaluation_test_case, path=["response"]) @@ -278,17 +278,17 @@ async def test_method_create(self, async_client: AsyncGradient) -> None: @parametrize async def test_method_create_with_all_params(self, async_client: AsyncGradient) -> None: evaluation_test_case = await async_client.agents.evaluation_test_cases.create( - dataset_uuid='"123e4567-e89b-12d3-a456-426614174000"', - description='"example string"', + dataset_uuid="123e4567-e89b-12d3-a456-426614174000", + description="example string", metrics=["example string"], - name='"example name"', + name="example name", star_metric={ - "metric_uuid": '"123e4567-e89b-12d3-a456-426614174000"', - "name": '"example name"', + "metric_uuid": "123e4567-e89b-12d3-a456-426614174000", + "name": "example name", "success_threshold": 123, "success_threshold_pct": 123, }, - workspace_uuid='"123e4567-e89b-12d3-a456-426614174000"', + workspace_uuid="123e4567-e89b-12d3-a456-426614174000", ) assert_matches_type(EvaluationTestCaseCreateResponse, evaluation_test_case, path=["response"]) @@ -378,17 +378,17 @@ async def test_method_update(self, async_client: AsyncGradient) -> None: async def test_method_update_with_all_params(self, async_client: AsyncGradient) -> None: evaluation_test_case = await async_client.agents.evaluation_test_cases.update( path_test_case_uuid='"123e4567-e89b-12d3-a456-426614174000"', - dataset_uuid='"123e4567-e89b-12d3-a456-426614174000"', - description='"example string"', + dataset_uuid="123e4567-e89b-12d3-a456-426614174000", + description="example string", metrics={"metric_uuids": ["example string"]}, - name='"example name"', + name="example name", star_metric={ - "metric_uuid": '"123e4567-e89b-12d3-a456-426614174000"', - "name": '"example name"', + "metric_uuid": "123e4567-e89b-12d3-a456-426614174000", + "name": "example name", "success_threshold": 123, "success_threshold_pct": 123, }, - body_test_case_uuid='"123e4567-e89b-12d3-a456-426614174000"', + body_test_case_uuid="123e4567-e89b-12d3-a456-426614174000", ) assert_matches_type(EvaluationTestCaseUpdateResponse, evaluation_test_case, path=["response"]) diff --git a/tests/api_resources/knowledge_bases/test_data_sources.py b/tests/api_resources/knowledge_bases/test_data_sources.py index a5734cea..0e44b584 100644 --- a/tests/api_resources/knowledge_bases/test_data_sources.py +++ b/tests/api_resources/knowledge_bases/test_data_sources.py @@ -35,20 +35,20 @@ def test_method_create_with_all_params(self, client: Gradient) -> None: data_source = client.knowledge_bases.data_sources.create( path_knowledge_base_uuid='"123e4567-e89b-12d3-a456-426614174000"', aws_data_source={ - "bucket_name": '"example name"', - "item_path": '"example string"', - "key_id": '"123e4567-e89b-12d3-a456-426614174000"', - "region": '"example string"', - "secret_key": '"example string"', + "bucket_name": "example name", + "item_path": "example string", + "key_id": "123e4567-e89b-12d3-a456-426614174000", + "region": "example string", + "secret_key": "example string", }, body_knowledge_base_uuid='"12345678-1234-1234-1234-123456789012"', spaces_data_source={ - "bucket_name": '"example name"', - "item_path": '"example string"', - "region": '"example string"', + "bucket_name": "example name", + "item_path": "example string", + "region": "example string", }, web_crawler_data_source={ - "base_url": '"example string"', + "base_url": "example string", "crawling_option": "UNKNOWN", "embed_media": True, }, @@ -215,20 +215,20 @@ async def test_method_create_with_all_params(self, async_client: AsyncGradient) data_source = await async_client.knowledge_bases.data_sources.create( path_knowledge_base_uuid='"123e4567-e89b-12d3-a456-426614174000"', aws_data_source={ - "bucket_name": '"example name"', - "item_path": '"example string"', - "key_id": '"123e4567-e89b-12d3-a456-426614174000"', - "region": '"example string"', - "secret_key": '"example string"', + "bucket_name": "example name", + "item_path": "example string", + "key_id": "123e4567-e89b-12d3-a456-426614174000", + "region": "example string", + "secret_key": "example string", }, body_knowledge_base_uuid='"12345678-1234-1234-1234-123456789012"', spaces_data_source={ - "bucket_name": '"example name"', - "item_path": '"example string"', - "region": '"example string"', + "bucket_name": "example name", + "item_path": "example string", + "region": "example string", }, web_crawler_data_source={ - "base_url": '"example string"', + "base_url": "example string", "crawling_option": "UNKNOWN", "embed_media": True, }, diff --git a/tests/api_resources/test_agents.py b/tests/api_resources/test_agents.py index 987f2eda..77825f7e 100644 --- a/tests/api_resources/test_agents.py +++ b/tests/api_resources/test_agents.py @@ -124,6 +124,7 @@ def test_method_update(self, client: Gradient) -> None: def test_method_update_with_all_params(self, client: Gradient) -> None: agent = client.agents.update( path_uuid='"123e4567-e89b-12d3-a456-426614174000"', + agent_log_insights_enabled=True, anthropic_key_uuid='"12345678-1234-1234-1234-123456789012"', conversation_logs_enabled=True, description='"My Agent Description"', @@ -415,6 +416,7 @@ async def test_method_update(self, async_client: AsyncGradient) -> None: async def test_method_update_with_all_params(self, async_client: AsyncGradient) -> None: agent = await async_client.agents.update( path_uuid='"123e4567-e89b-12d3-a456-426614174000"', + agent_log_insights_enabled=True, anthropic_key_uuid='"12345678-1234-1234-1234-123456789012"', conversation_logs_enabled=True, description='"My Agent Description"', diff --git a/tests/api_resources/test_knowledge_bases.py b/tests/api_resources/test_knowledge_bases.py index 1628fdbe..23945480 100644 --- a/tests/api_resources/test_knowledge_bases.py +++ b/tests/api_resources/test_knowledge_bases.py @@ -37,27 +37,31 @@ def test_method_create_with_all_params(self, client: Gradient) -> None: datasources=[ { "aws_data_source": { - "bucket_name": '"example name"', - "item_path": '"example string"', - "key_id": '"123e4567-e89b-12d3-a456-426614174000"', - "region": '"example string"', - "secret_key": '"example string"', + "bucket_name": "example name", + "item_path": "example string", + "key_id": "123e4567-e89b-12d3-a456-426614174000", + "region": "example string", + "secret_key": "example string", + }, + "bucket_name": "example name", + "bucket_region": "example string", + "dropbox_data_source": { + "folder": "example string", + "refresh_token": "example string", }, - "bucket_name": '"example name"', - "bucket_region": '"example string"', "file_upload_data_source": { - "original_file_name": '"example name"', - "size_in_bytes": '"12345"', - "stored_object_key": '"example string"', + "original_file_name": "example name", + "size_in_bytes": "12345", + "stored_object_key": "example string", }, - "item_path": '"example string"', + "item_path": "example string", "spaces_data_source": { - "bucket_name": '"example name"', - "item_path": '"example string"', - "region": '"example string"', + "bucket_name": "example name", + "item_path": "example string", + "region": "example string", }, "web_crawler_data_source": { - "base_url": '"example string"', + "base_url": "example string", "crawling_option": "UNKNOWN", "embed_media": True, }, @@ -291,27 +295,31 @@ async def test_method_create_with_all_params(self, async_client: AsyncGradient) datasources=[ { "aws_data_source": { - "bucket_name": '"example name"', - "item_path": '"example string"', - "key_id": '"123e4567-e89b-12d3-a456-426614174000"', - "region": '"example string"', - "secret_key": '"example string"', + "bucket_name": "example name", + "item_path": "example string", + "key_id": "123e4567-e89b-12d3-a456-426614174000", + "region": "example string", + "secret_key": "example string", + }, + "bucket_name": "example name", + "bucket_region": "example string", + "dropbox_data_source": { + "folder": "example string", + "refresh_token": "example string", }, - "bucket_name": '"example name"', - "bucket_region": '"example string"', "file_upload_data_source": { - "original_file_name": '"example name"', - "size_in_bytes": '"12345"', - "stored_object_key": '"example string"', + "original_file_name": "example name", + "size_in_bytes": "12345", + "stored_object_key": "example string", }, - "item_path": '"example string"', + "item_path": "example string", "spaces_data_source": { - "bucket_name": '"example name"', - "item_path": '"example string"', - "region": '"example string"', + "bucket_name": "example name", + "item_path": "example string", + "region": "example string", }, "web_crawler_data_source": { - "base_url": '"example string"', + "base_url": "example string", "crawling_option": "UNKNOWN", "embed_media": True, },