pydantic-ai 0.1.12__tar.gz → 0.2.1__tar.gz

{pydantic_ai-0.1.12 → pydantic_ai-0.2.1}/PKG-INFO

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

{pydantic_ai-0.1.12 → pydantic_ai-0.2.1}/tests/models/test_anthropic.py

@@ -927,6 +927,22 @@ async def test_anthropic_model_instructions(allow_model_requests: None, anthropi
     )
 
 
+async def test_multiple_system_prompt_formatting(allow_model_requests: None):
+    c = completion_message([TextBlock(text='world', type='text')], AnthropicUsage(input_tokens=5, output_tokens=10))
+    mock_client = MockAnthropic().create_mock(c)
+    m = AnthropicModel('claude-3-5-haiku-latest', provider=AnthropicProvider(anthropic_client=mock_client))
+    agent = Agent(m, system_prompt='this is the system prompt')
+
+    @agent.system_prompt
+    def system_prompt() -> str:
+        return 'and this is another'
+
+    await agent.run('hello')
+    completion_kwargs = get_mock_chat_completion_kwargs(mock_client)[0]
+    assert 'system' in completion_kwargs
+    assert completion_kwargs['system'] == 'this is the system prompt\n\nand this is another'
+
+
 def anth_msg(usage: AnthropicUsage) -> AnthropicMessage:
     return AnthropicMessage(
         id='x',

{pydantic_ai-0.1.12 → pydantic_ai-0.2.1}/tests/models/test_gemini.py

@@ -164,9 +164,17 @@ async def test_require_response_tool(allow_model_requests: None):
 
 
 async def test_json_def_replaced(allow_model_requests: None):
+    class Axis(BaseModel):
+        label: str
+
+    class Chart(BaseModel):
+        x_axis: Axis
+        y_axis: Axis
+
     class Location(BaseModel):
         lat: float
         lng: float = 1.1
+        chart: Chart
 
     class Locations(BaseModel):
         locations: list[Location]
@@ -175,15 +183,28 @@ async def test_json_def_replaced(allow_model_requests: None):
     assert json_schema == snapshot(
         {
             '$defs': {
+                'Axis': {
+                    'properties': {'label': {'title': 'Label', 'type': 'string'}},
+                    'required': ['label'],
+                    'title': 'Axis',
+                    'type': 'object',
+                },
+                'Chart': {
+                    'properties': {'x_axis': {'$ref': '#/$defs/Axis'}, 'y_axis': {'$ref': '#/$defs/Axis'}},
+                    'required': ['x_axis', 'y_axis'],
+                    'title': 'Chart',
+                    'type': 'object',
+                },
                 'Location': {
                     'properties': {
                         'lat': {'title': 'Lat', 'type': 'number'},
                         'lng': {'default': 1.1, 'title': 'Lng', 'type': 'number'},
+                        'chart': {'$ref': '#/$defs/Chart'},
                     },
-                    'required': ['lat'],
+                    'required': ['lat', 'chart'],
                     'title': 'Location',
                     'type': 'object',
-                }
+                },
             },
             'properties': {'locations': {'items': {'$ref': '#/$defs/Location'}, 'title': 'Locations', 'type': 'array'}},
             'required': ['locations'],
@@ -201,20 +222,36 @@ async def test_json_def_replaced(allow_model_requests: None):
     mrp = ModelRequestParameters(function_tools=[], allow_text_output=True, output_tools=[output_tool])
     mrp = m.customize_request_parameters(mrp)
     assert m._get_tools(mrp) == snapshot(
-        _GeminiTools(
-            function_declarations=[
-                _GeminiFunction(
-                    name='result',
-                    description='This is the tool for the final Result',
-                    parameters={
+        {
+            'function_declarations': [
+                {
+                    'name': 'result',
+                    'description': 'This is the tool for the final Result',
+                    'parameters': {
                         'properties': {
                             'locations': {
                                 'items': {
                                     'properties': {
                                         'lat': {'type': 'number'},
                                         'lng': {'type': 'number'},
+                                        'chart': {
+                                            'properties': {
+                                                'x_axis': {
+                                                    'properties': {'label': {'type': 'string'}},
+                                                    'required': ['label'],
+                                                    'type': 'object',
+                                                },
+                                                'y_axis': {
+                                                    'properties': {'label': {'type': 'string'}},
+                                                    'required': ['label'],
+                                                    'type': 'object',
+                                                },
+                                            },
+                                            'required': ['x_axis', 'y_axis'],
+                                            'type': 'object',
+                                        },
                                     },
-                                    'required': ['lat'],
+                                    'required': ['lat', 'chart'],
                                     'type': 'object',
                                 },
                                 'type': 'array',
@@ -223,9 +260,9 @@ async def test_json_def_replaced(allow_model_requests: None):
                         'required': ['locations'],
                         'type': 'object',
                     },
-                )
+                }
             ]
-        )
+        }
     )
 
 

{pydantic_ai-0.1.12 → pydantic_ai-0.2.1}/tests/models/test_instrumented.py

@@ -36,7 +36,7 @@ from pydantic_ai.models.instrumented import InstrumentationSettings, Instrumente
 from pydantic_ai.settings import ModelSettings
 from pydantic_ai.usage import Usage
 
-from ..conftest import try_import
+from ..conftest import IsStr, try_import
 
 with try_import() as imports_successful:
     from logfire.testing import CaptureLogfire
@@ -796,7 +796,7 @@ def test_messages_to_otel_events_image_url(document_content: BinaryContent):
                 'event.name': 'gen_ai.user.message',
             },
             {
-                'content': ['user_prompt6', {'kind': 'binary'}],
+                'content': ['user_prompt6', {'kind': 'binary', 'content': IsStr(), 'media_type': 'application/pdf'}],
                 'role': 'user',
                 'gen_ai.message.index': 5,
                 'event.name': 'gen_ai.user.message',

pydantic_ai-0.2.1/tests/models/test_model_request_parameters.py

@@ -0,0 +1,12 @@
+from pydantic import TypeAdapter
+
+from pydantic_ai.models import ModelRequestParameters
+
+
+def test_model_request_parameters_are_serializable():
+    params = ModelRequestParameters(function_tools=[], allow_text_output=False, output_tools=[])
+    assert TypeAdapter(ModelRequestParameters).dump_python(params) == {
+        'function_tools': [],
+        'allow_text_output': False,
+        'output_tools': [],
+    }

{pydantic_ai-0.1.12 → pydantic_ai-0.2.1}/tests/test_parts_manager.py

@@ -104,7 +104,7 @@ def test_handle_tool_call_deltas():
         PartDeltaEvent(
             index=0,
             delta=ToolCallPartDelta(
-                tool_name_delta='1', args_delta=None, tool_call_id=None, part_delta_kind='tool_call'
+                tool_name_delta='1', args_delta=None, tool_call_id=IsStr(), part_delta_kind='tool_call'
             ),
             event_kind='part_delta',
         )
@@ -118,7 +118,7 @@ def test_handle_tool_call_deltas():
         PartDeltaEvent(
             index=0,
             delta=ToolCallPartDelta(
-                tool_name_delta=None, args_delta='"value1"}', tool_call_id=None, part_delta_kind='tool_call'
+                tool_name_delta=None, args_delta='"value1"}', tool_call_id=IsStr(), part_delta_kind='tool_call'
             ),
             event_kind='part_delta',
         )
@@ -196,7 +196,7 @@ def test_handle_tool_call_part():
         PartDeltaEvent(
             index=0,
             delta=ToolCallPartDelta(
-                tool_name_delta=None, args_delta='"value1"}', tool_call_id=None, part_delta_kind='tool_call'
+                tool_name_delta=None, args_delta='"value1"}', tool_call_id=IsStr(), part_delta_kind='tool_call'
             ),
             event_kind='part_delta',
         )