feat(api): api update (#68)

Author: stainless-app[bot]

.stats.yml

@@ -1,2 +1,2 @@
 configured_endpoints: 82
-openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/datamini%2Fasktable-c4b030c81467ff9b675870c9a966dec4119d6ff1b92c39fc3aab8fac9fcbf1f7.yml
+openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/datamini%2Fasktable-ad23f3fb1367b9be154f172d0cde5cebc3bc454368108debe6bdf80da2f4bebc.yml

api.md

@@ -103,7 +103,7 @@ Methods:
 Types:
 
 ```python
-from asktable.types import Chat, Message
+from asktable.types import Chat, ChatSendMessageResponse
 ```
 
 Methods:
@@ -112,15 +112,25 @@ Methods:
 - <code title="get /chats/{chat_id}">client.chats.<a href="./src/asktable/resources/chats/chats.py">retrieve</a>(chat_id) -> <a href="./src/asktable/types/chat.py">Chat</a></code>
 - <code title="get /chats">client.chats.<a href="./src/asktable/resources/chats/chats.py">list</a>(\*\*<a href="src/asktable/types/chat_list_params.py">params</a>) -> <a href="./src/asktable/types/chat.py">SyncPage[Chat]</a></code>
 - <code title="delete /chats/{chat_id}">client.chats.<a href="./src/asktable/resources/chats/chats.py">delete</a>(chat_id) -> None</code>
-- <code title="post /chats/{chat_id}">client.chats.<a href="./src/asktable/resources/chats/chats.py">send_message</a>(chat_id, \*\*<a href="src/asktable/types/chat_send_message_params.py">params</a>) -> <a href="./src/asktable/types/message.py">Message</a></code>
+- <code title="post /chats/{chat_id}">client.chats.<a href="./src/asktable/resources/chats/chats.py">send_message</a>(chat_id, \*\*<a href="src/asktable/types/chat_send_message_params.py">params</a>) -> <a href="./src/asktable/types/chat_send_message_response.py">ChatSendMessageResponse</a></code>
 
 ## Messages
 
+Types:
+
+```python
+from asktable.types.chats import (
+    MessageRetrieveResponse,
+    MessageListResponse,
+    MessageSendMessageResponse,
+)
+```
+
 Methods:
 
-- <code title="get /chats/{chat_id}/messages/{message_id}">client.chats.messages.<a href="./src/asktable/resources/chats/messages.py">retrieve</a>(message_id, \*, chat_id) -> <a href="./src/asktable/types/message.py">Message</a></code>
-- <code title="get /chats/{chat_id}/messages">client.chats.messages.<a href="./src/asktable/resources/chats/messages.py">list</a>(chat_id, \*\*<a href="src/asktable/types/chats/message_list_params.py">params</a>) -> <a href="./src/asktable/types/message.py">SyncPage[Message]</a></code>
-- <code title="post /chats/{chat_id}/messages">client.chats.messages.<a href="./src/asktable/resources/chats/messages.py">send_message</a>(chat_id, \*\*<a href="src/asktable/types/chats/message_send_message_params.py">params</a>) -> <a href="./src/asktable/types/message.py">Message</a></code>
+- <code title="get /chats/{chat_id}/messages/{message_id}">client.chats.messages.<a href="./src/asktable/resources/chats/messages.py">retrieve</a>(message_id, \*, chat_id) -> <a href="./src/asktable/types/chats/message_retrieve_response.py">MessageRetrieveResponse</a></code>
+- <code title="get /chats/{chat_id}/messages">client.chats.messages.<a href="./src/asktable/resources/chats/messages.py">list</a>(chat_id, \*\*<a href="src/asktable/types/chats/message_list_params.py">params</a>) -> <a href="./src/asktable/types/chats/message_list_response.py">SyncPage[MessageListResponse]</a></code>
+- <code title="post /chats/{chat_id}/messages">client.chats.messages.<a href="./src/asktable/resources/chats/messages.py">send_message</a>(chat_id, \*\*<a href="src/asktable/types/chats/message_send_message_params.py">params</a>) -> <a href="./src/asktable/types/chats/message_send_message_response.py">MessageSendMessageResponse</a></code>
 
 # Datasources
 

src/asktable/resources/chats/chats.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from typing import Dict, Union, Optional
+from typing import Any, Dict, Union, Optional, cast
 
 import httpx
 
@@ -31,7 +31,7 @@
 from ...pagination import SyncPage, AsyncPage
 from ...types.chat import Chat
 from ..._base_client import AsyncPaginator, make_request_options
-from ...types.message import Message
+from ...types.chat_send_message_response import ChatSendMessageResponse
 
 __all__ = ["ChatsResource", "AsyncChatsResource"]
 
@@ -242,7 +242,7 @@ def send_message(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> Message:
+    ) -> ChatSendMessageResponse:
         """
         发消息
 
@@ -257,16 +257,21 @@ def send_message(
         """
         if not chat_id:
             raise ValueError(f"Expected a non-empty value for `chat_id` but received {chat_id!r}")
-        return self._post(
-            f"/chats/{chat_id}",
-            options=make_request_options(
-                extra_headers=extra_headers,
-                extra_query=extra_query,
-                extra_body=extra_body,
-                timeout=timeout,
-                query=maybe_transform({"question": question}, chat_send_message_params.ChatSendMessageParams),
+        return cast(
+            ChatSendMessageResponse,
+            self._post(
+                f"/chats/{chat_id}",
+                options=make_request_options(
+                    extra_headers=extra_headers,
+                    extra_query=extra_query,
+                    extra_body=extra_body,
+                    timeout=timeout,
+                    query=maybe_transform({"question": question}, chat_send_message_params.ChatSendMessageParams),
+                ),
+                cast_to=cast(
+                    Any, ChatSendMessageResponse
+                ),  # Union types cannot be passed in as arguments in the type system
             ),
-            cast_to=Message,
         )
 
 
@@ -476,7 +481,7 @@ async def send_message(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> Message:
+    ) -> ChatSendMessageResponse:
         """
         发消息
 
@@ -491,18 +496,23 @@ async def send_message(
         """
         if not chat_id:
             raise ValueError(f"Expected a non-empty value for `chat_id` but received {chat_id!r}")
-        return await self._post(
-            f"/chats/{chat_id}",
-            options=make_request_options(
-                extra_headers=extra_headers,
-                extra_query=extra_query,
-                extra_body=extra_body,
-                timeout=timeout,
-                query=await async_maybe_transform(
-                    {"question": question}, chat_send_message_params.ChatSendMessageParams
+        return cast(
+            ChatSendMessageResponse,
+            await self._post(
+                f"/chats/{chat_id}",
+                options=make_request_options(
+                    extra_headers=extra_headers,
+                    extra_query=extra_query,
+                    extra_body=extra_body,
+                    timeout=timeout,
+                    query=await async_maybe_transform(
+                        {"question": question}, chat_send_message_params.ChatSendMessageParams
+                    ),
                 ),
+                cast_to=cast(
+                    Any, ChatSendMessageResponse
+                ),  # Union types cannot be passed in as arguments in the type system
             ),
-            cast_to=Message,
         )
 
 

src/asktable/resources/chats/messages.py

@@ -2,6 +2,8 @@
 
 from __future__ import annotations
 
+from typing import Any, cast
+
 import httpx
 
 from ..._types import NOT_GIVEN, Body, Query, Headers, NotGiven
@@ -20,7 +22,9 @@
 from ...pagination import SyncPage, AsyncPage
 from ...types.chats import message_list_params, message_send_message_params
 from ..._base_client import AsyncPaginator, make_request_options
-from ...types.message import Message
+from ...types.chats.message_list_response import MessageListResponse
+from ...types.chats.message_retrieve_response import MessageRetrieveResponse
+from ...types.chats.message_send_message_response import MessageSendMessageResponse
 
 __all__ = ["MessagesResource", "AsyncMessagesResource"]
 
@@ -56,7 +60,7 @@ def retrieve(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> Message:
+    ) -> MessageRetrieveResponse:
         """
         查询某条消息
 
@@ -73,12 +77,17 @@ def retrieve(
             raise ValueError(f"Expected a non-empty value for `chat_id` but received {chat_id!r}")
         if not message_id:
             raise ValueError(f"Expected a non-empty value for `message_id` but received {message_id!r}")
-        return self._get(
-            f"/chats/{chat_id}/messages/{message_id}",
-            options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+        return cast(
+            MessageRetrieveResponse,
+            self._get(
+                f"/chats/{chat_id}/messages/{message_id}",
+                options=make_request_options(
+                    extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+                ),
+                cast_to=cast(
+                    Any, MessageRetrieveResponse
+                ),  # Union types cannot be passed in as arguments in the type system
             ),
-            cast_to=Message,
         )
 
     def list(
@@ -93,7 +102,7 @@ def list(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> SyncPage[Message]:
+    ) -> SyncPage[MessageListResponse]:
         """
         查询所有的消息
 
@@ -114,7 +123,7 @@ def list(
             raise ValueError(f"Expected a non-empty value for `chat_id` but received {chat_id!r}")
         return self._get_api_list(
             f"/chats/{chat_id}/messages",
-            page=SyncPage[Message],
+            page=SyncPage[MessageListResponse],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -128,7 +137,7 @@ def list(
                     message_list_params.MessageListParams,
                 ),
             ),
-            model=Message,
+            model=cast(Any, MessageListResponse),  # Union types cannot be passed in as arguments in the type system
         )
 
     def send_message(
@@ -142,7 +151,7 @@ def send_message(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> Message:
+    ) -> MessageSendMessageResponse:
         """
         发消息
 
@@ -157,16 +166,21 @@ def send_message(
         """
         if not chat_id:
             raise ValueError(f"Expected a non-empty value for `chat_id` but received {chat_id!r}")
-        return self._post(
-            f"/chats/{chat_id}/messages",
-            options=make_request_options(
-                extra_headers=extra_headers,
-                extra_query=extra_query,
-                extra_body=extra_body,
-                timeout=timeout,
-                query=maybe_transform({"question": question}, message_send_message_params.MessageSendMessageParams),
+        return cast(
+            MessageSendMessageResponse,
+            self._post(
+                f"/chats/{chat_id}/messages",
+                options=make_request_options(
+                    extra_headers=extra_headers,
+                    extra_query=extra_query,
+                    extra_body=extra_body,
+                    timeout=timeout,
+                    query=maybe_transform({"question": question}, message_send_message_params.MessageSendMessageParams),
+                ),
+                cast_to=cast(
+                    Any, MessageSendMessageResponse
+                ),  # Union types cannot be passed in as arguments in the type system
             ),
-            cast_to=Message,
         )
 
 
@@ -201,7 +215,7 @@ async def retrieve(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> Message:
+    ) -> MessageRetrieveResponse:
         """
         查询某条消息
 
@@ -218,12 +232,17 @@ async def retrieve(
             raise ValueError(f"Expected a non-empty value for `chat_id` but received {chat_id!r}")
         if not message_id:
             raise ValueError(f"Expected a non-empty value for `message_id` but received {message_id!r}")
-        return await self._get(
-            f"/chats/{chat_id}/messages/{message_id}",
-            options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+        return cast(
+            MessageRetrieveResponse,
+            await self._get(
+                f"/chats/{chat_id}/messages/{message_id}",
+                options=make_request_options(
+                    extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+                ),
+                cast_to=cast(
+                    Any, MessageRetrieveResponse
+                ),  # Union types cannot be passed in as arguments in the type system
             ),
-            cast_to=Message,
         )
 
     def list(
@@ -238,7 +257,7 @@ def list(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> AsyncPaginator[Message, AsyncPage[Message]]:
+    ) -> AsyncPaginator[MessageListResponse, AsyncPage[MessageListResponse]]:
         """
         查询所有的消息
 
@@ -259,7 +278,7 @@ def list(
             raise ValueError(f"Expected a non-empty value for `chat_id` but received {chat_id!r}")
         return self._get_api_list(
             f"/chats/{chat_id}/messages",
-            page=AsyncPage[Message],
+            page=AsyncPage[MessageListResponse],
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -273,7 +292,7 @@ def list(
                     message_list_params.MessageListParams,
                 ),
             ),
-            model=Message,
+            model=cast(Any, MessageListResponse),  # Union types cannot be passed in as arguments in the type system
         )
 
     async def send_message(
@@ -287,7 +306,7 @@ async def send_message(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> Message:
+    ) -> MessageSendMessageResponse:
         """
         发消息
 
@@ -302,18 +321,23 @@ async def send_message(
         """
         if not chat_id:
             raise ValueError(f"Expected a non-empty value for `chat_id` but received {chat_id!r}")
-        return await self._post(
-            f"/chats/{chat_id}/messages",
-            options=make_request_options(
-                extra_headers=extra_headers,
-                extra_query=extra_query,
-                extra_body=extra_body,
-                timeout=timeout,
-                query=await async_maybe_transform(
-                    {"question": question}, message_send_message_params.MessageSendMessageParams
+        return cast(
+            MessageSendMessageResponse,
+            await self._post(
+                f"/chats/{chat_id}/messages",
+                options=make_request_options(
+                    extra_headers=extra_headers,
+                    extra_query=extra_query,
+                    extra_body=extra_body,
+                    timeout=timeout,
+                    query=await async_maybe_transform(
+                        {"question": question}, message_send_message_params.MessageSendMessageParams
+                    ),
                 ),
+                cast_to=cast(
+                    Any, MessageSendMessageResponse
+                ),  # Union types cannot be passed in as arguments in the type system
             ),
-            cast_to=Message,
         )
 
 

src/asktable/types/__init__.py

@@ -10,7 +10,6 @@
 from .extapi import Extapi as Extapi
 from .shared import Policy as Policy
 from .chatbot import Chatbot as Chatbot
-from .message import Message as Message
 from .datasource import Datasource as Datasource
 from .secure_tunnel import SecureTunnel as SecureTunnel
 from .bot_list_params import BotListParams as BotListParams
@@ -38,6 +37,7 @@
 from .securetunnel_list_params import SecuretunnelListParams as SecuretunnelListParams
 from .role_get_polices_response import RoleGetPolicesResponse as RoleGetPolicesResponse
 from .role_get_variables_params import RoleGetVariablesParams as RoleGetVariablesParams
+from .chat_send_message_response import ChatSendMessageResponse as ChatSendMessageResponse
 from .securetunnel_create_params import SecuretunnelCreateParams as SecuretunnelCreateParams
 from .securetunnel_update_params import SecuretunnelUpdateParams as SecuretunnelUpdateParams
 from .business_glossary_list_params import BusinessGlossaryListParams as BusinessGlossaryListParams