쎒jV .ddlmZddlZddlZddlZddlZddlmZmZm Z ddl m Z ddl m Z ddl mZddl mZdd l mZdd l mZdd lmZdd lmZdd lmZddlmZddlmZmZddlmZm Z ej!dZ" d)dede e#e$efe%fdee#e$efde#e$effdZ& d)dede e#e$efe%fdee#e$efde#e$effdZ' d)de e#e$efe%fdee#e$efde#e$effdZ( d)de e#e$efe%fdee#e$efde#e$effdZ) d)de e#e$efe%fdee#e$efde#e$effdZ* d)de e#e$efe%fdee#e$efde#e$effdZ+ d)de e#e$efe%fdee#e$efde#e$effdZ, d)de e#e$efe%fdee#e$efde#e$effdZ- d)de e#e$efe%fdee#e$efde#e$effdZ. d)de e#e$efe%fdee#e$efde#e$effdZ/ d)de e#e$efe%fdee#e$efde#e$effd Z0 d)de e#e$efe%fdee#e$efde#e$effd!Z1 d)de e#e$efe%fdee#e$efde#e$effd"Z2 d)de e#e$efe%fdee#e$efde#e$effd#Z3 d)de e#e$efe%fdee#e$efde#e$effd$Z4Gd%d&e j5Z6Gd'd(e j5Z7dS)*)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 i}t|dg!t|dgt|dgt|dg4t|dgtj|t|dg|S)N display_name displayNameembedding_modelembeddingModel)getvsetvtt_modelrrr to_objects MC:\Projects\DSCons\.venv\Lib\site-packages\google\genai\file_search_stores.py%_CreateFileSearchStoreConfig_to_mldevr&)s !) +'((4kN;K)L)LMMM +)*++7  *d;1B0CDDEE cli}t|dg t|t|dg||SNconfig)rr&r#s r%)_CreateFileSearchStoreParameters_to_mldevr+=sH !) +z"".)Dxj119 r'cpi}t|dg"t|ddgt|dg|S)Nforce_queryrr rrr$s r%%_DeleteFileSearchStoreConfig_to_mldevr1KsD!) +y!!-7+T+y-I-IJJJ r'ci}t|dg"t|ddgt|dgt|dgtt|dg||S)Nname_urlr*)rr r1r0s r%)_DeleteFileSearchStoreParameters_to_mldevr5Ws{!) +x  ,VV$d;&A&ABBB +z"".) [8*%%y r'cpi}t|dg"t|ddgt|dg|S)Nr3r4r/r0s r%&_GetFileSearchStoreParameters_to_mldevr7gsD!) +x  ,VV$d;&A&ABBB r'c i}t|dg+t|dgdt|dgDt|dg!t|dgt|dg|S)Ncustom_metadatacustomMetadatacg|]}|Sr<.0items r% z._ImportFileConfig_to_mldev..|AAA$AAAr'chunking_configchunkingConfigr/r0s r%_ImportFileConfig_to_mldevrDrs!) +)*++7 AA${->,?@@AAA  +)*++7  [,-.. r'c i}t|dg!t|dgt|dgt|dg!t|dgt|dgt|dg!t|dgt|dgt|dg!t|dgt|dgt|dg/t|dgtt|dg||S)Nr3metadatadoneerrorresponse)rr _ImportFileResponse_from_mldevr0s r%_ImportFileOperation_from_mldevrKs#!) +x  ,VHd;99::: + |$$0ZL${ZL"A"ABBB +x  ,VHd;99::: +y!!-WItK';;<<< + |$$0  & zl + +Y   r'c4i}t|dg"t|ddgt|dgt|dg!t|dgt|dgt|dgtt|dg||S)Nfile_search_store_namer4 file_namefileNamer*)rr rDr0s r%_ImportFileParameters_to_mldevrPs!) +0122> )* [3455  + }%%1ZL${[M"B"BCCC +z"".tK(<>> +'((4_%tK.9I'J'JKKK r'ci}t|dg"t|ddgt|dgt|dg"t|ddgt|dg|S)N page_sizer.pageSize page_token pageTokenr/r0s r%$_ListFileSearchStoresConfig_to_mldevr\s!) + }%%1*-tK+/O/O + ~&&2 ; [<.)) r'cji}t|dgtt|dg||Sr))rr\r0s r%(_ListFileSearchStoresParameters_to_mldevr^sF!) +z"".( [8*%%y r'c Ji}t|dg!t|dgt|dgt|dg!t|dgt|dgt|dg+t|dgdt|dgD|S)NrRrS nextPageTokennext_page_tokenfileSearchStoresfile_search_storescg|]}|Sr<r<r=s r%r@z<_ListFileSearchStoresResponse_from_mldev..sBBB$BBBr'r/r0s r%(_ListFileSearchStoresResponse_from_mldevres!) +)*++7'(${=N.rAr'rBrCr/r0s r%'_UploadToFileSearchStoreConfig_to_mldevrjs!) + }%%1 d; &F&FGGG +'((4kN;K)L)LMMM +)*++7 AA${->,?@@AAA  +)*++7  [,-.. r'ci}t|dg"t|ddgt|dgt|dgtt|dg||S)NrMr4r*)rr rjr0s r%+_UploadToFileSearchStoreParameters_to_mldevrl$s!) +0122> )* [3455  +z"".+ [8*%%y r'cni}t|dg!t|dgt|dg|S)NrRrSr/r0s r%4_UploadToFileSearchStoreResumableResponse_from_mldevrn8sP!) +)*++7'(${=N>!##H--L1 0 0y.. 0 0dXt$$$04L*  " / ;$+8l*<88L6|DDL''lLH'mJBBHM1J1JM(77 ?Hd 3 3 #*#*,=t$$)0#*,BD))'.#*,@$''     !8L& %%l333 r'r3c tj||}|jjrt dt |}|d}|rd|}nd}|d}|r|dt|}| ddd}|j |j j |j j }tj |}tj|}|jd |||} | jsint#j| j} tj| t+|ddrDdt+|j d dt+|j d dt+|j d dd ini} |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. r3r*rzr4{name}r.r{r*Nrr}r~rrr)r _GetFileSearchStoreParametersrsrrr7rrrrr*rr rrrrrrrrrr rur3r*rrrrrrrIrrs r%rzFileSearchStores.gets$9 O     4   oNNl%))&11 "!,,-=>>!##H--L1 0 0y.. 0 0dXt$$$04L*  " / ;$+8l*<88L6|DDL''t\<PPH&mJBBHM1J1JM   $N>}MMm5DD ?Hd 3 3 #*#*,=t$$)0#*,BD))'.#*,@$''     !EL& %%l333 r'rMc Ttj||}|jjrt dt |}|d}|rd|}nd}|d}|r|dt|}| ddd}|j |j j |j j }tj |}tj|}|jd|||} |Bt!|d dr1tj| } |j| | S| jsint)j| j} |jjst-| } tj| t!|ddrDdt!|j d dt!|j d dt!|j d ddini} |j| | SNrMr*rzr4z>upload/v1beta/{file_search_store_name}:uploadToFileSearchStorer.r{r*r|should_return_http_response)rSr}r~rrr)r "_UploadToFileSearchStoreParametersrsrrrlrrrrr*rr rrrr(UploadToFileSearchStoreResumableResponserrrrrnr rurMr*rrrrrrrIrrs r%_upload_to_file_search_storez-FileSearchStores._upload_to_file_search_store`s >5O   P  4   A l&))&11 P L Z( ) )  P##H--L1 0 0y.. 0 0dXt$$$04L*  " / ;$+8l*<88L6|DDL''lLHg-tC$l '' 555 &mJBBHM1J1JM   $J m 6EE"$77'.'.0A4((-4'.0F--+2'.0Dd++    ! F  * %%l333 r'rNc tj|||}|jjrt dt |}|d}|rd|}nd}|d}|r|dt|}| ddd} |j |j j |j j } tj |}tj|}|jd ||| } | jsint#j| j} |jjst'| } tj| t-|ddrDdt-|j d dt-|j d dt-|j d dd ini} |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. )rMrNr*rzr4z#{file_search_store_name}:importFiler.r{r*Nr|r}r~rrr)r _ImportFileParametersrsrrrPrrrrr*rr rrrrrrrKImportFileOperationrrr rurMrNr*rrrrrrrIrrs r% import_filezFileSearchStores.import_files@,15O  5  4   4ODDl%))&11 54??   5##H--L1 0 0y.. 0 0dXt$$$04L*  " / ;$+8l*<88L6|DDL''lLH'mJBBHM1J1JM   $E5mDDm,;; ?Hd 3 3 #*#*,=t$$)0#*,BD))'.#*,@$''     !<L& %%l333 r'filec|jjrtd|i}n;t|trt |}n|d}t j||d|d\}}}i||dd}| |tj di| } | j (| j j d | j j vrd | j j vrtd d | j j vr| j j d } n| j j d } t|tjr|j|| || } n2t%j|} |j| | || } t)| j} tj| iS)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_unsetrrguser_http_optionsuser_mime_typerrrx-goog-upload-urlX-Goog-Upload-URL`Failed to upload file to file search store. Upload URL was not returned from the upload request.rrr<)rsrr isinstancedict model_dumpr prepare_resumable_uploadrrr UploadToFileSearchStoreConfigrSheadersKeyErrorioIOBase upload_fileosfspathrr UploadToFileSearchStoreOperationrrurMrr*user_config_dictr size_bytes_final_config_dictrI upload_urlupload_responsefs_pathrs r%upload_to_file_search_storez,FileSearchStores.upload_to_file_search_storesE(    I  ~)+ FD ! !?f***>>"."G *..~>>'++K88###L*a  $'+ 0052GG5FGG1H "*  % - 5 x'A'I I I#8+E+MMM  /    : B B B-56IJjj-56IJj$ ""(44  J\5oo $g(44 :z 5oAM  1 @ @r A  r'media_idc||jjrtd|d}d|vrtd|d|d}d}|r)t |t rt jd i|}n|}d}|r"t|dgt|dg}|j || }|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=mediaNrrr<) rsrrlstriprrr DownloadMediaConfigr download_filerurr*clean_idr config_modelrdatas r%download_mediazFileSearchStores.download_mediabs*    I  s##H   Fh F F F    " " "DL  FD ! !0::6::  L:\N+;<<H,(899l   ) ) ! *  D Kr'c\|j}td||||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}") rcrw)rrrur* list_requests r%listzFileSearchStores.lists7,:L  & !!   r')%__name__ __module__ __qualname__rrrvrr !CreateFileSearchStoreConfigOrDictrrstrGetFileSearchStoreConfigOrDictr!DeleteFileSearchStoreConfigOrDictr ListFileSearchStoresConfigOrDictrr#UploadToFileSearchStoreConfigOrDictrrImportFileConfigOrDictrrrrPathLikerrrrDownloadMediaConfigOrDictbytesrrrr<r'r%rprpEs8''''?'LPKKK GHK KKKKb@D MMM Mu;< M   MMMMfCG 5I5I5I 5Iu>? 5I  5I5I5I5IpKO@@@ FG@ )@@@@LEI XXX"Xu@A X  5 XXXX~8< YYY"Y Y u34 Y   YYYY@EI KKK"K #r{3'2 3 K u@A K  - KKKKb;? 2222u67 2  2222jKO FG U "#r'rpc ZeZdZedefdZdddeejdej fdZ ddde deej dej fdZ ddde deejddfd Zdddeejdejfd Zddd e deejdejfd Zddd e d e deejdejfdZddd e dee eje ejfdeejdejfdZddde deej de!fdZ"dddeejde#ej fdZ$dS)AsyncFileSearchStoresrc*t|jSrr)rrsrts r%rvzAsyncFileSearchStores.documentss $* + ++r'Nrwr*c Ktj|}|jjrt dt |j|}|d}|rd|}nd}|d}|r|dt|}| ddd}|j |j j |j j }tj |}tj|}|jd |||d{V}|jsint#j|j} tj| t+|ddrDdt+|j d dt+|j d dt+|j d dd ini} |j| | Sry)r rrsrrr+rrrrr*rr rr async_requestrrrrrrrrs r%rzAsyncFileSearchStores.createsF<O   "  4   ?  Ol&))&11 "!,,-=>>!##H--L1 0 0y.. 0 0dXt$$$04L*  " / ;$+8l*<88L6|DDL%33lLH'mJBBHM1J1JM(77 ?Hd 3 3 #*#*,=t$$)0#*,BD))'.#*,@$''     !8L& %%l333 r'r3c Ktj||}|jjrt dt |}|d}|rd|}nd}|d}|r|dt|}| ddd}|j |j j |j j }tj |}tj|}|jd |||d{V} | jsint#j| j} tj| t+|ddrDdt+|j d dt+|j d dt+|j d dd ini} |j| | Sr)r rrsrrr7rrrrr*rr rrrrrrrrrrrs r%rzAsyncFileSearchStores.gets=$9 O     4   oNNl%))&11 "!,,-=>>!##H--L1 0 0y.. 0 0dXt$$$04L*  " / ;$+8l*<88L6|DDL%33 t\<H'mJBBHM1J1JM   $N>}MMm5DD ?Hd 3 3 #*#*,=t$$)0#*,BD))'.#*,@$''     !EL& %%l333 r'rMc dKtj||}|jjrt dt |}|d}|rd|}nd}|d}|r|dt|}| ddd}|j |j j |j j }tj |}tj|}|jd|||d{V} |Bt!|d dr1tj| } |j| | S| jsint)j| j} |jjst-| } tj| t!|ddrDdt!|j d dt!|j d dt!|j d ddini} |j| | Sr)r rrsrrrlrrrrr*rr rrrrrrrrrrnrrs r%rz2AsyncFileSearchStores._upload_to_file_search_stores >5O   P  4   A l&))&11 P L Z( ) )  P##H--L1 0 0y.. 0 0dXt$$$04L*  " / ;$+8l*<88L6|DDL%33lLHg-tC$l '' 555 &mJBBHM1J1JM   $J m 6EE"$77'.'.0A4((-4'.0F--+2'.0Dd++    ! F  * %%l333 r'rNc Ktj|||}|jjrt dt |}|d}|rd|}nd}|d}|r|dt|}| ddd} |j |j j |j j } tj |}tj|}|jd ||| d{V} | jsint#j| j} |jjst'| } tj| t-|ddrDdt-|j d dt-|j d dt-|j d dd ini} |j| | Sr)r rrsrrrPrrrrr*rr rrrrrrrKrrrrrs r%rz!AsyncFileSearchStores.import_file0sb,15O  5  4   4ODDl%))&11 54??   5##H--L1 0 0y.. 0 0dXt$$$04L*  " / ;$+8l*<88L6|DDL%33lLH'mJBBHM1J1JM   $E5mDDm,;; ?Hd 3 3 #*#*,=t$$)0#*,BD))'.#*,@$''     !<L& %%l333 r'rcK|jjrtd|i}n;t|trt |}n|d}t j||d|d\}}}i||dd}| |tj di| d{V} | j (| j j d | j j vrd | j j vrtd d | j j vr| j j d } n| j j d } t|tjr%|j|| || d{V} n8t%j|} |j| | || d{V} t)| j} tj| iS)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. rNTrrrgrrrrrrrrr<)rsrrrrrr rrrr rrSrrrrasync_upload_filerrrrrrrs r%rz1AsyncFileSearchStores.upload_to_file_search_stores&    I  ~)+ FD ! !?f***>>"."G *..~>>'++K88###L*a  $'+ 6652GG5FGG7H "*  % - 5 x'A'I I I#8+E+MMM  /    : B B B-56IJjj-56IJj$ """.@@  J\Aoo $g".@@ :z AoAM  1 @ @r A  r'rcK|jjrtd|d}d|vrtd|d|d}d}|r)t |t rt jd i|}n|}d}|r"t|dgt|dg}|j || d{V}|Sr) rsrrrrrr rrasync_download_filers r%rz$AsyncFileSearchStores.download_medias:*    I  s##H   Fh F F F    " " "DL  FD ! !0::6::  L:\N+;<<H,(899l!55 !6      D Kr'clK|j}td|||d{V|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}") rcrwN)rrrs r%rzAsyncFileSearchStores.list sM,:L jjj''''''''   r')%rrrrrrvrr rrrrrrrrrrrrrrrrrrrrrrrrrrrrrr<r'r%rrs/,,,,?,LPKKK GHK KKKKb@D OOO Ou;< O   OOOOjCG 777 7u>? 7  7777tKOBBB FGB )BBBBPEI XXX"Xu@A X  5 XXXX~8< YYY"Y Y u34 Y   YYYY@EI JJJ"J #r{3'2 3 J u@A J  - JJJJ`;? 2222u67 2  2222jKO FG%'(r'rrr)8 functoolsrrrloggingrtypingrrr urllib.parserr r r r r!r rsrrrrr _operations_convertersrrvrrpagersrr getLoggerloggerrrobjectr&r+r1r5r7rDrKrPrJr\r^rerjrlrn BaseModulerprr<r'r%rsn$&%%%%%  ''''''''''"""""" &&&&&&............PPPPPP00000000%%%%%%%%  : ; ; /3tCH~v-.DcN+ #s(^ ./3   tCH~v-. DcN+  #s(^     /3  tCH~v-. DcN+  #s(^    /3  tCH~v-. DcN+  #s(^    $/3tCH~v-.DcN+ #s(^/3tCH~v-.DcN+ #s(^2/3tCH~v-.DcN+ #s(^>/3tCH~v-.DcN+ #s(^./3tCH~v-.DcN+ #s(^*/3tCH~v-.DcN+ #s(^./3  tCH~v-. DcN+  #s(^    /3tCH~v-.DcN+ #s(^2/3tCH~v-.DcN+ #s(^>/3tCH~v-.DcN+ #s(^,/3  tCH~v-. DcN+  #s(^    m m m m m {-m m m `r r r r r K2r r r r r r'