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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pydantic-ai
-Version: 0.5.1
+Version: 0.6.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,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.1
 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.1; 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.1}/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.1}/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.1}/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.6.1/tests/models/cassettes/test_bedrock/test_bedrock_anthropic_tool_with_thinking.yaml

@@ -0,0 +1,116 @@
+interactions:
+- request:
+    body: '{"messages": [{"role": "user", "content": [{"text": "What is the largest city in the user country?"}]}], "system":
+      [], "inferenceConfig": {}, "toolConfig": {"tools": [{"toolSpec": {"name": "get_user_country", "inputSchema": {"json":
+      {"additionalProperties": false, "properties": {}, "type": "object"}}}}]}, "additionalModelRequestFields": {"thinking":
+      {"type": "enabled", "budget_tokens": 1024}}}'
+    headers:
+      amz-sdk-invocation-id:
+      - !!binary |
+        MmM5YzRmZDctMmNlZS00Yzk2LWIwZWMtZjMxN2NkZDEwYmM5
+      amz-sdk-request:
+      - !!binary |
+        YXR0ZW1wdD0x
+      content-length:
+      - '396'
+      content-type:
+      - !!binary |
+        YXBwbGljYXRpb24vanNvbg==
+    method: POST
+    uri: https://bedrock-runtime.us-east-1.amazonaws.com/model/us.anthropic.claude-3-7-sonnet-20250219-v1%3A0/converse
+  response:
+    headers:
+      connection:
+      - keep-alive
+      content-length:
+      - '1085'
+      content-type:
+      - application/json
+    parsed_body:
+      metrics:
+        latencyMs: 3896
+      output:
+        message:
+          content:
+          - reasoningContent:
+              reasoningText:
+                signature: ErcBCkgIBhABGAIiQDYN+P1S3ACL3r24cAMKCrRNiuPPxvmT2uzREPLRyKEUXagRbXn97QCke6L7OEZvlh7NdA/MQNTwMZV8TuB4qPASDLNxYxDx1S3luCIfARoMIwLZXwhsvjjTN72XIjAJrEl5ryAvv6C1+6YMCPC73ffE+kgwB96IcZaOuDFQtyaoWwcFcDPBguM6YNp5e3cqHbDQ3QF5dR4PP5q+3K23pual3pUdT/0e7khyIxXkGAI=
+                text: |-
+                  The user is asking for the largest city in their country. To answer this, I first need to determine what country the user is from. I can use the `get_user_country` function to retrieve this information.
+
+                  Once I have the user's country, I can then provide information about the largest city in that country.
+          - text: I'll need to check what country you're from to answer that question.
+          - toolUse:
+              input: {}
+              name: get_user_country
+              toolUseId: tooluse_W9DaUFg4Tj2cRPpndqxWSg
+          role: assistant
+      stopReason: tool_use
+      usage:
+        cacheReadInputTokenCount: 0
+        cacheReadInputTokens: 0
+        cacheWriteInputTokenCount: 0
+        cacheWriteInputTokens: 0
+        inputTokens: 397
+        outputTokens: 130
+        totalTokens: 527
+    status:
+      code: 200
+      message: OK
+- request:
+    body: '{"messages": [{"role": "user", "content": [{"text": "What is the largest city in the user country?"}]}, {"role":
+      "assistant", "content": [{"reasoningContent": {"reasoningText": {"text": "The user is asking for the largest city in
+      their country. To answer this, I first need to determine what country the user is from. I can use the `get_user_country`
+      function to retrieve this information.\n\nOnce I have the user''s country, I can then provide information about the
+      largest city in that country.", "signature": "ErcBCkgIBhABGAIiQDYN+P1S3ACL3r24cAMKCrRNiuPPxvmT2uzREPLRyKEUXagRbXn97QCke6L7OEZvlh7NdA/MQNTwMZV8TuB4qPASDLNxYxDx1S3luCIfARoMIwLZXwhsvjjTN72XIjAJrEl5ryAvv6C1+6YMCPC73ffE+kgwB96IcZaOuDFQtyaoWwcFcDPBguM6YNp5e3cqHbDQ3QF5dR4PP5q+3K23pual3pUdT/0e7khyIxXkGAI="}}},
+      {"text": "I''ll need to check what country you''re from to answer that question."}, {"toolUse": {"toolUseId": "tooluse_W9DaUFg4Tj2cRPpndqxWSg",
+      "name": "get_user_country", "input": {}}}]}, {"role": "user", "content": [{"toolResult": {"toolUseId": "tooluse_W9DaUFg4Tj2cRPpndqxWSg",
+      "content": [{"text": "Mexico"}], "status": "success"}}]}], "system": [], "inferenceConfig": {}, "toolConfig": {"tools":
+      [{"toolSpec": {"name": "get_user_country", "inputSchema": {"json": {"additionalProperties": false, "properties": {},
+      "type": "object"}}}}]}, "additionalModelRequestFields": {"thinking": {"type": "enabled", "budget_tokens": 1024}}}'
+    headers:
+      amz-sdk-invocation-id:
+      - !!binary |
+        ZWM5NzBkMzYtZTZhYi00MjdlLWFmMzItMTBhNTc2ZjBiMWNl
+      amz-sdk-request:
+      - !!binary |
+        YXR0ZW1wdD0x
+      content-length:
+      - '1399'
+      content-type:
+      - !!binary |
+        YXBwbGljYXRpb24vanNvbg==
+    method: POST
+    uri: https://bedrock-runtime.us-east-1.amazonaws.com/model/us.anthropic.claude-3-7-sonnet-20250219-v1%3A0/converse
+  response:
+    headers:
+      connection:
+      - keep-alive
+      content-length:
+      - '756'
+      content-type:
+      - application/json
+    parsed_body:
+      metrics:
+        latencyMs: 4529
+      output:
+        message:
+          content:
+          - text: |-
+              Based on your location in Mexico, the largest city is Mexico City (Ciudad de México). It's not only the capital but also the most populous city in Mexico with a metropolitan area population of over 21 million people, making it one of the largest urban agglomerations in the world.
+
+              Mexico City is an important cultural, financial, and political center for the country and has a rich history dating back to the Aztec empire when it was known as Tenochtitlán.
+          role: assistant
+      stopReason: end_turn
+      usage:
+        cacheReadInputTokenCount: 0
+        cacheReadInputTokens: 0
+        cacheWriteInputTokenCount: 0
+        cacheWriteInputTokens: 0
+        inputTokens: 539
+        outputTokens: 106
+        totalTokens: 645
+    status:
+      code: 200
+      message: OK
+version: 1

{pydantic_ai-0.5.1 → pydantic_ai-0.6.1}/tests/models/cassettes/test_model_names/test_known_model_names.yaml

@@ -15,7 +15,7 @@ interactions:
   response:
     headers:
       content-length:
-      - '550'
+      - '760'
       content-security-policy:
       - default-src 'none'; frame-ancestors 'none'
       content-type:
@@ -56,6 +56,23 @@ interactions:
       - us
       type:
       - text-to-embedding
+    - model_id: gpt-oss-120b
+      regions:
+      - us
+      type:
+      - text-to-text
+    - model_id: nova-lite
+      regions:
+      - eu
+      - us
+      type:
+      - text-to-text
+    - model_id: nova-pro
+      regions:
+      - eu
+      - us
+      type:
+      - text-to-text
     - model_id: stable-image-ultra
       regions:
       - us

{pydantic_ai-0.5.1 → pydantic_ai-0.6.1}/tests/models/test_bedrock.py

@@ -654,6 +654,29 @@ async def test_bedrock_model_thinking_part(allow_model_requests: None, bedrock_p
     )
 
 
+async def test_bedrock_anthropic_tool_with_thinking(allow_model_requests: None, bedrock_provider: BedrockProvider):
+    """When using thinking with tool calls in Anthropic, we need to send the thinking part back to the provider.
+
+    This tests the issue raised in https://github.com/pydantic/pydantic-ai/issues/2453.
+    """
+    m = BedrockConverseModel('us.anthropic.claude-3-7-sonnet-20250219-v1:0', provider=bedrock_provider)
+    settings = BedrockModelSettings(
+        bedrock_additional_model_requests_fields={'thinking': {'type': 'enabled', 'budget_tokens': 1024}},
+    )
+    agent = Agent(m, model_settings=settings)
+
+    @agent.tool_plain
+    async def get_user_country() -> str:
+        return 'Mexico'
+
+    result = await agent.run('What is the largest city in the user country?')
+    assert result.output == snapshot("""\
+Based on your location in Mexico, the largest city is Mexico City (Ciudad de México). It's not only the capital but also the most populous city in Mexico with a metropolitan area population of over 21 million people, making it one of the largest urban agglomerations in the world.
+
+Mexico City is an important cultural, financial, and political center for the country and has a rich history dating back to the Aztec empire when it was known as Tenochtitlán.\
+""")
+
+
 async def test_bedrock_group_consecutive_tool_return_parts(bedrock_provider: BedrockProvider):
     """
     Test that consecutive ToolReturnPart objects are grouped into a single user message for Bedrock.

{pydantic_ai-0.5.1 → pydantic_ai-0.6.1}/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.1}/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.1}/tests/models/test_openai.py

@@ -12,7 +12,7 @@ import httpx
 import pytest
 from dirty_equals import IsListOrTuple
 from inline_snapshot import snapshot
-from pydantic import AnyUrl, BaseModel, Discriminator, Field, Tag
+from pydantic import AnyUrl, BaseModel, ConfigDict, Discriminator, Field, Tag
 from typing_extensions import NotRequired, TypedDict
 
 from pydantic_ai import Agent, ModelHTTPError, ModelRetry, UnexpectedModelBehavior
@@ -1777,6 +1777,44 @@ def test_strict_schema():
     )
 
 
+def test_native_output_strict_mode(allow_model_requests: None):
+    class CityLocation(BaseModel):
+        city: str
+        country: str
+
+    c = completion_message(
+        ChatCompletionMessage(content='{"city": "Mexico City", "country": "Mexico"}', role='assistant'),
+    )
+    mock_client = MockOpenAI.create_mock(c)
+    model = OpenAIModel('gpt-4o', provider=OpenAIProvider(openai_client=mock_client))
+
+    # Explicit strict=True
+    agent = Agent(model, output_type=NativeOutput(CityLocation, strict=True))
+
+    agent.run_sync('What is the capital of Mexico?')
+    assert get_mock_chat_completion_kwargs(mock_client)[-1]['response_format']['json_schema']['strict'] is True
+
+    # Explicit strict=False
+    agent = Agent(model, output_type=NativeOutput(CityLocation, strict=False))
+
+    agent.run_sync('What is the capital of Mexico?')
+    assert get_mock_chat_completion_kwargs(mock_client)[-1]['response_format']['json_schema']['strict'] is False
+
+    # Strict-compatible
+    agent = Agent(model, output_type=NativeOutput(CityLocation))
+
+    agent.run_sync('What is the capital of Mexico?')
+    assert get_mock_chat_completion_kwargs(mock_client)[-1]['response_format']['json_schema']['strict'] is True
+
+    # Strict-incompatible
+    CityLocation.model_config = ConfigDict(extra='allow')
+
+    agent = Agent(model, output_type=NativeOutput(CityLocation))
+
+    agent.run_sync('What is the capital of Mexico?')
+    assert get_mock_chat_completion_kwargs(mock_client)[-1]['response_format']['json_schema']['strict'] is False
+
+
 async def test_openai_instructions(allow_model_requests: None, openai_api_key: str):
     m = OpenAIModel('gpt-4o', provider=OpenAIProvider(api_key=openai_api_key))
     agent = Agent(m, instructions='You are a helpful assistant.')

{pydantic_ai-0.5.1 → pydantic_ai-0.6.1}/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.1}/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.1}/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.1}/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.1}/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.1}/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.1}/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