pydantic-ai 0.7.0__tar.gz → 0.7.1__tar.gz

{pydantic_ai-0.7.0 → pydantic_ai-0.7.1}/PKG-INFO

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

{pydantic_ai-0.7.0 → pydantic_ai-0.7.1}/pyproject.toml

@@ -102,6 +102,9 @@ dev = [
     "boto3-stubs[bedrock-runtime]",
     "strict-no-cover @ git+https://github.com/pydantic/strict-no-cover.git@7fc59da2c4dff919db2095a0f0e47101b657131d",
     "pytest-xdist>=3.6.1",
+    # Needed for PyCharm users
+    "pip>=25.2",
+    "genai-prices>=0.0.22",
 ]
 lint = ["mypy>=1.11.2", "pyright>=1.1.390", "ruff>=0.6.9"]
 docs = [

{pydantic_ai-0.7.0 → pydantic_ai-0.7.1}/tests/evals/test_dataset.py

@@ -7,7 +7,7 @@ from pathlib import Path
 from typing import Any
 
 import pytest
-from dirty_equals import HasRepr, IsNumber, IsPartialDict
+from dirty_equals import HasRepr, IsNumber
 from inline_snapshot import snapshot
 from pydantic import BaseModel, TypeAdapter
 
@@ -1172,115 +1172,238 @@ async def test_evaluate_async_logfire(
 
     await example_dataset.evaluate(mock_async_task)
 
-    spans = capfire.exporter.exported_spans_as_dict()
+    spans = capfire.exporter.exported_spans_as_dict(parse_json_attributes=True)
     spans.sort(key=lambda s: s['start_time'])
-    assert spans == [
-        {
-            'attributes': {
-                'averages': '{"name":"Averages","scores":{"confidence":1.0},"labels":{},"metrics":{},"assertions":1.0,"task_duration":1.0,"total_duration":5.0}',
-                'cases': '[{"name":"case1","inputs":{"query":"What is '
-                '2+2?"},"metadata":{"difficulty":"easy","category":"general"},"expected_output":{"answer":"4","confidence":1.0},"output":{"answer":"4","confidence":1.0},"metrics":{},"attributes":{},"scores":{"confidence":{"name":"confidence","value":1.0,"reason":null,"source":{"name":"SimpleEvaluator","arguments":null}}},"labels":{},"assertions":{"correct":{"name":"correct","value":true,"reason":null,"source":{"name":"SimpleEvaluator","arguments":null}}},"task_duration":1.0,"total_duration":6.0,"trace_id":"00000000000000000000000000000001","span_id":"0000000000000003"},{"name":"case2","inputs":{"query":"What '
-                'is the capital of '
-                'France?"},"metadata":{"difficulty":"medium","category":"geography"},"expected_output":{"answer":"Paris","confidence":1.0},"output":{"answer":"Paris","confidence":1.0},"metrics":{},"attributes":{},"scores":{"confidence":{"name":"confidence","value":1.0,"reason":null,"source":{"name":"SimpleEvaluator","arguments":null}}},"labels":{},"assertions":{"correct":{"name":"correct","value":true,"reason":null,"source":{"name":"SimpleEvaluator","arguments":null}}},"task_duration":1.0,"total_duration":4.0,"trace_id":"00000000000000000000000000000001","span_id":"0000000000000007"}]',
-                'code.filepath': 'test_dataset.py',
-                'code.function': 'test_evaluate_async_logfire',
-                'code.lineno': 123,
-                'logfire.json_schema': '{"type":"object","properties":{"name":{},"cases":{"type":"array"},"averages":{"type":"object"}}}',
-                'logfire.msg': 'evaluate mock_async_task',
-                'logfire.msg_template': 'evaluate {name}',
-                'logfire.span_type': 'span',
-                'name': 'mock_async_task',
-            },
-            'context': {'is_remote': False, 'span_id': 1, 'trace_id': 1},
-            'end_time': 10000000000,
-            'name': 'evaluate {name}',
-            'parent': None,
-            'start_time': 1000000000,
-        },
-        IsPartialDict(
-            {
-                'attributes': {
-                    'assertions': '{"correct":{"name":"correct","value":true,"reason":null,"source":{"name":"SimpleEvaluator","arguments":null}}}',
-                    'attributes': '{}',
+
+    for span in spans:
+        # These may or may not be present and may have weird values due to things running in async
+        span['attributes'].pop('code.filepath', None)
+        span['attributes'].pop('code.function', None)
+        span['attributes'].pop('code.lineno', None)
+
+    assert [(span['name'], span['attributes']) for span in spans] == snapshot(
+        [
+            (
+                'evaluate {name}',
+                {
+                    'name': 'mock_async_task',
+                    'logfire.msg_template': 'evaluate {name}',
+                    'logfire.msg': 'evaluate mock_async_task',
+                    'logfire.span_type': 'span',
+                    'cases': [
+                        {
+                            'name': 'case1',
+                            'inputs': {'query': 'What is 2+2?'},
+                            'metadata': {'difficulty': 'easy', 'category': 'general'},
+                            'expected_output': {'answer': '4', 'confidence': 1.0},
+                            'output': {'answer': '4', 'confidence': 1.0},
+                            'metrics': {},
+                            'attributes': {},
+                            'scores': {
+                                'confidence': {
+                                    'name': 'confidence',
+                                    'value': 1.0,
+                                    'reason': None,
+                                    'source': {'name': 'SimpleEvaluator', 'arguments': None},
+                                }
+                            },
+                            'labels': {},
+                            'assertions': {
+                                'correct': {
+                                    'name': 'correct',
+                                    'value': True,
+                                    'reason': None,
+                                    'source': {'name': 'SimpleEvaluator', 'arguments': None},
+                                }
+                            },
+                            'task_duration': 1.0,
+                            'total_duration': 6.0,
+                            'trace_id': '00000000000000000000000000000001',
+                            'span_id': '0000000000000003',
+                        },
+                        {
+                            'name': 'case2',
+                            'inputs': {'query': 'What is the capital of France?'},
+                            'metadata': {'difficulty': 'medium', 'category': 'geography'},
+                            'expected_output': {'answer': 'Paris', 'confidence': 1.0},
+                            'output': {'answer': 'Paris', 'confidence': 1.0},
+                            'metrics': {},
+                            'attributes': {},
+                            'scores': {
+                                'confidence': {
+                                    'name': 'confidence',
+                                    'value': 1.0,
+                                    'reason': None,
+                                    'source': {'name': 'SimpleEvaluator', 'arguments': None},
+                                }
+                            },
+                            'labels': {},
+                            'assertions': {
+                                'correct': {
+                                    'name': 'correct',
+                                    'value': True,
+                                    'reason': None,
+                                    'source': {'name': 'SimpleEvaluator', 'arguments': None},
+                                }
+                            },
+                            'task_duration': 1.0,
+                            'total_duration': 4.0,
+                            'trace_id': '00000000000000000000000000000001',
+                            'span_id': '0000000000000007',
+                        },
+                    ],
+                    'averages': {
+                        'name': 'Averages',
+                        'scores': {'confidence': 1.0},
+                        'labels': {},
+                        'metrics': {},
+                        'assertions': 1.0,
+                        'task_duration': 1.0,
+                        'total_duration': 5.0,
+                    },
+                    'logfire.json_schema': {
+                        'type': 'object',
+                        'properties': {'name': {}, 'cases': {'type': 'array'}, 'averages': {'type': 'object'}},
+                    },
+                },
+            ),
+            (
+                'case: {case_name}',
+                {
+                    'task_name': 'mock_async_task',
                     'case_name': 'case1',
-                    'expected_output': '{"answer":"4","confidence":1.0}',
-                    'inputs': '{"query":"What is 2+2?"}',
-                    'labels': '{}',
-                    'logfire.json_schema': '{"type":"object","properties":{"task_name":{},"case_name":{},"inputs":{"type":"object","title":"TaskInput","x-python-datatype":"PydanticModel"},"metadata":{"type":"object","title":"TaskMetadata","x-python-datatype":"PydanticModel"},"expected_output":{"type":"object","title":"TaskOutput","x-python-datatype":"PydanticModel"},"output":{"type":"object","title":"TaskOutput","x-python-datatype":"PydanticModel"},"task_duration":{},"metrics":{"type":"object"},"attributes":{"type":"object"},"assertions":{"type":"object"},"scores":{"type":"object"},"labels":{"type":"object"}}}',
-                    'logfire.msg': 'case: case1',
+                    'inputs': {'query': 'What is 2+2?'},
+                    'metadata': {'difficulty': 'easy', 'category': 'general'},
+                    'expected_output': {'answer': '4', 'confidence': 1.0},
                     'logfire.msg_template': 'case: {case_name}',
+                    'logfire.msg': 'case: case1',
                     'logfire.span_type': 'span',
-                    'metadata': '{"difficulty":"easy","category":"general"}',
-                    'metrics': '{}',
-                    'output': '{"answer":"4","confidence":1.0}',
-                    'scores': '{"confidence":{"name":"confidence","value":1.0,"reason":null,"source":{"name":"SimpleEvaluator","arguments":null}}}',
+                    'output': {'answer': '4', 'confidence': 1.0},
                     'task_duration': 1.0,
-                    'task_name': 'mock_async_task',
+                    'metrics': {},
+                    'attributes': {},
+                    'assertions': {
+                        'correct': {
+                            'name': 'correct',
+                            'value': True,
+                            'reason': None,
+                            'source': {'name': 'SimpleEvaluator', 'arguments': None},
+                        }
+                    },
+                    'scores': {
+                        'confidence': {
+                            'name': 'confidence',
+                            'value': 1.0,
+                            'reason': None,
+                            'source': {'name': 'SimpleEvaluator', 'arguments': None},
+                        }
+                    },
+                    'labels': {},
+                    'logfire.json_schema': {
+                        'type': 'object',
+                        'properties': {
+                            'task_name': {},
+                            'case_name': {},
+                            'inputs': {'type': 'object', 'title': 'TaskInput', 'x-python-datatype': 'PydanticModel'},
+                            'metadata': {
+                                'type': 'object',
+                                'title': 'TaskMetadata',
+                                'x-python-datatype': 'PydanticModel',
+                            },
+                            'expected_output': {
+                                'type': 'object',
+                                'title': 'TaskOutput',
+                                'x-python-datatype': 'PydanticModel',
+                            },
+                            'output': {'type': 'object', 'title': 'TaskOutput', 'x-python-datatype': 'PydanticModel'},
+                            'task_duration': {},
+                            'metrics': {'type': 'object'},
+                            'attributes': {'type': 'object'},
+                            'assertions': {'type': 'object'},
+                            'scores': {'type': 'object'},
+                            'labels': {'type': 'object'},
+                        },
+                    },
                 },
-                'context': {'is_remote': False, 'span_id': 3, 'trace_id': 1},
-                'end_time': 8000000000,
-                'name': 'case: {case_name}',
-                'parent': {'is_remote': False, 'span_id': 1, 'trace_id': 1},
-                'start_time': 2000000000,
-            }
-        ),
-        IsPartialDict(
-            {
-                'attributes': {
-                    'logfire.json_schema': '{"type":"object","properties":{"task":{}}}',
-                    'logfire.msg': 'execute mock_async_task',
+            ),
+            (
+                'execute {task}',
+                {
+                    'task': 'mock_async_task',
                     'logfire.msg_template': 'execute {task}',
+                    'logfire.msg': 'execute mock_async_task',
+                    'logfire.json_schema': {'type': 'object', 'properties': {'task': {}}},
                     'logfire.span_type': 'span',
-                    'task': 'mock_async_task',
                 },
-                'context': {'is_remote': False, 'span_id': 5, 'trace_id': 1},
-                'end_time': 4000000000,
-                'name': 'execute {task}',
-                'parent': {'is_remote': False, 'span_id': 3, 'trace_id': 1},
-                'start_time': 3000000000,
-            }
-        ),
-        IsPartialDict(
-            {
-                'attributes': {
-                    'assertions': '{"correct":{"name":"correct","value":true,"reason":null,"source":{"name":"SimpleEvaluator","arguments":null}}}',
-                    'attributes': '{}',
+            ),
+            (
+                'case: {case_name}',
+                {
+                    'task_name': 'mock_async_task',
                     'case_name': 'case2',
-                    'expected_output': '{"answer":"Paris","confidence":1.0}',
-                    'inputs': '{"query":"What is the capital of France?"}',
-                    'labels': '{}',
-                    'logfire.json_schema': '{"type":"object","properties":{"task_name":{},"case_name":{},"inputs":{"type":"object","title":"TaskInput","x-python-datatype":"PydanticModel"},"metadata":{"type":"object","title":"TaskMetadata","x-python-datatype":"PydanticModel"},"expected_output":{"type":"object","title":"TaskOutput","x-python-datatype":"PydanticModel"},"output":{"type":"object","title":"TaskOutput","x-python-datatype":"PydanticModel"},"task_duration":{},"metrics":{"type":"object"},"attributes":{"type":"object"},"assertions":{"type":"object"},"scores":{"type":"object"},"labels":{"type":"object"}}}',
-                    'logfire.msg': 'case: case2',
+                    'inputs': {'query': 'What is the capital of France?'},
+                    'metadata': {'difficulty': 'medium', 'category': 'geography'},
+                    'expected_output': {'answer': 'Paris', 'confidence': 1.0},
                     'logfire.msg_template': 'case: {case_name}',
+                    'logfire.msg': 'case: case2',
                     'logfire.span_type': 'span',
-                    'metadata': '{"difficulty":"medium","category":"geography"}',
-                    'metrics': '{}',
-                    'output': '{"answer":"Paris","confidence":1.0}',
-                    'scores': '{"confidence":{"name":"confidence","value":1.0,"reason":null,"source":{"name":"SimpleEvaluator","arguments":null}}}',
+                    'output': {'answer': 'Paris', 'confidence': 1.0},
                     'task_duration': 1.0,
-                    'task_name': 'mock_async_task',
+                    'metrics': {},
+                    'attributes': {},
+                    'assertions': {
+                        'correct': {
+                            'name': 'correct',
+                            'value': True,
+                            'reason': None,
+                            'source': {'name': 'SimpleEvaluator', 'arguments': None},
+                        }
+                    },
+                    'scores': {
+                        'confidence': {
+                            'name': 'confidence',
+                            'value': 1.0,
+                            'reason': None,
+                            'source': {'name': 'SimpleEvaluator', 'arguments': None},
+                        }
+                    },
+                    'labels': {},
+                    'logfire.json_schema': {
+                        'type': 'object',
+                        'properties': {
+                            'task_name': {},
+                            'case_name': {},
+                            'inputs': {'type': 'object', 'title': 'TaskInput', 'x-python-datatype': 'PydanticModel'},
+                            'metadata': {
+                                'type': 'object',
+                                'title': 'TaskMetadata',
+                                'x-python-datatype': 'PydanticModel',
+                            },
+                            'expected_output': {
+                                'type': 'object',
+                                'title': 'TaskOutput',
+                                'x-python-datatype': 'PydanticModel',
+                            },
+                            'output': {'type': 'object', 'title': 'TaskOutput', 'x-python-datatype': 'PydanticModel'},
+                            'task_duration': {},
+                            'metrics': {'type': 'object'},
+                            'attributes': {'type': 'object'},
+                            'assertions': {'type': 'object'},
+                            'scores': {'type': 'object'},
+                            'labels': {'type': 'object'},
+                        },
+                    },
                 },
-                'context': {'is_remote': False, 'span_id': 7, 'trace_id': 1},
-                'end_time': 9000000000,
-                'name': 'case: {case_name}',
-                'parent': {'is_remote': False, 'span_id': 1, 'trace_id': 1},
-                'start_time': 5000000000,
-            }
-        ),
-        IsPartialDict(
-            {
-                'attributes': {
-                    'logfire.json_schema': '{"type":"object","properties":{"task":{}}}',
-                    'logfire.msg': 'execute mock_async_task',
+            ),
+            (
+                'execute {task}',
+                {
+                    'task': 'mock_async_task',
                     'logfire.msg_template': 'execute {task}',
+                    'logfire.msg': 'execute mock_async_task',
+                    'logfire.json_schema': {'type': 'object', 'properties': {'task': {}}},
                     'logfire.span_type': 'span',
-                    'task': 'mock_async_task',
                 },
-                'context': {'is_remote': False, 'span_id': 9, 'trace_id': 1},
-                'end_time': 7000000000,
-                'name': 'execute {task}',
-                'parent': {'is_remote': False, 'span_id': 7, 'trace_id': 1},
-                'start_time': 6000000000,
-            }
-        ),
-    ]
+            ),
+        ]
+    )

pydantic_ai-0.7.1/tests/models/cassettes/test_google/test_google_model_usage_limit_exceeded.yaml

@@ -0,0 +1,46 @@
+interactions:
+- request:
+    body: '{"contents": [{"parts": [{"text": "The quick brown fox jumps over the lazydog."}],
+      "role": "user"}]}'
+    headers:
+      Content-Type:
+      - application/json
+      user-agent:
+      - google-genai-sdk/1.26.0 gl-python/3.12.7
+      x-goog-api-client:
+      - google-genai-sdk/1.26.0 gl-python/3.12.7
+    method: post
+    uri: https://generativelanguage.googleapis.com/v1beta/models/gemini-2.5-flash:countTokens
+  response:
+    body:
+      string: "{\n  \"totalTokens\": 12,\n  \"promptTokensDetails\": [\n    {\n      \"modality\":
+        \"TEXT\",\n      \"tokenCount\": 12\n    }\n  ]\n}\n"
+    headers:
+      Alt-Svc:
+      - h3=":443"; ma=2592000,h3-29=":443"; ma=2592000
+      Content-Type:
+      - application/json; charset=UTF-8
+      Date:
+      - Fri, 01 Aug 2025 15:59:25 GMT
+      Server:
+      - scaffolding on HTTPServer2
+      Server-Timing:
+      - gfet4t7; dur=1582
+      Transfer-Encoding:
+      - chunked
+      Vary:
+      - Origin
+      - X-Origin
+      - Referer
+      X-Content-Type-Options:
+      - nosniff
+      X-Frame-Options:
+      - SAMEORIGIN
+      X-XSS-Protection:
+      - '0'
+      content-length:
+      - '117'
+    status:
+      code: 200
+      message: OK
+version: 1

pydantic_ai-0.7.1/tests/models/cassettes/test_google/test_google_model_usage_limit_not_exceeded.yaml

@@ -0,0 +1,115 @@
+interactions:
+- request:
+    headers:
+      accept:
+      - '*/*'
+      accept-encoding:
+      - gzip, deflate
+      connection:
+      - keep-alive
+      content-length:
+      - '100'
+      content-type:
+      - application/json
+      host:
+      - generativelanguage.googleapis.com
+    method: POST
+    parsed_body:
+      contents:
+      - parts:
+        - text: The quick brown fox jumps over the lazydog.
+        role: user
+    uri: https://generativelanguage.googleapis.com/v1beta/models/gemini-2.5-flash:countTokens
+  response:
+    headers:
+      alt-svc:
+      - h3=":443"; ma=2592000,h3-29=":443"; ma=2592000
+      content-length:
+      - '117'
+      content-type:
+      - application/json; charset=UTF-8
+      server-timing:
+      - gfet4t7; dur=191
+      transfer-encoding:
+      - chunked
+      vary:
+      - Origin
+      - X-Origin
+      - Referer
+    parsed_body:
+      promptTokensDetails:
+      - modality: TEXT
+        tokenCount: 12
+      totalTokens: 12
+    status:
+      code: 200
+      message: OK
+- request:
+    headers:
+      accept:
+      - '*/*'
+      accept-encoding:
+      - gzip, deflate
+      connection:
+      - keep-alive
+      content-length:
+      - '124'
+      content-type:
+      - application/json
+      host:
+      - generativelanguage.googleapis.com
+    method: POST
+    parsed_body:
+      contents:
+      - parts:
+        - text: The quick brown fox jumps over the lazydog.
+        role: user
+      generationConfig: {}
+    uri: https://generativelanguage.googleapis.com/v1beta/models/gemini-2.5-flash:generateContent
+  response:
+    headers:
+      alt-svc:
+      - h3=":443"; ma=2592000,h3-29=":443"; ma=2592000
+      content-length:
+      - '979'
+      content-type:
+      - application/json; charset=UTF-8
+      server-timing:
+      - gfet4t7; dur=4808
+      transfer-encoding:
+      - chunked
+      vary:
+      - Origin
+      - X-Origin
+      - Referer
+    parsed_body:
+      candidates:
+      - content:
+          parts:
+          - text: |-
+              That's a classic! It's famously known as a **pangram**, which means it's a sentence that contains every letter of the alphabet.
+
+              It's often used for:
+              *   **Typing practice:** To ensure all keys are hit.
+              *   **Displaying font samples:** Because it showcases every character.
+
+              Just a small note, it's typically written as "lazy dog" (two words) and usually ends with a period:
+
+              **The quick brown fox jumps over the lazy dog.**
+          role: model
+        finishReason: STOP
+        index: 0
+      modelVersion: gemini-2.5-flash
+      responseId: ZwudaISALoquqtsP9uCG6Qw
+      usageMetadata:
+        candidatesTokenCount: 109
+        promptTokenCount: 12
+        promptTokensDetails:
+        - modality: TEXT
+          tokenCount: 12
+        thoughtsTokenCount: 806
+        totalTokenCount: 927
+    status:
+      code: 200
+      message: OK
+version: 1

pydantic_ai-0.7.1/tests/models/cassettes/test_google/test_google_vertexai_model_usage_limit_exceeded.yaml

@@ -0,0 +1,85 @@
+interactions:
+- request:
+    body: assertion=eyJ0eXAiOiAiSldUIiwgImFsZyI6ICJSUzI1NiIsICJraWQiOiAiNjc4MmViNzhmMGU0MmIwMjRkNjExNjY2MTg0NWZhZTdkOTdiMDY3NyJ9.eyJpYXQiOiAxNzU1MTIyMTU5LCAiZXhwIjogMTc1NTEyNTc1OSwgImlzcyI6ICJweWRhbnRpYy1haS12ZXJ0ZXgtYWktZGV2QHB5ZGFudGljLWFpLWRldi5pYW0uZ3NlcnZpY2VhY2NvdW50LmNvbSIsICJhdWQiOiAiaHR0cHM6Ly9vYXV0aDIuZ29vZ2xlYXBpcy5jb20vdG9rZW4iLCAic2NvcGUiOiAiaHR0cHM6Ly93d3cuZ29vZ2xlYXBpcy5jb20vYXV0aC9jbG91ZC1wbGF0Zm9ybSJ9.pnrHg7uMo_ow8OctKf4GM7Oe28kUhvzKewq0xCeaJz88bSeorgyd5ZLRtRdRhgZwiznSCevAahorDTxCVeCxenJp88UHf_5tuN6Jo7e7dcrzjA3Ta3DbQgBc_N58cGWc75JgJpE6GwDtHjehLVtIlo_qpux-FqLZL5N745-ECeKnyKmMTdkuVVArMyU96CGOss9L3xiWKukRIv-h8_uCmsoPVWQmyGEfQQOv--EIC4ERUOfwed1I-4ZyF5Z8w3dymKTHtJoRSL-fg8tn12acytLptC1eEZ2OzkZJI4UJWHmmtAy8HrZ-PqjwUiO83sC5iZcLICmCxXWUxJce5j-d_Q&grant_type=urn%3Aietf%3Aparams%3Aoauth%3Agrant-type%3Ajwt-bearer
+    headers:
+      accept:
+      - '*/*'
+      accept-encoding:
+      - gzip, deflate
+      connection:
+      - keep-alive
+      content-length:
+      - '819'
+      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
+      content-length:
+      - '1083'
+      content-type:
+      - application/json; charset=UTF-8
+      transfer-encoding:
+      - chunked
+      vary:
+      - Origin
+      - X-Origin
+      - Referer
+    parsed_body:
+      access_token: scrubbed
+      expires_in: 3599
+      token_type: Bearer
+    status:
+      code: 200
+      message: OK
+- request:
+    headers:
+      content-type:
+      - application/json
+    method: post
+    parsed_body:
+      contents:
+      - parts:
+        - text: What is the largest city in the user country? Use the get_user_country tool and then your own world knowledge.
+        role: user
+      generationConfig:
+        max_output_tokens: 100
+      systemInstruction:
+        parts:
+        - text: You are a chatbot.
+        role: user
+      tools:
+      - functionDeclarations:
+        - description: ''
+          name: get_user_country
+          parameters:
+            properties: {}
+            type: OBJECT
+    uri: https://us-central1-aiplatform.googleapis.com/v1beta1/projects/pydantic-ai/locations/us-central1/publishers/google/models/gemini-2.0-flash:countTokens
+  response:
+    headers:
+      alt-svc:
+      - h3=":443"; ma=2592000,h3-29=":443"; ma=2592000
+      content-length:
+      - '151'
+      content-type:
+      - application/json; charset=UTF-8
+      transfer-encoding:
+      - chunked
+      vary:
+      - Origin
+      - X-Origin
+      - Referer
+    parsed_body:
+      promptTokensDetails:
+      - modality: TEXT
+        tokenCount: 36
+      totalBillableCharacters: 123
+      totalTokens: 36
+    status:
+      code: 200
+      message: OK
+version: 1

{pydantic_ai-0.7.0 → pydantic_ai-0.7.1}/tests/models/cassettes/test_model_names/test_known_model_names.yaml

@@ -14,12 +14,20 @@ interactions:
     uri: https://us.inference.heroku.com/available-models
   response:
     headers:
+      cache-control:
+      - no-cache, no-store, must-revalidate
+      connection:
+      - keep-alive
       content-length:
       - '760'
       content-security-policy:
       - default-src 'none'; frame-ancestors 'none'
       content-type:
       - application/json
+      expires:
+      - '0'
+      pragma:
+      - no-cache
       strict-transport-security:
       - max-age=63072000
     parsed_body: