pydantic-ai-slim 0.8.0__py3-none-any.whl → 1.0.0b1__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 (70) hide show
  1. pydantic_ai/__init__.py +28 -2
  2. pydantic_ai/_agent_graph.py +310 -140
  3. pydantic_ai/_function_schema.py +5 -5
  4. pydantic_ai/_griffe.py +2 -1
  5. pydantic_ai/_otel_messages.py +2 -2
  6. pydantic_ai/_output.py +31 -35
  7. pydantic_ai/_parts_manager.py +4 -4
  8. pydantic_ai/_run_context.py +3 -1
  9. pydantic_ai/_system_prompt.py +2 -2
  10. pydantic_ai/_tool_manager.py +3 -22
  11. pydantic_ai/_utils.py +14 -26
  12. pydantic_ai/ag_ui.py +7 -8
  13. pydantic_ai/agent/__init__.py +84 -17
  14. pydantic_ai/agent/abstract.py +35 -4
  15. pydantic_ai/agent/wrapper.py +6 -0
  16. pydantic_ai/builtin_tools.py +2 -2
  17. pydantic_ai/common_tools/duckduckgo.py +4 -2
  18. pydantic_ai/durable_exec/temporal/__init__.py +70 -17
  19. pydantic_ai/durable_exec/temporal/_agent.py +23 -2
  20. pydantic_ai/durable_exec/temporal/_function_toolset.py +53 -6
  21. pydantic_ai/durable_exec/temporal/_logfire.py +6 -3
  22. pydantic_ai/durable_exec/temporal/_mcp_server.py +2 -1
  23. pydantic_ai/durable_exec/temporal/_model.py +2 -2
  24. pydantic_ai/durable_exec/temporal/_run_context.py +2 -1
  25. pydantic_ai/durable_exec/temporal/_toolset.py +2 -1
  26. pydantic_ai/exceptions.py +45 -2
  27. pydantic_ai/format_prompt.py +2 -2
  28. pydantic_ai/mcp.py +2 -2
  29. pydantic_ai/messages.py +81 -28
  30. pydantic_ai/models/__init__.py +19 -7
  31. pydantic_ai/models/anthropic.py +6 -6
  32. pydantic_ai/models/bedrock.py +63 -57
  33. pydantic_ai/models/cohere.py +3 -3
  34. pydantic_ai/models/fallback.py +2 -2
  35. pydantic_ai/models/function.py +25 -23
  36. pydantic_ai/models/gemini.py +10 -13
  37. pydantic_ai/models/google.py +4 -4
  38. pydantic_ai/models/groq.py +5 -5
  39. pydantic_ai/models/huggingface.py +5 -5
  40. pydantic_ai/models/instrumented.py +44 -21
  41. pydantic_ai/models/mcp_sampling.py +3 -1
  42. pydantic_ai/models/mistral.py +8 -8
  43. pydantic_ai/models/openai.py +20 -29
  44. pydantic_ai/models/test.py +24 -4
  45. pydantic_ai/output.py +27 -32
  46. pydantic_ai/profiles/__init__.py +3 -3
  47. pydantic_ai/profiles/groq.py +1 -1
  48. pydantic_ai/profiles/openai.py +25 -4
  49. pydantic_ai/providers/anthropic.py +2 -3
  50. pydantic_ai/providers/bedrock.py +3 -2
  51. pydantic_ai/result.py +173 -52
  52. pydantic_ai/retries.py +10 -29
  53. pydantic_ai/run.py +12 -5
  54. pydantic_ai/tools.py +126 -22
  55. pydantic_ai/toolsets/__init__.py +4 -1
  56. pydantic_ai/toolsets/_dynamic.py +4 -4
  57. pydantic_ai/toolsets/abstract.py +18 -2
  58. pydantic_ai/toolsets/approval_required.py +32 -0
  59. pydantic_ai/toolsets/combined.py +7 -12
  60. pydantic_ai/toolsets/{deferred.py → external.py} +11 -5
  61. pydantic_ai/toolsets/filtered.py +1 -1
  62. pydantic_ai/toolsets/function.py +13 -4
  63. pydantic_ai/toolsets/wrapper.py +2 -1
  64. pydantic_ai/usage.py +7 -5
  65. {pydantic_ai_slim-0.8.0.dist-info → pydantic_ai_slim-1.0.0b1.dist-info}/METADATA +6 -7
  66. pydantic_ai_slim-1.0.0b1.dist-info/RECORD +120 -0
  67. pydantic_ai_slim-0.8.0.dist-info/RECORD +0 -119
  68. {pydantic_ai_slim-0.8.0.dist-info → pydantic_ai_slim-1.0.0b1.dist-info}/WHEEL +0 -0
  69. {pydantic_ai_slim-0.8.0.dist-info → pydantic_ai_slim-1.0.0b1.dist-info}/entry_points.txt +0 -0
  70. {pydantic_ai_slim-0.8.0.dist-info → pydantic_ai_slim-1.0.0b1.dist-info}/licenses/LICENSE +0 -0
pydantic_ai/__init__.py CHANGED
@@ -1,9 +1,19 @@
1
1
  from importlib.metadata import version as _metadata_version
2
2
 
3
- from .agent import Agent, CallToolsNode, EndStrategy, ModelRequestNode, UserPromptNode, capture_run_messages
3
+ from .agent import (
4
+ Agent,
5
+ CallToolsNode,
6
+ EndStrategy,
7
+ InstrumentationSettings,
8
+ ModelRequestNode,
9
+ UserPromptNode,
10
+ capture_run_messages,
11
+ )
4
12
  from .builtin_tools import CodeExecutionTool, UrlContextTool, WebSearchTool, WebSearchUserLocation
5
13
  from .exceptions import (
6
14
  AgentRunError,
15
+ ApprovalRequired,
16
+ CallDeferred,
7
17
  FallbackExceptionGroup,
8
18
  ModelHTTPError,
9
19
  ModelRetry,
@@ -14,7 +24,9 @@ from .exceptions import (
14
24
  from .format_prompt import format_as_xml
15
25
  from .messages import AudioUrl, BinaryContent, DocumentUrl, ImageUrl, VideoUrl
16
26
  from .output import NativeOutput, PromptedOutput, StructuredDict, TextOutput, ToolOutput
17
- from .tools import RunContext, Tool
27
+ from .settings import ModelSettings
28
+ from .tools import DeferredToolRequests, DeferredToolResults, RunContext, Tool, ToolApproved, ToolDefinition, ToolDenied
29
+ from .usage import RequestUsage, RunUsage, UsageLimits
18
30
 
19
31
  __all__ = (
20
32
  '__version__',
@@ -25,8 +37,11 @@ __all__ = (
25
37
  'ModelRequestNode',
26
38
  'UserPromptNode',
27
39
  'capture_run_messages',
40
+ 'InstrumentationSettings',
28
41
  # exceptions
29
42
  'AgentRunError',
43
+ 'CallDeferred',
44
+ 'ApprovalRequired',
30
45
  'ModelRetry',
31
46
  'ModelHTTPError',
32
47
  'FallbackExceptionGroup',
@@ -41,7 +56,12 @@ __all__ = (
41
56
  'BinaryContent',
42
57
  # tools
43
58
  'Tool',
59
+ 'ToolDefinition',
44
60
  'RunContext',
61
+ 'DeferredToolRequests',
62
+ 'DeferredToolResults',
63
+ 'ToolApproved',
64
+ 'ToolDenied',
45
65
  # builtin_tools
46
66
  'WebSearchTool',
47
67
  'WebSearchUserLocation',
@@ -55,5 +75,11 @@ __all__ = (
55
75
  'StructuredDict',
56
76
  # format_prompt
57
77
  'format_as_xml',
78
+ # settings
79
+ 'ModelSettings',
80
+ # usage
81
+ 'RunUsage',
82
+ 'RequestUsage',
83
+ 'UsageLimits',
58
84
  )
59
85
  __version__ = _metadata_version('pydantic_ai_slim')