pydantic-ai 0.0.44__tar.gz → 0.0.45__tar.gz

{pydantic_ai-0.0.44 → pydantic_ai-0.0.45}/PKG-INFO

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

{pydantic_ai-0.0.44 → pydantic_ai-0.0.45}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "pydantic-ai"
-version = "0.0.44"
+version = "0.0.45"
 description = "Agent Framework / shim to use Pydantic with LLMs"
 authors = [
     { name = "Samuel Colvin", email = "samuel@pydantic.dev" },
@@ -36,7 +36,7 @@ classifiers = [
 ]
 requires-python = ">=3.9"
 dependencies = [
-    "pydantic-ai-slim[openai,vertexai,groq,anthropic,mistral,cohere,bedrock,cli,mcp]==0.0.44",
+    "pydantic-ai-slim[openai,vertexai,groq,anthropic,mistral,cohere,bedrock,cli,mcp]==0.0.45",
 ]
 
 [project.urls]
@@ -46,7 +46,7 @@ Documentation = "https://ai.pydantic.dev"
 Changelog = "https://github.com/pydantic/pydantic-ai/releases"
 
 [project.optional-dependencies]
-examples = ["pydantic-ai-examples==0.0.44"]
+examples = ["pydantic-ai-examples==0.0.45"]
 logfire = ["logfire>=2.3"]
 
 [tool.uv.sources]

{pydantic_ai-0.0.44 → pydantic_ai-0.0.45}/tests/conftest.py

@@ -40,7 +40,7 @@ else:
 
     def IsNow(*args: Any, **kwargs: Any):
         # Increase the default value of `delta` to 10 to reduce test flakiness on overburdened machines
-        if 'delta' not in kwargs:
+        if 'delta' not in kwargs:  # pragma: no branch
             kwargs['delta'] = 10
         return _IsNow(*args, **kwargs)
 
@@ -154,7 +154,7 @@ def create_module(tmp_path: Path, request: pytest.FixtureRequest) -> Callable[[s
 
 
 @contextmanager
-def try_import() -> Iterator[Callable[[], bool]]:
+def try_import() -> Iterator[Callable[[], bool]]:  # pragma: no cover
     import_success = False
 
     def check_import() -> bool:
@@ -251,6 +251,11 @@ def co_api_key() -> str:
     return os.getenv('CO_API_KEY', 'mock-api-key')
 
 
+@pytest.fixture(scope='session')
+def mistral_api_key() -> str:
+    return os.getenv('MISTRAL_API_KEY', 'mock-api-key')
+
+
 @pytest.fixture
 def mock_snapshot_id(mocker: MockerFixture):
     i = 0

{pydantic_ai-0.0.44 → pydantic_ai-0.0.45}/tests/models/cassettes/test_openai/test_audio_as_binary_content_input.yaml

@@ -37,13 +37,13 @@ interactions:
       connection:
       - keep-alive
       content-length:
-      - '882'
+      - '909'
       content-type:
       - application/json
       openai-organization:
       - pydantic-28gund
       openai-processing-ms:
-      - '1394'
+      - '1898'
       openai-version:
       - '2020-10-01'
       strict-transport-security:
@@ -55,15 +55,16 @@ interactions:
       - finish_reason: stop
         index: 0
         message:
+          annotations: []
           content: The name mentioned in the audio is Marcelo.
           refusal: null
           role: assistant
-      created: 1740408639
-      id: chatcmpl-B4U51malankDFzpReXZCXRgKQsAx1
+      created: 1742905314
+      id: chatcmpl-BExZy74Y67dd65ec2z4iuzM0Exnks
       model: gpt-4o-audio-preview-2024-12-17
       object: chat.completion
       service_tier: default
-      system_fingerprint: fp_31e26c9138
+      system_fingerprint: fp_bf4a46b2e5
       usage:
         completion_tokens: 9
         completion_tokens_details:

{pydantic_ai-0.0.44 → pydantic_ai-0.0.45}/tests/models/cassettes/test_openai/test_image_as_binary_content_input.yaml

@@ -36,13 +36,13 @@ interactions:
       connection:
       - keep-alive
       content-length:
-      - '812'
+      - '839'
       content-type:
       - application/json
       openai-organization:
       - pydantic-28gund
       openai-processing-ms:
-      - '3988'
+      - '3245'
       openai-version:
       - '2020-10-01'
       strict-transport-security:
@@ -55,15 +55,16 @@ interactions:
         index: 0
         logprobs: null
         message:
+          annotations: []
           content: The fruit in the image is a kiwi.
           refusal: null
           role: assistant
-      created: 1740408634
-      id: chatcmpl-B4U4wIOoauVGgr7Nqe5H2oj5tZoKK
+      created: 1742905309
+      id: chatcmpl-BExZtIgl2l91Y1IYJW3DSJj1yRlpY
       model: gpt-4o-2024-08-06
       object: chat.completion
       service_tier: default
-      system_fingerprint: fp_197415291b
+      system_fingerprint: fp_83df987f64
       usage:
         completion_tokens: 10
         completion_tokens_details:
@@ -71,11 +72,11 @@ interactions:
           audio_tokens: 0
           reasoning_tokens: 0
           rejected_prediction_tokens: 0
-        prompt_tokens: 1152
+        prompt_tokens: 1119
         prompt_tokens_details:
           audio_tokens: 0
           cached_tokens: 0
-        total_tokens: 1162
+        total_tokens: 1129
     status:
       code: 200
       message: OK

pydantic_ai-0.0.45/tests/models/cassettes/test_openai/test_multiple_agent_tool_calls.yaml

@@ -0,0 +1,391 @@
+interactions:
+- request:
+    headers:
+      accept:
+      - '*/*'
+      accept-encoding:
+      - gzip, deflate
+      connection:
+      - keep-alive
+      content-length:
+      - '318'
+      content-type:
+      - application/json
+      host:
+      - generativelanguage.googleapis.com
+    method: POST
+    parsed_body:
+      contents:
+      - parts:
+        - text: What is the capital of France?
+        role: user
+      tools:
+        function_declarations:
+        - description: Get the capital of a country.
+          name: get_capital
+          parameters:
+            properties:
+              country:
+                description: The country name.
+                type: string
+            required:
+            - country
+            type: object
+    uri: https://generativelanguage.googleapis.com/v1beta/models/gemini-2.0-flash-exp:generateContent
+  response:
+    headers:
+      alt-svc:
+      - h3=":443"; ma=2592000,h3-29=":443"; ma=2592000
+      content-length:
+      - '741'
+      content-type:
+      - application/json; charset=UTF-8
+      server-timing:
+      - gfet4t7; dur=407
+      transfer-encoding:
+      - chunked
+      vary:
+      - Origin
+      - X-Origin
+      - Referer
+    parsed_body:
+      candidates:
+      - avgLogprobs: 3.921892493963242e-06
+        content:
+          parts:
+          - functionCall:
+              args:
+                country: France
+              name: get_capital
+          role: model
+        finishReason: STOP
+      modelVersion: gemini-2.0-flash-exp
+      usageMetadata:
+        candidatesTokenCount: 5
+        candidatesTokensDetails:
+        - modality: TEXT
+          tokenCount: 5
+        promptTokenCount: 23
+        promptTokensDetails:
+        - modality: TEXT
+          tokenCount: 23
+        totalTokenCount: 28
+    status:
+      code: 200
+      message: OK
+- request:
+    headers:
+      accept:
+      - '*/*'
+      accept-encoding:
+      - gzip, deflate
+      connection:
+      - keep-alive
+      content-length:
+      - '519'
+      content-type:
+      - application/json
+      host:
+      - generativelanguage.googleapis.com
+    method: POST
+    parsed_body:
+      contents:
+      - parts:
+        - text: What is the capital of France?
+        role: user
+      - parts:
+        - functionCall:
+            args:
+              country: France
+            name: get_capital
+        role: model
+      - parts:
+        - functionResponse:
+            name: get_capital
+            response:
+              return_value: Paris
+        role: user
+      tools:
+        function_declarations:
+        - description: Get the capital of a country.
+          name: get_capital
+          parameters:
+            properties:
+              country:
+                description: The country name.
+                type: string
+            required:
+            - country
+            type: object
+    uri: https://generativelanguage.googleapis.com/v1beta/models/gemini-2.0-flash-exp:generateContent
+  response:
+    headers:
+      alt-svc:
+      - h3=":443"; ma=2592000,h3-29=":443"; ma=2592000
+      content-length:
+      - '640'
+      content-type:
+      - application/json; charset=UTF-8
+      server-timing:
+      - gfet4t7; dur=381
+      transfer-encoding:
+      - chunked
+      vary:
+      - Origin
+      - X-Origin
+      - Referer
+    parsed_body:
+      candidates:
+      - avgLogprobs: 5.035180947743356e-07
+        content:
+          parts:
+          - text: |
+              The capital of France is Paris.
+          role: model
+        finishReason: STOP
+      modelVersion: gemini-2.0-flash-exp
+      usageMetadata:
+        candidatesTokenCount: 8
+        candidatesTokensDetails:
+        - modality: TEXT
+          tokenCount: 8
+        promptTokenCount: 35
+        promptTokensDetails:
+        - modality: TEXT
+          tokenCount: 35
+        totalTokenCount: 43
+    status:
+      code: 200
+      message: OK
+- request:
+    headers:
+      accept:
+      - application/json
+      accept-encoding:
+      - gzip, deflate
+      connection:
+      - keep-alive
+      content-length:
+      - '801'
+      content-type:
+      - application/json
+      host:
+      - api.openai.com
+    method: POST
+    parsed_body:
+      messages:
+      - content: What is the capital of France?
+        role: user
+      - role: assistant
+        tool_calls:
+        - function:
+            arguments: '{"country":"France"}'
+            name: get_capital
+          id: pyd_ai_504f8147f83f44f3a5f14d87bfd01bda
+          type: function
+      - content: Paris
+        role: tool
+        tool_call_id: pyd_ai_504f8147f83f44f3a5f14d87bfd01bda
+      - content: |
+          The capital of France is Paris.
+        role: assistant
+      - content: What is the capital of England?
+        role: user
+      model: gpt-4o-mini
+      n: 1
+      stream: false
+      tool_choice: auto
+      tools:
+      - function:
+          description: Get the capital of a country.
+          name: get_capital
+          parameters:
+            additionalProperties: false
+            properties:
+              country:
+                description: The country name.
+                type: string
+            required:
+            - country
+            type: object
+        type: function
+    uri: https://api.openai.com/v1/chat/completions
+  response:
+    headers:
+      access-control-expose-headers:
+      - X-Request-ID
+      alt-svc:
+      - h3=":443"; ma=86400
+      connection:
+      - keep-alive
+      content-length:
+      - '1091'
+      content-type:
+      - application/json
+      openai-organization:
+      - pydantic-28gund
+      openai-processing-ms:
+      - '784'
+      openai-version:
+      - '2020-10-01'
+      strict-transport-security:
+      - max-age=31536000; includeSubDomains; preload
+      transfer-encoding:
+      - chunked
+    parsed_body:
+      choices:
+      - finish_reason: tool_calls
+        index: 0
+        logprobs: null
+        message:
+          annotations: []
+          content: null
+          refusal: null
+          role: assistant
+          tool_calls:
+          - function:
+              arguments: '{"country":"England"}'
+              name: get_capital
+            id: call_SkEQ3ZGSJC8m6AvaIGNuuKdm
+            type: function
+      created: 1742842885
+      id: chatcmpl-BEhL3fZWgTz2Z57jXexYbQPsOBUm3
+      model: gpt-4o-mini-2024-07-18
+      object: chat.completion
+      service_tier: default
+      system_fingerprint: fp_b8bc95a0ac
+      usage:
+        completion_tokens: 16
+        completion_tokens_details:
+          accepted_prediction_tokens: 0
+          audio_tokens: 0
+          reasoning_tokens: 0
+          rejected_prediction_tokens: 0
+        prompt_tokens: 104
+        prompt_tokens_details:
+          audio_tokens: 0
+          cached_tokens: 0
+        total_tokens: 120
+    status:
+      code: 200
+      message: OK
+- request:
+    headers:
+      accept:
+      - application/json
+      accept-encoding:
+      - gzip, deflate
+      connection:
+      - keep-alive
+      content-length:
+      - '1050'
+      content-type:
+      - application/json
+      cookie:
+      - __cf_bm=fEZONI5B238kQ6OXHK13zgt8ywFP33X_T.Rj7kOX_.U-1742842886-1.0.1.1-Qx39nivaihVZJFeyQKfx9cVGl_qxdo0BULRdHDhZN7JckuOkQmjCwesvWu7fLNf8qR8oZBKufDsfLmg4lUd3l314ZvU8xxx44rgtJTHVQTM;
+        _cfuvid=6dZO7_a2pjh.BwMhl506o0zCkNlkhADPSV7Hxx2Pqks-1742842886619-0.0.1.1-604800000
+      host:
+      - api.openai.com
+    method: POST
+    parsed_body:
+      messages:
+      - content: What is the capital of France?
+        role: user
+      - role: assistant
+        tool_calls:
+        - function:
+            arguments: '{"country":"France"}'
+            name: get_capital
+          id: pyd_ai_504f8147f83f44f3a5f14d87bfd01bda
+          type: function
+      - content: Paris
+        role: tool
+        tool_call_id: pyd_ai_504f8147f83f44f3a5f14d87bfd01bda
+      - content: |
+          The capital of France is Paris.
+        role: assistant
+      - content: What is the capital of England?
+        role: user
+      - role: assistant
+        tool_calls:
+        - function:
+            arguments: '{"country":"England"}'
+            name: get_capital
+          id: call_SkEQ3ZGSJC8m6AvaIGNuuKdm
+          type: function
+      - content: London
+        role: tool
+        tool_call_id: call_SkEQ3ZGSJC8m6AvaIGNuuKdm
+      model: gpt-4o-mini
+      n: 1
+      stream: false
+      tool_choice: auto
+      tools:
+      - function:
+          description: Get the capital of a country.
+          name: get_capital
+          parameters:
+            additionalProperties: false
+            properties:
+              country:
+                description: The country name.
+                type: string
+            required:
+            - country
+            type: object
+        type: function
+    uri: https://api.openai.com/v1/chat/completions
+  response:
+    headers:
+      access-control-expose-headers:
+      - X-Request-ID
+      alt-svc:
+      - h3=":443"; ma=86400
+      connection:
+      - keep-alive
+      content-length:
+      - '841'
+      content-type:
+      - application/json
+      openai-organization:
+      - pydantic-28gund
+      openai-processing-ms:
+      - '456'
+      openai-version:
+      - '2020-10-01'
+      strict-transport-security:
+      - max-age=31536000; includeSubDomains; preload
+      transfer-encoding:
+      - chunked
+    parsed_body:
+      choices:
+      - finish_reason: stop
+        index: 0
+        logprobs: null
+        message:
+          annotations: []
+          content: The capital of England is London.
+          refusal: null
+          role: assistant
+      created: 1742842886
+      id: chatcmpl-BEhL4jHN01U9VPVVYzgKrwORTJ0Pw
+      model: gpt-4o-mini-2024-07-18
+      object: chat.completion
+      service_tier: default
+      system_fingerprint: fp_b8bc95a0ac
+      usage:
+        completion_tokens: 9
+        completion_tokens_details:
+          accepted_prediction_tokens: 0
+          audio_tokens: 0
+          reasoning_tokens: 0
+          rejected_prediction_tokens: 0
+        prompt_tokens: 129
+        prompt_tokens_details:
+          audio_tokens: 0
+          cached_tokens: 0
+        total_tokens: 138
+    status:
+      code: 200
+      message: OK
+version: 1

{pydantic_ai-0.0.44 → pydantic_ai-0.0.45}/tests/models/cassettes/test_openai/test_openai_o1_mini_system_role[developer].yaml

@@ -11,9 +11,6 @@ interactions:
       - '149'
       content-type:
       - application/json
-      cookie:
-      - __cf_bm=k1xoKY2.Jtgshh4aAfJbgzfCta3eg2_pKpOubx52AqY-1740396355-1.0.1.1-piAnJhOpL2bGY4paQ8VURT3k0nvStOyXvtsja2kOYPgRmSNpE1vUf87UU.AnB_di_8lc63rtIkZjSJSSZPPgDQ;
-        _cfuvid=9Q7wbAVWuVxqmo8CZYy4FZvphfjrBGzqwRIt0TKETsE-1740396355005-0.0.1.1-604800000
       host:
       - api.openai.com
     method: POST
@@ -42,7 +39,7 @@ interactions:
       openai-organization:
       - pydantic-28gund
       openai-processing-ms:
-      - '23'
+      - '36'
       openai-version:
       - '2020-10-01'
       strict-transport-security:

{pydantic_ai-0.0.44 → pydantic_ai-0.0.45}/tests/models/cassettes/test_openai/test_openai_o1_mini_system_role[system].yaml

@@ -39,14 +39,9 @@ interactions:
       openai-organization:
       - pydantic-28gund
       openai-processing-ms:
-      - '15'
+      - '157'
       openai-version:
       - '2020-10-01'
-      set-cookie:
-      - __cf_bm=k1xoKY2.Jtgshh4aAfJbgzfCta3eg2_pKpOubx52AqY-1740396355-1.0.1.1-piAnJhOpL2bGY4paQ8VURT3k0nvStOyXvtsja2kOYPgRmSNpE1vUf87UU.AnB_di_8lc63rtIkZjSJSSZPPgDQ;
-        path=/; expires=Mon, 24-Feb-25 11:55:55 GMT; domain=.api.openai.com; HttpOnly; Secure; SameSite=None
-      - _cfuvid=9Q7wbAVWuVxqmo8CZYy4FZvphfjrBGzqwRIt0TKETsE-1740396355005-0.0.1.1-604800000; path=/; domain=.api.openai.com;
-        HttpOnly; Secure; SameSite=None
       strict-transport-security:
       - max-age=31536000; includeSubDomains; preload
     parsed_body:

pydantic_ai-0.0.45/tests/models/cassettes/test_openai/test_user_id.yaml

@@ -0,0 +1,79 @@
+interactions:
+- request:
+    headers:
+      accept:
+      - application/json
+      accept-encoding:
+      - gzip, deflate
+      connection:
+      - keep-alive
+      content-length:
+      - '103'
+      content-type:
+      - application/json
+      host:
+      - api.openai.com
+    method: POST
+    parsed_body:
+      messages:
+      - content: hello
+        role: user
+      model: gpt-4o
+      n: 1
+      stream: false
+      user: user_id
+    uri: https://api.openai.com/v1/chat/completions
+  response:
+    headers:
+      access-control-expose-headers:
+      - X-Request-ID
+      alt-svc:
+      - h3=":443"; ma=86400
+      connection:
+      - keep-alive
+      content-length:
+      - '835'
+      content-type:
+      - application/json
+      openai-organization:
+      - pydantic-28gund
+      openai-processing-ms:
+      - '235'
+      openai-version:
+      - '2020-10-01'
+      strict-transport-security:
+      - max-age=31536000; includeSubDomains; preload
+      transfer-encoding:
+      - chunked
+    parsed_body:
+      choices:
+      - finish_reason: stop
+        index: 0
+        logprobs: null
+        message:
+          annotations: []
+          content: Hello! How can I assist you today?
+          refusal: null
+          role: assistant
+      created: 1742906323
+      id: chatcmpl-BExqF10T3Bt833bEyp7OAqjVznPeS
+      model: gpt-4o-2024-08-06
+      object: chat.completion
+      service_tier: default
+      system_fingerprint: fp_90d33c15d4
+      usage:
+        completion_tokens: 10
+        completion_tokens_details:
+          accepted_prediction_tokens: 0
+          audio_tokens: 0
+          reasoning_tokens: 0
+          rejected_prediction_tokens: 0
+        prompt_tokens: 8
+        prompt_tokens_details:
+          audio_tokens: 0
+          cached_tokens: 0
+        total_tokens: 18
+    status:
+      code: 200
+      message: OK
+version: 1

{pydantic_ai-0.0.44 → pydantic_ai-0.0.45}/tests/models/test_anthropic.py

@@ -7,7 +7,6 @@ from dataclasses import dataclass, field
 from datetime import timezone
 from functools import cached_property
 from typing import Any, TypeVar, Union, cast
-from unittest.mock import patch
 
 import httpx
 import pytest
@@ -30,7 +29,7 @@ from pydantic_ai.messages import (
 from pydantic_ai.result import Usage
 from pydantic_ai.settings import ModelSettings
 
-from ..conftest import IsDatetime, IsNow, IsStr, raise_if_exception, try_import
+from ..conftest import IsDatetime, IsNow, IsStr, TestEnv, raise_if_exception, try_import
 from .mock_async_stream import MockAsyncStream
 
 with try_import() as imports_successful:
@@ -680,8 +679,8 @@ def test_init_with_provider():
     assert model.client == provider.client
 
 
-def test_init_with_provider_string():
-    with patch.dict(os.environ, {'ANTHROPIC_API_KEY': 'env-api-key'}, clear=False):
-        model = AnthropicModel('claude-3-opus-latest', provider='anthropic')
-        assert model.model_name == 'claude-3-opus-latest'
-        assert model.client is not None
+def test_init_with_provider_string(env: TestEnv):
+    env.set('ANTHROPIC_API_KEY', 'env-api-key')
+    model = AnthropicModel('claude-3-opus-latest', provider='anthropic')
+    assert model.model_name == 'claude-3-opus-latest'
+    assert model.client is not None