feat(api): support file management, prolong timeout (#102)

Author: stainless-app[bot]

.stats.yml

@@ -1,2 +1,2 @@
-configured_endpoints: 90
+configured_endpoints: 92
 openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/datamini%2Fasktable-812fbef9135838630557abadedcc8c593c10d2133388aaea70b9a313fc546941.yml

README.md

@@ -202,15 +202,15 @@ client.with_options(max_retries=5).datasources.create(
 
 ### Timeouts
 
-By default requests time out after 1 minute. You can configure this with a `timeout` option,
+By default requests time out after 5 minutes. You can configure this with a `timeout` option,
 which accepts a float or an [`httpx.Timeout`](https://www.python-httpx.org/advanced/#fine-tuning-the-configuration) object:
 
 ```python
 from asktable import Asktable
 
 # Configure the default for all requests:
 client = Asktable(
-    # 20 seconds (default is 1 minute)
+    # 20 seconds (default is 5 minutes)
     timeout=20.0,
 )
 

api.md

@@ -150,6 +150,8 @@ from asktable.types import (
     Meta,
     DatasourceRetrieveResponse,
     DatasourceDeleteResponse,
+    DatasourceAddFileResponse,
+    DatasourceDeleteFileResponse,
 )
 ```
 
@@ -160,6 +162,8 @@ Methods:
 - <code title="patch /datasources/{datasource_id}">client.datasources.<a href="./src/asktable/resources/datasources/datasources.py">update</a>(datasource_id, \*\*<a href="src/asktable/types/datasource_update_params.py">params</a>) -> <a href="./src/asktable/types/datasource.py">Datasource</a></code>
 - <code title="get /datasources">client.datasources.<a href="./src/asktable/resources/datasources/datasources.py">list</a>(\*\*<a href="src/asktable/types/datasource_list_params.py">params</a>) -> <a href="./src/asktable/types/datasource.py">SyncPage[Datasource]</a></code>
 - <code title="delete /datasources/{datasource_id}">client.datasources.<a href="./src/asktable/resources/datasources/datasources.py">delete</a>(datasource_id) -> <a href="./src/asktable/types/datasource_delete_response.py">object</a></code>
+- <code title="post /datasources/{datasource_id}/files">client.datasources.<a href="./src/asktable/resources/datasources/datasources.py">add_file</a>(datasource_id, \*\*<a href="src/asktable/types/datasource_add_file_params.py">params</a>) -> <a href="./src/asktable/types/datasource_add_file_response.py">object</a></code>
+- <code title="delete /datasources/{datasource_id}/files/{file_id}">client.datasources.<a href="./src/asktable/resources/datasources/datasources.py">delete_file</a>(file_id, \*, datasource_id) -> <a href="./src/asktable/types/datasource_delete_file_response.py">object</a></code>
 
 ## Meta
 

src/asktable/_constants.py

@@ -5,8 +5,8 @@
 RAW_RESPONSE_HEADER = "X-Stainless-Raw-Response"
 OVERRIDE_CAST_TO_HEADER = "____stainless_override_cast_to"
 
-# default timeout is 1 minute
-DEFAULT_TIMEOUT = httpx.Timeout(timeout=60.0, connect=5.0)
+# default timeout is 5 minutes
+DEFAULT_TIMEOUT = httpx.Timeout(timeout=300.0, connect=5.0)
 DEFAULT_MAX_RETRIES = 2
 DEFAULT_CONNECTION_LIMITS = httpx.Limits(max_connections=100, max_keepalive_connections=20)
 

src/asktable/resources/datasources/datasources.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from typing import Optional
+from typing import List, Mapping, Optional, cast
 from typing_extensions import Literal
 
 import httpx
@@ -15,7 +15,12 @@
     MetaResourceWithStreamingResponse,
     AsyncMetaResourceWithStreamingResponse,
 )
-from ...types import datasource_list_params, datasource_create_params, datasource_update_params
+from ...types import (
+    datasource_list_params,
+    datasource_create_params,
+    datasource_update_params,
+    datasource_add_file_params,
+)
 from .indexes import (
     IndexesResource,
     AsyncIndexesResource,
@@ -24,9 +29,11 @@
     IndexesResourceWithStreamingResponse,
     AsyncIndexesResourceWithStreamingResponse,
 )
-from ..._types import NOT_GIVEN, Body, Query, Headers, NotGiven
+from ..._types import NOT_GIVEN, Body, Query, Headers, NotGiven, FileTypes
 from ..._utils import (
+    extract_files,
     maybe_transform,
+    deepcopy_minimal,
     async_maybe_transform,
 )
 from ..._compat import cached_property
@@ -340,6 +347,84 @@ def delete(
             cast_to=object,
         )
 
+    def add_file(
+        self,
+        datasource_id: str,
+        *,
+        files: List[FileTypes],
+        # 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,
+    ) -> object:
+        """
+        为数据源添加文件
+
+        Args:
+          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
+        """
+        if not datasource_id:
+            raise ValueError(f"Expected a non-empty value for `datasource_id` but received {datasource_id!r}")
+        body = deepcopy_minimal({"files": files})
+        extracted_files = extract_files(cast(Mapping[str, object], body), paths=[["files", "<array>"]])
+        # It should be noted that the actual Content-Type header that will be
+        # sent to the server will contain a `boundary` parameter, e.g.
+        # multipart/form-data; boundary=---abc--
+        extra_headers = {"Content-Type": "multipart/form-data", **(extra_headers or {})}
+        return self._post(
+            f"/datasources/{datasource_id}/files",
+            body=maybe_transform(body, datasource_add_file_params.DatasourceAddFileParams),
+            files=extracted_files,
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=object,
+        )
+
+    def delete_file(
+        self,
+        file_id: str,
+        *,
+        datasource_id: str,
+        # 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,
+    ) -> object:
+        """
+        删除数据源的单个文件
+
+        Args:
+          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
+        """
+        if not datasource_id:
+            raise ValueError(f"Expected a non-empty value for `datasource_id` but received {datasource_id!r}")
+        if not file_id:
+            raise ValueError(f"Expected a non-empty value for `file_id` but received {file_id!r}")
+        return self._delete(
+            f"/datasources/{datasource_id}/files/{file_id}",
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=object,
+        )
+
 
 class AsyncDatasourcesResource(AsyncAPIResource):
     @cached_property
@@ -628,6 +713,84 @@ async def delete(
             cast_to=object,
         )
 
+    async def add_file(
+        self,
+        datasource_id: str,
+        *,
+        files: List[FileTypes],
+        # 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,
+    ) -> object:
+        """
+        为数据源添加文件
+
+        Args:
+          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
+        """
+        if not datasource_id:
+            raise ValueError(f"Expected a non-empty value for `datasource_id` but received {datasource_id!r}")
+        body = deepcopy_minimal({"files": files})
+        extracted_files = extract_files(cast(Mapping[str, object], body), paths=[["files", "<array>"]])
+        # It should be noted that the actual Content-Type header that will be
+        # sent to the server will contain a `boundary` parameter, e.g.
+        # multipart/form-data; boundary=---abc--
+        extra_headers = {"Content-Type": "multipart/form-data", **(extra_headers or {})}
+        return await self._post(
+            f"/datasources/{datasource_id}/files",
+            body=await async_maybe_transform(body, datasource_add_file_params.DatasourceAddFileParams),
+            files=extracted_files,
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=object,
+        )
+
+    async def delete_file(
+        self,
+        file_id: str,
+        *,
+        datasource_id: str,
+        # 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,
+    ) -> object:
+        """
+        删除数据源的单个文件
+
+        Args:
+          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
+        """
+        if not datasource_id:
+            raise ValueError(f"Expected a non-empty value for `datasource_id` but received {datasource_id!r}")
+        if not file_id:
+            raise ValueError(f"Expected a non-empty value for `file_id` but received {file_id!r}")
+        return await self._delete(
+            f"/datasources/{datasource_id}/files/{file_id}",
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=object,
+        )
+
 
 class DatasourcesResourceWithRawResponse:
     def __init__(self, datasources: DatasourcesResource) -> None:
@@ -648,6 +811,12 @@ def __init__(self, datasources: DatasourcesResource) -> None:
         self.delete = to_raw_response_wrapper(
             datasources.delete,
         )
+        self.add_file = to_raw_response_wrapper(
+            datasources.add_file,
+        )
+        self.delete_file = to_raw_response_wrapper(
+            datasources.delete_file,
+        )
 
     @cached_property
     def meta(self) -> MetaResourceWithRawResponse:
@@ -681,6 +850,12 @@ def __init__(self, datasources: AsyncDatasourcesResource) -> None:
         self.delete = async_to_raw_response_wrapper(
             datasources.delete,
         )
+        self.add_file = async_to_raw_response_wrapper(
+            datasources.add_file,
+        )
+        self.delete_file = async_to_raw_response_wrapper(
+            datasources.delete_file,
+        )
 
     @cached_property
     def meta(self) -> AsyncMetaResourceWithRawResponse:
@@ -714,6 +889,12 @@ def __init__(self, datasources: DatasourcesResource) -> None:
         self.delete = to_streamed_response_wrapper(
             datasources.delete,
         )
+        self.add_file = to_streamed_response_wrapper(
+            datasources.add_file,
+        )
+        self.delete_file = to_streamed_response_wrapper(
+            datasources.delete_file,
+        )
 
     @cached_property
     def meta(self) -> MetaResourceWithStreamingResponse:
@@ -747,6 +928,12 @@ def __init__(self, datasources: AsyncDatasourcesResource) -> None:
         self.delete = async_to_streamed_response_wrapper(
             datasources.delete,
         )
+        self.add_file = async_to_streamed_response_wrapper(
+            datasources.add_file,
+        )
+        self.delete_file = async_to_streamed_response_wrapper(
+            datasources.delete_file,
+        )
 
     @cached_property
     def meta(self) -> AsyncMetaResourceWithStreamingResponse:

src/asktable/types/__init__.py

@@ -58,6 +58,7 @@
 from .role_get_polices_response import RoleGetPolicesResponse as RoleGetPolicesResponse
 from .role_get_variables_params import RoleGetVariablesParams as RoleGetVariablesParams
 from .chat_send_message_response import ChatSendMessageResponse as ChatSendMessageResponse
+from .datasource_add_file_params import DatasourceAddFileParams as DatasourceAddFileParams
 from .preference_create_response import PreferenceCreateResponse as PreferenceCreateResponse
 from .preference_update_response import PreferenceUpdateResponse as PreferenceUpdateResponse
 from .securetunnel_create_params import SecuretunnelCreateParams as SecuretunnelCreateParams

src/asktable/types/datasource_add_file_params.py

@@ -0,0 +1,14 @@
+# 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 Required, TypedDict
+
+from .._types import FileTypes
+
+__all__ = ["DatasourceAddFileParams"]
+
+
+class DatasourceAddFileParams(TypedDict, total=False):
+    files: Required[List[FileTypes]]