scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-07 02:59:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

fix: ci test

Browse source 
gemini 2.5 depricated
This commit is contained in:
Yuta Saito 2026-01-17 09:17:31 +09:00
parent bcc23a32b8
commit 34e8e97222
3 changed files with 5 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
tests/llm_translation/test_gemini.py
View file

@ -293,7 +293,7 @@ def test_gemini_image_generation():
@pytest.mark.parametrize( @pytest.mark.parametrize(
"model_name", "model_name",
[ [
"gemini/gemini-2.5-flash-image-preview", "gemini/gemini-2.5-flash-image",
"gemini/gemini-2.0-flash-preview-image-generation", "gemini/gemini-2.0-flash-preview-image-generation",
"gemini/gemini-3-pro-image-preview", "gemini/gemini-3-pro-image-preview",
], ],
@ -733,7 +733,7 @@ async def test_gemini_image_generation_async():
"content": "Generate an image of a banana wearing a costume that says LiteLLM", "content": "Generate an image of a banana wearing a costume that says LiteLLM",
} }
], ],
model="gemini/gemini-2.5-flash-image-preview", model="gemini/gemini-2.5-flash-image",
) )
CONTENT = response.choices[0].message.content CONTENT = response.choices[0].message.content
@ -762,7 +762,7 @@ async def test_gemini_image_generation_async_stream():
"content": "Generate an image of a banana wearing a costume that says LiteLLM", "content": "Generate an image of a banana wearing a costume that says LiteLLM",
} }
], ],
model="gemini/gemini-2.5-flash-image-preview", model="gemini/gemini-2.5-flash-image",
stream=True, stream=True,
) )

3
tests/llm_translation/test_gemini_image_usage.py
View file

@ -13,7 +13,7 @@ from litellm.types.utils import ImageResponse, ImageObject, ImageUsage
@pytest.mark.parametrize( @pytest.mark.parametrize(
"model_name", "model_name",
[ [
"gemini/gemini-2.5-flash-image-preview", "gemini/gemini-2.5-flash-image",
"gemini/gemini-2.0-flash-preview-image-generation", "gemini/gemini-2.0-flash-preview-image-generation",
"gemini/gemini-3-pro-image-preview", "gemini/gemini-3-pro-image-preview",
], ],
@ -211,4 +211,3 @@ def test_gemini_imagen_models_no_usage_extraction():
# For Imagen models, we don't extract usage from the predictions format # For Imagen models, we don't extract usage from the predictions format
# This test just ensures we don't crash # This test just ensures we don't crash

2
tests/llm_translation/test_openrouter.py
View file

@ -22,7 +22,7 @@ def test_completion_openrouter_reasoning_content():
def test_completion_openrouter_image_generation(): def test_completion_openrouter_image_generation():
litellm._turn_on_debug() litellm._turn_on_debug()
resp = litellm.completion( resp = litellm.completion(
model="openrouter/google/gemini-2.5-flash-image-preview", model="openrouter/google/gemini-2.5-flash-image",
messages=[{"role": "user", "content": "Generate an image of a cat"}], messages=[{"role": "user", "content": "Generate an image of a cat"}],
modalities=["image", "text"], modalities=["image", "text"],
) )