pydantic-ai 0.5.1__tar.gz → 0.6.0__tar.gz

{pydantic_ai-0.5.1 → pydantic_ai-0.6.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pydantic-ai
-Version: 0.5.1
+Version: 0.6.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,11 +28,11 @@ 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[ag-ui,anthropic,bedrock,cli,cohere,evals,google,groq,huggingface,mcp,mistral,openai,retries,vertexai]==0.5.1
+Requires-Dist: pydantic-ai-slim[ag-ui,anthropic,bedrock,cli,cohere,evals,google,groq,huggingface,mcp,mistral,openai,retries,vertexai]==0.6.0
 Provides-Extra: a2a
 Requires-Dist: fasta2a>=0.4.1; extra == 'a2a'
 Provides-Extra: examples
-Requires-Dist: pydantic-ai-examples==0.5.1; extra == 'examples'
+Requires-Dist: pydantic-ai-examples==0.6.0; extra == 'examples'
 Provides-Extra: logfire
 Requires-Dist: logfire>=3.11.0; extra == 'logfire'
 Description-Content-Type: text/markdown

{pydantic_ai-0.5.1 → pydantic_ai-0.6.0}/pyproject.toml

@@ -92,6 +92,7 @@ docs = [
     "mkdocs-llmstxt>=0.2.0",
     "mkdocs-material[imaging]>=9.5.45",
     "mkdocstrings-python>=1.12.2",
+    "griffe-warnings-deprecated>=1.1.0",
 ]
 docs-upload = ["algoliasearch>=4.12.0", "pydantic>=2.10.1"]
 

{pydantic_ai-0.5.1 → pydantic_ai-0.6.0}/tests/conftest.py

@@ -479,7 +479,7 @@ def model(
             return CohereModel('command-r-plus', provider=CohereProvider(api_key=co_api_key))
         elif request.param == 'gemini':
             from pydantic_ai.models.gemini import GeminiModel  # type: ignore[reportDeprecated]
-            from pydantic_ai.providers.google_gla import GoogleGLAProvider
+            from pydantic_ai.providers.google_gla import GoogleGLAProvider  # type: ignore[reportDeprecated]
 
             return GeminiModel('gemini-1.5-flash', provider=GoogleGLAProvider(api_key=gemini_api_key))  # type: ignore[reportDeprecated]
         elif request.param == 'google':

{pydantic_ai-0.5.1 → pydantic_ai-0.6.0}/tests/graph/test_graph.py

@@ -401,39 +401,6 @@ async def test_iter_next_error(mock_snapshot_id: object):
             await run.next()
 
 
-async def test_next(mock_snapshot_id: object):
-    @dataclass
-    class Foo(BaseNode):
-        async def run(self, ctx: GraphRunContext) -> Bar:
-            return Bar()
-
-    @dataclass
-    class Bar(BaseNode):
-        async def run(self, ctx: GraphRunContext) -> Foo:
-            return Foo()  # pragma: no cover
-
-    g = Graph(nodes=(Foo, Bar))
-    assert g.name is None
-    sp = FullStatePersistence()
-    with pytest.warns(DeprecationWarning, match='`next` is deprecated, use `async with graph.iter(...)'):
-        n = await g.next(Foo(), persistence=sp)  # pyright: ignore[reportDeprecated]
-    assert n == Bar()
-    assert g.name == 'g'
-    assert sp.history == snapshot(
-        [
-            NodeSnapshot(
-                state=None,
-                node=Foo(),
-                start_ts=IsNow(tz=timezone.utc),
-                duration=IsFloat(),
-                status='success',
-                id='Foo:1',
-            ),
-            NodeSnapshot(state=None, node=Bar(), id='Bar:2'),
-        ]
-    )
-
-
 async def test_deps(mock_snapshot_id: object):
     @dataclass
     class Deps:

{pydantic_ai-0.5.1 → pydantic_ai-0.6.0}/tests/models/test_gemini.py

@@ -61,7 +61,11 @@ from pydantic_ai.tools import ToolDefinition
 
 from ..conftest import ClientWithHandler, IsDatetime, IsInstance, IsNow, IsStr, TestEnv, try_import
 
-pytestmark = [pytest.mark.anyio, pytest.mark.filterwarnings('ignore:Use `GoogleModel` instead.:DeprecationWarning')]
+pytestmark = [
+    pytest.mark.anyio,
+    pytest.mark.filterwarnings('ignore:Use `GoogleModel` instead.:DeprecationWarning'),
+    pytest.mark.filterwarnings('ignore:`GoogleGLAProvider` is deprecated.:DeprecationWarning'),
+]
 
 
 async def test_model_simple(allow_model_requests: None):

{pydantic_ai-0.5.1 → pydantic_ai-0.6.0}/tests/models/test_gemini_vertex.py

@@ -34,6 +34,7 @@ pytestmark = [
     pytest.mark.skipif(not imports_successful(), reason='google-auth not installed'),
     pytest.mark.anyio,
     pytest.mark.filterwarnings('ignore:Use `GoogleModel` instead.:DeprecationWarning'),
+    pytest.mark.filterwarnings('ignore:`GoogleVertexProvider` is deprecated.:DeprecationWarning'),
 ]
 
 

{pydantic_ai-0.5.1 → pydantic_ai-0.6.0}/tests/providers/test_google_gla.py

@@ -1,3 +1,4 @@
+# pyright: reportDeprecated=false
 import re
 
 import pytest
@@ -7,6 +8,10 @@ from pydantic_ai.providers.google_gla import GoogleGLAProvider
 
 from ..conftest import TestEnv
 
+pytestmark = [
+    pytest.mark.filterwarnings('ignore:`GoogleGLAProvider` is deprecated.:DeprecationWarning'),
+]
+
 
 def test_api_key_arg(env: TestEnv):
     env.set('GEMINI_API_KEY', 'via-env-var')

{pydantic_ai-0.5.1 → pydantic_ai-0.6.0}/tests/providers/test_google_vertex.py

@@ -1,3 +1,4 @@
+# pyright: reportDeprecated=false
 from __future__ import annotations as _annotations
 
 import json
@@ -13,7 +14,7 @@ from inline_snapshot import snapshot
 from pytest_mock import MockerFixture
 
 from pydantic_ai.agent import Agent
-from pydantic_ai.models.gemini import GeminiModel  # type: ignore[reportDeprecated]
+from pydantic_ai.models.gemini import GeminiModel
 
 from ..conftest import try_import
 
@@ -26,6 +27,7 @@ pytestmark = [
     pytest.mark.skipif(not imports_successful(), reason='google-genai not installed'),
     pytest.mark.anyio(),
     pytest.mark.filterwarnings('ignore:Use `GoogleModel` instead.:DeprecationWarning'),
+    pytest.mark.filterwarnings('ignore:`GoogleVertexProvider` is deprecated.:DeprecationWarning'),
 ]
 
 
@@ -168,7 +170,7 @@ def vertex_provider_auth(mocker: MockerFixture) -> None:  # pragma: lax no cover
 )
 @pytest.mark.vcr()
 async def test_vertexai_provider(allow_model_requests: None):  # pragma: lax no cover
-    m = GeminiModel('gemini-2.0-flash', provider='google-vertex')  # type: ignore[reportDeprecated]
+    m = GeminiModel('gemini-2.0-flash', provider='google-vertex')
     agent = Agent(m)
 
     result = await agent.run('What is the capital of France?')

{pydantic_ai-0.5.1 → pydantic_ai-0.6.0}/tests/providers/test_provider_names.py

@@ -20,8 +20,8 @@ with try_import() as imports_successful:
     from pydantic_ai.providers.deepseek import DeepSeekProvider
     from pydantic_ai.providers.fireworks import FireworksProvider
     from pydantic_ai.providers.github import GitHubProvider
-    from pydantic_ai.providers.google_gla import GoogleGLAProvider
-    from pydantic_ai.providers.google_vertex import GoogleVertexProvider
+    from pydantic_ai.providers.google_gla import GoogleGLAProvider  # type: ignore[reportDeprecated]
+    from pydantic_ai.providers.google_vertex import GoogleVertexProvider  # type: ignore[reportDeprecated]
     from pydantic_ai.providers.grok import GrokProvider
     from pydantic_ai.providers.groq import GroqProvider
     from pydantic_ai.providers.heroku import HerokuProvider
@@ -40,8 +40,8 @@ with try_import() as imports_successful:
         ('vercel', VercelProvider, 'VERCEL_AI_GATEWAY_API_KEY'),
         ('openai', OpenAIProvider, 'OPENAI_API_KEY'),
         ('azure', AzureProvider, 'AZURE_OPENAI'),
-        ('google-vertex', GoogleVertexProvider, None),
-        ('google-gla', GoogleGLAProvider, 'GEMINI_API_KEY'),
+        ('google-vertex', GoogleVertexProvider, None),  # type: ignore[reportDeprecated]
+        ('google-gla', GoogleGLAProvider, 'GEMINI_API_KEY'),  # type: ignore[reportDeprecated]
         ('groq', GroqProvider, 'GROQ_API_KEY'),
         ('mistral', MistralProvider, 'MISTRAL_API_KEY'),
         ('grok', GrokProvider, 'GROK_API_KEY'),
@@ -55,7 +55,11 @@ with try_import() as imports_successful:
 if not imports_successful():
     test_infer_provider_params = []  # pragma: lax no cover
 
-pytestmark = pytest.mark.skipif(not imports_successful(), reason='need to install all extra packages')
+pytestmark = [
+    pytest.mark.skipif(not imports_successful(), reason='need to install all extra packages'),
+    pytest.mark.filterwarnings('ignore:`GoogleGLAProvider` is deprecated:DeprecationWarning'),
+    pytest.mark.filterwarnings('ignore:`GoogleVertexProvider` is deprecated:DeprecationWarning'),
+]
 
 
 @pytest.fixture(autouse=True)

{pydantic_ai-0.5.1 → pydantic_ai-0.6.0}/tests/test_agent.py

@@ -2617,13 +2617,6 @@ def test_heterogeneous_responses_non_streaming() -> None:
     )
 
 
-def test_last_run_messages() -> None:
-    agent = Agent('test')
-
-    with pytest.raises(AttributeError, match='The `last_run_messages` attribute has been removed,'):
-        agent.last_run_messages  # pyright: ignore[reportDeprecated]
-
-
 def test_nested_capture_run_messages() -> None:
     agent = Agent('test')
 
@@ -3638,46 +3631,13 @@ def test_deprecated_kwargs_still_work():
     """Test that valid deprecated kwargs still work with warnings."""
     import warnings
 
-    with warnings.catch_warnings(record=True) as w:
-        warnings.simplefilter('always')
-
-        agent = Agent('test', result_type=str)  # type: ignore[call-arg]
-        assert len(w) == 1
-        assert issubclass(w[0].category, DeprecationWarning)
-        assert '`result_type` is deprecated' in str(w[0].message)
-        assert agent.output_type is str
-
-    with warnings.catch_warnings(record=True) as w:
-        warnings.simplefilter('always')
-
-        agent = Agent('test', result_tool_name='test_tool')  # type: ignore[call-arg]
-        assert len(w) == 1
-        assert issubclass(w[0].category, DeprecationWarning)
-        assert '`result_tool_name` is deprecated' in str(w[0].message)
-
-    with warnings.catch_warnings(record=True) as w:
-        warnings.simplefilter('always')
-
-        agent = Agent('test', result_tool_description='test description')  # type: ignore[call-arg]
-        assert len(w) == 1
-        assert issubclass(w[0].category, DeprecationWarning)
-        assert '`result_tool_description` is deprecated' in str(w[0].message)
-
-    with warnings.catch_warnings(record=True) as w:
-        warnings.simplefilter('always')
-
-        agent = Agent('test', result_retries=3)  # type: ignore[call-arg]
-        assert len(w) == 1
-        assert issubclass(w[0].category, DeprecationWarning)
-        assert '`result_retries` is deprecated' in str(w[0].message)
-
     try:
         from pydantic_ai.mcp import MCPServerStdio
 
         with warnings.catch_warnings(record=True) as w:
             warnings.simplefilter('always')
 
-            agent = Agent('test', mcp_servers=[MCPServerStdio('python', ['-m', 'tests.mcp_server'])])  # type: ignore[call-arg]
+            Agent('test', mcp_servers=[MCPServerStdio('python', ['-m', 'tests.mcp_server'])])  # type: ignore[call-arg]
             assert len(w) == 1
             assert issubclass(w[0].category, DeprecationWarning)
             assert '`mcp_servers` is deprecated' in str(w[0].message)
@@ -3685,21 +3645,6 @@ def test_deprecated_kwargs_still_work():
         pass
 
 
-def test_deprecated_kwargs_mixed_valid_invalid():
-    """Test that mix of valid deprecated and invalid kwargs raises error for invalid ones."""
-    import warnings
-
-    with pytest.raises(UserError, match='Unknown keyword arguments: `usage_limits`'):
-        with warnings.catch_warnings():
-            warnings.simplefilter('ignore', DeprecationWarning)  # Ignore the deprecation warning for result_type
-            Agent('test', result_type=str, usage_limits='invalid')  # type: ignore[call-arg]
-
-    with pytest.raises(UserError, match='Unknown keyword arguments: `foo`, `bar`'):
-        with warnings.catch_warnings():
-            warnings.simplefilter('ignore', DeprecationWarning)  # Ignore the deprecation warning for result_tool_name
-            Agent('test', result_tool_name='test', foo='value1', bar='value2')  # type: ignore[call-arg]
-
-
 def test_override_toolsets():
     foo_toolset = FunctionToolset()
 

{pydantic_ai-0.5.1 → pydantic_ai-0.6.0}/tests/test_examples.py

@@ -45,9 +45,7 @@ from .conftest import ClientWithHandler, TestEnv, try_import
 
 try:
     from pydantic_ai.providers.google import GoogleProvider
-    from pydantic_ai.providers.google_vertex import GoogleVertexProvider
 except ImportError:  # pragma: lax no cover
-    GoogleVertexProvider = None
     GoogleProvider = None
 
 
@@ -63,10 +61,7 @@ with try_import() as imports_successful:
 
 pytestmark = [
     pytest.mark.skipif(not imports_successful(), reason='extras not installed'),
-    pytest.mark.skipif(
-        GoogleVertexProvider is None or logfire is None or GoogleProvider is None,
-        reason='google-auth or logfire or google-provider not installed',
-    ),
+    pytest.mark.skipif(logfire is None or GoogleProvider is None, reason='logfire or google-provider not installed'),
 ]
 code_examples: dict[str, CodeExample] = {}
 

{pydantic_ai-0.5.1 → pydantic_ai-0.6.0}/tests/test_streaming.py

@@ -926,7 +926,7 @@ async def test_stream_iter_structured_validator() -> None:
     class NotOutputType(BaseModel):
         not_value: str
 
-    agent = Agent[None, Union[OutputType, NotOutputType]]('test', output_type=Union[OutputType, NotOutputType])  # pyright: ignore[reportArgumentType]
+    agent = Agent[None, Union[OutputType, NotOutputType]]('test', output_type=Union[OutputType, NotOutputType])
 
     @agent.output_validator
     def output_validator(data: OutputType | NotOutputType) -> OutputType | NotOutputType:

{pydantic_ai-0.5.1 → pydantic_ai-0.6.0}/tests/typed_agent.py

@@ -1,4 +1,5 @@
 """This file is used to test static typing, it's analyzed with pyright and mypy."""
+# pyright: reportUnnecessaryTypeIgnoreComment=false
 
 import re
 from collections.abc import Awaitable