o %g @sddlZddlZddlZddlZddlZddlmZmZmZddl m Z ddl m Z ddl m Z ddl mZddl mZdd lmZdd l mZdd l mZdd lmZmZed Z d.dedeeefdeedefddZ d.dedeeefdeedefddZ d.dedeeefdeedefddZ d.dedeeefdeedefddZ! d.dedeeefdeedefddZ" d.dedeeefdeedefddZ# d.dedeeefdeedefddZ$ d.dedeeefdeedefd d!Z% d.dedeeefdeedefd"d#Z& d.dedeeefdeedefd$d%Z' d.dedeeefdeedefd&d'Z( d.dedeeefdeedefd(d)Z)Gd*d+d+e j*Z+Gd,d-d-e j*Z,dS)/N)AnyOptionalUnion) urlencode) _api_module)_common) _transformers)types) BaseApiClient)get_value_by_path)set_value_by_path) AsyncPagerPagerzgoogle_genai.files api_client from_object parent_objectreturncCsXi}t|dgdurt|ddgt|dgt|dgdur*t|ddgt|dg|S)N page_size_querypageSize page_token pageTokengetvsetvrrr to_objectrT/mnt/skqttb/ctump_chatbot/chatbot/lib/python3.10/site-packages/google/genai/files.py_ListFilesConfig_to_mldev%s r cCs6i}t|dgdurt|dgt|t|dg||S)Nconfig)rrr rrrr_ListFilesParameters_to_mldev;sr"cCzi}t|dgdurt|dgt|dgt|dgdur(t|dgt|dgt|dgdur;t|dgt|dg|SNdetailsmessagecoderrrrr_FileStatus_to_mldevMr(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_mldev_HrBcCs\i}t|dgdurt|dgt|t|dg|t|dgdur,t|dgt|dg|S)Nfiler!)rrrBrrrr_CreateFileParameters_to_mldevsrEc C^i}t|dgdurt|ddgt|t|dgt|dgdur-t|dgt|dg|SNr+_urlrDr!rrt t_file_namerrrr_GetFileParameters_to_mldevrLc CrFrGrIrrrr_DeleteFileParameters_to_mldevrMrNcCr#r$rrrrr_FileStatus_from_mldevr)rOcCr*)Nr+r-r,r/r.r1r0r3r2r5r4r7r6r9r8r:r<r;r=r>r@r?rA)rrrOrrrr_File_from_mldevrCrPcsdit|dgdurtdgt|dgt|dgdur0tdgfddt|dgDS)N nextPageTokennext_page_tokenfilescsg|]}t|qSr)rP).0itemrrrr )s z1_ListFilesResponse_from_mldev..r)rrrrrVr_ListFilesResponse_from_mldevs   rXcCs.i}t|dgdurt|dgt|dg|S)N httpHeaders http_headersrrrrr_CreateFileResponse_from_mldev2sr[cCsi}|SNrrrrr_DeleteFileResponse_from_mldev>sr]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}|j durJ|j j durJ|j j }t |}t |}|jd |||}|jjsgt|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_=This method is only supported in the Gemini Developer client.rHrSr?r!Ngetresponsekwargs)r _ListFilesParameters _api_clientvertexai ValueErrorr"rb format_maprpopr! http_optionsrconvert_to_dictencode_unserializable_typesrequestrXListFilesResponse_from_response model_dump_verify_response selfr!parameter_model request_dictrequest_url_dictpath query_paramsrl response_dict return_valuerrr_listJsH         z Files._listrDc Cstj||d}|jjrtdt|j|}|d}|r"d|}nd}|d}|r4|dt|}| ddd}|j durK|j j durK|j j }t |}t |}|jd|||} |jjsht|j| } tjj| |d } |j| | S NrDr!r`rHzupload/v1beta/filesrrar!postrc)r _CreateFileParametersrgrhrirErbrjrrkr!rlrrmrnror[CreateFileResponserqrrrs rurDr!rvrwrxryrzrlr{r|rrr_createsJ         z Files._creater+c Ctj||d}|jjrtdt|j|}|d}|r"d|}nd}|d}|r4|dt|}| ddd}|j durK|j j durK|j j }t |}t |}|jd |||} |jjsht|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+r!r`rH files/{file}rrar!Nrbrc)r _GetFileParametersrgrhrirLrbrjrrkr!rlrrmrnrorPFilerqrrrs rur+r!rvrwrxryrzrlr{r|rrrrbsF          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`rHrrrar!Ndeleterc)r _DeleteFileParametersrgrhrirNrbrjrrkr!rlrrmrnror]DeleteFileResponserqrrrsrrrrrsJ         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| 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` argumentapplication/json resumablestartz Content-TypezX-Goog-Upload-ProtocolzX-Goog-Upload-Commandz#X-Goog-Upload-Header-Content-Lengthz!X-Goog-Upload-Header-Content-Type api_versionheadersrlrx-goog-upload-urlPFailed to create file. Upload URL did not returned from the create file request.rDrcr)(rgrhrir UploadFileConfig isinstancedictrr.r+r, startswithioIOBasehasattrrtellseekosSEEK_ENDr0SEEK_SETfspathryisfileFileNotFoundErrorgetsize mimetypes guess_typerl HttpOptionsrCreateFileConfigrZKeyError upload_filerqrPjsonrr rurDr! config_modelfile_objoffsetfs_path_rlrd upload_url return_filerrruploadTs               z Files.uploadcCstd|j|j|d|SNrSr_)rr}rur!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 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 :downloadaltmediararlr)rgrhrirrr DownloadFileConfigrr;rJrKrr download_fileVideo video_bytesGeneratedVideovideo) rurDr!rr+ryrzrldatarrrdownloads<)    zFiles.download)#__name__ __module__ __qualname__rr ListFilesConfigOrDictrpr} FileOrDictCreateFileConfigOrDictrrstrGetFileConfigOrDictrrbDeleteFileConfigOrDictrrrpathlibPathrPathLikerrUploadFileConfigOrDictrrrrrDownloadFileConfigOrDictbytesrrrrrr^Hsj J : F J i 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}|j durK|j j durK|j j }t |}t |}|jd |||IdH}|jjskt|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`rHrSrrar!Nrbrc)r rfrgrhrir"rbrjrrkr!rlrrmrn async_requestrXrprqrrrsrtrrrr}sJ          zAsyncFiles._listrDc stj||d}|jjrtdt|j|}|d}|r#d|}nd}|d}|r5|dt|}| ddd}|j durL|j j durL|j j }t |}t |}|jd|||IdH} |jjslt|j| } tjj| |d } |j| | Sr~)r rrgrhrirErbrjrrkr!rlrrmrnrr[rrqrrrsrrrrr`sL          zAsyncFiles._creater+c tj||d}|jjrtdt|j|}|d}|r#d|}nd}|d}|r5|dt|}| ddd}|j durL|j j durL|j j }t |}t |}|jd |||IdH} |jjslt|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`rHrrrar!Nrbrc)r rrgrhrirLrbrjrrkr!rlrrmrnrrPrrqrrrsrrrrrbsH           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`rHrrrar!Nrrc)r rrgrhrirNrbrjrrkr!rlrrmrnrr]rrqrrrsrrrrrsL          zAsyncFiles.deletec s6|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| 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`. r`rNrrrrrrrrrrrrrrrrrrDrcr)(rgrhrir rrrrr.r+r,rrrrrrrrrr0rrryrrrrrrlrrrrZrasync_upload_filerqrPrrrrrrrr$s                 zAsyncFiles.uploadcs td|j|j|dIdH|Sr)rr}rrrrrszAsyncFiles.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) r`Nrrrlrrrar) rgrhrirrr rrJrKrrasync_download_file) rurDr!rr+ryrlrzrrrrrs.#    zAsyncFiles.download)!rrrrr rrpr}rrrrrrrrbrrrrrrrrrrrrrrrrrrrrrrsj J : F J h  rr\)-rloggingrrrtypingrrr urllib.parserrrrr rJr rgr r rr rpagersrr getLoggerloggerrobjectr r"r(rBrErLrNrOrPrXr[r] BaseModuler^rrrrrs                 <          <       S