o Yi @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)NfilegetvsetvrrZ to_objectrL/home/kim/smarthome/.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+cCrrrrrrr_DeleteFileResponse_from_mldevNr#r-cCr$r%r(rrrr_GetFileParameters_to_mldev[r,r.cCr)Nurisrrrrr)_InternalRegisterFilesParameters_to_mldevhrr0cCsXi}t|dgdurt|ddgt|dgt|dgdur*t|ddgt|dg|S)NZ page_size_queryZpageSizeZ page_tokenZ pageTokenrrrrr_ListFilesConfig_to_mldevss r2cCs*i}t|dgdurtt|dg||S)Nconfig)rr2rrrr_ListFilesParameters_to_mldevsr4cCsi}t|dgdurt|dgt|dgt|dgdur(t|dgt|dgt|dgdur@t|dgddt|dgD|S)Nr r!Z nextPageTokenZnext_page_tokenfilescSg|]}|qSrr.0itemrrr z1_ListFilesResponse_from_mldev..rrrrr_ListFilesResponse_from_mldevs r<cCs^i}t|dgdurt|dgt|dgt|dgdur-t|dgddt|dgD|S)Nr r!r5cSr6rrr7rrrr:r;z5_RegisterFilesResponse_from_mldev..rrrrr!_RegisterFilesResponse_from_mldevs r=c @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)FilesNr3r3rc 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 Nr?=This method is only supported in the Gemini Developer client.r'r5r1?r3getresponsekwargsheaders)r _ListFilesParametersrvertexai ValueErrorr4rC format_maprpopr3 http_optionsr convert_to_dictencode_unserializable_typesrequestbodyjsonloadsr<ListFilesResponse_from_response model_dump HttpResponserHr!_verify_response selfr3parameter_model request_dictrequest_url_dictpath query_paramsrNrE 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 Nrr3rAr'zupload/v1beta/filesr1rBr3postshould_return_http_responser!rD)r _CreateFileParametersrrJrKrrCrLrrMr3rNr rOrPrQgetattrCreateFileResponserYrRrSrTr"rVrW r[rr3r\r]r^r_r`rNrErbrarrr_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&r3rAr' files/{file}r1rBr3NrCrD)r _GetFileParametersrrJrKr.rCrLrrMr3rNr rOrPrQrRrSrTFilerVrWrY r[r&r3r\r]r^r_r`rNrErarbrrrrC+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/...') rprAr'rqr1rBr3NdeleterDrG)r _DeleteFileParametersrrJrKr+rCrLrrMr3rNr rOrPrQrRrSrTr-DeleteFileResponserVrWrXrHr!rYrtrrrrulsJ         z Files.deleter/c Crd Nr/r3rAr'zfiles:registerr1rBr3rgrhrirD)r _InternalRegisterFilesParametersrrJrKr0rCrLrrMr3rNr rOrPrQrkRegisterFilesResponserYrRrSrTr=rVrW r[r/r3r\r]r^r_r`rNrErbrarrr_register_filesrozFiles._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`. rA mime_typer& display_nameNfiles/Zuser_http_optionsZuser_mime_typeTrNrhrfx-goog-upload-urlPFailed to create file. Upload URL did not returned from the create file request.rNrrDr)rrJrKr UploadFileConfig isinstancedictrsrr&r startswithr prepare_resumable_uploadrN size_bytesrnCreateFileConfigr!rHKeyErrorioIOBaseZ upload_fileosfspathrVrSrW r[rr3 config_modelZfile_objrNrrrEZ upload_urlZ return_fileZfs_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 rANzOnly 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 :downloadaltmediarBrNrr)rrJrKrrr DownloadFileConfigrsZ download_urir)r*rrZ download_fileVideoZ video_bytesGeneratedVideoZvideo) r[rr3rr&r_r`rNdatarrrdownload>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|] \}}||qSrlowerr8kvrrr z(Files.register_files..Bearer authorizationx-goog-user-projectry)rgoogler credentials CredentialsrKr RegisterFilesConfigmodel_validate model_copyrN HttpOptionsrHitemsrZget_token_from_credentialsquota_project_idr}r[rr/r3rNrHtokenrrrregister_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) r5r?)rcrr[r3Z list_requestrrrlists z Files.list))__name__ __module__ __qualname__rr ListFilesConfigOrDictrUrc FileOrDictCreateFileConfigOrDictrlrnstrGetFileConfigOrDictrsrCDeleteFileConfigOrDictrwrurRegisterFilesConfigOrDictr{r}rrPathLikerrUploadFileConfigOrDictrrrDownloadFileConfigOrDictbytesrrrrrrrrrrrr>s 8 @ B K C P V "r>c @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) AsyncFilesNr?r3rc 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| | Sr@)r rIrrJrKr4rCrLrrMr3rNr rOrP async_requestrRrSrTr<rUrVrWrXrHr!rYrZrrrrcsJ          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| | Sre)r rjrrJrKrrCrLrrMr3rNr rOrPrrkrlrYrRrSrTr"rVrWrmrrrrnR            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) rprAr'rqr1rBr3NrCrD)r rrrrJrKr.rCrLrrMr3rNr rOrPrrRrSrTrsrVrWrYrtrrrrCDsB          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/...') rprAr'rqr1rBr3NrurDrG)r rvrrJrKr+rCrLrrMr3rNr rOrPrrRrSrTr-rwrVrWrXrHr!rYrtrrrrusL          zAsyncFiles.deleter/c rrx)r rzrrJrKr0rCrLrrMr3rNr rOrPrrkr{rYrRrSrTr=rVrWr|rrrr}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)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`. rAr~NrrTrrfrzX-Goog-Upload-URLrrrrDr)rrJrKr rrrrsrr&rrr rrNrrnrr!rHrrrZasync_upload_filerrrVrSrWrrrrr 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) rANrrrNrrrBrr) rrJrKrrr rr)r*rrZasync_download_file) r[rr3rr&r_rNr`rrrrr^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..rrrry)rrrrrrKr rrrrNrrHrrZ async_get_token_from_credentialsrr}rrrrrs*      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) r5r?N)rcrrrrrrszAsyncFiles.list)'rrrrr rrUrcrrrlrnrrrsrCrrwrurrr{r}rrrrrrrrrrrrrrrrrrrrrs : @ D K C U  G $r)N)-rrSloggingrtypingrrr urllib.parserZ google.authrrrr r r r)r r rrrZpagersrr getLoggerloggerrrobjectrr"r+r-r.r0r2r4r<r=Z BaseModuler>rrrrrs