R-j% vddlmZddlZddlZddlZddlZddlmZmZm Z ddl m Z ddl m Z ddl mZddl mZdd l mZdd l mZdd lmZdd lmZdd lmZddlmZddlmZmZddlmZm Z ejBdZ" 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 jjZ6Gd'd(e jjZ7y)*)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|dgt|dgt|dgt|dg-t|dgtj|t|dg|S)N display_name displayNameembedding_modelembeddingModel)getvsetvtt_modelrrr to_objects m/Users/jameslopez/projects/TradingBot25/.venv/lib/python3.12/site-packages/google/genai/file_search_stores.py%_CreateFileSearchStoreConfig_to_mldevr&)st !) +'(4kN;K)LM +)*+7  *d;1B0CDE cTi}t|dgt|t|dg||SNconfig)rr&r#s r%)_CreateFileSearchStoreParameters_to_mldevr+=s8 !) +z".)Dxj19 r'cXi}t|dgt|ddgt|dg|S)Nforce_queryrr rrr$s r%%_DeleteFileSearchStoreConfig_to_mldevr1Ks8!) +y!-7+T+y-IJ r'ci}t|dgt|ddgt|dgt|dgtt|dg||S)Nname_urlr*)rr r1r0s r%)_DeleteFileSearchStoreParameters_to_mldevr5Ws_!) +x ,VV$d;&AB +z".) [8*%y r'cXi}t|dgt|ddgt|dg|S)Nr3r4r/r0s r%&_GetFileSearchStoreParameters_to_mldevr7gs8!) +x ,VV$d;&AB r'ci}t|dg't|dgt|dgDcgc]}|c}t|dgt|dgt|dg|Scc}w)Ncustom_metadatacustomMetadatachunking_configchunkingConfigr/rrr$items r%_ImportFileConfig_to_mldevr?rs!) +)*+7 {->,?@A@$@A  +)*+7  [,-.  Bs A c i}t|dgt|dgt|dgt|dgt|dgt|dgt|dgt|dgt|dgt|dgt|dgt|dgt|dg#t|dgtt|dg||S)Nr3metadatadoneerrorresponse)rr _ImportFileResponse_from_mldevr0s r%_ImportFileOperation_from_mldevrFs!) +x ,VHd;9: + |$0ZL${ZL"AB +x ,VHd;9: +y!-WItK';< + |$0  & zl +Y  r'ci}t|dgt|ddgt|dgt|dgt|dgt|dgt|dgtt|dg||S)Nfile_search_store_namer4 file_namefileNamer*)rr r?r0s r%_ImportFileParameters_to_mldevrKs!) +012> )* [345  + }%1ZL${[M"BC +z".tK( +'(4_%tK.9I'JK r'ci}t|dgt|ddgt|dgt|dgt|ddgt|dg|S)N page_sizer.pageSize page_token pageTokenr/r0s r%$_ListFileSearchStoresConfig_to_mldevrWsm!) + }%1*-tK+/O + ~&2 ; [<.) r'cRi}t|dgtt|dg||Sr))rrWr0s r%(_ListFileSearchStoresParameters_to_mldevrYs6!) +z".( [8*%y r'ci}t|dgt|dgt|dgt|dgt|dgt|dgt|dg't|dgt|dgDcgc]}|c}|Scc}w)NrMrN nextPageTokennext_page_tokenfileSearchStoresfile_search_storesr/r=s r%(_ListFileSearchStoresResponse_from_mldevr_s!) +)*+7'(${=N,?@A@$@A  +)*+7  [,-.  Bs3 B+ ci}t|dgt|ddgt|dgt|dgtt|dg||S)NrHr4r*)rr rcr0s r%+_UploadToFileSearchStoreParameters_to_mldevre$sk!) +012> )* [345  +z".+ [8*%y r'cVi}t|dgt|dgt|dg|S)NrMrNr/r0s r%4_UploadToFileSearchStoreResumableResponse_from_mldevrg8s@!) +)*+7'(${=N!##H-LvQy./ 0dXt$04L*  " " / / ;$++88l**<8L66|DL''lLH'mmBHMM1JM((77 ?Hd 3 #*#**,=t$)0#**,BD)'.#**,@$'   !8L& %%l3 r'r3c tj||}|jjr t dt |}|j d}|rdj|}nd}|j d}|r|dt|}|jddd}|j,|jj|jj}tj|}tj|}|jjd |||} | j sint#j$| j } tj&j)| t+|ddrGdt+|jd dt+|jd dt+|jd dd ini} |jj-| | 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*rsr4{name}r.rtr*Nrrvrwrxryrz)r _GetFileSearchStoreParametersrlr}r~r7rrrrr*rr rrrrrrrrrr rnr3r*rrrrrrrDrrs r%rzFileSearchStores.gets$99 O     4  oNl%))&1 !,,-=>!##H-LvQy./ 0dXt$04L*  " " / / ;$++88l**<8L66|DL''t\<PH&mmBHMM1JM    $ $>}Mm55DD ?Hd 3 #*#**,=t$)0#**,BD)'.#**,@$'   !EL& %%l3 r'rHc tj||}|jjr t dt |}|j d}|rdj|}nd}|j d}|r|dt|}|jddd}|j,|jj|jj}tj|}tj|}|jjd|||} |@t!|d dr3tj"| } |jj%| | S| j&sint)j*| j&} |jjs t-| } tj"j/| t!|ddrGdt!|jd dt!|jd dt!|jd ddini} |jj%| | SNrHr*rsr4z>upload/v1beta/{file_search_store_name}:uploadToFileSearchStorer.rtr*rushould_return_http_response)rNrvrwrxryrz)r "_UploadToFileSearchStoreParametersrlr}r~rerrrrr*rr rrrr(UploadToFileSearchStoreResumableResponserrrrrgr rnrHr*rrrrrrrDrrs r%_upload_to_file_search_storez-FileSearchStores._upload_to_file_search_store`sP >>5O     4  A l&))&1  L Z( ) P##H-LvQy./ 0dXt$04L*  " " / / ;$++88l**<8L66|DL''lLHg-tCC$l '' 5 &mmBHMM1JM    $ $J m 66EE"$7'.'..0A4(-4'..0F-+2'..0Dd+  ! F * %%l3 r'rIc tj|||}|jjr t dt |}|j d}|rdj|}nd}|j d}|r|dt|}|jddd} |j,|jj|jj} tj|}tj|}|jjd ||| } | j sint#j$| j } |jjs t'| } tj(j+| t-|ddrGdt-|jd dt-|jd dt-|jd dd ini} |jj/| | 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. )rHrIr*rsr4z#{file_search_store_name}:importFiler.rtr*Nrurvrwrxryrz)r _ImportFileParametersrlr}r~rKrrrrr*rr rrrrrrrFImportFileOperationrrr rnrHrIr*rrrrrrrDrrs r% import_filezFileSearchStores.import_files,115O    4  4ODl%))&1 4??  5##H-LvQy./ 0dXt$04L*  " " / / ;$++88l**<8L66|DL''lLH'mmBHMM1JM    $ $5mDm,,;; ?Hd 3 #*#**,=t$)0#**,BD)'.#**,@$'   !<L& %%l3 r'filec|jjr td|i}n.t|tr t |}n|j d}t j||jd|jd\}}}i||dd}|j|tjdi|} | jF| jj0d | jjvr#d | jjvr td d | jjvr| jjd } n| jjd } t|tj r |jj#|| || } n4t%j&|} |jj#| | || } t)| j*} tj,j/| i S)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. =This method is only supported in the Gemini Developer client.T exclude_unsetrrauser_http_optionsuser_mime_typerrrx-goog-upload-urlX-Goog-Upload-URL`Failed to upload file to file search store. Upload URL did not returned from the upload request.rrz)rlr}r~ isinstancedict model_dumpr prepare_resumable_uploadrrr UploadToFileSearchStoreConfigrNheadersKeyErrorioIOBase upload_fileosfspathrr UploadToFileSearchStoreOperationrrnrHrr*user_config_dictr size_bytes_final_config_dictrD upload_urlupload_responsefs_pathrs r%upload_to_file_search_storez,FileSearchStores.upload_to_file_search_stores&    I ~)+ FD !f***>"."G"G *..~>'++K8#L*a  $'+ 00522G5FG1H ""*  % % - - 5 x'A'A'I'I I#8+E+E+M+MM  /   : : B B B--556IJj--556IJj$ "((44  J\5o $g((44 :z 5oAM  1 1 @ @r A r'media_idch|jjr td|jd}d|vrtd|d|d}d}|r(t |t rt jd i|}n|}d}|rt|dg t|dg}|jj|| }|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) rlr}r~lstriprrr DownloadMediaConfigr download_filernrr*clean_idr config_modelrdatas r%download_mediazFileSearchStores.download_mediaas*    I s#H  %h\2F F  Zz "DL FD !00:6:  L\N+;<H,(89l    ) ) ! * D Kr'cV|j}td||j||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}") r^rp)rrrnr* list_requests r%listzFileSearchStores.lists1,::L  & !  r')%__name__ __module__ __qualname__rrrorr !CreateFileSearchStoreConfigOrDictrrstrGetFileSearchStoreConfigOrDictr!DeleteFileSearchStoreConfigOrDictr ListFileSearchStoresConfigOrDictrr#UploadToFileSearchStoreConfigOrDictrrImportFileConfigOrDictrrrrPathLikerrrrDownloadMediaConfigOrDictbytesrrrrr'r%ririEsJ'''LPK G GHK Kb@D M Mu;;< M   MfCG 5I 5Iu>>? 5I  5IpKO@ F FG@ ))@LEI X"Xu@@A X  55 X~8< Y"Y Y u334 Y    Y@EI J"J #r{{3'2 3 J u@@A J  -- J`;? 22u667 2  2jKO F FG U " "#r'ric eZdZedefdZdddeejdejfdZ ddde deejdejfdZ ddde deejddfd Zdddeej dej"fd Zddd e deej&dej(fd Zddd e d e deej,dej.fdZddd e dee ej6e ej:fdeej&dej<fdZddde deej@de!fdZ"dddeej de#ejfdZ$y)AsyncFileSearchStoresrc,t|jSrk)rrlrms r%rozAsyncFileSearchStores.documentss $** ++r'Nrpr*c Ktj|}|jjr t dt |j|}|j d}|rdj|}nd}|j d}|r|dt|}|jddd}|j,|jj|jj}tj|}tj|}|jjd |||d{}|j sint#j$|j } tj&j)| t+|ddrGdt+|jd dt+|jd dt+|jd dd ini} |jj-| | S7íwrr)r r|rlr}r~r+rrrrr*rr rr async_requestrrrrrrrrs r%rzAsyncFileSearchStores.creates<<O     4  ?   Ol&))&1 !,,-=>!##H-LvQy./ 0dXt$04L*  " " / / ;$++88l**<8L66|DL%%33lLH'mmBHMM1JM((77 ?Hd 3 #*#**,=t$)0#**,BD)'.#**,@$'   !8L& %%l3 5sD/G71G52CG7r3c Ktj||}|jjr t dt |}|j d}|rdj|}nd}|j d}|r|dt|}|jddd}|j,|jj|jj}tj|}tj|}|jjd |||d{} | j sint#j$| j } tj&j)| t+|ddrGdt+|jd dt+|jd dt+|jd dd ini} |jj-| | S7íwr)r rrlr}r~r7rrrrr*rr rrrrrrrrrrrs r%rzAsyncFileSearchStores.gets$99 O     4  oNl%))&1 !,,-=>!##H-LvQy./ 0dXt$04L*  " " / / ;$++88l**<8L66|DL%%33 t\<H'mmBHMM1JM    $ $>}Mm55DD ?Hd 3 #*#**,=t$)0#**,BD)'.#**,@$'   !EL& %%l3 ;sD$H &H 'C%H rHc Ktj||}|jjr t dt |}|j d}|rdj|}nd}|j d}|r|dt|}|jddd}|j,|jj|jj}tj|}tj|}|jjd|||d{} |@t!|d dr3tj"| } |jj%| | S| j&sint)j*| j&} |jjs t-| } tj"j/| t!|ddrGdt!|jd dt!|jd dt!|jd ddini} |jj%| | S7'wr)r rrlr}r~rerrrrr*rr rrrrrrrrrrgrrs r%rz2AsyncFileSearchStores._upload_to_file_search_stores_ >>5O     4  A l&))&1  L Z( ) P##H-LvQy./ 0dXt$04L*  " " / / ;$++88l**<8L66|DL%%33lLHg-tCC$l '' 5 &mmBHMM1JM    $ $J m 66EE"$7'.'..0A4(-4'..0F-+2'..0Dd+  ! F * %%l3 UsD%I'I(D'IrIc "Ktj|||}|jjr t dt |}|j d}|rdj|}nd}|j d}|r|dt|}|jddd} |j,|jj|jj} tj|}tj|}|jjd ||| d{} | j sint#j$| j } |jjs t'| } tj(j+| t-|ddrGdt-|jd dt-|jd dt-|jd dd ini} |jj/| | S7wr)r rrlr}r~rKrrrrr*rr rrrrrrrFrrrrrs r%rz!AsyncFileSearchStores.import_file/s ,115O    4  4ODl%))&1 4??  5##H-LvQy./ 0dXt$04L*  " " / / ;$++88l**<8L66|DL%%33lLH'mmBHMM1JM    $ $5mDm,,;; ?Hd 3 #*#**,=t$)0#**,BD)'.#**,@$'   !<L& %%l3 ;sD&H(H )C%HrcZK|jjr td|i}n.t|tr t |}n|j d}t j||jd|jd\}}}i||dd}|j|tjdi| d{} | jF| jj0d | jjvr#d | jjvr td d | jjvr| jjd } n| jjd } t|tj r(|jj#|| || d{} n"."G"G *..~>'++K8#L*a  $'+ 66522G5FG7H ""*  % % - - 5 x'A'A'I'I I#8+E+E+M+MM  /   : : B B B--556IJj--556IJj$ ""..@@  J\Ao $g"..@@ :z AoAM  1 1 @ @r A A* s7CH+H$C(H+,H'-8CcrK|j}td||j|d{|S7 w)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}") r^rpN)rrrs r%rzAsyncFileSearchStores.list s=,::L jjj''   (s )75 7)%rrrrrrorr rrrrrrrrrrrrrrrrrrrrrrrrrrrrrrr'r%rrsF,,,LPK G GHK Kb@D O Ou;;< O   OjCG 7 7u>>? 7  7tKOB F FGB ))BPEI X"Xu@@A X  55 X~8< Y"Y Y u334 Y    Y@EI J"J #r{{3'2 3 J u@@A J  -- J`;? 22u667 2  2jKO F FG%''(r'rrk)8 functoolsrrrloggingrtypingrrr urllib.parserr r r r r!r rlrrrrr _operations_convertersrrorrpagersrr getLoggerloggerrrobjectr&r+r1r5r7r?rFrKrErWrYr_rcrerg BaseModulerirrr'r%rs$&  ''" &..P0%   : ; /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(^ l {--l ^r K22r r'