o Yiʡ @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 displayNamegetvsetvrrZ to_objectrY/home/kim/smarthome/.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 r6cCsi}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!)rrr6rrrr_ImportFileParameters_to_mldevs r@cCszi}t|dgdurt|dgt|dgt|dgdur(t|dgt|dgt|dgdur;t|dgt|dg|S)NsdkHttpResponsesdk_http_responseparentZ documentNameZ document_namerrrrrr;sr;cCsXi}t|dgdurt|ddgt|dgt|dgdur*t|ddgt|dg|S)NZ page_sizer%ZpageSizeZ page_tokenZ pageTokenrrrrr$_ListFileSearchStoresConfig_to_mldevs rDcCrr )rrDrrrr(_ListFileSearchStoresParameters_to_mldevr#rEcCsi}t|dgdurt|dgt|dgt|dgdur(t|dgt|dgt|dgdur@t|dgddt|dgD|S) NrArBZ nextPageTokenZnext_page_tokenfileSearchStoresfile_search_storescSr.rrr/rrrr2r3z<_ListFileSearchStoresResponse_from_mldev..rrrrr(_ListFileSearchStoresResponse_from_mldevsrHcCsi}t|dgdurt|dgt|dgt|dgdur(t|dgt|dgt|dgdur@t|dgddt|dgDt|d gdurSt|d gt|d g|S) N mime_typeZmimeTyperrr,r-cSr.rrr/rrrr2 r3z;_UploadToFileSearchStoreConfig_to_mldev..r4r5rrrrr'_UploadToFileSearchStoreConfig_to_mldevs$ rJcCr')Nr=r)r!)rrrJrrrr+_UploadToFileSearchStoreParameters_to_mldevs  rKcCs.i}t|dgdurt|dgt|dg|S)NrArBrrrrr4_UploadToFileSearchStoreResumableResponse_from_mldev.s rLc @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 rV=This method is only supported in the Gemini Developer client.r)rFr%?r!Npostr:kwargs)r _CreateFileSearchStoreParametersrQvertexai ValueErrorr"get format_maprpopr! http_optionsr convert_to_dictencode_unserializable_typesrequestbodyjsonloadsFileSearchStore_from_response model_dump_verify_response rSr!parameter_model request_dictrequest_url_dictpath query_paramsrcr: 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!rXr){name}r%rYr!Nr`r[)r _GetFileSearchStoreParametersrQr^r_r+r`rarrbr!rcr rdrerfrgrhrirjrkrlrm rSr(r!rorprqrrrsrcr:rtrurrrr`|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 rxrXr)ryr%rYr!Ndelete)r _DeleteFileSearchStoreParametersrQr^r_r*r`rarrbr!rcr rdrerf rSr(r!rorprqrrrsrcr:rrrr}s4        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 NrVrXr)rFr%rYr!r`r[)r _ListFileSearchStoresParametersrQr^r_rEr`rarrbr!rcr rdrerfrgrhrirHListFileSearchStoresResponserkrlrmrnrrr_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!rXr)z>upload/v1beta/{file_search_store_name}:uploadToFileSearchStorer%rYr!rZshould_return_http_response)rBr[)r "_UploadToFileSearchStoreParametersrQr^r_rKr`rarrbr!rcr rdrerfgetattr(UploadToFileSearchStoreResumableResponsermrgrhrirLrkrl rSr=r!rorprqrrrsrcr:rurtrrr_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!rXr)z#{file_search_store_name}:importFiler%rYr!NrZr[)r _ImportFileParametersrQr^r_r@r`rarrbr!rcr rdrerfrgrhrir<ImportFileOperationrkrlrm rSr=r>r!rorprqrrrsrcr:rtrurrr import_fileosJ        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. rXNTZ exclude_unsetrcrIZuser_http_optionsZuser_mime_typercrrx-goog-upload-urlX-Goog-Upload-URL`Failed to upload file to file search store. Upload URL did not returned from the upload request.rcr[r)rQr^r_ isinstancedictrlr prepare_resumable_uploadr`rr UploadToFileSearchStoreConfigrBheadersKeyErrorioIOBaseZ upload_fileosfspathrrh UploadToFileSearchStoreOperationrkrSr=rr!Zuser_config_dictrcZ size_bytes_Zfinal_config_dictr:Z upload_urlZupload_responseZfs_pathrtrrrupload_to_file_search_stores`            z,FileSearchStores.upload_to_file_search_storecCs|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}") rGrV)rrrSr!Z list_requestrrrlists zFileSearchStores.list)"__name__ __module__ __qualname__rrrTrr !CreateFileSearchStoreConfigOrDictrjrvstrGetFileSearchStoreConfigOrDictr`!DeleteFileSearchStoreConfigOrDictr} ListFileSearchStoresConfigOrDictrr#UploadToFileSearchStoreConfigOrDictrrImportFileConfigOrDictrrrrPathLikerrrrrrrrrrrN;s ? C 9 6 O P MrNc @rM)AsyncFileSearchStoresrcCrOrP)rrQrRrrrrT'rUzAsyncFileSearchStores.documentsNrVr!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| | SrW)r r]rQr^r_r"r`rarrbr!rcr rdre async_requestrgrhrirjrkrlrmrnrrrrv+s@           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| | Srw)r rzrQr^r_r+r`rarrbr!rcr rdrerrgrhrirjrkrlrmr{rrrr`fsB          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 r~rQr^r_r*r`rarrbr!rcr rdrerrrrrr}s6        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 rrQr^r_rEr`rarrbr!rcr rdrerrgrhrirHrrkrlrmrnrrrrsD          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 rrQr^r_rKr`rarrbr!rcr rdrerrrrmrgrhrirLrkrlrrrrrsd          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 rrQr^r_r@r`rarrbr!rcr rdrerrgrhrir<rrkrlrmrrrrr]sL         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. rXNTrrcrIrrrrrrrr[r)rQr^r_rrrlr rr`rr rrBrrrrZasync_upload_filerrrrhrrkrrrrrsb             z1AsyncFileSearchStores.upload_to_file_search_storecs$|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}") rGrVN)rrrrrrrszAsyncFileSearchStores.list)"rrrrrrTrr rrjrvrrr`rr}rrrrrrrrrrrrrrrrrrrrrrr%s ? E 9 8 O P MrrP)4 functoolsrrrhloggingrtypingrrr urllib.parserrr r r r rr rZ_operations_convertersrrTrrZpagersrr getLoggerloggerrrobjectrr"r&r*r+r6r<r@r;rDrErHrJrKrLZ BaseModulerNrrrrrs                                          m