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

test(ocr): cover supported raw image signatures
Some checks failed
LiteLLM Rust / rustfmt, clippy, test (push) Waiting to run
Details
Terraform Provider / gofmt, vet, build, test (push) Has been cancelled
Details
Terraform Provider / Provider endpoints vs proxy OpenAPI schema (push) Has been cancelled
Details

Browse source 
Co-Authored-By: Ishaan Jaffer <155045088+ishaan-berri@users.noreply.github.com>
This commit is contained in:
Devin AI 2026-07-18 04:35:56 +00:00
parent 2ce8be19c3
commit e4aafe41c4
1 changed files with 15 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
tests/test_litellm/ocr/test_ocr_file_input.py
View file

@ -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"