uPj%UdZddlmZddlmZddlmZmZddlm Z ddl m Z m Z ddl mZddlmZd Zeed <Gd d eZy )z- Azure AI OCR transformation implementation. )Final)verbose_logger)async_convert_url_to_base64convert_url_to_base64)get_azure_ai_auth_headers) DocumentTypeOCRRequestData)MistralOCRConfig)get_secret_strAZURE_AI_API_KEYAZURE_AI_OCR_API_KEY_ENV_VARc eZdZdZdfd ZdedzfdZ ddedededzd edzd edzdef d Z dd edzded ed edzdef d Z dedefdZ dedefdZ dede d edede f fd Zdede d edede f fd ZxZS)AzureAIOCRConfiga Azure AI OCR transformation configuration. Azure AI uses Mistral's OCR API but with a different endpoint format. Inherits transformation logic from MistralOCRConfig since they use the same format. Reference: Azure AI Foundry OCR documentation Important: Azure AI only supports base64 data URIs (data:image/..., data:application/pdf;base64,...). Regular URLs are not supported. returnNc"t|yN)super__init__)self __class__s m/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/azure_ai/ocr/transformation.pyrzAzureAIOCRConfig.__init__!s ctSr)r )rs rget_api_key_env_varz$AzureAIOCRConfig.get_api_key_env_var$s++rheadersmodelapi_keyapi_baselitellm_paramsc |tt}| td}| tdit||ddi|}|S)z Validate environment and return headers for Azure AI OCR. Authenticates with AZURE_AI_API_KEY, or with an Entra ID / OAuth token when no key is set. AZURE_AI_API_BASEaMissing Azure AI API Base - Set AZURE_AI_API_BASE environment variable or pass api_base parameter)rrz Content-Typezapplication/json)r r ValueErrorr)rrrrrrkwargss rvalidate_environmentz%AzureAIOCRConfig.validate_environment'so ?$%ABG  %&9:H  s  'W .   roptional_paramsc H| td|jd}|dS)aq Get complete URL for Azure AI OCR endpoint. Azure AI endpoint format: https:///providers/mistral/azure/ocr Args: api_base: Azure AI API base URL model: Model name (not used in URL construction) optional_params: Optional parameters Returns: Complete URL for Azure AI OCR endpoint r"/z/providers/mistral/azure/ocr)r#rstrip)rrrr&rr$s rget_complete_urlz!AzureAIOCRConfig.get_complete_urlJs:(  s  ??3'788rurlctjd|t|}tjdt||S)a& Synchronously convert a URL to a base64 data URI. Azure AI OCR doesn't have internet access, so we need to fetch URLs and convert them to base64 data URIs. Args: url: The URL to convert Returns: Base64 data URI string z:Azure AI OCR: Converting URL to base64 data URI (sync): %sr+4Azure AI OCR: Converted URL to data URI (length: %s))rdebugrlenrr+data_uris r_convert_url_to_data_uri_syncz.AzureAIOCRConfig._convert_url_to_data_uri_syncis? Y[^_0C8SUXYaUbcrcKtjd|t|d{}tjdt||S7%w)a' Asynchronously convert a URL to a base64 data URI. Azure AI OCR doesn't have internet access, so we need to fetch URLs and convert them to base64 data URIs. Args: url: The URL to convert Returns: Base64 data URI string z;Azure AI OCR: Converting URL to base64 data URI (async): %sr-Nr.)rr/rr0r1s r_convert_url_to_data_uri_asyncz/AzureAIOCRConfig._convert_url_to_data_uri_asyncsL Z\_`!< DDSUXYaUbc Es&AA&Adocumentc Btjd|t|tst dt ||j d}|j}|dk(rR|j dd}|r|jdstjd|j|} | |d<nV|d k(rQ|j d d} | r=| jds,tjd |j| } | |d <t |,d ||||d |S) a Transform OCR request for Azure AI, converting URLs to base64 data URIs (sync). Azure AI OCR doesn't have internet access, so we automatically fetch any URLs and convert them to base64 data URIs synchronously. Args: model: Model name document: Document dict from user optional_params: Already mapped optional parameters headers: Request headers **kwargs: Additional arguments Returns: OCRRequestData with JSON data z5Azure AI OCR transform_ocr_request (sync) - model: %sExpected document dict, got type document_urldata:z?Azure AI OCR: Converting document URL to base64 data URI (sync)r- image_urlz$$%cd==)=L4<$[1w, )+     rcrKtjd|t|tst dt ||j d}|j}|dk(rZ|j dd}|r|jdstjd|j|d {} | |d<n^|d k(rY|j d d} | rE| jds4tjd |j| d {} | |d <t |,d ||||d |S7|7w)a Transform OCR request for Azure AI, converting URLs to base64 data URIs (async). Azure AI OCR doesn't have internet access, so we automatically fetch any URLs and convert them to base64 data URIs asynchronously. Args: model: Model name document: Document dict from user optional_params: Already mapped optional parameters headers: Request headers **kwargs: Additional arguments Returns: OCRRequestData with JSON data z4Azure AI OCR async_transform_ocr_request - model: %sr8r9r:r;r<z@Azure AI OCR: Converting document URL to base64 data URI (async)r-Nr=z=Azure AI OCR: Converting image URL to base64 data URI (async)r>r?) rr/r@rAr#r9rBrCrDr5rrErFs rasync_transform_ocr_requestz,AzureAIOCRConfig.async_transform_ocr_requestsE0 SUZ[(D);DN;KLM M#,,v.&.mmo ~ %"*,,~r"BLL$;$;G$D$$%gh!%!D!D!D!VV7?$^4  $'||K$$%de!%!D!D!D!SS4<$[1w, )+     WTs%B3D76D37AD7D5D75D7)rN)NNNr)__name__ __module__ __qualname____doc__rstrrrAr%r*r3r5rr rErJ __classcell__)rs@rrrsL ,S4Z,##&* !!!t ! * ! t ! !P'+ 9*99 9 t 9 9>..7 7 7  7  7  7 r7 7 7  7  7  7 7 rrN)rNtypingrlitellm._loggingr:litellm.litellm_core_utils.prompt_templates.image_handlingrr"litellm.llms.azure_ai.common_utilsr(litellm.llms.base_llm.ocr.transformationrr 'litellm.llms.mistral.ocr.transformationr litellm.secret_managers.mainr r __annotations__rr?rrrYs@+IQD7&8e8s 's r