feat(api): update via SDK Studio

Author: stainless-bot

.stats.yml

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

api.md

@@ -142,7 +142,7 @@ from asktable.types import Chat, ChatOut, ChatListResponse
 
 Methods:
 
-- <code title="post /chats/{chat_id}">client.chats.<a href="./src/asktable/resources/chats/chats.py">create</a>(chat_id, \*\*<a href="src/asktable/types/chat_create_params.py">params</a>) -> <a href="./src/asktable/types/shared/message.py">Message</a></code>
+- <code title="post /chats">client.chats.<a href="./src/asktable/resources/chats/chats.py">create</a>(\*\*<a href="src/asktable/types/chat_create_params.py">params</a>) -> <a href="./src/asktable/types/chat.py">Chat</a></code>
 - <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_out.py">ChatOut</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_list_response.py">ChatListResponse</a></code>
 - <code title="delete /chats/{chat_id}">client.chats.<a href="./src/asktable/resources/chats/chats.py">delete</a>(chat_id) -> None</code>
@@ -157,6 +157,7 @@ from asktable.types.chats import MessageListResponse
 
 Methods:
 
+- <code title="post /chats/{chat_id}">client.chats.messages.<a href="./src/asktable/resources/chats/messages.py">create</a>(chat_id, \*\*<a href="src/asktable/types/chats/message_create_params.py">params</a>) -> <a href="./src/asktable/types/shared/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/shared/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/chats/message_list_response.py">MessageListResponse</a></code>
 

src/asktable/resources/chats/chats.py

@@ -2,6 +2,8 @@
 
 from __future__ import annotations
 
+from typing import Dict, List, Union, Optional
+
 import httpx
 
 from ...types import chat_list_params, chat_create_params
@@ -26,9 +28,9 @@
     async_to_raw_response_wrapper,
     async_to_streamed_response_wrapper,
 )
+from ...types.chat import Chat
 from ..._base_client import make_request_options
 from ...types.chat_out import ChatOut
-from ...types.shared.message import Message
 from ...types.chat_list_response import ChatListResponse
 
 __all__ = ["ChatsResource", "AsyncChatsResource"]
@@ -60,20 +62,38 @@ def with_streaming_response(self) -> ChatsResourceWithStreamingResponse:
 
     def create(
         self,
-        chat_id: str,
         *,
-        question: str,
+        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,
+        user_profile: Optional[Dict[str, str]] | 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,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> Message:
+    ) -> Chat:
         """
-        发消息
+        创建对话
 
         Args:
+          bot_id: 机器人 ID，如果需要使用高级功能，请使用 bot_id 来创建对话。在机器人中你可以定义
+              可以访问的数据、可以执行的任务以及是否开启调试模式等设置。
+
+          datasource_ids: 直接与数据源对话，我们会自动为你创建一个机器人, only support one datasource
+
+          name: New name for the chat
+
+          role_id: 角色 ID，将扮演这个角色来执行对话，用于权限控制。若无，则跳过鉴权，即可查询所有
+              数据
+
+          role_variables: 在扮演这个角色时需要传递的变量值，用 Key-Value 形式传递
+
+          user_profile: 用户信息，用于在对话中传递用户的信息，用 Key-Value 形式传递
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -82,18 +102,23 @@ def create(
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        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}",
+            "/chats",
+            body=maybe_transform(
+                {
+                    "bot_id": bot_id,
+                    "datasource_ids": datasource_ids,
+                    "name": name,
+                    "role_id": role_id,
+                    "role_variables": role_variables,
+                    "user_profile": user_profile,
+                },
+                chat_create_params.ChatCreateParams,
+            ),
             options=make_request_options(
-                extra_headers=extra_headers,
-                extra_query=extra_query,
-                extra_body=extra_body,
-                timeout=timeout,
-                query=maybe_transform({"question": question}, chat_create_params.ChatCreateParams),
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=Message,
+            cast_to=Chat,
         )
 
     def retrieve(
@@ -236,20 +261,38 @@ def with_streaming_response(self) -> AsyncChatsResourceWithStreamingResponse:
 
     async def create(
         self,
-        chat_id: str,
         *,
-        question: str,
+        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,
+        user_profile: Optional[Dict[str, str]] | 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,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> Message:
+    ) -> Chat:
         """
-        发消息
+        创建对话
 
         Args:
+          bot_id: 机器人 ID，如果需要使用高级功能，请使用 bot_id 来创建对话。在机器人中你可以定义
+              可以访问的数据、可以执行的任务以及是否开启调试模式等设置。
+
+          datasource_ids: 直接与数据源对话，我们会自动为你创建一个机器人, only support one datasource
+
+          name: New name for the chat
+
+          role_id: 角色 ID，将扮演这个角色来执行对话，用于权限控制。若无，则跳过鉴权，即可查询所有
+              数据
+
+          role_variables: 在扮演这个角色时需要传递的变量值，用 Key-Value 形式传递
+
+          user_profile: 用户信息，用于在对话中传递用户的信息，用 Key-Value 形式传递
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -258,18 +301,23 @@ async def create(
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        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}",
+            "/chats",
+            body=await async_maybe_transform(
+                {
+                    "bot_id": bot_id,
+                    "datasource_ids": datasource_ids,
+                    "name": name,
+                    "role_id": role_id,
+                    "role_variables": role_variables,
+                    "user_profile": user_profile,
+                },
+                chat_create_params.ChatCreateParams,
+            ),
             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_create_params.ChatCreateParams),
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=Message,
+            cast_to=Chat,
         )
 
     async def retrieve(

src/asktable/resources/chats/messages.py

@@ -17,7 +17,7 @@
     async_to_raw_response_wrapper,
     async_to_streamed_response_wrapper,
 )
-from ...types.chats import message_list_params
+from ...types.chats import message_list_params, message_create_params
 from ..._base_client import make_request_options
 from ...types.shared.message import Message
 from ...types.chats.message_list_response import MessageListResponse
@@ -45,6 +45,44 @@ def with_streaming_response(self) -> MessagesResourceWithStreamingResponse:
         """
         return MessagesResourceWithStreamingResponse(self)
 
+    def create(
+        self,
+        chat_id: str,
+        *,
+        question: str,
+        # 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,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> Message:
+        """
+        发消息
+
+        Args:
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        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}, message_create_params.MessageCreateParams),
+            ),
+            cast_to=Message,
+        )
+
     def retrieve(
         self,
         message_id: str,
@@ -151,6 +189,44 @@ def with_streaming_response(self) -> AsyncMessagesResourceWithStreamingResponse:
         """
         return AsyncMessagesResourceWithStreamingResponse(self)
 
+    async def create(
+        self,
+        chat_id: str,
+        *,
+        question: str,
+        # 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,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> Message:
+        """
+        发消息
+
+        Args:
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        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}, message_create_params.MessageCreateParams),
+            ),
+            cast_to=Message,
+        )
+
     async def retrieve(
         self,
         message_id: str,
@@ -241,6 +317,9 @@ class MessagesResourceWithRawResponse:
     def __init__(self, messages: MessagesResource) -> None:
         self._messages = messages
 
+        self.create = to_raw_response_wrapper(
+            messages.create,
+        )
         self.retrieve = to_raw_response_wrapper(
             messages.retrieve,
         )
@@ -253,6 +332,9 @@ class AsyncMessagesResourceWithRawResponse:
     def __init__(self, messages: AsyncMessagesResource) -> None:
         self._messages = messages
 
+        self.create = async_to_raw_response_wrapper(
+            messages.create,
+        )
         self.retrieve = async_to_raw_response_wrapper(
             messages.retrieve,
         )
@@ -265,6 +347,9 @@ class MessagesResourceWithStreamingResponse:
     def __init__(self, messages: MessagesResource) -> None:
         self._messages = messages
 
+        self.create = to_streamed_response_wrapper(
+            messages.create,
+        )
         self.retrieve = to_streamed_response_wrapper(
             messages.retrieve,
         )
@@ -277,6 +362,9 @@ class AsyncMessagesResourceWithStreamingResponse:
     def __init__(self, messages: AsyncMessagesResource) -> None:
         self._messages = messages
 
+        self.create = async_to_streamed_response_wrapper(
+            messages.create,
+        )
         self.retrieve = async_to_streamed_response_wrapper(
             messages.retrieve,
         )

src/asktable/types/__init__.py

@@ -2,6 +2,7 @@
 
 from __future__ import annotations
 
+from .chat import Chat as Chat
 from .role import Role as Role
 from .shared import Policy as Policy, Message as Message, AnswerModel as AnswerModel
 from .chat_bot import ChatBot as ChatBot

src/asktable/types/chat.py

@@ -0,0 +1,45 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Dict, List, Union, Optional
+from datetime import datetime
+
+from .._models import BaseModel
+
+__all__ = ["Chat"]
+
+
+class Chat(BaseModel):
+    id: str
+    """对话 ID"""
+
+    created: datetime
+
+    modified: datetime
+
+    project_id: str
+
+    bot_id: Optional[str] = None
+    """
+    机器人 ID，如果需要使用高级功能，请使用 bot_id 来创建对话。在机器人中你可以定义
+    可以访问的数据、可以执行的任务以及是否开启调试模式等设置。
+    """
+
+    datasource_ids: Optional[List[str]] = None
+    """直接与数据源对话，我们会自动为你创建一个机器人, only support one datasource"""
+
+    latest_msg: Optional[datetime] = None
+
+    name: Optional[str] = None
+    """New name for the chat"""
+
+    role_id: Optional[str] = None
+    """
+    角色 ID，将扮演这个角色来执行对话，用于权限控制。若无，则跳过鉴权，即可查询所有
+    数据
+    """
+
+    role_variables: Optional[Dict[str, Union[str, int, bool]]] = None
+    """在扮演这个角色时需要传递的变量值，用 Key-Value 形式传递"""
+
+    user_profile: Optional[Dict[str, str]] = None
+    """用户信息，用于在对话中传递用户的信息，用 Key-Value 形式传递"""