feat(api): update via SDK Studio

Author: stainless-app[bot]

.stats.yml

@@ -1,4 +1,4 @@
-configured_endpoints: 174
+configured_endpoints: 175
 openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/digitalocean%2Fgradient-40c154e2fdc4fef9ca1cf8329c29b7102bf324acfc9589571d6f3452d1ca579c.yml
 openapi_spec_hash: 83a3d092965fde776b29b61f785459f9
-config_hash: 3752c9f438752ab61938a8921bb3befd
+config_hash: dd3a0f16fb9e072bb63c570b14beccd2

api.md

@@ -852,6 +852,7 @@ from gradient.types.knowledge_bases import (
     DataSourceCreateResponse,
     DataSourceListResponse,
     DataSourceDeleteResponse,
+    DataSourceCreatePresignedURLsResponse,
 )
 ```
 
@@ -860,6 +861,7 @@ Methods:
 - <code title="post /v2/gen-ai/knowledge_bases/{knowledge_base_uuid}/data_sources">client.knowledge_bases.data_sources.<a href="./src/gradient/resources/knowledge_bases/data_sources.py">create</a>(path_knowledge_base_uuid, \*\*<a href="src/gradient/types/knowledge_bases/data_source_create_params.py">params</a>) -> <a href="./src/gradient/types/knowledge_bases/data_source_create_response.py">DataSourceCreateResponse</a></code>
 - <code title="get /v2/gen-ai/knowledge_bases/{knowledge_base_uuid}/data_sources">client.knowledge_bases.data_sources.<a href="./src/gradient/resources/knowledge_bases/data_sources.py">list</a>(knowledge_base_uuid, \*\*<a href="src/gradient/types/knowledge_bases/data_source_list_params.py">params</a>) -> <a href="./src/gradient/types/knowledge_bases/data_source_list_response.py">DataSourceListResponse</a></code>
 - <code title="delete /v2/gen-ai/knowledge_bases/{knowledge_base_uuid}/data_sources/{data_source_uuid}">client.knowledge_bases.data_sources.<a href="./src/gradient/resources/knowledge_bases/data_sources.py">delete</a>(data_source_uuid, \*, knowledge_base_uuid) -> <a href="./src/gradient/types/knowledge_bases/data_source_delete_response.py">DataSourceDeleteResponse</a></code>
+- <code title="post /v2/gen-ai/knowledge_bases/data_sources/file_upload_presigned_urls">client.knowledge_bases.data_sources.<a href="./src/gradient/resources/knowledge_bases/data_sources.py">create_presigned_urls</a>(\*\*<a href="src/gradient/types/knowledge_bases/data_source_create_presigned_urls_params.py">params</a>) -> <a href="./src/gradient/types/knowledge_bases/data_source_create_presigned_urls_response.py">DataSourceCreatePresignedURLsResponse</a></code>
 
 ## IndexingJobs
 

src/gradient/resources/knowledge_bases/data_sources.py

@@ -2,6 +2,8 @@
 
 from __future__ import annotations
 
+from typing import Iterable
+
 import httpx
 
 from ..._types import Body, Omit, Query, Headers, NotGiven, omit, not_given
@@ -18,13 +20,15 @@
 from ...types.knowledge_bases import (
     data_source_list_params,
     data_source_create_params,
+    data_source_create_presigned_urls_params,
 )
 from ...types.knowledge_bases.aws_data_source_param import AwsDataSourceParam
 from ...types.knowledge_bases.data_source_list_response import DataSourceListResponse
 from ...types.knowledge_bases.data_source_create_response import DataSourceCreateResponse
 from ...types.knowledge_bases.data_source_delete_response import DataSourceDeleteResponse
 from ...types.knowledge_bases.api_spaces_data_source_param import APISpacesDataSourceParam
 from ...types.knowledge_bases.api_web_crawler_data_source_param import APIWebCrawlerDataSourceParam
+from ...types.knowledge_bases.data_source_create_presigned_urls_response import DataSourceCreatePresignedURLsResponse
 
 __all__ = ["DataSourcesResource", "AsyncDataSourcesResource"]
 
@@ -203,6 +207,45 @@ def delete(
             cast_to=DataSourceDeleteResponse,
         )
 
+    def create_presigned_urls(
+        self,
+        *,
+        files: Iterable[data_source_create_presigned_urls_params.File] | 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,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
+    ) -> DataSourceCreatePresignedURLsResponse:
+        """
+        To create presigned URLs for knowledge base data source file upload, send a POST
+        request to `/v2/gen-ai/knowledge_bases/data_sources/file_upload_presigned_urls`.
+
+        Args:
+          files: A list of files to generate presigned URLs for.
+
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        return self._post(
+            "/v2/gen-ai/knowledge_bases/data_sources/file_upload_presigned_urls"
+            if self._client._base_url_overridden
+            else "https://api.digitalocean.com/v2/gen-ai/knowledge_bases/data_sources/file_upload_presigned_urls",
+            body=maybe_transform(
+                {"files": files}, data_source_create_presigned_urls_params.DataSourceCreatePresignedURLsParams
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=DataSourceCreatePresignedURLsResponse,
+        )
+
 
 class AsyncDataSourcesResource(AsyncAPIResource):
     @cached_property
@@ -378,6 +421,45 @@ async def delete(
             cast_to=DataSourceDeleteResponse,
         )
 
+    async def create_presigned_urls(
+        self,
+        *,
+        files: Iterable[data_source_create_presigned_urls_params.File] | 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,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
+    ) -> DataSourceCreatePresignedURLsResponse:
+        """
+        To create presigned URLs for knowledge base data source file upload, send a POST
+        request to `/v2/gen-ai/knowledge_bases/data_sources/file_upload_presigned_urls`.
+
+        Args:
+          files: A list of files to generate presigned URLs for.
+
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        return await self._post(
+            "/v2/gen-ai/knowledge_bases/data_sources/file_upload_presigned_urls"
+            if self._client._base_url_overridden
+            else "https://api.digitalocean.com/v2/gen-ai/knowledge_bases/data_sources/file_upload_presigned_urls",
+            body=await async_maybe_transform(
+                {"files": files}, data_source_create_presigned_urls_params.DataSourceCreatePresignedURLsParams
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=DataSourceCreatePresignedURLsResponse,
+        )
+
 
 class DataSourcesResourceWithRawResponse:
     def __init__(self, data_sources: DataSourcesResource) -> None:
@@ -392,6 +474,9 @@ def __init__(self, data_sources: DataSourcesResource) -> None:
         self.delete = to_raw_response_wrapper(
             data_sources.delete,
         )
+        self.create_presigned_urls = to_raw_response_wrapper(
+            data_sources.create_presigned_urls,
+        )
 
 
 class AsyncDataSourcesResourceWithRawResponse:
@@ -407,6 +492,9 @@ def __init__(self, data_sources: AsyncDataSourcesResource) -> None:
         self.delete = async_to_raw_response_wrapper(
             data_sources.delete,
         )
+        self.create_presigned_urls = async_to_raw_response_wrapper(
+            data_sources.create_presigned_urls,
+        )
 
 
 class DataSourcesResourceWithStreamingResponse:
@@ -422,6 +510,9 @@ def __init__(self, data_sources: DataSourcesResource) -> None:
         self.delete = to_streamed_response_wrapper(
             data_sources.delete,
         )
+        self.create_presigned_urls = to_streamed_response_wrapper(
+            data_sources.create_presigned_urls,
+        )
 
 
 class AsyncDataSourcesResourceWithStreamingResponse:
@@ -437,3 +528,6 @@ def __init__(self, data_sources: AsyncDataSourcesResource) -> None:
         self.delete = async_to_streamed_response_wrapper(
             data_sources.delete,
         )
+        self.create_presigned_urls = async_to_streamed_response_wrapper(
+            data_sources.create_presigned_urls,
+        )

src/gradient/types/knowledge_bases/__init__.py

@@ -24,6 +24,12 @@
 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 .data_source_create_presigned_urls_params import (
+    DataSourceCreatePresignedURLsParams as DataSourceCreatePresignedURLsParams,
+)
+from .data_source_create_presigned_urls_response import (
+    DataSourceCreatePresignedURLsResponse as DataSourceCreatePresignedURLsResponse,
+)
 from .indexing_job_retrieve_data_sources_response import (
     IndexingJobRetrieveDataSourcesResponse as IndexingJobRetrieveDataSourcesResponse,
 )

src/gradient/types/knowledge_bases/data_source_create_presigned_urls_params.py

@@ -0,0 +1,21 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import Iterable
+from typing_extensions import TypedDict
+
+__all__ = ["DataSourceCreatePresignedURLsParams", "File"]
+
+
+class DataSourceCreatePresignedURLsParams(TypedDict, total=False):
+    files: Iterable[File]
+    """A list of files to generate presigned URLs for."""
+
+
+class File(TypedDict, total=False):
+    file_name: str
+    """Local filename"""
+
+    file_size: str
+    """The size of the file in bytes."""

src/gradient/types/knowledge_bases/data_source_create_presigned_urls_response.py

@@ -0,0 +1,30 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import List, Optional
+from datetime import datetime
+
+from ..._models import BaseModel
+
+__all__ = ["DataSourceCreatePresignedURLsResponse", "Upload"]
+
+
+class Upload(BaseModel):
+    expires_at: Optional[datetime] = None
+    """The time the url expires at."""
+
+    object_key: Optional[str] = None
+    """The unique object key to store the file as."""
+
+    original_file_name: Optional[str] = None
+    """The original file name."""
+
+    presigned_url: Optional[str] = None
+    """The actual presigned URL the client can use to upload the file directly."""
+
+
+class DataSourceCreatePresignedURLsResponse(BaseModel):
+    request_id: Optional[str] = None
+    """The ID generated for the request for Presigned URLs."""
+
+    uploads: Optional[List[Upload]] = None
+    """A list of generated presigned URLs and object keys, one per file."""

tests/api_resources/knowledge_bases/test_data_sources.py

@@ -13,6 +13,7 @@
     DataSourceListResponse,
     DataSourceCreateResponse,
     DataSourceDeleteResponse,
+    DataSourceCreatePresignedURLsResponse,
 )
 
 base_url = os.environ.get("TEST_API_BASE_URL", "http://127.0.0.1:4010")
@@ -195,6 +196,47 @@ def test_path_params_delete(self, client: Gradient) -> None:
                 knowledge_base_uuid='"123e4567-e89b-12d3-a456-426614174000"',
             )
 
+    @pytest.mark.skip(reason="Prism tests are disabled")
+    @parametrize
+    def test_method_create_presigned_urls(self, client: Gradient) -> None:
+        data_source = client.knowledge_bases.data_sources.create_presigned_urls()
+        assert_matches_type(DataSourceCreatePresignedURLsResponse, data_source, path=["response"])
+
+    @pytest.mark.skip(reason="Prism tests are disabled")
+    @parametrize
+    def test_method_create_presigned_urls_with_all_params(self, client: Gradient) -> None:
+        data_source = client.knowledge_bases.data_sources.create_presigned_urls(
+            files=[
+                {
+                    "file_name": "example name",
+                    "file_size": "file_size",
+                }
+            ],
+        )
+        assert_matches_type(DataSourceCreatePresignedURLsResponse, data_source, path=["response"])
+
+    @pytest.mark.skip(reason="Prism tests are disabled")
+    @parametrize
+    def test_raw_response_create_presigned_urls(self, client: Gradient) -> None:
+        response = client.knowledge_bases.data_sources.with_raw_response.create_presigned_urls()
+
+        assert response.is_closed is True
+        assert response.http_request.headers.get("X-Stainless-Lang") == "python"
+        data_source = response.parse()
+        assert_matches_type(DataSourceCreatePresignedURLsResponse, data_source, path=["response"])
+
+    @pytest.mark.skip(reason="Prism tests are disabled")
+    @parametrize
+    def test_streaming_response_create_presigned_urls(self, client: Gradient) -> None:
+        with client.knowledge_bases.data_sources.with_streaming_response.create_presigned_urls() as response:
+            assert not response.is_closed
+            assert response.http_request.headers.get("X-Stainless-Lang") == "python"
+
+            data_source = response.parse()
+            assert_matches_type(DataSourceCreatePresignedURLsResponse, data_source, path=["response"])
+
+        assert cast(Any, response.is_closed) is True
+
 
 class TestAsyncDataSources:
     parametrize = pytest.mark.parametrize(
@@ -374,3 +416,46 @@ async def test_path_params_delete(self, async_client: AsyncGradient) -> None:
                 data_source_uuid="",
                 knowledge_base_uuid='"123e4567-e89b-12d3-a456-426614174000"',
             )
+
+    @pytest.mark.skip(reason="Prism tests are disabled")
+    @parametrize
+    async def test_method_create_presigned_urls(self, async_client: AsyncGradient) -> None:
+        data_source = await async_client.knowledge_bases.data_sources.create_presigned_urls()
+        assert_matches_type(DataSourceCreatePresignedURLsResponse, data_source, path=["response"])
+
+    @pytest.mark.skip(reason="Prism tests are disabled")
+    @parametrize
+    async def test_method_create_presigned_urls_with_all_params(self, async_client: AsyncGradient) -> None:
+        data_source = await async_client.knowledge_bases.data_sources.create_presigned_urls(
+            files=[
+                {
+                    "file_name": "example name",
+                    "file_size": "file_size",
+                }
+            ],
+        )
+        assert_matches_type(DataSourceCreatePresignedURLsResponse, data_source, path=["response"])
+
+    @pytest.mark.skip(reason="Prism tests are disabled")
+    @parametrize
+    async def test_raw_response_create_presigned_urls(self, async_client: AsyncGradient) -> None:
+        response = await async_client.knowledge_bases.data_sources.with_raw_response.create_presigned_urls()
+
+        assert response.is_closed is True
+        assert response.http_request.headers.get("X-Stainless-Lang") == "python"
+        data_source = await response.parse()
+        assert_matches_type(DataSourceCreatePresignedURLsResponse, data_source, path=["response"])
+
+    @pytest.mark.skip(reason="Prism tests are disabled")
+    @parametrize
+    async def test_streaming_response_create_presigned_urls(self, async_client: AsyncGradient) -> None:
+        async with (
+            async_client.knowledge_bases.data_sources.with_streaming_response.create_presigned_urls()
+        ) as response:
+            assert not response.is_closed
+            assert response.http_request.headers.get("X-Stainless-Lang") == "python"
+
+            data_source = await response.parse()
+            assert_matches_type(DataSourceCreatePresignedURLsResponse, data_source, path=["response"])
+
+        assert cast(Any, response.is_closed) is True