tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-07-03 19:07:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix test images gemini

Browse Source
This commit is contained in:
Ishaan Jaff
2025-09-08 17:22:59 -07:00
parent 6400552308
commit e37c1069a1
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/llm_translation/test_gemini.py
+5 -3
View File
@@ -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():