Fix update deprecated model test

This commit is contained in:
Sameer Kankute 2026-03-12 18:34:15 +05:30
parent 7c70015a5f
commit 374c35a6b7

View file

@ -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"],
)