From 30476bc35b927f8e5f35f5b3a80450a7caa83e91 Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Sat, 7 Mar 2026 16:50:59 -0800 Subject: [PATCH] fix(test): update openrouter image generation assertion to accept any image format 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. --- tests/llm_translation/test_openrouter.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/llm_translation/test_openrouter.py b/tests/llm_translation/test_openrouter.py index ba70e99ebce..1c75e6d664d 100644 --- a/tests/llm_translation/test_openrouter.py +++ b/tests/llm_translation/test_openrouter.py @@ -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/") )