feat(api): api update (#34)

Author: stainless-app[bot]

.stats.yml

@@ -1,2 +1,2 @@
 configured_endpoints: 70
-openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/datamini%2Fasktable-55b1b1224cfcd2083a3df4a1752689ebf1b16821d0baeef75030854010fd6cd4.yml
+openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/datamini%2Fasktable-5af6abcc29b8b798ae7baa30849249d4a0ad087a19105fd8b3143f9f18fcfba1.yml

src/asktable/types/data_source.py

@@ -10,6 +10,9 @@
 
 
 class AccessConfig(BaseModel):
+    atst_link_id: Optional[str] = None
+    """安全隧道链接 ID"""
+
     db: Optional[str] = None
     """数据库引擎可以管理多个数据库，此参数用于指定数据库名称"""
 
@@ -25,6 +28,12 @@ class AccessConfig(BaseModel):
     port: Optional[int] = None
     """数据库端口"""
 
+    proxy_host: Optional[str] = None
+    """数据源代理地址"""
+
+    proxy_port: Optional[int] = None
+    """数据源代理端口"""
+
     securetunnel_id: Optional[str] = None
     """安全隧道 ID"""
 

src/asktable/types/datasource_create_params.py

@@ -22,6 +22,9 @@ class DatasourceCreateParams(TypedDict, total=False):
 
 
 class AccessConfig(TypedDict, total=False):
+    atst_link_id: Optional[str]
+    """安全隧道链接 ID"""
+
     db: Optional[str]
     """数据库引擎可以管理多个数据库，此参数用于指定数据库名称"""
 
@@ -40,6 +43,12 @@ class AccessConfig(TypedDict, total=False):
     port: Optional[int]
     """数据库端口"""
 
+    proxy_host: Optional[str]
+    """数据源代理地址"""
+
+    proxy_port: Optional[int]
+    """数据源代理端口"""
+
     securetunnel_id: Optional[str]
     """安全隧道 ID"""
 

src/asktable/types/datasource_update_params.py

@@ -35,6 +35,9 @@ class DatasourceUpdateParams(TypedDict, total=False):
 
 
 class AccessConfig(TypedDict, total=False):
+    atst_link_id: Optional[str]
+    """安全隧道链接 ID"""
+
     db: Optional[str]
     """数据库引擎可以管理多个数据库，此参数用于指定数据库名称"""
 
@@ -53,6 +56,12 @@ class AccessConfig(TypedDict, total=False):
     port: Optional[int]
     """数据库端口"""
 
+    proxy_host: Optional[str]
+    """数据源代理地址"""
+
+    proxy_port: Optional[int]
+    """数据源代理端口"""
+
     securetunnel_id: Optional[str]
     """安全隧道 ID"""
 

tests/api_resources/test_datasources.py

@@ -33,12 +33,15 @@ def test_method_create_with_all_params(self, client: Asktable) -> None:
             engine="mysql",
             async_process_meta=True,
             access_config={
+                "atst_link_id": "atst_link_123456",
                 "db": "test",
                 "host": "192.168.0.10",
                 "location_type": "local",
                 "location_url": "http://example.com/data.csv",
                 "password": "root",
                 "port": 3306,
+                "proxy_host": "192.168.0.10",
+                "proxy_port": 3306,
                 "securetunnel_id": "atst_123456",
                 "user": "root",
             },
@@ -120,12 +123,15 @@ def test_method_update_with_all_params(self, client: Asktable) -> None:
         datasource = client.datasources.update(
             datasource_id="datasource_id",
             access_config={
+                "atst_link_id": "atst_link_123456",
                 "db": "test",
                 "host": "192.168.0.10",
                 "location_type": "local",
                 "location_url": "http://example.com/data.csv",
                 "password": "root",
                 "port": 3306,
+                "proxy_host": "192.168.0.10",
+                "proxy_port": 3306,
                 "securetunnel_id": "atst_123456",
                 "user": "root",
             },
@@ -302,12 +308,15 @@ async def test_method_create_with_all_params(self, async_client: AsyncAsktable)
             engine="mysql",
             async_process_meta=True,
             access_config={
+                "atst_link_id": "atst_link_123456",
                 "db": "test",
                 "host": "192.168.0.10",
                 "location_type": "local",
                 "location_url": "http://example.com/data.csv",
                 "password": "root",
                 "port": 3306,
+                "proxy_host": "192.168.0.10",
+                "proxy_port": 3306,
                 "securetunnel_id": "atst_123456",
                 "user": "root",
             },
@@ -389,12 +398,15 @@ async def test_method_update_with_all_params(self, async_client: AsyncAsktable)
         datasource = await async_client.datasources.update(
             datasource_id="datasource_id",
             access_config={
+                "atst_link_id": "atst_link_123456",
                 "db": "test",
                 "host": "192.168.0.10",
                 "location_type": "local",
                 "location_url": "http://example.com/data.csv",
                 "password": "root",
                 "port": 3306,
+                "proxy_host": "192.168.0.10",
+                "proxy_port": 3306,
                 "securetunnel_id": "atst_123456",
                 "user": "root",
             },