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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pydantic-ai
-Version: 0.7.1
+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.1
+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.1; 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.1 → 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.1 → 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.1 → 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.1 → 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.1 → 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

pydantic_ai-0.7.2/tests/models/cassettes/test_anthropic/test_anthropic_model_thinking_part.yaml

@@ -0,0 +1,255 @@
+interactions:
+- request:
+    headers:
+      accept:
+      - application/json
+      accept-encoding:
+      - gzip, deflate
+      connection:
+      - keep-alive
+      content-length:
+      - '213'
+      content-type:
+      - application/json
+      host:
+      - api.anthropic.com
+    method: POST
+    parsed_body:
+      max_tokens: 4096
+      messages:
+      - content:
+        - text: How do I cross the street?
+          type: text
+        role: user
+      model: claude-3-7-sonnet-latest
+      stream: false
+      thinking:
+        budget_tokens: 1024
+        type: enabled
+    uri: https://api.anthropic.com/v1/messages?beta=true
+  response:
+    headers:
+      connection:
+      - keep-alive
+      content-length:
+      - '2197'
+      content-type:
+      - application/json
+      strict-transport-security:
+      - max-age=31536000; includeSubDomains; preload
+      transfer-encoding:
+      - chunked
+    parsed_body:
+      content:
+      - signature: ErUBCkYIBhgCIkB9AyHADyBknnHL4dh+Yj3rg3javltU/bz1MLHKCQTEVZwvjis+DKTOFSYqZU0F2xasSofECVAmYmgtRf87AL52EgyXRs8lh+1HtZ0V+wAaDBo0eAabII+t1pdHzyIweFpD2l4j1eeUwN8UQOW+bxcN3mwu144OdOoUxmEKeOcU97wv+VF2pCsm07qcvucSKh1P/rZzWuYm7vxdnD4EVFHdBeewghoO0Ngc1MTNsxgC
+        thinking: |-
+          This is a straightforward question about a common everyday task - crossing the street safely. I should provide clear, helpful instructions that emphasize safety.
+
+          The basic steps for crossing a street safely include:
+          1. Find a designated crossing area if possible (crosswalk, pedestrian crossing)
+          2. Look both ways before crossing
+          3. Make eye contact with drivers if possible
+          4. Follow traffic signals if present
+          5. Cross quickly but don't run
+          6. Continue to be aware of traffic while crossing
+
+          I'll provide this information in a clear, helpful way, emphasizing safety without being condescending.
+        type: thinking
+      - text: |-
+          # How to Cross the Street Safely
+
+          1. **Find the right spot**
+             - Use crosswalks, pedestrian crossings, or intersections when available
+             - If there's a traffic light, cross at the corner
+
+          2. **Wait for the right time**
+             - At traffic lights, wait for the "WALK" signal or green light
+             - At crosswalks without signals, wait for a gap in traffic
+
+          3. **Look carefully**
+             - Look left, then right, then left again (reverse in countries with left-side driving)
+             - Make eye contact with drivers to ensure they see you
+
+          4. **Cross with caution**
+             - Walk at a steady pace - don't run
+             - Continue watching for traffic as you cross
+             - Stay within the crosswalk if there is one
+
+          5. **Stay alert**
+             - Put away distractions like phones
+             - Remove headphones or keep volume low to hear traffic
+
+          Would you like me to explain any of these steps in more detail?
+        type: text
+      id: msg_01BnZvs3naGorn93wjjCDwbd
+      model: claude-3-7-sonnet-20250219
+      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: 42
+        output_tokens: 363
+        service_tier: standard
+    status:
+      code: 200
+      message: OK
+- request:
+    headers:
+      accept:
+      - application/json
+      accept-encoding:
+      - gzip, deflate
+      connection:
+      - keep-alive
+      content-length:
+      - '2195'
+      content-type:
+      - application/json
+      host:
+      - api.anthropic.com
+    method: POST
+    parsed_body:
+      max_tokens: 4096
+      messages:
+      - content:
+        - text: How do I cross the street?
+          type: text
+        role: user
+      - content:
+        - signature: ErUBCkYIBhgCIkB9AyHADyBknnHL4dh+Yj3rg3javltU/bz1MLHKCQTEVZwvjis+DKTOFSYqZU0F2xasSofECVAmYmgtRf87AL52EgyXRs8lh+1HtZ0V+wAaDBo0eAabII+t1pdHzyIweFpD2l4j1eeUwN8UQOW+bxcN3mwu144OdOoUxmEKeOcU97wv+VF2pCsm07qcvucSKh1P/rZzWuYm7vxdnD4EVFHdBeewghoO0Ngc1MTNsxgC
+          thinking: |-
+            This is a straightforward question about a common everyday task - crossing the street safely. I should provide clear, helpful instructions that emphasize safety.
+
+            The basic steps for crossing a street safely include:
+            1. Find a designated crossing area if possible (crosswalk, pedestrian crossing)
+            2. Look both ways before crossing
+            3. Make eye contact with drivers if possible
+            4. Follow traffic signals if present
+            5. Cross quickly but don't run
+            6. Continue to be aware of traffic while crossing
+
+            I'll provide this information in a clear, helpful way, emphasizing safety without being condescending.
+          type: thinking
+        - text: |-
+            # How to Cross the Street Safely
+
+            1. **Find the right spot**
+               - Use crosswalks, pedestrian crossings, or intersections when available
+               - If there's a traffic light, cross at the corner
+
+            2. **Wait for the right time**
+               - At traffic lights, wait for the "WALK" signal or green light
+               - At crosswalks without signals, wait for a gap in traffic
+
+            3. **Look carefully**
+               - Look left, then right, then left again (reverse in countries with left-side driving)
+               - Make eye contact with drivers to ensure they see you
+
+            4. **Cross with caution**
+               - Walk at a steady pace - don't run
+               - Continue watching for traffic as you cross
+               - Stay within the crosswalk if there is one
+
+            5. **Stay alert**
+               - Put away distractions like phones
+               - Remove headphones or keep volume low to hear traffic
+
+            Would you like me to explain any of these steps in more detail?
+          type: text
+        role: assistant
+      - content:
+        - text: Considering the way to cross the street, analogously, how do I cross the river?
+          type: text
+        role: user
+      model: claude-3-7-sonnet-latest
+      stream: false
+      thinking:
+        budget_tokens: 1024
+        type: enabled
+    uri: https://api.anthropic.com/v1/messages?beta=true
+  response:
+    headers:
+      connection:
+      - keep-alive
+      content-length:
+      - '2676'
+      content-type:
+      - application/json
+      strict-transport-security:
+      - max-age=31536000; includeSubDomains; preload
+      transfer-encoding:
+      - chunked
+    parsed_body:
+      content:
+      - signature: ErUBCkYIBhgCIkDvSvKCs5ePyYmR6zFw5i+jF7KEmortSIleqDa4gfa3pbuBclQt0TPdacouhdXFHdVSqR4qOAAAOpN7RQEUz2o6Egy9MPee6H8U4SW/G2QaDP/9ysoEvk+yNyVYZSIw+/+5wuRyc3oajwV3w0EdL9CIAXXd5thQH7DwAe3HTFvoJuF4oZ4fU+Kh6LRqxnEaKh3SSRqAH4UH/sD86duzg0jox4J/NH4C9iILVesEERgC
+        thinking: |-
+          The person is asking me to draw an analogy between crossing a street and crossing a river. I'll structure my response similarly to my street-crossing guidelines, but adapt it for river crossing, which has different safety considerations and methods.
+
+          For crossing a river, I should include:
+          1. Finding the right spot (bridges, shallow areas, ferry points)
+          2. Assessing safety (current speed, depth, obstacles)
+          3. Choosing the appropriate method (walking across shallow areas, using bridges, boats, etc.)
+          4. Safety precautions (life vests, ropes, etc.)
+          5. The actual crossing technique
+          6. What to do in emergencies
+
+          I'll keep the format similar to my street-crossing response for consistency.
+        type: thinking
+      - text: |-
+          # How to Cross a River Safely
+
+          1. **Find the right spot**
+             - Use bridges, ferries, or designated crossing points when available
+             - Look for narrow or shallow sections if no infrastructure exists
+             - Avoid areas with fast currents, whitewater, or dangerous debris
+
+          2. **Assess the conditions**
+             - Check water depth and speed of current
+             - Look for hazards like submerged objects or slippery rocks
+             - Consider weather conditions that might affect water levels
+
+          3. **Choose the appropriate method**
+             - Bridge or ferry: The safest options
+             - Wade across: Only in shallow, calm water with stable footing
+             - Swim: Only if you're a strong swimmer and conditions are safe
+             - Boat/raft: For wider or deeper crossings
+
+          4. **Prepare properly**
+             - Wear appropriate footwear for wading
+             - Use life jackets/flotation devices when swimming or boating
+             - Secure belongings in waterproof containers
+             - Consider using a walking stick for balance when wading
+
+          5. **Cross with caution**
+             - Face upstream when wading to brace against current
+             - Move diagonally downstream if swimming
+             - Stay alert for changing conditions
+             - Use the buddy system when possible
+
+          Is there a specific river crossing scenario you're curious about?
+        type: text
+      id: msg_019Z9a1qnqUCxd7Fj6PuuetE
+      model: claude-3-7-sonnet-20250219
+      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: 291
+        output_tokens: 471
+        service_tier: standard
+    status:
+      code: 200
+      message: OK
+version: 1