diff --git a/litellm/tests/test_image_generation.py b/litellm/tests/test_image_generation.py index 886953c1a..6acb28e5b 100644 --- a/litellm/tests/test_image_generation.py +++ b/litellm/tests/test_image_generation.py @@ -173,6 +173,9 @@ async def test_aimage_generation_bedrock_with_optional_params(): @pytest.mark.asyncio async def test_aimage_generation_vertex_ai(): + from test_amazing_vertex_completion import load_vertex_ai_credentials + + load_vertex_ai_credentials() try: response = await litellm.aimage_generation( prompt="An olympic size swimming pool",