o pkjk @sbddlZddlZddlZddlZddlZddlmZmZmZm Z ddl m Z ddl Z ddlmZddlmZddlmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZmZedZ d*deee efe!fdeee efdee effddZ" d*deee efe!fdeee efdee effddZ# d*deee efe!fdeee efdee effddZ$ d*deee efe!fdeee efdee effddZ% d*deee efe!fdeee efdee effddZ& d*deee efe!fdeee efdee effddZ' d*deee efe!fdeee efdee effddZ( d*deee efe!fdeee efdee effd d!Z) d*deee efe!fdeee efdee effd"d#Z* d*deee efe!fdeee efdee effd$d%Z+Gd&d'd'ej,Z-Gd(d)d)ej,Z.dS)+N)AnyOptionalUnionoverload) urlencode) _api_client) _api_module)_common) _extra_utils) _transformers)types)get_value_by_path)set_value_by_path) AsyncPagerPagerzgoogle_genai.files from_object parent_objectreturncC.i}t|dgdurt|dgt|dg|S)Nfilegetvsetvrr to_objectrP/home/djax/ivt_ai_plugin/venv/lib/python3.10/site-packages/google/genai/files.py_CreateFileParameters_to_mldev)rcC.i}t|dgdurt|dgt|dg|SNsdkHttpResponsesdk_http_responserrrrr_CreateFileResponse_from_mldev4 r$cC6i}t|dgdurt|ddgtt|dg|SNname_urlrrrt t_file_namerrrr_DeleteFileParameters_to_mldevA r-cCr r!rrrrr_DeleteFileResponse_from_mldevNr%r/cCr&r'r*rrrr_GetFileParameters_to_mldev[r.r0cCr)Nurisrrrrr)_InternalRegisterFilesParameters_to_mldevhrr2cCsXi}t|dgdurt|ddgt|dgt|dgdur*t|ddgt|dg|S)N page_size_querypageSize page_token pageTokenrrrrr_ListFilesConfig_to_mldevss r8cCs*i}t|dgdurtt|dg||S)Nconfig)rr8rrrr_ListFilesParameters_to_mldevsr:cCsi}t|dgdurt|dgt|dgt|dgdur(t|dgt|dgt|dgdur@t|dgddt|dgD|S)Nr"r# nextPageTokennext_page_tokenfilescSg|]}|qSrr.0itemrrr z1_ListFilesResponse_from_mldev..rrrrr_ListFilesResponse_from_mldevs rDcCs^i}t|dgdurt|dgt|dgt|dgdur-t|dgddt|dgD|S)Nr"r#r=cSr>rrr?rrrrBrCz5_RegisterFilesResponse_from_mldev..rrrrr!_RegisterFilesResponse_from_mldevs rEc @2eZdZdddeejdejfddZdddejdeej dej fdd Z ddd e deej dejfd d Zddd e deejdejfd dZdddeje deejdejfddZdddee eje ejfdeejdejfddZeddddee ejej ej!fdddeej"de#fddZ$edddee ejej ej!fdee eje ejfdeej"ddfddZ$ddddee ejej ej!fdeee eje ejfdeej"dee#fddZ$ddde%j&j'j(deje deejdejfddZ)dddeejde*ejfddZdS)FilesNr9r9rc CsFtj|d}|jjrtdt|}|d}|rd|}nd}|d}|r1|dt|}| ddd}|j durH|j j durH|j j }t |}t |}|jd|||}|js`int|j} |jjsnt| } tjj| t|ddrdt|j d dt|j d dt|j d dd inid } tj|jd| _|j| | SNrHiThis method is only supported in Gemini Developer API mode, not in Gemini Enterprise Agent Platform mode.r)r=r4?r9getresponse_schemaresponse_json_schemainclude_all_fieldsrMrNrOresponsekwargsheaders)r _ListFilesParametersrvertexai ValueErrorr:rL format_maprpopr9 http_optionsr convert_to_dictencode_unserializable_typesrequestbodyjsonloadsrDListFilesResponse_from_responsegetattr HttpResponserUr#_verify_response selfr9parameter_model request_dictrequest_url_dictpath query_paramsr[rR response_dict return_valuerrr_listsb          z Files._listrc Chtj||d}|jjrtdt|}|d}|r d|}nd}|d}|r2|dt|}| ddd}|j durI|j j durI|j j }t |}t |}|jd|||} |durtt|d drttj| d } |j| | S| jsyint| j} |jjst| } tjj| t|ddrdt|j d dt|j d dt|j d ddinid} |j| | SNrr9rJr)zupload/v1beta/filesr4rKr9postshould_return_http_responser#rMrNrOrPrQ)r _CreateFileParametersrrWrXrrLrYrrZr9r[r r\r]r^rdCreateFileResponserfr_r`rar$rc rhrr9rirjrkrlrmr[rRrornrrr_createn             z Files._creater(c Cs(tj||d}|jjrtdt|}|d}|r d|}nd}|d}|r2|dt|}| ddd}|j durI|j j durI|j j }t |}t |}|jd |||} | jsaint| j} tjj| t|ddrdt|j d dt|j d dt|j d dd inid} |j| | S)aiRetrieves the file information from the service. Args: name (str): The name identifier for the file to retrieve. config (GetFileConfig): Optional, configuration for the get method. Returns: File: The file information. Usage: .. code-block:: python file = client.files.get(name='files/...') print(file.uri) r(r9rJr) files/{file}r4rKr9NrLrMrNrOrPrQ)r _GetFileParametersrrWrXr0rLrYrrZr9r[r r\r]r^r_r`raFilercrdrf rhr(r9rirjrkrlrmr[rRrnrorrrrLKsZ          z Files.getc CsHtj||d}|jjrtdt|}|d}|r d|}nd}|d}|r2|dt|}| ddd}|j durI|j j durI|j j }t |}t |}|jd |||} | jsaint| j} |jjsot| } tjj| t|ddrdt|j d dt|j d dt|j d dd inid} tj| jd| _|j| | S)a]Deletes a remotely stored file. Args: name (str): The name identifier for the file to delete. config (DeleteFileConfig): Optional, configuration for the delete method. Returns: DeleteFileResponse: The response for the delete method Usage: .. code-block:: python client.files.delete(name='files/...') r|rJr)r}r4rKr9NdeleterMrNrOrPrQrT)r _DeleteFileParametersrrWrXr-rLrYrrZr9r[r r\r]r^r_r`rar/DeleteFileResponsercrdrerUr#rfrrrrrsh          z Files.deleter1c CrqNr1r9rJr)zfiles:registerr4rKr9rtrurvrMrNrOrPrQ)r _InternalRegisterFilesParametersrrWrXr2rLrYrrZr9r[r r\r]r^rdRegisterFilesResponserfr_r`rarErc rhr1r9rirjrkrlrmr[rRrornrrr_register_filesr{zFiles._register_filesc Cs^|jjrtdt}|r*t|trtjdi|}n|}tj|j|j |j d}nt}|j dur@|j ds@d|j |_ t j ||j|jd\}}}||_||_|j|tj|ddd}|jdusp|jjduspd |jjvrttd |jjd } t|tjr|jj|| |j|d } nt|} |jj| | |j|d } tjj| jd |r|d Sid S)aCalls the API to upload a file using a supported file service. Args: file: A path to the file or an `IOBase` object to be uploaded. If it's an IOBase object, it must be opened in blocking (the default) mode and binary mode. In other words, do not use non-blocking mode or text mode. The given stream must be seekable, that is, it must be able to call `seek()` on 'path'. config: Optional parameters to set `display_name`, `mime_type`, and `name`. =This method is only supported in the Gemini Developer client. mime_typer( display_nameNfiles/user_http_optionsuser_mime_typeTr[rursx-goog-upload-urlPFailed to create file. Upload URL was not returned from the create file request.r[rrQr)rrWrXr UploadFileConfig isinstancedictrrr(r startswithr prepare_resumable_uploadr[ size_bytesrzCreateFileConfigr#rUKeyErrorioIOBase upload_fileosfspathrcr` model_dump rhrr9 config_modelfile_objr[rrrR upload_url return_filefs_pathrrruploadBsf           z Files.upload destinationr9rcCdSNrrhrrr9rrrdownloadzFiles.downloadcCrrrrrrrrrc Cs|jjrtdd}|rt|trtjd i|}n|}t|tjr+|jdur+tdt |}d|d}ddi}|dt |}d}t |d gdurSt |d g}|jj |||d } |durzt|tjrk| |_| St|tjrz|jdurz| |j_| S) aDownloads a file's data from storage. Files created by `upload` can't be downloaded. You can tell which files are downloadable by checking the `source` or `download_uri` property. Note: When destination is None, this method returns the data as bytes. For `Video` and `GeneratedVideo` objects, if destination is None, it also sets the `video_bytes` property on the `Video` object. When destination is provided, chunks are streamed directly to disk or the writable stream without loading the full file into memory, and None is returned. Args: file: A file name, uri, or file object identifying which file to download. destination: Optional, a local file path or writable binary stream to write the file directly to in chunks, avoiding holding the full file in memory. config: Optional, configuration for the get method. Returns: bytes: The file data as bytes if destination is None; otherwise None. Usage: .. code-block:: python for file in client.files.list(): if file.download_uri is not None: break else: raise ValueError('No files found with a `download_uri`.') # Download to memory: data = client.files.download(file=file) # Stream directly to a file path (avoids memory spike): client.files.download(file=file, destination="output.mp4") # Download video object: video = types.Video(uri=file.uri) video_bytes = client.files.download(file=video) video.video_bytes rNOnly generated files can be downloaded, uploaded files can't be downloaded. You can tell which files are downloadable by checking the `source` or `download_uri` property.r :downloadaltmediarKr[r[rr)rrWrXrrr DownloadFileConfigr download_urir+r,rr download_fileVideo video_bytesGeneratedVideovideo) rhrrr9rr(rlrmr[datarrrrs@2     authcCst|tjjjs td|durt}n tj|}|j dd}|j p't }|j p,i}dd| D}t|j|}d||d<|jrL|j|d <||_ ||_ |j||d S) *Registers gcs files with the file service.:auth must be a google.auth.credentials.Credentials object.NTdeepcSi|] \}}||qSrlowerr@kvrrr z(Files.register_files..Bearer authorizationx-goog-user-projectr)rgoogler credentials CredentialsrXr RegisterFilesConfigmodel_validate model_copyr[ HttpOptionsrUitemsrget_token_from_credentialsquota_project_idrrhrr1r9r[rUtokenrrrregister_filess$     zFiles.register_filescCs|j}td||j|d|S)aLists all files from the service. Args: config (ListFilesConfig): Optional, configuration for the list method. Returns: A Pager object that contains one page of files. When iterating over the pager, it automatically fetches the next page if there are more. Usage: .. code-block:: python for file in client.files.list(config={'page_size': 10}): print(file.name) r=rH)rprrhr9 list_requestrrrlist s z Files.list)+__name__ __module__ __qualname__rr ListFilesConfigOrDictrbrp FileOrDictCreateFileConfigOrDictrxrzstrGetFileConfigOrDictrrLDeleteFileConfigOrDictrrbuiltinsrRegisterFilesConfigOrDictrrrrPathLikerrUploadFileConfigOrDictrrrrDownloadFileConfigOrDictbytesrrrrrrrrrrrrGs H P R [ S M     a "rGc @rF) AsyncFilesNrHr9rc sNtj|d}|jjrtdt|}|d}|r d|}nd}|d}|r2|dt|}| ddd}|j durI|j j durI|j j }t |}t |}|jd|||IdH}|jsdint|j} |jjsrt| } tjj| t|ddrdt|j d dt|j d dt|j d dd inid } tj|jd| _|j| | SrI)r rVrrWrXr:rLrYrrZr9r[r r\r] async_requestr_r`rarDrbrcrdrerUr#rfrgrrrrp?sh           zAsyncFiles._listrc ptj||d}|jjrtdt|}|d}|r!d|}nd}|d}|r3|dt|}| ddd}|j durJ|j j durJ|j j }t |}t |}|jd|||IdH} |durxt|d drxtj| d } |j| | S| js}int| j} |jjst| } tjj| t|ddrdt|j d dt|j d dt|j d ddinid} |j| | Srr)r rwrrWrXrrLrYrrZr9r[r r\r]rrdrxrfr_r`rar$rcryrrrrzp             zAsyncFiles._creater(c s0tj||d}|jjrtdt|}|d}|r!d|}nd}|d}|r3|dt|}| ddd}|j durJ|j j durJ|j j }t |}t |}|jd |||IdH} | jseint| j} tjj| t|ddrdt|j d dt|j d dt|j d dd inid} |j| | S)asRetrieves the file information from the service. Args: name (str): The name identifier for the file to retrieve. config (GetFileConfig): Optional, configuration for the get method. Returns: File: The file information. Usage: .. code-block:: python file = await client.aio.files.get(name='files/...') print(file.uri) r|rJr)r}r4rKr9NrLrMrNrOrPrQ)r r~rrWrXr0rLrYrrZr9r[r r\r]rr_r`rarrcrdrfrrrrrLs`           zAsyncFiles.getc sPtj||d}|jjrtdt|}|d}|r!d|}nd}|d}|r3|dt|}| ddd}|j durJ|j j durJ|j j }t |}t |}|jd |||IdH} | jseint| j} |jjsst| } tjj| t|ddrdt|j d dt|j d dt|j d dd inid} tj| jd| _|j| | S)agDeletes a remotely stored file. Args: name (str): The name identifier for the file to delete. config (DeleteFileConfig): Optional, configuration for the delete method. Returns: DeleteFileResponse: The response for the delete method Usage: .. code-block:: python await client.aio.files.delete(name='files/...') r|rJr)r}r4rKr9NrrMrNrOrPrQrT)r rrrWrXr-rLrYrrZr9r[r r\r]rr_r`rar/rrcrdrerUr#rfrrrrr'sj           zAsyncFiles.deleter1c rr)r rrrWrXr2rLrYrrZr9r[r r\r]rrdrrfr_r`rarErcrrrrr~rzAsyncFiles._register_filesc s|jjr tdt}|r+t|trtjdi|}n|}tj|j|j |j d}nt}|j durA|j dsAd|j |_ t j ||j|jd\}}}||_||_|j|tj|dddIdH}|jdusz|jjduszd |jjvr~d |jjvr~td d |jjvr|jjd } n|jjd } t|tjr|jj|| |j|d IdH} nt|} |jj| | |j|d IdH} tjj| jd |r|dSidS)a Calls the API to upload a file asynchronously using a supported file service. Args: file: A path to the file or an `IOBase` object to be uploaded. If it's an IOBase object, it must be opened in blocking (the default) mode and binary mode. In other words, do not use non-blocking mode or text mode. The given stream must be seekable, that is, it must be able to call `seek()` on 'path'. config: Optional parameters to set `display_name`, `mime_type`, and `name`. rrNrrTrrsrzX-Goog-Upload-URLrrrrQr)rrWrXr rrrrrr(rrr rr[rrzrr#rUrrrasync_upload_filerrrcr`rrrrrrsn              zAsyncFiles.uploadrrcdSrrrrrrrzAsyncFiles.downloadcrrrrrrrr)rc s|jjr tdd}|rt|trtjd i|}n|}t|tjr,|jdur,tdt |}d|d}d}t |dgdurGt |dg}ddi}|rV|d t |}|jj |||d IdH} |durt|tjrq| |_| St|tjr|jdur| |j_| S) aDownloads a file's data from the file service asynchronously. The Gemini Enterprise Agent Platform implementation of the API does not include the file service. Files created by `upload` can't be downloaded. You can tell which files are downloadable by checking the `download_uri` property. Note: When destination is None, this method returns the data as bytes. For `Video` and `GeneratedVideo` objects, if destination is None, it also sets the `video_bytes` property on the `Video` object. When destination is provided, chunks are streamed directly to disk or the writable stream without loading the full file into memory, and None is returned. Args: file: A file name, uri, or file object identifying which file to download. destination: Optional, a local file path or writable binary stream to write the file directly to in chunks, avoiding holding the full file in memory. config: Optional, configuration for the get method. Returns: bytes: The file data as bytes if destination is None; otherwise None. Usage: .. code-block:: python async for file in await client.aio.files.list(): if file.download_uri is not None: break else: raise ValueError('No files found with a `download_uri`.') # Download to memory: data = await client.aio.files.download(file=file) # Stream directly to a file path (avoids memory spike): await client.aio.files.download(file=file, destination="output.mp4") rNrrrr[rrrKrr)rrWrXrrr rrrr+r,rrasync_download_filerrrr) rhrrr9rr(rlr[rmrrrrr3sD0      rcst|tjjjs td|durt}n tj|}|j dd}|j p(t }|j p-i}dd| D}t|j|IdH}d||d<|jrP|j|d <||_ ||_ |j||d IdHS) rrNTrcSrrrrrrrrrz-AsyncFiles.register_files..rrrr)rrrrrrXr rrrr[rrUrr async_get_token_from_credentialsrrrrrrrs*      zAsyncFiles.register_filescs$|j}td||j|dIdH|S)aLists all files from the service asynchronously. Args: config (ListFilesConfig): Optional, configuration for the list method. Returns: A Pager object that contains one page of files. When iterating over the pager, it automatically fetches the next page if there are more. Usage: .. code-block:: python async for file in await client.aio.files.list(config={'page_size': 10}): print(file.name) r=rHN)rprrrrrrszAsyncFiles.list)+rrrrr rrbrprrrxrzrrrrLrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrr=s J P T [ S R     a $rr)/rrr`loggingrtypingrrrr urllib.parser google.authrrr r r r r+r rrrrpagersrr getLoggerloggerrrobjectrr$r-r/r0r2r8r:rDrE BaseModulerGrrrrrs