feat(api): api update

Author: stainless-app[bot]

.stats.yml

@@ -1,4 +1,4 @@
 configured_endpoints: 96
-openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/datamini%2Fasktable-0d2c202b105ede34bc28a2532b7ff64cfb703b35a0e4ac110c663308f3db2ac5.yml
-openapi_spec_hash: 144caed1d3c1d1d2dd4af71dcb80ca2d
+openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/datamini%2Fasktable-a1d971d9b78dce18435845fd9ff02282e95647ffee9b7406fb5f5cd0078c630b.yml
+openapi_spec_hash: ca7ca362f716b3a9b065f74f14ba43bc
 config_hash: a572ab842ea60ce13f1d1a1358440cbe

src/asktable/resources/sqls.py

@@ -49,9 +49,9 @@ def create(
         *,
         datasource_id: str,
         question: str,
+        parameterize: bool | NotGiven = NOT_GIVEN,
         role_id: Optional[str] | NotGiven = NOT_GIVEN,
         role_variables: Optional[object] | NotGiven = NOT_GIVEN,
-        separate_params: 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,
@@ -67,13 +67,13 @@ def create(
 
           question: 查询语句
 
+          parameterize: 是否将参数分开传递
+
           role_id: 角色 ID，将扮演这个角色来执行对话，用于权限控制。若无，则跳过鉴权，即可查询所有
               数据
 
           role_variables: 在扮演这个角色时需要传递的变量值，用 Key-Value 形式传递
 
-          separate_params: 是否将参数分开传递
-
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -88,9 +88,9 @@ def create(
                 {
                     "datasource_id": datasource_id,
                     "question": question,
+                    "parameterize": parameterize,
                     "role_id": role_id,
                     "role_variables": role_variables,
-                    "separate_params": separate_params,
                 },
                 sql_create_params.SqlCreateParams,
             ),
@@ -177,9 +177,9 @@ async def create(
         *,
         datasource_id: str,
         question: str,
+        parameterize: bool | NotGiven = NOT_GIVEN,
         role_id: Optional[str] | NotGiven = NOT_GIVEN,
         role_variables: Optional[object] | NotGiven = NOT_GIVEN,
-        separate_params: 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,
@@ -195,13 +195,13 @@ async def create(
 
           question: 查询语句
 
+          parameterize: 是否将参数分开传递
+
           role_id: 角色 ID，将扮演这个角色来执行对话，用于权限控制。若无，则跳过鉴权，即可查询所有
               数据
 
           role_variables: 在扮演这个角色时需要传递的变量值，用 Key-Value 形式传递
 
-          separate_params: 是否将参数分开传递
-
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -216,9 +216,9 @@ async def create(
                 {
                     "datasource_id": datasource_id,
                     "question": question,
+                    "parameterize": parameterize,
                     "role_id": role_id,
                     "role_variables": role_variables,
-                    "separate_params": separate_params,
                 },
                 sql_create_params.SqlCreateParams,
             ),

src/asktable/types/query_response.py

@@ -12,6 +12,9 @@ class Query(BaseModel):
     sql: str
     """SQL 语句"""
 
+    parameterized_sql: Optional[str] = None
+    """参数化后的 SQL 语句"""
+
     params: Optional[object] = None
     """参数"""
 
@@ -23,6 +26,9 @@ class Request(BaseModel):
     question: str
     """查询语句"""
 
+    parameterize: Optional[bool] = None
+    """是否将参数分开传递"""
+
     role_id: Optional[str] = None
     """
     角色 ID，将扮演这个角色来执行对话，用于权限控制。若无，则跳过鉴权，即可查询所有
@@ -32,9 +38,6 @@ class Request(BaseModel):
     role_variables: Optional[object] = None
     """在扮演这个角色时需要传递的变量值，用 Key-Value 形式传递"""
 
-    separate_params: Optional[bool] = None
-    """是否将参数分开传递"""
-
 
 class QueryResponse(BaseModel):
     id: str

src/asktable/types/sql_create_params.py

@@ -15,6 +15,9 @@ class SqlCreateParams(TypedDict, total=False):
     question: Required[str]
     """查询语句"""
 
+    parameterize: bool
+    """是否将参数分开传递"""
+
     role_id: Optional[str]
     """
     角色 ID，将扮演这个角色来执行对话，用于权限控制。若无，则跳过鉴权，即可查询所有
@@ -23,6 +26,3 @@ class SqlCreateParams(TypedDict, total=False):
 
     role_variables: Optional[object]
     """在扮演这个角色时需要传递的变量值，用 Key-Value 形式传递"""
-
-    separate_params: bool
-    """是否将参数分开传递"""

tests/api_resources/test_sqls.py

@@ -31,9 +31,9 @@ def test_method_create_with_all_params(self, client: Asktable) -> None:
         sql = client.sqls.create(
             datasource_id="datasource_id",
             question="question",
+            parameterize=True,
             role_id="role_id",
             role_variables={},
-            separate_params=True,
         )
         assert_matches_type(QueryResponse, sql, path=["response"])
 
@@ -114,9 +114,9 @@ async def test_method_create_with_all_params(self, async_client: AsyncAsktable)
         sql = await async_client.sqls.create(
             datasource_id="datasource_id",
             question="question",
+            parameterize=True,
             role_id="role_id",
             role_variables={},
-            separate_params=True,
         )
         assert_matches_type(QueryResponse, sql, path=["response"])