pydantic-ai 0.3.5__tar.gz → 0.3.6__tar.gz

{pydantic_ai-0.3.5 → pydantic_ai-0.3.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pydantic-ai
-Version: 0.3.5
+Version: 0.3.6
 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.5
+Requires-Dist: pydantic-ai-slim[a2a,anthropic,bedrock,cli,cohere,evals,google,groq,mcp,mistral,openai,vertexai]==0.3.6
 Provides-Extra: examples
-Requires-Dist: pydantic-ai-examples==0.3.5; extra == 'examples'
+Requires-Dist: pydantic-ai-examples==0.3.6; extra == 'examples'
 Provides-Extra: logfire
 Requires-Dist: logfire>=3.11.0; extra == 'logfire'
 Description-Content-Type: text/markdown

{pydantic_ai-0.3.5 → pydantic_ai-0.3.6}/tests/models/test_bedrock.py

@@ -413,8 +413,7 @@ async def test_bedrock_model_iter_stream(allow_model_requests: None, bedrock_pro
                     content='30°C',
                     tool_call_id='tooluse_lAG_zP8QRHmSYOwZzzaCqA',
                     timestamp=IsDatetime(),
-                ),
-                tool_call_id='tooluse_lAG_zP8QRHmSYOwZzzaCqA',
+                )
             ),
             PartStartEvent(index=0, part=TextPart(content='The')),
             FinalResultEvent(tool_name=None, tool_call_id=None),

{pydantic_ai-0.3.5 → pydantic_ai-0.3.6}/tests/models/test_google.py

@@ -375,8 +375,7 @@ async def test_google_model_iter_stream(allow_model_requests: None, google_provi
             FunctionToolResultEvent(
                 result=ToolReturnPart(
                     tool_name='get_capital', content='Paris', tool_call_id=IsStr(), timestamp=IsDatetime()
-                ),
-                tool_call_id=IsStr(),
+                )
             ),
             PartStartEvent(
                 index=0,
@@ -386,8 +385,7 @@ async def test_google_model_iter_stream(allow_model_requests: None, google_provi
             FunctionToolResultEvent(
                 result=ToolReturnPart(
                     tool_name='get_temperature', content='30°C', tool_call_id=IsStr(), timestamp=IsDatetime()
-                ),
-                tool_call_id=IsStr(),
+                )
             ),
             PartStartEvent(index=0, part=TextPart(content='The temperature in Paris')),
             FinalResultEvent(tool_name=None, tool_call_id=None),

{pydantic_ai-0.3.5 → pydantic_ai-0.3.6}/tests/models/test_instrumented.py

@@ -240,6 +240,7 @@ Fix the errors and try again.\
             {
                 'body': {
                     'content': """\
+Validation feedback:
 retry_prompt2
 
 Fix the errors and try again.\
@@ -619,6 +620,7 @@ Fix the errors and try again.\
                                 {
                                     'event.name': 'gen_ai.user.message',
                                     'content': """\
+Validation feedback:
 retry_prompt2
 
 Fix the errors and try again.\

{pydantic_ai-0.3.5 → pydantic_ai-0.3.6}/tests/test_agent.py

@@ -301,7 +301,7 @@ def test_plain_response_then_tuple():
                 parts=[
                     ToolCallPart(tool_name='final_result', args='{"response": ["foo", "bar"]}', tool_call_id=IsStr())
                 ],
-                usage=Usage(requests=1, request_tokens=72, response_tokens=8, total_tokens=80),
+                usage=Usage(requests=1, request_tokens=74, response_tokens=8, total_tokens=82),
                 model_name='function:return_tuple:',
                 timestamp=IsNow(tz=timezone.utc),
             ),
@@ -930,7 +930,7 @@ def test_output_type_text_output_function_with_retry():
             ),
             ModelResponse(
                 parts=[TextPart(content='Mexico City')],
-                usage=Usage(requests=1, request_tokens=68, response_tokens=5, total_tokens=73),
+                usage=Usage(requests=1, request_tokens=70, response_tokens=5, total_tokens=75),
                 model_name='function:call_tool:',
                 timestamp=IsDatetime(),
             ),
@@ -1584,7 +1584,7 @@ Don't include any text or Markdown fencing before or after.\
             ),
             ModelResponse(
                 parts=[TextPart(content='{"city": "Mexico City"}')],
-                usage=Usage(requests=1, request_tokens=68, response_tokens=11, total_tokens=79),
+                usage=Usage(requests=1, request_tokens=70, response_tokens=11, total_tokens=81),
                 model_name='function:call_tool:',
                 timestamp=IsDatetime(),
             ),

{pydantic_ai-0.3.5 → pydantic_ai-0.3.6}/tests/test_logfire.py

@@ -554,6 +554,46 @@ def test_include_tool_args_span_attributes(
     ]
 
     if include_content:
-        assert tool_attributes['tool_arguments'] == snapshot('{"x":42,"y":42}')
+        assert tool_attributes == snapshot(
+            {
+                'gen_ai.tool.name': 'add_numbers',
+                'gen_ai.tool.call.id': IsStr(),
+                'tool_arguments': '{"x":42,"y":42}',
+                'tool_response': '84',
+                'logfire.msg': 'running tool: add_numbers',
+                'logfire.json_schema': IsJson(
+                    snapshot(
+                        {
+                            'type': 'object',
+                            'properties': {
+                                'tool_arguments': {'type': 'object'},
+                                'tool_response': {'type': 'object'},
+                                'gen_ai.tool.name': {},
+                                'gen_ai.tool.call.id': {},
+                            },
+                        }
+                    )
+                ),
+                'logfire.span_type': 'span',
+            }
+        )
     else:
-        assert 'tool_arguments' not in tool_attributes
+        assert tool_attributes == snapshot(
+            {
+                'gen_ai.tool.name': 'add_numbers',
+                'gen_ai.tool.call.id': IsStr(),
+                'logfire.msg': 'running tool: add_numbers',
+                'logfire.json_schema': IsJson(
+                    snapshot(
+                        {
+                            'type': 'object',
+                            'properties': {
+                                'gen_ai.tool.name': {},
+                                'gen_ai.tool.call.id': {},
+                            },
+                        }
+                    )
+                ),
+                'logfire.span_type': 'span',
+            }
+        )

{pydantic_ai-0.3.5 → pydantic_ai-0.3.6}/tests/test_streaming.py

@@ -988,8 +988,7 @@ async def test_unknown_tool_call_events():
                     tool_name='unknown_tool',
                     tool_call_id=IsStr(),
                     timestamp=IsNow(tz=timezone.utc),
-                ),
-                tool_call_id=IsStr(),
+                )
             ),
             FunctionToolCallEvent(
                 part=ToolCallPart(tool_name='known_tool', args={'x': 5}, tool_call_id=IsStr()),
@@ -1000,8 +999,7 @@ async def test_unknown_tool_call_events():
                     content=10,
                     tool_call_id=IsStr(),
                     timestamp=IsNow(tz=timezone.utc),
-                ),
-                tool_call_id=IsStr(),
+                )
             ),
             FunctionToolCallEvent(
                 part=ToolCallPart(
@@ -1052,8 +1050,7 @@ async def test_output_tool_validation_failure_events():
                     content='Output tool not used - result failed validation.',
                     tool_call_id=IsStr(),
                     timestamp=IsNow(tz=timezone.utc),
-                ),
-                tool_call_id=IsStr(),
+                )
             ),
         ]
     )
@@ -1080,3 +1077,20 @@ async def test_stream_prompted_output():
             ]
         )
         assert result.is_complete
+
+
+def test_function_tool_event_tool_call_id_properties():
+    """Ensure that the `tool_call_id` property on function tool events mirrors the underlying part's ID."""
+    # Prepare a ToolCallPart with a fixed ID
+    call_part = ToolCallPart(tool_name='sample_tool', args={'a': 1}, tool_call_id='call_id_123')
+    call_event = FunctionToolCallEvent(part=call_part)
+
+    # The event should expose the same `tool_call_id` as the part
+    assert call_event.tool_call_id == call_part.tool_call_id == 'call_id_123'
+
+    # Prepare a ToolReturnPart with a fixed ID
+    return_part = ToolReturnPart(tool_name='sample_tool', content='ok', tool_call_id='return_id_456')
+    result_event = FunctionToolResultEvent(result=return_part)
+
+    # The event should expose the same `tool_call_id` as the result part
+    assert result_event.tool_call_id == return_part.tool_call_id == 'return_id_456'