o i @sddlmZddlZddlZddlZddlZddlmZmZm Z ddl m Z ddl m Z ddl mZddl mZdd l mZdd lmZdd lmZdd lmZdd lmZmZddlmZmZedZ d5de ee efe!fdeee efdee effddZ" d5de ee efe!fdeee efdee effddZ# d5de ee efe!fdeee efdee effddZ$ d5de ee efe!fdeee efdee effddZ% d5de ee efe!fdeee efdee effddZ& d5de ee efe!fdeee efdee effddZ' d5de ee efe!fdeee efdee effdd Z( d5de ee efe!fdeee efdee effd!d"Z) d5de ee efe!fdeee efdee effd#d$Z* d5de ee efe!fdeee efdee effd%d&Z+ d5de ee efe!fdeee efdee effd'd(Z, d5de ee efe!fdeee efdee effd)d*Z- d5de ee efe!fdeee efdee effd+d,Z. d5de ee efe!fdeee efdee effd-d.Z/ d5de ee efe!fdeee efdee effd/d0Z0Gd1d2d2e j1Z2Gd3d4d4e j1Z3dS)6)cached_propertyN)AnyOptionalUnion) urlencode) _api_module)_common) _extra_utils)types)get_value_by_path)set_value_by_path),_UploadToFileSearchStoreOperation_from_mldev)AsyncDocuments Documents) AsyncPagerPagerzgoogle_genai.filesearchstores from_object parent_objectreturncCs.i}t|dgdurt|dgt|dg|S)N display_name displayNamegetvsetvrr to_objectre/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/google/genai/file_search_stores.py%_CreateFileSearchStoreConfig_to_mldev(srcC*i}t|dgdurtt|dg||SNconfig)rrrrrr)_CreateFileSearchStoreParameters_to_mldev4  r#cCs0i}t|dgdurt|ddgt|dg|S)Nforce_queryrrrrr%_DeleteFileSearchStoreConfig_to_mldevAsr'cCRi}t|dgdurt|ddgt|dgt|dgdur'tt|dg||S)Nname_urlr")rrr'rrrr)_DeleteFileSearchStoreParameters_to_mldevMs r+cCs0i}t|dgdurt|ddgt|dg|S)Nr)r*rrrrr&_GetFileSearchStoreParameters_to_mldev]sr,cCs^i}t|dgdurt|dgddt|dgDt|dgdur-t|dgt|dg|S)Ncustom_metadatacustomMetadatacSg|]}|qSrr.0itemrrr rz._ImportFileConfig_to_mldev..chunking_configchunkingConfigrrrrr_ImportFileConfig_to_mldevhs r7cCsi}t|dgdurt|dgt|dgt|dgdur(t|dgt|dgt|dgdur;t|dgt|dgt|dgdurNt|dgt|dgt|dgdurdt|dgtt|dg||S)Nr)metadatadoneerrorresponse)rr_ImportFileResponse_from_mldevrrrr_ImportFileOperation_from_mldevs$ r=cCsxi}t|dgdurt|ddgt|dgt|dgdur)t|dgt|dgt|dgdur:tt|dg||S)Nfile_search_store_namer* file_namefileNamer")rrr7rrrr_ImportFileParameters_to_mldevs rAcCszi}t|dgdurt|dgt|dgt|dgdur(t|dgt|dgt|dgdur;t|dgt|dg|S)NsdkHttpResponsesdk_http_responseparent documentName document_namerrrrrr<sr<cCsXi}t|dgdurt|ddgt|dgt|dgdur*t|ddgt|dg|S)N page_sizer&pageSize page_token pageTokenrrrrr$_ListFileSearchStoresConfig_to_mldevs rKcCr r!)rrKrrrr(_ListFileSearchStoresParameters_to_mldevr$rLcCsi}t|dgdurt|dgt|dgt|dgdur(t|dgt|dgt|dgdur@t|dgddt|dgD|S) NrBrC nextPageTokennext_page_tokenfileSearchStoresfile_search_storescSr/rrr0rrrr3r4z<_ListFileSearchStoresResponse_from_mldev..rrrrr(_ListFileSearchStoresResponse_from_mldevsrQcCsi}t|dgdurt|dgt|dgt|dgdur(t|dgt|dgt|dgdur@t|dgddt|dgDt|d gdurSt|d gt|d g|S) N mime_typemimeTyperrr-r.cSr/rrr0rrrr3 r4z;_UploadToFileSearchStoreConfig_to_mldev..r5r6rrrrr'_UploadToFileSearchStoreConfig_to_mldevs$ rTcCr()Nr>r*r")rrrTrrrr+_UploadToFileSearchStoreParameters_to_mldevs  rUcCs.i}t|dgdurt|dgt|dg|S)NrBrCrrrrr4_UploadToFileSearchStoreResumableResponse_from_mldev.s rVc @NeZdZedefddZdddeejdej fddZ ddd e deej dej fd d Z ddd e deejddfd d ZdddeejdejfddZddde deejdejfddZddde de deejdejfddZddde dee eje ejfdeejdejfddZdddeejde ej fddZ!dS)FileSearchStoresrcC t|jSN)r _api_clientselfrrr documents= zFileSearchStores.documentsNr"r"c Cstj|d}|jjrtdt|}|d}|rd|}nd}|d}|r1|dt|}| ddd}|j durH|j j durH|j j }t |}t |}|jd |||}|js`int|j} tjj| |d } |j| | S zCreates a File Search Store. Args: config (CreateFileSearchStoreConfig | None): Optional parameters for the request. Returns: FileSearchStore r`=This method is only supported in the Gemini Developer client.r*rOr&?r"Npostr;kwargs)r _CreateFileSearchStoreParametersr[vertexai ValueErrorr#get format_maprpopr" http_optionsr convert_to_dictencode_unserializable_typesrequestbodyjsonloadsFileSearchStore_from_response model_dump_verify_response r]r"parameter_model request_dictrequest_url_dictpath query_paramsrmr; response_dict return_valuerrrcreateAs>          zFileSearchStores.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 aRGets metadata about a FileSearchStore. Args: name (str): The resource name of the FileSearchStore. Example: `FileSearchStores/my-file-search-store-123` config (GetFileSearchStoreConfig | None): Optional parameters for the request. Returns: A FileSearchStore object containing the metadata. r)r"rbr*{name}r&rcr"Nrjre)r _GetFileSearchStoreParametersr[rhrir,rjrkrrlr"rmr rnrorprqrrrsrtrurvrw r]r)r"ryrzr{r|r}rmr;r~rrrrrj|s<         zFileSearchStores.getc Cstj||d}|jjrtdt|}|d}|r d|}nd}|d}|r2|dt|}| ddd}|j durI|j j durI|j j }t |}t |}|jd |||} dS aDeletes a FileSearchStore. Args: name (str): The resource name of the FileSearchStore. Example: `FileSearchStores/my-file-search-store-123` config (DeleteFileSearchStoreConfig | None): Optional parameters for the request. Returns: None rrbr*rr&rcr"Ndelete)r _DeleteFileSearchStoreParametersr[rhrir+rjrkrrlr"rmr rnrorp r]r)r"ryrzr{r|r}rmr;rrrrs4        zFileSearchStores.deletec 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 } |j| | S aLists all FileSearchStore owned by the user. Args: config (ListFileSearchStoresConfig | None): Optional parameters for the request, such as page_size. Returns: ListFileSearchStoresResponse: A paginated list of FileSearchStore. r`rbr*rOr&rcr"Nrjre)r _ListFileSearchStoresParametersr[rhrirLrjrkrrlr"rmr rnrorprqrrrsrQListFileSearchStoresResponserurvrwrxrrr_lists>          zFileSearchStores._listr>c 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 Nr>r"rbr*z>upload/v1beta/{file_search_store_name}:uploadToFileSearchStorer&rcr"rdshould_return_http_response)rCre)r "_UploadToFileSearchStoreParametersr[rhrirUrjrkrrlr"rmr rnrorpgetattr(UploadToFileSearchStoreResumableResponserwrqrrrsrVrurv r]r>r"ryrzr{r|r}rmr;rr~rrr_upload_to_file_search_store/sb          z-FileSearchStores._upload_to_file_search_storer?c Cstj|||d}|jjrtdt|}|d}|r!d|}nd}|d}|r3|dt|}| ddd} |j durJ|j j durJ|j j } t |}t |}|jd ||| } | jsbint| j} |jjspt| } tjj| |d } |j| | S aImports a File from File Service to a FileSearchStore. This is a long-running operation, see aip.dev/151 Args: file_search_store_name (str): The resource name of the FileSearchStore. Example: `fileSearchStores/my-file-search-store-123` file_name (str): The resource name of the File to import. Example: `files/abc-123` config (ImportFileConfig | None): Optional parameters for the request. Returns: ImportFileOperation. )r>r?r"rbr*z#{file_search_store_name}:importFiler&rcr"Nrdre)r _ImportFileParametersr[rhrirArjrkrrlr"rmr rnrorprqrrrsr=ImportFileOperationrurvrw r]r>r?r"ryrzr{r|r}rmr;r~rrrr import_fileysJ        zFileSearchStores.import_filefilecCsB|jjrtd|duri}nt|trt|}n|jdd}tj||d|dd\}}}i||dd}|j |t j di|d } | j dus]| j j dus]d | j j vrad | j j vratd d | j j vrn| j j d } n| j j d } t|tjr|jj|| ||d } nt|} |jj| | ||d } t| j} t jj| idS)auCalls the API to upload a file to the given file search store. Args: file_search_store_name: The resource name of the FileSearchStore. Example: `fileSearchStores/file-search-store-123` 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 others. rbNT exclude_unsetrmrRuser_http_optionsuser_mime_typermrrx-goog-upload-urlX-Goog-Upload-URL`Failed to upload file to file search store. Upload URL did not returned from the upload request.rmrer)r[rhri isinstancedictrvr prepare_resumable_uploadrjrr UploadToFileSearchStoreConfigrCheadersKeyErrorioIOBase upload_fileosfspathrrr UploadToFileSearchStoreOperationrur]r>rr"user_config_dictrm size_bytes_final_config_dictr; upload_urlupload_responsefs_pathr~rrrupload_to_file_search_stores`            z,FileSearchStores.upload_to_file_search_storecCstd|j|j|d|S)aLists FileSearchStores. Args: config (ListFileSearchStoresConfig): Optional configuration for the list request. Returns: A Pager object that contains one page of file search stores. When iterating over the pager, it automatically fetches the next page if there are more. Usage: .. code-block:: python for file_search_store in client.file_search_stores.list(): print(f"file search store: {file_search_store.name} - {file_search_store.display_name}") rPr`)rrr]r"rrrlists  zFileSearchStores.list)"__name__ __module__ __qualname__rrr^rr !CreateFileSearchStoreConfigOrDictrtrstrGetFileSearchStoreConfigOrDictrj!DeleteFileSearchStoreConfigOrDictr ListFileSearchStoresConfigOrDictrr#UploadToFileSearchStoreConfigOrDictrrImportFileConfigOrDictrrrrPathLikerrrrrrrrrrrX;s ? C 9 @ O P MrXc @rW)AsyncFileSearchStoresrcCrYrZ)rr[r\rrrr^/r_zAsyncFileSearchStores.documentsNr`r"c stj|d}|jjrtdt|}|d}|r d|}nd}|d}|r2|dt|}| ddd}|j durI|j j durI|j j }t |}t |}|jd |||IdH}|jsdint|j} tjj| |d } |j| | Sra)r rgr[rhrir#rjrkrrlr"rmr rnro async_requestrqrrrsrtrurvrwrxrrrr3s@           zAsyncFileSearchStores.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| | Sr)r rr[rhrir,rjrkrrlr"rmr rnrorrqrrrsrtrurvrwrrrrrjnsB          zAsyncFileSearchStores.getc stj||d}|jjrtdt|}|d}|r!d|}nd}|d}|r3|dt|}| ddd}|j durJ|j j durJ|j j }t |}t |}|jd |||IdH} dSr)r rr[rhrir+rjrkrrlr"rmr rnrorrrrrrs6        zAsyncFileSearchStores.deletec 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 } |j| | Sr)r rr[rhrirLrjrkrrlr"rmr rnrorrqrrrsrQrrurvrwrxrrrrsD           zAsyncFileSearchStores._listr>c 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| | Sr)r rr[rhrirUrjrkrrlr"rmr rnrorrrrwrqrrrsrVrurvrrrrr%sd          z2AsyncFileSearchStores._upload_to_file_search_storer?c s tj|||d}|jjrtdt|}|d}|r"d|}nd}|d}|r4|dt|}| ddd} |j durK|j j durK|j j } t |}t |}|jd ||| IdH} | jsfint| j} |jjstt| } tjj| |d } |j| | Sr)r rr[rhrirArjrkrrlr"rmr rnrorrqrrrsr=rrurvrwrrrrrosL         z!AsyncFileSearchStores.import_filercsV|jjr td|duri}nt|trt|}n|jdd}tj||d|dd\}}}i||dd}|j |t j di|d IdH} | j dusa| j j dusad | j j vred | j j vretd d | j j vrr| j j d } n| j j d } t|tjr|jj|| ||d IdH} nt|} |jj| | ||d IdH} t| j} t jj| idS)atCalls the API to upload a file to the given file search store. Args: file_search_store_name: The resource name of the FileSearchStore. Example: `fileSearchStores/file-search-store-123` 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 others. rbNTrrmrRrrrrrrrrer)r[rhrirrrvr rrjrr rrCrrrrasync_upload_filerrrrrrrurrrrrsb             z1AsyncFileSearchStores.upload_to_file_search_storecs td|j|j|dIdH|S)a(Lists FileSearchStores asynchronously. Args: config (ListFileSearchStoresConfig): Optional configuration for the list request. Returns: A Pager object that contains one page of FileSearchStores. When iterating over the pager, it automatically fetches the next page if there are more. Usage: .. code-block:: python async for file_search_store in await client.file_search_stores.list(): print(f"file search store: {file_search_store.name} - {file_search_store.display_name}") rPr`N)rrrrrrrszAsyncFileSearchStores.list)"rrrrrr^rr rrtrrrrjrrrrrrrrrrrrrrrrrrrrrrrrr-s ? E 9 B O P MrrZ)4 functoolsrrrrloggingrtypingrrr urllib.parserrr r r r rr r_operations_convertersrr^rrpagersrr getLoggerloggerrrobjectrr#r'r+r,r7r=rAr<rKrLrQrTrUrV BaseModulerXrrrrrs                                          u