pydantic-ai 0.1.1__tar.gz → 0.1.2__tar.gz

{pydantic_ai-0.1.1 → pydantic_ai-0.1.2}/PKG-INFO

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

{pydantic_ai-0.1.1 → pydantic_ai-0.1.2}/tests/models/test_openai.py

@@ -4,6 +4,7 @@ import json
 from collections.abc import Sequence
 from dataclasses import dataclass, field
 from datetime import datetime, timezone
+from enum import Enum
 from functools import cached_property
 from typing import Annotated, Any, Callable, Literal, Union, cast
 
@@ -730,9 +731,15 @@ class MyDefaultDc:
     x: int = 1
 
 
+class MyEnum(Enum):
+    a = 'a'
+    b = 'b'
+
+
 @dataclass
 class MyRecursiveDc:
     field: MyRecursiveDc | None
+    my_enum: MyEnum = Field(description='my enum')
 
 
 @dataclass
@@ -826,9 +833,13 @@ def tool_with_tuples(x: tuple[int], y: tuple[str] = ('abc',)) -> str:
                             },
                             'type': 'object',
                         },
+                        'MyEnum': {'enum': ['a', 'b'], 'type': 'string'},
                         'MyRecursiveDc': {
-                            'properties': {'field': {'anyOf': [{'$ref': '#/$defs/MyRecursiveDc'}, {'type': 'null'}]}},
-                            'required': ['field'],
+                            'properties': {
+                                'field': {'anyOf': [{'$ref': '#/$defs/MyRecursiveDc'}, {'type': 'null'}]},
+                                'my_enum': {'description': 'my enum', 'anyOf': [{'$ref': '#/$defs/MyEnum'}]},
+                            },
+                            'required': ['field', 'my_enum'],
                             'type': 'object',
                         },
                     },
@@ -857,11 +868,15 @@ def tool_with_tuples(x: tuple[int], y: tuple[str] = ('abc',)) -> str:
                             'additionalProperties': False,
                             'required': ['field'],
                         },
+                        'MyEnum': {'enum': ['a', 'b'], 'type': 'string'},
                         'MyRecursiveDc': {
-                            'properties': {'field': {'anyOf': [{'$ref': '#/$defs/MyRecursiveDc'}, {'type': 'null'}]}},
+                            'properties': {
+                                'field': {'anyOf': [{'$ref': '#/$defs/MyRecursiveDc'}, {'type': 'null'}]},
+                                'my_enum': {'description': 'my enum', 'anyOf': [{'$ref': '#/$defs/MyEnum'}]},
+                            },
                             'type': 'object',
                             'additionalProperties': False,
-                            'required': ['field'],
+                            'required': ['field', 'my_enum'],
                         },
                     },
                     'additionalProperties': False,
@@ -998,7 +1013,7 @@ def tool_with_tuples(x: tuple[int], y: tuple[str] = ('abc',)) -> str:
                         }
                     },
                     'additionalProperties': False,
-                    'properties': {'x': {'oneOf': [{'type': 'integer'}, {'$ref': '#/$defs/MyDefaultDc'}]}},
+                    'properties': {'x': {'anyOf': [{'type': 'integer'}, {'$ref': '#/$defs/MyDefaultDc'}]}},
                     'required': ['x'],
                     'type': 'object',
                 }
@@ -1079,12 +1094,15 @@ def tool_with_tuples(x: tuple[int], y: tuple[str] = ('abc',)) -> str:
                 {
                     'additionalProperties': False,
                     'properties': {
-                        'x': {'maxItems': 1, 'minItems': 1, 'prefixItems': [{'type': 'integer'}], 'type': 'array'},
+                        'x': {
+                            'prefixItems': [{'type': 'integer'}],
+                            'type': 'array',
+                            'description': 'minItems=1, maxItems=1',
+                        },
                         'y': {
-                            'maxItems': 1,
-                            'minItems': 1,
                             'prefixItems': [{'type': 'string'}],
                             'type': 'array',
+                            'description': 'minItems=1, maxItems=1',
                         },
                     },
                     'required': ['x', 'y'],
@@ -1160,28 +1178,46 @@ def test_strict_schema():
                 'MyModel': {
                     'additionalProperties': False,
                     'properties': {
-                        'my_discriminated_union': {'oneOf': [{'$ref': '#/$defs/Apple'}, {'$ref': '#/$defs/Banana'}]},
+                        'my_discriminated_union': {'anyOf': [{'$ref': '#/$defs/Apple'}, {'$ref': '#/$defs/Banana'}]},
                         'my_list': {'items': {'type': 'number'}, 'type': 'array'},
                         'my_patterns': {
                             'additionalProperties': False,
-                            'patternProperties': {'^my-pattern$': {'type': 'string'}},
+                            'description': "patternProperties={'^my-pattern$': {'type': 'string'}}",
                             'type': 'object',
                             'properties': {},
                             'required': [],
                         },
-                        'my_recursive': {'anyOf': [{'$ref': '#/$defs/MyModel'}, {'type': 'null'}]},
+                        'my_recursive': {'anyOf': [{'$ref': '#'}, {'type': 'null'}]},
                         'my_tuple': {
-                            'maxItems': 1,
-                            'minItems': 1,
                             'prefixItems': [{'type': 'integer'}],
                             'type': 'array',
+                            'description': 'minItems=1, maxItems=1',
                         },
                     },
                     'required': ['my_recursive', 'my_patterns', 'my_tuple', 'my_list', 'my_discriminated_union'],
                     'type': 'object',
                 },
             },
-            '$ref': '#/$defs/MyModel',
+            'properties': {
+                'my_recursive': {'anyOf': [{'$ref': '#'}, {'type': 'null'}]},
+                'my_patterns': {
+                    'type': 'object',
+                    'description': "patternProperties={'^my-pattern$': {'type': 'string'}}",
+                    'additionalProperties': False,
+                    'properties': {},
+                    'required': [],
+                },
+                'my_tuple': {
+                    'prefixItems': [{'type': 'integer'}],
+                    'type': 'array',
+                    'description': 'minItems=1, maxItems=1',
+                },
+                'my_list': {'items': {'type': 'number'}, 'type': 'array'},
+                'my_discriminated_union': {'anyOf': [{'$ref': '#/$defs/Apple'}, {'$ref': '#/$defs/Banana'}]},
+            },
+            'required': ['my_recursive', 'my_patterns', 'my_tuple', 'my_list', 'my_discriminated_union'],
+            'type': 'object',
+            'additionalProperties': False,
         }
     )
 

{pydantic_ai-0.1.1 → pydantic_ai-0.1.2}/tests/test_agent.py

@@ -15,6 +15,7 @@ from pydantic_ai import Agent, ModelRetry, RunContext, UnexpectedModelBehavior,
 from pydantic_ai.messages import (
     BinaryContent,
     ModelMessage,
+    ModelMessagesTypeAdapter,
     ModelRequest,
     ModelResponse,
     ModelResponsePart,
@@ -1675,8 +1676,11 @@ def test_custom_output_type_invalid() -> None:
 def test_binary_content_all_messages_json():
     agent = Agent('test')
 
-    result = agent.run_sync(['Hello', BinaryContent(data=b'Hello', media_type='text/plain')])
-    assert json.loads(result.all_messages_json()) == snapshot(
+    content = BinaryContent(data=b'Hello', media_type='text/plain')
+    result = agent.run_sync(['Hello', content])
+
+    serialized = result.all_messages_json()
+    assert json.loads(serialized) == snapshot(
         [
             {
                 'parts': [
@@ -1698,6 +1702,10 @@ def test_binary_content_all_messages_json():
         ]
     )
 
+    # We also need to be able to round trip the serialized messages.
+    messages = ModelMessagesTypeAdapter.validate_json(serialized)
+    assert messages == result.all_messages()
+
 
 def test_instructions_raise_error_when_system_prompt_is_set():
     agent = Agent('test', instructions='An instructions!')

{pydantic_ai-0.1.1 → pydantic_ai-0.1.2}/tests/test_mcp.py

@@ -1,5 +1,7 @@
 """Tests for the MCP (Model Context Protocol) server implementation."""
 
+from pathlib import Path
+
 import pytest
 from dirty_equals import IsInstance
 from inline_snapshot import snapshot
@@ -38,6 +40,14 @@ async def test_stdio_server():
         assert result.content == snapshot([TextContent(type='text', text='32.0')])
 
 
+async def test_stdio_server_with_cwd():
+    test_dir = Path(__file__).parent
+    server = MCPServerStdio('python', ['mcp_server.py'], cwd=test_dir)
+    async with server:
+        tools = await server.list_tools()
+        assert len(tools) == 1
+
+
 def test_sse_server():
     sse_server = MCPServerHTTP(url='http://localhost:8000/sse')
     assert sse_server.url == 'http://localhost:8000/sse'