pydantic-ai 0.2.13__tar.gz → 0.2.14__tar.gz

{pydantic_ai-0.2.13 → pydantic_ai-0.2.14}/PKG-INFO

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

{pydantic_ai-0.2.13 → pydantic_ai-0.2.14}/tests/test_mcp.py

@@ -1,7 +1,6 @@
 """Tests for the MCP (Model Context Protocol) server implementation."""
 
 import re
-from datetime import timedelta
 from pathlib import Path
 
 import pytest
@@ -71,40 +70,25 @@ async def test_stdio_server_with_cwd():
         assert len(tools) == 10
 
 
-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():
+    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_with_header_and_timeout():
-    http_server = MCPServerHTTP(
+def test_sse_server_with_header_and_timeout():
+    sse_server = MCPServerHTTP(
         url='http://localhost:8000/sse',
         headers={'my-custom-header': 'my-header-value'},
         timeout=10,
         sse_read_timeout=100,
         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]
-
-
-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]
+    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]
 
 
 @pytest.mark.vcr()