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

test(ocr): assert rust_ocr returns the raw bridge dict

Browse source
This commit is contained in:
Ishaan Jaffer 2026-06-22 20:37:21 -07:00
parent 17791a2174
commit ca904f5607
No known key found for this signature in database
1 changed files with 5 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
tests/test_litellm/ocr/test_rust_bridge.py
View file

@ -66,7 +66,7 @@ def test_use_litellm_rust_toggles_flag():
assert rust_bridge.rust_ocr_enabled() is False
def test_rust_ocr_wraps_bridge_response(fake_bridge):
def test_rust_ocr_returns_bridge_dict(fake_bridge):
response = rust_bridge.rust_ocr(
model="mistral-ocr-latest",
document=DOCUMENT,
@ -75,9 +75,10 @@ def test_rust_ocr_wraps_bridge_response(fake_bridge):
optional_params={"include_image_base64": True},
)
assert isinstance(response, OCRResponse)
assert response.pages[0].markdown == "hello world"
assert response.model == "mistral-ocr-2505-completion"
# rust_ocr returns the raw dict; main.py wraps it into an OCRResponse.
assert isinstance(response, dict)
assert response["pages"][0]["markdown"] == "hello world"
assert response["model"] == "mistral-ocr-2505-completion"
assert fake_bridge[0]["model"] == "mistral-ocr-latest"
assert fake_bridge[0]["optional_params"] == {"include_image_base64": True}