diff --git a/litellm/ocr/main.py b/litellm/ocr/main.py index 53edc9f087e..d8b711a20e5 100644 --- a/litellm/ocr/main.py +++ b/litellm/ocr/main.py @@ -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] = (