|
22 | 22 | QuestionResourceWithStreamingResponse, |
23 | 23 | AsyncQuestionResourceWithStreamingResponse, |
24 | 24 | ) |
25 | | -from ...._types import Body, Query, Headers, NotGiven, not_given |
| 25 | +from ...._types import Body, Omit, Query, Headers, NotGiven, omit, not_given |
26 | 26 | from ...._utils import maybe_transform, async_maybe_transform |
27 | 27 | from ...._compat import cached_property |
28 | 28 | from ...._resource import SyncAPIResource, AsyncAPIResource |
|
33 | 33 | async_to_streamed_response_wrapper, |
34 | 34 | ) |
35 | 35 | from ...._base_client import make_request_options |
36 | | -from ....types.agents import chat_deliver_permission_params |
| 36 | +from ....types.agents import chat_rehydrate_chat_params, chat_deliver_permission_params |
37 | 37 | from ....types.agents.chat_get_chat_state_response import ChatGetChatStateResponse |
38 | 38 | from ....types.agents.chat_rehydrate_chat_response import ChatRehydrateChatResponse |
39 | 39 | from ....types.agents.chat_deliver_permission_response import ChatDeliverPermissionResponse |
@@ -155,18 +155,35 @@ def list_slash_commands( |
155 | 155 | def rehydrate_chat( |
156 | 156 | self, |
157 | 157 | *, |
| 158 | + session_id: str | Omit = omit, |
158 | 159 | # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs. |
159 | 160 | # The extra values given here take precedence over values defined on the client or passed to this method. |
160 | 161 | extra_headers: Headers | None = None, |
161 | 162 | extra_query: Query | None = None, |
162 | 163 | extra_body: Body | None = None, |
163 | 164 | timeout: float | httpx.Timeout | None | NotGiven = not_given, |
164 | 165 | ) -> ChatRehydrateChatResponse: |
165 | | - """Rehydrate the user's chat history. Does not wake a hibernated machine.""" |
| 166 | + """Rehydrate the user's chat history. |
| 167 | +
|
| 168 | + Does not wake a hibernated machine. |
| 169 | +
|
| 170 | + Args: |
| 171 | + extra_headers: Send extra headers |
| 172 | +
|
| 173 | + extra_query: Add additional query parameters to the request |
| 174 | +
|
| 175 | + extra_body: Add additional JSON properties to the request |
| 176 | +
|
| 177 | + timeout: Override the client-level default timeout for this request, in seconds |
| 178 | + """ |
166 | 179 | return self._get( |
167 | 180 | "/agents/chat/messages", |
168 | 181 | options=make_request_options( |
169 | | - extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout |
| 182 | + extra_headers=extra_headers, |
| 183 | + extra_query=extra_query, |
| 184 | + extra_body=extra_body, |
| 185 | + timeout=timeout, |
| 186 | + query=maybe_transform({"session_id": session_id}, chat_rehydrate_chat_params.ChatRehydrateChatParams), |
170 | 187 | ), |
171 | 188 | cast_to=ChatRehydrateChatResponse, |
172 | 189 | ) |
@@ -285,18 +302,37 @@ async def list_slash_commands( |
285 | 302 | async def rehydrate_chat( |
286 | 303 | self, |
287 | 304 | *, |
| 305 | + session_id: str | Omit = omit, |
288 | 306 | # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs. |
289 | 307 | # The extra values given here take precedence over values defined on the client or passed to this method. |
290 | 308 | extra_headers: Headers | None = None, |
291 | 309 | extra_query: Query | None = None, |
292 | 310 | extra_body: Body | None = None, |
293 | 311 | timeout: float | httpx.Timeout | None | NotGiven = not_given, |
294 | 312 | ) -> ChatRehydrateChatResponse: |
295 | | - """Rehydrate the user's chat history. Does not wake a hibernated machine.""" |
| 313 | + """Rehydrate the user's chat history. |
| 314 | +
|
| 315 | + Does not wake a hibernated machine. |
| 316 | +
|
| 317 | + Args: |
| 318 | + extra_headers: Send extra headers |
| 319 | +
|
| 320 | + extra_query: Add additional query parameters to the request |
| 321 | +
|
| 322 | + extra_body: Add additional JSON properties to the request |
| 323 | +
|
| 324 | + timeout: Override the client-level default timeout for this request, in seconds |
| 325 | + """ |
296 | 326 | return await self._get( |
297 | 327 | "/agents/chat/messages", |
298 | 328 | options=make_request_options( |
299 | | - extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout |
| 329 | + extra_headers=extra_headers, |
| 330 | + extra_query=extra_query, |
| 331 | + extra_body=extra_body, |
| 332 | + timeout=timeout, |
| 333 | + query=await async_maybe_transform( |
| 334 | + {"session_id": session_id}, chat_rehydrate_chat_params.ChatRehydrateChatParams |
| 335 | + ), |
300 | 336 | ), |
301 | 337 | cast_to=ChatRehydrateChatResponse, |
302 | 338 | ) |
|
0 commit comments