scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-08 03:08:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 115

Fix: test_transform_request_image_extraction

Browse source
This commit is contained in:
Sameer Kankute 2026-01-05 17:02:54 +05:30
parent 935824e4f9
commit c0f74c994a
1 changed files with 5 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
tests/test_litellm/llms/ollama/test_ollama_chat_transformation.py
View file

@ -216,10 +216,8 @@ class TestOllamaChatConfigResponseFormat:
# Verify image was extracted to images list
assert "images" in result["messages"][0]
assert len(result["messages"][0]["images"]) == 1
assert (
result["messages"][0]["images"][0]
== "data:image/jpeg;base64,/9j/4AAQSkZJRgABAQAAAQ..."
)
# Ollama expects pure base64 data without the data URL prefix
assert result["messages"][0]["images"][0] == "/9j/4AAQSkZJRgABAQAAAQ..."
def test_transform_request_multiple_images_extraction(self):
"""Test extraction of multiple images from a single message"""
@ -263,12 +261,9 @@ class TestOllamaChatConfigResponseFormat:
# Verify both images were extracted
assert "images" in result["messages"][0]
assert len(result["messages"][0]["images"]) == 2
assert (
result["messages"][0]["images"][0] == "data:image/jpeg;base64,image1data..."
)
assert (
result["messages"][0]["images"][1] == "data:image/png;base64,image2data..."
)
# Ollama expects pure base64 data without the data URL prefix
assert result["messages"][0]["images"][0] == "image1data..."
assert result["messages"][0]["images"][1] == "image2data..."
def test_transform_request_image_url_as_string(self):
"""Test handling of image_url as direct string (edge case)"""