uPj2 6UddlZddlmZddlmZmZmZddlZddlm Z m Z m Z m Z m Z mZddlmZddlmZddlmZmZmZmZddlmZmZdd ldd lmZmZdd lm Z e Z!ee"d <d e#de$dzde$dzde%e$e$ffdZ&dee$efde dee$effdZ'de(de de dzfdZ)de de%e$effdZ*de dee$effdZ+de de%e$effdZ,e!j[de egedge!j[de egedge efde de defd Z.y)!N)Mapping)AnyFinalcast) APIRouterDepends HTTPExceptionRequestResponse UploadFile)ORJSONResponse)verbose_proxy_logger)OCR_REQUEST_FORMAT_HEADEROCR_REQUEST_FORMAT_PARAM OCRResponseparse_ocr_request_format)%convert_file_document_to_url_document get_mime_type)*)UserAPIKeyAuthuser_api_key_auth)ProxyBaseLLMRequestProcessingrouter file_contentfilename content_typereturnc|r"|jddjnd}|r|dk(r |r t|}td||xsddS)z Convert uploaded file bytes into a Mistral-format document dict with base64 data URI. Delegates to convert_file_document_to_url_document after resolving MIME type from the upload's content_type header or filename. ;rNzapplication/octet-streamfile)typer mime_type)splitstriprr)rrrr"s j/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/proxy/ocr_endpoints/endpoints.py_build_document_from_uploadr&s`7C ""3'*002I  %?? %h/I 0 "@&@  datarequestcL|jt}|jjt}||n|}||S t t |t r|jjn|}i|t|iS#t$r}tdd|id}~wwxYw)z Resolve the requested response format from the body or the `x-req-format` header. An explicit `req_format` in the body wins over the header. Nierror) status_codedetail) getrheadersrr isinstancestrr$lower ValueErrorr )r(r) body_value header_value raw_valuerequest_formates r%_with_request_formatr93s !9:J!//--.GHL%/%;zI G 8)3Is)CIOO  # # %!  >d =,n == GW4EFFGs9B B#BB#responsefastapi_responsec t|tsy|j}|ytt j |d|j jDcic]\}}|jdvr||c}}Scc}}w)z Return the provider's native payload when the caller asked for `req_format=native` and the provider config captured it, carrying over the LiteLLM response headers (cost, call id, etc.) built for the normalized response. Nzapplication/json)zcontent-length content-type)content media_typer/) r0rget_provider_native_responser orjsondumpsr/itemsr2)r:r;native_payloadkeyvalues r%_native_responserGGs h ,$AACN  ^,%/66<<> Uyy{"DD J   sBc2K |jd{}|jd}|t |t st |ds tdtt |}|jdd{|jd{}|s tdt||j|j }d |i}|jD]9\}}|d vr t |tr tj |||<5|||<;t%j&d |jd |d|j|S75#t$r}td|dd}~wwxYw77#tj"tf$r|||<YwxYww)z Extract OCR data from a multipart form request. Uses the cached form if already parsed by auth middleware, otherwise parses the form from the request. Returns: A dict with 'document', 'model', and any other OCR params. Nz%Failed to parse multipart form data: u. When using curl with --form/-F, do NOT set the Content-Type header manually — curl will set it automatically with the required boundary.r readzNMultipart OCR request must include a 'file' field with the document to processrzUploaded file is empty)rrrdocument)r rJzNOCR multipart form request parsed - model: %s, document_type: %s, filename: %smodelr!)form Exceptionr3r.r0r hasattrrseekrIr&rrrCr1jsonloadsJSONDecodeErrorrdebug) r)rLr8 uploaded_filerrJr( field_name field_values r%_parse_multipart_formrW]s #LLN*HHV$MZ z%JSZ[hjpSqijj]3M   Q  - 2 2 44L 1221!''"//H $.x"8D#'::< + K - -  k3 ' /#'::k#:Z  +D  +X   Ke+  3A37V V   " 4&((*5 /#.Z  /s{FEE EAF:E/;FE1A"F6E3>F E E,E''E,,F1F3FFFFcHKtt|d{|S7 w)zAParse an OCR request and apply the `x-req-format` header, if any.N)r9_parse_ocr_request_body)r)s r%_parse_ocr_requestrZs &=g&F F PP Fs "  "cHK|jjdd}d|jvrt|d{S |j d{}|s@t |dd(tjdt|d{Std tj|}t|tr|jd nd}t|tr|jd d k(r tdt|trEdD]@}|j|}t|ts%|j!ds7td|S77 #t $rd}YwxYw7#tj$r}td |d d}~wwxYww)a Parse an OCR request, supporting both JSON and multipart form data. JSON body (existing behavior): { "model": "mistral/mistral-ocr-latest", "document": {"type": "document_url", "document_url": "https://..."} } Multipart form data (new): - file: the uploaded file - model: model name (form field) - Any other OCR params as form fields (pages, include_image_base64, etc.) Returns: A dict suitable for passing to the OCR processing pipeline. r=zmultipart/form-dataNr'_formufOCR request body is empty but form data is available from middleware — processing as multipart form.zEmpty request body. For file uploads, use multipart/form-data content type with a file field. When using curl with --form/-F, do NOT set the Content-Type header manually.zInvalid JSON in request body: zy. Ensure the request body is valid JSON with Content-Type: application/json, or use multipart/form-data for file uploads.rJr!r zdocument type 'file' is not supported through the JSON API. To upload a local file, use multipart/form-data with a 'file' field. For JSON requests, use 'document_url' or 'image_url' document types.) document_url image_urlz reducto://zreducto:// file IDs are not accepted through the proxy OCR API; upload the file in the same request via multipart/form-data with a 'file' field, or pass an inline base64 data URI as the document URL.)r/r.r2rWbody RuntimeErrorgetattrrrSr3rArQrRr0dictr1 startswith)r)rr`r(r8doc url_field url_values r%rYrYs$"//--nbAL 2 2 44*7333\\^#  7GT * 6 & &x /w77 7    ll4(*4D$)?*%TC#tF!: S  #t6 I *I)S)i.B.B<.P B  KE4$ 8  ! ! ,QC0; ;   s=F"EF"E$E!E$2F"E6F" E85BF">F"F"!E$$ E3/F"2E33F"8F FFF"z/v1/ocrocr) dependenciesresponse_classtagsz/ocruser_api_key_dictcKddlm}m}m}m}m}m}m} m} m } m } m } i} tt|d{}t|}|j|||d|||||d| | | | || d{}t!||xs|S7M7#t"$r/}t|}|j%|||| d{7d}~wwxYww)a OCR endpoint for extracting text from documents and images. Supports two input modes: **1. JSON body** (Mistral OCR API compatible): ```bash curl -X POST "http://localhost:4000/v1/ocr" -H "Authorization: Bearer sk-1234" -H "Content-Type: application/json" -d '{ "model": "mistral-ocr", "document": { "type": "document_url", "document_url": "https://arxiv.org/pdf/2201.04234" } }' ``` **2. Multipart form file upload**: ```bash curl -X POST "http://localhost:4000/v1/ocr" -H "Authorization: Bearer sk-1234" -F "model=mistral-ocr" -F "file=@document.pdf" ``` Response format is normalized to the LiteLLM OCR schema by default. Providers that support it (Azure Document Intelligence) can return their own payload instead, with cost tracking unchanged, via `x-req-format: native` (or `"req_format": "native"` in the body). r) general_settings llm_router proxy_configproxy_logging_objselect_data_generator user_api_baseuser_max_tokens user_modeluser_request_timeoutuser_temperatureversionN)r(aocr)r)r;rl route_typerqrornrprrrKrurwrvrtrsrx)r8rlrqrx)litellm.proxy.proxy_serverrnrorprqrrrsrtrurvrwrxrcrZrbase_process_llm_requestrGrM_handle_llm_api_exception)r)r;rlrnrorprqrrrsrtrurvrwrxr( processorr:r8s r%rhrhsb    D" ,W5562t< ) B B-//!-%"7!-!5+'!!C!  & *:;GxG16  (  1t< 77// 8     sVCBB8B,B-BCBB B=$B82B53B88B==C)/rPcollections.abcrtypingrrrrAfastapirrr r r r fastapi.responsesr litellm._loggingr(litellm.llms.base_llm.ocr.transformationrrrrlitellm.ocr.mainrrlitellm.proxy._types$litellm.proxy.auth.user_api_key_authrr'litellm.proxy.common_request_processingrr__annotations__bytesr1rcr&r9objectrGrWrZrYpostrhr'r%rs ### TT,1 R"RQ Dj* #s(^ 2>wsCx0>7>wsTWxGX>(vhQUo,==T#s(^=@QgQ'#s(2CQ W7WtCH~Wt +,-!     +,-!   )00A(BV V V &V  V r'