o k Di| @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_objectrR/home/air/goalskill_t/back/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 ejfdeejdejfddZdddee ejejejfdeejdefddZdddeejdeejfddZ 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 Nr==This method is only supported in the Gemini Developer client.r%r7r.?r3getresponsekwargsheaders)r _ListFilesParameters _api_clientvertexai ValueErrorr4rA format_maprpopr3 http_optionsrconvert_to_dictencode_unserializable_typesrequestbodyjsonloadsr;ListFilesResponse_from_response model_dump HttpResponserFr_verify_response selfr3parameter_model request_dictrequest_url_dictpath query_paramsrMrC 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)rrB)r _CreateFileParametersrHrIrJrrArKrrLr3rMrrNrOrPgetattrCreateFileResponserXrQrRrSr rUrV rZrr3r[r\r]r^r_rMrCrar`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@r3NrArB)r _GetFileParametersrHrIrJr,rArKrrLr3rMrrNrOrPrQrRrSFilerUrVrX rZr$r3r[r\r]r^r_rMrCr`rarrrrA 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/...') rlr?r%rmr.r@r3NdeleterBrE)r _DeleteFileParametersrHrIrJr)rArKrrLr3rMrrNrOrPrQrRrSr+DeleteFileResponserUrVrWrFrrXrprrrrqNsJ         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_typeTrMrfrdx-goog-upload-urlPFailed to create file. Upload URL did not returned from the create file request.rMrrBr)rHrIrJr UploadFileConfig isinstancedictrorur$rv startswithr prepare_resumable_uploadrM size_bytesrkCreateFileConfigrrFKeyErrorioIOBase upload_fileosfspathrUrRrV rZrr3 config_modelfile_objrMrrurC 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 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.rw :downloadaltmediar@rMr~r)rHrIrJrrr DownloadFileConfigro download_urir'r(rr download_fileVideo video_bytesGeneratedVideovideo) rZrr3rr$r^r_rMdatarrrdownloads<)     zFiles.downloadcCs|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) r7r=)rbrrZr3 list_requestrrrlist2s z Files.list)!__name__ __module__ __qualname__rr ListFilesConfigOrDictrTrb FileOrDictCreateFileConfigOrDictrirkstrGetFileConfigOrDictrorADeleteFileConfigOrDictrsrqrrPathLikerrUploadFileConfigOrDictrrrDownloadFileConfigOrDictbytesrrrrrrrr<sj 8 @ B K P Rr<c@s eZdZdddeejdejfddZdddejdeej dej fdd Z ddd e deej dejfd d Zddd e deejdejfd dZdddee eje ejfdeejdejfddZdddee ejfdeejdefddZdddeejdeejfddZdS) 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 rGrHrIrJr4rArKrrLr3rMrrNrO async_requestrQrRrSr;rTrUrVrWrFrrXrYrrrrbQsJ          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| | Src)r rgrHrIrJrrArKrrLr3rMrrNrOrrhrirXrQrRrSr rUrVrjrrrrksR            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) rlr?r%rmr.r@r3NrArB)r rnrHrIrJr,rArKrrLr3rMrrNrOrrQrRrSrorUrVrXrprrrrAsB          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/...') rlr?r%rmr.r@r3NrqrBrE)r rrrHrIrJr)rArKrrLr3rMrrNrOrrQrRrSr+rsrUrVrWrFrrXrprrrrq sL          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?rtNrwrxTr{rdr|zX-Goog-Upload-URLr}r~rrBr)rHrIrJr rrrrorur$rvrr rrMrrkrrrFrrrasync_upload_filerrrUrRrVrrrrrPsn              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) r?NrwrrMrrr@r~r) rHrIrJrrr rr'r(rrasync_download_file) rZrr3rr$r^rMr_rrrrrs.#     zAsyncFiles.downloadcs$|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) r7r=N)rbrrrrrrszAsyncFiles.list)rrrrr rrTrbrrrirkrrrorArrsrqrrrrrrrrrrrrrrrrrOsj : @ D K U  Cr)N)(rrRloggingrtypingrrr urllib.parserrrr r r'r r rr rpagersrr getLoggerloggerrrobjectrr r)r+r,r2r4r; BaseModuler<rrrrrs                           :