pydantic-ai-slim 1.7.0__py3-none-any.whl → 1.11.0__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.

This version of pydantic-ai-slim might be problematic. Click here for more details.

Files changed (47) hide show
  1. pydantic_ai/__init__.py +2 -0
  2. pydantic_ai/_agent_graph.py +3 -0
  3. pydantic_ai/_cli.py +2 -2
  4. pydantic_ai/_run_context.py +8 -2
  5. pydantic_ai/_tool_manager.py +1 -0
  6. pydantic_ai/_utils.py +18 -0
  7. pydantic_ai/ag_ui.py +50 -696
  8. pydantic_ai/agent/__init__.py +13 -3
  9. pydantic_ai/agent/abstract.py +172 -9
  10. pydantic_ai/agent/wrapper.py +5 -0
  11. pydantic_ai/direct.py +16 -4
  12. pydantic_ai/durable_exec/dbos/_agent.py +31 -0
  13. pydantic_ai/durable_exec/prefect/_agent.py +28 -0
  14. pydantic_ai/durable_exec/temporal/_agent.py +28 -0
  15. pydantic_ai/durable_exec/temporal/_function_toolset.py +23 -73
  16. pydantic_ai/durable_exec/temporal/_mcp_server.py +30 -30
  17. pydantic_ai/durable_exec/temporal/_run_context.py +9 -3
  18. pydantic_ai/durable_exec/temporal/_toolset.py +67 -3
  19. pydantic_ai/messages.py +49 -8
  20. pydantic_ai/models/__init__.py +42 -1
  21. pydantic_ai/models/google.py +5 -12
  22. pydantic_ai/models/groq.py +9 -1
  23. pydantic_ai/models/openai.py +6 -3
  24. pydantic_ai/profiles/openai.py +5 -2
  25. pydantic_ai/providers/anthropic.py +2 -2
  26. pydantic_ai/providers/openrouter.py +3 -0
  27. pydantic_ai/result.py +178 -11
  28. pydantic_ai/tools.py +10 -6
  29. pydantic_ai/ui/__init__.py +16 -0
  30. pydantic_ai/ui/_adapter.py +386 -0
  31. pydantic_ai/ui/_event_stream.py +591 -0
  32. pydantic_ai/ui/_messages_builder.py +28 -0
  33. pydantic_ai/ui/ag_ui/__init__.py +9 -0
  34. pydantic_ai/ui/ag_ui/_adapter.py +187 -0
  35. pydantic_ai/ui/ag_ui/_event_stream.py +236 -0
  36. pydantic_ai/ui/ag_ui/app.py +148 -0
  37. pydantic_ai/ui/vercel_ai/__init__.py +16 -0
  38. pydantic_ai/ui/vercel_ai/_adapter.py +199 -0
  39. pydantic_ai/ui/vercel_ai/_event_stream.py +187 -0
  40. pydantic_ai/ui/vercel_ai/_utils.py +16 -0
  41. pydantic_ai/ui/vercel_ai/request_types.py +275 -0
  42. pydantic_ai/ui/vercel_ai/response_types.py +230 -0
  43. {pydantic_ai_slim-1.7.0.dist-info → pydantic_ai_slim-1.11.0.dist-info}/METADATA +10 -6
  44. {pydantic_ai_slim-1.7.0.dist-info → pydantic_ai_slim-1.11.0.dist-info}/RECORD +47 -33
  45. {pydantic_ai_slim-1.7.0.dist-info → pydantic_ai_slim-1.11.0.dist-info}/WHEEL +0 -0
  46. {pydantic_ai_slim-1.7.0.dist-info → pydantic_ai_slim-1.11.0.dist-info}/entry_points.txt +0 -0
  47. {pydantic_ai_slim-1.7.0.dist-info → pydantic_ai_slim-1.11.0.dist-info}/licenses/LICENSE +0 -0
pydantic_ai/durable_exec/dbos/_agent.py CHANGED
@@ -116,6 +116,7 @@ class DBOSAgent(WrapperAgent[AgentDepsT, OutputDataT], DBOSConfiguredInstance):
116
116
  message_history: Sequence[_messages.ModelMessage] | None = None,
117
117
  deferred_tool_results: DeferredToolResults | None = None,
118
118
  model: models.Model | models.KnownModelName | str | None = None,
119
+ instructions: Instructions[AgentDepsT] = None,
119
120
  deps: AgentDepsT,
120
121
  model_settings: ModelSettings | None = None,
121
122
  usage_limits: _usage.UsageLimits | None = None,
@@ -133,6 +134,7 @@ class DBOSAgent(WrapperAgent[AgentDepsT, OutputDataT], DBOSConfiguredInstance):
133
134
  message_history=message_history,
134
135
  deferred_tool_results=deferred_tool_results,
135
136
  model=model,
137
+ instructions=instructions,
136
138
  deps=deps,
137
139
  model_settings=model_settings,
138
140
  usage_limits=usage_limits,
@@ -157,6 +159,7 @@ class DBOSAgent(WrapperAgent[AgentDepsT, OutputDataT], DBOSConfiguredInstance):
157
159
  model: models.Model | models.KnownModelName | str | None = None,
158
160
  deps: AgentDepsT,
159
161
  model_settings: ModelSettings | None = None,
162
+ instructions: Instructions[AgentDepsT] = None,
160
163
  usage_limits: _usage.UsageLimits | None = None,
161
164
  usage: _usage.RunUsage | None = None,
162
165
  infer_name: bool = True,
@@ -172,6 +175,7 @@ class DBOSAgent(WrapperAgent[AgentDepsT, OutputDataT], DBOSConfiguredInstance):
172
175
  message_history=message_history,
173
176
  deferred_tool_results=deferred_tool_results,
174
177
  model=model,
178
+ instructions=instructions,
175
179
  deps=deps,
176
180
  model_settings=model_settings,
177
181
  usage_limits=usage_limits,
@@ -245,6 +249,7 @@ class DBOSAgent(WrapperAgent[AgentDepsT, OutputDataT], DBOSConfiguredInstance):
245
249
  message_history: Sequence[_messages.ModelMessage] | None = None,
246
250
  deferred_tool_results: DeferredToolResults | None = None,
247
251
  model: models.Model | models.KnownModelName | str | None = None,
252
+ instructions: Instructions[AgentDepsT] = None,
248
253
  deps: AgentDepsT = None,
249
254
  model_settings: ModelSettings | None = None,
250
255
  usage_limits: _usage.UsageLimits | None = None,
@@ -264,6 +269,7 @@ class DBOSAgent(WrapperAgent[AgentDepsT, OutputDataT], DBOSConfiguredInstance):
264
269
  message_history: Sequence[_messages.ModelMessage] | None = None,
265
270
  deferred_tool_results: DeferredToolResults | None = None,
266
271
  model: models.Model | models.KnownModelName | str | None = None,
272
+ instructions: Instructions[AgentDepsT] = None,
267
273
  deps: AgentDepsT = None,
268
274
  model_settings: ModelSettings | None = None,
269
275
  usage_limits: _usage.UsageLimits | None = None,
@@ -282,6 +288,7 @@ class DBOSAgent(WrapperAgent[AgentDepsT, OutputDataT], DBOSConfiguredInstance):
282
288
  message_history: Sequence[_messages.ModelMessage] | None = None,
283
289
  deferred_tool_results: DeferredToolResults | None = None,
284
290
  model: models.Model | models.KnownModelName | str | None = None,
291
+ instructions: Instructions[AgentDepsT] = None,
285
292
  deps: AgentDepsT = None,
286
293
  model_settings: ModelSettings | None = None,
287
294
  usage_limits: _usage.UsageLimits | None = None,
@@ -316,6 +323,7 @@ class DBOSAgent(WrapperAgent[AgentDepsT, OutputDataT], DBOSConfiguredInstance):
316
323
  message_history: History of the conversation so far.
317
324
  deferred_tool_results: Optional results for deferred tool calls in the message history.
318
325
  model: Optional model to use for this run, required if `model` was not set when creating the agent.
326
+ instructions: Optional additional instructions to use for this run.
319
327
  deps: Optional dependencies to use for this run.
320
328
  model_settings: Optional settings to use for this model's request.
321
329
  usage_limits: Optional limits on model request count or token usage.
@@ -334,6 +342,7 @@ class DBOSAgent(WrapperAgent[AgentDepsT, OutputDataT], DBOSConfiguredInstance):
334
342
  message_history=message_history,
335
343
  deferred_tool_results=deferred_tool_results,
336
344
  model=model,
345
+ instructions=instructions,
337
346
  deps=deps,
338
347
  model_settings=model_settings,
339
348
  usage_limits=usage_limits,
@@ -354,6 +363,7 @@ class DBOSAgent(WrapperAgent[AgentDepsT, OutputDataT], DBOSConfiguredInstance):
354
363
  message_history: Sequence[_messages.ModelMessage] | None = None,
355
364
  deferred_tool_results: DeferredToolResults | None = None,
356
365
  model: models.Model | models.KnownModelName | str | None = None,
366
+ instructions: Instructions[AgentDepsT] = None,
357
367
  deps: AgentDepsT = None,
358
368
  model_settings: ModelSettings | None = None,
359
369
  usage_limits: _usage.UsageLimits | None = None,
@@ -373,6 +383,7 @@ class DBOSAgent(WrapperAgent[AgentDepsT, OutputDataT], DBOSConfiguredInstance):
373
383
  message_history: Sequence[_messages.ModelMessage] | None = None,
374
384
  deferred_tool_results: DeferredToolResults | None = None,
375
385
  model: models.Model | models.KnownModelName | str | None = None,
386
+ instructions: Instructions[AgentDepsT] = None,
376
387
  deps: AgentDepsT = None,
377
388
  model_settings: ModelSettings | None = None,
378
389
  usage_limits: _usage.UsageLimits | None = None,
@@ -391,6 +402,7 @@ class DBOSAgent(WrapperAgent[AgentDepsT, OutputDataT], DBOSConfiguredInstance):
391
402
  message_history: Sequence[_messages.ModelMessage] | None = None,
392
403
  deferred_tool_results: DeferredToolResults | None = None,
393
404
  model: models.Model | models.KnownModelName | str | None = None,
405
+ instructions: Instructions[AgentDepsT] = None,
394
406
  deps: AgentDepsT = None,
395
407
  model_settings: ModelSettings | None = None,
396
408
  usage_limits: _usage.UsageLimits | None = None,
@@ -424,6 +436,7 @@ class DBOSAgent(WrapperAgent[AgentDepsT, OutputDataT], DBOSConfiguredInstance):
424
436
  message_history: History of the conversation so far.
425
437
  deferred_tool_results: Optional results for deferred tool calls in the message history.
426
438
  model: Optional model to use for this run, required if `model` was not set when creating the agent.
439
+ instructions: Optional additional instructions to use for this run.
427
440
  deps: Optional dependencies to use for this run.
428
441
  model_settings: Optional settings to use for this model's request.
429
442
  usage_limits: Optional limits on model request count or token usage.
@@ -442,6 +455,7 @@ class DBOSAgent(WrapperAgent[AgentDepsT, OutputDataT], DBOSConfiguredInstance):
442
455
  message_history=message_history,
443
456
  deferred_tool_results=deferred_tool_results,
444
457
  model=model,
458
+ instructions=instructions,
445
459
  deps=deps,
446
460
  model_settings=model_settings,
447
461
  usage_limits=usage_limits,
@@ -462,6 +476,7 @@ class DBOSAgent(WrapperAgent[AgentDepsT, OutputDataT], DBOSConfiguredInstance):
462
476
  message_history: Sequence[_messages.ModelMessage] | None = None,
463
477
  deferred_tool_results: DeferredToolResults | None = None,
464
478
  model: models.Model | models.KnownModelName | str | None = None,
479
+ instructions: Instructions[AgentDepsT] = None,
465
480
  deps: AgentDepsT = None,
466
481
  model_settings: ModelSettings | None = None,
467
482
  usage_limits: _usage.UsageLimits | None = None,
@@ -482,6 +497,7 @@ class DBOSAgent(WrapperAgent[AgentDepsT, OutputDataT], DBOSConfiguredInstance):
482
497
  deferred_tool_results: DeferredToolResults | None = None,
483
498
  model: models.Model | models.KnownModelName | str | None = None,
484
499
  deps: AgentDepsT = None,
500
+ instructions: Instructions[AgentDepsT] = None,
485
501
  model_settings: ModelSettings | None = None,
486
502
  usage_limits: _usage.UsageLimits | None = None,
487
503
  usage: _usage.RunUsage | None = None,
@@ -500,6 +516,7 @@ class DBOSAgent(WrapperAgent[AgentDepsT, OutputDataT], DBOSConfiguredInstance):
500
516
  message_history: Sequence[_messages.ModelMessage] | None = None,
501
517
  deferred_tool_results: DeferredToolResults | None = None,
502
518
  model: models.Model | models.KnownModelName | str | None = None,
519
+ instructions: Instructions[AgentDepsT] = None,
503
520
  deps: AgentDepsT = None,
504
521
  model_settings: ModelSettings | None = None,
505
522
  usage_limits: _usage.UsageLimits | None = None,
@@ -531,6 +548,7 @@ class DBOSAgent(WrapperAgent[AgentDepsT, OutputDataT], DBOSConfiguredInstance):
531
548
  message_history: History of the conversation so far.
532
549
  deferred_tool_results: Optional results for deferred tool calls in the message history.
533
550
  model: Optional model to use for this run, required if `model` was not set when creating the agent.
551
+ instructions: Optional additional instructions to use for this run.
534
552
  deps: Optional dependencies to use for this run.
535
553
  model_settings: Optional settings to use for this model's request.
536
554
  usage_limits: Optional limits on model request count or token usage.
@@ -555,6 +573,7 @@ class DBOSAgent(WrapperAgent[AgentDepsT, OutputDataT], DBOSConfiguredInstance):
555
573
  message_history=message_history,
556
574
  deferred_tool_results=deferred_tool_results,
557
575
  model=model,
576
+ instructions=instructions,
558
577
  deps=deps,
559
578
  model_settings=model_settings,
560
579
  usage_limits=usage_limits,
@@ -576,6 +595,7 @@ class DBOSAgent(WrapperAgent[AgentDepsT, OutputDataT], DBOSConfiguredInstance):
576
595
  message_history: Sequence[_messages.ModelMessage] | None = None,
577
596
  deferred_tool_results: DeferredToolResults | None = None,
578
597
  model: models.Model | models.KnownModelName | str | None = None,
598
+ instructions: Instructions[AgentDepsT] = None,
579
599
  deps: AgentDepsT = None,
580
600
  model_settings: ModelSettings | None = None,
581
601
  usage_limits: _usage.UsageLimits | None = None,
@@ -594,6 +614,7 @@ class DBOSAgent(WrapperAgent[AgentDepsT, OutputDataT], DBOSConfiguredInstance):
594
614
  message_history: Sequence[_messages.ModelMessage] | None = None,
595
615
  deferred_tool_results: DeferredToolResults | None = None,
596
616
  model: models.Model | models.KnownModelName | str | None = None,
617
+ instructions: Instructions[AgentDepsT] = None,
597
618
  deps: AgentDepsT = None,
598
619
  model_settings: ModelSettings | None = None,
599
620
  usage_limits: _usage.UsageLimits | None = None,
@@ -611,6 +632,7 @@ class DBOSAgent(WrapperAgent[AgentDepsT, OutputDataT], DBOSConfiguredInstance):
611
632
  message_history: Sequence[_messages.ModelMessage] | None = None,
612
633
  deferred_tool_results: DeferredToolResults | None = None,
613
634
  model: models.Model | models.KnownModelName | str | None = None,
635
+ instructions: Instructions[AgentDepsT] = None,
614
636
  deps: AgentDepsT = None,
615
637
  model_settings: ModelSettings | None = None,
616
638
  usage_limits: _usage.UsageLimits | None = None,
@@ -640,6 +662,9 @@ class DBOSAgent(WrapperAgent[AgentDepsT, OutputDataT], DBOSConfiguredInstance):
640
662
  PartStartEvent(index=0, part=TextPart(content='The capital of ')),
641
663
  FinalResultEvent(tool_name=None, tool_call_id=None),
642
664
  PartDeltaEvent(index=0, delta=TextPartDelta(content_delta='France is Paris. ')),
665
+ PartEndEvent(
666
+ index=0, part=TextPart(content='The capital of France is Paris. ')
667
+ ),
643
668
  AgentRunResultEvent(
644
669
  result=AgentRunResult(output='The capital of France is Paris. ')
645
670
  ),
@@ -657,6 +682,7 @@ class DBOSAgent(WrapperAgent[AgentDepsT, OutputDataT], DBOSConfiguredInstance):
657
682
  message_history: History of the conversation so far.
658
683
  deferred_tool_results: Optional results for deferred tool calls in the message history.
659
684
  model: Optional model to use for this run, required if `model` was not set when creating the agent.
685
+ instructions: Optional additional instructions to use for this run.
660
686
  deps: Optional dependencies to use for this run.
661
687
  model_settings: Optional settings to use for this model's request.
662
688
  usage_limits: Optional limits on model request count or token usage.
@@ -683,6 +709,7 @@ class DBOSAgent(WrapperAgent[AgentDepsT, OutputDataT], DBOSConfiguredInstance):
683
709
  message_history: Sequence[_messages.ModelMessage] | None = None,
684
710
  deferred_tool_results: DeferredToolResults | None = None,
685
711
  model: models.Model | models.KnownModelName | str | None = None,
712
+ instructions: Instructions[AgentDepsT] = None,
686
713
  deps: AgentDepsT = None,
687
714
  model_settings: ModelSettings | None = None,
688
715
  usage_limits: _usage.UsageLimits | None = None,
@@ -702,6 +729,7 @@ class DBOSAgent(WrapperAgent[AgentDepsT, OutputDataT], DBOSConfiguredInstance):
702
729
  message_history: Sequence[_messages.ModelMessage] | None = None,
703
730
  deferred_tool_results: DeferredToolResults | None = None,
704
731
  model: models.Model | models.KnownModelName | str | None = None,
732
+ instructions: Instructions[AgentDepsT] = None,
705
733
  deps: AgentDepsT = None,
706
734
  model_settings: ModelSettings | None = None,
707
735
  usage_limits: _usage.UsageLimits | None = None,
@@ -721,6 +749,7 @@ class DBOSAgent(WrapperAgent[AgentDepsT, OutputDataT], DBOSConfiguredInstance):
721
749
  message_history: Sequence[_messages.ModelMessage] | None = None,
722
750
  deferred_tool_results: DeferredToolResults | None = None,
723
751
  model: models.Model | models.KnownModelName | str | None = None,
752
+ instructions: Instructions[AgentDepsT] = None,
724
753
  deps: AgentDepsT = None,
725
754
  model_settings: ModelSettings | None = None,
726
755
  usage_limits: _usage.UsageLimits | None = None,
@@ -795,6 +824,7 @@ class DBOSAgent(WrapperAgent[AgentDepsT, OutputDataT], DBOSConfiguredInstance):
795
824
  message_history: History of the conversation so far.
796
825
  deferred_tool_results: Optional results for deferred tool calls in the message history.
797
826
  model: Optional model to use for this run, required if `model` was not set when creating the agent.
827
+ instructions: Optional additional instructions to use for this run.
798
828
  deps: Optional dependencies to use for this run.
799
829
  model_settings: Optional settings to use for this model's request.
800
830
  usage_limits: Optional limits on model request count or token usage.
@@ -818,6 +848,7 @@ class DBOSAgent(WrapperAgent[AgentDepsT, OutputDataT], DBOSConfiguredInstance):
818
848
  message_history=message_history,
819
849
  deferred_tool_results=deferred_tool_results,
820
850
  model=model,
851
+ instructions=instructions,
821
852
  deps=deps,
822
853
  model_settings=model_settings,
823
854
  usage_limits=usage_limits,
pydantic_ai/durable_exec/prefect/_agent.py CHANGED
@@ -180,6 +180,7 @@ class PrefectAgent(WrapperAgent[AgentDepsT, OutputDataT]):
180
180
  message_history: Sequence[_messages.ModelMessage] | None = None,
181
181
  deferred_tool_results: DeferredToolResults | None = None,
182
182
  model: models.Model | models.KnownModelName | str | None = None,
183
+ instructions: Instructions[AgentDepsT] = None,
183
184
  deps: AgentDepsT = None,
184
185
  model_settings: ModelSettings | None = None,
185
186
  usage_limits: _usage.UsageLimits | None = None,
@@ -199,6 +200,7 @@ class PrefectAgent(WrapperAgent[AgentDepsT, OutputDataT]):
199
200
  message_history: Sequence[_messages.ModelMessage] | None = None,
200
201
  deferred_tool_results: DeferredToolResults | None = None,
201
202
  model: models.Model | models.KnownModelName | str | None = None,
203
+ instructions: Instructions[AgentDepsT] = None,
202
204
  deps: AgentDepsT = None,
203
205
  model_settings: ModelSettings | None = None,
204
206
  usage_limits: _usage.UsageLimits | None = None,
@@ -217,6 +219,7 @@ class PrefectAgent(WrapperAgent[AgentDepsT, OutputDataT]):
217
219
  message_history: Sequence[_messages.ModelMessage] | None = None,
218
220
  deferred_tool_results: DeferredToolResults | None = None,
219
221
  model: models.Model | models.KnownModelName | str | None = None,
222
+ instructions: Instructions[AgentDepsT] = None,
220
223
  deps: AgentDepsT = None,
221
224
  model_settings: ModelSettings | None = None,
222
225
  usage_limits: _usage.UsageLimits | None = None,
@@ -251,6 +254,7 @@ class PrefectAgent(WrapperAgent[AgentDepsT, OutputDataT]):
251
254
  message_history: History of the conversation so far.
252
255
  deferred_tool_results: Optional results for deferred tool calls in the message history.
253
256
  model: Optional model to use for this run, required if `model` was not set when creating the agent.
257
+ instructions: Optional additional instructions to use for this run.
254
258
  deps: Optional dependencies to use for this run.
255
259
  model_settings: Optional settings to use for this model's request.
256
260
  usage_limits: Optional limits on model request count or token usage.
@@ -276,6 +280,7 @@ class PrefectAgent(WrapperAgent[AgentDepsT, OutputDataT]):
276
280
  message_history=message_history,
277
281
  deferred_tool_results=deferred_tool_results,
278
282
  model=model,
283
+ instructions=instructions,
279
284
  deps=deps,
280
285
  model_settings=model_settings,
281
286
  usage_limits=usage_limits,
@@ -299,6 +304,7 @@ class PrefectAgent(WrapperAgent[AgentDepsT, OutputDataT]):
299
304
  message_history: Sequence[_messages.ModelMessage] | None = None,
300
305
  deferred_tool_results: DeferredToolResults | None = None,
301
306
  model: models.Model | models.KnownModelName | str | None = None,
307
+ instructions: Instructions[AgentDepsT] = None,
302
308
  deps: AgentDepsT = None,
303
309
  model_settings: ModelSettings | None = None,
304
310
  usage_limits: _usage.UsageLimits | None = None,
@@ -318,6 +324,7 @@ class PrefectAgent(WrapperAgent[AgentDepsT, OutputDataT]):
318
324
  message_history: Sequence[_messages.ModelMessage] | None = None,
319
325
  deferred_tool_results: DeferredToolResults | None = None,
320
326
  model: models.Model | models.KnownModelName | str | None = None,
327
+ instructions: Instructions[AgentDepsT] = None,
321
328
  deps: AgentDepsT = None,
322
329
  model_settings: ModelSettings | None = None,
323
330
  usage_limits: _usage.UsageLimits | None = None,
@@ -336,6 +343,7 @@ class PrefectAgent(WrapperAgent[AgentDepsT, OutputDataT]):
336
343
  message_history: Sequence[_messages.ModelMessage] | None = None,
337
344
  deferred_tool_results: DeferredToolResults | None = None,
338
345
  model: models.Model | models.KnownModelName | str | None = None,
346
+ instructions: Instructions[AgentDepsT] = None,
339
347
  deps: AgentDepsT = None,
340
348
  model_settings: ModelSettings | None = None,
341
349
  usage_limits: _usage.UsageLimits | None = None,
@@ -369,6 +377,7 @@ class PrefectAgent(WrapperAgent[AgentDepsT, OutputDataT]):
369
377
  message_history: History of the conversation so far.
370
378
  deferred_tool_results: Optional results for deferred tool calls in the message history.
371
379
  model: Optional model to use for this run, required if `model` was not set when creating the agent.
380
+ instructions: Optional additional instructions to use for this run.
372
381
  deps: Optional dependencies to use for this run.
373
382
  model_settings: Optional settings to use for this model's request.
374
383
  usage_limits: Optional limits on model request count or token usage.
@@ -396,6 +405,7 @@ class PrefectAgent(WrapperAgent[AgentDepsT, OutputDataT]):
396
405
  message_history=message_history,
397
406
  deferred_tool_results=deferred_tool_results,
398
407
  model=model,
408
+ instructions=instructions,
399
409
  deps=deps,
400
410
  model_settings=model_settings,
401
411
  usage_limits=usage_limits,
@@ -420,6 +430,7 @@ class PrefectAgent(WrapperAgent[AgentDepsT, OutputDataT]):
420
430
  message_history: Sequence[_messages.ModelMessage] | None = None,
421
431
  deferred_tool_results: DeferredToolResults | None = None,
422
432
  model: models.Model | models.KnownModelName | str | None = None,
433
+ instructions: Instructions[AgentDepsT] = None,
423
434
  deps: AgentDepsT = None,
424
435
  model_settings: ModelSettings | None = None,
425
436
  usage_limits: _usage.UsageLimits | None = None,
@@ -439,6 +450,7 @@ class PrefectAgent(WrapperAgent[AgentDepsT, OutputDataT]):
439
450
  message_history: Sequence[_messages.ModelMessage] | None = None,
440
451
  deferred_tool_results: DeferredToolResults | None = None,
441
452
  model: models.Model | models.KnownModelName | str | None = None,
453
+ instructions: Instructions[AgentDepsT] = None,
442
454
  deps: AgentDepsT = None,
443
455
  model_settings: ModelSettings | None = None,
444
456
  usage_limits: _usage.UsageLimits | None = None,
@@ -458,6 +470,7 @@ class PrefectAgent(WrapperAgent[AgentDepsT, OutputDataT]):
458
470
  message_history: Sequence[_messages.ModelMessage] | None = None,
459
471
  deferred_tool_results: DeferredToolResults | None = None,
460
472
  model: models.Model | models.KnownModelName | str | None = None,
473
+ instructions: Instructions[AgentDepsT] = None,
461
474
  deps: AgentDepsT = None,
462
475
  model_settings: ModelSettings | None = None,
463
476
  usage_limits: _usage.UsageLimits | None = None,
@@ -489,6 +502,7 @@ class PrefectAgent(WrapperAgent[AgentDepsT, OutputDataT]):
489
502
  message_history: History of the conversation so far.
490
503
  deferred_tool_results: Optional results for deferred tool calls in the message history.
491
504
  model: Optional model to use for this run, required if `model` was not set when creating the agent.
505
+ instructions: Optional additional instructions to use for this run.
492
506
  deps: Optional dependencies to use for this run.
493
507
  model_settings: Optional settings to use for this model's request.
494
508
  usage_limits: Optional limits on model request count or token usage.
@@ -513,6 +527,7 @@ class PrefectAgent(WrapperAgent[AgentDepsT, OutputDataT]):
513
527
  message_history=message_history,
514
528
  deferred_tool_results=deferred_tool_results,
515
529
  model=model,
530
+ instructions=instructions,
516
531
  deps=deps,
517
532
  model_settings=model_settings,
518
533
  usage_limits=usage_limits,
@@ -534,6 +549,7 @@ class PrefectAgent(WrapperAgent[AgentDepsT, OutputDataT]):
534
549
  message_history: Sequence[_messages.ModelMessage] | None = None,
535
550
  deferred_tool_results: DeferredToolResults | None = None,
536
551
  model: models.Model | models.KnownModelName | str | None = None,
552
+ instructions: Instructions[AgentDepsT] = None,
537
553
  deps: AgentDepsT = None,
538
554
  model_settings: ModelSettings | None = None,
539
555
  usage_limits: _usage.UsageLimits | None = None,
@@ -552,6 +568,7 @@ class PrefectAgent(WrapperAgent[AgentDepsT, OutputDataT]):
552
568
  message_history: Sequence[_messages.ModelMessage] | None = None,
553
569
  deferred_tool_results: DeferredToolResults | None = None,
554
570
  model: models.Model | models.KnownModelName | str | None = None,
571
+ instructions: Instructions[AgentDepsT] = None,
555
572
  deps: AgentDepsT = None,
556
573
  model_settings: ModelSettings | None = None,
557
574
  usage_limits: _usage.UsageLimits | None = None,
@@ -569,6 +586,7 @@ class PrefectAgent(WrapperAgent[AgentDepsT, OutputDataT]):
569
586
  message_history: Sequence[_messages.ModelMessage] | None = None,
570
587
  deferred_tool_results: DeferredToolResults | None = None,
571
588
  model: models.Model | models.KnownModelName | str | None = None,
589
+ instructions: Instructions[AgentDepsT] = None,
572
590
  deps: AgentDepsT = None,
573
591
  model_settings: ModelSettings | None = None,
574
592
  usage_limits: _usage.UsageLimits | None = None,
@@ -598,6 +616,9 @@ class PrefectAgent(WrapperAgent[AgentDepsT, OutputDataT]):
598
616
  PartStartEvent(index=0, part=TextPart(content='The capital of ')),
599
617
  FinalResultEvent(tool_name=None, tool_call_id=None),
600
618
  PartDeltaEvent(index=0, delta=TextPartDelta(content_delta='France is Paris. ')),
619
+ PartEndEvent(
620
+ index=0, part=TextPart(content='The capital of France is Paris. ')
621
+ ),
601
622
  AgentRunResultEvent(
602
623
  result=AgentRunResult(output='The capital of France is Paris. ')
603
624
  ),
@@ -615,6 +636,7 @@ class PrefectAgent(WrapperAgent[AgentDepsT, OutputDataT]):
615
636
  message_history: History of the conversation so far.
616
637
  deferred_tool_results: Optional results for deferred tool calls in the message history.
617
638
  model: Optional model to use for this run, required if `model` was not set when creating the agent.
639
+ instructions: Optional additional instructions to use for this run.
618
640
  deps: Optional dependencies to use for this run.
619
641
  model_settings: Optional settings to use for this model's request.
620
642
  usage_limits: Optional limits on model request count or token usage.
@@ -639,6 +661,7 @@ class PrefectAgent(WrapperAgent[AgentDepsT, OutputDataT]):
639
661
  message_history=message_history,
640
662
  deferred_tool_results=deferred_tool_results,
641
663
  model=model,
664
+ instructions=instructions,
642
665
  deps=deps,
643
666
  model_settings=model_settings,
644
667
  usage_limits=usage_limits,
@@ -657,6 +680,7 @@ class PrefectAgent(WrapperAgent[AgentDepsT, OutputDataT]):
657
680
  message_history: Sequence[_messages.ModelMessage] | None = None,
658
681
  deferred_tool_results: DeferredToolResults | None = None,
659
682
  model: models.Model | models.KnownModelName | str | None = None,
683
+ instructions: Instructions[AgentDepsT] = None,
660
684
  deps: AgentDepsT = None,
661
685
  model_settings: ModelSettings | None = None,
662
686
  usage_limits: _usage.UsageLimits | None = None,
@@ -675,6 +699,7 @@ class PrefectAgent(WrapperAgent[AgentDepsT, OutputDataT]):
675
699
  message_history: Sequence[_messages.ModelMessage] | None = None,
676
700
  deferred_tool_results: DeferredToolResults | None = None,
677
701
  model: models.Model | models.KnownModelName | str | None = None,
702
+ instructions: Instructions[AgentDepsT] = None,
678
703
  deps: AgentDepsT = None,
679
704
  model_settings: ModelSettings | None = None,
680
705
  usage_limits: _usage.UsageLimits | None = None,
@@ -693,6 +718,7 @@ class PrefectAgent(WrapperAgent[AgentDepsT, OutputDataT]):
693
718
  message_history: Sequence[_messages.ModelMessage] | None = None,
694
719
  deferred_tool_results: DeferredToolResults | None = None,
695
720
  model: models.Model | models.KnownModelName | str | None = None,
721
+ instructions: Instructions[AgentDepsT] = None,
696
722
  deps: AgentDepsT = None,
697
723
  model_settings: ModelSettings | None = None,
698
724
  usage_limits: _usage.UsageLimits | None = None,
@@ -767,6 +793,7 @@ class PrefectAgent(WrapperAgent[AgentDepsT, OutputDataT]):
767
793
  deferred_tool_results: Optional results for deferred tool calls in the message history.
768
794
  model: Optional model to use for this run, required if `model` was not set when creating the agent.
769
795
  deps: Optional dependencies to use for this run.
796
+ instructions: Optional additional instructions to use for this run.
770
797
  model_settings: Optional settings to use for this model's request.
771
798
  usage_limits: Optional limits on model request count or token usage.
772
799
  usage: Optional usage to start with, useful for resuming a conversation or agents used in tools.
@@ -789,6 +816,7 @@ class PrefectAgent(WrapperAgent[AgentDepsT, OutputDataT]):
789
816
  message_history=message_history,
790
817
  deferred_tool_results=deferred_tool_results,
791
818
  model=model,
819
+ instructions=instructions,
792
820
  deps=deps,
793
821
  model_settings=model_settings,
794
822
  usage_limits=usage_limits,
pydantic_ai/durable_exec/temporal/_agent.py CHANGED
@@ -263,6 +263,7 @@ class TemporalAgent(WrapperAgent[AgentDepsT, OutputDataT]):
263
263
  message_history: Sequence[_messages.ModelMessage] | None = None,
264
264
  deferred_tool_results: DeferredToolResults | None = None,
265
265
  model: models.Model | models.KnownModelName | str | None = None,
266
+ instructions: Instructions[AgentDepsT] = None,
266
267
  deps: AgentDepsT = None,
267
268
  model_settings: ModelSettings | None = None,
268
269
  usage_limits: _usage.UsageLimits | None = None,
@@ -282,6 +283,7 @@ class TemporalAgent(WrapperAgent[AgentDepsT, OutputDataT]):
282
283
  message_history: Sequence[_messages.ModelMessage] | None = None,
283
284
  deferred_tool_results: DeferredToolResults | None = None,
284
285
  model: models.Model | models.KnownModelName | str | None = None,
286
+ instructions: Instructions[AgentDepsT] = None,
285
287
  deps: AgentDepsT = None,
286
288
  model_settings: ModelSettings | None = None,
287
289
  usage_limits: _usage.UsageLimits | None = None,
@@ -300,6 +302,7 @@ class TemporalAgent(WrapperAgent[AgentDepsT, OutputDataT]):
300
302
  message_history: Sequence[_messages.ModelMessage] | None = None,
301
303
  deferred_tool_results: DeferredToolResults | None = None,
302
304
  model: models.Model | models.KnownModelName | str | None = None,
305
+ instructions: Instructions[AgentDepsT] = None,
303
306
  deps: AgentDepsT = None,
304
307
  model_settings: ModelSettings | None = None,
305
308
  usage_limits: _usage.UsageLimits | None = None,
@@ -334,6 +337,7 @@ class TemporalAgent(WrapperAgent[AgentDepsT, OutputDataT]):
334
337
  message_history: History of the conversation so far.
335
338
  deferred_tool_results: Optional results for deferred tool calls in the message history.
336
339
  model: Optional model to use for this run, required if `model` was not set when creating the agent.
340
+ instructions: Optional additional instructions to use for this run.
337
341
  deps: Optional dependencies to use for this run.
338
342
  model_settings: Optional settings to use for this model's request.
339
343
  usage_limits: Optional limits on model request count or token usage.
@@ -358,6 +362,7 @@ class TemporalAgent(WrapperAgent[AgentDepsT, OutputDataT]):
358
362
  message_history=message_history,
359
363
  deferred_tool_results=deferred_tool_results,
360
364
  model=model,
365
+ instructions=instructions,
361
366
  deps=deps,
362
367
  model_settings=model_settings,
363
368
  usage_limits=usage_limits,
@@ -378,6 +383,7 @@ class TemporalAgent(WrapperAgent[AgentDepsT, OutputDataT]):
378
383
  message_history: Sequence[_messages.ModelMessage] | None = None,
379
384
  deferred_tool_results: DeferredToolResults | None = None,
380
385
  model: models.Model | models.KnownModelName | str | None = None,
386
+ instructions: Instructions[AgentDepsT] = None,
381
387
  deps: AgentDepsT = None,
382
388
  model_settings: ModelSettings | None = None,
383
389
  usage_limits: _usage.UsageLimits | None = None,
@@ -397,6 +403,7 @@ class TemporalAgent(WrapperAgent[AgentDepsT, OutputDataT]):
397
403
  message_history: Sequence[_messages.ModelMessage] | None = None,
398
404
  deferred_tool_results: DeferredToolResults | None = None,
399
405
  model: models.Model | models.KnownModelName | str | None = None,
406
+ instructions: Instructions[AgentDepsT] = None,
400
407
  deps: AgentDepsT = None,
401
408
  model_settings: ModelSettings | None = None,
402
409
  usage_limits: _usage.UsageLimits | None = None,
@@ -415,6 +422,7 @@ class TemporalAgent(WrapperAgent[AgentDepsT, OutputDataT]):
415
422
  message_history: Sequence[_messages.ModelMessage] | None = None,
416
423
  deferred_tool_results: DeferredToolResults | None = None,
417
424
  model: models.Model | models.KnownModelName | str | None = None,
425
+ instructions: Instructions[AgentDepsT] = None,
418
426
  deps: AgentDepsT = None,
419
427
  model_settings: ModelSettings | None = None,
420
428
  usage_limits: _usage.UsageLimits | None = None,
@@ -448,6 +456,7 @@ class TemporalAgent(WrapperAgent[AgentDepsT, OutputDataT]):
448
456
  message_history: History of the conversation so far.
449
457
  deferred_tool_results: Optional results for deferred tool calls in the message history.
450
458
  model: Optional model to use for this run, required if `model` was not set when creating the agent.
459
+ instructions: Optional additional instructions to use for this run.
451
460
  deps: Optional dependencies to use for this run.
452
461
  model_settings: Optional settings to use for this model's request.
453
462
  usage_limits: Optional limits on model request count or token usage.
@@ -471,6 +480,7 @@ class TemporalAgent(WrapperAgent[AgentDepsT, OutputDataT]):
471
480
  message_history=message_history,
472
481
  deferred_tool_results=deferred_tool_results,
473
482
  model=model,
483
+ instructions=instructions,
474
484
  deps=deps,
475
485
  model_settings=model_settings,
476
486
  usage_limits=usage_limits,
@@ -491,6 +501,7 @@ class TemporalAgent(WrapperAgent[AgentDepsT, OutputDataT]):
491
501
  message_history: Sequence[_messages.ModelMessage] | None = None,
492
502
  deferred_tool_results: DeferredToolResults | None = None,
493
503
  model: models.Model | models.KnownModelName | str | None = None,
504
+ instructions: Instructions[AgentDepsT] = None,
494
505
  deps: AgentDepsT = None,
495
506
  model_settings: ModelSettings | None = None,
496
507
  usage_limits: _usage.UsageLimits | None = None,
@@ -510,6 +521,7 @@ class TemporalAgent(WrapperAgent[AgentDepsT, OutputDataT]):
510
521
  message_history: Sequence[_messages.ModelMessage] | None = None,
511
522
  deferred_tool_results: DeferredToolResults | None = None,
512
523
  model: models.Model | models.KnownModelName | str | None = None,
524
+ instructions: Instructions[AgentDepsT] = None,
513
525
  deps: AgentDepsT = None,
514
526
  model_settings: ModelSettings | None = None,
515
527
  usage_limits: _usage.UsageLimits | None = None,
@@ -529,6 +541,7 @@ class TemporalAgent(WrapperAgent[AgentDepsT, OutputDataT]):
529
541
  message_history: Sequence[_messages.ModelMessage] | None = None,
530
542
  deferred_tool_results: DeferredToolResults | None = None,
531
543
  model: models.Model | models.KnownModelName | str | None = None,
544
+ instructions: Instructions[AgentDepsT] = None,
532
545
  deps: AgentDepsT = None,
533
546
  model_settings: ModelSettings | None = None,
534
547
  usage_limits: _usage.UsageLimits | None = None,
@@ -560,6 +573,7 @@ class TemporalAgent(WrapperAgent[AgentDepsT, OutputDataT]):
560
573
  message_history: History of the conversation so far.
561
574
  deferred_tool_results: Optional results for deferred tool calls in the message history.
562
575
  model: Optional model to use for this run, required if `model` was not set when creating the agent.
576
+ instructions: Optional additional instructions to use for this run.
563
577
  deps: Optional dependencies to use for this run.
564
578
  model_settings: Optional settings to use for this model's request.
565
579
  usage_limits: Optional limits on model request count or token usage.
@@ -584,6 +598,7 @@ class TemporalAgent(WrapperAgent[AgentDepsT, OutputDataT]):
584
598
  message_history=message_history,
585
599
  deferred_tool_results=deferred_tool_results,
586
600
  model=model,
601
+ instructions=instructions,
587
602
  deps=deps,
588
603
  model_settings=model_settings,
589
604
  usage_limits=usage_limits,
@@ -605,6 +620,7 @@ class TemporalAgent(WrapperAgent[AgentDepsT, OutputDataT]):
605
620
  message_history: Sequence[_messages.ModelMessage] | None = None,
606
621
  deferred_tool_results: DeferredToolResults | None = None,
607
622
  model: models.Model | models.KnownModelName | str | None = None,
623
+ instructions: Instructions[AgentDepsT] = None,
608
624
  deps: AgentDepsT = None,
609
625
  model_settings: ModelSettings | None = None,
610
626
  usage_limits: _usage.UsageLimits | None = None,
@@ -623,6 +639,7 @@ class TemporalAgent(WrapperAgent[AgentDepsT, OutputDataT]):
623
639
  message_history: Sequence[_messages.ModelMessage] | None = None,
624
640
  deferred_tool_results: DeferredToolResults | None = None,
625
641
  model: models.Model | models.KnownModelName | str | None = None,
642
+ instructions: Instructions[AgentDepsT] = None,
626
643
  deps: AgentDepsT = None,
627
644
  model_settings: ModelSettings | None = None,
628
645
  usage_limits: _usage.UsageLimits | None = None,
@@ -640,6 +657,7 @@ class TemporalAgent(WrapperAgent[AgentDepsT, OutputDataT]):
640
657
  message_history: Sequence[_messages.ModelMessage] | None = None,
641
658
  deferred_tool_results: DeferredToolResults | None = None,
642
659
  model: models.Model | models.KnownModelName | str | None = None,
660
+ instructions: Instructions[AgentDepsT] = None,
643
661
  deps: AgentDepsT = None,
644
662
  model_settings: ModelSettings | None = None,
645
663
  usage_limits: _usage.UsageLimits | None = None,
@@ -669,6 +687,9 @@ class TemporalAgent(WrapperAgent[AgentDepsT, OutputDataT]):
669
687
  PartStartEvent(index=0, part=TextPart(content='The capital of ')),
670
688
  FinalResultEvent(tool_name=None, tool_call_id=None),
671
689
  PartDeltaEvent(index=0, delta=TextPartDelta(content_delta='France is Paris. ')),
690
+ PartEndEvent(
691
+ index=0, part=TextPart(content='The capital of France is Paris. ')
692
+ ),
672
693
  AgentRunResultEvent(
673
694
  result=AgentRunResult(output='The capital of France is Paris. ')
674
695
  ),
@@ -686,6 +707,7 @@ class TemporalAgent(WrapperAgent[AgentDepsT, OutputDataT]):
686
707
  message_history: History of the conversation so far.
687
708
  deferred_tool_results: Optional results for deferred tool calls in the message history.
688
709
  model: Optional model to use for this run, required if `model` was not set when creating the agent.
710
+ instructions: Optional additional instructions to use for this run.
689
711
  deps: Optional dependencies to use for this run.
690
712
  model_settings: Optional settings to use for this model's request.
691
713
  usage_limits: Optional limits on model request count or token usage.
@@ -710,6 +732,7 @@ class TemporalAgent(WrapperAgent[AgentDepsT, OutputDataT]):
710
732
  message_history=message_history,
711
733
  deferred_tool_results=deferred_tool_results,
712
734
  model=model,
735
+ instructions=instructions,
713
736
  deps=deps,
714
737
  model_settings=model_settings,
715
738
  usage_limits=usage_limits,
@@ -728,6 +751,7 @@ class TemporalAgent(WrapperAgent[AgentDepsT, OutputDataT]):
728
751
  message_history: Sequence[_messages.ModelMessage] | None = None,
729
752
  deferred_tool_results: DeferredToolResults | None = None,
730
753
  model: models.Model | models.KnownModelName | str | None = None,
754
+ instructions: Instructions[AgentDepsT] = None,
731
755
  deps: AgentDepsT = None,
732
756
  model_settings: ModelSettings | None = None,
733
757
  usage_limits: _usage.UsageLimits | None = None,
@@ -747,6 +771,7 @@ class TemporalAgent(WrapperAgent[AgentDepsT, OutputDataT]):
747
771
  message_history: Sequence[_messages.ModelMessage] | None = None,
748
772
  deferred_tool_results: DeferredToolResults | None = None,
749
773
  model: models.Model | models.KnownModelName | str | None = None,
774
+ instructions: Instructions[AgentDepsT] = None,
750
775
  deps: AgentDepsT = None,
751
776
  model_settings: ModelSettings | None = None,
752
777
  usage_limits: _usage.UsageLimits | None = None,
@@ -766,6 +791,7 @@ class TemporalAgent(WrapperAgent[AgentDepsT, OutputDataT]):
766
791
  message_history: Sequence[_messages.ModelMessage] | None = None,
767
792
  deferred_tool_results: DeferredToolResults | None = None,
768
793
  model: models.Model | models.KnownModelName | str | None = None,
794
+ instructions: Instructions[AgentDepsT] = None,
769
795
  deps: AgentDepsT = None,
770
796
  model_settings: ModelSettings | None = None,
771
797
  usage_limits: _usage.UsageLimits | None = None,
@@ -840,6 +866,7 @@ class TemporalAgent(WrapperAgent[AgentDepsT, OutputDataT]):
840
866
  message_history: History of the conversation so far.
841
867
  deferred_tool_results: Optional results for deferred tool calls in the message history.
842
868
  model: Optional model to use for this run, required if `model` was not set when creating the agent.
869
+ instructions: Optional additional instructions to use for this run.
843
870
  deps: Optional dependencies to use for this run.
844
871
  model_settings: Optional settings to use for this model's request.
845
872
  usage_limits: Optional limits on model request count or token usage.
@@ -873,6 +900,7 @@ class TemporalAgent(WrapperAgent[AgentDepsT, OutputDataT]):
873
900
  message_history=message_history,
874
901
  deferred_tool_results=deferred_tool_results,
875
902
  model=model,
903
+ instructions=instructions,
876
904
  deps=deps,
877
905
  model_settings=model_settings,
878
906
  usage_limits=usage_limits,