pydantic-ai 0.2.9__tar.gz → 0.2.11__tar.gz

{pydantic_ai-0.2.9 → pydantic_ai-0.2.11}/PKG-INFO

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

pydantic_ai-0.2.11/tests/models/cassettes/test_anthropic/test_anthropic_model_empty_message_on_history.yaml

@@ -0,0 +1,73 @@
+interactions:
+- request:
+    headers:
+      accept:
+      - application/json
+      accept-encoding:
+      - gzip, deflate
+      connection:
+      - keep-alive
+      content-length:
+      - '281'
+      content-type:
+      - application/json
+      host:
+      - api.anthropic.com
+    method: POST
+    parsed_body:
+      max_tokens: 1024
+      messages:
+      - content:
+        - text: Hello, how can I help you?
+          type: text
+        role: assistant
+      - content:
+        - text: I need a potato!
+          type: text
+        role: user
+      model: claude-3-5-sonnet-latest
+      stream: false
+      system: |+
+        You are a helpful assistant.
+
+    uri: https://api.anthropic.com/v1/messages?beta=true
+  response:
+    headers:
+      connection:
+      - keep-alive
+      content-length:
+      - '671'
+      content-type:
+      - application/json
+      strict-transport-security:
+      - max-age=31536000; includeSubDomains; preload
+      transfer-encoding:
+      - chunked
+    parsed_body:
+      content:
+      - text: |-
+          I can't physically give you a potato since I'm a computer program. 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
+
+          What specifically would you like to know about potatoes?
+        type: text
+      id: msg_01UjnDmX3B57Drosu49sMteT
+      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: 41
+        output_tokens: 82
+        service_tier: standard
+    status:
+      code: 200
+      message: OK
+version: 1

{pydantic_ai-0.2.9 → pydantic_ai-0.2.11}/tests/models/test_anthropic.py

@@ -1035,3 +1035,31 @@ def anth_msg(usage: BetaUsage) -> BetaMessage:
 )
 def test_usage(message_callback: Callable[[], BetaMessage | BetaRawMessageStreamEvent], usage: Usage):
     assert _map_usage(message_callback()) == usage
+
+
+@pytest.mark.vcr()
+async def test_anthropic_model_empty_message_on_history(allow_model_requests: None, anthropic_api_key: str):
+    """The Anthropic API will error if you send an empty message on the history.
+
+    Check <https://github.com/pydantic/pydantic-ai/pull/1027> for more details.
+    """
+    m = AnthropicModel('claude-3-5-sonnet-latest', provider=AnthropicProvider(api_key=anthropic_api_key))
+    agent = Agent(m, instructions='You are a helpful assistant.')
+
+    result = await agent.run(
+        'I need a potato!',
+        message_history=[
+            ModelRequest(parts=[], instructions='You are a helpful assistant.', kind='request'),
+            ModelResponse(parts=[TextPart(content='Hello, how can I help you?')], kind='response'),
+        ],
+    )
+    assert result.output == snapshot("""\
+I can't physically give you a potato since I'm a computer program. 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
+
+What specifically would you like to know about potatoes?\
+""")

{pydantic_ai-0.2.9 → pydantic_ai-0.2.11}/tests/test_mcp.py

@@ -1,6 +1,7 @@
 """Tests for the MCP (Model Context Protocol) server implementation."""
 
 import re
+from datetime import timedelta
 from pathlib import Path
 
 import pytest
@@ -70,25 +71,40 @@ async def test_stdio_server_with_cwd():
         assert len(tools) == 10
 
 
-def test_sse_server():
-    sse_server = MCPServerHTTP(url='http://localhost:8000/sse')
-    assert sse_server.url == 'http://localhost:8000/sse'
-    assert sse_server._get_log_level() is None  # pyright: ignore[reportPrivateUsage]
+def test_http_server():
+    http_server = MCPServerHTTP(url='http://localhost:8000/sse')
+    assert http_server.url == 'http://localhost:8000/sse'
+    assert http_server._get_log_level() is None  # pyright: ignore[reportPrivateUsage]
 
 
-def test_sse_server_with_header_and_timeout():
-    sse_server = MCPServerHTTP(
+def test_http_server_with_header_and_timeout():
+    http_server = MCPServerHTTP(
         url='http://localhost:8000/sse',
         headers={'my-custom-header': 'my-header-value'},
         timeout=10,
         sse_read_timeout=100,
         log_level='info',
     )
-    assert sse_server.url == 'http://localhost:8000/sse'
-    assert sse_server.headers is not None and sse_server.headers['my-custom-header'] == 'my-header-value'
-    assert sse_server.timeout == 10
-    assert sse_server.sse_read_timeout == 100
-    assert sse_server._get_log_level() == 'info'  # pyright: ignore[reportPrivateUsage]
+    assert http_server.url == 'http://localhost:8000/sse'
+    assert http_server.headers is not None and http_server.headers['my-custom-header'] == 'my-header-value'
+    assert http_server.timeout == 10
+    assert http_server.sse_read_timeout == 100
+    assert http_server._get_log_level() == 'info'  # pyright: ignore[reportPrivateUsage]
+
+
+def test_http_server_with_timedelta_arguments():
+    http_server = MCPServerHTTP(
+        url='http://localhost:8000/sse',
+        headers={'my-custom-header': 'my-header-value'},
+        timeout=timedelta(seconds=10),  # type: ignore[arg-type]
+        sse_read_timeout=timedelta(seconds=100),  # type: ignore[arg-type]
+        log_level='info',
+    )
+    assert http_server.url == 'http://localhost:8000/sse'
+    assert http_server.headers is not None and http_server.headers['my-custom-header'] == 'my-header-value'
+    assert http_server.timeout == 10
+    assert http_server.sse_read_timeout == 100
+    assert http_server._get_log_level() == 'info'  # pyright: ignore[reportPrivateUsage]
 
 
 @pytest.mark.vcr()