o _ik @sVddlZddlZddlZddlZddlmZmZmZddlm Z ddl Z ddl m Z ddl mZddl mZddl mZdd l mZdd l mZdd lmZdd lmZdd lmZmZedZ d*deeeefefdeeeefdeeeffddZ d*deeeefefdeeeefdeeeffddZ! d*deeeefefdeeeefdeeeffddZ" d*deeeefefdeeeefdeeeffddZ# d*deeeefefdeeeefdeeeffddZ$ d*deeeefefdeeeefdeeeffddZ% d*deeeefefdeeeefdeeeffddZ& d*deeeefefdeeeefdeeeffd d!Z' d*deeeefefdeeeefdeeeffd"d#Z( d*deeeefefdeeeefdeeeffd$d%Z)Gd&d'd'ej*Z+Gd(d)d)ej*Z,dS)+N)AnyOptionalUnion) 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_objectrK/home/air/demo/back/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,cCrr rrrrr_DeleteFileResponse_from_mldevNr$r.cCr%r&r)rrrr_GetFileParameters_to_mldev[r-r/cCsXi}t|dgdurt|ddgt|dgt|dgdur*t|ddgt|dg|S)N page_size_querypageSize page_token pageTokenrrrrr_ListFilesConfig_to_mldevhs r5cCs*i}t|dgdurtt|dg||S)Nconfig)rr5rrrr_ListFilesParameters_to_mldev}sr7cCsi}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 rAcCr)Nurisrrrrr!_RegisterFilesParameters_to_mldevrrCcCs^i}t|dgdurt|dgt|dgt|dgdur-t|dgddt|dgD|S)Nr!r"r:cSr;rrr<rrrr?r@z5_RegisterFilesResponse_from_mldev..rrrrr!_RegisterFilesResponse_from_mldevs rDc @seZdZdddeejdejfddZdddejdeej dej fdd Z ddd e deej dejfd d Zddd e deejdejfd dZdddee deejdejfddZdddee eje ejfdeejdejfddZdddee ejejejfdeej de!fddZ"ddde#j$j%j&dee deejdejfddZ'dddeejde(ejfddZdS)FilesNr6r6rc Cstj|d}|jjrtdt|}|d}|rd|}nd}|d}|r1|dt|}| ddd}|j durH|j j durH|j j }t |}t |}|jd|||}|js`int|j} |jjsnt| } tjj| |d } tj|jd | _|j| | S NrF=This method is only supported in the Gemini Developer client.r(r:r1?r6getresponsekwargsheaders)r _ListFilesParametersrvertexai ValueErrorr7rJ format_maprpopr6 http_optionsr convert_to_dictencode_unserializable_typesrequestbodyjsonloadsrAListFilesResponse_from_response model_dump HttpResponserOr"_verify_response selfr6parameter_model request_dictrequest_url_dictpath query_paramsrUrL response_dict return_valuerrr_listsD         z Files._listrc C2tj||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| |d } |j| | S Nrr6rHr(zupload/v1beta/filesr1rIr6postshould_return_http_responser"rK)r _CreateFileParametersrrQrRrrJrSrrTr6rUr rVrWrXgetattrCreateFileResponser`rYrZr[r#r]r^ rbrr6rcrdrerfrgrUrLrirhrrr_createP            z Files._creater'c Cstj||d}|jjrtdt|}|d}|r d|}nd}|d}|r2|dt|}| ddd}|j durI|j j durI|j j }t |}t |}|jd |||} | jsaint| j} 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'r6rHr( files/{file}r1rIr6NrJrK)r _GetFileParametersrrQrRr/rJrSrrTr6rUr rVrWrXrYrZr[Filer]r^r` rbr'r6rcrdrerfrgrUrLrhrirrrrJ+s<         z Files.getc Cstj||d}|jjrtdt|}|d}|r d|}nd}|d}|r2|dt|}| ddd}|j durI|j j durI|j j }t |}t |}|jd |||} | jsaint| j} |jjsot| } tjj| |d } 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/...') rwrHr(rxr1rIr6NdeleterKrN)r _DeleteFileParametersrrQrRr,rJrSrrTr6rUr rVrWrXrYrZr[r.DeleteFileResponser]r^r_rOr"r`r{rrrr|lsJ         z Files.deleterBc Crk NrBr6rHr(zfiles:registerr1rIr6rnrorprK)r _RegisterFilesParametersrrQrRrCrJrSrrTr6rUr rVrWrXrrRegisterFilesResponser`rYrZr[rDr]r^ rbrBr6rcrdrerfrgrUrLrirhrrr_register_filesrvzFiles._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 `diplay_name`, `mime_type`, and `name`. rH mime_typer' display_nameNfiles/user_http_optionsuser_mime_typeTrUrormx-goog-upload-urlPFailed to create file. Upload URL did not returned from the create file request.rUrrKr)rrQrRr UploadFileConfig isinstancedictrzrr'r startswithr prepare_resumable_uploadrU size_bytesruCreateFileConfigr"rOKeyErrorioIOBase upload_fileosfspathr]rZr^ rbrr6 config_modelfile_objrUrrrL upload_url return_filefs_pathrrruploadsf           z Files.uploadc 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 rHNzOnly 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 :downloadaltmediarIrUrr)rrQrRrrr DownloadFileConfigrz download_urir*r+rr download_fileVideo video_bytesGeneratedVideovideo) rbrr6rr'rfrgrUdatarrrdownload>s<)     zFiles.downloadauthcCst|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 CredentialsrRr RegisterFilesConfigmodel_validate model_copyrU HttpOptionsrOitemsrget_token_from_credentialsquota_project_idrrbrrBr6rUrOtokenrrrregister_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:rF)rjrrbr6 list_requestrrrlists z Files.list))__name__ __module__ __qualname__rr ListFilesConfigOrDictr\rj FileOrDictCreateFileConfigOrDictrsrustrGetFileConfigOrDictrzrJDeleteFileConfigOrDictr~r|rRegisterFilesConfigOrDictrrrrPathLikerrUploadFileConfigOrDictrrrDownloadFileConfigOrDictbytesrrrrrrrrrrrrEs 8 @ B K C P V "rEc @szeZdZdddeejdejfddZdddejdeej dej fdd Z ddd e deej dejfd d Zddd e deejdejfd dZdddee deejdejfddZdddee eje ejfdeejdejfddZdddee ejfdeejdefddZ ddde!j"j#j$dee deejdejfddZ%dddeejde&ejfddZdS) AsyncFilesNrFr6rc stj|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| |d } tj|jd | _|j| | SrG)r rPrrQrRr7rJrSrrTr6rUr rVrW async_requestrYrZr[rAr\r]r^r_rOr"r`rarrrrjsJ          zAsyncFiles._listrc :tj||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| |d } |j| | Srl)r rqrrQrRrrJrSrrTr6rUr rVrWrrrrsr`rYrZr[r#r]r^rtrrrruR            zAsyncFiles._creater'c stj||d}|jjrtdt|}|d}|r!d|}nd}|d}|r3|dt|}| ddd}|j durJ|j j durJ|j j }t |}t |}|jd |||IdH} | jseint| j} 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) rwrHr(rxr1rIr6NrJrK)r ryrrQrRr/rJrSrrTr6rUr rVrWrrYrZr[rzr]r^r`r{rrrrJDsB          zAsyncFiles.getc stj||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| |d } 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/...') rwrHr(rxr1rIr6Nr|rKrN)r r}rrQrRr,rJrSrrTr6rUr rVrWrrYrZr[r.r~r]r^r_rOr"r`r{rrrr|sL          zAsyncFiles.deleterBc rr)r rrrQrRrCrJrSrrTr6rUr rVrWrrrrr`rYrZr[rDr]r^rrrrrrzAsyncFiles._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)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`. rHrNrrTrrmrzX-Goog-Upload-URLrrrrKr)rrQrRr rrrrzrr'rrr rrUrrurr"rOrrrasync_upload_filerrr]rZr^rrrrr sn              zAsyncFiles.uploadc 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) rHNrrrUrrrIrr) rrQrRrrr rr*r+rrasync_download_file) rbrr6rr'rfrUrgrrrrr^s.#     zAsyncFiles.downloadrcst|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)rrrrrrRr rrrrUrrOrr 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:rFN)rjrrrrrrszAsyncFiles.list)'rrrrr rr\rjrrrsrurrrzrJrr~r|rrrrrrrrrrrrrrrrrrrrrrrrrs : @ D K C U  G $r)N)-rrZloggingrtypingrrr urllib.parser google.authrrrr r r r*r r rrrpagersrr getLoggerloggerrrobjectrr#r,r.r/r5r7rArCrD BaseModulerErrrrrs