Skip to content

Commit 22f59a3

Browse files
qianl15qianl15
committed
Support model retry
1 parent 6a3ef13 commit 22f59a3

File tree

2 files changed

+454
-1
lines changed

2 files changed

+454
-1
lines changed

‎tests/cassettes/test_dbos/test_dbos_agent_with_model_retry.yaml‎

Lines changed: 335 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,335 @@
1+
interactions:
2+
- request:
3+
headers:
4+
accept:
5+
- application/json
6+
accept-encoding:
7+
- gzip, deflate
8+
connection:
9+
- keep-alive
10+
content-length:
11+
- '347'
12+
content-type:
13+
- application/json
14+
host:
15+
- api.openai.com
16+
method: POST
17+
parsed_body:
18+
messages:
19+
- content: What is the weather in CDMX?
20+
role: user
21+
model: gpt-4o
22+
stream: false
23+
tool_choice: auto
24+
tools:
25+
- function:
26+
description: ''
27+
name: get_weather_in_city
28+
parameters:
29+
additionalProperties: false
30+
properties:
31+
city:
32+
type: string
33+
required:
34+
- city
35+
type: object
36+
strict: true
37+
type: function
38+
uri: https://api.openai.com/v1/chat/completions
39+
response:
40+
headers:
41+
access-control-expose-headers:
42+
- X-Request-ID
43+
alt-svc:
44+
- h3=":443"; ma=86400
45+
connection:
46+
- keep-alive
47+
content-length:
48+
- '1086'
49+
content-type:
50+
- application/json
51+
openai-organization:
52+
- pydantic-28gund
53+
openai-processing-ms:
54+
- '327'
55+
openai-project:
56+
- proj_dKobscVY9YJxeEaDJen54e3d
57+
openai-version:
58+
- '2020-10-01'
59+
strict-transport-security:
60+
- max-age=31536000; includeSubDomains; preload
61+
transfer-encoding:
62+
- chunked
63+
parsed_body:
64+
choices:
65+
- finish_reason: tool_calls
66+
index: 0
67+
logprobs: null
68+
message:
69+
annotations: []
70+
content: null
71+
refusal: null
72+
role: assistant
73+
tool_calls:
74+
- function:
75+
arguments: '{"city":"CDMX"}'
76+
name: get_weather_in_city
77+
id: call_fFAB8MNL3tUdfNIIdsIJTo0H
78+
type: function
79+
created: 1756423190
80+
id: chatcmpl-C9gCExiXILzHBQ4ZuERdiURkHUZZM
81+
model: gpt-4o-2024-08-06
82+
object: chat.completion
83+
service_tier: default
84+
system_fingerprint: fp_ea40d5097a
85+
usage:
86+
completion_tokens: 17
87+
completion_tokens_details:
88+
accepted_prediction_tokens: 0
89+
audio_tokens: 0
90+
reasoning_tokens: 0
91+
rejected_prediction_tokens: 0
92+
prompt_tokens: 47
93+
prompt_tokens_details:
94+
audio_tokens: 0
95+
cached_tokens: 0
96+
total_tokens: 64
97+
status:
98+
code: 200
99+
message: OK
100+
- request:
101+
headers:
102+
accept:
103+
- application/json
104+
accept-encoding:
105+
- gzip, deflate
106+
connection:
107+
- keep-alive
108+
content-length:
109+
- '665'
110+
content-type:
111+
- application/json
112+
cookie:
113+
- __cf_bm=yUpeQm6kQWb.kRgbX0F391IMtd8nqGUrSigYrEakdXQ-1756423190-1.0.1.1-FPxZURmrG14EfyWc8ZgNIHTHoWxQ3IPV9X4uCZ2O_z1KiiSc66VTYHq86EbI5aUnd26QXGFdwTsriLpMHbGRxfqlOj.DSK3daVg9q_XBQDE;
114+
_cfuvid=t8Kvct8UEUS4P6_7AIQ7_ayX0J6QaCWrtHqGpTB5uMo-1756423190751-0.0.1.1-604800000
115+
host:
116+
- api.openai.com
117+
method: POST
118+
parsed_body:
119+
messages:
120+
- content: What is the weather in CDMX?
121+
role: user
122+
- content: null
123+
role: assistant
124+
tool_calls:
125+
- function:
126+
arguments: '{"city":"CDMX"}'
127+
name: get_weather_in_city
128+
id: call_fFAB8MNL3tUdfNIIdsIJTo0H
129+
type: function
130+
- content: |-
131+
Did you mean Mexico City?
132+
133+
Fix the errors and try again.
134+
role: tool
135+
tool_call_id: call_fFAB8MNL3tUdfNIIdsIJTo0H
136+
model: gpt-4o
137+
stream: false
138+
tool_choice: auto
139+
tools:
140+
- function:
141+
description: ''
142+
name: get_weather_in_city
143+
parameters:
144+
additionalProperties: false
145+
properties:
146+
city:
147+
type: string
148+
required:
149+
- city
150+
type: object
151+
strict: true
152+
type: function
153+
uri: https://api.openai.com/v1/chat/completions
154+
response:
155+
headers:
156+
access-control-expose-headers:
157+
- X-Request-ID
158+
alt-svc:
159+
- h3=":443"; ma=86400
160+
connection:
161+
- keep-alive
162+
content-length:
163+
- '1094'
164+
content-type:
165+
- application/json
166+
openai-organization:
167+
- pydantic-28gund
168+
openai-processing-ms:
169+
- '352'
170+
openai-project:
171+
- proj_dKobscVY9YJxeEaDJen54e3d
172+
openai-version:
173+
- '2020-10-01'
174+
strict-transport-security:
175+
- max-age=31536000; includeSubDomains; preload
176+
transfer-encoding:
177+
- chunked
178+
parsed_body:
179+
choices:
180+
- finish_reason: tool_calls
181+
index: 0
182+
logprobs: null
183+
message:
184+
annotations: []
185+
content: null
186+
refusal: null
187+
role: assistant
188+
tool_calls:
189+
- function:
190+
arguments: '{"city":"Mexico City"}'
191+
name: get_weather_in_city
192+
id: call_hLYHO5lK5lmiukTZv6VQzz3x
193+
type: function
194+
created: 1756423191
195+
id: chatcmpl-C9gCF2OpzQojDQTsp31IsAagNqEC6
196+
model: gpt-4o-2024-08-06
197+
object: chat.completion
198+
service_tier: default
199+
system_fingerprint: fp_ea40d5097a
200+
usage:
201+
completion_tokens: 17
202+
completion_tokens_details:
203+
accepted_prediction_tokens: 0
204+
audio_tokens: 0
205+
reasoning_tokens: 0
206+
rejected_prediction_tokens: 0
207+
prompt_tokens: 87
208+
prompt_tokens_details:
209+
audio_tokens: 0
210+
cached_tokens: 0
211+
total_tokens: 104
212+
status:
213+
code: 200
214+
message: OK
215+
- request:
216+
headers:
217+
accept:
218+
- application/json
219+
accept-encoding:
220+
- gzip, deflate
221+
connection:
222+
- keep-alive
223+
content-length:
224+
- '937'
225+
content-type:
226+
- application/json
227+
cookie:
228+
- __cf_bm=yUpeQm6kQWb.kRgbX0F391IMtd8nqGUrSigYrEakdXQ-1756423190-1.0.1.1-FPxZURmrG14EfyWc8ZgNIHTHoWxQ3IPV9X4uCZ2O_z1KiiSc66VTYHq86EbI5aUnd26QXGFdwTsriLpMHbGRxfqlOj.DSK3daVg9q_XBQDE;
229+
_cfuvid=t8Kvct8UEUS4P6_7AIQ7_ayX0J6QaCWrtHqGpTB5uMo-1756423190751-0.0.1.1-604800000
230+
host:
231+
- api.openai.com
232+
method: POST
233+
parsed_body:
234+
messages:
235+
- content: What is the weather in CDMX?
236+
role: user
237+
- content: null
238+
role: assistant
239+
tool_calls:
240+
- function:
241+
arguments: '{"city":"CDMX"}'
242+
name: get_weather_in_city
243+
id: call_fFAB8MNL3tUdfNIIdsIJTo0H
244+
type: function
245+
- content: |-
246+
Did you mean Mexico City?
247+
248+
Fix the errors and try again.
249+
role: tool
250+
tool_call_id: call_fFAB8MNL3tUdfNIIdsIJTo0H
251+
- content: null
252+
role: assistant
253+
tool_calls:
254+
- function:
255+
arguments: '{"city":"Mexico City"}'
256+
name: get_weather_in_city
257+
id: call_hLYHO5lK5lmiukTZv6VQzz3x
258+
type: function
259+
- content: sunny
260+
role: tool
261+
tool_call_id: call_hLYHO5lK5lmiukTZv6VQzz3x
262+
model: gpt-4o
263+
stream: false
264+
tool_choice: auto
265+
tools:
266+
- function:
267+
description: ''
268+
name: get_weather_in_city
269+
parameters:
270+
additionalProperties: false
271+
properties:
272+
city:
273+
type: string
274+
required:
275+
- city
276+
type: object
277+
strict: true
278+
type: function
279+
uri: https://api.openai.com/v1/chat/completions
280+
response:
281+
headers:
282+
access-control-expose-headers:
283+
- X-Request-ID
284+
alt-svc:
285+
- h3=":443"; ma=86400
286+
connection:
287+
- keep-alive
288+
content-length:
289+
- '850'
290+
content-type:
291+
- application/json
292+
openai-organization:
293+
- pydantic-28gund
294+
openai-processing-ms:
295+
- '312'
296+
openai-project:
297+
- proj_dKobscVY9YJxeEaDJen54e3d
298+
openai-version:
299+
- '2020-10-01'
300+
strict-transport-security:
301+
- max-age=31536000; includeSubDomains; preload
302+
transfer-encoding:
303+
- chunked
304+
parsed_body:
305+
choices:
306+
- finish_reason: stop
307+
index: 0
308+
logprobs: null
309+
message:
310+
annotations: []
311+
content: The weather in Mexico City is currently sunny.
312+
refusal: null
313+
role: assistant
314+
created: 1756423192
315+
id: chatcmpl-C9gCGg6DDdUlo7CuS04nK9k6dnkZG
316+
model: gpt-4o-2024-08-06
317+
object: chat.completion
318+
service_tier: default
319+
system_fingerprint: fp_ea40d5097a
320+
usage:
321+
completion_tokens: 10
322+
completion_tokens_details:
323+
accepted_prediction_tokens: 0
324+
audio_tokens: 0
325+
reasoning_tokens: 0
326+
rejected_prediction_tokens: 0
327+
prompt_tokens: 116
328+
prompt_tokens_details:
329+
audio_tokens: 0
330+
cached_tokens: 0
331+
total_tokens: 126
332+
status:
333+
code: 200
334+
message: OK
335+
version: 1

0 commit comments

Comments
 (0)