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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pydantic-ai
-Version: 0.7.0
+Version: 0.7.2
 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.2
 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.2; 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.2}/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.2}/tests/conftest.py

@@ -24,6 +24,7 @@ from typing_extensions import TypeAlias
 from vcr import VCR, request as vcr_request
 
 import pydantic_ai.models
+from pydantic_ai import Agent
 from pydantic_ai.messages import BinaryContent
 from pydantic_ai.models import Model, cached_async_http_client
 
@@ -230,6 +231,24 @@ def event_loop() -> Iterator[None]:
     new_loop.close()
 
 
+@pytest.fixture(autouse=True)
+def no_instrumentation_by_default():
+    Agent.instrument_all(False)
+
+
+try:
+    import logfire
+
+    logfire.DEFAULT_LOGFIRE_INSTANCE.config.ignore_no_config = True
+
+    @pytest.fixture(autouse=True)
+    def fresh_logfire():
+        logfire.shutdown(flush=False)
+
+except ImportError:
+    pass
+
+
 def raise_if_exception(e: Any) -> None:
     if isinstance(e, Exception):
         raise e

{pydantic_ai-0.7.0 → pydantic_ai-0.7.2}/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.0 → pydantic_ai-0.7.2}/tests/evals/test_otel.py

@@ -336,9 +336,6 @@ async def test_span_node_repr(span_tree: SpanTree):
 
 async def test_span_tree_ancestors_methods():
     """Test the ancestor traversal methods in SpanNode."""
-    # Configure logfire
-    logfire.configure()
-
     # Create spans with a deep structure for testing ancestor methods
     with context_subtree() as tree:
         with logfire.span('root', depth=0):
@@ -398,9 +395,6 @@ async def test_span_tree_ancestors_methods():
 
 async def test_span_tree_descendants_methods():
     """Test the descendant traversal methods in SpanNode."""
-    # Configure logfire
-    logfire.configure()
-
     # Create spans with a deep structure for testing descendant methods
     with context_subtree() as tree:
         with logfire.span('root', depth=0):
@@ -488,9 +482,6 @@ async def test_span_tree_descendants_methods():
 
 async def test_log_levels_and_exceptions():
     """Test recording different log levels and exceptions in spans."""
-    # Configure logfire
-    logfire.configure()
-
     with context_subtree() as tree:
         # Test different log levels
         with logfire.span('parent_span'):
@@ -891,7 +882,6 @@ async def test_context_subtree_not_configured(mocker: MockerFixture):
     """Test that context_subtree correctly records spans in independent async contexts."""
     from opentelemetry.trace import ProxyTracerProvider
 
-    # from opentelemetry.sdk.trace import TracerProvider
     mocker.patch(
         'pydantic_evals.otel._context_in_memory_span_exporter.get_tracer_provider', return_value=ProxyTracerProvider()
     )

{pydantic_ai-0.7.0 → pydantic_ai-0.7.2}/tests/models/cassettes/test_anthropic/test_anthropic_code_execution_tool.yaml

@@ -15,7 +15,7 @@ interactions:
       - api.anthropic.com
     method: POST
     parsed_body:
-      max_tokens: 1024
+      max_tokens: 4096
       messages:
       - content:
         - text: How much is 3 * 12390?
@@ -34,7 +34,7 @@ interactions:
       connection:
       - keep-alive
       content-length:
-      - '927'
+      - '994'
       content-type:
       - application/json
       strict-transport-security:
@@ -43,12 +43,12 @@ interactions:
       - chunked
     parsed_body:
       container:
-        expires_at: '2025-05-26T13:30:45.703429+00:00'
-        id: container_011CPW9LpfbF8dmXMvVNCiQJ
+        expires_at: '2025-08-14T12:35:01.802902Z'
+        id: container_011CS7XodG8mEYj1PnST46H5
       content:
       - text: I'll calculate 3 * 12390 for you.
         type: text
-      - id: srvtoolu_01CPfaeVC7ju4VsdzxjSLDrY
+      - id: srvtoolu_01PZTDQyDyxVNeVikBoNo2fb
         input:
           code: |-
             result = 3 * 12390
@@ -62,21 +62,24 @@ interactions:
           stdout: |
             3 * 12390 = 37170
           type: code_execution_result
-        tool_use_id: srvtoolu_01CPfaeVC7ju4VsdzxjSLDrY
+        tool_use_id: srvtoolu_01PZTDQyDyxVNeVikBoNo2fb
         type: code_execution_tool_result
-      - text: The answer is **37,170**.
+      - text: 3 * 12390 = 37,170
         type: text
-      id: msg_015H6Emn2T8vZhE52mU2jF1U
+      id: msg_01RJnbK7VMxvS2SyvtyJAQVU
       model: claude-sonnet-4-20250514
       role: assistant
       stop_reason: end_turn
       stop_sequence: null
       type: message
       usage:
+        cache_creation:
+          ephemeral_1h_input_tokens: 0
+          ephemeral_5m_input_tokens: 0
         cache_creation_input_tokens: 0
         cache_read_input_tokens: 0
         input_tokens: 1630
-        output_tokens: 105
+        output_tokens: 109
         server_tool_use:
           web_search_requests: 0
         service_tier: standard

{pydantic_ai-0.7.0 → pydantic_ai-0.7.2}/tests/models/cassettes/test_anthropic/test_anthropic_model_empty_message_on_history.yaml

@@ -15,7 +15,7 @@ interactions:
       - api.anthropic.com
     method: POST
     parsed_body:
-      max_tokens: 1024
+      max_tokens: 4096
       messages:
       - content:
         - text: Hello, how can I help you?
@@ -36,7 +36,7 @@ interactions:
       connection:
       - keep-alive
       content-length:
-      - '671'
+      - '777'
       content-type:
       - application/json
       strict-transport-security:
@@ -46,26 +46,29 @@ interactions:
     parsed_body:
       content:
       - text: |-
-          I can't physically give you a potato since I'm a computer program. However, I can:
+          I can't physically give you a potato since I'm a digital assistant. However, I can:
 
           1. Help you find recipes that use potatoes
-          2. Give you tips on how to select, store, or cook potatoes
-          3. Share information about different potato varieties
-          4. Provide guidance on growing potatoes
+          2. Give you tips on how to select, store, or prepare potatoes
+          3. Share information about different types of potatoes
+          4. Suggest where you might buy potatoes locally
 
-          What specifically would you like to know about potatoes?
+          What specific information about potatoes would be most helpful to you?
         type: text
-      id: msg_01UjnDmX3B57Drosu49sMteT
+      id: msg_01PAZFa5ciacA9ptgEDMbkZM
       model: claude-3-5-sonnet-20241022
       role: assistant
       stop_reason: end_turn
       stop_sequence: null
       type: message
       usage:
+        cache_creation:
+          ephemeral_1h_input_tokens: 0
+          ephemeral_5m_input_tokens: 0
         cache_creation_input_tokens: 0
         cache_read_input_tokens: 0
         input_tokens: 41
-        output_tokens: 82
+        output_tokens: 88
         service_tier: standard
     status:
       code: 200

{pydantic_ai-0.7.0 → pydantic_ai-0.7.2}/tests/models/cassettes/test_anthropic/test_anthropic_model_instructions.yaml

@@ -15,7 +15,7 @@ interactions:
       - api.anthropic.com
     method: POST
     parsed_body:
-      max_tokens: 1024
+      max_tokens: 4096
       messages:
       - content:
         - text: What is the capital of France?
@@ -32,7 +32,7 @@ interactions:
       connection:
       - keep-alive
       content-length:
-      - '354'
+      - '433'
       content-type:
       - application/json
       strict-transport-security:
@@ -43,13 +43,16 @@ interactions:
       content:
       - text: The capital of France is Paris.
         type: text
-      id: msg_01BznVNBje2zyfpCfNQCD5en
+      id: msg_01Fg1JVgvCYUHWsxrj9GkpEv
       model: claude-3-opus-20240229
       role: assistant
       stop_reason: end_turn
       stop_sequence: null
       type: message
       usage:
+        cache_creation:
+          ephemeral_1h_input_tokens: 0
+          ephemeral_5m_input_tokens: 0
         cache_creation_input_tokens: 0
         cache_read_input_tokens: 0
         input_tokens: 20