pydantic-ai 0.2.5__tar.gz → 0.2.6__tar.gz

{pydantic_ai-0.2.5 → pydantic_ai-0.2.6}/PKG-INFO

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

pydantic_ai-0.2.6/tests/fasta2a/test_applications.py

@@ -0,0 +1,45 @@
+from __future__ import annotations as _annotations
+
+from contextlib import asynccontextmanager
+
+import httpx
+import pytest
+from asgi_lifespan import LifespanManager
+from inline_snapshot import snapshot
+
+from ..conftest import try_import
+
+with try_import() as imports_successful:
+    from fasta2a.applications import FastA2A
+    from fasta2a.broker import InMemoryBroker
+    from fasta2a.storage import InMemoryStorage
+
+
+pytestmark = [pytest.mark.anyio, pytest.mark.skipif(not imports_successful(), reason='fasta2a not installed')]
+
+
+@asynccontextmanager
+async def create_test_client(app: FastA2A):
+    async with LifespanManager(app=app) as manager:
+        transport = httpx.ASGITransport(app=manager.app)
+        async with httpx.AsyncClient(transport=transport, base_url='http://testclient') as client:
+            yield client
+
+
+async def test_agent_card():
+    app = FastA2A(storage=InMemoryStorage(), broker=InMemoryBroker())
+    async with create_test_client(app) as client:
+        response = await client.get('/.well-known/agent.json')
+        assert response.status_code == 200
+        assert response.json() == snapshot(
+            {
+                'name': 'Agent',
+                'url': 'http://localhost:8000',
+                'version': '1.0.0',
+                'skills': [],
+                'defaultInputModes': ['application/json'],
+                'defaultOutputModes': ['application/json'],
+                'capabilities': {'streaming': False, 'pushNotifications': False, 'stateTransitionHistory': False},
+                'authentication': {'schemes': []},
+            }
+        )

{pydantic_ai-0.2.5 → pydantic_ai-0.2.6}/tests/test_agent.py

@@ -1968,3 +1968,10 @@ def test_agent_run_result_serialization() -> None:
     # Check that we can load the data back
     deserialized_result = adapter.validate_json(serialized_data)
     assert deserialized_result == result
+
+
+def test_agent_repr() -> None:
+    agent = Agent()
+    assert repr(agent) == snapshot(
+        "Agent(model=None, name=None, end_strategy='early', model_settings=None, output_type=<class 'str'>, instrument=None)"
+    )

{pydantic_ai-0.2.5 → pydantic_ai-0.2.6}/tests/test_cli.py

@@ -224,27 +224,21 @@ def test_code_theme_unset(mocker: MockerFixture, env: TestEnv):
     env.set('OPENAI_API_KEY', 'test')
     mock_run_chat = mocker.patch('pydantic_ai._cli.run_chat')
     cli([])
-    mock_run_chat.assert_awaited_once_with(
-        IsInstance(PromptSession), True, IsInstance(Agent), IsInstance(Console), 'monokai', 'pai'
-    )
+    mock_run_chat.assert_awaited_once_with(True, IsInstance(Agent), IsInstance(Console), 'monokai', 'pai')
 
 
 def test_code_theme_light(mocker: MockerFixture, env: TestEnv):
     env.set('OPENAI_API_KEY', 'test')
     mock_run_chat = mocker.patch('pydantic_ai._cli.run_chat')
     cli(['--code-theme=light'])
-    mock_run_chat.assert_awaited_once_with(
-        IsInstance(PromptSession), True, IsInstance(Agent), IsInstance(Console), 'default', 'pai'
-    )
+    mock_run_chat.assert_awaited_once_with(True, IsInstance(Agent), IsInstance(Console), 'default', 'pai')
 
 
 def test_code_theme_dark(mocker: MockerFixture, env: TestEnv):
     env.set('OPENAI_API_KEY', 'test')
     mock_run_chat = mocker.patch('pydantic_ai._cli.run_chat')
     cli(['--code-theme=dark'])
-    mock_run_chat.assert_awaited_once_with(
-        IsInstance(PromptSession), True, IsInstance(Agent), IsInstance(Console), 'monokai', 'pai'
-    )
+    mock_run_chat.assert_awaited_once_with(True, IsInstance(Agent), IsInstance(Console), 'monokai', 'pai')
 
 
 def test_agent_to_cli_sync(mocker: MockerFixture, env: TestEnv):
@@ -252,7 +246,6 @@ def test_agent_to_cli_sync(mocker: MockerFixture, env: TestEnv):
     mock_run_chat = mocker.patch('pydantic_ai._cli.run_chat')
     cli_agent.to_cli_sync()
     mock_run_chat.assert_awaited_once_with(
-        session=IsInstance(PromptSession),
         stream=True,
         agent=IsInstance(Agent),
         console=IsInstance(Console),
@@ -268,7 +261,6 @@ async def test_agent_to_cli_async(mocker: MockerFixture, env: TestEnv):
     mock_run_chat = mocker.patch('pydantic_ai._cli.run_chat')
     await cli_agent.to_cli()
     mock_run_chat.assert_awaited_once_with(
-        session=IsInstance(PromptSession),
         stream=True,
         agent=IsInstance(Agent),
         console=IsInstance(Console),

{pydantic_ai-0.2.5 → pydantic_ai-0.2.6}/tests/test_tools.py

@@ -1,5 +1,5 @@
 import json
-from dataclasses import dataclass
+from dataclasses import dataclass, replace
 from typing import Annotated, Any, Callable, Literal, Union
 
 import pydantic_core
@@ -954,3 +954,37 @@ def test_tool_parameters_with_attribute_docstrings():
             'strict': None,
         }
     )
+
+
+def test_dynamic_tools_agent_wide():
+    async def prepare_tool_defs(
+        ctx: RunContext[int], tool_defs: list[ToolDefinition]
+    ) -> Union[list[ToolDefinition], None]:
+        if ctx.deps == 42:
+            return []
+        elif ctx.deps == 43:
+            return None
+        elif ctx.deps == 21:
+            return [replace(tool_def, strict=True) for tool_def in tool_defs]
+        return tool_defs
+
+    agent = Agent('test', deps_type=int, prepare_tools=prepare_tool_defs)
+
+    @agent.tool
+    def foobar(ctx: RunContext[int], x: int, y: str) -> str:
+        return f'{ctx.deps} {x} {y}'
+
+    result = agent.run_sync('', deps=42)
+    assert result.output == snapshot('success (no tool calls)')
+
+    result = agent.run_sync('', deps=43)
+    assert result.output == snapshot('success (no tool calls)')
+
+    with agent.override(model=FunctionModel(get_json_schema)):
+        result = agent.run_sync('', deps=21)
+        json_schema = json.loads(result.output)
+        assert agent._function_tools['foobar'].strict is None
+        assert json_schema['strict'] is True
+
+    result = agent.run_sync('', deps=1)
+    assert result.output == snapshot('{"foobar":"1 0 a"}')