o pkj/ @sddlmZddlZddlZddlZddlZddlmZmZm Z ddl m Z ddl m Z ddl mZddl mZdd l mZdd l mZdd lmZdd lmZdd lmZddlmZddlmZmZddlmZm Z e!dZ" d8dede e#e$efe%fdee#e$efde#e$effddZ& d8dede e#e$efe%fdee#e$efde#e$effddZ' d8de e#e$efe%fdee#e$efde#e$effddZ( d8de e#e$efe%fdee#e$efde#e$effddZ) d8de e#e$efe%fdee#e$efde#e$effddZ* d8de e#e$efe%fdee#e$efde#e$effd d!Z+ d8de e#e$efe%fdee#e$efde#e$effd"d#Z, d8de e#e$efe%fdee#e$efde#e$effd$d%Z- d8de e#e$efe%fdee#e$efde#e$effd&d'Z. d8de e#e$efe%fdee#e$efde#e$effd(d)Z/ d8de e#e$efe%fdee#e$efde#e$effd*d+Z0 d8de e#e$efe%fdee#e$efde#e$effd,d-Z1 d8de e#e$efe%fdee#e$efde#e$effd.d/Z2 d8de e#e$efe%fdee#e$efde#e$effd0d1Z3 d8de e#e$efe%fdee#e$efde#e$effd2d3Z4Gd4d5d5e j5Z6Gd6d7d7e j5Z7dS)9)cached_propertyN)AnyOptionalUnion) urlencode) _api_module)_common) _extra_utils) _transformers)types) BaseApiClient)get_value_by_path)set_value_by_path),_UploadToFileSearchStoreOperation_from_mldev)AsyncDocuments Documents) AsyncPagerPagerzgoogle_genai.filesearchstores api_client from_object parent_objectreturnc Cs\i}t|dgdurt|dgt|dgt|dgdur,t|dgt|t|dg|S)N display_name displayNameembedding_modelembeddingModel)getvsetvtt_modelrrr to_objectr#]/home/djax/ivt_ai_plugin/venv/lib/python3.10/site-packages/google/genai/file_search_stores.py%_CreateFileSearchStoreConfig_to_mldev)sr%cCs,i}t|dgdurt|t|dg||SNconfig)rr%r!r#r#r$)_CreateFileSearchStoreParameters_to_mldev=s r(cCs0i}t|dgdurt|ddgt|dg|S)Nforce_queryrrrrr"r#r#r$%_DeleteFileSearchStoreConfig_to_mldevKsr-cCRi}t|dgdurt|ddgt|dgt|dgdur'tt|dg||S)Nname_urlr')rrr-r,r#r#r$)_DeleteFileSearchStoreParameters_to_mldevWs r1cCs0i}t|dgdurt|ddgt|dg|S)Nr/r0r+r,r#r#r$&_GetFileSearchStoreParameters_to_mldevgsr2cCs^i}t|dgdurt|dgddt|dgDt|dgdur-t|dgt|dg|S)Ncustom_metadatacustomMetadatacSg|]}|qSr#r#.0itemr#r#r$ |z._ImportFileConfig_to_mldev..chunking_configchunkingConfigr+r,r#r#r$_ImportFileConfig_to_mldevrs r=cCsi}t|dgdurt|dgt|dgt|dgdur(t|dgt|dgt|dgdur;t|dgt|dgt|dgdurNt|dgt|dgt|dgdurdt|dgtt|dg||S)Nr/metadatadoneerrorresponse)rr_ImportFileResponse_from_mldevr,r#r#r$_ImportFileOperation_from_mldevs$ rCcCsxi}t|dgdurt|ddgt|dgt|dgdur)t|dgt|dgt|dgdur:tt|dg||S)Nfile_search_store_namer0 file_namefileNamer')rrr=r,r#r#r$_ImportFileParameters_to_mldevs rGcCszi}t|dgdurt|dgt|dgt|dgdur(t|dgt|dgt|dgdur;t|dgt|dg|S)NsdkHttpResponsesdk_http_responseparent documentName document_namer+r,r#r#r$rBsrBcCsXi}t|dgdurt|ddgt|dgt|dgdur*t|ddgt|dg|S)N page_sizer*pageSize page_token pageTokenr+r,r#r#r$$_ListFileSearchStoresConfig_to_mldevs rQcCs*i}t|dgdurtt|dg||Sr&)rrQr,r#r#r$(_ListFileSearchStoresParameters_to_mldevs  rRcCsi}t|dgdurt|dgt|dgt|dgdur(t|dgt|dgt|dgdur@t|dgddt|dgD|S) NrHrI nextPageTokennext_page_tokenfileSearchStoresfile_search_storescSr5r#r#r6r#r#r$r9r:z<_ListFileSearchStoresResponse_from_mldev..r+r,r#r#r$(_ListFileSearchStoresResponse_from_mldevsrWcCsi}t|dgdurt|dgt|dgt|dgdur(t|dgt|dgt|dgdur@t|dgddt|dgDt|d gdurSt|d gt|d g|S) N mime_typemimeTyperrr3r4cSr5r#r#r6r#r#r$r9r:z;_UploadToFileSearchStoreConfig_to_mldev..r;r<r+r,r#r#r$'_UploadToFileSearchStoreConfig_to_mldevs$ rZcCr.)NrDr0r')rrrZr,r#r#r$+_UploadToFileSearchStoreParameters_to_mldev$s  r[cCs.i}t|dgdurt|dgt|dg|S)NrHrIr+r,r#r#r$4_UploadToFileSearchStoreResumableResponse_from_mldev8s r\c @peZdZedefddZdddeejdej fddZ ddd e deej dej fd d Z ddd e deejddfd d ZdddeejdejfddZddde deejdejfddZddde de deejdejfddZddde dee eje ejfdeejdejfddZddde deej de!fddZ"dddeejde#ej fddZ$dS)FileSearchStoresrcC t|jSN)r _api_clientselfr#r#r$ documentsG zFileSearchStores.documentsNr'r'c Cs*tj|d}|jjrtdt|j|}|d}|r!d|}nd}|d}|r3|dt|}| ddd}|j durJ|j j durJ|j j }t |}t |}|jd |||}|jsbint|j} tjj| t|ddrdt|j d dt|j d dt|j d dd inid} |j| | SzCreates a File Search Store. Args: config (CreateFileSearchStoreConfig | None): Optional parameters for the request. Returns: FileSearchStore rfiThis method is only supported in Gemini Developer API mode, not in Gemini Enterprise Agent Platform mode.r0rUr*?r'Npostresponse_schemaresponse_json_schemainclude_all_fieldsrkrlrmrAkwargs)r _CreateFileSearchStoreParametersravertexai ValueErrorr(get format_maprpopr' http_optionsr convert_to_dictencode_unserializable_typesrequestbodyjsonloadsFileSearchStore_from_responsegetattr_verify_response rcr'parameter_model request_dictrequest_url_dictpath query_paramsrwrA response_dict return_valuer#r#r$createKs`           zFileSearchStores.creater/c Cs(tj||d}|jjrtdt|}|d}|r d|}nd}|d}|r2|dt|}| ddd}|j durI|j j durI|j j }t |}t |}|jd |||} | jsaint| j} tjj| t|ddrdt|j d dt|j d dt|j d dd inid} |j| | SaRGets 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'rhr0{name}r*rir'Nrtrkrlrmrnro)r _GetFileSearchStoreParametersrarrrsr2rtrurrvr'rwr rxryrzr{r|r}r~rrr rcr/r'rrrrrrwrArrr#r#r$rtsZ          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 rrhr0rr*rir'Ndelete)r _DeleteFileSearchStoreParametersrarrrsr1rtrurrvr'rwr rxryrz rcr/r'rrrrrrwr#r#r$rs0        zFileSearchStores.deletec Cs6tj|d}|jjrtdt|}|d}|rd|}nd}|d}|r1|dt|}| ddd}|j durH|j j durH|j j }t |}t |}|jd|||}|js`int|j} |jjsnt| } tjj| t|ddrdt|j d dt|j d dt|j d dd inid } |j| | SNrfrhr0rUr*rir'rtrkrlrmrnro)r _ListFileSearchStoresParametersrarrrsrRrtrurrvr'rwr rxryrzr{r|r}rWListFileSearchStoresResponserrrrr#r#r$_lists\          zFileSearchStores._listrDc Cshtj||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| t|ddrdt|j d dt|j d dt|j d ddinid} |j| | SNrDr'rhr0z>upload/v1beta/{file_search_store_name}:uploadToFileSearchStorer*rir'rjshould_return_http_response)rIrkrlrmrnro)r "_UploadToFileSearchStoreParametersrarrrsr[rtrurrvr'rwr rxryrzr(UploadToFileSearchStoreResumableResponserr{r|r}r\r rcrDr'rrrrrrwrArrr#r#r$_upload_to_file_search_store`s           z-FileSearchStores._upload_to_file_search_storerEc Cs:tj|||d}|jjrtdt|}|d}|r!d|}nd}|d}|r3|dt|}| ddd} |j durJ|j j durJ|j j } t |}t |}|jd ||| } | jsbint| j} |jjspt| } tjj| t|ddrdt|j d dt|j d dt|j d dd inid} |j| | SaImports 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. )rDrEr'rhr0z#{file_search_store_name}:importFiler*rir'Nrjrkrlrmrnro)r _ImportFileParametersrarrrsrGrtrurrvr'rwr rxryrzr{r|r}rCImportFileOperationrrr rcrDrEr'rrrrrrwrArrr#r#r$ import_filesh         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)a~Calls 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 `display_name`, `mime_type`, and others. =This method is only supported in the Gemini Developer client.NT exclude_unsetrwrXuser_http_optionsuser_mime_typerwrrx-goog-upload-urlX-Goog-Upload-URL`Failed to upload file to file search store. Upload URL was not returned from the upload request.rwror#)rarrrs isinstancedict model_dumpr prepare_resumable_uploadrtrr UploadToFileSearchStoreConfigrIheadersKeyErrorioIOBase upload_fileosfspathrr| UploadToFileSearchStoreOperationrrcrDrr'user_config_dictrw size_bytes_final_config_dictrA upload_urlupload_responsefs_pathrr#r#r$upload_to_file_search_stores`            z,FileSearchStores.upload_to_file_search_storemedia_idcCs|jjrtd|d}d|vrtd|d|d}d}|r2t|tr0tjd i|}n|}d}|rDt|dgdurDt|dg}|jj ||d }|S aDownloads media using a Media ID. The media_id has the format: fileSearchStores//media/ This is mapped to the DownloadMedia RPC which expects: GET /{name=fileSearchStores/*/media/*} Args: media_id: The Media ID from grounding metadata. config: Optional configuration for the download. Returns: bytes: The media data. r/z/media/zInvalid media_id format: z;. Expected format: fileSearchStores//media/z ?alt=mediaNrwrr#) rarrrslstriprrr DownloadMediaConfigr download_filercrr'clean_idr config_modelrwdatar#r#r$download_mediabs.     zFileSearchStores.download_mediacCs|j}td||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}") rVrf)rrrcr' list_requestr#r#r$lists zFileSearchStores.list)%__name__ __module__ __qualname__rrrdrr !CreateFileSearchStoreConfigOrDictr~rstrGetFileSearchStoreConfigOrDictrt!DeleteFileSearchStoreConfigOrDictr ListFileSearchStoresConfigOrDictrr#UploadToFileSearchStoreConfigOrDictrrImportFileConfigOrDictrrrrPathLikerrrrDownloadMediaConfigOrDictbytesrrrr#r#r#r$r^Es Q S 8 F _ ` Q 5r^c @r])AsyncFileSearchStoresrcCr_r`)rrarbr#r#r$rdrezAsyncFileSearchStores.documentsNrfr'c s2tj|d}|jjrtdt|j|}|d}|r"d|}nd}|d}|r4|dt|}| ddd}|j durK|j j durK|j j }t |}t |}|jd |||IdH}|jsfint|j} tjj| t|ddrdt|j d dt|j d dt|j d dd inid} |j| | Srg)r rqrarrrsr(rtrurrvr'rwr rxry async_requestr{r|r}r~rrrrr#r#r$rsb            zAsyncFileSearchStores.creater/c s0tj||d}|jjrtdt|}|d}|r!d|}nd}|d}|r3|dt|}| ddd}|j durJ|j j durJ|j j }t |}t |}|jd |||IdH} | jseint| j} tjj| t|ddrdt|j d dt|j d dt|j d dd inid} |j| | Sr)r rrarrrsr2rtrurrvr'rwr rxryrr{r|r}r~rrrrr#r#r$rts`           zAsyncFileSearchStores.getc stj||d}|jjrtdt|}|d}|r!d|}nd}|d}|r3|dt|}| ddd}|j durJ|j j durJ|j j }t |}t |}|jd |||IdHdSr)r rrarrrsr1rtrurrvr'rwr rxryrrr#r#r$rYs6        zAsyncFileSearchStores.deletec s>tj|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| t|ddrdt|j d dt|j d dt|j d dd inid } |j| | Sr)r rrarrrsrRrtrurrvr'rwr rxryrr{r|r}rWrrrrrr#r#r$rsb           zAsyncFileSearchStores._listrDc sptj||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| t|ddrdt|j d dt|j d dt|j d ddinid} |j| | Sr)r rrarrrsr[rtrurrvr'rwr rxryrrrrr{r|r}r\rrr#r#r$rs           z2AsyncFileSearchStores._upload_to_file_search_storerEc sBtj|||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| t|ddrdt|j d dt|j d dt|j d dd inid} |j| | Sr)r rrarrrsrGrtrurrvr'rwr rxryrr{r|r}rCrrrrrr#r#r$r0sj          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)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 `display_name`, `mime_type` and others. rNTrrwrXrrrrrrrror#)rarrrsrrrr rrtrr rrIrrrrasync_upload_filerrrr|rrrr#r#r$rsb             z1AsyncFileSearchStores.upload_to_file_search_storercs|jjr td|d}d|vrtd|d|d}d}|r3t|tr1tjd i|}n|}d}|rEt|dgdurEt|dg}|jj ||d IdH}|Sr) rarrrsrrrr rrasync_download_filerr#r#r$rs0      z$AsyncFileSearchStores.download_mediacs$|j}td||j|dIdH|S)a7Lists FileSearchStores asynchronously. Args: config (ListFileSearchStoresConfig): Optional parameters for the request, such as page_size. 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.aio.file_search_stores.list(): print(f"file search store: {file_search_store.name} - {file_search_store.display_name}") rVrfN)rrrr#r#r$r szAsyncFileSearchStores.list)%rrrrrrdrr rr~rrrrtrrrrrrrrrrrrrrrrrrrrrrrr#r#r#r$rs Q U : H _ ` P 5rr`)8 functoolsrrr|loggingrtypingrrr urllib.parserrr r r rr rar rrrr_operations_convertersrrdrrpagersrr getLoggerloggerrrobjectr%r(r-r1r2r=rCrGrBrQrRrWrZr[r\ BaseModuler^rr#r#r#r$s*                                            t