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: make rust_bridge a leaf (return raw dict, no litellm import) so the CodeQL autofix stops re-breaking it

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

16
litellm/ocr/rust_bridge.py
View file

@ -28,11 +28,15 @@ def rust_ocr(
api_key: str | None,
api_base: str | None,
optional_params: dict,
) -> "OCRResponse":
"""Run a Mistral OCR call end-to-end in Rust and wrap the result as ``OCRResponse``."""
import litellm_python_bridge
from litellm.llms.base_llm.ocr.transformation import OCRResponse
) -> dict:
"""Call the Rust bridge and return the raw OCR response dict.
return OCRResponse(
**litellm_python_bridge.ocr(model, document, api_key, api_base, optional_params)
Kept free of ``litellm`` imports so this module stays a leaf — the caller
(``litellm/ocr/main.py``) wraps the dict into an ``OCRResponse``. This avoids
the import edge CodeQL repeatedly flags (and auto-"fixes") as a cyclic import.
"""
import litellm_python_bridge
return litellm_python_bridge.ocr(
model, document, api_key, api_base, optional_params
)