pydantic-ai 0.3.7__tar.gz → 0.4.0__tar.gz

{pydantic_ai-0.3.7 → pydantic_ai-0.4.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pydantic-ai
-Version: 0.3.7
+Version: 0.4.0
 Summary: Agent Framework / shim to use Pydantic with LLMs
 Project-URL: Homepage, https://ai.pydantic.dev
 Project-URL: Source, https://github.com/pydantic/pydantic-ai
@@ -28,9 +28,9 @@ Classifier: Topic :: Internet
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Requires-Python: >=3.9
-Requires-Dist: pydantic-ai-slim[a2a,anthropic,bedrock,cli,cohere,evals,google,groq,mcp,mistral,openai,vertexai]==0.3.7
+Requires-Dist: pydantic-ai-slim[a2a,anthropic,bedrock,cli,cohere,evals,google,groq,mcp,mistral,openai,vertexai]==0.4.0
 Provides-Extra: examples
-Requires-Dist: pydantic-ai-examples==0.3.7; extra == 'examples'
+Requires-Dist: pydantic-ai-examples==0.4.0; extra == 'examples'
 Provides-Extra: logfire
 Requires-Dist: logfire>=3.11.0; extra == 'logfire'
 Description-Content-Type: text/markdown

{pydantic_ai-0.3.7 → pydantic_ai-0.4.0}/tests/evals/test_dataset.py

@@ -33,7 +33,7 @@ with try_import() as imports_successful:
         def evaluate(self, ctx: EvaluatorContext[object, object, object]) -> EvaluatorOutput:
             return self.output
 
-    from pydantic_evals.reporting import ReportCase
+    from pydantic_evals.reporting import ReportCase, ReportCaseAdapter
 
 pytestmark = [pytest.mark.skipif(not imports_successful(), reason='pydantic-evals not installed'), pytest.mark.anyio]
 
@@ -196,7 +196,7 @@ async def test_evaluate(
 
     assert report is not None
     assert len(report.cases) == 2
-    assert report.cases[0].model_dump() == snapshot(
+    assert ReportCaseAdapter.dump_python(report.cases[0]) == snapshot(
         {
             'assertions': {
                 'correct': {
@@ -248,7 +248,7 @@ async def test_evaluate_with_concurrency(
 
     assert report is not None
     assert len(report.cases) == 2
-    assert report.cases[0].model_dump() == snapshot(
+    assert ReportCaseAdapter.dump_python(report.cases[0]) == snapshot(
         {
             'assertions': {
                 'correct': {

{pydantic_ai-0.3.7 → pydantic_ai-0.4.0}/tests/evals/test_otel.py

@@ -65,9 +65,10 @@ async def test_context_subtree_concurrent():
     task1_root = tree1.roots[0]
     assert len(task1_root.children) == 2, 'task1 should have exactly two children'
     task1_child_names = {child.name for child in task1_root.children}
-    assert task1_child_names == {'task1_child1', 'task1_child2'}, (
-        "task1's children should be task1_child1 and task1_child2"
-    )
+    assert task1_child_names == {
+        'task1_child1',
+        'task1_child2',
+    }, "task1's children should be task1_child1 and task1_child2"
 
     # Verify that tree2 only contains spans from task2
     assert len(tree2.roots) == 1, 'tree2 should have exactly one root span'

{pydantic_ai-0.3.7 → pydantic_ai-0.4.0}/tests/evals/test_reports.py

@@ -12,9 +12,11 @@ with try_import() as imports_successful:
     from pydantic_evals.evaluators import EvaluationResult, Evaluator, EvaluatorContext
     from pydantic_evals.reporting import (
         EvaluationReport,
+        EvaluationReportAdapter,
         RenderNumberConfig,
         RenderValueConfig,
         ReportCase,
+        ReportCaseAdapter,
         ReportCaseAggregate,
     )
 
@@ -157,7 +159,7 @@ async def test_report_case_aggregate():
 async def test_report_serialization(sample_report: EvaluationReport):
     """Test serializing a report to dict."""
     # Serialize the report
-    serialized = sample_report.model_dump()
+    serialized = EvaluationReportAdapter.dump_python(sample_report)
 
     # Check the serialized structure
     assert 'cases' in serialized
@@ -202,7 +204,7 @@ async def test_report_with_error(mock_evaluator: Evaluator[TaskInput, TaskOutput
         name='error_report',
     )
 
-    assert report.cases[0].model_dump() == snapshot(
+    assert ReportCaseAdapter.dump_python(report.cases[0]) == snapshot(
         {
             'assertions': {
                 'error_evaluator': {

{pydantic_ai-0.3.7 → pydantic_ai-0.4.0}/tests/models/test_bedrock.py

@@ -1443,9 +1443,9 @@ async def test_bedrock_mistral_tool_result_format(bedrock_provider: BedrockProvi
 
 async def test_bedrock_anthropic_no_tool_choice(bedrock_provider: BedrockProvider):
     my_tool = ToolDefinition(
-        'my_tool',
-        'This is my tool',
-        {'type': 'object', 'title': 'Result', 'properties': {'spam': {'type': 'number'}}},
+        name='my_tool',
+        description='This is my tool',
+        parameters_json_schema={'type': 'object', 'title': 'Result', 'properties': {'spam': {'type': 'number'}}},
     )
     mrp = ModelRequestParameters(output_mode='tool', function_tools=[my_tool], allow_text_output=False, output_tools=[])
 

{pydantic_ai-0.3.7 → pydantic_ai-0.4.0}/tests/models/test_gemini.py

@@ -83,14 +83,18 @@ async def test_model_tools(allow_model_requests: None):
     m = GeminiModel('gemini-1.5-flash', provider=GoogleGLAProvider(api_key='via-arg'))
     tools = [
         ToolDefinition(
-            'foo',
-            'This is foo',
-            {'type': 'object', 'title': 'Foo', 'properties': {'bar': {'type': 'number', 'title': 'Bar'}}},
+            name='foo',
+            description='This is foo',
+            parameters_json_schema={
+                'type': 'object',
+                'title': 'Foo',
+                'properties': {'bar': {'type': 'number', 'title': 'Bar'}},
+            },
         ),
         ToolDefinition(
-            'apple',
-            'This is apple',
-            {
+            name='apple',
+            description='This is apple',
+            parameters_json_schema={
                 'type': 'object',
                 'properties': {
                     'banana': {'type': 'array', 'title': 'Banana', 'items': {'type': 'number', 'title': 'Bar'}}
@@ -99,9 +103,14 @@ async def test_model_tools(allow_model_requests: None):
         ),
     ]
     output_tool = ToolDefinition(
-        'result',
-        'This is the tool for the final Result',
-        {'type': 'object', 'title': 'Result', 'properties': {'spam': {'type': 'number'}}, 'required': ['spam']},
+        name='result',
+        description='This is the tool for the final Result',
+        parameters_json_schema={
+            'type': 'object',
+            'title': 'Result',
+            'properties': {'spam': {'type': 'number'}},
+            'required': ['spam'],
+        },
     )
 
     mrp = ModelRequestParameters(
@@ -148,9 +157,9 @@ async def test_model_tools(allow_model_requests: None):
 async def test_require_response_tool(allow_model_requests: None):
     m = GeminiModel('gemini-1.5-flash', provider=GoogleGLAProvider(api_key='via-arg'))
     output_tool = ToolDefinition(
-        'result',
-        'This is the tool for the final Result',
-        {'type': 'object', 'title': 'Result', 'properties': {'spam': {'type': 'number'}}},
+        name='result',
+        description='This is the tool for the final Result',
+        parameters_json_schema={'type': 'object', 'title': 'Result', 'properties': {'spam': {'type': 'number'}}},
     )
     mrp = ModelRequestParameters(
         function_tools=[],
@@ -235,9 +244,9 @@ async def test_json_def_replaced(allow_model_requests: None):
 
     m = GeminiModel('gemini-1.5-flash', provider=GoogleGLAProvider(api_key='via-arg'))
     output_tool = ToolDefinition(
-        'result',
-        'This is the tool for the final Result',
-        json_schema,
+        name='result',
+        description='This is the tool for the final Result',
+        parameters_json_schema=json_schema,
     )
     mrp = ModelRequestParameters(
         function_tools=[],
@@ -320,9 +329,9 @@ async def test_json_def_enum(allow_model_requests: None):
     )
     m = GeminiModel('gemini-1.5-flash', provider=GoogleGLAProvider(api_key='via-arg'))
     output_tool = ToolDefinition(
-        'result',
-        'This is the tool for the final Result',
-        json_schema,
+        name='result',
+        description='This is the tool for the final Result',
+        parameters_json_schema=json_schema,
     )
     mrp = ModelRequestParameters(
         function_tools=[],
@@ -368,9 +377,9 @@ async def test_json_def_replaced_any_of(allow_model_requests: None):
 
     m = GeminiModel('gemini-1.5-flash', provider=GoogleGLAProvider(api_key='via-arg'))
     output_tool = ToolDefinition(
-        'result',
-        'This is the tool for the final Result',
-        json_schema,
+        name='result',
+        description='This is the tool for the final Result',
+        parameters_json_schema=json_schema,
     )
     mrp = ModelRequestParameters(
         function_tools=[],
@@ -437,9 +446,9 @@ async def test_json_def_recursive(allow_model_requests: None):
 
     m = GeminiModel('gemini-1.5-flash', provider=GoogleGLAProvider(api_key='via-arg'))
     output_tool = ToolDefinition(
-        'result',
-        'This is the tool for the final Result',
-        json_schema,
+        name='result',
+        description='This is the tool for the final Result',
+        parameters_json_schema=json_schema,
     )
     with pytest.raises(UserError, match=r'Recursive `\$ref`s in JSON Schema are not supported by Gemini'):
         mrp = ModelRequestParameters(
@@ -476,9 +485,9 @@ async def test_json_def_date(allow_model_requests: None):
 
     m = GeminiModel('gemini-1.5-flash', provider=GoogleGLAProvider(api_key='via-arg'))
     output_tool = ToolDefinition(
-        'result',
-        'This is the tool for the final Result',
-        json_schema,
+        name='result',
+        description='This is the tool for the final Result',
+        parameters_json_schema=json_schema,
     )
     mrp = ModelRequestParameters(
         function_tools=[],

{pydantic_ai-0.3.7 → pydantic_ai-0.4.0}/tests/models/test_openai.py

@@ -1095,7 +1095,10 @@ def tool_with_tuples(x: tuple[int], y: tuple[str] = ('abc',)) -> str:
                     '$defs': {
                         'MyDefaultRecursiveDc': {
                             'properties': {
-                                'field': {'anyOf': [{'$ref': '#/$defs/MyDefaultRecursiveDc'}, {'type': 'null'}]}
+                                'field': {
+                                    'anyOf': [{'$ref': '#/$defs/MyDefaultRecursiveDc'}, {'type': 'null'}],
+                                    'default': None,
+                                }
                             },
                             'type': 'object',
                         },
@@ -1213,7 +1216,7 @@ def tool_with_tuples(x: tuple[int], y: tuple[str] = ('abc',)) -> str:
                 {
                     '$defs': {
                         'MyDefaultDc': {
-                            'properties': {'x': {'type': 'integer'}},
+                            'properties': {'x': {'default': 1, 'type': 'integer'}},
                             'type': 'object',
                         }
                     },
@@ -1253,7 +1256,7 @@ def tool_with_tuples(x: tuple[int], y: tuple[str] = ('abc',)) -> str:
                 {
                     '$defs': {
                         'MyDefaultDc': {
-                            'properties': {'x': {'type': 'integer'}},
+                            'properties': {'x': {'default': 1, 'type': 'integer'}},
                             'type': 'object',
                         }
                     },
@@ -1293,7 +1296,7 @@ def tool_with_tuples(x: tuple[int], y: tuple[str] = ('abc',)) -> str:
                 {
                     '$defs': {
                         'MyDefaultDc': {
-                            'properties': {'x': {'type': 'integer'}},
+                            'properties': {'x': {'default': 1, 'type': 'integer'}},
                             'type': 'object',
                         }
                     },
@@ -1913,9 +1916,9 @@ def test_openai_model_profile_from_provider():
 
 def test_model_profile_strict_not_supported():
     my_tool = ToolDefinition(
-        'my_tool',
-        'This is my tool',
-        {'type': 'object', 'title': 'Result', 'properties': {'spam': {'type': 'number'}}},
+        name='my_tool',
+        description='This is my tool',
+        parameters_json_schema={'type': 'object', 'title': 'Result', 'properties': {'spam': {'type': 'number'}}},
         strict=True,
     )
 

{pydantic_ai-0.3.7 → pydantic_ai-0.4.0}/tests/models/test_openai_responses.py

@@ -473,9 +473,9 @@ async def test_openai_responses_model_instructions(allow_model_requests: None, o
 
 def test_model_profile_strict_not_supported():
     my_tool = ToolDefinition(
-        'my_tool',
-        'This is my tool',
-        {'type': 'object', 'title': 'Result', 'properties': {'spam': {'type': 'number'}}},
+        name='my_tool',
+        description='This is my tool',
+        parameters_json_schema={'type': 'object', 'title': 'Result', 'properties': {'spam': {'type': 'number'}}},
         strict=True,
     )
 

{pydantic_ai-0.3.7 → pydantic_ai-0.4.0}/tests/test_direct.py

@@ -54,9 +54,7 @@ async def test_model_request_tool_call():
         'test',
         [ModelRequest.user_text_prompt('x')],
         model_request_parameters=ModelRequestParameters(
-            function_tools=[
-                ToolDefinition(name='tool_name', description='', parameters_json_schema={'type': 'object'})
-            ],
+            function_tools=[ToolDefinition(name='tool_name', parameters_json_schema={'type': 'object'})],
             allow_text_output=False,
         ),
     )

{pydantic_ai-0.3.7 → pydantic_ai-0.4.0}/tests/test_logfire.py

@@ -280,7 +280,7 @@ def test_logfire(
                         'function_tools': [
                             {
                                 'name': 'my_ret',
-                                'description': '',
+                                'description': None,
                                 'parameters_json_schema': {
                                     'additionalProperties': False,
                                     'properties': {'x': {'type': 'integer'}},

{pydantic_ai-0.3.7 → pydantic_ai-0.4.0}/tests/test_mcp.py

@@ -60,6 +60,7 @@ async def test_stdio_server():
         tools = await server.list_tools()
         assert len(tools) == snapshot(13)
         assert tools[0].name == 'celsius_to_fahrenheit'
+        assert isinstance(tools[0].description, str)
         assert tools[0].description.startswith('Convert Celsius to Fahrenheit.')
 
         # Test calling the temperature conversion tool

{pydantic_ai-0.3.7 → pydantic_ai-0.4.0}/tests/test_messages.py

@@ -125,6 +125,10 @@ def test_binary_content_document(media_type: str, format: str):
     [
         pytest.param(AudioUrl('foobar.mp3'), 'audio/mpeg', 'mp3', id='mp3'),
         pytest.param(AudioUrl('foobar.wav'), 'audio/wav', 'wav', id='wav'),
+        pytest.param(AudioUrl('foobar.oga'), 'audio/ogg', 'oga', id='oga'),
+        pytest.param(AudioUrl('foobar.flac'), 'audio/flac', 'flac', id='flac'),
+        pytest.param(AudioUrl('foobar.aiff'), 'audio/aiff', 'aiff', id='aiff'),
+        pytest.param(AudioUrl('foobar.aac'), 'audio/aac', 'aac', id='aac'),
     ],
 )
 def test_audio_url(audio_url: AudioUrl, media_type: str, format: str):

{pydantic_ai-0.3.7 → pydantic_ai-0.4.0}/tests/test_tools.py

@@ -107,11 +107,6 @@ def test_docstring_google(docstring_format: Literal['google', 'auto']):
             'strict': None,
         }
     )
-    keys = list(json_schema.keys())
-    # name should be the first key
-    assert keys[0] == 'name'
-    # description should be the second key
-    assert keys[1] == 'description'
 
 
 def sphinx_style_docstring(foo: int, /) -> str:  # pragma: no cover
@@ -444,7 +439,7 @@ def test_takes_model_and_int():
     assert json_schema == snapshot(
         {
             'name': 'takes_just_model',
-            'description': '',
+            'description': None,
             'parameters_json_schema': {
                 '$defs': {
                     'Foo': {
@@ -862,7 +857,7 @@ def test_json_schema_required_parameters():
     assert json_schema == snapshot(
         [
             {
-                'description': '',
+                'description': None,
                 'name': 'my_tool',
                 'outer_typed_dict_key': None,
                 'parameters_json_schema': {
@@ -874,7 +869,7 @@ def test_json_schema_required_parameters():
                 'strict': None,
             },
             {
-                'description': '',
+                'description': None,
                 'name': 'my_tool_plain',
                 'outer_typed_dict_key': None,
                 'parameters_json_schema': {
@@ -960,7 +955,7 @@ def test_schema_generator():
     assert json_schema == snapshot(
         [
             {
-                'description': '',
+                'description': None,
                 'name': 'my_tool_1',
                 'outer_typed_dict_key': None,
                 'parameters_json_schema': {
@@ -970,7 +965,7 @@ def test_schema_generator():
                 'strict': None,
             },
             {
-                'description': '',
+                'description': None,
                 'name': 'my_tool_2',
                 'outer_typed_dict_key': None,
                 'parameters_json_schema': {