scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-09 22:31:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 86

Fix tests related to mediaResolution

Browse source
This commit is contained in:
Sameer Kankute 2025-11-26 21:45:50 +05:30
parent e49f21c918
commit 9a85ffceff
3 changed files with 13 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
tests/llm_translation/test_prompt_factory.py
View file

@ -567,7 +567,7 @@ def test_vertex_only_image_user_message():
},
]
response = _gemini_convert_messages_with_history(messages=messages)
response = _gemini_convert_messages_with_history(messages=messages, model="gemini-1.5-pro")
expected_response = [
{
@ -576,7 +576,7 @@ def test_vertex_only_image_user_message():
{
"inline_data": {
"data": "/9j/2wCEAAgGBgcGBQ",
"mimeType": "image/jpeg",
"mime_type": "image/jpeg",
}
},
{"text": " "},

20
tests/test_litellm/llms/vertex_ai/gemini/test_vertex_and_google_ai_studio_gemini.py
View file

@ -1811,9 +1811,9 @@ def test_media_resolution_from_detail_parameter():
break
assert image_part is not None
assert "inline_data" in image_part
# The TypedDict uses snake_case internally, but mediaResolution is camelCase in the dict
assert "mediaResolution" in image_part["inline_data"]
assert image_part["inline_data"]["mediaResolution"] == "high"
# The TypedDict uses snake_case internally, and we keep it as snake_case
assert "media_resolution" in image_part["inline_data"]
assert image_part["inline_data"]["media_resolution"] == "high"
def test_media_resolution_low_detail():
@ -1851,7 +1851,7 @@ def test_media_resolution_low_detail():
break
assert image_part is not None
assert "inline_data" in image_part
assert image_part["inline_data"]["mediaResolution"] == "low"
assert image_part["inline_data"]["media_resolution"] == "low"
def test_media_resolution_auto_detail():
@ -1888,8 +1888,8 @@ def test_media_resolution_auto_detail():
break
assert image_part is not None
assert "inline_data" in image_part
# mediaResolution should not be set for auto
assert "mediaResolution" not in image_part["inline_data"] or image_part["inline_data"].get("mediaResolution") is None
# media_resolution should not be set for auto
assert "media_resolution" not in image_part["inline_data"] or image_part["inline_data"].get("media_resolution") is None
# Test with None
messages_none = [
@ -1915,8 +1915,8 @@ def test_media_resolution_auto_detail():
break
assert image_part is not None
assert "inline_data" in image_part
# mediaResolution should not be set
assert "mediaResolution" not in image_part["inline_data"] or image_part["inline_data"].get("mediaResolution") is None
# media_resolution should not be set
assert "media_resolution" not in image_part["inline_data"] or image_part["inline_data"].get("media_resolution") is None
def test_media_resolution_per_part():
@ -1966,12 +1966,12 @@ def test_media_resolution_per_part():
# First image should have low resolution (first part is the image)
image1_part = contents[0]["parts"][0]
assert "inline_data" in image1_part
assert image1_part["inline_data"]["mediaResolution"] == "low"
assert image1_part["inline_data"]["media_resolution"] == "low"
# Second image should have high resolution (third part is the second image)
image2_part = contents[0]["parts"][2]
assert "inline_data" in image2_part
assert image2_part["inline_data"]["mediaResolution"] == "high"
assert image2_part["inline_data"]["media_resolution"] == "high"
def test_media_resolution_only_for_gemini_3_models():

2
tests/test_litellm/llms/vertex_ai/test_vertex.py
View file

@ -1241,7 +1241,7 @@ def test_process_gemini_image():
base64_image = "data:image/jpeg;base64,/9j/4AAQSkZJRg..."
base64_result = _process_gemini_image(base64_image)
print("base64_result", base64_result)
assert base64_result["inline_data"]["mimeType"] == "image/jpeg"
assert base64_result["inline_data"]["mime_type"] == "image/jpeg"
assert base64_result["inline_data"]["data"] == "/9j/4AAQSkZJRg..."