From 17791a2174f828f26428e0d5224cbf27ea89330e Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Mon, 22 Jun 2026 20:37:21 -0700 Subject: [PATCH] ocr: wrap rust bridge dict into OCRResponse at the call site --- litellm/ocr/main.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) 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] = (