phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix img gen basic test

Browse source
This commit is contained in:
Ishaan Jaff 2024-11-12 16:46:21 -08:00
parent d2a1ac804b
commit e0f1f18339
1 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
tests/image_gen_tests/test_image_generation.py
View file

@ -104,7 +104,9 @@ def load_vertex_ai_credentials():
class TestVertexImageGeneration(BaseImageGenTest): class TestVertexImageGeneration(BaseImageGenTest):
def get_base_image_generation_call_args(self) -> dict: def get_base_image_generation_call_args(self) -> dict:
# load_vertex_ai_credentials() # comment this when running locally
load_vertex_ai_credentials()
litellm.in_memory_llm_clients_cache = {} litellm.in_memory_llm_clients_cache = {}
return { return {
"model": "vertex_ai/imagegeneration@006", "model": "vertex_ai/imagegeneration@006",