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

ocr: wrap rust bridge dict into OCRResponse at the call site

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

4
litellm/ocr/main.py
View file

@ -268,7 +268,9 @@ def ocr(
rust_bridge = importlib.import_module("litellm.ocr.rust_bridge")
if custom_llm_provider == "mistral" and rust_bridge.rust_ocr_enabled():
return rust_bridge.rust_ocr(model, document, api_key, api_base, kwargs)
return OCRResponse(
**rust_bridge.rust_ocr(model, document, api_key, api_base, kwargs)
)
# Get provider config
ocr_provider_config: Optional[BaseOCRConfig] = (