From e4aafe41c4a7e1a597a2cf8c600368a86da83224 Mon Sep 17 00:00:00 2001 From: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> Date: Sat, 18 Jul 2026 04:35:56 +0000 Subject: [PATCH] test(ocr): cover supported raw image signatures Co-Authored-By: Ishaan Jaffer <155045088+ishaan-berri@users.noreply.github.com> --- tests/test_litellm/ocr/test_ocr_file_input.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/tests/test_litellm/ocr/test_ocr_file_input.py b/tests/test_litellm/ocr/test_ocr_file_input.py index 5202e4dbada..a42ffd34231 100644 --- a/tests/test_litellm/ocr/test_ocr_file_input.py +++ b/tests/test_litellm/ocr/test_ocr_file_input.py @@ -153,6 +153,21 @@ class TestConvertFileDocumentToUrlDocument: assert result["type"] == "image_url" assert result["image_url"].startswith("data:image/png;base64,") + @pytest.mark.parametrize( + ("content", "mime_type"), + ( + (b"\xff\xd8\xff image", "image/jpeg"), + (b"GIF89a image", "image/gif"), + (b"RIFF\x00\x00\x00\x00WEBP image", "image/webp"), + (b"II*\x00 image", "image/tiff"), + ), + ) + def test_should_infer_supported_image_mime_from_raw_bytes(self, content: bytes, mime_type: str) -> None: + result = convert_file_document_to_url_document({"type": "file", "file": content}) + + assert result["type"] == "image_url" + assert result["image_url"].startswith(f"data:{mime_type};base64,") + def test_should_prefer_explicit_mime_over_sniffed_magic_number(self): content = b"%PDF-1.4 pretends to be a pdf"