scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-11-11 08:38:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 359

test(a2a): expect the bounded connect timeout in the pydantic_ai provider tests

Browse source 
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.
This commit is contained in:
Oliver Yu 2026-10-20 04:40:49 +00:00
parent 1165261baf
commit 122701f224
1 changed files with 13 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
tests/unit/a2a_protocol/providers/pydantic_ai_agents/test_pydantic_ai_agent_headers.py
View file

@ -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"