uPj dUdZddlZddlZddlZddlZddlmZddlmZddl m Z m Z ddl m Z ddlmZmZd Zeed <iZeeeefed <d ed efdZded eeeefdzfdZd ed eeeeffdZd ed eeeeffdZ ddededzdedzd edzfdZ ddededzdedzd edzfdZy)z GigaChat File Handler Handles file uploads to GigaChat API for image processing. GigaChat requires files to be uploaded first, then referenced by file_id. N)Final)verbose_logger)_get_httpx_clientget_async_httpx_client) LlmProviders)get_access_tokenget_access_token_asyncz+https://gigachat.devices.sberbank.ru/api/v1GIGACHAT_BASE_URL _file_cacheurlreturncdtj|jjS)z*Generate hash for URL to use as cache key.)hashlibsha256encode hexdigest)r s g/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/gigachat/file_handler.py _get_url_hashrs >>#**, ' 1 1 33data_urlctjd|}|sy|jd}|jd}tj|}|j ddj ddxsd }|||fS) zx Parse data URL (base64 image). Returns: Tuple of (content_bytes, content_type, extension) or None zdata:([^;]+);base64,(.+)Nr/;rjpg)rematchgroupbase64 b64decodesplit)rr content_type base64_data content_bytesexts r_parse_data_urlr(#s|887BE ++a.LQK!++K8M##C(,2237:CeC , ++rc tddi}|j|}|j|jjdd}|j ddj dd xsd }|j ||fS) z&Download image from URL synchronously. ssl_verifyFparams content-type image/jpegrrrrr)rgetraise_for_statusheadersr#contentr clientresponser$r's r_download_image_syncr66s%lE-BCFjjoH "**..~|LL##C(,2237:CeC   \3 ..rcDKttjddi}|j|d{}|j |j jdd}|j ddj d d xsd }|j||fS7iw) z'Download image from URL asynchronously.r*F llm_providerr,Nr-r.rrrrr)rrGIGACHATr/r0r1r#r2r3s r_download_image_asyncr;Bs*!**e$F#JJsO+H "**..~|LL##C(,2237:CeC   \3 .. ,s2B BA*B  image_url credentialsapi_basect|}|tvr"tjd|ddt|S t |}|r|\}}}tjdn(tjd|ddt |\}}}t jd|}t|} |xst} | d} td d i } | j| d d | id|||fiddid} | j| j}|jd}|r|t|<tjd||S#t$r }tj d|Yd}~yd}~wwxYw)a Upload file to GigaChat and return file_id (sync). Args: image_url: URL or base64 data URL of the image credentials: GigaChat credentials for auth api_base: Optional custom API base URL Returns: file_id string or None if upload failed Image found in cache: %s...NDecoded base64 image!Downloading image from URL: %s...P./filesr*Fr+ AuthorizationBearer filepurposegeneral<r1filesdatatimeoutid'File uploaded successfully, file_id: %s$Error uploading file to GigaChat: %s)rr rdebugr(r6uuiduuid4r r rpostr0jsonr/ Exceptionerrorr<r=r>url_hashparsedr&r$r'filename access_tokenbase_url upload_urlr4r5resultfile_ides rupload_file_syncreQs $I.H;:HSbMJ8$$'' 2 /5 ,M<  !7 8  !DiPSQSn U/CI/N ,M<!ZZ\N!C51/{; #7&7'j/ ),1FG ++ $ ~&>?Hm\BCY' &  !!# D) $+K !  !JG T CQGsDD?? E(E##E(c8Kt|}|tvr"tjd|ddt|S t |}|r|\}}}tjdn0tjd|ddt |d{\}}}t jd|}t|d{} |xst} | d} ttjd d i } | j| d d | id|||fiddidd{} | j| j}|j!d}|r|t|<tjd||S777\#t"$r }tj$d|Yd}~yd}~wwxYww)a Upload file to GigaChat and return file_id (async). Args: image_url: URL or base64 data URL of the image credentials: GigaChat credentials for auth api_base: Optional custom API base URL Returns: file_id string or None if upload failed r@NrArBrCrDrErFr*Fr8rGrHrIrJrKrLrMrQrRrS)rr rrTr(r;rUrVr r rrr:rWr0rXr/rYrZr[s rupload_file_asyncrgs $I.H;:HSbMJ8$$*' 2 /5 ,M<  !7 8  !DiPSQSn U5J95U/U ,M<!ZZ\N!C51%;;$GG #7&7'j/ .%.. %( !' $ ~&>?Hm\BCY' !,!   !!# D) $+K !  !JG T=0V H " CQGsg6FAE. E( /E.9E*:AE.E,AE.'F(E.*E.,E.. F7F FFF)NN)__doc__r!rrrUtypingrlitellm._loggingr&litellm.llms.custom_httpx.http_handlerrrlitellm.types.utilsr authenticatorr r r __annotations__r dictstrrtuplebytesr(r6r;rergrrrtsD +-CI5H&( U4S> "'4s4s4 ,c,eE3O&>t>Dj> 4Z >F#AAtADjA 4Z Ar