o Zhܱ @sddlZddlZddlZddlZddlZddlmZmZmZddl m Z ddl m Z ddl m Z ddl mZddl mZdd lmZdd l mZdd l mZdd lmZmZed Z dKdedeeefdeedefddZ dKdedeeefdeedefddZ dKdedeeefdeedefddZ dKdedeeefdeedefddZ! dKdedeeefdeedefddZ" dKdedeeefdeedefddZ# dKdedeeefdeedefddZ$ dKdedeeefdeedefd d!Z% dKdedeeefdeedefd"d#Z& dKdedeeefdeedefd$d%Z' dKdedeeefdeedefd&d'Z( dKdedeeefdeedefd(d)Z) dKdedeeefdeedefd*d+Z* dKdedeeefdeedefd,d-Z+d.efd/d0Z,d.efd1d2Z- dKdedeeefdeedefd3d4Z. dKdedeeefdeedefd5d6Z/ dKdedeeefdeedefd7d8Z0 dKdedeeefdeedefd9d:Z1 dKdedeeefdeedefd;d<Z2 dKdedeeefdeedefd=d>Z3 dKdedeeefdeedefd?d@Z4 dKdedeeefdeedefdAdBZ5 dKdedeeefdeedefdCdDZ6 dKdedeeefdeedefdEdFZ7GdGdHdHe j8Z9GdIdJdJe j8Z:dS)LN)AnyOptionalUnion) urlencode) _api_module)_common) _transformers)types) BaseApiClient)get_value_by_path)set_value_by_path) AsyncPagerPagerzgoogle_genai.files api_client from_object parent_objectreturncCXi}t|dgdurt|ddgt|dgt|dgdur*t|ddgt|dg|SNZ page_size_queryZpageSizeZ page_tokenZ pageTokengetvsetvrrr to_objectrI/var/www/html/lang_env/lib/python3.10/site-packages/google/genai/files.py_ListFilesConfig_to_mldev% rcCrrrrrrr_ListFilesConfig_to_vertex;rr cCs6i}t|dgdurt|dgt|t|dg||S)Nconfig)rrrrrrr_ListFilesParameters_to_mldevQsr"cCs i}t|dgdurtd|S)Nr!/config parameter is not supported in Vertex AI.r ValueErrorrrrr_ListFilesParameters_to_vertexcsr&cCzi}t|dgdurt|dgt|dgt|dgdur(t|dgt|dgt|dgdur;t|dgt|dg|SNdetailsmessagecoderrrrr_FileStatus_to_mldevor,cCsPi}t|dgdurtdt|dgdurtdt|dgdur&td|S)Nr)z0details parameter is not supported in Vertex AI.r*z0message parameter is not supported in Vertex AI.r+z-code parameter is not supported in Vertex AI.r$rrrr_FileStatus_to_vertexsr.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|dgt|t|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_mldevHrGcCsXi}t|dgdurtdt|dgdurtdt|dgdur&tdt|dgdur2tdt|d gdur>td t|d gdurJtd t|d gdurVtdt|dgdurbtdt|dgdurntdt|dgdurztdt|dgdurtdt|dgdurtdt|dgdurtdt|dgdurtd|S)Nr0-name parameter is not supported in Vertex AI.r1z5display_name parameter is not supported in Vertex AI.r3z2mime_type parameter is not supported in Vertex AI.r5z3size_bytes parameter is not supported in Vertex AI.r7z4create_time parameter is not supported in Vertex AI.r9z8expiration_time parameter is not supported in Vertex AI.r;z4update_time parameter is not supported in Vertex AI.r=z4sha256_hash parameter is not supported in Vertex AI.r?z,uri parameter is not supported in Vertex AI.r@z5download_uri parameter is not supported in Vertex AI.rBz.state parameter is not supported in Vertex AI.rCz/source parameter is not supported in Vertex AI.rDz7video_metadata parameter is not supported in Vertex AI.rFz.error parameter is not supported in Vertex AI.r$rrrr_File_to_vertexs<rJcCs\i}t|dgdurt|dgt|t|dg|t|dgdur,t|dgt|dg|S)Nfiler!)rrrGrrrr_CreateFileParameters_to_mldevsrLcC8i}t|dgdurtdt|dgdurtd|S)NrKz-file parameter is not supported in Vertex AI.r!r#r$rrrr_CreateFileParameters_to_vertex rNc C^i}t|dgdurt|ddgt|t|dgt|dgdur-t|dgt|dg|SNr0_urlrKr!rrt t_file_namerrrr_GetFileParameters_to_mldev!rVcCrMNr0rIr!r#r$rrrr_GetFileParameters_to_vertex4rOrYc CrPrQrSrrrr_DeleteFileParameters_to_mldevCrWrZcCrMrXr$rrrr_DeleteFileParameters_to_vertexVrOr[ enum_valuecC"|tgdvrt|ddS)N)ZSTATE_UNSPECIFIED PROCESSINGZACTIVEZFAILED* enum value is not supported in Vertex AI.setr%r\rrr"_FileState_to_vertex_enum_validateerccCr])N)ZSOURCE_UNSPECIFIEDZUPLOADEDZ GENERATEDr_r`rbrrr#_FileSource_to_vertex_enum_validatejrdrecCr'r(rrrrr_FileStatus_from_mldevor-rfcCi}|SNrrrrr_FileStatus_from_vertexricCr/)Nr0r2r1r4r3r6r5r8r7r:r9r<r;r>r=r?rAr@rBrCrErDrF)rrrfrrrr_File_from_mldevrHrkcCrgrhrrrrr_File_from_vertexrjrlcsdit|dgdurtdgt|dgt|dgdur0tdgfddt|dgDS)NZ nextPageTokenZnext_page_tokenfilescsg|]}t|qSr)rk).0itemrrrr s z1_ListFilesResponse_from_mldev..r)rrrrrpr_ListFilesResponse_from_mldevs   rrcCrgrhrrrrr_ListFilesResponse_from_vertexrjrscC.i}t|dgdurt|dgt|dg|SNZ httpHeaders http_headersrrrrr_CreateFileResponse_from_mldevrwcCrtrurrrrr_CreateFileResponse_from_vertexrxrycCrgrhrrrrr_DeleteFileResponse_from_mldevrjrzcCrgrhrrrrr_DeleteFileResponse_from_vertexrjr{c@s(eZdZdddeejdejfddZdddejdeej dej fdd Z ddd e deej dejfd d Zddd e deejdejfd dZdddee ejejejfdeejdejfddZdddeejdeejfddZdddee ejejejfdeej de!fddZ"dS)FilesNr!r!rc Cstj|d}|jjrtdt|j|}|d}|r!d|}nd}|d}|r3|dt|}| ddd}t |t rG|d d}n t |d rS|durS|j }t|}t|}|jd |||}|jjrqt|j|}nt|j|}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}4This method is only supported in the default client.rRrmr?r!N http_optionsgetresponsekwargs)r _ListFilesParameters _api_clientvertexair%r"r format_maprpop isinstancedicthasattrrrconvert_to_dictencode_unserializable_typesrequestrsrrListFilesResponse_from_response model_dump_verify_response selfr!parameter_model request_dictrequest_url_dictpath query_paramsr response_dict return_valuerrr_listsL        z Files._listrKc Cstj||d}|jjrtdt|j|}|d}|r"d|}nd}|d}|r4|dt|}| ddd}t |t rH|dd}n t |drT|durT|j }t|}t|}|jd |||} |jjrrt|j| } nt|j| } tjj| |d } |j| | S NrKr!r~rRzupload/v1beta/filesrrr!rpostr)r _CreateFileParametersrrr%rLrrrrrrrrrrrrryrwCreateFileResponserrr rrKr!rrrrrrrrrrr_createcsN        z Files._creater0c Ctj||d}|jjrtdt|j|}|d}|r"d|}nd}|d}|r4|dt|}| ddd}t |t rH|d d}n t |d rT|durT|j }t|}t|}|jd |||} |jjrrt|j| } nt|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) r0r!r~rR files/{file}rrr!Nrrr)r _GetFileParametersrrr%rVrrrrrrrrrrrrrlrkFilerrr rr0r!rrrrrrrrrrrrsF         z Files.getc Cr) ahDeletes an existing file from the service. 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~rRrrrr!Nrdeleter)r _DeleteFileParametersrrr%rZrrrrrrrrrrrrr{rzDeleteFileResponserrrrrrrrsN        z Files.deletec Cs |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}ntjd d dd|j|jdd}|j|tj |dd} | j!dusd| j!vrt"d| j!d} t|t j r|j#|| |j} n |j#|| |j} 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`. UVertex AI does not support creating files. You can upload files to GCS files instead.r3r0r1Nfiles/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` argumentapplication/json resumablestartz Content-TypezX-Goog-Upload-ProtocolzX-Goog-Upload-Commandz#X-Goog-Upload-Header-Content-Lengthz!X-Goog-Upload-Header-Content-Type api_versionheadersrrx-goog-upload-urlPFailed to create file. Upload URL did not returned from the create file request.rKrr)'rrr%r UploadFileConfigrrrr3r0r1 startswithioIOBaserrtellseekosSEEK_ENDr5SEEK_SETfspathrisfileFileNotFoundErrorgetsize mimetypes guess_typer HttpOptionsrCreateFileConfigrvKeyErrorZ upload_filerrkr rrKr! config_modelZfile_objoffsetZfs_path_rrZ upload_urlZ return_filerrrupload(s               z Files.uploadcCstd|j|j|d|SNrmr})rrrr!rrrlists  z Files.listc Cs|jjrtdd}|rt|trtjd i|}n|}t|tjr+|jdur+tdt |j|}d|d}ddi}|dt |}d}t |d gdurUt |d g}|j ||}t|tjrg||_|St|tjrv|jdurv||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 @Vertex AI does not support the Files API. Use GCS files instead.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 :downloadaltmediarrr)rrr%rrr DownloadFileConfigrr@rTrUrrZ download_fileVideoZ video_bytesGeneratedVideoZvideo) rrKr!rr0rrrdatarrrdownloads<)    zFiles.download)#__name__ __module__ __qualname__rr ListFilesConfigOrDictrr FileOrDictCreateFileConfigOrDictrrstrGetFileConfigOrDictrrDeleteFileConfigOrDictrrrpathlibPathrPathLikerrUploadFileConfigOrDictrrrrrDownloadFileConfigOrDictbytesrrrrrr|sj K ; E K j r|c@s eZdZdddeejdejfddZdddejdeej dej fdd Z ddd e deej dejfd d Zddd e deejdejfd dZdddee ejejejfdeejdejfddZdddeejdeejfddZdddee ejfdeejdefddZ dS) AsyncFilesNr}r!rc stj|d}|jjrtdt|j|}|d}|r"d|}nd}|d}|r4|dt|}| ddd}t |t rH|d d}n t |d rT|durT|j }t|}t|}|jd |||IdH}|jjrut|j|}nt|j|}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~rRrmrrr!Nrrr)r rrrr%r"rrrrrrrrrrr async_requestrsrrrrrrrrrrrsN         zAsyncFiles._listrKc stj||d}|jjrtdt|j|}|d}|r#d|}nd}|d}|r5|dt|}| ddd}t |t rI|dd}n t |drU|durU|j }t|}t|}|jd |||IdH} |jjrvt|j| } nt|j| } tjj| |d } |j| | Sr)r rrrr%rLrrrrrrrrrrrrryrwrrrrrrrrr6sP         zAsyncFiles._creater0c tj||d}|jjrtdt|j|}|d}|r#d|}nd}|d}|r5|dt|}| ddd}t |t rI|d d}n t |d rU|durU|j }t|}t|}|jd |||IdH} |jjrvt|j| } nt|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) rr~rRrrrr!Nrrr)r rrrr%rVrrrrrrrrrrrrrlrkrrrrrrrrrpsH          zAsyncFiles.getc r) arDeletes an existing file from the service. 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~rRrrrr!Nrrr)r rrrr%rZrrrrrrrrrrrrr{rzrrrrrrrrrsP         zAsyncFiles.deletec s4|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}ntjd d d d|j|jdd}|j|tj |ddIdH} | j!dusd| j!vrt"d| j!d} t|t j r|j#|| |jIdH} n |j#|| |jIdH} tjj$t%|j| d|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`. rrNrrrrrrrrrrrrrrrrrrKrr)'rrr%r rrrrr3r0r1rrrrrrrrrr5rrrrrrrrrrrrrvrZasync_upload_filerrkrrrrrrs                 zAsyncFiles.uploadcs td|j|j|dIdH|Sr)rrrrrrrcszAsyncFiles.listc s|jjr tdd}|rt|trtjd i|}n|}t|j|}d|d}d}t |dgdur:t |dg}ddi}|rI|dt |}|j ||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) rNrrrrrrr) rrr%rrr rrTrUrrZasync_download_file) rrKr!rr0rrrrrrrrms.#    zAsyncFiles.download)!rrrrr rrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrsj K ; E K i  rrh);rloggingrrrtypingrrr urllib.parserrrrr rTr rr r rr rZpagersrr getLoggerloggerrobjectrr r"r&r,r.rGrJrLrNrVrYrZr[rcrerfrirkrlrrrsrwryrzr{Z BaseModuler|rrrrrs                       <  6                   <                  V