feat(api): api update (#10)

Author: stainless-app[bot]

.stats.yml

@@ -1,2 +1,2 @@
-configured_endpoints: 67
-openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/datamini%2Fasktable-40b24dabc4429084d4928122661865fb5b2feebac81de578d7de95c4d351621c.yml
+configured_endpoints: 68
+openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/datamini%2Fasktable-f8d4f4ee9ccccf65054df0c205156f6022a8d207441b1494fa56bd950b44593b.yml

README.md

@@ -84,7 +84,7 @@ client = Asktable()
 
 try:
     client.datasources.create(
-        access_config={"engine": "mysql"},
+        access_config={},
         engine="mysql",
     )
 except asktable.APIConnectionError as e:
@@ -130,7 +130,7 @@ client = Asktable(
 
 # Or, configure per-request:
 client.with_options(max_retries=5).datasources.create(
-    access_config={"engine": "mysql"},
+    access_config={},
     engine="mysql",
 )
 ```
@@ -156,7 +156,7 @@ client = Asktable(
 
 # Override per-request:
 client.with_options(timeout=5.0).datasources.create(
-    access_config={"engine": "mysql"},
+    access_config={},
     engine="mysql",
 )
 ```
@@ -198,9 +198,7 @@ from asktable import Asktable
 
 client = Asktable()
 response = client.datasources.with_raw_response.create(
-    access_config={
-        "engine": "mysql"
-    },
+    access_config={},
     engine="mysql",
 )
 print(response.headers.get('X-My-Header'))
@@ -221,7 +219,7 @@ To stream the response body, use `.with_streaming_response` instead, which requi
 
 ```python
 with client.datasources.with_streaming_response.create(
-    access_config={"engine": "mysql"},
+    access_config={},
     engine="mysql",
 ) as response:
     print(response.headers.get("X-My-Header"))

api.md

@@ -132,6 +132,7 @@ 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/data_source.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_list_response.py">DatasourceListResponse</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/file">client.datasources.<a href="./src/asktable/resources/datasources/datasources.py">create_from_file</a>(\*\*<a href="src/asktable/types/datasource_create_from_file_params.py">params</a>) -> <a href="./src/asktable/types/data_source.py">DataSource</a></code>
 
 ## Meta
 

src/asktable/resources/datasources/datasources.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from typing import Optional
+from typing import Mapping, Optional, cast
 from typing_extensions import Literal
 
 import httpx
@@ -15,10 +15,17 @@
     MetaResourceWithStreamingResponse,
     AsyncMetaResourceWithStreamingResponse,
 )
-from ...types import datasource_list_params, datasource_create_params, datasource_update_params
-from ..._types import NOT_GIVEN, Body, Query, Headers, NotGiven
+from ...types import (
+    datasource_list_params,
+    datasource_create_params,
+    datasource_update_params,
+    datasource_create_from_file_params,
+)
+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
@@ -319,6 +326,63 @@ def delete(
             cast_to=object,
         )
 
+    def create_from_file(
+        self,
+        *,
+        name: str,
+        file: FileTypes,
+        async_process_meta: bool | NotGiven = NOT_GIVEN,
+        skip_process_meta: bool | NotGiven = NOT_GIVEN,
+        # 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,
+    ) -> DataSource:
+        """
+        上传文件并创建数据源
+
+        Args:
+          async_process_meta: 是否异步处理元数据
+
+          skip_process_meta: 是否跳过元数据处理
+
+          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
+        """
+        body = deepcopy_minimal({"file": file})
+        files = extract_files(cast(Mapping[str, object], body), paths=[["file"]])
+        # 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(
+            "/datasources/file",
+            body=maybe_transform(body, datasource_create_from_file_params.DatasourceCreateFromFileParams),
+            files=files,
+            options=make_request_options(
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                query=maybe_transform(
+                    {
+                        "name": name,
+                        "async_process_meta": async_process_meta,
+                        "skip_process_meta": skip_process_meta,
+                    },
+                    datasource_create_from_file_params.DatasourceCreateFromFileParams,
+                ),
+            ),
+            cast_to=DataSource,
+        )
+
 
 class AsyncDatasourcesResource(AsyncAPIResource):
     @cached_property
@@ -595,6 +659,63 @@ async def delete(
             cast_to=object,
         )
 
+    async def create_from_file(
+        self,
+        *,
+        name: str,
+        file: FileTypes,
+        async_process_meta: bool | NotGiven = NOT_GIVEN,
+        skip_process_meta: bool | NotGiven = NOT_GIVEN,
+        # 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,
+    ) -> DataSource:
+        """
+        上传文件并创建数据源
+
+        Args:
+          async_process_meta: 是否异步处理元数据
+
+          skip_process_meta: 是否跳过元数据处理
+
+          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
+        """
+        body = deepcopy_minimal({"file": file})
+        files = extract_files(cast(Mapping[str, object], body), paths=[["file"]])
+        # 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(
+            "/datasources/file",
+            body=await async_maybe_transform(body, datasource_create_from_file_params.DatasourceCreateFromFileParams),
+            files=files,
+            options=make_request_options(
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                query=await async_maybe_transform(
+                    {
+                        "name": name,
+                        "async_process_meta": async_process_meta,
+                        "skip_process_meta": skip_process_meta,
+                    },
+                    datasource_create_from_file_params.DatasourceCreateFromFileParams,
+                ),
+            ),
+            cast_to=DataSource,
+        )
+
 
 class DatasourcesResourceWithRawResponse:
     def __init__(self, datasources: DatasourcesResource) -> None:
@@ -615,6 +736,9 @@ def __init__(self, datasources: DatasourcesResource) -> None:
         self.delete = to_raw_response_wrapper(
             datasources.delete,
         )
+        self.create_from_file = to_raw_response_wrapper(
+            datasources.create_from_file,
+        )
 
     @cached_property
     def meta(self) -> MetaResourceWithRawResponse:
@@ -644,6 +768,9 @@ def __init__(self, datasources: AsyncDatasourcesResource) -> None:
         self.delete = async_to_raw_response_wrapper(
             datasources.delete,
         )
+        self.create_from_file = async_to_raw_response_wrapper(
+            datasources.create_from_file,
+        )
 
     @cached_property
     def meta(self) -> AsyncMetaResourceWithRawResponse:
@@ -673,6 +800,9 @@ def __init__(self, datasources: DatasourcesResource) -> None:
         self.delete = to_streamed_response_wrapper(
             datasources.delete,
         )
+        self.create_from_file = to_streamed_response_wrapper(
+            datasources.create_from_file,
+        )
 
     @cached_property
     def meta(self) -> MetaResourceWithStreamingResponse:
@@ -702,6 +832,9 @@ def __init__(self, datasources: AsyncDatasourcesResource) -> None:
         self.delete = async_to_streamed_response_wrapper(
             datasources.delete,
         )
+        self.create_from_file = async_to_streamed_response_wrapper(
+            datasources.create_from_file,
+        )
 
     @cached_property
     def meta(self) -> AsyncMetaResourceWithStreamingResponse:

src/asktable/types/__init__.py

@@ -40,3 +40,4 @@
 from .securetunnel_list_response import SecuretunnelListResponse as SecuretunnelListResponse
 from .securetunnel_update_params import SecuretunnelUpdateParams as SecuretunnelUpdateParams
 from .integration_excel_csv_ask_params import IntegrationExcelCsvAskParams as IntegrationExcelCsvAskParams
+from .datasource_create_from_file_params import DatasourceCreateFromFileParams as DatasourceCreateFromFileParams

src/asktable/types/data_source.py

@@ -10,9 +10,6 @@
 
 
 class AccessConfig(BaseModel):
-    engine: Literal["mysql", "tidb", "postgresql", "oceanbase", "clickhouse", "csv", "excel"]
-    """数据库引擎"""
-
     db: Optional[str] = None
     """数据库引擎可以管理多个数据库，此参数用于指定数据库名称"""
 

src/asktable/types/datasource_create_from_file_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_extensions import Required, TypedDict
+
+from .._types import FileTypes
+
+__all__ = ["DatasourceCreateFromFileParams"]
+
+
+class DatasourceCreateFromFileParams(TypedDict, total=False):
+    name: Required[str]
+
+    file: Required[FileTypes]
+
+    async_process_meta: bool
+    """是否异步处理元数据"""
+
+    skip_process_meta: bool
+    """是否跳过元数据处理"""

src/asktable/types/datasource_create_params.py

@@ -26,9 +26,6 @@ class DatasourceCreateParams(TypedDict, total=False):
 
 
 class AccessConfig(TypedDict, total=False):
-    engine: Required[Literal["mysql", "tidb", "postgresql", "oceanbase", "clickhouse", "csv", "excel"]]
-    """数据库引擎"""
-
     db: Optional[str]
     """数据库引擎可以管理多个数据库，此参数用于指定数据库名称"""
 

src/asktable/types/datasource_update_params.py

@@ -3,7 +3,7 @@
 from __future__ import annotations
 
 from typing import Optional
-from typing_extensions import Literal, Required, TypedDict
+from typing_extensions import Literal, TypedDict
 
 __all__ = ["DatasourceUpdateParams", "AccessConfig"]
 
@@ -35,9 +35,6 @@ class DatasourceUpdateParams(TypedDict, total=False):
 
 
 class AccessConfig(TypedDict, total=False):
-    engine: Required[Literal["mysql", "tidb", "postgresql", "oceanbase", "clickhouse", "csv", "excel"]]
-    """数据库引擎"""
-
     db: Optional[str]
     """数据库引擎可以管理多个数据库，此参数用于指定数据库名称"""