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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pydantic-ai
-Version: 0.0.43
+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.43
+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.43; 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.43 → pydantic_ai-0.0.45}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "pydantic-ai"
-version = "0.0.43"
+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.43",
+    "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.43"]
+examples = ["pydantic-ai-examples==0.0.45"]
 logfire = ["logfire>=2.3"]
 
 [tool.uv.sources]

{pydantic_ai-0.0.43 → 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:
@@ -246,6 +246,16 @@ def anthropic_api_key() -> str:
     return os.getenv('ANTHROPIC_API_KEY', 'mock-api-key')
 
 
+@pytest.fixture(scope='session')
+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.45/tests/graph/test_utils.py

@@ -0,0 +1,20 @@
+from threading import Thread
+
+from pydantic_graph._utils import run_until_complete
+
+
+def test_run_until_complete_in_main_thread():
+    async def run(): ...
+
+    run_until_complete(run())
+
+
+def test_run_until_complete_in_thread():
+    async def run(): ...
+
+    def get_and_close_event_loop():
+        run_until_complete(run())
+
+    thread = Thread(target=get_and_close_event_loop)
+    thread.start()
+    thread.join()

pydantic_ai-0.0.45/tests/models/cassettes/test_cohere/test_request_simple_success_with_vcr.yaml

@@ -0,0 +1,64 @@
+interactions:
+- request:
+    headers:
+      accept:
+      - '*/*'
+      accept-encoding:
+      - gzip, deflate
+      connection:
+      - keep-alive
+      content-length:
+      - '93'
+      content-type:
+      - application/json
+      host:
+      - api.cohere.com
+    method: POST
+    parsed_body:
+      messages:
+      - content: hello
+        role: user
+      model: command-r7b-12-2024
+      stream: false
+    uri: https://api.cohere.com/v2/chat
+  response:
+    headers:
+      access-control-expose-headers:
+      - X-Debug-Trace-ID
+      alt-svc:
+      - h3=":443"; ma=2592000,h3-29=":443"; ma=2592000
+      cache-control:
+      - no-cache, no-store, no-transform, must-revalidate, private, max-age=0
+      content-length:
+      - '286'
+      content-type:
+      - application/json
+      expires:
+      - Thu, 01 Jan 1970 00:00:00 UTC
+      num_chars:
+      - '2583'
+      num_tokens:
+      - '10'
+      pragma:
+      - no-cache
+      vary:
+      - Origin
+    parsed_body:
+      finish_reason: COMPLETE
+      id: f17a5f6c-1734-4098-bd0d-733ef000ac7b
+      message:
+        content:
+        - text: Hello! How can I assist you today?
+          type: text
+        role: assistant
+      usage:
+        billed_units:
+          input_tokens: 1
+          output_tokens: 9
+        tokens:
+          input_tokens: 496
+          output_tokens: 11
+    status:
+      code: 200
+      message: OK
+version: 1

{pydantic_ai-0.0.43 → 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.43 → 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_max_completion_tokens[gpt-4.5-preview].yaml

@@ -0,0 +1,78 @@
+interactions:
+- request:
+    headers:
+      accept:
+      - application/json
+      accept-encoding:
+      - gzip, deflate
+      connection:
+      - keep-alive
+      content-length:
+      - '123'
+      content-type:
+      - application/json
+      host:
+      - api.openai.com
+    method: POST
+    parsed_body:
+      max_completion_tokens: 100
+      messages:
+      - content: hello
+        role: user
+      model: gpt-4.5-preview
+      n: 1
+      stream: false
+    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:
+      - '807'
+      content-type:
+      - application/json
+      openai-organization:
+      - pydantic-28gund
+      openai-processing-ms:
+      - '1408'
+      openai-version:
+      - '2020-10-01'
+      strict-transport-security:
+      - max-age=31536000; includeSubDomains; preload
+      transfer-encoding:
+      - chunked
+    parsed_body:
+      choices:
+      - finish_reason: stop
+        index: 0
+        message:
+          annotations: []
+          content: Hello! How can I help you today?
+          refusal: null
+          role: assistant
+      created: 1742636225
+      id: chatcmpl-BDpZplWguNLn40wA5mIpCR3OIzvYP
+      model: gpt-4.5-preview-2025-02-27
+      object: chat.completion
+      service_tier: default
+      system_fingerprint: null
+      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.45/tests/models/cassettes/test_openai/test_max_completion_tokens[gpt-4o-mini].yaml

@@ -0,0 +1,79 @@
+interactions:
+- request:
+    headers:
+      accept:
+      - application/json
+      accept-encoding:
+      - gzip, deflate
+      connection:
+      - keep-alive
+      content-length:
+      - '119'
+      content-type:
+      - application/json
+      host:
+      - api.openai.com
+    method: POST
+    parsed_body:
+      max_completion_tokens: 100
+      messages:
+      - content: hello
+        role: user
+      model: gpt-4o-mini
+      n: 1
+      stream: false
+    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:
+      - '840'
+      content-type:
+      - application/json
+      openai-organization:
+      - pydantic-28gund
+      openai-processing-ms:
+      - '278'
+      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: 1742636224
+      id: chatcmpl-BDpZoxw4i90ZesN8iyrwLmGWRJ5lz
+      model: gpt-4o-mini-2024-07-18
+      object: chat.completion
+      service_tier: default
+      system_fingerprint: fp_b8bc95a0ac
+      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.45/tests/models/cassettes/test_openai/test_max_completion_tokens[o3-mini].yaml

@@ -0,0 +1,78 @@
+interactions:
+- request:
+    headers:
+      accept:
+      - application/json
+      accept-encoding:
+      - gzip, deflate
+      connection:
+      - keep-alive
+      content-length:
+      - '115'
+      content-type:
+      - application/json
+      host:
+      - api.openai.com
+    method: POST
+    parsed_body:
+      max_completion_tokens: 100
+      messages:
+      - content: hello
+        role: user
+      model: o3-mini
+      n: 1
+      stream: false
+    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:
+      - '817'
+      content-type:
+      - application/json
+      openai-organization:
+      - pydantic-28gund
+      openai-processing-ms:
+      - '1895'
+      openai-version:
+      - '2020-10-01'
+      strict-transport-security:
+      - max-age=31536000; includeSubDomains; preload
+      transfer-encoding:
+      - chunked
+    parsed_body:
+      choices:
+      - finish_reason: stop
+        index: 0
+        message:
+          annotations: []
+          content: Hello there! How can I help you today?
+          refusal: null
+          role: assistant
+      created: 1742636222
+      id: chatcmpl-BDpZm1SiItIXIcDA0xRV9QGZhD97e
+      model: o3-mini-2025-01-31
+      object: chat.completion
+      service_tier: default
+      system_fingerprint: fp_617f206dd9
+      usage:
+        completion_tokens: 85
+        completion_tokens_details:
+          accepted_prediction_tokens: 0
+          audio_tokens: 0
+          reasoning_tokens: 64
+          rejected_prediction_tokens: 0
+        prompt_tokens: 7
+        prompt_tokens_details:
+          audio_tokens: 0
+          cached_tokens: 0
+        total_tokens: 92
+    status:
+      code: 200
+      message: OK
+version: 1