From 8cb66168bcf2967699c534732a221ed5ac47162b Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Fri, 10 Oct 2025 19:57:17 -0700 Subject: [PATCH] test fix --- tests/litellm_utils_tests/test_utils.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/tests/litellm_utils_tests/test_utils.py b/tests/litellm_utils_tests/test_utils.py index aec4a88d60f..889e3918b39 100644 --- a/tests/litellm_utils_tests/test_utils.py +++ b/tests/litellm_utils_tests/test_utils.py @@ -1182,10 +1182,8 @@ def test_async_http_handler(mock_async_client): assert call_args["transport"] == mock_transport assert call_args["event_hooks"] == event_hooks assert call_args["headers"] == headers - assert isinstance(call_args["limits"], httpx.Limits) - assert call_args["limits"].max_connections == concurrent_limit - assert call_args["limits"].max_keepalive_connections == concurrent_limit assert call_args["timeout"] == timeout + assert call_args["follow_redirects"] is True @mock.patch("httpx.AsyncClient") @@ -1224,12 +1222,10 @@ def test_async_http_handler_force_ipv4(mock_async_client): # Assert other parameters match assert call_args["event_hooks"] == event_hooks assert call_args["headers"] == headers - assert isinstance(call_args["limits"], httpx.Limits) - assert call_args["limits"].max_connections == concurrent_limit - assert call_args["limits"].max_keepalive_connections == concurrent_limit assert call_args["timeout"] == timeout assert isinstance(call_args["verify"], ssl.SSLContext) assert call_args["cert"] is None + assert call_args["follow_redirects"] is True finally: # Reset force_ipv4 to default