pydantic-ai 0.2.8__tar.gz → 0.2.10__tar.gz

{pydantic_ai-0.2.8 → pydantic_ai-0.2.10}/PKG-INFO

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

pydantic_ai-0.2.10/tests/models/cassettes/test_gemini/test_labels_are_ignored_with_gla_provider.yaml

@@ -0,0 +1,62 @@
+interactions:
+  - request:
+      headers:
+        accept:
+          - "*/*"
+        accept-encoding:
+          - gzip, deflate
+        connection:
+          - keep-alive
+        content-length:
+          - "82"
+        content-type:
+          - application/json
+        host:
+          - generativelanguage.googleapis.com
+      method: POST
+      parsed_body:
+        contents:
+          - parts:
+              - text: What is the capital of France?
+            role: user
+      uri: https://generativelanguage.googleapis.com/v1beta/models/gemini-2.0-flash:generateContent
+    response:
+      headers:
+        alt-svc:
+          - h3=":443"; ma=2592000,h3-29=":443"; ma=2592000
+        content-length:
+          - "637"
+        content-type:
+          - application/json; charset=UTF-8
+        server-timing:
+          - gfet4t7; dur=426
+        transfer-encoding:
+          - chunked
+        vary:
+          - Origin
+          - X-Origin
+          - Referer
+      parsed_body:
+        candidates:
+          - avgLogprobs: -0.02703852951526642
+            content:
+              parts:
+                - text: |
+                    The capital of France is **Paris**.
+              role: model
+            finishReason: STOP
+        modelVersion: gemini-2.0-flash
+        usageMetadata:
+          candidatesTokenCount: 9
+          candidatesTokensDetails:
+            - modality: TEXT
+              tokenCount: 9
+          promptTokenCount: 7
+          promptTokensDetails:
+            - modality: TEXT
+              tokenCount: 7
+          totalTokenCount: 16
+      status:
+        code: 200
+        message: OK
+version: 1

pydantic_ai-0.2.10/tests/models/cassettes/test_gemini_vertexai/test_labels.yaml

@@ -0,0 +1,110 @@
+interactions:
+  - request:
+      body: grant_type=%5B%27refresh_token%27%5D&client_id=%5B%27764086051850-6qr4p6gpi6hn506pt8ejuq83di341hur.apps.googleusercontent.com%27%5D&client_secret=%5B%27scrubbed%27%5D&refresh_token=%5B%27scrubbed%27%5D
+      headers:
+        accept:
+          - "*/*"
+        accept-encoding:
+          - gzip, deflate
+        connection:
+          - keep-alive
+        content-length:
+          - "268"
+        content-type:
+          - application/x-www-form-urlencoded
+      method: POST
+      uri: https://oauth2.googleapis.com/token
+    response:
+      headers:
+        alt-svc:
+          - h3=":443"; ma=2592000,h3-29=":443"; ma=2592000
+        cache-control:
+          - no-cache, no-store, max-age=0, must-revalidate
+        content-length:
+          - "1419"
+        content-type:
+          - application/json; charset=utf-8
+        expires:
+          - Mon, 01 Jan 1990 00:00:00 GMT
+        pragma:
+          - no-cache
+        transfer-encoding:
+          - chunked
+        vary:
+          - Origin
+          - X-Origin
+          - Referer
+      parsed_body:
+        access_token: scrubbed
+        expires_in: 3599
+        id_token: eyJhbGciOiJSUzI1NiIsImtpZCI6IjgyMWYzYmM2NmYwNzUxZjc4NDA2MDY3OTliMWFkZjllOWZiNjBkZmIiLCJ0eXAiOiJKV1QifQ.eyJpc3MiOiJodHRwczovL2FjY291bnRzLmdvb2dsZS5jb20iLCJhenAiOiI3NjQwODYwNTE4NTAtNnFyNHA2Z3BpNmhuNTA2cHQ4ZWp1cTgzZGkzNDFodXIuYXBwcy5nb29nbGV1c2VyY29udGVudC5jb20iLCJhdWQiOiI3NjQwODYwNTE4NTAtNnFyNHA2Z3BpNmhuNTA2cHQ4ZWp1cTgzZGkzNDFodXIuYXBwcy5nb29nbGV1c2VyY29udGVudC5jb20iLCJzdWIiOiIxMDY1Njg0NzQzMTU3NzkyMTI1NTkiLCJoZCI6InB5ZGFudGljLmRldiIsImVtYWlsIjoibWFyY2Vsb0BweWRhbnRpYy5kZXYiLCJlbWFpbF92ZXJpZmllZCI6dHJ1ZSwiYXRfaGFzaCI6ImlyckNRNE00c0Z0Z2dfS2hRTVNjekEiLCJpYXQiOjE3NDM0MTM3NzcsImV4cCI6MTc0MzQxNzM3N30.BAvb4TlcIoYcQODNLFqwtUQoSNJJbpAR0lk2OyFxXK9rSZ7m1e1_Dp1O4ApxPUS7f_NX34eSCuDJN2IXgh8VBv4k3IhI7CbMydYeqXuwlbgOOp1Z0farGEKneU1M7TvdngigAJ9wT-2LHjKd_GEcGau-CUvzXpcT1IOnNNyXGVqtuGmEfcw5jjPkKJNECUheeNHE3zeImatTstOLuKmI1ZK-etl41l3poSNuQkZkrbQ80Vst8BdT-b1tnJnXP1KGATBIamDy99OOiB9a7a9m_ikXYEyN91yR76DYot3hpDPlOX0H9hF-BOSqoOtlSS2TMBkMvFiiYWjID1e_9VlNUg
+        scope: https://www.googleapis.com/auth/cloud-platform https://www.googleapis.com/auth/userinfo.email openid https://www.googleapis.com/auth/sqlservice.login
+        token_type: Bearer
+      status:
+        code: 200
+        message: OK
+  - request:
+      headers:
+        accept:
+          - "*/*"
+        accept-encoding:
+          - gzip, deflate
+        connection:
+          - keep-alive
+        content-length:
+          - "133"
+        content-type:
+          - application/json
+        host:
+          - us-central1-aiplatform.googleapis.com
+      method: POST
+      parsed_body:
+        contents:
+          - parts:
+              - text: What is the capital of France?
+            role: user
+        labels:
+          environment: test
+          team: analytics
+      uri: https://us-central1-aiplatform.googleapis.com/v1/projects/pydantic-ai/locations/us-central1/publishers/google/models/gemini-2.0-flash:generateContent
+    response:
+      headers:
+        alt-svc:
+          - h3=":443"; ma=2592000,h3-29=":443"; ma=2592000
+        content-length:
+          - "759"
+        content-type:
+          - application/json; charset=UTF-8
+        transfer-encoding:
+          - chunked
+        vary:
+          - Origin
+          - X-Origin
+          - Referer
+      parsed_body:
+        candidates:
+          - avgLogprobs: -0.02703852951526642
+            content:
+              parts:
+                - text: |
+                    The capital of France is **Paris**.
+              role: model
+            finishReason: STOP
+        createTime: "2025-05-23T07:53:55.494386Z"
+        modelVersion: gemini-2.0-flash
+        responseId: kykwaLKWHti5nvgPmN2T8AE
+        usageMetadata:
+          candidatesTokenCount: 9
+          candidatesTokensDetails:
+            - modality: TEXT
+              tokenCount: 9
+          promptTokenCount: 7
+          promptTokensDetails:
+            - modality: TEXT
+              tokenCount: 7
+          totalTokenCount: 16
+          trafficType: ON_DEMAND
+      status:
+        code: 200
+        message: OK
+version: 1

pydantic_ai-0.2.10/tests/models/cassettes/test_google/test_google_model_vertex_labels.yaml

@@ -0,0 +1,115 @@
+interactions:
+  - request:
+      body: grant_type=%5B%27refresh_token%27%5D&client_id=%5B%27764086051850-6qr4p6gpi6hn506pt8ejuq83di341hur.apps.googleusercontent.com%27%5D&client_secret=%5B%27scrubbed%27%5D&refresh_token=%5B%27scrubbed%27%5D
+      headers:
+        accept:
+          - "*/*"
+        accept-encoding:
+          - gzip, deflate
+        connection:
+          - keep-alive
+        content-length:
+          - "268"
+        content-type:
+          - application/x-www-form-urlencoded
+      method: POST
+      uri: https://oauth2.googleapis.com/token
+    response:
+      headers:
+        alt-svc:
+          - h3=":443"; ma=2592000,h3-29=":443"; ma=2592000
+        cache-control:
+          - no-cache, no-store, max-age=0, must-revalidate
+        content-length:
+          - "1420"
+        content-type:
+          - application/json; charset=utf-8
+        expires:
+          - Mon, 01 Jan 1990 00:00:00 GMT
+        pragma:
+          - no-cache
+        transfer-encoding:
+          - chunked
+        vary:
+          - Origin
+          - X-Origin
+          - Referer
+      parsed_body:
+        access_token: scrubbed
+        expires_in: 3599
+        id_token: eyJhbGciOiJSUzI1NiIsImtpZCI6IjY2MGVmM2I5Nzg0YmRmNTZlYmU4NTlmNTc3ZjdmYjJlOGMxY2VmZmIiLCJ0eXAiOiJKV1QifQ.eyJpc3MiOiJodHRwczovL2FjY291bnRzLmdvb2dsZS5jb20iLCJhenAiOiI3NjQwODYwNTE4NTAtNnFyNHA2Z3BpNmhuNTA2cHQ4ZWp1cTgzZGkzNDFodXIuYXBwcy5nb29nbGV1c2VyY29udGVudC5jb20iLCJhdWQiOiI3NjQwODYwNTE4NTAtNnFyNHA2Z3BpNmhuNTA2cHQ4ZWp1cTgzZGkzNDFodXIuYXBwcy5nb29nbGV1c2VyY29udGVudC5jb20iLCJzdWIiOiIxMDY1Njg0NzQzMTU3NzkyMTI1NTkiLCJoZCI6InB5ZGFudGljLmRldiIsImVtYWlsIjoibWFyY2Vsb0BweWRhbnRpYy5kZXYiLCJlbWFpbF92ZXJpZmllZCI6dHJ1ZSwiYXRfaGFzaCI6Ii1CeV9XOWwtRHg1ekg0YTVOV25fV3ciLCJpYXQiOjE3NDc1NzQxOTEsImV4cCI6MTc0NzU3Nzc5MX0.dHg3qRlYoQ8WyIml7-kGqsuefvkl5deuZ0yTQM-RvKuuqtF_t6p8TrWbndEuSbZpRn9JhVPnsoEAYVPexbGy-pon4gu1aHH0dJNq3ghhdim7qp5JWpegLaZqvNvELvEHjj2VNLWXQ70-5wEaI_HCtAWTjlROAHQxvoWHJAdeH0Yf9zoljEBQvx3VLDLEpdCcMd-UGNCBucpQlFHcCJs5Qq8yj8R8f27BCEmRo7z9K3Axuedj_wcJ_tWV1x1tWxojUloJaKsIfztFOPFxzOdNPOlTHXsE47d4v43v87a8LhdDGloD72xN_kLapfIqyTIwRTj4cQvQp5H0u7As49fvMA
+        scope: https://www.googleapis.com/auth/userinfo.email openid https://www.googleapis.com/auth/sqlservice.login https://www.googleapis.com/auth/cloud-platform
+        token_type: Bearer
+      status:
+        code: 200
+        message: OK
+  - request:
+      headers:
+        accept:
+          - "*/*"
+        accept-encoding:
+          - gzip, deflate
+        connection:
+          - keep-alive
+        content-length:
+          - "257"
+        content-type:
+          - application/json
+        host:
+          - aiplatform.googleapis.com
+      method: POST
+      parsed_body:
+        contents:
+          - parts:
+              - text: What is the capital of France?
+            role: user
+        generationConfig: {}
+        labels:
+          environment: test
+          team: analytics
+        systemInstruction:
+          parts:
+            - text: You are a helpful chatbot.
+          role: user
+      uri: https://aiplatform.googleapis.com/v1beta1/projects/pydantic-ai/locations/global/publishers/google/models/gemini-2.0-flash:generateContent
+    response:
+      headers:
+        alt-svc:
+          - h3=":443"; ma=2592000,h3-29=":443"; ma=2592000
+        content-length:
+          - "759"
+        content-type:
+          - application/json; charset=UTF-8
+        transfer-encoding:
+          - chunked
+        vary:
+          - Origin
+          - X-Origin
+          - Referer
+      parsed_body:
+        candidates:
+          - avgLogprobs: -0.0005532301729544997
+            content:
+              parts:
+                - text: |
+                    The capital of France is Paris.
+              role: model
+            finishReason: STOP
+        createTime: "2025-05-23T07:09:59.524624Z"
+        modelVersion: gemini-2.0-flash
+        responseId: sN0paKOZFtmtyOgPqMyL6AE
+        usageMetadata:
+          candidatesTokenCount: 8
+          candidatesTokensDetails:
+            - modality: TEXT
+              tokenCount: 8
+          promptTokenCount: 13
+          promptTokensDetails:
+            - modality: TEXT
+              tokenCount: 13
+          totalTokenCount: 21
+          trafficType: ON_DEMAND
+      status:
+        code: 200
+        message: OK
+version: 1

{pydantic_ai-0.2.8 → pydantic_ai-0.2.10}/tests/models/test_gemini.py

@@ -1127,6 +1127,18 @@ I need to use the `get_image` tool to see the image first.
     )
 
 
+@pytest.mark.vcr()
+async def test_labels_are_ignored_with_gla_provider(allow_model_requests: None, gemini_api_key: str) -> None:
+    m = GeminiModel('gemini-2.0-flash', provider=GoogleGLAProvider(api_key=gemini_api_key))
+    agent = Agent(m)
+
+    result = await agent.run(
+        'What is the capital of France?',
+        model_settings=GeminiModelSettings(gemini_labels={'environment': 'test', 'team': 'analytics'}),
+    )
+    assert result.output == snapshot('The capital of France is **Paris**.\n')
+
+
 @pytest.mark.vcr()
 async def test_image_as_binary_content_input(
     allow_model_requests: None, gemini_api_key: str, image_content: BinaryContent

pydantic_ai-0.2.10/tests/models/test_gemini_vertexai.py

@@ -0,0 +1,51 @@
+import os
+from dataclasses import dataclass
+
+import pytest
+from inline_snapshot import snapshot
+from pytest_mock import MockerFixture
+
+from pydantic_ai import Agent
+from pydantic_ai.models.gemini import GeminiModel, GeminiModelSettings
+
+from ..conftest import try_import
+
+with try_import() as imports_successful:
+    from google.auth.transport.requests import Request
+
+
+pytestmark = [
+    pytest.mark.skipif(not imports_successful(), reason='google-auth not installed'),
+    pytest.mark.anyio,
+]
+
+
+@pytest.fixture(autouse=True)
+def vertex_provider_auth(mocker: MockerFixture) -> None:  # pragma: lax no cover
+    # Locally, we authenticate via `gcloud` CLI, so we don't need to patch anything.
+    if not os.getenv('CI'):
+        return
+
+    @dataclass
+    class NoOpCredentials:
+        token = 'my-token'
+
+        def refresh(self, request: Request): ...
+
+    return_value = (NoOpCredentials(), 'pydantic-ai')
+    mocker.patch('pydantic_ai.providers.google_vertex.google.auth.default', return_value=return_value)
+
+
+@pytest.mark.skipif(
+    not os.getenv('CI', False), reason='Requires properly configured local google vertex config to pass'
+)
+@pytest.mark.vcr()
+async def test_labels(allow_model_requests: None) -> None:
+    m = GeminiModel('gemini-2.0-flash', provider='google-vertex')
+    agent = Agent(m)
+
+    result = await agent.run(
+        'What is the capital of France?',
+        model_settings=GeminiModelSettings(gemini_labels={'environment': 'test', 'team': 'analytics'}),
+    )
+    assert result.output == snapshot('The capital of France is **Paris**.\n')

{pydantic_ai-0.2.8 → pydantic_ai-0.2.10}/tests/models/test_google.py

@@ -262,6 +262,28 @@ async def test_google_model_thinking_config(allow_model_requests: None, google_p
     assert result.output == snapshot('The capital of France is **Paris**.')
 
 
+@pytest.mark.skipif(
+    not os.getenv('CI', False), reason='Requires properly configured local google vertex config to pass'
+)
+async def test_google_model_vertex_labels(allow_model_requests: None):
+    provider = GoogleProvider(location='global', project='pydantic-ai')
+    model = GoogleModel('gemini-2.0-flash', provider=provider)
+    settings = GoogleModelSettings(google_labels={'environment': 'test', 'team': 'analytics'})
+    agent = Agent(model=model, system_prompt='You are a helpful chatbot.', model_settings=settings)
+    result = await agent.run('What is the capital of France?')
+    assert result.output == snapshot('The capital of France is Paris.\n')
+
+
+async def test_google_model_gla_labels_raises_value_error(allow_model_requests: None, google_provider: GoogleProvider):
+    model = GoogleModel('gemini-2.0-flash', provider=google_provider)
+    settings = GoogleModelSettings(google_labels={'environment': 'test', 'team': 'analytics'})
+    agent = Agent(model=model, system_prompt='You are a helpful chatbot.', model_settings=settings)
+
+    # Raises before any request is made.
+    with pytest.raises(ValueError, match='labels parameter is not supported in Gemini API.'):
+        await agent.run('What is the capital of France?')
+
+
 @pytest.fixture(autouse=True)
 def vertex_provider_auth(mocker: MockerFixture) -> None:  # pragma: lax no cover
     # Locally, we authenticate via `gcloud` CLI, so we don't need to patch anything.

{pydantic_ai-0.2.8 → pydantic_ai-0.2.10}/tests/test_mcp.py

@@ -1,6 +1,7 @@
 """Tests for the MCP (Model Context Protocol) server implementation."""
 
 import re
+from datetime import timedelta
 from pathlib import Path
 
 import pytest
@@ -70,25 +71,40 @@ async def test_stdio_server_with_cwd():
         assert len(tools) == 10
 
 
-def test_sse_server():
-    sse_server = MCPServerHTTP(url='http://localhost:8000/sse')
-    assert sse_server.url == 'http://localhost:8000/sse'
-    assert sse_server._get_log_level() is None  # pyright: ignore[reportPrivateUsage]
+def test_http_server():
+    http_server = MCPServerHTTP(url='http://localhost:8000/sse')
+    assert http_server.url == 'http://localhost:8000/sse'
+    assert http_server._get_log_level() is None  # pyright: ignore[reportPrivateUsage]
 
 
-def test_sse_server_with_header_and_timeout():
-    sse_server = MCPServerHTTP(
+def test_http_server_with_header_and_timeout():
+    http_server = MCPServerHTTP(
         url='http://localhost:8000/sse',
         headers={'my-custom-header': 'my-header-value'},
         timeout=10,
         sse_read_timeout=100,
         log_level='info',
     )
-    assert sse_server.url == 'http://localhost:8000/sse'
-    assert sse_server.headers is not None and sse_server.headers['my-custom-header'] == 'my-header-value'
-    assert sse_server.timeout == 10
-    assert sse_server.sse_read_timeout == 100
-    assert sse_server._get_log_level() == 'info'  # pyright: ignore[reportPrivateUsage]
+    assert http_server.url == 'http://localhost:8000/sse'
+    assert http_server.headers is not None and http_server.headers['my-custom-header'] == 'my-header-value'
+    assert http_server.timeout == 10
+    assert http_server.sse_read_timeout == 100
+    assert http_server._get_log_level() == 'info'  # pyright: ignore[reportPrivateUsage]
+
+
+def test_http_server_with_timedelta_arguments():
+    http_server = MCPServerHTTP(
+        url='http://localhost:8000/sse',
+        headers={'my-custom-header': 'my-header-value'},
+        timeout=timedelta(seconds=10),  # type: ignore[arg-type]
+        sse_read_timeout=timedelta(seconds=100),  # type: ignore[arg-type]
+        log_level='info',
+    )
+    assert http_server.url == 'http://localhost:8000/sse'
+    assert http_server.headers is not None and http_server.headers['my-custom-header'] == 'my-header-value'
+    assert http_server.timeout == 10
+    assert http_server.sse_read_timeout == 100
+    assert http_server._get_log_level() == 'info'  # pyright: ignore[reportPrivateUsage]
 
 
 @pytest.mark.vcr()