feat(api): api update (#17)

Author: stainless-app[bot]

.stats.yml

@@ -1,2 +1,2 @@
 configured_endpoints: 68
-openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/datamini%2Fasktable-f31385daf5ebf616e46fc8458a1eeafbec0849e12dbfa221bcb2d409db409e10.yml
+openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/datamini%2Fasktable-70caff6d751e62dc360c72766ea91a07da36288a58a83a412fd64d1e8ca0aa2a.yml

src/asktable/_base_client.py

@@ -143,6 +143,12 @@ def __init__(
         self.url = url
         self.params = params
 
+    @override
+    def __repr__(self) -> str:
+        if self.url:
+            return f"{self.__class__.__name__}(url={self.url})"
+        return f"{self.__class__.__name__}(params={self.params})"
+
 
 class BasePage(GenericModel, Generic[_T]):
     """
@@ -689,7 +695,8 @@ def _calculate_retry_timeout(
         if retry_after is not None and 0 < retry_after <= 60:
             return retry_after
 
-        nb_retries = max_retries - remaining_retries
+        # Also cap retry count to 1000 to avoid any potential overflows with `pow`
+        nb_retries = min(max_retries - remaining_retries, 1000)
 
         # Apply exponential backoff, but not more than the max.
         sleep_seconds = min(INITIAL_RETRY_DELAY * pow(2.0, nb_retries), MAX_RETRY_DELAY)

src/asktable/_response.py

@@ -192,6 +192,9 @@ def _parse(self, *, to: type[_T] | None = None) -> R | _T:
         if cast_to == float:
             return cast(R, float(response.text))
 
+        if cast_to == bool:
+            return cast(R, response.text.lower() == "true")
+
         origin = get_origin(cast_to) or cast_to
 
         if origin == APIResponse:

src/asktable/resources/chats/chats.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from typing import Dict, List, Union, Optional
+from typing import Dict, Union, Optional
 
 import httpx
 
@@ -64,7 +64,6 @@ def create(
         self,
         *,
         bot_id: Optional[str] | NotGiven = NOT_GIVEN,
-        datasource_ids: Optional[List[str]] | NotGiven = NOT_GIVEN,
         name: Optional[str] | NotGiven = NOT_GIVEN,
         role_id: Optional[str] | NotGiven = NOT_GIVEN,
         role_variables: Optional[Dict[str, Union[str, int, bool]]] | NotGiven = NOT_GIVEN,
@@ -83,8 +82,6 @@ def create(
           bot_id: 机器人 ID，如果需要使用高级功能，请使用 bot_id 来创建对话。在机器人中你可以定义
               可以访问的数据、可以执行的任务以及是否开启调试模式等设置。
 
-          datasource_ids: 直接与数据源对话，我们会自动为你创建一个机器人, only support one datasource
-
           name: New name for the chat
 
           role_id: 角色 ID，将扮演这个角色来执行对话，用于权限控制。若无，则跳过鉴权，即可查询所有
@@ -107,7 +104,6 @@ def create(
             body=maybe_transform(
                 {
                     "bot_id": bot_id,
-                    "datasource_ids": datasource_ids,
                     "name": name,
                     "role_id": role_id,
                     "role_variables": role_variables,
@@ -263,7 +259,6 @@ async def create(
         self,
         *,
         bot_id: Optional[str] | NotGiven = NOT_GIVEN,
-        datasource_ids: Optional[List[str]] | NotGiven = NOT_GIVEN,
         name: Optional[str] | NotGiven = NOT_GIVEN,
         role_id: Optional[str] | NotGiven = NOT_GIVEN,
         role_variables: Optional[Dict[str, Union[str, int, bool]]] | NotGiven = NOT_GIVEN,
@@ -282,8 +277,6 @@ async def create(
           bot_id: 机器人 ID，如果需要使用高级功能，请使用 bot_id 来创建对话。在机器人中你可以定义
               可以访问的数据、可以执行的任务以及是否开启调试模式等设置。
 
-          datasource_ids: 直接与数据源对话，我们会自动为你创建一个机器人, only support one datasource
-
           name: New name for the chat
 
           role_id: 角色 ID，将扮演这个角色来执行对话，用于权限控制。若无，则跳过鉴权，即可查询所有
@@ -306,7 +299,6 @@ async def create(
             body=await async_maybe_transform(
                 {
                     "bot_id": bot_id,
-                    "datasource_ids": datasource_ids,
                     "name": name,
                     "role_id": role_id,
                     "role_variables": role_variables,

src/asktable/types/chat.py

@@ -1,6 +1,6 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
-from typing import Dict, List, Union, Optional
+from typing import Dict, Union, Optional
 from datetime import datetime
 
 from .._models import BaseModel
@@ -24,9 +24,6 @@ class Chat(BaseModel):
     可以访问的数据、可以执行的任务以及是否开启调试模式等设置。
     """
 
-    datasource_ids: Optional[List[str]] = None
-    """直接与数据源对话，我们会自动为你创建一个机器人, only support one datasource"""
-
     latest_msg: Optional[datetime] = None
 
     name: Optional[str] = None

src/asktable/types/chat_create_params.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from typing import Dict, List, Union, Optional
+from typing import Dict, Union, Optional
 from typing_extensions import TypedDict
 
 __all__ = ["ChatCreateParams"]
@@ -15,9 +15,6 @@ class ChatCreateParams(TypedDict, total=False):
     可以访问的数据、可以执行的任务以及是否开启调试模式等设置。
     """
 
-    datasource_ids: Optional[List[str]]
-    """直接与数据源对话，我们会自动为你创建一个机器人, only support one datasource"""
-
     name: Optional[str]
     """New name for the chat"""
 

src/asktable/types/chat_list_response.py

@@ -32,9 +32,6 @@ class Item(BaseModel):
     可以访问的数据、可以执行的任务以及是否开启调试模式等设置。
     """
 
-    datasource_ids: Optional[List[str]] = None
-    """直接与数据源对话，我们会自动为你创建一个机器人, only support one datasource"""
-
     latest_msg: Optional[datetime] = None
     """最新消息时间"""
 

src/asktable/types/chat_out.py

@@ -1,6 +1,6 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
-from typing import Dict, List, Union, Optional
+from typing import Dict, Union, Optional
 from datetime import datetime
 
 from .._models import BaseModel
@@ -24,9 +24,6 @@ class ChatOut(BaseModel):
     可以访问的数据、可以执行的任务以及是否开启调试模式等设置。
     """
 
-    datasource_ids: Optional[List[str]] = None
-    """直接与数据源对话，我们会自动为你创建一个机器人, only support one datasource"""
-
     latest_msg: Optional[datetime] = None
 
     name: Optional[str] = None

tests/api_resources/test_chats.py

@@ -26,7 +26,6 @@ def test_method_create(self, client: Asktable) -> None:
     def test_method_create_with_all_params(self, client: Asktable) -> None:
         chat = client.chats.create(
             bot_id="bot_42",
-            datasource_ids=["ds_42"],
             name="name",
             role_id="role_42",
             role_variables={"id": "123123123"},
@@ -180,7 +179,6 @@ async def test_method_create(self, async_client: AsyncAsktable) -> None:
     async def test_method_create_with_all_params(self, async_client: AsyncAsktable) -> None:
         chat = await async_client.chats.create(
             bot_id="bot_42",
-            datasource_ids=["ds_42"],
             name="name",
             role_id="role_42",
             role_variables={"id": "123123123"},

tests/test_client.py

@@ -682,6 +682,7 @@ class Model(BaseModel):
             [3, "", 0.5],
             [2, "", 0.5 * 2.0],
             [1, "", 0.5 * 4.0],
+            [-1100, "", 7.8],  # test large number potentially overflowing
         ],
     )
     @mock.patch("time.time", mock.MagicMock(return_value=1696004797))
@@ -1441,6 +1442,7 @@ class Model(BaseModel):
             [3, "", 0.5],
             [2, "", 0.5 * 2.0],
             [1, "", 0.5 * 4.0],
+            [-1100, "", 7.8],  # test large number potentially overflowing
         ],
     )
     @mock.patch("time.time", mock.MagicMock(return_value=1696004797))