pydantic-ai 0.2.19__tar.gz → 0.3.0__tar.gz

{pydantic_ai-0.2.19 → pydantic_ai-0.3.0}/PKG-INFO

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

{pydantic_ai-0.2.19 → pydantic_ai-0.3.0}/pyproject.toml

@@ -243,6 +243,7 @@ exclude_lines = [
     'if __name__ == .__main__.:',
     'except ImportError as _import_error:',
     '$\s*pass$',
+    'assert False',
 ]
 
 [tool.logfire]

{pydantic_ai-0.2.19 → pydantic_ai-0.3.0}/tests/conftest.py

@@ -37,7 +37,6 @@ if TYPE_CHECKING:
     T = TypeVar('T')
 
     def IsInstance(arg: type[T]) -> T: ...
-
     def IsDatetime(*args: Any, **kwargs: Any) -> datetime: ...
     def IsFloat(*args: Any, **kwargs: Any) -> float: ...
     def IsInt(*args: Any, **kwargs: Any) -> int: ...
@@ -252,6 +251,11 @@ def document_content(assets_path: Path) -> BinaryContent:
     return BinaryContent(data=pdf_bytes, media_type='application/pdf')
 
 
+@pytest.fixture(scope='session')
+def deepseek_api_key() -> str:
+    return os.getenv('DEEPSEEK_API_KEY', 'mock-api-key')
+
+
 @pytest.fixture(scope='session')
 def openai_api_key() -> str:
     return os.getenv('OPENAI_API_KEY', 'mock-api-key')

{pydantic_ai-0.2.19 → pydantic_ai-0.3.0}/tests/mcp_server.py

@@ -3,6 +3,8 @@ from pathlib import Path
 from typing import Any
 
 from mcp.server.fastmcp import Context, FastMCP, Image
+from mcp.server.session import ServerSessionT
+from mcp.shared.context import LifespanContextT, RequestT
 from mcp.types import BlobResourceContents, EmbeddedResource, TextResourceContents
 from pydantic import AnyUrl
 
@@ -118,6 +120,22 @@ async def get_log_level(ctx: Context) -> str:  # type: ignore
     return log_level
 
 
+@mcp.tool()
+async def echo_deps(ctx: Context[ServerSessionT, LifespanContextT, RequestT]) -> dict[str, Any]:
+    """Echo the run context.
+
+    Args:
+        ctx: Context object containing request and session information.
+
+    Returns:
+        Dictionary with an echo message and the deps.
+    """
+    await ctx.info('This is an info message')
+
+    deps: Any = getattr(ctx.request_context.meta, 'deps')
+    return {'echo': 'This is an echo message', 'deps': deps}
+
+
 @mcp._mcp_server.set_logging_level()  # pyright: ignore[reportPrivateUsage]
 async def set_logging_level(level: str) -> None:
     global log_level

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

@@ -0,0 +1,230 @@
+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: 2048
+      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:
+      - '1876'
+      content-type:
+      - application/json
+      strict-transport-security:
+      - max-age=31536000; includeSubDomains; preload
+      transfer-encoding:
+      - chunked
+    parsed_body:
+      content:
+      - signature: ErUBCkYIBBgCIkDdtS5sPfAhQSct3TDKHzeqm87m7bk/P0ecMKVxqofk9q15fEDVWXxuIzQIYZCLUfcJzFi4/IYnpQYrgP34x4pnEgzeA7mWRCy/f1bK+IYaDH5i0Q5hgZkqPeMdwSIwMzHMBPM4Xae4czWnzjHGLR9Xg7DN+sb+MXKFgdXY4bcaOKzhImS05aqIjqBs4CHyKh1dTzSnHd76MAHgM1qjBQ2eIZJJ7s5WGkRkbvWzTxgC
+        thinking: This is a basic question about pedestrian safety when crossing a street. I'll provide a clear, step-by-step
+          explanation of how to safely cross a street. This is important safety information that applies to most places, though
+          specific rules might vary slightly by country.
+        type: thinking
+      - text: |-
+          # How to Cross the Street Safely
+
+          1. **Find the right spot to cross**
+             - Use marked crosswalks when available
+             - Cross at intersections with traffic signals if possible
+             - Avoid crossing between parked cars or on blind curves
+
+          2. **Stop at the edge of the street**
+             - Stand where you can see traffic clearly
+             - Make sure drivers can see you too
+
+          3. **Look in all directions**
+             - Look left, right, then left again (or right, left, right again in left-driving countries)
+             - Check for turning vehicles at intersections
+
+          4. **Wait for a safe gap in traffic**
+             - If at a signal, wait for the "WALK" sign or green light
+             - Make sure all vehicles have stopped before stepping into the road
+
+          5. **Cross with purpose**
+             - Walk at a steady pace - don't run
+             - Continue watching for traffic as you cross
+             - Don't use your phone or wear headphones while crossing
+
+          Remember that local customs and laws may vary, but safety always comes first!
+        type: text
+      id: msg_01FWiSVNCRHvHUYU21BRandY
+      model: claude-3-7-sonnet-20250219
+      role: assistant
+      stop_reason: end_turn
+      stop_sequence: null
+      type: message
+      usage:
+        cache_creation_input_tokens: 0
+        cache_read_input_tokens: 0
+        input_tokens: 42
+        output_tokens: 302
+        service_tier: standard
+    status:
+      code: 200
+      message: OK
+- request:
+    headers:
+      accept:
+      - application/json
+      accept-encoding:
+      - gzip, deflate
+      connection:
+      - keep-alive
+      content-length:
+      - '1953'
+      content-type:
+      - application/json
+      host:
+      - api.anthropic.com
+    method: POST
+    parsed_body:
+      max_tokens: 2048
+      messages:
+      - content:
+        - text: How do I cross the street?
+          type: text
+        role: user
+      - content:
+        - signature: ErUBCkYIBBgCIkDdtS5sPfAhQSct3TDKHzeqm87m7bk/P0ecMKVxqofk9q15fEDVWXxuIzQIYZCLUfcJzFi4/IYnpQYrgP34x4pnEgzeA7mWRCy/f1bK+IYaDH5i0Q5hgZkqPeMdwSIwMzHMBPM4Xae4czWnzjHGLR9Xg7DN+sb+MXKFgdXY4bcaOKzhImS05aqIjqBs4CHyKh1dTzSnHd76MAHgM1qjBQ2eIZJJ7s5WGkRkbvWzTxgC
+          thinking: This is a basic question about pedestrian safety when crossing a street. I'll provide a clear, step-by-step
+            explanation of how to safely cross a street. This is important safety information that applies to most places,
+            though specific rules might vary slightly by country.
+          type: thinking
+        - text: |-
+            # How to Cross the Street Safely
+
+            1. **Find the right spot to cross**
+               - Use marked crosswalks when available
+               - Cross at intersections with traffic signals if possible
+               - Avoid crossing between parked cars or on blind curves
+
+            2. **Stop at the edge of the street**
+               - Stand where you can see traffic clearly
+               - Make sure drivers can see you too
+
+            3. **Look in all directions**
+               - Look left, right, then left again (or right, left, right again in left-driving countries)
+               - Check for turning vehicles at intersections
+
+            4. **Wait for a safe gap in traffic**
+               - If at a signal, wait for the "WALK" sign or green light
+               - Make sure all vehicles have stopped before stepping into the road
+
+            5. **Cross with purpose**
+               - Walk at a steady pace - don't run
+               - Continue watching for traffic as you cross
+               - Don't use your phone or wear headphones while crossing
+
+            Remember that local customs and laws may vary, but safety always comes first!
+          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:
+      - '2807'
+      content-type:
+      - application/json
+      strict-transport-security:
+      - max-age=31536000; includeSubDomains; preload
+      transfer-encoding:
+      - chunked
+    parsed_body:
+      content:
+      - signature: ErUBCkYIBBgCIkCNPEqIUXmAqiaqIqaHEmtiTi5sG6jBLYWmyfr9ELAh9dLAPPiq0Bnp2YQFJB2kz0aWYC8pJW9ylay8cJPFOGdIEgwcoJGGceEVCihog7MaDBZNwmI8LweQANgdvCIwvYrhAAqUDGHfQUYWuVB3ay4ySnmnROCDhtjOe6zTA2N2NC+BCePcZQBGQh/tnuoXKh37QqP3KRrKdVU5j1x4vAtUNtxQhbh4ip5qU5J12xgC
+        thinking: |-
+          This is an interesting analogy question. The person is asking how to cross a river by comparing it to crossing a street. I should outline the key principles of river crossing that parallel street crossing safety, while adapting for the unique challenges of a river environment.
+
+          For crossing a river, I should consider:
+          1. Finding the right spot (like shallow areas, bridges, or ferry crossings)
+          2. Assessing the conditions (river speed, depth, width, obstacles)
+          3. Choosing the appropriate method based on the river conditions
+          4. Safety precautions specific to water crossings
+          5. The actual crossing technique
+
+          I'll structure this as a parallel to the street crossing guide, highlighting the similarities in approach while acknowledging the different hazards and considerations.
+        type: thinking
+      - text: |-
+          # How to Cross a River Safely
+
+          1. **Find the right spot to cross**
+             - Use bridges, ferries, or designated crossing points when available
+             - Look for narrow, shallow sections with slower currents
+             - Avoid areas with rapids, fallen trees, or deep channels
+
+          2. **Assess the river before attempting to cross**
+             - Check the water speed, depth, and clarity
+             - Look for hazards like submerged objects or strong currents
+             - Consider seasonal factors (spring floods, winter ice)
+
+          3. **Choose the appropriate crossing method**
+             - Bridge or ferry: Safest options when available
+             - Wading: For shallow, calm waters (typically below knee height)
+             - Swimming: Only for those with strong swimming skills in appropriate conditions
+             - Boat/raft: For wider or deeper crossings
+
+          4. **Prepare for the crossing**
+             - Secure belongings in waterproof containers
+             - Remove shoes or change to water shoes if wading
+             - Consider using a walking stick for balance when wading
+
+          5. **Cross with caution**
+             - Face upstream while crossing to better resist the current
+             - Move steadily and deliberately - don't rush
+             - Stay aware of changing conditions during the crossing
+
+          Remember that river crossings involve significantly greater risks than street crossings. When in doubt, seek an established crossing point or go around rather than risking a dangerous crossing.
+        type: text
+      id: msg_01T32uJ4PiazXZjT1tfhpP8g
+      model: claude-3-7-sonnet-20250219
+      role: assistant
+      stop_reason: end_turn
+      stop_sequence: null
+      type: message
+      usage:
+        cache_creation_input_tokens: 0
+        cache_read_input_tokens: 0
+        input_tokens: 303
+        output_tokens: 486
+        service_tier: standard
+    status:
+      code: 200
+      message: OK
+version: 1

pydantic_ai-0.3.0/tests/models/cassettes/test_anthropic/test_anthropic_model_thinking_part_stream.yaml

@@ -0,0 +1,231 @@
+interactions:
+- request:
+    headers:
+      accept:
+      - application/json
+      accept-encoding:
+      - gzip, deflate
+      connection:
+      - keep-alive
+      content-length:
+      - '212'
+      content-type:
+      - application/json
+      host:
+      - api.anthropic.com
+    method: POST
+    parsed_body:
+      max_tokens: 2048
+      messages:
+      - content:
+        - text: How do I cross the street?
+          type: text
+        role: user
+      model: claude-3-7-sonnet-latest
+      stream: true
+      thinking:
+        budget_tokens: 1024
+        type: enabled
+    uri: https://api.anthropic.com/v1/messages?beta=true
+  response:
+    body:
+      string: |+
+        event: message_start
+        data: {"type":"message_start","message":{"id":"msg_01K8dnhxH3xD11atV9TZWkNM","type":"message","role":"assistant","model":"claude-3-7-sonnet-20250219","content":[],"stop_reason":null,"stop_sequence":null,"usage":{"input_tokens":42,"cache_creation_input_tokens":0,"cache_read_input_tokens":0,"output_tokens":5,"service_tier":"standard"}}}
+
+        event: content_block_start
+        data: {"type":"content_block_start","index":0,"content_block":{"type":"thinking","thinking":"","signature":""}     }
+
+        event: ping
+        data: {"type": "ping"}
+
+        event: content_block_delta
+        data: {"type":"content_block_delta","index":0,"delta":{"type":"thinking_delta","thinking":"This question is asking"}              }
+
+        event: content_block_delta
+        data: {"type":"content_block_delta","index":0,"delta":{"type":"thinking_delta","thinking":" about how to cross a street"}               }
+
+        event: content_block_delta
+        data: {"type":"content_block_delta","index":0,"delta":{"type":"thinking_delta","thinking":" safely, which is a"}         }
+
+        event: content_block_delta
+        data: {"type":"content_block_delta","index":0,"delta":{"type":"thinking_delta","thinking":" basic pedestrian safety skill"}             }
+
+        event: content_block_delta
+        data: {"type":"content_block_delta","index":0,"delta":{"type":"thinking_delta","thinking":". Let me provide a"}         }
+
+        event: content_block_delta
+        data: {"type":"content_block_delta","index":0,"delta":{"type":"thinking_delta","thinking":" clear, helpful answer"}       }
+
+        event: content_block_delta
+        data: {"type":"content_block_delta","index":0,"delta":{"type":"thinking_delta","thinking":" with the standard safety procedures for"} }
+
+        event: content_block_delta
+        data: {"type":"content_block_delta","index":0,"delta":{"type":"thinking_delta","thinking":" crossing a street."}           }
+
+        event: content_block_delta
+        data: {"type":"content_block_delta","index":0,"delta":{"type":"signature_delta","signature":"ErUBCkYIBBgCIkCvb+W28uqOVW1QcFdQfS4Qmlvhuf3Ji+y2HrT70+C+1Y3+bsJWI7TalhkbNqYrrsiwNiZD/V36NRWv13CwaLU0EgwKvWjc4TaALe+mux4aDBvydYDAXS63lck0WiIwe6i1uYVGUdtG2sDLh0hykB4CFO8QKO2jguCYqNxB3XMC0f/6zTQOsCIGqOUxaDRqKh0Wa/4Poohf60tzp4tl4c3ft1J+02/lKjVXEg28PhgC"}}
+
+        event: content_block_stop
+        data: {"type":"content_block_stop","index":0     }
+
+        event: content_block_start
+        data: {"type":"content_block_start","index":1,"content_block":{"type":"text","text":""}             }
+
+        event: content_block_delta
+        data: {"type":"content_block_delta","index":1,"delta":{"type":"text_delta","text":"# How to Cross a Street Safely"}}
+
+        event: content_block_delta
+        data: {"type":"content_block_delta","index":1,"delta":{"type":"text_delta","text":"\n\n1. **Fin"}           }
+
+        event: content_block_delta
+        data: {"type":"content_block_delta","index":1,"delta":{"type":"text_delta","text":"d a designated crossing point** if"} }
+
+        event: content_block_delta
+        data: {"type":"content_block_delta","index":1,"delta":{"type":"text_delta","text":" possible:\n   -"}         }
+
+        event: content_block_delta
+        data: {"type":"content_block_delta","index":1,"delta":{"type":"text_delta","text":" Crosswalks"}        }
+
+        event: content_block_delta
+        data: {"type":"content_block_delta","index":1,"delta":{"type":"text_delta","text":"\n   - Pedestrian signals\n   -"}   }
+
+        event: content_block_delta
+        data: {"type":"content_block_delta","index":1,"delta":{"type":"text_delta","text":" Pedestrian bridges"} }
+
+        event: content_block_delta
+        data: {"type":"content_block_delta","index":1,"delta":{"type":"text_delta","text":"\n   - Inters"}           }
+
+        event: content_block_delta
+        data: {"type":"content_block_delta","index":1,"delta":{"type":"text_delta","text":"ections\n\n2. **Before"}       }
+
+        event: content_block_delta
+        data: {"type":"content_block_delta","index":1,"delta":{"type":"text_delta","text":" crossing:**\n   - Stop"}   }
+
+        event: content_block_delta
+        data: {"type":"content_block_delta","index":1,"delta":{"type":"text_delta","text":" at the curb or edge of the road"}      }
+
+        event: content_block_delta
+        data: {"type":"content_block_delta","index":1,"delta":{"type":"text_delta","text":"\n   - Look left,"}         }
+
+        event: content_block_delta
+        data: {"type":"content_block_delta","index":1,"delta":{"type":"text_delta","text":" right, then left again ("}            }
+
+        event: content_block_delta
+        data: {"type":"content_block_delta","index":1,"delta":{"type":"text_delta","text":"or right, left, right again"}         }
+
+        event: content_block_delta
+        data: {"type":"content_block_delta","index":1,"delta":{"type":"text_delta","text":" in countries with left"} }
+
+        event: content_block_delta
+        data: {"type":"content_block_delta","index":1,"delta":{"type":"text_delta","text":"-side driving)\n   -"}   }
+
+        event: content_block_delta
+        data: {"type":"content_block_delta","index":1,"delta":{"type":"text_delta","text":" Make eye contact with nearby"}         }
+
+        event: content_block_delta
+        data: {"type":"content_block_delta","index":1,"delta":{"type":"text_delta","text":" drivers if possible\n   "}            }
+
+        event: content_block_delta
+        data: {"type":"content_block_delta","index":1,"delta":{"type":"text_delta","text":"- Listen for approaching vehicles\n\n3. **"}     }
+
+        event: content_block_delta
+        data: {"type":"content_block_delta","index":1,"delta":{"type":"text_delta","text":"When crossing:**\n   -"}          }
+
+        event: content_block_delta
+        data: {"type":"content_block_delta","index":1,"delta":{"type":"text_delta","text":" Walk, don't run"}               }
+
+        event: content_block_delta
+        data: {"type":"content_block_delta","index":1,"delta":{"type":"text_delta","text":"\n   - Continue looking in"}               }
+
+        event: content_block_delta
+        data: {"type":"content_block_delta","index":1,"delta":{"type":"text_delta","text":" both directions as you cross\n   -"}    }
+
+        event: content_block_delta
+        data: {"type":"content_block_delta","index":1,"delta":{"type":"text_delta","text":" Use the crosswalk if available"}         }
+
+        event: content_block_delta
+        data: {"type":"content_block_delta","index":1,"delta":{"type":"text_delta","text":"\n   - Cross in"}               }
+
+        event: content_block_delta
+        data: {"type":"content_block_delta","index":1,"delta":{"type":"text_delta","text":" a straight line, not"}}
+
+        event: content_block_delta
+        data: {"type":"content_block_delta","index":1,"delta":{"type":"text_delta","text":" diagonally"}         }
+
+        event: content_block_delta
+        data: {"type":"content_block_delta","index":1,"delta":{"type":"text_delta","text":"\n\n4. **If there"} }
+
+        event: content_block_delta
+        data: {"type":"content_block_delta","index":1,"delta":{"type":"text_delta","text":" are pedestrian signals:**\n   "}             }
+
+        event: content_block_delta
+        data: {"type":"content_block_delta","index":1,"delta":{"type":"text_delta","text":"- Wait for the \""}           }
+
+        event: content_block_delta
+        data: {"type":"content_block_delta","index":1,"delta":{"type":"text_delta","text":"WALK\" signal or"} }
+
+        event: content_block_delta
+        data: {"type":"content_block_delta","index":1,"delta":{"type":"text_delta","text":" green walking figure\n   - Don't"}    }
+
+        event: content_block_delta
+        data: {"type":"content_block_delta","index":1,"delta":{"type":"text_delta","text":" start crossing if the signal is flashing"}        }
+
+        event: content_block_delta
+        data: {"type":"content_block_delta","index":1,"delta":{"type":"text_delta","text":" or showing \"DON'"}    }
+
+        event: content_block_delta
+        data: {"type":"content_block_delta","index":1,"delta":{"type":"text_delta","text":"T WALK\""}       }
+
+        event: content_block_delta
+        data: {"type":"content_block_delta","index":1,"delta":{"type":"text_delta","text":"\n\n5. **Avoi"}     }
+
+        event: content_block_delta
+        data: {"type":"content_block_delta","index":1,"delta":{"type":"text_delta","text":"d distractions**"}       }
+
+        event: content_block_delta
+        data: {"type":"content_block_delta","index":1,"delta":{"type":"text_delta","text":" like texting, looking"}    }
+
+        event: content_block_delta
+        data: {"type":"content_block_delta","index":1,"delta":{"type":"text_delta","text":" at your phone, or"}          }
+
+        event: content_block_delta
+        data: {"type":"content_block_delta","index":1,"delta":{"type":"text_delta","text":" wearing headphones that prevent"}           }
+
+        event: content_block_delta
+        data: {"type":"content_block_delta","index":1,"delta":{"type":"text_delta","text":" you from hearing traffic."}               }
+
+        event: content_block_delta
+        data: {"type":"content_block_delta","index":1,"delta":{"type":"text_delta","text":"\n\nIs there a"}          }
+
+        event: content_block_delta
+        data: {"type":"content_block_delta","index":1,"delta":{"type":"text_delta","text":" specific situation or location"}       }
+
+        event: content_block_delta
+        data: {"type":"content_block_delta","index":1,"delta":{"type":"text_delta","text":" you're asking about?"}      }
+
+        event: content_block_stop
+        data: {"type":"content_block_stop","index":1          }
+
+        event: message_delta
+        data: {"type":"message_delta","delta":{"stop_reason":"end_turn","stop_sequence":null},"usage":{"output_tokens":305}             }
+
+        event: message_stop
+        data: {"type":"message_stop"              }
+
+    headers:
+      cache-control:
+      - no-cache
+      connection:
+      - keep-alive
+      content-type:
+      - text/event-stream; charset=utf-8
+      strict-transport-security:
+      - max-age=31536000; includeSubDomains; preload
+      transfer-encoding:
+      - chunked
+    status:
+      code: 200
+      message: OK
+version: 1
+...