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