pydantic-ai-slim 1.9.1__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 (26) hide show
  1. pydantic_ai/_run_context.py +8 -2
  2. pydantic_ai/_tool_manager.py +1 -0
  3. pydantic_ai/_utils.py +18 -0
  4. pydantic_ai/agent/__init__.py +13 -3
  5. pydantic_ai/agent/abstract.py +155 -3
  6. pydantic_ai/agent/wrapper.py +5 -0
  7. pydantic_ai/durable_exec/dbos/_agent.py +28 -0
  8. pydantic_ai/durable_exec/prefect/_agent.py +25 -0
  9. pydantic_ai/durable_exec/temporal/_agent.py +25 -0
  10. pydantic_ai/durable_exec/temporal/_function_toolset.py +23 -73
  11. pydantic_ai/durable_exec/temporal/_mcp_server.py +30 -30
  12. pydantic_ai/durable_exec/temporal/_run_context.py +9 -3
  13. pydantic_ai/durable_exec/temporal/_toolset.py +67 -3
  14. pydantic_ai/messages.py +10 -1
  15. pydantic_ai/models/openai.py +4 -0
  16. pydantic_ai/profiles/openai.py +5 -2
  17. pydantic_ai/providers/openrouter.py +3 -0
  18. pydantic_ai/result.py +159 -4
  19. pydantic_ai/tools.py +10 -6
  20. pydantic_ai/ui/_event_stream.py +4 -4
  21. pydantic_ai/ui/ag_ui/_event_stream.py +11 -2
  22. {pydantic_ai_slim-1.9.1.dist-info → pydantic_ai_slim-1.11.0.dist-info}/METADATA +8 -6
  23. {pydantic_ai_slim-1.9.1.dist-info → pydantic_ai_slim-1.11.0.dist-info}/RECORD +26 -26
  24. {pydantic_ai_slim-1.9.1.dist-info → pydantic_ai_slim-1.11.0.dist-info}/WHEEL +0 -0
  25. {pydantic_ai_slim-1.9.1.dist-info → pydantic_ai_slim-1.11.0.dist-info}/entry_points.txt +0 -0
  26. {pydantic_ai_slim-1.9.1.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,
@@ -660,6 +682,7 @@ class DBOSAgent(WrapperAgent[AgentDepsT, OutputDataT], DBOSConfiguredInstance):
660
682
  message_history: History of the conversation so far.
661
683
  deferred_tool_results: Optional results for deferred tool calls in the message history.
662
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.
663
686
  deps: Optional dependencies to use for this run.
664
687
  model_settings: Optional settings to use for this model's request.
665
688
  usage_limits: Optional limits on model request count or token usage.
@@ -686,6 +709,7 @@ class DBOSAgent(WrapperAgent[AgentDepsT, OutputDataT], DBOSConfiguredInstance):
686
709
  message_history: Sequence[_messages.ModelMessage] | None = None,
687
710
  deferred_tool_results: DeferredToolResults | None = None,
688
711
  model: models.Model | models.KnownModelName | str | None = None,
712
+ instructions: Instructions[AgentDepsT] = None,
689
713
  deps: AgentDepsT = None,
690
714
  model_settings: ModelSettings | None = None,
691
715
  usage_limits: _usage.UsageLimits | None = None,
@@ -705,6 +729,7 @@ class DBOSAgent(WrapperAgent[AgentDepsT, OutputDataT], DBOSConfiguredInstance):
705
729
  message_history: Sequence[_messages.ModelMessage] | None = None,
706
730
  deferred_tool_results: DeferredToolResults | None = None,
707
731
  model: models.Model | models.KnownModelName | str | None = None,
732
+ instructions: Instructions[AgentDepsT] = None,
708
733
  deps: AgentDepsT = None,
709
734
  model_settings: ModelSettings | None = None,
710
735
  usage_limits: _usage.UsageLimits | None = None,
@@ -724,6 +749,7 @@ class DBOSAgent(WrapperAgent[AgentDepsT, OutputDataT], DBOSConfiguredInstance):
724
749
  message_history: Sequence[_messages.ModelMessage] | None = None,
725
750
  deferred_tool_results: DeferredToolResults | None = None,
726
751
  model: models.Model | models.KnownModelName | str | None = None,
752
+ instructions: Instructions[AgentDepsT] = None,
727
753
  deps: AgentDepsT = None,
728
754
  model_settings: ModelSettings | None = None,
729
755
  usage_limits: _usage.UsageLimits | None = None,
@@ -798,6 +824,7 @@ class DBOSAgent(WrapperAgent[AgentDepsT, OutputDataT], DBOSConfiguredInstance):
798
824
  message_history: History of the conversation so far.
799
825
  deferred_tool_results: Optional results for deferred tool calls in the message history.
800
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.
801
828
  deps: Optional dependencies to use for this run.
802
829
  model_settings: Optional settings to use for this model's request.
803
830
  usage_limits: Optional limits on model request count or token usage.
@@ -821,6 +848,7 @@ class DBOSAgent(WrapperAgent[AgentDepsT, OutputDataT], DBOSConfiguredInstance):
821
848
  message_history=message_history,
822
849
  deferred_tool_results=deferred_tool_results,
823
850
  model=model,
851
+ instructions=instructions,
824
852
  deps=deps,
825
853
  model_settings=model_settings,
826
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,
@@ -618,6 +636,7 @@ class PrefectAgent(WrapperAgent[AgentDepsT, OutputDataT]):
618
636
  message_history: History of the conversation so far.
619
637
  deferred_tool_results: Optional results for deferred tool calls in the message history.
620
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.
621
640
  deps: Optional dependencies to use for this run.
622
641
  model_settings: Optional settings to use for this model's request.
623
642
  usage_limits: Optional limits on model request count or token usage.
@@ -642,6 +661,7 @@ class PrefectAgent(WrapperAgent[AgentDepsT, OutputDataT]):
642
661
  message_history=message_history,
643
662
  deferred_tool_results=deferred_tool_results,
644
663
  model=model,
664
+ instructions=instructions,
645
665
  deps=deps,
646
666
  model_settings=model_settings,
647
667
  usage_limits=usage_limits,
@@ -660,6 +680,7 @@ class PrefectAgent(WrapperAgent[AgentDepsT, OutputDataT]):
660
680
  message_history: Sequence[_messages.ModelMessage] | None = None,
661
681
  deferred_tool_results: DeferredToolResults | None = None,
662
682
  model: models.Model | models.KnownModelName | str | None = None,
683
+ instructions: Instructions[AgentDepsT] = None,
663
684
  deps: AgentDepsT = None,
664
685
  model_settings: ModelSettings | None = None,
665
686
  usage_limits: _usage.UsageLimits | None = None,
@@ -678,6 +699,7 @@ class PrefectAgent(WrapperAgent[AgentDepsT, OutputDataT]):
678
699
  message_history: Sequence[_messages.ModelMessage] | None = None,
679
700
  deferred_tool_results: DeferredToolResults | None = None,
680
701
  model: models.Model | models.KnownModelName | str | None = None,
702
+ instructions: Instructions[AgentDepsT] = None,
681
703
  deps: AgentDepsT = None,
682
704
  model_settings: ModelSettings | None = None,
683
705
  usage_limits: _usage.UsageLimits | None = None,
@@ -696,6 +718,7 @@ class PrefectAgent(WrapperAgent[AgentDepsT, OutputDataT]):
696
718
  message_history: Sequence[_messages.ModelMessage] | None = None,
697
719
  deferred_tool_results: DeferredToolResults | None = None,
698
720
  model: models.Model | models.KnownModelName | str | None = None,
721
+ instructions: Instructions[AgentDepsT] = None,
699
722
  deps: AgentDepsT = None,
700
723
  model_settings: ModelSettings | None = None,
701
724
  usage_limits: _usage.UsageLimits | None = None,
@@ -770,6 +793,7 @@ class PrefectAgent(WrapperAgent[AgentDepsT, OutputDataT]):
770
793
  deferred_tool_results: Optional results for deferred tool calls in the message history.
771
794
  model: Optional model to use for this run, required if `model` was not set when creating the agent.
772
795
  deps: Optional dependencies to use for this run.
796
+ instructions: Optional additional instructions to use for this run.
773
797
  model_settings: Optional settings to use for this model's request.
774
798
  usage_limits: Optional limits on model request count or token usage.
775
799
  usage: Optional usage to start with, useful for resuming a conversation or agents used in tools.
@@ -792,6 +816,7 @@ class PrefectAgent(WrapperAgent[AgentDepsT, OutputDataT]):
792
816
  message_history=message_history,
793
817
  deferred_tool_results=deferred_tool_results,
794
818
  model=model,
819
+ instructions=instructions,
795
820
  deps=deps,
796
821
  model_settings=model_settings,
797
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,
@@ -689,6 +707,7 @@ class TemporalAgent(WrapperAgent[AgentDepsT, OutputDataT]):
689
707
  message_history: History of the conversation so far.
690
708
  deferred_tool_results: Optional results for deferred tool calls in the message history.
691
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.
692
711
  deps: Optional dependencies to use for this run.
693
712
  model_settings: Optional settings to use for this model's request.
694
713
  usage_limits: Optional limits on model request count or token usage.
@@ -713,6 +732,7 @@ class TemporalAgent(WrapperAgent[AgentDepsT, OutputDataT]):
713
732
  message_history=message_history,
714
733
  deferred_tool_results=deferred_tool_results,
715
734
  model=model,
735
+ instructions=instructions,
716
736
  deps=deps,
717
737
  model_settings=model_settings,
718
738
  usage_limits=usage_limits,
@@ -731,6 +751,7 @@ class TemporalAgent(WrapperAgent[AgentDepsT, OutputDataT]):
731
751
  message_history: Sequence[_messages.ModelMessage] | None = None,
732
752
  deferred_tool_results: DeferredToolResults | None = None,
733
753
  model: models.Model | models.KnownModelName | str | None = None,
754
+ instructions: Instructions[AgentDepsT] = None,
734
755
  deps: AgentDepsT = None,
735
756
  model_settings: ModelSettings | None = None,
736
757
  usage_limits: _usage.UsageLimits | None = None,
@@ -750,6 +771,7 @@ class TemporalAgent(WrapperAgent[AgentDepsT, OutputDataT]):
750
771
  message_history: Sequence[_messages.ModelMessage] | None = None,
751
772
  deferred_tool_results: DeferredToolResults | None = None,
752
773
  model: models.Model | models.KnownModelName | str | None = None,
774
+ instructions: Instructions[AgentDepsT] = None,
753
775
  deps: AgentDepsT = None,
754
776
  model_settings: ModelSettings | None = None,
755
777
  usage_limits: _usage.UsageLimits | None = None,
@@ -769,6 +791,7 @@ class TemporalAgent(WrapperAgent[AgentDepsT, OutputDataT]):
769
791
  message_history: Sequence[_messages.ModelMessage] | None = None,
770
792
  deferred_tool_results: DeferredToolResults | None = None,
771
793
  model: models.Model | models.KnownModelName | str | None = None,
794
+ instructions: Instructions[AgentDepsT] = None,
772
795
  deps: AgentDepsT = None,
773
796
  model_settings: ModelSettings | None = None,
774
797
  usage_limits: _usage.UsageLimits | None = None,
@@ -843,6 +866,7 @@ class TemporalAgent(WrapperAgent[AgentDepsT, OutputDataT]):
843
866
  message_history: History of the conversation so far.
844
867
  deferred_tool_results: Optional results for deferred tool calls in the message history.
845
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.
846
870
  deps: Optional dependencies to use for this run.
847
871
  model_settings: Optional settings to use for this model's request.
848
872
  usage_limits: Optional limits on model request count or token usage.
@@ -876,6 +900,7 @@ class TemporalAgent(WrapperAgent[AgentDepsT, OutputDataT]):
876
900
  message_history=message_history,
877
901
  deferred_tool_results=deferred_tool_results,
878
902
  model=model,
903
+ instructions=instructions,
879
904
  deps=deps,
880
905
  model_settings=model_settings,
881
906
  usage_limits=usage_limits,