From e37c1069a1abafbd12ae329aba599d5800d62cff Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Mon, 8 Sep 2025 17:22:59 -0700 Subject: [PATCH] fix test images gemini --- tests/llm_translation/test_gemini.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/tests/llm_translation/test_gemini.py b/tests/llm_translation/test_gemini.py index 5bb4440d4d..47e3aaa814 100644 --- a/tests/llm_translation/test_gemini.py +++ b/tests/llm_translation/test_gemini.py @@ -265,9 +265,11 @@ def test_gemini_image_generation(): ######################################################### # Important: Validate we did get an image in the response ######################################################### - assert response.choices[0].message.image is not None - assert response.choices[0].message.image["url"] is not None - assert response.choices[0].message.image["url"].startswith("data:image/png;base64,") + assert response.choices[0].message.images is not None + assert len(response.choices[0].message.images) > 0 + assert response.choices[0].message.images[0]["image_url"] is not None + assert response.choices[0].message.images[0]["image_url"]["url"] is not None + assert response.choices[0].message.images[0]["image_url"]["url"].startswith("data:image/png;base64,") def test_gemini_thinking():