From 415c7b8f6ee3d637c8eefd12e29a1490fde4fd4a Mon Sep 17 00:00:00 2001 From: Sameer Kankute Date: Fri, 24 Apr 2026 17:27:16 +0530 Subject: [PATCH] Fix failing tests --- tests/test_litellm/integrations/test_rubrik.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/test_litellm/integrations/test_rubrik.py b/tests/test_litellm/integrations/test_rubrik.py index d543516f18..c2fe643a57 100644 --- a/tests/test_litellm/integrations/test_rubrik.py +++ b/tests/test_litellm/integrations/test_rubrik.py @@ -14,6 +14,7 @@ import pytest from litellm.integrations.custom_guardrail import ModifyResponseException from litellm.integrations.rubrik import RubrikLogger +from litellm.llms.custom_httpx.http_handler import AsyncHTTPHandler from tests.test_litellm.integrations.rubrik_test_helpers import ( make_inputs_with_tools, @@ -54,7 +55,7 @@ class TestInitialization: ) assert handler.logging_endpoint == "http://localhost:8080/v1/litellm/batch" assert handler.key == "test-api-key" - assert isinstance(handler.tool_blocking_client, httpx.AsyncClient) + assert isinstance(handler.tool_blocking_client, AsyncHTTPHandler) def test_init_with_constructor_params(self): with patch("asyncio.create_task", Mock()):