diff --git a/tests/llm_translation/test_gemini.py b/tests/llm_translation/test_gemini.py index c9ee3625395..796b35b436e 100644 --- a/tests/llm_translation/test_gemini.py +++ b/tests/llm_translation/test_gemini.py @@ -271,7 +271,7 @@ def test_gemini_context_caching_separate_messages(): def test_gemini_image_generation(): # litellm._turn_on_debug() response = completion( - model="gemini/gemini-2.0-flash-exp-image-generation", + model="gemini/gemini-2.5-flash-image-preview", messages=[{"role": "user", "content": "Generate an image of a cat"}], modalities=["image", "text"], )