Skip to content

Commit 95f8c27

Browse files
stainless-app[bot]stainless-bot
stainless-app[bot]
authored and
stainless-bot
committed
feat(api): api update (#41)
1 parent 7d4f6fb commit 95f8c27

94 files changed

Lines changed: 2136 additions & 781 deletions

File tree

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

‎.stats.yml‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,2 +1,2 @@
11
configured_endpoints: 69
2-
openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/datamini%2Fasktable-4d166de341d22528d6d116bb49704008e9511df1de7d5132e95b1a034a2ce95b.yml
2+
openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/datamini%2Fasktable-07404bb852c826d0cd06edde3a9b736629f599ac2308ea8ffc91ebdfc7a9403e.yml

‎api.md‎

Lines changed: 95 additions & 54 deletions
Large diffs are not rendered by default.

‎src/asktable/resources/bots.py‎

Lines changed: 15 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -21,8 +21,10 @@
2121
async_to_streamed_response_wrapper,
2222
)
2323
from .._base_client import make_request_options
24-
from ..types.chat_bot import ChatBot
2524
from ..types.bot_list_response import BotListResponse
25+
from ..types.bot_create_response import BotCreateResponse
26+
from ..types.bot_update_response import BotUpdateResponse
27+
from ..types.bot_retrieve_response import BotRetrieveResponse
2628

2729
__all__ = ["BotsResource", "AsyncBotsResource"]
2830

@@ -66,7 +68,7 @@ def create(
6668
extra_query: Query | None = None,
6769
extra_body: Body | None = None,
6870
timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
69-
) -> ChatBot:
71+
) -> BotCreateResponse:
7072
"""
7173
创建一个新的 Bot
7274
@@ -119,7 +121,7 @@ def create(
119121
options=make_request_options(
120122
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
121123
),
122-
cast_to=ChatBot,
124+
cast_to=BotCreateResponse,
123125
)
124126

125127
def retrieve(
@@ -132,7 +134,7 @@ def retrieve(
132134
extra_query: Query | None = None,
133135
extra_body: Body | None = None,
134136
timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
135-
) -> ChatBot:
137+
) -> BotRetrieveResponse:
136138
"""
137139
获取某个 Bot
138140
@@ -152,7 +154,7 @@ def retrieve(
152154
options=make_request_options(
153155
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
154156
),
155-
cast_to=ChatBot,
157+
cast_to=BotRetrieveResponse,
156158
)
157159

158160
def update(
@@ -176,7 +178,7 @@ def update(
176178
extra_query: Query | None = None,
177179
extra_body: Body | None = None,
178180
timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
179-
) -> ChatBot:
181+
) -> BotUpdateResponse:
180182
"""
181183
更新某个 Bot
182184
@@ -234,7 +236,7 @@ def update(
234236
options=make_request_options(
235237
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
236238
),
237-
cast_to=ChatBot,
239+
cast_to=BotUpdateResponse,
238240
)
239241

240242
def list(
@@ -398,7 +400,7 @@ async def create(
398400
extra_query: Query | None = None,
399401
extra_body: Body | None = None,
400402
timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
401-
) -> ChatBot:
403+
) -> BotCreateResponse:
402404
"""
403405
创建一个新的 Bot
404406
@@ -451,7 +453,7 @@ async def create(
451453
options=make_request_options(
452454
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
453455
),
454-
cast_to=ChatBot,
456+
cast_to=BotCreateResponse,
455457
)
456458

457459
async def retrieve(
@@ -464,7 +466,7 @@ async def retrieve(
464466
extra_query: Query | None = None,
465467
extra_body: Body | None = None,
466468
timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
467-
) -> ChatBot:
469+
) -> BotRetrieveResponse:
468470
"""
469471
获取某个 Bot
470472
@@ -484,7 +486,7 @@ async def retrieve(
484486
options=make_request_options(
485487
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
486488
),
487-
cast_to=ChatBot,
489+
cast_to=BotRetrieveResponse,
488490
)
489491

490492
async def update(
@@ -508,7 +510,7 @@ async def update(
508510
extra_query: Query | None = None,
509511
extra_body: Body | None = None,
510512
timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
511-
) -> ChatBot:
513+
) -> BotUpdateResponse:
512514
"""
513515
更新某个 Bot
514516
@@ -566,7 +568,7 @@ async def update(
566568
options=make_request_options(
567569
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
568570
),
569-
cast_to=ChatBot,
571+
cast_to=BotUpdateResponse,
570572
)
571573

572574
async def list(

‎src/asktable/resources/chats/messages.py‎

Lines changed: 10 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -19,8 +19,9 @@
1919
)
2020
from ...types.chats import message_list_params, message_create_params
2121
from ..._base_client import make_request_options
22-
from ...types.shared.message import Message
2322
from ...types.chats.message_list_response import MessageListResponse
23+
from ...types.chats.message_create_response import MessageCreateResponse
24+
from ...types.chats.message_retrieve_response import MessageRetrieveResponse
2425

2526
__all__ = ["MessagesResource", "AsyncMessagesResource"]
2627

@@ -56,7 +57,7 @@ def create(
5657
extra_query: Query | None = None,
5758
extra_body: Body | None = None,
5859
timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
59-
) -> Message:
60+
) -> MessageCreateResponse:
6061
"""
6162
发消息
6263
@@ -80,7 +81,7 @@ def create(
8081
timeout=timeout,
8182
query=maybe_transform({"question": question}, message_create_params.MessageCreateParams),
8283
),
83-
cast_to=Message,
84+
cast_to=MessageCreateResponse,
8485
)
8586

8687
def retrieve(
@@ -94,7 +95,7 @@ def retrieve(
9495
extra_query: Query | None = None,
9596
extra_body: Body | None = None,
9697
timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
97-
) -> Message:
98+
) -> MessageRetrieveResponse:
9899
"""
99100
查询某条消息
100101
@@ -116,7 +117,7 @@ def retrieve(
116117
options=make_request_options(
117118
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
118119
),
119-
cast_to=Message,
120+
cast_to=MessageRetrieveResponse,
120121
)
121122

122123
def list(
@@ -200,7 +201,7 @@ async def create(
200201
extra_query: Query | None = None,
201202
extra_body: Body | None = None,
202203
timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
203-
) -> Message:
204+
) -> MessageCreateResponse:
204205
"""
205206
发消息
206207
@@ -224,7 +225,7 @@ async def create(
224225
timeout=timeout,
225226
query=await async_maybe_transform({"question": question}, message_create_params.MessageCreateParams),
226227
),
227-
cast_to=Message,
228+
cast_to=MessageCreateResponse,
228229
)
229230

230231
async def retrieve(
@@ -238,7 +239,7 @@ async def retrieve(
238239
extra_query: Query | None = None,
239240
extra_body: Body | None = None,
240241
timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
241-
) -> Message:
242+
) -> MessageRetrieveResponse:
242243
"""
243244
查询某条消息
244245
@@ -260,7 +261,7 @@ async def retrieve(
260261
options=make_request_options(
261262
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
262263
),
263-
cast_to=Message,
264+
cast_to=MessageRetrieveResponse,
264265
)
265266

266267
async def list(

‎src/asktable/resources/datasources/datasources.py‎

Lines changed: 20 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -45,8 +45,11 @@
4545
AsyncUploadParamsResourceWithStreamingResponse,
4646
)
4747
from ..._base_client import make_request_options
48-
from ...types.data_source import DataSource
4948
from ...types.datasource_list_response import DatasourceListResponse
49+
from ...types.datasource_create_response import DatasourceCreateResponse
50+
from ...types.datasource_update_response import DatasourceUpdateResponse
51+
from ...types.datasource_retrieve_response import DatasourceRetrieveResponse
52+
from ...types.datasource_create_from_file_response import DatasourceCreateFromFileResponse
5053

5154
__all__ = ["DatasourcesResource", "AsyncDatasourcesResource"]
5255

@@ -92,7 +95,7 @@ def create(
9295
extra_query: Query | None = None,
9396
extra_body: Body | None = None,
9497
timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
95-
) -> DataSource:
98+
) -> DatasourceCreateResponse:
9699
"""
97100
创建一个新的数据源
98101
@@ -130,7 +133,7 @@ def create(
130133
{"async_process_meta": async_process_meta}, datasource_create_params.DatasourceCreateParams
131134
),
132135
),
133-
cast_to=DataSource,
136+
cast_to=DatasourceCreateResponse,
134137
)
135138

136139
def retrieve(
@@ -143,7 +146,7 @@ def retrieve(
143146
extra_query: Query | None = None,
144147
extra_body: Body | None = None,
145148
timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
146-
) -> DataSource:
149+
) -> DatasourceRetrieveResponse:
147150
"""
148151
根据 id 获取指定数据源
149152
@@ -163,7 +166,7 @@ def retrieve(
163166
options=make_request_options(
164167
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
165168
),
166-
cast_to=DataSource,
169+
cast_to=DatasourceRetrieveResponse,
167170
)
168171

169172
def update(
@@ -185,7 +188,7 @@ def update(
185188
extra_query: Query | None = None,
186189
extra_body: Body | None = None,
187190
timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
188-
) -> DataSource:
191+
) -> DatasourceUpdateResponse:
189192
"""
190193
更新指定数据源信息
191194
@@ -237,7 +240,7 @@ def update(
237240
options=make_request_options(
238241
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
239242
),
240-
cast_to=DataSource,
243+
cast_to=DatasourceUpdateResponse,
241244
)
242245

243246
def list(
@@ -333,7 +336,7 @@ def create_from_file(
333336
extra_query: Query | None = None,
334337
extra_body: Body | None = None,
335338
timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
336-
) -> DataSource:
339+
) -> DatasourceCreateFromFileResponse:
337340
"""
338341
上传文件并创建数据源
339342
@@ -369,7 +372,7 @@ def create_from_file(
369372
datasource_create_from_file_params.DatasourceCreateFromFileParams,
370373
),
371374
),
372-
cast_to=DataSource,
375+
cast_to=DatasourceCreateFromFileResponse,
373376
)
374377

375378

@@ -414,7 +417,7 @@ async def create(
414417
extra_query: Query | None = None,
415418
extra_body: Body | None = None,
416419
timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
417-
) -> DataSource:
420+
) -> DatasourceCreateResponse:
418421
"""
419422
创建一个新的数据源
420423
@@ -452,7 +455,7 @@ async def create(
452455
{"async_process_meta": async_process_meta}, datasource_create_params.DatasourceCreateParams
453456
),
454457
),
455-
cast_to=DataSource,
458+
cast_to=DatasourceCreateResponse,
456459
)
457460

458461
async def retrieve(
@@ -465,7 +468,7 @@ async def retrieve(
465468
extra_query: Query | None = None,
466469
extra_body: Body | None = None,
467470
timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
468-
) -> DataSource:
471+
) -> DatasourceRetrieveResponse:
469472
"""
470473
根据 id 获取指定数据源
471474
@@ -485,7 +488,7 @@ async def retrieve(
485488
options=make_request_options(
486489
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
487490
),
488-
cast_to=DataSource,
491+
cast_to=DatasourceRetrieveResponse,
489492
)
490493

491494
async def update(
@@ -507,7 +510,7 @@ async def update(
507510
extra_query: Query | None = None,
508511
extra_body: Body | None = None,
509512
timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
510-
) -> DataSource:
513+
) -> DatasourceUpdateResponse:
511514
"""
512515
更新指定数据源信息
513516
@@ -559,7 +562,7 @@ async def update(
559562
options=make_request_options(
560563
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
561564
),
562-
cast_to=DataSource,
565+
cast_to=DatasourceUpdateResponse,
563566
)
564567

565568
async def list(
@@ -655,7 +658,7 @@ async def create_from_file(
655658
extra_query: Query | None = None,
656659
extra_body: Body | None = None,
657660
timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
658-
) -> DataSource:
661+
) -> DatasourceCreateFromFileResponse:
659662
"""
660663
上传文件并创建数据源
661664
@@ -691,7 +694,7 @@ async def create_from_file(
691694
datasource_create_from_file_params.DatasourceCreateFromFileParams,
692695
),
693696
),
694-
cast_to=DataSource,
697+
cast_to=DatasourceCreateFromFileResponse,
695698
)
696699

697700

0 commit comments

Comments
 (0)