Skip to content

Commit cef8afa

Browse files
Wang Zesongdayflyshao
authored andcommitted
fix: create with reasoning
1 parent 6597ad2 commit cef8afa

File tree

1 file changed

+40
-26
lines changed

1 file changed

+40
-26
lines changed

volcenginesdkarkruntime/resources/responses/responses.py

Lines changed: 40 additions & 26 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,3 @@
1-
21
# Copyright (c) [2025] [OpenAI]
32
# Copyright (c) [2025] [ByteDance Ltd. and/or its affiliates.]
43
# SPDX-License-Identifier: Apache-2.0
@@ -41,14 +40,15 @@
4140
from ...types.responses.response_stream_event import ResponseStreamEvent
4241
from ...types.responses.response_text_config_param import ResponseTextConfigParam
4342
from ...types.responses.response_caching_param import ResponseCaching
44-
from ...types.shared import Reasoning,ReasoningEffort
45-
from ...types.shared_params import Reasoning,ReasoningEffort
43+
from ...types.shared_params import Reasoning
4644
from volcenginesdkarkruntime.types.shared_params.thinking import Thinking
4745

4846
__all__ = ["Responses", "AsyncResponses"]
4947

5048

51-
def _add_beta_headers(extra_headers: Headers | None = None, tools: Iterable[ToolParam] | None = ()) -> Headers:
49+
def _add_beta_headers(
50+
extra_headers: Headers | None = None, tools: Iterable[ToolParam] | None = ()
51+
) -> Headers:
5252
if tools is None:
5353
return extra_headers
5454
for tool_param in tools:
@@ -138,14 +138,15 @@ def retrieve(
138138
self,
139139
response_id: str,
140140
*,
141-
142141
extra_headers: Headers | None = None,
143142
extra_query: Query | None = None,
144143
extra_body: Body | None = None,
145144
timeout: float | httpx.Timeout | None = None,
146145
) -> Response:
147146
if not response_id:
148-
raise ValueError(f"Expected a non-empty value for `response_id` but received {response_id!r}")
147+
raise ValueError(
148+
f"Expected a non-empty value for `response_id` but received {response_id!r}"
149+
)
149150
return self._get(
150151
f"/responses/{response_id}",
151152
options=make_request_options(
@@ -169,12 +170,17 @@ def delete(
169170
timeout: float | httpx.Timeout | None = None,
170171
) -> None:
171172
if not response_id:
172-
raise ValueError(f"Expected a non-empty value for `response_id` but received {response_id!r}")
173+
raise ValueError(
174+
f"Expected a non-empty value for `response_id` but received {response_id!r}"
175+
)
173176
extra_headers = {"Accept": "*/*", **(extra_headers or {})}
174177
return self._delete(
175178
f"/responses/{response_id}",
176179
options=make_request_options(
177-
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
180+
extra_headers=extra_headers,
181+
extra_query=extra_query,
182+
extra_body=extra_body,
183+
timeout=timeout,
178184
),
179185
cast_to=type(None),
180186
)
@@ -214,6 +220,7 @@ async def create(
214220
extra_query: Query | None = None,
215221
extra_body: Body | None = None,
216222
timeout: float | httpx.Timeout | None | None = None,
223+
reasoning: Optional[Reasoning] | None = None,
217224
) -> Response | AsyncStream[ResponseStreamEvent]:
218225
extra_headers = _add_beta_headers(extra_headers, tools)
219226
resp = await self._post(
@@ -236,6 +243,7 @@ async def create(
236243
"top_p": top_p,
237244
"max_tool_calls": max_tool_calls,
238245
"expire_at": expire_at,
246+
"reasoning": reasoning,
239247
},
240248
options=make_request_options(
241249
extra_headers=extra_headers,
@@ -251,17 +259,18 @@ async def create(
251259
return resp
252260

253261
async def retrieve(
254-
self,
255-
response_id: str,
256-
*,
257-
258-
extra_headers: Headers | None = None,
259-
extra_query: Query | None = None,
260-
extra_body: Body | None = None,
261-
timeout: float | httpx.Timeout | None = None,
262+
self,
263+
response_id: str,
264+
*,
265+
extra_headers: Headers | None = None,
266+
extra_query: Query | None = None,
267+
extra_body: Body | None = None,
268+
timeout: float | httpx.Timeout | None = None,
262269
) -> Response:
263270
if not response_id:
264-
raise ValueError(f"Expected a non-empty value for `response_id` but received {response_id!r}")
271+
raise ValueError(
272+
f"Expected a non-empty value for `response_id` but received {response_id!r}"
273+
)
265274
return await self._get(
266275
f"/responses/{response_id}",
267276
options=make_request_options(
@@ -276,21 +285,26 @@ async def retrieve(
276285
)
277286

278287
async def delete(
279-
self,
280-
response_id: str,
281-
*,
282-
extra_headers: Headers | None = None,
283-
extra_query: Query | None = None,
284-
extra_body: Body | None = None,
285-
timeout: float | httpx.Timeout | None = None,
288+
self,
289+
response_id: str,
290+
*,
291+
extra_headers: Headers | None = None,
292+
extra_query: Query | None = None,
293+
extra_body: Body | None = None,
294+
timeout: float | httpx.Timeout | None = None,
286295
) -> None:
287296
if not response_id:
288-
raise ValueError(f"Expected a non-empty value for `response_id` but received {response_id!r}")
297+
raise ValueError(
298+
f"Expected a non-empty value for `response_id` but received {response_id!r}"
299+
)
289300
extra_headers = {"Accept": "*/*", **(extra_headers or {})}
290301
return await self._delete(
291302
f"/responses/{response_id}",
292303
options=make_request_options(
293-
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
304+
extra_headers=extra_headers,
305+
extra_query=extra_query,
306+
extra_body=extra_body,
307+
timeout=timeout,
294308
),
295309
cast_to=type(None),
296310
)

0 commit comments

Comments
 (0)