qiLdZddlZddlmZmZmZddlZddlmZGddZ y)a Gemini File API client for uploading and managing files. This module provides a client for Google's Gemini File API, enabling file uploads, metadata retrieval, listing, and deletion. Files uploaded through this API can be referenced in Gemini generative model calls. N)AnyDictOptional)loggerc eZdZdZ ddedefdZ ddedeedeeeffd Z d edeeeffd Z dd e d eedeeeffdZ d ede fdZy) GeminiFileAPIaqClient for the Gemini File API. This class provides methods for uploading, fetching, listing, and deleting files through Google's Gemini File API. Files uploaded through this API remain available for 48 hours and can be referenced in calls to the Gemini generative models. Maximum file size is 2GB, with total project storage limited to 20GB. api_keybase_urlc.||_||_d|_y)zInitialize the Gemini File API client. Args: api_key: Google AI API key base_url: Base URL for the Gemini File API (default is the v1beta endpoint) z=https://generativelanguage.googleapis.com/upload/v1beta/filesN)_api_key _base_urlupload_base_url)selfr r s ^/opt/pipecat/venv/lib/python3.12/site-packages/pipecat/services/google/gemini_live/file_api.py__init__zGeminiFileAPI.__init__ s  !^N file_path display_namereturnc Ktjd|tj4d{}t j |\}}|sd}t |d5}|j}dddi}|rdd|ii}ddtt|d d } tjd |j|j|jd |j| | 4d{} | jdk7rK| jd{} tj d| t#d| jd| | j$j'd} | s6tj dt)| j$t#dtjd| dddd{tt|ddd} tjd |j| | |4d{} | jdk7rK| jd{} tj d| t#d| jd| | j+d{}tjd|j'dij'd|cdddd{cdddd{S7#1swYxxYw7775#1d{7swYFxYw7777L7=#1d{7swYnxYwdddd{7y#1d{7swYyxYww)a*Upload a file to the Gemini File API using the correct resumable upload protocol. Args: file_path: Path to the file to upload display_name: Optional display name for the file Returns: File metadata including uri, name, and display_name zUploading file: Nzapplication/octet-streamrbfiler resumablestartzapplication/json)zX-Goog-Upload-ProtocolX-Goog-Upload-Commandz#X-Goog-Upload-Header-Content-Lengthz!X-Goog-Upload-Header-Content-Typez Content-Typez Step 1: Getting upload URL from ?key=)headersjsonzError initiating file upload: zFailed to initiate upload: z - zX-Goog-Upload-URLzResponse headers: z!No upload URL in response headerszGot upload URL: 0zupload, finalize)zContent-LengthzX-Goog-Upload-OffsetrzStep 2: Uploading file data to )rdatazError uploading file data: zFailed to upload file: zFile uploaded successfully: name)rinfoaiohttp ClientSession mimetypes guess_typeopenreadstrlendebugrpostr statustexterror Exceptionrgetdictr)rrrsession mime_type_f file_datametadatarresponse error_text upload_urlupload_headers file_infos r upload_filezGeminiFileAPI.upload_file.s,  &yk23((*9 !9 !g$// :LIq6 i& %!FFH  %H"^\$BC+6)07:3y>7J5> 2 G LL;D >??c)'/}}!6JLL#A*!NO#&A(//ARRUV`Ua$bcc&--112EF !LL#5d8;K;K6L5M!NO#$GHH / |<= > >$#&c)n"5(+);N LL::,G H||JY|W ! ![c??c)'/}}!6JLL#>zl!KL#&=hoo=NcR\Q]$^__"*--/1  :9==QS;T;X;XY_;`:abc  ! ! !c9 !9 !9 ! % %$ >"7  > > > >0 !!62  !c9 !b ! ! !c9 !9 !9 !9 !9 !s,1MK$M)M K'1A?M0K41M4#K=K7B"K=: MK:A MLM#L9L:A LLM?MM M MMr"cKd|vr|jdd}tj4d{}|j|jd|d|j 4d{}|j dk7rH|jd{}tjd|td|j |jd{}|cdddd{cdddd{S777t7.77#1d{7swYnxYwdddd{7y#1d{7swYyxYww)zGet metadata for a file. Args: name: File name (or full path) Returns: File metadata /NrrzError getting file metadata: zFailed to get file metadata: ) splitr$r%r2r r r.r/rr0r1r)rr"r4r:r;r>s rget_filezGeminiFileAPI.get_filews' $;::c?2&D((* ! !g{{dnn%5QtfE$--#QR ! !V^??c)'/}}!6JLL#@ !MN#&CHOOCT$UVV"*--/1   ! ! ! ! ! ! !!62  ! ! ! ! ! ! ! ! ! !s1E DE 2D4)D*D4-#D DAD DD  D4*D +D4/ E ;D <E D4D D  D4 E  D DD D4" E -D0.E 4E:D= ;EE  page_size page_tokencrK|j|d}|r||d<tj4d{}|j|j|4d{}|j dk7rH|j d{}tjd|td|j |jd{}|cdddd{cdddd{S777t7.77#1d{7swYnxYwdddd{7y#1d{7swYyxYww)zList uploaded files. Args: page_size: Number of files to return per page page_token: Token for pagination Returns: List of files and next page token if available )keypageSize pageTokenN)paramsrzError listing files: zFailed to list files: ) r r$r%r2r r.r/rr0r1r)rrErFrKr4r:r;results r list_fileszGeminiFileAPI.list_filessI> ",F; ((*  g{{4>>&{A  X??c)'/}}!6JLL#8 !EF#&C3?AC;C5C; D"C7D" D7)C9*D71D"3C;5C;7D"9D7;D DD D" D7DD7"D4(D+ )D40D7c^Kd|vr|jdd}tj4d{}|j|jd|d|j 4d{}|j dk7rH|jd{}tjd|td|j  dddd{dddd{y777Z77 #1d{7swYnxYwdddd{7y#1d{7swYyxYww) zDelete a file. Args: name: File name (or full path) Returns: True if deleted successfully rArBNrrzError deleting file: zFailed to delete file: T) rCr$r%deleter r r.r/rr0r1)rr"r4r:r;s r delete_filezGeminiFileAPI.delete_files $;::c?2&D((*  g~~(8$uT]]O&TU  Ya??c)'/}}!6JLL#8 !EF#&=hoo=N$OPP       !6          s1D-C'D-2D)C)*D-#C1C+5C1 DC-D D-!C/"D-)D+C1-D/D-1D 7C:8D ?D D-DD-D*D! D*&D-)z6https://generativelanguage.googleapis.com/v1beta/files)N) N)__name__ __module__ __qualname____doc__r*rrrrr?rDintrMboolrPrrrrs-e _ _&) _=AG!G!,4SMG! c3hG!R!3!4S>!0@D/7} c3h6cdrr) rUr&typingrrrr$logururrrXrrr[s'&&hhr