From 6ddc9873e5fbf36ed0ea9360cf39650c72ee1b04 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Mon, 20 May 2024 11:14:16 -0700 Subject: [PATCH] test vertex image gen test --- litellm/tests/test_image_generation.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/litellm/tests/test_image_generation.py b/litellm/tests/test_image_generation.py index 6acb28e5b4..4a5a8dac91 100644 --- a/litellm/tests/test_image_generation.py +++ b/litellm/tests/test_image_generation.py @@ -175,11 +175,14 @@ async def test_aimage_generation_bedrock_with_optional_params(): async def test_aimage_generation_vertex_ai(): from test_amazing_vertex_completion import load_vertex_ai_credentials + litellm.set_verbose = True + load_vertex_ai_credentials() try: response = await litellm.aimage_generation( prompt="An olympic size swimming pool", model="vertex_ai/imagegeneration@006", + vertex_ai_project="adroit-crow-413218", ) assert response.data is not None assert len(response.data) > 0