open-git/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2025-12-06 11:33:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
Ishaan Jaffer
2025-11-26 17:08:06 -08:00
parent 831694897e
commit 0346d1ea23
1 changed files with 0 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
tests/image_gen_tests/test_image_generation.py
View File

@@ -135,22 +135,6 @@ class TestVertexAIGeminiImageGeneration(BaseImageGenTest):
}
class TestVertexAIGemini3ProImageGeneration(BaseImageGenTest):
"""Test Gemini 3 Pro image generation model"""
def get_base_image_generation_call_args(self) -> dict:
# comment this when running locally
load_vertex_ai_credentials()
litellm.in_memory_llm_clients_cache = InMemoryCache()
return {
"model": "vertex_ai/gemini-3-pro-image-preview",
"vertex_ai_project": "pathrise-convert-1606954137718",
"vertex_ai_location": "us-central1",
"n": 1,
"size": "1024x1024",
}
class TestBedrockNovaCanvasTextToImage(BaseImageGenTest):
def get_base_image_generation_call_args(self) -> dict:
litellm.in_memory_llm_clients_cache = InMemoryCache()