feat(api): manual updates (#167)

Author: stainless-app[bot]

.stats.yml

@@ -1,2 +1,2 @@
-configured_endpoints: 91
+configured_endpoints: 94
 openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/datamini%2Fasktable-2d9923ffe6d2d59041bac841db518829933a84e3251a372c9ad2306473fd7488.yml

api.md

@@ -103,7 +103,14 @@ Methods:
 Types:
 
 ```python
-from asktable.types import AIMessage, Chat, ToolMessage, UserMessage, ChatRetrieveResponse
+from asktable.types import (
+    AIMessage,
+    Chat,
+    ToolMessage,
+    UserMessage,
+    ChatRetrieveResponse,
+    ChatPostMessageResponse,
+)
 ```
 
 Methods:
@@ -112,6 +119,7 @@ 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_retrieve_response.py">ChatRetrieveResponse</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">post_message</a>(chat_id, \*\*<a href="src/asktable/types/chat_post_message_params.py">params</a>) -> <a href="./src/asktable/types/chat_post_message_response.py">ChatPostMessageResponse</a></code>
 
 ## Messages
 
@@ -381,3 +389,27 @@ from asktable.types import ScoreCreateResponse
 Methods:
 
 - <code title="post /score">client.scores.<a href="./src/asktable/resources/scores.py">create</a>(\*\*<a href="src/asktable/types/score_create_params.py">params</a>) -> <a href="./src/asktable/types/score_create_response.py">ScoreCreateResponse</a></code>
+
+# Files
+
+Types:
+
+```python
+from asktable.types import FileRetrieveResponse
+```
+
+Methods:
+
+- <code title="get /files/{file_id}">client.files.<a href="./src/asktable/resources/files.py">retrieve</a>(file_id) -> <a href="./src/asktable/types/file_retrieve_response.py">object</a></code>
+
+# Dataframes
+
+Types:
+
+```python
+from asktable.types import DataframeRetrieveResponse
+```
+
+Methods:
+
+- <code title="get /dataframes/{dataframe_id}">client.dataframes.<a href="./src/asktable/resources/dataframes.py">retrieve</a>(dataframe_id) -> <a href="./src/asktable/types/dataframe_retrieve_response.py">DataframeRetrieveResponse</a></code>

src/asktable/_client.py

@@ -28,13 +28,15 @@
     auth,
     bots,
     sqls,
+    files,
     roles,
     caches,
     scores,
     answers,
     project,
     policies,
     trainings,
+    dataframes,
     integration,
     preferences,
     securetunnels,
@@ -83,6 +85,8 @@ class Asktable(SyncAPIClient):
     trainings: trainings.TrainingsResource
     project: project.ProjectResource
     scores: scores.ScoresResource
+    files: files.FilesResource
+    dataframes: dataframes.DataframesResource
     with_raw_response: AsktableWithRawResponse
     with_streaming_response: AsktableWithStreamedResponse
 
@@ -158,6 +162,8 @@ def __init__(
         self.trainings = trainings.TrainingsResource(self)
         self.project = project.ProjectResource(self)
         self.scores = scores.ScoresResource(self)
+        self.files = files.FilesResource(self)
+        self.dataframes = dataframes.DataframesResource(self)
         self.with_raw_response = AsktableWithRawResponse(self)
         self.with_streaming_response = AsktableWithStreamedResponse(self)
 
@@ -285,6 +291,8 @@ class AsyncAsktable(AsyncAPIClient):
     trainings: trainings.AsyncTrainingsResource
     project: project.AsyncProjectResource
     scores: scores.AsyncScoresResource
+    files: files.AsyncFilesResource
+    dataframes: dataframes.AsyncDataframesResource
     with_raw_response: AsyncAsktableWithRawResponse
     with_streaming_response: AsyncAsktableWithStreamedResponse
 
@@ -360,6 +368,8 @@ def __init__(
         self.trainings = trainings.AsyncTrainingsResource(self)
         self.project = project.AsyncProjectResource(self)
         self.scores = scores.AsyncScoresResource(self)
+        self.files = files.AsyncFilesResource(self)
+        self.dataframes = dataframes.AsyncDataframesResource(self)
         self.with_raw_response = AsyncAsktableWithRawResponse(self)
         self.with_streaming_response = AsyncAsktableWithStreamedResponse(self)
 
@@ -488,6 +498,8 @@ def __init__(self, client: Asktable) -> None:
         self.trainings = trainings.TrainingsResourceWithRawResponse(client.trainings)
         self.project = project.ProjectResourceWithRawResponse(client.project)
         self.scores = scores.ScoresResourceWithRawResponse(client.scores)
+        self.files = files.FilesResourceWithRawResponse(client.files)
+        self.dataframes = dataframes.DataframesResourceWithRawResponse(client.dataframes)
 
 
 class AsyncAsktableWithRawResponse:
@@ -512,6 +524,8 @@ def __init__(self, client: AsyncAsktable) -> None:
         self.trainings = trainings.AsyncTrainingsResourceWithRawResponse(client.trainings)
         self.project = project.AsyncProjectResourceWithRawResponse(client.project)
         self.scores = scores.AsyncScoresResourceWithRawResponse(client.scores)
+        self.files = files.AsyncFilesResourceWithRawResponse(client.files)
+        self.dataframes = dataframes.AsyncDataframesResourceWithRawResponse(client.dataframes)
 
 
 class AsktableWithStreamedResponse:
@@ -536,6 +550,8 @@ def __init__(self, client: Asktable) -> None:
         self.trainings = trainings.TrainingsResourceWithStreamingResponse(client.trainings)
         self.project = project.ProjectResourceWithStreamingResponse(client.project)
         self.scores = scores.ScoresResourceWithStreamingResponse(client.scores)
+        self.files = files.FilesResourceWithStreamingResponse(client.files)
+        self.dataframes = dataframes.DataframesResourceWithStreamingResponse(client.dataframes)
 
 
 class AsyncAsktableWithStreamedResponse:
@@ -560,6 +576,8 @@ def __init__(self, client: AsyncAsktable) -> None:
         self.trainings = trainings.AsyncTrainingsResourceWithStreamingResponse(client.trainings)
         self.project = project.AsyncProjectResourceWithStreamingResponse(client.project)
         self.scores = scores.AsyncScoresResourceWithStreamingResponse(client.scores)
+        self.files = files.AsyncFilesResourceWithStreamingResponse(client.files)
+        self.dataframes = dataframes.AsyncDataframesResourceWithStreamingResponse(client.dataframes)
 
 
 Client = Asktable

src/asktable/resources/__init__.py

@@ -40,6 +40,14 @@
     ChatsResourceWithStreamingResponse,
     AsyncChatsResourceWithStreamingResponse,
 )
+from .files import (
+    FilesResource,
+    AsyncFilesResource,
+    FilesResourceWithRawResponse,
+    AsyncFilesResourceWithRawResponse,
+    FilesResourceWithStreamingResponse,
+    AsyncFilesResourceWithStreamingResponse,
+)
 from .roles import (
     RolesResource,
     AsyncRolesResource,
@@ -104,6 +112,14 @@
     TrainingsResourceWithStreamingResponse,
     AsyncTrainingsResourceWithStreamingResponse,
 )
+from .dataframes import (
+    DataframesResource,
+    AsyncDataframesResource,
+    DataframesResourceWithRawResponse,
+    AsyncDataframesResourceWithRawResponse,
+    DataframesResourceWithStreamingResponse,
+    AsyncDataframesResourceWithStreamingResponse,
+)
 from .datasources import (
     DatasourcesResource,
     AsyncDatasourcesResource,
@@ -254,4 +270,16 @@
     "AsyncScoresResourceWithRawResponse",
     "ScoresResourceWithStreamingResponse",
     "AsyncScoresResourceWithStreamingResponse",
+    "FilesResource",
+    "AsyncFilesResource",
+    "FilesResourceWithRawResponse",
+    "AsyncFilesResourceWithRawResponse",
+    "FilesResourceWithStreamingResponse",
+    "AsyncFilesResourceWithStreamingResponse",
+    "DataframesResource",
+    "AsyncDataframesResource",
+    "DataframesResourceWithRawResponse",
+    "AsyncDataframesResourceWithRawResponse",
+    "DataframesResourceWithStreamingResponse",
+    "AsyncDataframesResourceWithStreamingResponse",
 ]

src/asktable/resources/chats/chats.py

@@ -2,11 +2,11 @@
 
 from __future__ import annotations
 
-from typing import Dict, Union, Optional
+from typing import Any, Dict, Union, Optional, cast
 
 import httpx
 
-from ...types import chat_list_params, chat_create_params
+from ...types import chat_list_params, chat_create_params, chat_post_message_params
 from ..._types import NOT_GIVEN, Body, Query, Headers, NoneType, NotGiven
 from ..._utils import (
     maybe_transform,
@@ -32,6 +32,7 @@
 from ...types.chat import Chat
 from ..._base_client import AsyncPaginator, make_request_options
 from ...types.chat_retrieve_response import ChatRetrieveResponse
+from ...types.chat_post_message_response import ChatPostMessageResponse
 
 __all__ = ["ChatsResource", "AsyncChatsResource"]
 
@@ -231,6 +232,49 @@ def delete(
             cast_to=NoneType,
         )
 
+    def post_message(
+        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,
+    ) -> ChatPostMessageResponse:
+        """
+        发消息
+
+        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 cast(
+            ChatPostMessageResponse,
+            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_post_message_params.ChatPostMessageParams),
+                ),
+                cast_to=cast(
+                    Any, ChatPostMessageResponse
+                ),  # Union types cannot be passed in as arguments in the type system
+            ),
+        )
+
 
 class AsyncChatsResource(AsyncAPIResource):
     @cached_property
@@ -427,6 +471,51 @@ async def delete(
             cast_to=NoneType,
         )
 
+    async def post_message(
+        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,
+    ) -> ChatPostMessageResponse:
+        """
+        发消息
+
+        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 cast(
+            ChatPostMessageResponse,
+            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_post_message_params.ChatPostMessageParams
+                    ),
+                ),
+                cast_to=cast(
+                    Any, ChatPostMessageResponse
+                ),  # Union types cannot be passed in as arguments in the type system
+            ),
+        )
+
 
 class ChatsResourceWithRawResponse:
     def __init__(self, chats: ChatsResource) -> None:
@@ -444,6 +533,9 @@ def __init__(self, chats: ChatsResource) -> None:
         self.delete = to_raw_response_wrapper(
             chats.delete,
         )
+        self.post_message = to_raw_response_wrapper(
+            chats.post_message,
+        )
 
     @cached_property
     def messages(self) -> MessagesResourceWithRawResponse:
@@ -466,6 +558,9 @@ def __init__(self, chats: AsyncChatsResource) -> None:
         self.delete = async_to_raw_response_wrapper(
             chats.delete,
         )
+        self.post_message = async_to_raw_response_wrapper(
+            chats.post_message,
+        )
 
     @cached_property
     def messages(self) -> AsyncMessagesResourceWithRawResponse:
@@ -488,6 +583,9 @@ def __init__(self, chats: ChatsResource) -> None:
         self.delete = to_streamed_response_wrapper(
             chats.delete,
         )
+        self.post_message = to_streamed_response_wrapper(
+            chats.post_message,
+        )
 
     @cached_property
     def messages(self) -> MessagesResourceWithStreamingResponse:
@@ -510,6 +608,9 @@ def __init__(self, chats: AsyncChatsResource) -> None:
         self.delete = async_to_streamed_response_wrapper(
             chats.delete,
         )
+        self.post_message = async_to_streamed_response_wrapper(
+            chats.post_message,
+        )
 
     @cached_property
     def messages(self) -> AsyncMessagesResourceWithStreamingResponse: