From 6c36dc269b4d493f4be85c36dfe4aca4c2172679 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Tue, 27 May 2025 18:51:50 -0700 Subject: [PATCH] test: fix test_vertexai_model_garden_model_completion --- tests/local_testing/test_amazing_vertex_completion.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/local_testing/test_amazing_vertex_completion.py b/tests/local_testing/test_amazing_vertex_completion.py index 4b54a631bc6..e35ab0a1348 100644 --- a/tests/local_testing/test_amazing_vertex_completion.py +++ b/tests/local_testing/test_amazing_vertex_completion.py @@ -3143,6 +3143,7 @@ async def test_vertexai_model_garden_model_completion( Using OpenAI compatible models from Vertex Model Garden """ litellm.use_aiohttp_transport = False # since this uses respx, we need to set use_aiohttp_transport to False + litellm.module_level_aclient = httpx.AsyncClient() load_vertex_ai_credentials() litellm.set_verbose = True