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: modernize rust_bridge typing (PEP 604, drop typing.Any/Dict) to satisfy strict-rule gate

Browse source
This commit is contained in:
Ishaan Jaffer 2026-06-22 18:56:24 -07:00
parent 88a9e6ed57
commit a4b4965888
No known key found for this signature in database
1 changed files with 8 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
litellm/ocr/rust_bridge.py
View file

@ -6,7 +6,9 @@ then routes supported Mistral calls through the compiled ``litellm_python_bridge
extension, which performs the whole OCR call (URL, headers, HTTP, parse) in Rust.
"""
from typing import TYPE_CHECKING, Any, Dict, Optional
from __future__ import annotations
from typing import TYPE_CHECKING
if TYPE_CHECKING:
from litellm.llms.base_llm.ocr.transformation import OCRResponse
@ -27,11 +29,11 @@ def rust_ocr_enabled() -> bool:
def rust_ocr(
model: str,
document: Dict[str, Any],
api_key: Optional[str],
api_base: Optional[str],
optional_params: Dict[str, Any],
) -> "OCRResponse":
document: dict,
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