o Thߚ @sddlZddlZddlZddlZddlZddlmZmZmZddl m Z ddl m Z ddl m Z ddl mZddl mZdd l mZdd l mZdd lmZmZed Z d,d eeeefefdeeeefdeeeffddZ d,d eeeefefdeeeefdeeeffddZ d,d eeeefefdeeeefdeeeffddZ d,d eeeefefdeeeefdeeeffddZ d,d eeeefefdeeeefdeeeffddZ! d,d eeeefefdeeeefdeeeffddZ" d,d eeeefefdeeeefdeeeffddZ# d,d eeeefefdeeeefdeeeffddZ$ d,d eeeefefdeeeefdeeeffd d!Z% d,d eeeefefdeeeefdeeeffd"d#Z& d,d eeeefefdeeeefdeeeffd$d%Z' d,d eeeefefdeeeefdeeeffd&d'Z(Gd(d)d)e j)Z*Gd*d+d+e j)Z+dS)-N)AnyOptionalUnion) urlencode) _api_module)_common) _transformers)types)get_value_by_path)set_value_by_path) AsyncPagerPagerzgoogle_genai.files from_object parent_objectreturncCsXi}t|dgdurt|ddgt|dgt|dgdur*t|ddgt|dg|S)N page_size_querypageSize page_token pageTokengetvsetvrr to_objectrU/home/air/segue/gemini/backup/venv/lib/python3.10/site-packages/google/genai/files.py_ListFilesConfig_to_mldev%s rcCs4i}t|dgdurt|dgtt|dg||S)Nconfig)rrrrrrr_ListFilesParameters_to_mldev:sr cCzi}t|dgdurt|dgt|dgt|dgdur(t|dgt|dgt|dgdur;t|dgt|dg|SNdetailsmessagecoderrrrr_FileStatus_to_mldevIr&cC$i}t|dgdurt|dgt|dgt|dgdur(t|dgt|dgt|dgdur;t|dgt|dgt|dgdurNt|dgt|dgt|dgdurat|d gt|dgt|d gdurtt|d gt|d gt|d gdurt|d gt|d gt|dgdurt|dgt|dgt|dgdurt|dgt|dgt|dgdurt|dgt|dgt|dgdurt|dgt|dgt|dgdurt|dgt|dgt|dgdurt|dgt|dgt|dgdurt|dgtt|dg||S)Nname display_name displayName mime_typemimeType size_bytes sizeBytes create_time createTimeexpiration_timeexpirationTime update_time updateTime sha256_hash sha256Hashuri download_uri downloadUristatesourcevideo_metadata videoMetadataerror)rrr&rrrr_File_to_mldevZDr@cCsZi}t|dgdurt|dgtt|dg|t|dgdur+t|dgt|dg|S)Nfiler)rrr@rrrr_CreateFileParameters_to_mldevsrCcC\i}t|dgdurt|ddgtt|dgt|dgdur,t|dgt|dg|SNr)_urlrBrrrt t_file_namerrrr_GetFileParameters_to_mldevrJcCrDrErGrrrr_DeleteFileParameters_to_mldevrKrLcCr!r"rrrrr_FileStatus_from_mldevr'rMcCr()Nr)r+r*r-r,r/r.r1r0r3r2r5r4r7r6r8r:r9r;r<r>r=r?)rrrMrrrr_File_from_mldevrArNcsbit|dgdurtdgt|dgt|dgdur/tdgfddt|dgDS)N nextPageTokennext_page_tokenfilescsg|]}t|qSr)rN).0itemrrr sz1_ListFilesResponse_from_mldev..r)rrrrTr_ListFilesResponse_from_mldev s   rVcCi}|SNrrrrr_CreateFileResponse_from_mldevrYcCrWrXrrrrr_DeleteFileResponse_from_mldev'rZr[c@s(eZdZdddeejdejfddZdddejdeej dej fdd Z ddd e deej dejfd d Zddd e deejdejfd dZdddee eje ejfdeejdejfddZdddeejdeejfddZdddee ejejejfdeejdefddZ dS)FilesNrrrc Cstj|d}|jjrtdt|}|d}|rd|}nd}|d}|r1|dt|}| ddd}|j durH|j j durH|j j }t |}t |}|jd |||}|js`d nt|j} |jjsnt| } tjj| |d } |j| | S) aaLists all files from the service. Args: config (ListFilesConfig): Optional, configuration for the list method. Returns: ListFilesResponse: The response for the list method. Usage: .. code-block:: python pager = client.files.list(config={'page_size': 10}) for file in pager.page: print(file.name) r]=This method is only supported in the Gemini Developer client.rFrQr?rNgetresponsekwargs)r _ListFilesParameters _api_clientvertexai ValueErrorr r` format_maprpopr http_optionsrconvert_to_dictencode_unserializable_typesrequestbodyjsonloadsrVListFilesResponse_from_response model_dump_verify_response selfrparameter_model request_dictrequest_url_dictpath query_paramsrkrc response_dict return_valuerrr_list2s>         z Files._listrBc Cs&tj||d}|jjrtdt|}|d}|r d|}nd}|d}|r2|dt|}| ddd}|j durI|j j durI|j j }t |}t |}|jd|||} |durnt|d drntj| d } | S| jssd nt| j} |jjst| } tjj| |d } |j| | S NrBrr^rFzupload/v1beta/filesrr_rpostshould_return_http_response)sdk_http_responserarb)r _CreateFileParametersrfrgrhrCr`rirrjrrkrrlrmrngetattrCreateFileResponserorprqrYrsrtru rwrBrrxryrzr{r|rkrcr~r}rrr_createtsN           z Files._creater)c Ctj||d}|jjrtdt|}|d}|r d|}nd}|d}|r2|dt|}| ddd}|j durI|j j durI|j j }t |}t |}|jd |||} | jsad nt| j} |jjsot| } tjj| |d } |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)rr^rF files/{file}rr_rNr`rarb)r _GetFileParametersrfrgrhrJr`rirrjrrkrrlrmrnrorprqrNFilersrtru rwr)rrxryrzr{r|rkrcr}r~rrrr`s@         z Files.getc Cr) 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/...') rr^rFrrr_rNdeleterarb)r _DeleteFileParametersrfrgrhrLr`rirrjrrkrrlrmrnrorprqr[DeleteFileResponsersrtrurrrrrsD         z Files.deletec CsZ|jjrtdt}|r*t|trtjdi|}n|}tj|j|j |j d}nt}|j dur@|j ds@d|j |_ t|t j rw|jdurOtdt|dr]d|jvr]td|}|d tj|||_||tjn1t|}|rtj|st|d tj||_|jdurt|\|_}|jdurtd |r|jr|j}d |_d dd|j|jd|_ntj d d dd|j|jdd}|j!|tj"|ddd} | j#dus| j#jdusd| j#jvrt$d| j#jd} t|t j r |jj%|| |j|d} n |jj%|| |j|d} tjj&t'| j(d|r)|)dSidS)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 `diplay_name`, `mime_type`, and `name`. r^r,r)r*Nfiles/zfUnknown mime type: Could not determine the mimetype for your file please set the `mime_type` argumentmodeb'The file must be opened in binary mode.r is not a valid file path.iUnknown mime type: Could not determine the mimetype for your file please set the `mime_type` argumentraapplication/json resumablestartz Content-TypezX-Goog-Upload-ProtocolzX-Goog-Upload-Commandz#X-Goog-Upload-Header-Content-Lengthz!X-Goog-Upload-Header-Content-Type api_versionheadersTrkrrx-goog-upload-urlPFailed to create file. Upload URL did not returned from the create file request.rkrBrbr)*rfrgrhr UploadFileConfig isinstancedictrr,r)r* startswithioIOBasehasattrrtellseekosSEEK_ENDr.SEEK_SETfspathr{isfileFileNotFoundErrorgetsize mimetypes guess_typerkrr HttpOptionsrCreateFileConfigrKeyError upload_filersrNrprt rwrBr config_modelfile_objoffsetfs_path_rkrc upload_url return_filerrrupload8s                   z Files.uploadcCstd|j|j|d|SNrQr])rrrwrrrrlists  z Files.listc Cs|jjrtdd}|rt|trtjd i|}n|}t|tjr+|jdur+tdt |}d|d}ddi}|dt |}d}t |d gdurSt |d g}|jj ||d }t|tjrf||_|St|tjru|jduru||j_|S) arDownloads 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: This method returns the data as bytes. For `Video` and `GeneratedVideo` objects there is an additional side effect, that it also sets the `video_bytes` property on the `Video` object. Args: file (str): A file name, uri, or file object. Identifying which file to download. config (DownloadFileConfigOrDict): Optional, configuration for the get method. Returns: File: The file data as bytes. Usage: .. code-block:: python for file client.files.list(): if file.download_uri is not None: break else: raise ValueError('No files found with a `download_uri`.') data = client.files.download(file=file) # data = client.files.download(file=file.name) # data = client.files.download(file=file.download_uri) video = types.Video(uri=file.uri) video_bytes = client.files.download(file=video) video.video_bytes r^NzOnly 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 :downloadaltmediar_rkrr)rfrgrhrrr DownloadFileConfigrr9rHrIrr download_fileVideo video_bytesGeneratedVideovideo) rwrBrrr)r{r|rkdatarrrdownloads<)     zFiles.download)!__name__ __module__ __qualname__rr ListFilesConfigOrDictrrr FileOrDictCreateFileConfigOrDictrrstrGetFileConfigOrDictrr`DeleteFileConfigOrDictrrrrPathLikerrUploadFileConfigOrDictrrrrrDownloadFileConfigOrDictbytesrrrrrr\0sj F > D H u r\c@s eZdZdddeejdejfddZdddejdeej dej fdd Z ddd e deej dejfd d Zddd e deejdejfd dZdddee eje ejfdeejdejfddZdddeejdeejfddZdddee ejfdeejdefddZdS) AsyncFilesNr]rrc stj|d}|jjrtdt|}|d}|r d|}nd}|d}|r2|dt|}| ddd}|j durI|j j durI|j j }t |}t |}|jd |||IdH}|jsdd nt|j} |jjsrt| } tjj| |d } |j| | S) akLists all files from the service. Args: config (ListFilesConfig): Optional, configuration for the list method. Returns: ListFilesResponse: The response for the list method. Usage: .. code-block:: python pager = await client.aio.files.list(config={'page_size': 10}) for file in pager.page: print(file.name) r]r^rFrQrr_rNr`rarb)r rerfrgrhr r`rirrjrrkrrlrm async_requestrorprqrVrrrsrtrurvrrrr sD          zAsyncFiles._listrBc s.tj||d}|jjrtdt|}|d}|r!d|}nd}|d}|r3|dt|}| ddd}|j durJ|j j durJ|j j }t |}t |}|jd|||IdH} |durrt|d drrtj| d } | S| jswd nt| j} |jjst| } tjj| |d } |j| | Sr)r rrfrgrhrCr`rirrjrrkrrlrmrrrrorprqrYrsrtrurrrrrNsP           zAsyncFiles._creater)c  tj||d}|jjrtdt|}|d}|r!d|}nd}|d}|r3|dt|}| ddd}|j durJ|j j durJ|j j }t |}t |}|jd |||IdH} | jsed nt| j} |jjsst| } tjj| |d } |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) rr^rFrrr_rNr`rarb)r rrfrgrhrJr`rirrjrrkrrlrmrrorprqrNrrsrtrurrrrr`sF          zAsyncFiles.getc r) 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/...') rr^rFrrr_rNrrarb)r rrfrgrhrLr`rirrjrrkrrlrmrrorprqr[rrsrtrurrrrrsF          zAsyncFiles.deletec s|jjr tdt}|r+t|trtjdi|}n|}tj|j|j |j d}nt}|j durA|j dsAd|j |_ t|t j rx|jdurPtdt|dr^d|jvr^td|}|d tj|||_||tjn1t|}|rtj|st|d tj||_|jdurt|\|_}|jdurtd|r|jr|j}d |_d d d|j|jd|_ntj d d d d|j|jdd}|j!|tj"|dddIdH} | j#dus| j#jdusd| j#jvrd| j#jvrt$dd| j#jvr | j#jd} n| j#jd} t|t j r'|jj%|| |j|dIdH} n|jj%|| |j|dIdH} tjj&t'| j(d|rG|)dSidS)aCalls 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 `diplay_name`, `mime_type`, and `name`. r^rNrrrrrrrrarrrrrTrrrzX-Goog-Upload-URLrrrBrbr)*rfrgrhr rrrrr,r)r*rrrrrrrrrr.rrr{rrrrrrkrrrrrrrasync_upload_filersrNrprtrrrrrs                     zAsyncFiles.uploadcs td|j|j|dIdH|Sr)r rrrrrrszAsyncFiles.listc s|jjr tdd}|rt|trtjd i|}n|}t|}d|d}d}t |dgdur8t |dg}ddi}|rG|dt |}|jj ||d IdH}|S) atDownloads a file's data from the file service. The Vertex-AI implementation of the API foes 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. Args: File (str): A file name, uri, or file object. Identifying which file to download. config (DownloadFileConfigOrDict): Optional, configuration for the get method. Returns: File: The file data as bytes. Usage: .. code-block:: python for file client.files.list(): if file.download_uri is not None: break else: raise ValueError('No files found with a `download_uri`.') data = client.files.download(file=file) # data = client.files.download(file=file.name) # data = client.files.download(file=file.uri) r^Nrrrkrrr_rr) rfrgrhrrr rrHrIrrasync_download_file) rwrBrrr)r{rkr|rrrrrs.#     zAsyncFiles.download)rrrrr rrrrrrrrrrrr`rrrrrrrrrrr rrrrrrrrrsj H > F H z  rrX),rrploggingrrtypingrrr urllib.parserrarrr rHr r rr rpagersr r getLoggerloggerrrobjectrr r&r@rCrJrLrMrNrVrYr[ BaseModuler\rrrrrs                8          8        [