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

{pydantic_ai-0.3.1 → pydantic_ai-0.3.3}/.gitignore

@@ -19,3 +19,4 @@ examples/pydantic_ai_examples/.chat_app_messages.sqlite
 node_modules/
 **.idea/
 .coverage*
+/test_tmp/

{pydantic_ai-0.3.1 → pydantic_ai-0.3.3}/Makefile

@@ -64,7 +64,7 @@ test: ## Run tests and collect coverage data
 	@uv run coverage report
 
 .PHONY: test-fast
-test-fast: ## Same as test except no coverage. ~1/4th the time depending on hardware.
+test-fast: ## Same as test except no coverage and 4x faster depending on hardware
 	uv run pytest -n auto --dist=loadgroup
 
 .PHONY: test-all-python
@@ -78,12 +78,12 @@ test-all-python: ## Run tests on Python 3.9 to 3.13
 	@uv run coverage report
 
 .PHONY: testcov
-testcov: test ## Run tests and generate a coverage report
+testcov: test ## Run tests and generate an HTML coverage report
 	@echo "building coverage html"
 	@uv run coverage html
 
 .PHONY: test-mrp
-test-mrp: ## Build and  tests of mcp-run-python
+test-mrp: ## Build and tests of mcp-run-python
 	cd mcp-run-python && deno task build
 	uv run --package mcp-run-python pytest mcp-run-python -v
 

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pydantic-ai
-Version: 0.3.1
+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.1
+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.1; 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.3/tests/example_modules/mcp_server.py

@@ -0,0 +1,26 @@
+from typing import Any
+
+from mcp.server.fastmcp import Context, FastMCP
+from mcp.server.session import ServerSessionT
+from mcp.shared.context import LifespanContextT, RequestT
+
+mcp = FastMCP('PydanticAI MCP Server')
+
+
+@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.
+    """
+
+    deps: Any = getattr(ctx.request_context.meta, 'deps')
+    return {'echo': 'This is an echo message', 'deps': deps}
+
+
+if __name__ == '__main__':
+    mcp.run()

{pydantic_ai-0.3.1 → 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.1 → pydantic_ai-0.3.3}/tests/mcp_server.py

@@ -5,7 +5,7 @@ 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 mcp.types import BlobResourceContents, EmbeddedResource, SamplingMessage, TextContent, TextResourceContents
 from pydantic import AnyUrl
 
 mcp = FastMCP('PydanticAI MCP Server')
@@ -136,6 +136,23 @@ async def echo_deps(ctx: Context[ServerSessionT, LifespanContextT, RequestT]) ->
     return {'echo': 'This is an echo message', 'deps': deps}
 
 
+@mcp.tool()
+async def use_sampling(ctx: Context, foo: str) -> str:  # type: ignore
+    """Use sampling callback."""
+
+    result = await ctx.session.create_message(
+        [
+            SamplingMessage(role='assistant', content=TextContent(type='text', text='')),
+            SamplingMessage(role='user', content=TextContent(type='text', text=foo)),
+        ],
+        max_tokens=1_024,
+        system_prompt='this is a test of MCP sampling',
+        temperature=0.5,
+        stop_sequences=['potato'],
+    )
+    return result.model_dump_json(indent=2)
+
+
 @mcp._mcp_server.set_logging_level()  # pyright: ignore[reportPrivateUsage]
 async def set_logging_level(level: str) -> None:
     global log_level

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