Skip to content

Commit b236a47

Browse files
adtyavrdhnadtyavrdhn
committed
Fixing doc tests
1 parent a7dd57a commit b236a47

File tree

15 files changed

+105
-44
lines changed

15 files changed

+105
-44
lines changed

‎docs/agents.md‎

Lines changed: 12 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -320,7 +320,8 @@ async def main():
320320
content='What is the capital of France?',
321321
timestamp=datetime.datetime(...),
322322
)
323-
]
323+
],
324+
run_id='...',
324325
)
325326
),
326327
CallToolsNode(
@@ -329,6 +330,7 @@ async def main():
329330
usage=RequestUsage(input_tokens=56, output_tokens=7),
330331
model_name='gpt-5',
331332
timestamp=datetime.datetime(...),
333+
run_id='...',
332334
)
333335
),
334336
End(data=FinalResult(output='The capital of France is Paris.')),
@@ -382,7 +384,8 @@ async def main():
382384
content='What is the capital of France?',
383385
timestamp=datetime.datetime(...),
384386
)
385-
]
387+
],
388+
run_id='...',
386389
)
387390
),
388391
CallToolsNode(
@@ -391,6 +394,7 @@ async def main():
391394
usage=RequestUsage(input_tokens=56, output_tokens=7),
392395
model_name='gpt-5',
393396
timestamp=datetime.datetime(...),
397+
run_id='...',
394398
)
395399
),
396400
End(data=FinalResult(output='The capital of France is Paris.')),
@@ -1044,7 +1048,8 @@ with capture_run_messages() as messages: # (2)!
10441048
content='Please get me the volume of a box with size 6.',
10451049
timestamp=datetime.datetime(...),
10461050
)
1047-
]
1051+
],
1052+
run_id='...',
10481053
),
10491054
ModelResponse(
10501055
parts=[
@@ -1057,6 +1062,7 @@ with capture_run_messages() as messages: # (2)!
10571062
usage=RequestUsage(input_tokens=62, output_tokens=4),
10581063
model_name='gpt-5',
10591064
timestamp=datetime.datetime(...),
1065+
run_id='...',
10601066
),
10611067
ModelRequest(
10621068
parts=[
@@ -1066,7 +1072,8 @@ with capture_run_messages() as messages: # (2)!
10661072
tool_call_id='pyd_ai_tool_call_id',
10671073
timestamp=datetime.datetime(...),
10681074
)
1069-
]
1075+
],
1076+
run_id='...',
10701077
),
10711078
ModelResponse(
10721079
parts=[
@@ -1079,6 +1086,7 @@ with capture_run_messages() as messages: # (2)!
10791086
usage=RequestUsage(input_tokens=72, output_tokens=8),
10801087
model_name='gpt-5',
10811088
timestamp=datetime.datetime(...),
1089+
run_id='...',
10821090
),
10831091
]
10841092
"""

‎docs/api/models/function.md‎

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,8 @@ async def model_function(
2929
content='Testing my agent...',
3030
timestamp=datetime.datetime(...),
3131
)
32-
]
32+
],
33+
run_id='...',
3334
)
3435
]
3536
"""

‎docs/deferred-tools.md‎

Lines changed: 14 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -106,7 +106,8 @@ print(result.all_messages())
106106
content='Delete `__init__.py`, write `Hello, world!` to `README.md`, and clear `.env`',
107107
timestamp=datetime.datetime(...),
108108
)
109-
]
109+
],
110+
run_id='...',
110111
),
111112
ModelResponse(
112113
parts=[
@@ -129,6 +130,7 @@ print(result.all_messages())
129130
usage=RequestUsage(input_tokens=63, output_tokens=21),
130131
model_name='gpt-5',
131132
timestamp=datetime.datetime(...),
133+
run_id='...',
132134
),
133135
ModelRequest(
134136
parts=[
@@ -138,7 +140,8 @@ print(result.all_messages())
138140
tool_call_id='update_file_readme',
139141
timestamp=datetime.datetime(...),
140142
)
141-
]
143+
],
144+
run_id='...',
142145
),
143146
ModelRequest(
144147
parts=[
@@ -154,7 +157,8 @@ print(result.all_messages())
154157
tool_call_id='delete_file',
155158
timestamp=datetime.datetime(...),
156159
),
157-
]
160+
],
161+
run_id='...',
158162
),
159163
ModelResponse(
160164
parts=[
@@ -165,6 +169,7 @@ print(result.all_messages())
165169
usage=RequestUsage(input_tokens=79, output_tokens=39),
166170
model_name='gpt-5',
167171
timestamp=datetime.datetime(...),
172+
run_id='...',
168173
),
169174
]
170175
"""
@@ -275,7 +280,8 @@ async def main():
275280
content='Calculate the answer to the ultimate question of life, the universe, and everything',
276281
timestamp=datetime.datetime(...),
277282
)
278-
]
283+
],
284+
run_id='...',
279285
),
280286
ModelResponse(
281287
parts=[
@@ -290,6 +296,7 @@ async def main():
290296
usage=RequestUsage(input_tokens=63, output_tokens=13),
291297
model_name='gpt-5',
292298
timestamp=datetime.datetime(...),
299+
run_id='...',
293300
),
294301
ModelRequest(
295302
parts=[
@@ -299,7 +306,8 @@ async def main():
299306
tool_call_id='pyd_ai_tool_call_id',
300307
timestamp=datetime.datetime(...),
301308
)
302-
]
309+
],
310+
run_id='...',
303311
),
304312
ModelResponse(
305313
parts=[
@@ -310,6 +318,7 @@ async def main():
310318
usage=RequestUsage(input_tokens=64, output_tokens=28),
311319
model_name='gpt-5',
312320
timestamp=datetime.datetime(...),
321+
run_id='...',
313322
),
314323
]
315324
"""

‎docs/message-history.md‎

Lines changed: 18 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,8 @@ print(result.all_messages())
5050
content='Tell me a joke.',
5151
timestamp=datetime.datetime(...),
5252
),
53-
]
53+
],
54+
run_id='...',
5455
),
5556
ModelResponse(
5657
parts=[
@@ -61,6 +62,7 @@ print(result.all_messages())
6162
usage=RequestUsage(input_tokens=60, output_tokens=12),
6263
model_name='gpt-5',
6364
timestamp=datetime.datetime(...),
65+
run_id='...',
6466
),
6567
]
6668
"""
@@ -118,7 +120,8 @@ async def main():
118120
content='Tell me a joke.',
119121
timestamp=datetime.datetime(...),
120122
),
121-
]
123+
],
124+
run_id='...',
122125
),
123126
ModelResponse(
124127
parts=[
@@ -129,6 +132,7 @@ async def main():
129132
usage=RequestUsage(input_tokens=50, output_tokens=12),
130133
model_name='gpt-5',
131134
timestamp=datetime.datetime(...),
135+
run_id='...',
132136
),
133137
]
134138
"""
@@ -172,7 +176,8 @@ print(result2.all_messages())
172176
content='Tell me a joke.',
173177
timestamp=datetime.datetime(...),
174178
),
175-
]
179+
],
180+
run_id='...',
176181
),
177182
ModelResponse(
178183
parts=[
@@ -183,14 +188,16 @@ print(result2.all_messages())
183188
usage=RequestUsage(input_tokens=60, output_tokens=12),
184189
model_name='gpt-5',
185190
timestamp=datetime.datetime(...),
191+
run_id='...',
186192
),
187193
ModelRequest(
188194
parts=[
189195
UserPromptPart(
190196
content='Explain?',
191197
timestamp=datetime.datetime(...),
192198
)
193-
]
199+
],
200+
run_id='...',
194201
),
195202
ModelResponse(
196203
parts=[
@@ -201,6 +208,7 @@ print(result2.all_messages())
201208
usage=RequestUsage(input_tokens=61, output_tokens=26),
202209
model_name='gpt-5',
203210
timestamp=datetime.datetime(...),
211+
run_id='...',
204212
),
205213
]
206214
"""
@@ -293,7 +301,8 @@ print(result2.all_messages())
293301
content='Tell me a joke.',
294302
timestamp=datetime.datetime(...),
295303
),
296-
]
304+
],
305+
run_id='...',
297306
),
298307
ModelResponse(
299308
parts=[
@@ -304,14 +313,16 @@ print(result2.all_messages())
304313
usage=RequestUsage(input_tokens=60, output_tokens=12),
305314
model_name='gpt-5',
306315
timestamp=datetime.datetime(...),
316+
run_id='...',
307317
),
308318
ModelRequest(
309319
parts=[
310320
UserPromptPart(
311321
content='Explain?',
312322
timestamp=datetime.datetime(...),
313323
)
314-
]
324+
],
325+
run_id='...',
315326
),
316327
ModelResponse(
317328
parts=[
@@ -322,6 +333,7 @@ print(result2.all_messages())
322333
usage=RequestUsage(input_tokens=61, output_tokens=26),
323334
model_name='gemini-2.5-pro',
324335
timestamp=datetime.datetime(...),
336+
run_id='...',
325337
),
326338
]
327339
"""

‎docs/tools.md‎

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -87,7 +87,8 @@ print(dice_result.all_messages())
8787
content='My guess is 4',
8888
timestamp=datetime.datetime(...),
8989
),
90-
]
90+
],
91+
run_id='...',
9192
),
9293
ModelResponse(
9394
parts=[
@@ -98,6 +99,7 @@ print(dice_result.all_messages())
9899
usage=RequestUsage(input_tokens=90, output_tokens=2),
99100
model_name='gemini-2.5-flash',
100101
timestamp=datetime.datetime(...),
102+
run_id='...',
101103
),
102104
ModelRequest(
103105
parts=[
@@ -107,7 +109,8 @@ print(dice_result.all_messages())
107109
tool_call_id='pyd_ai_tool_call_id',
108110
timestamp=datetime.datetime(...),
109111
)
110-
]
112+
],
113+
run_id='...',
111114
),
112115
ModelResponse(
113116
parts=[
@@ -118,6 +121,7 @@ print(dice_result.all_messages())
118121
usage=RequestUsage(input_tokens=91, output_tokens=4),
119122
model_name='gemini-2.5-flash',
120123
timestamp=datetime.datetime(...),
124+
run_id='...',
121125
),
122126
ModelRequest(
123127
parts=[
@@ -127,7 +131,8 @@ print(dice_result.all_messages())
127131
tool_call_id='pyd_ai_tool_call_id',
128132
timestamp=datetime.datetime(...),
129133
)
130-
]
134+
],
135+
run_id='...',
131136
),
132137
ModelResponse(
133138
parts=[
@@ -138,6 +143,7 @@ print(dice_result.all_messages())
138143
usage=RequestUsage(input_tokens=92, output_tokens=12),
139144
model_name='gemini-2.5-flash',
140145
timestamp=datetime.datetime(...),
146+
run_id='...',
141147
),
142148
]
143149
"""

‎pydantic_ai_slim/pydantic_ai/agent/__init__.py‎

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -501,7 +501,8 @@ async def main():
501501
content='What is the capital of France?',
502502
timestamp=datetime.datetime(...),
503503
)
504-
]
504+
],
505+
run_id='...',
505506
)
506507
),
507508
CallToolsNode(
@@ -510,6 +511,7 @@ async def main():
510511
usage=RequestUsage(input_tokens=56, output_tokens=7),
511512
model_name='gpt-4o',
512513
timestamp=datetime.datetime(...),
514+
run_id='...',
513515
)
514516
),
515517
End(data=FinalResult(output='The capital of France is Paris.')),

‎pydantic_ai_slim/pydantic_ai/agent/abstract.py‎

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1003,7 +1003,8 @@ async def main():
10031003
content='What is the capital of France?',
10041004
timestamp=datetime.datetime(...),
10051005
)
1006-
]
1006+
],
1007+
run_id='...',
10071008
)
10081009
),
10091010
CallToolsNode(
@@ -1012,6 +1013,7 @@ async def main():
10121013
usage=RequestUsage(input_tokens=56, output_tokens=7),
10131014
model_name='gpt-4o',
10141015
timestamp=datetime.datetime(...),
1016+
run_id='...',
10151017
)
10161018
),
10171019
End(data=FinalResult(output='The capital of France is Paris.')),

‎pydantic_ai_slim/pydantic_ai/agent/wrapper.py‎

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -163,7 +163,8 @@ async def main():
163163
content='What is the capital of France?',
164164
timestamp=datetime.datetime(...),
165165
)
166-
]
166+
],
167+
run_id='...',
167168
)
168169
),
169170
CallToolsNode(
@@ -172,6 +173,7 @@ async def main():
172173
usage=RequestUsage(input_tokens=56, output_tokens=7),
173174
model_name='gpt-4o',
174175
timestamp=datetime.datetime(...),
176+
run_id='...',
175177
)
176178
),
177179
End(data=FinalResult(output='The capital of France is Paris.')),

‎pydantic_ai_slim/pydantic_ai/durable_exec/dbos/_agent.py‎

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -799,7 +799,8 @@ async def main():
799799
content='What is the capital of France?',
800800
timestamp=datetime.datetime(...),
801801
)
802-
]
802+
],
803+
run_id='...',
803804
)
804805
),
805806
CallToolsNode(
@@ -808,6 +809,7 @@ async def main():
808809
usage=RequestUsage(input_tokens=56, output_tokens=7),
809810
model_name='gpt-4o',
810811
timestamp=datetime.datetime(...),
812+
run_id='...',
811813
)
812814
),
813815
End(data=FinalResult(output='The capital of France is Paris.')),

0 commit comments

Comments
 (0)