scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-14 23:21:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 156

fix(test): update openrouter image generation assertion to accept any image format

Browse source 
gemini-2.5-flash-image returns JPEG, not PNG. The assertion was hardcoded
to png after the model was swapped from gemini-2.5-flash-image-preview
(which returned PNG) in commit 34e8e972.
This commit is contained in:
Ishaan Jaffer 2026-03-07 16:50:59 -08:00
parent 596ae55108
commit 30476bc35b
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
tests/llm_translation/test_openrouter.py
View file

@ -30,7 +30,7 @@ def test_completion_openrouter_image_generation():
assert (
resp.choices[0]
.message.images[0]["image_url"]["url"]
.startswith("data:image/png;base64,")
.startswith("data:image/")
)