From 122701f2249aeed07eb5db476052844f4b26052a Mon Sep 17 00:00:00 2001 From: Oliver Yu Date: Tue, 20 Oct 2026 04:40:49 +0000 Subject: [PATCH] test(a2a): expect the bounded connect timeout in the pydantic_ai provider tests PydanticAIProviderConfig sends its float timeout through AsyncHTTPHandler.post, so connect is now min(timeout, HTTP_HANDLER_CONNECT_TIMEOUT_SECONDS) while read, write and pool keep the caller's value. --- .../test_pydantic_ai_agent_headers.py | 15 +++++++++++++-- 1 file changed, 13 insertions(+), 2 deletions(-) diff --git a/tests/unit/a2a_protocol/providers/pydantic_ai_agents/test_pydantic_ai_agent_headers.py b/tests/unit/a2a_protocol/providers/pydantic_ai_agents/test_pydantic_ai_agent_headers.py index 8d5e15b4b71..c11b069282c 100644 --- a/tests/unit/a2a_protocol/providers/pydantic_ai_agents/test_pydantic_ai_agent_headers.py +++ b/tests/unit/a2a_protocol/providers/pydantic_ai_agents/test_pydantic_ai_agent_headers.py @@ -11,6 +11,7 @@ import respx import litellm from litellm.a2a_protocol.providers.pydantic_ai_agents.config import PydanticAIProviderConfig +from litellm.constants import HTTP_HANDLER_CONNECT_TIMEOUT_SECONDS from litellm.a2a_protocol.providers.pydantic_ai_agents.transformation import ( PydanticAITransformation, ) @@ -240,7 +241,12 @@ async def test_provider_config_non_streaming_defaults_to_sixty_second_timeout(ag "id": "req-5", "result": {"kind": "message", "role": "agent", "parts": [{"kind": "text", "text": "ok"}], "messageId": ANY}, } - assert request.extensions["timeout"] == {"connect": 60.0, "read": 60.0, "write": 60.0, "pool": 60.0} + assert request.extensions["timeout"] == { + "connect": min(60.0, HTTP_HANDLER_CONNECT_TIMEOUT_SECONDS), + "read": 60.0, + "write": 60.0, + "pool": 60.0, + } assert request.headers["content-type"] == "application/json" @@ -259,7 +265,12 @@ async def test_provider_config_non_streaming_forwards_timeout_and_ignores_unrela request: Final = agent_route.calls.last.request assert response["id"] == "req-5" - assert request.extensions["timeout"] == {"connect": 5.5, "read": 5.5, "write": 5.5, "pool": 5.5} + assert request.extensions["timeout"] == { + "connect": min(5.5, HTTP_HANDLER_CONNECT_TIMEOUT_SECONDS), + "read": 5.5, + "write": 5.5, + "pool": 5.5, + } assert request.headers["x-trace-id"] == "abc-123"