/j$ SSKrSSKJrJrJr SSKrSSKJr SSKr\R"5r Sr Sr Sr "SS5rg! Sr N=f) N)UnionOptionalLiteral) TavilyClientc  SnURR5HnUSUR:waMUSS:wa[SURS35eSnUSSHdnUS UR:waMUSS :wa[S URS 35eUS S:wa[S URSUS S35eSn O U(d[S URS35eSnM U(d[SURS35eg)a- Check that the index specified by the parameters exists and is a valid vector search index. Raises: ValueError: If the index does not exist, is not of type 'vectorSearch', or if the embeddings field does not exist, is not of type 'vector', or has similarity other than 'cosine'. Fnametype vectorSearchzIndex 'z+' exists but is not of type 'vectorSearch'.latestDefinitionfieldspathvectorzField 'z%' exists but is not of type 'vector'. similaritycosinez' exists but has similarity 'z' instead of 'cosine'.Tz+' does not exist in index '{client.index}'.z' does not exist.N) collectionlist_search_indexesindex ValueErrorembeddings_field)client index_existsr field_existsfields M/app/agent/.venv/lib/python3.13/site-packages/tavily/hybrid_rag/hybrid_rag.py_validate_indexr siL""668 =FLL (  =N *wv||n5//0 0 -.x8EV} 7 77V}( 76+B+B*CD@"@AA|$0 76+B+B*CD005l0C/DDZ"\]] L 9wv'>'>&?@778 8 79: 76<<.0ABCC c@[RSUUS9R$)Nzembed-english-v3.0)modeltexts input_type)coembed embeddings)rr s r _cohere_embedr$6s( 88"  j rc [RSUUVs/sHo3SPM snUS9nURVs/sH nXRSUR0-PM" sn$s snfs snf)Nzrerank-english-v3.0content)rquery documentstop_nscore)r!rerankresultsrrelevance_score)r'r(r)docresponseresults r_cohere_rerankr1=s|yy4E>G#His Ni#HPUWH && &F ,,7F,B,B"CC& $I s A" 'A'c\rSrSrSS\\S4S\SS\S\S\S \\S \\S \\ R4S jjr SS jr Sr g)TavilyHybridClientFNapi_key db_providermongodbrr content_fieldembedding_functionranking_functionsessionc [XS9UlUS:wa [S5eX0lX@lXPlX`lUc[OUUlUc[OUUl [U5 g)a A client for performing hybrid RAG using both the Tavily API and a local database collection. Parameters: api_key (str): The Tavily API key. If this is set to None, it will be loaded from the environment variable TAVILY_API_KEY. db_provider (str): The database provider. Currently only 'mongodb' is supported. collection (str): The name of the collection in the database that will be used for local search. index (str): The name of the collection's vector search index. embeddings_field (str): The name of the field in the collection that contains the embeddings. content_field (str): The name of the field in the collection that contains the content. embedding_function (callable): If provided, this function will be used to generate embeddings for the search query and documents. ranking_function (callable): If provided, this function will be used to rerank the combined results. session (requests.Session): If provided, this pre-configured session will be used for HTTP requests. When set, api_key is optional. )r;r7z;Only MongoDB is currently supported as a database provider.N) rtavilyrrrrr8r$r9r1r:r) selfr5r6rrrr8r9r:r;s r__init__TavilyHybridClient.__init__Gsb6#7< ) #Z[ [$ 0*3E3M-Se2B2JP`rc UcUnUcUnURU/S5Sn[URRSURUR UUS-US.0SSSUR 3S S 0S S .0/55nUS:a#URR"U4S U0UD6Sn O/n U V s/sHn U SU SSS.PM n n X-n [U 5S:Xa/$URXU5n [U 5U:aU SUn US:aUS:wa/n URU V s/sHoSPM sn S5n[U 5HeupXU S'US:Xa1U RUR U SUR U S05 MCU"U 5n U (dMTU RU 5 Mg URRU 5 U $s sn fs sn f)a Return results for the given query from both the tavily API (foreign) and the specified mongo collection (local). Parameters: query (str): The query to search for. max_results (int): The maximum number of results to return. max_local (int): The maximum number of local results to return. max_foreign (int): The maximum number of foreign results to return. save_foreign (bool or function): Whether to save the foreign results in the collection. If a function is provided, it will be used to transform the foreign results before saving. N search_queryrz $vectorSearch)rr queryVector numCandidateslimitz$project$z$metavectorSearchScorelocal)_idr&r*origin max_resultsr,r&r*foreign)r&r*rKFsearch_documentr#T)r9listr aggregaterrr8r=searchlenr: enumerateappend insert_many)r>r'rL max_local max_foreign save_foreignkwargsquery_embeddings local_resultsforeign_resultsr0projected_foreign_resultscombined_resultsr(r#is rrQTavilyHybridClient.searchqsH  #I  %K22E7NKANT__66!ZZ 11#3%.]& " !"4#5#5"67!4&  8   . ?"kk00ZKZSYZ[deO O* % * "),#  * "% )D  A %I 00+V  ; ./ =  ?|u4I00Ra1bRa2CRa1bduvJ&7 '1}|$4'$$**F9,=--vl/C& *&1Fv!((08 OO ' ' 2U% .2cs +GG)rr8r9rrr:r=)r#r&NNN) NNF)__name__ __module__ __qualname____firstlineno__rstrrrcallablerequestsSessionr?rQ__static_attributes__rrr3r3Fs%1!*593726(39%(!+(  ( " ((!) 2('x0(h../(TIM!_ rr3)ostypingrrrrhr=rcohereClientr!rr$r1r3rkrrrpsQ ++ B'DRJ J y Bs AA