Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion .stats.yml
Original file line number Diff line number Diff line change
@@ -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
34 changes: 33 additions & 1 deletion api.md
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand All @@ -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

Expand Down Expand Up @@ -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>
18 changes: 18 additions & 0 deletions src/asktable/_client.py
Original file line number Diff line number Diff line change
Expand Up @@ -28,13 +28,15 @@
auth,
bots,
sqls,
files,
roles,
caches,
scores,
answers,
project,
policies,
trainings,
dataframes,
integration,
preferences,
securetunnels,
Expand Down Expand Up @@ -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

Expand Down Expand Up @@ -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)

Expand Down Expand Up @@ -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

Expand Down Expand Up @@ -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)

Expand Down Expand Up @@ -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:
Expand All @@ -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:
Expand All @@ -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:
Expand All @@ -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
Expand Down
28 changes: 28 additions & 0 deletions src/asktable/resources/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -40,6 +40,14 @@
ChatsResourceWithStreamingResponse,
AsyncChatsResourceWithStreamingResponse,
)
from .files import (
FilesResource,
AsyncFilesResource,
FilesResourceWithRawResponse,
AsyncFilesResourceWithRawResponse,
FilesResourceWithStreamingResponse,
AsyncFilesResourceWithStreamingResponse,
)
from .roles import (
RolesResource,
AsyncRolesResource,
Expand Down Expand Up @@ -104,6 +112,14 @@
TrainingsResourceWithStreamingResponse,
AsyncTrainingsResourceWithStreamingResponse,
)
from .dataframes import (
DataframesResource,
AsyncDataframesResource,
DataframesResourceWithRawResponse,
AsyncDataframesResourceWithRawResponse,
DataframesResourceWithStreamingResponse,
AsyncDataframesResourceWithStreamingResponse,
)
from .datasources import (
DatasourcesResource,
AsyncDatasourcesResource,
Expand Down Expand Up @@ -254,4 +270,16 @@
"AsyncScoresResourceWithRawResponse",
"ScoresResourceWithStreamingResponse",
"AsyncScoresResourceWithStreamingResponse",
"FilesResource",
"AsyncFilesResource",
"FilesResourceWithRawResponse",
"AsyncFilesResourceWithRawResponse",
"FilesResourceWithStreamingResponse",
"AsyncFilesResourceWithStreamingResponse",
"DataframesResource",
"AsyncDataframesResource",
"DataframesResourceWithRawResponse",
"AsyncDataframesResourceWithRawResponse",
"DataframesResourceWithStreamingResponse",
"AsyncDataframesResourceWithStreamingResponse",
]
105 changes: 103 additions & 2 deletions src/asktable/resources/chats/chats.py
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand All @@ -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"]

Expand Down Expand Up @@ -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
Expand Down Expand Up @@ -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:
Expand All @@ -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:
Expand All @@ -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:
Expand All @@ -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:
Expand All @@ -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:
Expand Down
Loading