pydantic-ai 0.3.2__tar.gz → 0.3.3__tar.gz

{pydantic_ai-0.3.2 → pydantic_ai-0.3.3}/PKG-INFO

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

{pydantic_ai-0.3.2 → pydantic_ai-0.3.3}/tests/graph/test_mermaid.py

@@ -5,7 +5,7 @@ from collections.abc import Iterator
 from dataclasses import dataclass
 from datetime import timezone
 from pathlib import Path
-from typing import Annotated, Callable
+from typing import Annotated, Callable, Union
 
 import httpx
 import pytest
@@ -426,3 +426,16 @@ def test_wrong_return_type():
 
     with pytest.raises(GraphSetupError, match="Invalid return type: <class 'int'>"):
         NoReturnType.get_node_def({})
+
+
+def test_edge_union():
+    """Test that a union of things annotated with an Edge doesn't raise a TypeError.
+
+    This is important because such unions may occur as a return type for a graph, and needs to be evaluated when
+    generating a mermaid diagram.
+    """
+    # This would raise an error on 3.10 if Edge was not hashable:
+    edges_union = Union[
+        Annotated[End[None], Edge(label='first label')], Annotated[End[None], Edge(label='second label')]
+    ]
+    assert edges_union

pydantic_ai-0.3.3/tests/models/cassettes/test_anthropic/test_anthropic_prompted_output.yaml

@@ -0,0 +1,161 @@
+interactions:
+- request:
+    headers:
+      accept:
+      - application/json
+      accept-encoding:
+      - gzip, deflate
+      connection:
+      - keep-alive
+      content-length:
+      - '740'
+      content-type:
+      - application/json
+      host:
+      - api.anthropic.com
+    method: POST
+    parsed_body:
+      max_tokens: 1024
+      messages:
+      - content:
+        - text: What is the largest city in the user country? Use the get_user_country tool and then your own world knowledge.
+          type: text
+        role: user
+      model: claude-3-5-sonnet-latest
+      stream: false
+      system: |+
+        Always respond with a JSON object that's compatible with this schema:
+
+        {"properties": {"city": {"type": "string"}, "country": {"type": "string"}}, "required": ["city", "country"], "title": "CityLocation", "type": "object"}
+
+        Don't include any text or Markdown fencing before or after.
+
+      tool_choice:
+        type: auto
+      tools:
+      - description: ''
+        input_schema:
+          additionalProperties: false
+          properties: {}
+          type: object
+        name: get_user_country
+    uri: https://api.anthropic.com/v1/messages?beta=true
+  response:
+    headers:
+      connection:
+      - keep-alive
+      content-length:
+      - '397'
+      content-type:
+      - application/json
+      strict-transport-security:
+      - max-age=31536000; includeSubDomains; preload
+      transfer-encoding:
+      - chunked
+    parsed_body:
+      content:
+      - id: toolu_017UryVwtsKsjonhFV3cgV3X
+        input: {}
+        name: get_user_country
+        type: tool_use
+      id: msg_014CpBKzioMqUyLWrMihpvsz
+      model: claude-3-5-sonnet-20241022
+      role: assistant
+      stop_reason: tool_use
+      stop_sequence: null
+      type: message
+      usage:
+        cache_creation_input_tokens: 0
+        cache_read_input_tokens: 0
+        input_tokens: 459
+        output_tokens: 38
+        service_tier: standard
+    status:
+      code: 200
+      message: OK
+- request:
+    headers:
+      accept:
+      - application/json
+      accept-encoding:
+      - gzip, deflate
+      connection:
+      - keep-alive
+      content-length:
+      - '1002'
+      content-type:
+      - application/json
+      host:
+      - api.anthropic.com
+    method: POST
+    parsed_body:
+      max_tokens: 1024
+      messages:
+      - content:
+        - text: What is the largest city in the user country? Use the get_user_country tool and then your own world knowledge.
+          type: text
+        role: user
+      - content:
+        - id: toolu_017UryVwtsKsjonhFV3cgV3X
+          input: {}
+          name: get_user_country
+          type: tool_use
+        role: assistant
+      - content:
+        - content: Mexico
+          is_error: false
+          tool_use_id: toolu_017UryVwtsKsjonhFV3cgV3X
+          type: tool_result
+        role: user
+      model: claude-3-5-sonnet-latest
+      stream: false
+      system: |+
+        Always respond with a JSON object that's compatible with this schema:
+
+        {"properties": {"city": {"type": "string"}, "country": {"type": "string"}}, "required": ["city", "country"], "title": "CityLocation", "type": "object"}
+
+        Don't include any text or Markdown fencing before or after.
+
+      tool_choice:
+        type: auto
+      tools:
+      - description: ''
+        input_schema:
+          additionalProperties: false
+          properties: {}
+          type: object
+        name: get_user_country
+    uri: https://api.anthropic.com/v1/messages?beta=true
+  response:
+    headers:
+      connection:
+      - keep-alive
+      content-length:
+      - '380'
+      content-type:
+      - application/json
+      strict-transport-security:
+      - max-age=31536000; includeSubDomains; preload
+      transfer-encoding:
+      - chunked
+    parsed_body:
+      content:
+      - text: '{"city": "Mexico City", "country": "Mexico"}'
+        type: text
+      id: msg_014JeWCouH6DpdqzMTaBdkpJ
+      model: claude-3-5-sonnet-20241022
+      role: assistant
+      stop_reason: end_turn
+      stop_sequence: null
+      type: message
+      usage:
+        cache_creation_input_tokens: 0
+        cache_read_input_tokens: 0
+        input_tokens: 510
+        output_tokens: 17
+        service_tier: standard
+    status:
+      code: 200
+      message: OK
+version: 1
+...

pydantic_ai-0.3.3/tests/models/cassettes/test_anthropic/test_anthropic_prompted_output_multiple.yaml

@@ -0,0 +1,66 @@
+interactions:
+- request:
+    headers:
+      accept:
+      - application/json
+      accept-encoding:
+      - gzip, deflate
+      connection:
+      - keep-alive
+      content-length:
+      - '1268'
+      content-type:
+      - application/json
+      host:
+      - api.anthropic.com
+    method: POST
+    parsed_body:
+      max_tokens: 1024
+      messages:
+      - content:
+        - text: What is the largest city in Mexico?
+          type: text
+        role: user
+      model: claude-3-5-sonnet-latest
+      stream: false
+      system: |+
+        Always respond with a JSON object that's compatible with this schema:
+
+        {"type": "object", "properties": {"result": {"anyOf": [{"type": "object", "properties": {"kind": {"type": "string", "const": "CityLocation"}, "data": {"properties": {"city": {"type": "string"}, "country": {"type": "string"}}, "required": ["city", "country"], "title": "CityLocation", "type": "object"}}, "description": "CityLocation", "required": ["kind", "data"], "additionalProperties": false}, {"type": "object", "properties": {"kind": {"type": "string", "const": "CountryLanguage"}, "data": {"properties": {"country": {"type": "string"}, "language": {"type": "string"}}, "required": ["country", "language"], "title": "CountryLanguage", "type": "object"}}, "description": "CountryLanguage", "required": ["kind", "data"], "additionalProperties": false}]}}, "required": ["result"], "additionalProperties": false}
+
+        Don't include any text or Markdown fencing before or after.
+
+    uri: https://api.anthropic.com/v1/messages?beta=true
+  response:
+    headers:
+      connection:
+      - keep-alive
+      content-length:
+      - '434'
+      content-type:
+      - application/json
+      strict-transport-security:
+      - max-age=31536000; includeSubDomains; preload
+      transfer-encoding:
+      - chunked
+    parsed_body:
+      content:
+      - text: '{"result": {"kind": "CityLocation", "data": {"city": "Mexico City", "country": "Mexico"}}}'
+        type: text
+      id: msg_013ttUi3HCcKt7PkJpoWs5FT
+      model: claude-3-5-sonnet-20241022
+      role: assistant
+      stop_reason: end_turn
+      stop_sequence: null
+      type: message
+      usage:
+        cache_creation_input_tokens: 0
+        cache_read_input_tokens: 0
+        input_tokens: 281
+        output_tokens: 31
+        service_tier: standard
+    status:
+      code: 200
+      message: OK
+version: 1
+...

pydantic_ai-0.3.3/tests/models/cassettes/test_anthropic/test_anthropic_text_output_function.yaml

@@ -0,0 +1,156 @@
+interactions:
+- request:
+    headers:
+      accept:
+      - application/json
+      accept-encoding:
+      - gzip, deflate
+      connection:
+      - keep-alive
+      content-length:
+      - '409'
+      content-type:
+      - application/json
+      host:
+      - api.anthropic.com
+    method: POST
+    parsed_body:
+      max_tokens: 1024
+      messages:
+      - content:
+        - text: What is the largest city in the user country? Use the get_user_country tool and then your own world knowledge.
+          type: text
+        role: user
+      model: claude-3-5-sonnet-latest
+      stream: false
+      tool_choice:
+        type: auto
+      tools:
+      - description: ''
+        input_schema:
+          additionalProperties: false
+          properties: {}
+          type: object
+        name: get_user_country
+    uri: https://api.anthropic.com/v1/messages?beta=true
+  response:
+    headers:
+      connection:
+      - keep-alive
+      content-length:
+      - '540'
+      content-type:
+      - application/json
+      strict-transport-security:
+      - max-age=31536000; includeSubDomains; preload
+      transfer-encoding:
+      - chunked
+    parsed_body:
+      content:
+      - text: I'll help you find the largest city in your country. Let me first check your country using the get_user_country
+          tool.
+        type: text
+      - id: toolu_01EZuxfc6MsPsPgrAKQohw3e
+        input: {}
+        name: get_user_country
+        type: tool_use
+      id: msg_014NE4yfV1Yz2vLAJzapxxef
+      model: claude-3-5-sonnet-20241022
+      role: assistant
+      stop_reason: tool_use
+      stop_sequence: null
+      type: message
+      usage:
+        cache_creation_input_tokens: 0
+        cache_read_input_tokens: 0
+        input_tokens: 383
+        output_tokens: 66
+        service_tier: standard
+    status:
+      code: 200
+      message: OK
+- request:
+    headers:
+      accept:
+      - application/json
+      accept-encoding:
+      - gzip, deflate
+      connection:
+      - keep-alive
+      content-length:
+      - '814'
+      content-type:
+      - application/json
+      host:
+      - api.anthropic.com
+    method: POST
+    parsed_body:
+      max_tokens: 1024
+      messages:
+      - content:
+        - text: What is the largest city in the user country? Use the get_user_country tool and then your own world knowledge.
+          type: text
+        role: user
+      - content:
+        - text: I'll help you find the largest city in your country. Let me first check your country using the get_user_country
+            tool.
+          type: text
+        - id: toolu_01EZuxfc6MsPsPgrAKQohw3e
+          input: {}
+          name: get_user_country
+          type: tool_use
+        role: assistant
+      - content:
+        - content: Mexico
+          is_error: false
+          tool_use_id: toolu_01EZuxfc6MsPsPgrAKQohw3e
+          type: tool_result
+        role: user
+      model: claude-3-5-sonnet-latest
+      stream: false
+      tool_choice:
+        type: auto
+      tools:
+      - description: ''
+        input_schema:
+          additionalProperties: false
+          properties: {}
+          type: object
+        name: get_user_country
+    uri: https://api.anthropic.com/v1/messages?beta=true
+  response:
+    headers:
+      connection:
+      - keep-alive
+      content-length:
+      - '801'
+      content-type:
+      - application/json
+      strict-transport-security:
+      - max-age=31536000; includeSubDomains; preload
+      transfer-encoding:
+      - chunked
+    parsed_body:
+      content:
+      - text: Based on the result, you are located in Mexico. The largest city in Mexico is Mexico City (Ciudad de México),
+          which is also the nation's capital. Mexico City has a population of approximately 9.2 million people in the city
+          proper, and over 21 million people in its metropolitan area, making it one of the largest urban agglomerations in
+          the world. It is both the political and economic center of Mexico, located in the Valley of Mexico in the central
+          part of the country.
+        type: text
+      id: msg_0193srwo7TCx49h97wDwc7K7
+      model: claude-3-5-sonnet-20241022
+      role: assistant
+      stop_reason: end_turn
+      stop_sequence: null
+      type: message
+      usage:
+        cache_creation_input_tokens: 0
+        cache_read_input_tokens: 0
+        input_tokens: 461
+        output_tokens: 107
+        service_tier: standard
+    status:
+      code: 200
+      message: OK
+version: 1

pydantic_ai-0.3.3/tests/models/cassettes/test_anthropic/test_anthropic_tool_output.yaml

@@ -0,0 +1,176 @@
+interactions:
+- request:
+    headers:
+      accept:
+      - application/json
+      accept-encoding:
+      - gzip, deflate
+      connection:
+      - keep-alive
+      content-length:
+      - '585'
+      content-type:
+      - application/json
+      host:
+      - api.anthropic.com
+    method: POST
+    parsed_body:
+      max_tokens: 1024
+      messages:
+      - content:
+        - text: What is the largest city in the user country?
+          type: text
+        role: user
+      model: claude-3-5-sonnet-latest
+      stream: false
+      tool_choice:
+        type: any
+      tools:
+      - description: ''
+        input_schema:
+          additionalProperties: false
+          properties: {}
+          type: object
+        name: get_user_country
+      - description: The final response which ends this conversation
+        input_schema:
+          properties:
+            city:
+              type: string
+            country:
+              type: string
+          required:
+          - city
+          - country
+          title: CityLocation
+          type: object
+        name: final_result
+    uri: https://api.anthropic.com/v1/messages?beta=true
+  response:
+    headers:
+      connection:
+      - keep-alive
+      content-length:
+      - '397'
+      content-type:
+      - application/json
+      strict-transport-security:
+      - max-age=31536000; includeSubDomains; preload
+      transfer-encoding:
+      - chunked
+    parsed_body:
+      content:
+      - id: toolu_019pMboNVRg5jkw4PKkofQ6Y
+        input: {}
+        name: get_user_country
+        type: tool_use
+      id: msg_01EnfsDTixCmHjqvk9QarBj4
+      model: claude-3-5-sonnet-20241022
+      role: assistant
+      stop_reason: tool_use
+      stop_sequence: null
+      type: message
+      usage:
+        cache_creation_input_tokens: 0
+        cache_read_input_tokens: 0
+        input_tokens: 445
+        output_tokens: 23
+        service_tier: standard
+    status:
+      code: 200
+      message: OK
+- request:
+    headers:
+      accept:
+      - application/json
+      accept-encoding:
+      - gzip, deflate
+      connection:
+      - keep-alive
+      content-length:
+      - '847'
+      content-type:
+      - application/json
+      host:
+      - api.anthropic.com
+    method: POST
+    parsed_body:
+      max_tokens: 1024
+      messages:
+      - content:
+        - text: What is the largest city in the user country?
+          type: text
+        role: user
+      - content:
+        - id: toolu_019pMboNVRg5jkw4PKkofQ6Y
+          input: {}
+          name: get_user_country
+          type: tool_use
+        role: assistant
+      - content:
+        - content: Mexico
+          is_error: false
+          tool_use_id: toolu_019pMboNVRg5jkw4PKkofQ6Y
+          type: tool_result
+        role: user
+      model: claude-3-5-sonnet-latest
+      stream: false
+      tool_choice:
+        type: any
+      tools:
+      - description: ''
+        input_schema:
+          additionalProperties: false
+          properties: {}
+          type: object
+        name: get_user_country
+      - description: The final response which ends this conversation
+        input_schema:
+          properties:
+            city:
+              type: string
+            country:
+              type: string
+          required:
+          - city
+          - country
+          title: CityLocation
+          type: object
+        name: final_result
+    uri: https://api.anthropic.com/v1/messages?beta=true
+  response:
+    headers:
+      connection:
+      - keep-alive
+      content-length:
+      - '432'
+      content-type:
+      - application/json
+      strict-transport-security:
+      - max-age=31536000; includeSubDomains; preload
+      transfer-encoding:
+      - chunked
+    parsed_body:
+      content:
+      - id: toolu_01V4d2H4EWp5LDM2aXaeyR6W
+        input:
+          city: Mexico City
+          country: Mexico
+        name: final_result
+        type: tool_use
+      id: msg_01Hbm5BtKzfVtWs8Eb7rCNNx
+      model: claude-3-5-sonnet-20241022
+      role: assistant
+      stop_reason: tool_use
+      stop_sequence: null
+      type: message
+      usage:
+        cache_creation_input_tokens: 0
+        cache_read_input_tokens: 0
+        input_tokens: 497
+        output_tokens: 56
+        service_tier: standard
+    status:
+      code: 200
+      message: OK
+version: 1

pydantic_ai-0.3.3/tests/models/cassettes/test_gemini/test_gemini_native_output.yaml

@@ -0,0 +1,79 @@
+interactions:
+- request:
+    headers:
+      accept:
+      - '*/*'
+      accept-encoding:
+      - gzip, deflate
+      connection:
+      - keep-alive
+      content-length:
+      - '305'
+      content-type:
+      - application/json
+      host:
+      - generativelanguage.googleapis.com
+    method: POST
+    parsed_body:
+      contents:
+      - parts:
+        - text: What is the largest city in Mexico?
+        role: user
+      generationConfig:
+        response_mime_type: application/json
+        response_schema:
+          properties:
+            city:
+              type: string
+            country:
+              type: string
+          required:
+          - city
+          - country
+          title: CityLocation
+          type: object
+    uri: https://generativelanguage.googleapis.com/v1beta/models/gemini-2.0-flash:generateContent
+  response:
+    headers:
+      alt-svc:
+      - h3=":443"; ma=2592000,h3-29=":443"; ma=2592000
+      content-length:
+      - '710'
+      content-type:
+      - application/json; charset=UTF-8
+      server-timing:
+      - gfet4t7; dur=819
+      transfer-encoding:
+      - chunked
+      vary:
+      - Origin
+      - X-Origin
+      - Referer
+    parsed_body:
+      candidates:
+      - avgLogprobs: -0.00018302639946341515
+        content:
+          parts:
+          - text: |-
+              {
+                "city": "Mexico City",
+                "country": "Mexico"
+              }
+          role: model
+        finishReason: STOP
+      modelVersion: gemini-2.0-flash
+      responseId: SEVIaJvJHICK7dcP3OzRiQQ
+      usageMetadata:
+        candidatesTokenCount: 20
+        candidatesTokensDetails:
+        - modality: TEXT
+          tokenCount: 20
+        promptTokenCount: 17
+        promptTokensDetails:
+        - modality: TEXT
+          tokenCount: 17
+        totalTokenCount: 37
+    status:
+      code: 200
+      message: OK
+version: 1