pydantic-ai 0.0.55__tar.gz → 0.1.1__tar.gz

{pydantic_ai-0.0.55 → pydantic_ai-0.1.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pydantic-ai
-Version: 0.0.55
+Version: 0.1.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,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,evals,groq,mcp,mistral,openai,vertexai]==0.0.55
+Requires-Dist: pydantic-ai-slim[anthropic,bedrock,cli,cohere,evals,groq,mcp,mistral,openai,vertexai]==0.1.1
 Provides-Extra: examples
-Requires-Dist: pydantic-ai-examples==0.0.55; extra == 'examples'
+Requires-Dist: pydantic-ai-examples==0.1.1; extra == 'examples'
 Provides-Extra: logfire
 Requires-Dist: logfire>=3.11.0; extra == 'logfire'
 Description-Content-Type: text/markdown
@@ -86,14 +86,14 @@ Designed to make [type checking](https://ai.pydantic.dev/agents/#static-type-che
 Leverages Python's familiar control flow and agent composition to build your AI-driven projects, making it easy to apply standard Python best practices you'd use in any other (non-AI) project.
 
 * __Structured Responses__
-Harnesses the power of [Pydantic](https://docs.pydantic.dev/latest/) to [validate and structure](https://ai.pydantic.dev/results/#structured-result-validation) model outputs, ensuring responses are consistent across runs.
+Harnesses the power of [Pydantic](https://docs.pydantic.dev/latest/) to [validate and structure](https://ai.pydantic.dev/output/#structured-output) model outputs, ensuring responses are consistent across runs.
 
 * __Dependency Injection System__
-Offers an optional [dependency injection](https://ai.pydantic.dev/dependencies/) system to provide data and services to your agent's [system prompts](https://ai.pydantic.dev/agents/#system-prompts), [tools](https://ai.pydantic.dev/tools/) and [result validators](https://ai.pydantic.dev/results/#result-validators-functions).
+Offers an optional [dependency injection](https://ai.pydantic.dev/dependencies/) system to provide data and services to your agent's [system prompts](https://ai.pydantic.dev/agents/#system-prompts), [tools](https://ai.pydantic.dev/tools/) and [output validators](https://ai.pydantic.dev/output/#output-validator-functions).
 This is useful for testing and eval-driven iterative development.
 
 * __Streamed Responses__
-Provides the ability to [stream](https://ai.pydantic.dev/results/#streamed-results) LLM outputs continuously, with immediate validation, ensuring rapid and accurate results.
+Provides the ability to [stream](https://ai.pydantic.dev/output/#streamed-results) LLM outputs continuously, with immediate validation, ensuring rapid and accurate outputs.
 
 * __Graph Support__
 [Pydantic Graph](https://ai.pydantic.dev/graph) provides a powerful way to define graphs using typing hints, this is useful in complex applications where standard control flow can degrade to spaghetti code.
@@ -117,7 +117,7 @@ agent = Agent(
 # Here the exchange should be very short: PydanticAI will send the system prompt and the user query to the LLM,
 # the model will return a text response. See below for a more complex run.
 result = agent.run_sync('Where does "hello world" come from?')
-print(result.data)
+print(result.output)
 """
 The first known use of "hello, world" was in a 1974 textbook about the C programming language.
 """
@@ -150,22 +150,22 @@ class SupportDependencies:
     db: DatabaseConn
 
 
-# This pydantic model defines the structure of the result returned by the agent.
-class SupportResult(BaseModel):
+# This pydantic model defines the structure of the output returned by the agent.
+class SupportOutput(BaseModel):
     support_advice: str = Field(description='Advice returned to the customer')
     block_card: bool = Field(description="Whether to block the customer's card")
     risk: int = Field(description='Risk level of query', ge=0, le=10)
 
 
 # This agent will act as first-tier support in a bank.
-# Agents are generic in the type of dependencies they accept and the type of result they return.
-# In this case, the support agent has type `Agent[SupportDependencies, SupportResult]`.
+# Agents are generic in the type of dependencies they accept and the type of output they return.
+# In this case, the support agent has type `Agent[SupportDependencies, SupportOutput]`.
 support_agent = Agent(
     'openai:gpt-4o',
     deps_type=SupportDependencies,
-    # The response from the agent will, be guaranteed to be a SupportResult,
+    # The response from the agent will, be guaranteed to be a SupportOutput,
     # if validation fails the agent is prompted to try again.
-    result_type=SupportResult,
+    output_type=SupportOutput,
     system_prompt=(
         'You are a support agent in our bank, give the '
         'customer support and judge the risk level of their query.'
@@ -187,7 +187,7 @@ async def add_customer_name(ctx: RunContext[SupportDependencies]) -> str:
 # Pydantic is used to validate these arguments, and errors are passed back to the LLM so it can retry.
 @support_agent.tool
 async def customer_balance(
-    ctx: RunContext[SupportDependencies], include_pending: bool
+        ctx: RunContext[SupportDependencies], include_pending: bool
 ) -> float:
     """Returns the customer's current account balance."""
     # The docstring of a tool is also passed to the LLM as the description of the tool.
@@ -205,17 +205,17 @@ async def customer_balance(
 async def main():
     deps = SupportDependencies(customer_id=123, db=DatabaseConn())
     # Run the agent asynchronously, conducting a conversation with the LLM until a final response is reached.
-    # Even in this fairly simple case, the agent will exchange multiple messages with the LLM as tools are called to retrieve a result.
+    # Even in this fairly simple case, the agent will exchange multiple messages with the LLM as tools are called to retrieve an output.
     result = await support_agent.run('What is my balance?', deps=deps)
-    # The result will be validated with Pydantic to guarantee it is a `SupportResult`, since the agent is generic,
-    # it'll also be typed as a `SupportResult` to aid with static type checking.
-    print(result.data)
+    # The `result.output` will be validated with Pydantic to guarantee it is a `SupportOutput`. Since the agent is generic,
+    # it'll also be typed as a `SupportOutput` to aid with static type checking.
+    print(result.output)
     """
     support_advice='Hello John, your current account balance, including pending transactions, is $123.45.' block_card=False risk=1
     """
 
     result = await support_agent.run('I just lost my card!', deps=deps)
-    print(result.data)
+    print(result.output)
     """
     support_advice="I'm sorry to hear that, John. We are temporarily blocking your card to prevent unauthorized transactions." block_card=True risk=8
     """

{pydantic_ai-0.0.55 → pydantic_ai-0.1.1}/README.md

@@ -49,14 +49,14 @@ Designed to make [type checking](https://ai.pydantic.dev/agents/#static-type-che
 Leverages Python's familiar control flow and agent composition to build your AI-driven projects, making it easy to apply standard Python best practices you'd use in any other (non-AI) project.
 
 * __Structured Responses__
-Harnesses the power of [Pydantic](https://docs.pydantic.dev/latest/) to [validate and structure](https://ai.pydantic.dev/results/#structured-result-validation) model outputs, ensuring responses are consistent across runs.
+Harnesses the power of [Pydantic](https://docs.pydantic.dev/latest/) to [validate and structure](https://ai.pydantic.dev/output/#structured-output) model outputs, ensuring responses are consistent across runs.
 
 * __Dependency Injection System__
-Offers an optional [dependency injection](https://ai.pydantic.dev/dependencies/) system to provide data and services to your agent's [system prompts](https://ai.pydantic.dev/agents/#system-prompts), [tools](https://ai.pydantic.dev/tools/) and [result validators](https://ai.pydantic.dev/results/#result-validators-functions).
+Offers an optional [dependency injection](https://ai.pydantic.dev/dependencies/) system to provide data and services to your agent's [system prompts](https://ai.pydantic.dev/agents/#system-prompts), [tools](https://ai.pydantic.dev/tools/) and [output validators](https://ai.pydantic.dev/output/#output-validator-functions).
 This is useful for testing and eval-driven iterative development.
 
 * __Streamed Responses__
-Provides the ability to [stream](https://ai.pydantic.dev/results/#streamed-results) LLM outputs continuously, with immediate validation, ensuring rapid and accurate results.
+Provides the ability to [stream](https://ai.pydantic.dev/output/#streamed-results) LLM outputs continuously, with immediate validation, ensuring rapid and accurate outputs.
 
 * __Graph Support__
 [Pydantic Graph](https://ai.pydantic.dev/graph) provides a powerful way to define graphs using typing hints, this is useful in complex applications where standard control flow can degrade to spaghetti code.
@@ -80,7 +80,7 @@ agent = Agent(
 # Here the exchange should be very short: PydanticAI will send the system prompt and the user query to the LLM,
 # the model will return a text response. See below for a more complex run.
 result = agent.run_sync('Where does "hello world" come from?')
-print(result.data)
+print(result.output)
 """
 The first known use of "hello, world" was in a 1974 textbook about the C programming language.
 """
@@ -113,22 +113,22 @@ class SupportDependencies:
     db: DatabaseConn
 
 
-# This pydantic model defines the structure of the result returned by the agent.
-class SupportResult(BaseModel):
+# This pydantic model defines the structure of the output returned by the agent.
+class SupportOutput(BaseModel):
     support_advice: str = Field(description='Advice returned to the customer')
     block_card: bool = Field(description="Whether to block the customer's card")
     risk: int = Field(description='Risk level of query', ge=0, le=10)
 
 
 # This agent will act as first-tier support in a bank.
-# Agents are generic in the type of dependencies they accept and the type of result they return.
-# In this case, the support agent has type `Agent[SupportDependencies, SupportResult]`.
+# Agents are generic in the type of dependencies they accept and the type of output they return.
+# In this case, the support agent has type `Agent[SupportDependencies, SupportOutput]`.
 support_agent = Agent(
     'openai:gpt-4o',
     deps_type=SupportDependencies,
-    # The response from the agent will, be guaranteed to be a SupportResult,
+    # The response from the agent will, be guaranteed to be a SupportOutput,
     # if validation fails the agent is prompted to try again.
-    result_type=SupportResult,
+    output_type=SupportOutput,
     system_prompt=(
         'You are a support agent in our bank, give the '
         'customer support and judge the risk level of their query.'
@@ -150,7 +150,7 @@ async def add_customer_name(ctx: RunContext[SupportDependencies]) -> str:
 # Pydantic is used to validate these arguments, and errors are passed back to the LLM so it can retry.
 @support_agent.tool
 async def customer_balance(
-    ctx: RunContext[SupportDependencies], include_pending: bool
+        ctx: RunContext[SupportDependencies], include_pending: bool
 ) -> float:
     """Returns the customer's current account balance."""
     # The docstring of a tool is also passed to the LLM as the description of the tool.
@@ -168,17 +168,17 @@ async def customer_balance(
 async def main():
     deps = SupportDependencies(customer_id=123, db=DatabaseConn())
     # Run the agent asynchronously, conducting a conversation with the LLM until a final response is reached.
-    # Even in this fairly simple case, the agent will exchange multiple messages with the LLM as tools are called to retrieve a result.
+    # Even in this fairly simple case, the agent will exchange multiple messages with the LLM as tools are called to retrieve an output.
     result = await support_agent.run('What is my balance?', deps=deps)
-    # The result will be validated with Pydantic to guarantee it is a `SupportResult`, since the agent is generic,
-    # it'll also be typed as a `SupportResult` to aid with static type checking.
-    print(result.data)
+    # The `result.output` will be validated with Pydantic to guarantee it is a `SupportOutput`. Since the agent is generic,
+    # it'll also be typed as a `SupportOutput` to aid with static type checking.
+    print(result.output)
     """
     support_advice='Hello John, your current account balance, including pending transactions, is $123.45.' block_card=False risk=1
     """
 
     result = await support_agent.run('I just lost my card!', deps=deps)
-    print(result.data)
+    print(result.output)
     """
     support_advice="I'm sorry to hear that, John. We are temporarily blocking your card to prevent unauthorized transactions." block_card=True risk=8
     """

{pydantic_ai-0.0.55 → pydantic_ai-0.1.1}/pyproject.toml

@@ -83,10 +83,9 @@ members = [
 lint = ["mypy>=1.11.2", "pyright>=1.1.390", "ruff>=0.6.9"]
 docs = [
     "black>=24.10.0",
-    "bs4>=0.0.2",
-    "markdownify>=0.14.1",
     "mkdocs>=1.6.1",
     "mkdocs-glightbox>=0.4.0",
+    "mkdocs-llmstxt>=0.2.0",
     "mkdocs-material[imaging]>=9.5.45",
     "mkdocstrings-python>=1.12.2",
 ]
@@ -192,6 +191,9 @@ filterwarnings = [
     # uvicorn (mcp server)
     "ignore:websockets.legacy is deprecated.*:DeprecationWarning:websockets.legacy",
     "ignore:websockets.server.WebSocketServerProtocol is deprecated:DeprecationWarning",
+    # random resource warnings; I suspect these are coming from vendor SDKs when running examples..
+    "ignore:unclosed <socket:ResourceWarning",
+    "ignore:unclosed event loop:ResourceWarning"
 ]
 
 # https://coverage.readthedocs.io/en/latest/config.html#run
@@ -218,6 +220,7 @@ exclude_lines = [
     'if TYPE_CHECKING:',
     'if typing.TYPE_CHECKING:',
     '@overload',
+    '@deprecated',
     '@typing.overload',
     '@abstractmethod',
     '\(Protocol\):$',

{pydantic_ai-0.0.55 → pydantic_ai-0.1.1}/tests/conftest.py

@@ -211,6 +211,7 @@ async def close_cached_httpx_client() -> AsyncIterator[None]:
         'mistral',
         'cohere',
         'deepseek',
+        None,
     ]:
         await cached_async_http_client(provider=provider).aclose()
 
@@ -232,6 +233,12 @@ def image_content(assets_path: Path) -> BinaryContent:
     return BinaryContent(data=image_bytes, media_type='image/png')
 
 
+@pytest.fixture(scope='session')
+def video_content(assets_path: Path) -> BinaryContent:
+    video_bytes = assets_path.joinpath('small_video.mp4').read_bytes()
+    return BinaryContent(data=video_bytes, media_type='video/mp4')
+
+
 @pytest.fixture(scope='session')
 def document_content(assets_path: Path) -> BinaryContent:
     pdf_bytes = assets_path.joinpath('dummy.pdf').read_bytes()

{pydantic_ai-0.0.55 → pydantic_ai-0.1.1}/tests/evals/test_dataset.py

@@ -688,8 +688,8 @@ async def test_duplicate_evaluator_failure(example_dataset: Dataset[TaskInput, T
     assert str(exc_info.value) == snapshot("Duplicate evaluator class name: 'FirstEvaluator'")
 
 
-async def test_invalid_evaluator_result_type(example_dataset: Dataset[TaskInput, TaskOutput, TaskMetadata]):
-    """Test that an invalid evaluator result type raises an error."""
+async def test_invalid_evaluator_output_type(example_dataset: Dataset[TaskInput, TaskOutput, TaskMetadata]):
+    """Test that an invalid evaluator output type raises an error."""
     invalid_evaluator = Python(expression='...')
     example_dataset.add_evaluator(invalid_evaluator)
 
@@ -996,31 +996,31 @@ def test_import_generate_dataset():
 def test_evaluate_non_serializable_inputs():
     @dataclass
     class MyInputs:
-        result_type: type[str] | type[int]
+        output_type: type[str] | type[int]
 
     my_dataset = Dataset[MyInputs, Any, Any](
         cases=[
             Case(
                 name='str',
-                inputs=MyInputs(result_type=str),
+                inputs=MyInputs(output_type=str),
                 expected_output='abc',
             ),
             Case(
                 name='int',
-                inputs=MyInputs(result_type=int),
+                inputs=MyInputs(output_type=int),
                 expected_output=123,
             ),
         ],
     )
 
     async def my_task(my_inputs: MyInputs) -> int | str:
-        if issubclass(my_inputs.result_type, str):
-            return my_inputs.result_type('abc')
+        if issubclass(my_inputs.output_type, str):
+            return my_inputs.output_type('abc')
         else:
-            return my_inputs.result_type(123)
+            return my_inputs.output_type(123)
 
     report = my_dataset.evaluate_sync(task=my_task)
-    assert [c.inputs for c in report.cases] == snapshot([MyInputs(result_type=str), MyInputs(result_type=int)])
+    assert [c.inputs for c in report.cases] == snapshot([MyInputs(output_type=str), MyInputs(output_type=int)])
 
     table = report.console_table(include_input=True)
     assert render_table(table) == snapshot("""\
@@ -1028,9 +1028,9 @@ def test_evaluate_non_serializable_inputs():
 ┏━━━━━━━━━━┳━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┳━━━━━━━━━━┓
 ┃ Case ID  ┃ Inputs                                                                             ┃ Duration ┃
 ┡━━━━━━━━━━╇━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━╇━━━━━━━━━━┩
-│ str      │ test_evaluate_non_serializable_inputs.<locals>.MyInputs(result_type=<class 'str'>) │     1.0s │
+│ str      │ test_evaluate_non_serializable_inputs.<locals>.MyInputs(output_type=<class 'str'>) │     1.0s │
 ├──────────┼────────────────────────────────────────────────────────────────────────────────────┼──────────┤
-│ int      │ test_evaluate_non_serializable_inputs.<locals>.MyInputs(result_type=<class 'int'>) │     1.0s │
+│ int      │ test_evaluate_non_serializable_inputs.<locals>.MyInputs(output_type=<class 'int'>) │     1.0s │
 ├──────────┼────────────────────────────────────────────────────────────────────────────────────┼──────────┤
 │ Averages │                                                                                    │     1.0s │
 └──────────┴────────────────────────────────────────────────────────────────────────────────────┴──────────┘

{pydantic_ai-0.0.55 → pydantic_ai-0.1.1}/tests/evals/test_llm_as_a_judge.py

@@ -70,15 +70,15 @@ async def test_judge_output_mock(mocker: MockerFixture):
     """Test judge_output function with mocked agent."""
     # Mock the agent run method
     mock_result = mocker.MagicMock()
-    mock_result.data = GradingOutput(reason='Test passed', pass_=True, score=1.0)
+    mock_result.output = GradingOutput(reason='Test passed', pass_=True, score=1.0)
     mock_run = mocker.patch('pydantic_ai.Agent.run', return_value=mock_result)
 
     # Test with string output
-    result = await judge_output('Hello world', 'Content contains a greeting')
-    assert isinstance(result, GradingOutput)
-    assert result.reason == 'Test passed'
-    assert result.pass_ is True
-    assert result.score == 1.0
+    grading_output = await judge_output('Hello world', 'Content contains a greeting')
+    assert isinstance(grading_output, GradingOutput)
+    assert grading_output.reason == 'Test passed'
+    assert grading_output.pass_ is True
+    assert grading_output.score == 1.0
 
     # Verify the agent was called with correct prompt
     mock_run.assert_called_once()
@@ -92,7 +92,7 @@ async def test_judge_input_output_mock(mocker: MockerFixture):
     """Test judge_input_output function with mocked agent."""
     # Mock the agent run method
     mock_result = mocker.MagicMock()
-    mock_result.data = GradingOutput(reason='Test passed', pass_=True, score=1.0)
+    mock_result.output = GradingOutput(reason='Test passed', pass_=True, score=1.0)
     mock_run = mocker.patch('pydantic_ai.Agent.run', return_value=mock_result)
 
     # Test with string input and output

pydantic_ai-0.1.1/tests/models/cassettes/test_anthropic/test_anthropic_model_instructions.yaml

@@ -0,0 +1,56 @@
+interactions:
+- request:
+    headers:
+      accept:
+      - application/json
+      accept-encoding:
+      - gzip, deflate
+      connection:
+      - keep-alive
+      content-length:
+      - '202'
+      content-type:
+      - application/json
+      host:
+      - api.anthropic.com
+    method: POST
+    parsed_body:
+      max_tokens: 1024
+      messages:
+      - content:
+        - text: What is the capital of France?
+          type: text
+        role: user
+      model: claude-3-opus-latest
+      stream: false
+      system: You are a helpful assistant.
+    uri: https://api.anthropic.com/v1/messages
+  response:
+    headers:
+      connection:
+      - keep-alive
+      content-length:
+      - '328'
+      content-type:
+      - application/json
+      transfer-encoding:
+      - chunked
+    parsed_body:
+      content:
+      - text: The capital of France is Paris.
+        type: text
+      id: msg_01U58nruzfn9BrXrrF2hhb4m
+      model: claude-3-opus-20240229
+      role: assistant
+      stop_reason: end_turn
+      stop_sequence: null
+      type: message
+      usage:
+        cache_creation_input_tokens: 0
+        cache_read_input_tokens: 0
+        input_tokens: 20
+        output_tokens: 10
+    status:
+      code: 200
+      message: OK
+version: 1

pydantic_ai-0.1.1/tests/models/cassettes/test_bedrock/test_bedrock_model_guardrail_config.yaml

@@ -0,0 +1,52 @@
+interactions:
+- request:
+    body: '{"messages": [{"role": "user", "content": [{"text": "What is the capital of France?"}]}], "system": [{"text": "You
+      are a helpful chatbot."}], "guardrailConfig": {"guardrailIdentifier": "guardrailv1", "guardrailVersion": "v1", "trace": "enabled"}'
+    headers:
+      amz-sdk-invocation-id:
+      - !!binary |
+        ZmZhYjMyZmItODRjOS00YWZjLWE4NTAtNTQ4OTUxMjI5NmU4
+      amz-sdk-request:
+      - !!binary |
+        YXR0ZW1wdD0x
+      content-length:
+      - '178'
+      content-type:
+      - !!binary |
+        YXBwbGljYXRpb24vanNvbg==
+    method: POST
+    uri: https://bedrock-runtime.us-east-1.amazonaws.com/model/us.amazon.nova-micro-v1%3A0/converse
+  response:
+    headers:
+      connection:
+      - keep-alive
+      content-length:
+      - '206'
+      content-type:
+      - application/json
+    parsed_body:
+      metrics:
+        latencyMs: 159
+      output:
+        message:
+          content:
+          - text: The capital of France is Paris. Paris is not only the capital city but also the most populous city in France, known for its significant cultural,
+              political, and economic influence both within the country and globally. It is famous for landmarks such as the Eiffel Tower, the Louvre Museum,
+              and the Notre-Dame Cathedral, among many other historical and architectural treasures.
+
+          role: assistant
+      stopReason: max_tokens
+      usage:
+        inputTokens: 13
+        outputTokens: 5
+        totalTokens: 18
+      trace:
+        guardrail:
+          modelOutput:
+            - test
+      performanceConfig:
+        latency: optimized
+    status:
+      code: 200
+      message: OK
+version: 1

pydantic_ai-0.1.1/tests/models/cassettes/test_bedrock/test_bedrock_model_instructions.yaml

@@ -0,0 +1,46 @@
+interactions:
+- request:
+    body: '{"messages": [{"role": "user", "content": [{"text": "What is the capital of France?"}]}], "system": [{"text": "You
+      are a helpful assistant."}], "inferenceConfig": {}}'
+    headers:
+      amz-sdk-invocation-id:
+      - !!binary |
+        YmM2NjFmZjctZDZhYi00NzI5LWIyNTQtNjc5MzA1MGMyOGU3
+      amz-sdk-request:
+      - !!binary |
+        YXR0ZW1wdD0x
+      content-length:
+      - '166'
+      content-type:
+      - !!binary |
+        YXBwbGljYXRpb24vanNvbg==
+    method: POST
+    uri: https://bedrock-runtime.us-east-1.amazonaws.com/model/us.amazon.nova-pro-v1%3A0/converse
+  response:
+    headers:
+      connection:
+      - keep-alive
+      content-length:
+      - '521'
+      content-type:
+      - application/json
+    parsed_body:
+      metrics:
+        latencyMs: 863
+      output:
+        message:
+          content:
+          - text: The capital of France is Paris. Paris is not only the political and economic hub of the country but also
+              a major center for culture, fashion, art, and tourism. It is renowned for its rich history, iconic landmarks
+              such as the Eiffel Tower, Notre-Dame Cathedral, and the Louvre Museum, as well as its influence on global culture
+              and cuisine.
+          role: assistant
+      stopReason: end_turn
+      usage:
+        inputTokens: 13
+        outputTokens: 71
+        totalTokens: 84
+    status:
+      code: 200
+      message: OK
+version: 1

pydantic_ai-0.1.1/tests/models/cassettes/test_bedrock/test_bedrock_model_other_parameters.yaml

@@ -0,0 +1,47 @@
+interactions:
+- request:
+    body: '{"messages": [{"role": "user", "content": [{"text": "What is the capital of France?"}]}], "system": [{"text": "You
+      are a helpful chatbot."}], "requestMetadata": {"test": "test"}, "additionalModelResponseFieldPaths": ["test"], "additionalModelRequestFields": {"test": "test"}, "promptVariables": {"leo": {"text": "aaaa"}}'
+    headers:
+      amz-sdk-invocation-id:
+      - !!binary |
+        ZmZhYjMyZmItODRjOS00YWZjLWE4NTAtNTQ4OTUxMjI5NmU4
+      amz-sdk-request:
+      - !!binary |
+        YXR0ZW1wdD0x
+      content-length:
+      - '178'
+      content-type:
+      - !!binary |
+        YXBwbGljYXRpb24vanNvbg==
+    method: POST
+    uri: https://bedrock-runtime.us-east-1.amazonaws.com/model/us.amazon.nova-micro-v1%3A0/converse
+  response:
+    headers:
+      connection:
+      - keep-alive
+      content-length:
+      - '206'
+      content-type:
+      - application/json
+    parsed_body:
+      metrics:
+        latencyMs: 159
+      output:
+        message:
+          content:
+          - text: The capital of France is Paris. Paris is not only the capital city but also the most populous city in France, known for its significant cultural,
+              political, and economic influence both within the country and globally. It is famous for landmarks such as the Eiffel Tower, the Louvre Museum,
+              and the Notre-Dame Cathedral, among many other historical and architectural treasures.
+
+          role: assistant
+      stopReason: max_tokens
+      usage:
+        inputTokens: 13
+        outputTokens: 5
+        totalTokens: 18
+      additionalModelResponseFields: "test"
+    status:
+      code: 200
+      message: OK
+version: 1

pydantic_ai-0.1.1/tests/models/cassettes/test_bedrock/test_bedrock_model_performance_config.yaml

@@ -0,0 +1,48 @@
+interactions:
+- request:
+    body: '{"messages": [{"role": "user", "content": [{"text": "What is the capital of France?"}]}], "system": [{"text": "You
+      are a helpful chatbot."}], "performanceConfig": {"latency": "optimized"}}'
+    headers:
+      amz-sdk-invocation-id:
+      - !!binary |
+        ZmZhYjMyZmItODRjOS00YWZjLWE4NTAtNTQ4OTUxMjI5NmU4
+      amz-sdk-request:
+      - !!binary |
+        YXR0ZW1wdD0x
+      content-length:
+      - '178'
+      content-type:
+      - !!binary |
+        YXBwbGljYXRpb24vanNvbg==
+    method: POST
+    uri: https://bedrock-runtime.us-east-1.amazonaws.com/model/us.amazon.nova-micro-v1%3A0/converse
+  response:
+    headers:
+      connection:
+      - keep-alive
+      content-length:
+      - '206'
+      content-type:
+      - application/json
+    parsed_body:
+      metrics:
+        latencyMs: 159
+      output:
+        message:
+          content:
+          - text: The capital of France is Paris. Paris is not only the capital city but also the most populous city in France, known for its significant cultural,
+              political, and economic influence both within the country and globally. It is famous for landmarks such as the Eiffel Tower, the Louvre Museum,
+              and the Notre-Dame Cathedral, among many other historical and architectural treasures.
+
+          role: assistant
+      stopReason: max_tokens
+      usage:
+        inputTokens: 13
+        outputTokens: 5
+        totalTokens: 18
+      performanceConfig:
+        latency: optimized
+    status:
+      code: 200
+      message: OK
+version: 1