uPj UdZddlmZddlZddlmZddlmZmZm Z m Z m Z ddl Z ddl mZerddlmZmZGdd e ZGd d e ZGd d ZeZded<ddZeadaded<daded< deeeeeeed ddZddZddZd dZ d!dZ d!dZ!y)"z3Thin Python wrapper for the native Rust OCR bridge.) annotationsN) Awaitable) TYPE_CHECKINGAnyFinalProtocolcast)timeout_to_seconds) RustAmessages RustMessagesc8eZdZ ddZy)RustOcrc tNNotImplementedError selfmodeldocumentapi_keyapi_basecustom_llm_provider extra_headersoptional_paramstimeout_secondss \/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/rust_bridge/ocr.py__call__zRustOcr.__call__ "!N)rstrrdict[str, object]r str | Nonerr#rr#rdict[str, object] | Nonerr"r float | Nonereturnr"__name__ __module__ __qualname__rr rrrs_ " "$ " "  " ( "0 "+ "& "  "r rc8eZdZ ddZy)RustAocrc trrrs rrzRustAocr.__call__!rr N)rr!rr"rr#rr#rr#rr$rr"rr%r&zAwaitable[dict[str, object]]r'r+r rr-r- s_ " "$ " "  " ( "0 "+ "& " & "r r-c eZdZy)_UnsetN)r(r)r*r+r rr0r0/sr r0z Final[_Unset]_UNSETcjtjddjjdvS)NLITELLM_USE_RUST_OCR>1onyestrue)osgetenvstriplowerr+r r_env_enables_rust_ocrr=6s3 99+R 0 6 6 8 > > @E r RustOcr | None_rust_ocr_implRustAocr | None_rust_aocr_impl)ocraocrmessages amessagesresponses_websocket transcriptionatranscriptioncVt|t xst|t }t|t xst|t } t|t } t|t xst|t } |s| s| s|at|ts|at|ts|a| rddlm} | |||| s| sy| rTddlm} t|tst|ts | ||n#t|ts | |n | || rddl m }|| yy) Nr)configure_rust_transcription)enabledrGrH)set_rust_messages)rDrE)rD)rE)set_rust_responses_websocket) connection) isinstancer0_rust_ocr_enabledr?rA!litellm.rust_bridge.transcriptionrJlitellm.rust_bridge.messagesrL'litellm.rust_bridge.responses_websocketrM)rKrBrCrDrErFrGrHconfiguring_ocrconfiguring_messagesconfiguring_responses_websocketconfiguring_transcriptionrJrLrMs ruse_litellm_rustrXDs",C!88X 4QW@X ??r c Nt}|y||||||||t|SN)rrrrrrrr)ra_timeout_to_seconds) rrrrrrrtimeoutrust_ocrs rrBrBs>$oH /#'+G4  r c jKt}|y||||||||t|d{S7wrf)rdrg) rrrrrrrrh rust_aocrs rrCrCsM&'I/#'+G4    s *313)r&bool)T)rKrlrBzRustOcr | None | _UnsetrCzRustAocr | None | _UnsetrDzRustMessages | None | _UnsetrEzRustAmessages | None | _UnsetrFAny | None | _UnsetrGrmrHrmr&None)r&r>)r&r@)rr!rr"rr#rr#rr#rr$rr"rhzfloat | httpx.Timeout | Noner&r$)"__doc__ __future__rr9collections.abcrtypingrrrrr httpxlitellm.rust_bridge.timeoutsr rgrRr r rr-r0r1__annotations__r=rPr?rArXrZrardrBrCr+r rrvs9" %<< RH "h " "x "   *+!%%#'',E$*%+-3/5/5)/*0,E ,E !,E # ,E + ,E - ,E-,E',E(,E ,E^,@      $ ,'*4      $ ,'*r