feat(api): api update

Author: stainless-app[bot]

.stats.yml

@@ -1,4 +1,4 @@
 configured_endpoints: 101
-openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/datamini%2Fasktable-433d41b79e0e9c47c18aa56fbda33ce5a3645c224012d74f8a30da676905a16a.yml
-openapi_spec_hash: ff827305d98a6c808e7931a0298e7729
+openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/datamini%2Fasktable-06a1ae7f466edd478e449b5d27c152c689a75228b0a132d1fc75282b3b004b29.yml
+openapi_spec_hash: 38321574df76b7905271862ed1e200f0
 config_hash: acdf4142177ed1932c2d82372693f811

src/asktable/_qs.py

@@ -2,17 +2,13 @@
 
 from typing import Any, List, Tuple, Union, Mapping, TypeVar
 from urllib.parse import parse_qs, urlencode
-from typing_extensions import Literal, get_args
+from typing_extensions import get_args
 
-from ._types import NotGiven, not_given
+from ._types import NotGiven, ArrayFormat, NestedFormat, not_given
 from ._utils import flatten
 
 _T = TypeVar("_T")
 
-
-ArrayFormat = Literal["comma", "repeat", "indices", "brackets"]
-NestedFormat = Literal["dots", "brackets"]
-
 PrimitiveData = Union[str, int, float, bool, None]
 # this should be Data = Union[PrimitiveData, "List[Data]", "Tuple[Data]", "Mapping[str, Data]"]
 # https://github.com/microsoft/pyright/issues/3555

src/asktable/_types.py

@@ -47,6 +47,9 @@
 ModelT = TypeVar("ModelT", bound=pydantic.BaseModel)
 _T = TypeVar("_T")
 
+ArrayFormat = Literal["comma", "repeat", "indices", "brackets"]
+NestedFormat = Literal["dots", "brackets"]
+
 
 # Approximates httpx internal ProxiesTypes and RequestFiles types
 # while adding support for `PathLike` instances

src/asktable/_utils/_utils.py

@@ -17,11 +17,11 @@
 )
 from pathlib import Path
 from datetime import date, datetime
-from typing_extensions import TypeGuard
+from typing_extensions import TypeGuard, get_args
 
 import sniffio
 
-from .._types import Omit, NotGiven, FileTypes, HeadersLike
+from .._types import Omit, NotGiven, FileTypes, ArrayFormat, HeadersLike
 
 _T = TypeVar("_T")
 _TupleT = TypeVar("_TupleT", bound=Tuple[object, ...])
@@ -40,25 +40,45 @@ def extract_files(
     query: Mapping[str, object],
     *,
     paths: Sequence[Sequence[str]],
+    array_format: ArrayFormat = "brackets",
 ) -> list[tuple[str, FileTypes]]:
     """Recursively extract files from the given dictionary based on specified paths.
 
     A path may look like this ['foo', 'files', '<array>', 'data'].
 
+    ``array_format`` controls how ``<array>`` segments contribute to the emitted
+    field name. Supported values: ``"brackets"`` (``foo[]``), ``"repeat"`` and
+    ``"comma"`` (``foo``), ``"indices"`` (``foo[0]``, ``foo[1]``).
+
     Note: this mutates the given dictionary.
     """
     files: list[tuple[str, FileTypes]] = []
     for path in paths:
-        files.extend(_extract_items(query, path, index=0, flattened_key=None))
+        files.extend(_extract_items(query, path, index=0, flattened_key=None, array_format=array_format))
     return files
 
 
+def _array_suffix(array_format: ArrayFormat, array_index: int) -> str:
+    if array_format == "brackets":
+        return "[]"
+    if array_format == "indices":
+        return f"[{array_index}]"
+    if array_format == "repeat" or array_format == "comma":
+        # Both repeat the bare field name for each file part; there is no
+        # meaningful way to comma-join binary parts.
+        return ""
+    raise NotImplementedError(
+        f"Unknown array_format value: {array_format}, choose from {', '.join(get_args(ArrayFormat))}"
+    )
+
+
 def _extract_items(
     obj: object,
     path: Sequence[str],
     *,
     index: int,
     flattened_key: str | None,
+    array_format: ArrayFormat,
 ) -> list[tuple[str, FileTypes]]:
     try:
         key = path[index]
@@ -75,9 +95,11 @@ def _extract_items(
 
         if is_list(obj):
             files: list[tuple[str, FileTypes]] = []
-            for entry in obj:
-                assert_is_file_content(entry, key=flattened_key + "[]" if flattened_key else "")
-                files.append((flattened_key + "[]", cast(FileTypes, entry)))
+            for array_index, entry in enumerate(obj):
+                suffix = _array_suffix(array_format, array_index)
+                emitted_key = (flattened_key + suffix) if flattened_key else suffix
+                assert_is_file_content(entry, key=emitted_key)
+                files.append((emitted_key, cast(FileTypes, entry)))
             return files
 
         assert_is_file_content(obj, key=flattened_key)
@@ -106,6 +128,7 @@ def _extract_items(
             path,
             index=index,
             flattened_key=flattened_key,
+            array_format=array_format,
         )
     elif is_list(obj):
         if key != "<array>":
@@ -117,9 +140,12 @@ def _extract_items(
                     item,
                     path,
                     index=index,
-                    flattened_key=flattened_key + "[]" if flattened_key is not None else "[]",
+                    flattened_key=(
+                        (flattened_key if flattened_key is not None else "") + _array_suffix(array_format, array_index)
+                    ),
+                    array_format=array_format,
                 )
-                for item in obj
+                for array_index, item in enumerate(obj)
             ]
         )
 

src/asktable/resources/datasources/datasources.py

@@ -131,6 +131,7 @@ def create(
             "gaussdbdws",
             "bitable",
             "dap",
+            "duckdb",
         ],
         access_config: Optional[datasource_create_params.AccessConfig] | Omit = omit,
         name: Optional[str] | Omit = omit,
@@ -249,6 +250,7 @@ def update(
                 "gaussdbdws",
                 "bitable",
                 "dap",
+                "duckdb",
             ]
         ]
         | Omit = omit,
@@ -657,6 +659,7 @@ async def create(
             "gaussdbdws",
             "bitable",
             "dap",
+            "duckdb",
         ],
         access_config: Optional[datasource_create_params.AccessConfig] | Omit = omit,
         name: Optional[str] | Omit = omit,
@@ -775,6 +778,7 @@ async def update(
                 "gaussdbdws",
                 "bitable",
                 "dap",
+                "duckdb",
             ]
         ]
         | Omit = omit,

src/asktable/types/datasource.py

@@ -50,6 +50,7 @@ class Datasource(BaseModel):
         "gaussdbdws",
         "bitable",
         "dap",
+        "duckdb",
     ]
     """数据源引擎"""
 

src/asktable/types/datasource_create_params.py

@@ -51,6 +51,7 @@ class DatasourceCreateParams(TypedDict, total=False):
             "gaussdbdws",
             "bitable",
             "dap",
+            "duckdb",
         ]
     ]
     """数据源引擎"""

src/asktable/types/datasource_retrieve_response.py

@@ -105,6 +105,7 @@ class DatasourceRetrieveResponse(BaseModel):
         "gaussdbdws",
         "bitable",
         "dap",
+        "duckdb",
     ]
     """数据源引擎"""
 

src/asktable/types/datasource_update_params.py

@@ -58,6 +58,7 @@ class DatasourceUpdateParams(TypedDict, total=False):
             "gaussdbdws",
             "bitable",
             "dap",
+            "duckdb",
         ]
     ]
     """数据源引擎"""

tests/test_extract_files.py

@@ -4,7 +4,7 @@
 
 import pytest
 
-from asktable._types import FileTypes
+from asktable._types import FileTypes, ArrayFormat
 from asktable._utils import extract_files
 
 
@@ -37,10 +37,7 @@ def test_multiple_files() -> None:
 
 def test_top_level_file_array() -> None:
     query = {"files": [b"file one", b"file two"], "title": "hello"}
-    assert extract_files(query, paths=[["files", "<array>"]]) == [
-        ("files[]", b"file one"),
-        ("files[]", b"file two"),
-    ]
+    assert extract_files(query, paths=[["files", "<array>"]]) == [("files[]", b"file one"), ("files[]", b"file two")]
     assert query == {"title": "hello"}
 
 
@@ -71,3 +68,24 @@ def test_ignores_incorrect_paths(
     expected: list[tuple[str, FileTypes]],
 ) -> None:
     assert extract_files(query, paths=paths) == expected
+
+
+@pytest.mark.parametrize(
+    "array_format,expected_top_level,expected_nested",
+    [
+        ("brackets", [("files[]", b"a"), ("files[]", b"b")], [("items[][file]", b"a"), ("items[][file]", b"b")]),
+        ("repeat", [("files", b"a"), ("files", b"b")], [("items[file]", b"a"), ("items[file]", b"b")]),
+        ("comma", [("files", b"a"), ("files", b"b")], [("items[file]", b"a"), ("items[file]", b"b")]),
+        ("indices", [("files[0]", b"a"), ("files[1]", b"b")], [("items[0][file]", b"a"), ("items[1][file]", b"b")]),
+    ],
+)
+def test_array_format_controls_file_field_names(
+    array_format: ArrayFormat,
+    expected_top_level: list[tuple[str, FileTypes]],
+    expected_nested: list[tuple[str, FileTypes]],
+) -> None:
+    top_level = {"files": [b"a", b"b"]}
+    assert extract_files(top_level, paths=[["files", "<array>"]], array_format=array_format) == expected_top_level
+
+    nested = {"items": [{"file": b"a"}, {"file": b"b"}]}
+    assert extract_files(nested, paths=[["items", "<array>", "file"]], array_format=array_format) == expected_nested