o i?{ @sddlZddlZddlZddlZddlmZmZmZddlm Z ddl m Z ddl m Z ddl m Z ddl mZdd l mZdd l mZdd l mZdd lmZmZed Z d%deeeefefdeeeefdeeeffddZ d%deeeefefdeeeefdeeeffddZ d%deeeefefdeeeefdeeeffddZ d%deeeefefdeeeefdeeeffddZ d%deeeefefdeeeefdeeeffddZ! d%deeeefefdeeeefdeeeffddZ" d%deeeefefdeeeefdeeeffddZ# d%deeeefefdeeeefdeeeffdd Z$Gd!d"d"e j%Z&Gd#d$d$e j%Z'dS)&N)AnyOptionalUnion) urlencode) _api_module)_common) _extra_utils) _transformers)types)get_value_by_path)set_value_by_path) AsyncPagerPagerzgoogle_genai.files from_object parent_objectreturncCs.i}t|dgdurt|dgt|dg|S)Nfilegetvsetvrr to_objectrX/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/google/genai/files.py_CreateFileParameters_to_mldev&srcC.i}t|dgdurt|dgt|dg|SNsdkHttpResponsesdk_http_responserrrrr_CreateFileResponse_from_mldev1 r cC6i}t|dgdurt|ddgtt|dg|SNname_urlrrrt t_file_namerrrr_DeleteFileParameters_to_mldev> r)cCrrrrrrr_DeleteFileResponse_from_mldevKr!r+cCr"r#r&rrrr_GetFileParameters_to_mldevXr*r,cCsXi}t|dgdurt|ddgt|dgt|dgdur*t|ddgt|dg|S)N page_size_querypageSize page_token pageTokenrrrrr_ListFilesConfig_to_mldeves r2cCs*i}t|dgdurtt|dg||S)Nconfig)rr2rrrr_ListFilesParameters_to_mldevzsr4cCsi}t|dgdurt|dgt|dgt|dgdur(t|dgt|dgt|dgdur@t|dgddt|dgD|S)Nrr nextPageTokennext_page_tokenfilescSsg|]}|qSrr).0itemrrr sz1_ListFilesResponse_from_mldev..rrrrr_ListFilesResponse_from_mldevs r;c@s(eZdZdddeejdejfddZdddejdeej dej fdd Z ddd e deej dejfd d Zddd e deejdejfd dZdddee eje ejfdeejdejfddZdddeejdeejfddZdddee ejejejfdeejdefddZ dS)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) 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.r%r7r.?r3Ngetresponsekwargsheaders)r _ListFilesParameters _api_clientvertexai ValueErrorr4r@ format_maprpopr3 http_optionsrconvert_to_dictencode_unserializable_typesrequestbodyjsonloadsr;ListFilesResponse_from_response model_dump HttpResponserEr_verify_response selfr3parameter_model request_dictrequest_url_dictpath query_paramsrLrB response_dict return_valuerrr_listsD         z Files._listrc Cs2tj||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 Nrr3r>r%zupload/v1beta/filesr.r?r3postshould_return_http_response)rrA)r _CreateFileParametersrGrHrIrr@rJrrKr3rLrrMrNrOgetattrCreateFileResponserWrPrQrRr rTrU rYrr3rZr[r\r]r^rLrBr`r_rrr_createsP            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$r3r>r% files/{file}r.r?r3Nr@rA)r _GetFileParametersrGrHrIr,r@rJrrKr3rLrrMrNrOrPrQrRFilerTrUrW rYr$r3rZr[r\r]r^rLrBr_r`rrrr@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/...') rkr>r%rlr.r?r3NdeleterArD)r _DeleteFileParametersrGrHrIr)r@rJrrKr3rLrrMrNrOrPrQrRr+DeleteFileResponserTrUrVrErrWrorrrrp_sJ         z Files.deletec 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`. r> mime_typer$ display_nameNfiles/user_http_optionsuser_mime_typeTrLrercx-goog-upload-urlPFailed to create file. Upload URL did not returned from the create file request.rLrrAr)rGrHrIr UploadFileConfig isinstancedictrnrtr$ru startswithr prepare_resumable_uploadrL size_bytesrjCreateFileConfigrrEKeyErrorioIOBase upload_fileosfspathrTrQrU rYrr3 config_modelfile_objrLrrtrB upload_url return_filefs_pathrrruploadsf           z Files.uploadcCstd|j|j|d|SNr7r=)rrarYr3rrrlists  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.rv :downloadaltmediar?rLr}r)rGrHrIrrr DownloadFileConfigrn download_urir'r(rr download_fileVideo video_bytesGeneratedVideovideo) rYrr3rr$r]r^rLdatarrrdownloads<)     zFiles.download)!__name__ __module__ __qualname__rr ListFilesConfigOrDictrSra FileOrDictCreateFileConfigOrDictrhrjstrGetFileConfigOrDictrnr@DeleteFileConfigOrDictrrrprrPathLikerrUploadFileConfigOrDictrrrrrDownloadFileConfigOrDictbytesrrrrrr<sj I @ B K M r<c@s eZdZdddeejdejfddZdddejdeej dej fdd Z ddd e deej dejfd d Zddd e deejdejfd dZdddee eje ejfdeejdejfddZdddeejdeejfddZdddee ejfdeejdefddZdS) 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| | 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>r%r7r.r?r3Nr@rArD)r rFrGrHrIr4r@rJrrKr3rLrrMrN async_requestrPrQrRr;rSrTrUrVrErrWrXrrrraPsJ          zAsyncFiles._listrc s: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| | Srb)r rfrGrHrIrr@rJrrKr3rLrrMrNrrgrhrWrPrQrRr rTrUrirrrrjsR            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) rkr>r%rlr.r?r3Nr@rA)r rmrGrHrIr,r@rJrrKr3rLrrMrNrrPrQrRrnrTrUrWrorrrr@sB          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/...') rkr>r%rlr.r?r3NrprArD)r rqrGrHrIr)r@rJrrKr3rLrrMrNrrPrQrRr+rrrTrUrVrErrWrorrrrpsL          zAsyncFiles.deletec 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`. r>rsNrvrwTrzrcr{zX-Goog-Upload-URLr|r}rrAr)rGrHrIr r~rrrnrtr$rurr rrLrrjrrrErrrasync_upload_filerrrTrQrUrrrrr`sn              zAsyncFiles.uploadcs td|j|j|dIdH|Sr)rrarrrrrszAsyncFiles.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>NrvrrLrrr?r}r) rGrHrIrrr rr'r(rrasync_download_file) rYrr3rr$r]rLr^rrrrrs.#     zAsyncFiles.download)rrrrr rrSrarrrhrjrrrnr@rrrrprrrrrrrrrrrrrrrrrNsj K @ D K R  r)N)(rrQloggingrtypingrrr urllib.parserrrr r r'r r rr rpagersrr getLoggerloggerrrobjectrr r)r+r,r2r4r; BaseModuler<rrrrrs                           9