pydantic-ai 0.0.45__tar.gz → 0.0.46__tar.gz

{pydantic_ai-0.0.45 → pydantic_ai-0.0.46}/PKG-INFO

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

{pydantic_ai-0.0.45 → pydantic_ai-0.0.46}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "pydantic-ai"
-version = "0.0.45"
+version = "0.0.46"
 description = "Agent Framework / shim to use Pydantic with LLMs"
 authors = [
     { name = "Samuel Colvin", email = "samuel@pydantic.dev" },
@@ -36,7 +36,7 @@ classifiers = [
 ]
 requires-python = ">=3.9"
 dependencies = [
-    "pydantic-ai-slim[openai,vertexai,groq,anthropic,mistral,cohere,bedrock,cli,mcp]==0.0.45",
+    "pydantic-ai-slim[openai,vertexai,groq,anthropic,mistral,cohere,bedrock,cli,mcp]==0.0.46",
 ]
 
 [project.urls]
@@ -46,7 +46,7 @@ Documentation = "https://ai.pydantic.dev"
 Changelog = "https://github.com/pydantic/pydantic-ai/releases"
 
 [project.optional-dependencies]
-examples = ["pydantic-ai-examples==0.0.45"]
+examples = ["pydantic-ai-examples==0.0.46"]
 logfire = ["logfire>=2.3"]
 
 [tool.uv.sources]

{pydantic_ai-0.0.45 → pydantic_ai-0.0.46}/tests/conftest.py

@@ -200,7 +200,18 @@ def vcr_config():
 @pytest.fixture(autouse=True)
 async def close_cached_httpx_client() -> AsyncIterator[None]:
     yield
-    await cached_async_http_client().aclose()
+    for provider in [
+        'openai',
+        'anthropic',
+        'azure',
+        'google-gla',
+        'google-vertex',
+        'groq',
+        'mistral',
+        'cohere',
+        'deepseek',
+    ]:
+        await cached_async_http_client(provider=provider).aclose()
 
 
 @pytest.fixture(scope='session')

pydantic_ai-0.0.46/tests/graph/test_utils.py

@@ -0,0 +1,13 @@
+from threading import Thread
+
+from pydantic_graph._utils import get_event_loop
+
+
+def test_get_event_loop_in_thread():
+    def get_and_close_event_loop():
+        event_loop = get_event_loop()
+        event_loop.close()
+
+    thread = Thread(target=get_and_close_event_loop)
+    thread.start()
+    thread.join()

{pydantic_ai-0.0.45 → pydantic_ai-0.0.46}/tests/test_mcp.py

@@ -43,6 +43,19 @@ def test_sse_server():
     assert sse_server.url == 'http://localhost:8000/sse'
 
 
+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,
+    )
+    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
+
+
 async def test_agent_with_stdio_server(allow_model_requests: None, openai_api_key: str):
     server = MCPServerStdio('python', ['-m', 'tests.mcp_server'])
     model = OpenAIModel('gpt-4o', provider=OpenAIProvider(api_key=openai_api_key))

pydantic_ai-0.0.45/tests/graph/test_utils.py

@@ -1,20 +0,0 @@
-from threading import Thread
-
-from pydantic_graph._utils import run_until_complete
-
-
-def test_run_until_complete_in_main_thread():
-    async def run(): ...
-
-    run_until_complete(run())
-
-
-def test_run_until_complete_in_thread():
-    async def run(): ...
-
-    def get_and_close_event_loop():
-        run_until_complete(run())
-
-    thread = Thread(target=get_and_close_event_loop)
-    thread.start()
-    thread.join()