KKio*&SrSSKrSSKJr SSKJrJrJrJr SSK J r SSK J r J r Jr SSKJrJr SSKJr SS KJrJr SS KJr SS KJrJr SS KJrJr SS KJ r SSK!J"r" \ "S5r#\ "S\S9r$S\\#S\\#/\$4S\\#4Sjr%"SS\5r&g)zEnsemble Retriever. Ensemble retriever that ensemble the results of multiple retrievers by using weighted Reciprocal Rank Fusion. N) defaultdict)CallableHashableIterableIterator)chain)AnyTypeVarcast)#AsyncCallbackManagerForRetrieverRunCallbackManagerForRetrieverRun)Document) BaseRetriever RetrieverLike)RunnableConfig) ensure_config patch_config)ConfigurableFieldSpecget_unique_config_specs)model_validator)overrideTH)bounditerablekeyreturnc#z# [5nUH'nU"U5=oB;dMURU5 Uv M) g7f)zYield unique elements of an iterable based on a key function. Args: iterable: The iterable to filter. key: A function that returns a hashable key for each element. Yields: Unique elements of the iterable based on the key function. N)setadd)rrseeneks g/var/www/html/dynamic-report/venv/lib/python3.13/site-packages/langchain_classic/retrievers/ensemble.py unique_by_keyr%$s6 5D QKA $ HHQKGs;;c \rSrSr%Sr\\\S'\\\S'Sr \ \S'Sr \ S-\S'\ S \\4S j5r\"S S 9\S \\ \4S \4Sj55r\SS\ S\S-S\S \\4Sjj5r\SS\ S\S-S\S \\4Sjj5rS\ S\S \\4SjrS\ S\S \\4SjrSS.S\ S\S\S-S \\4SjjrSS.S\ S\S\S-S \\4SjjrS\\\S \\4Sjr Sr!g)EnsembleRetriever5a"Retriever that ensembles the multiple retrievers. It uses a rank fusion. Args: retrievers: A list of retrievers to ensemble. weights: A list of weights corresponding to the retrievers. Defaults to equal weighting for all retrievers. c: A constant added to the rank, controlling the balance between the importance of high-ranked items and the consideration given to lower-ranked items. id_key: The key in the document's metadata used to determine unique documents. If not specified, page_content is used. retrieversweights<cNid_keyrc:[SUR55$)z+List configurable fields for this runnable.c3J# UHoRHo"v M M g7fN) config_specs).0 retrieverspecs r$ 1EnsembleRetriever.config_specs..Ls ' "1Y>T>TdD>TD/s!#)rr))selfs r$r1EnsembleRetriever.config_specsIs"'' "&//'   before)modevaluescfURS5(d[US5nSU- /U-US'U$)Nr*r))getlen)clsr< n_retrieverss r$ _set_weightsEnsembleRetriever._set_weightsPs>zz)$$vl34L!"\!1 2\ AF9  r9inputconfigkwargsc SSKJn [U5nURUR S5SUR SS5UR S/5UR UR S05UR S9nUR"SU4S UR S 5=(d UR50UD6nURXUS 9nUR"U40UD6 U$![anURU5 eSnAff=f) Nr)CallbackManager callbacksverboseFtagsmetadatarKinheritable_tags local_tagsinheritable_metadatalocal_metadatanamerun_name run_managerrF) langchain_core.callbacksrIr configurer?rLrMon_retriever_startget_name rank_fusionon_retriever_end Exceptionon_retriever_error) r7rErFrGrIcallback_managerrVresultr"s r$invokeEnsembleRetriever.invokeXs  =v&*44 JJ{ # JJy%0#ZZ3yy!'J!;==5 '99   J':4==?   %%eV%TF  ( (  M   * *1 -  s7C C=&C88C=c X# SSKJn [U5nURUR S5SUR SS5UR S/5UR UR S05UR S9nUR"SU4S UR S 5=(d UR50UD6IShvN nURUUUS 9IShvN nUR"U40UD6IShvN U$N;N#N ![a nURU5IShvN eSnAff=f7f) Nr)AsyncCallbackManagerrJrKFrLrMrNrSrTrU) rWrdrrXr?rLrMrYrZ arank_fusionr\r]r^) r7rErFrGrdr_rVr`r"s r$ainvokeEnsembleRetriever.ainvoke}sB Bv&/99 JJ{ # JJy%0#ZZ3yy!'J!;==: -??   J':4==?     ,,'-F..  M)    003 3 3  sfB9D*;C7<D*C=C9C=D*0C;1D*9C=;D*= D'D"DD""D''D*queryrVc$URX5$)zGet the relevant documents for a given query. Args: query: The query to search for. run_manager: The callback handler to use. Returns: A list of reranked documents. )r[r7rhrVs r$_get_relevant_documents)EnsembleRetriever._get_relevant_documentss 33r9c@# URX5IShvN $N7f)zAsynchronously get the relevant documents for a given query. Args: query: The query to search for. run_manager: The callback handler to use. Returns: A list of reranked documents. N)rerjs r$_aget_relevant_documents*EnsembleRetriever._aget_relevant_documentss &&u::::s )rFc[UR5VVs/sH2upEURU[UUR SUS-3S9S95PM4 nnn[ [ U55HBnXdVs/sH-n[U[5(a[[SU5S9OUPM/ snXd'MD URU5$s snnfs snf)a#Rank fusion. Retrieve the results of the retrievers and use rank_fusion_func to get the final result. Args: query: The query to search for. run_manager: The callback handler to use. config: Optional configuration for the retrievers. Returns: A list of reranked documents. retriever_r>tagrJstr page_content) enumerater)rar get_childranger@ isinstancerurr weighted_reciprocal_rankr7rhrVrFir3retriever_docsdocs r$r[EnsembleRetriever.rank_fusions:!*$// :  !;    )33*QUG8L3M !;   s>*+A*,!,CrrrtNrv) asynciogatherrxr)rfrryrzr@r{rr|r}s r$reEnsembleRetriever.arank_fusions* '~~%.doo$> %?LA!! "-"7"7jQ*+A*,!,C3=S(2K2Kc*QTT,!N ,,,^<<)   !s.)C9C $C,C-!C*C8!CC doc_listsc^^[U5[TR5:wa Sn[U5e[[5m[ UTRSS9Hdup4[ USS9HQupVTTRc UROURTR==XETR-- - ss'MS Mf [R"U5n[[UU4Sj5SUU4SjS 9$) alPerform weighted Reciprocal Rank Fusion on multiple rank lists. You can find more details about RRF here: https://plg.uwaterloo.ca/~gvcormac/cormacksigir09-rrf.pdf. Args: doc_lists: A list of rank lists, where each rank list contains unique items. Returns: The final aggregated list of items sorted by their weighted RRF scores in descending order. z)startch>TRc UR$URTR$r0r-rwrM)rr7s r$.Fs2{{*$$3dkk23r9Tcr>TTRcUR$URTR$r0r)r rrf_scorer7s r$rrMs1I$(KK$7  =@\\$++=Vr9)reverser)r@r* ValueErrorrfloatziprxr-rwrMr,r from_iterablesortedr%) r7rmsgdoc_listweightrankrall_docsrs ` @r$r|*EnsembleRetriever.weighted_reciprocal_rank s y>S. .PCS/ !'2%&8 #It||E J H&xq9  ;;.(( \\$++6  dff}- .:!K&&y1    r9r0)"__name__ __module__ __qualname____firstlineno____doc__listr__annotations__rr,intr-rupropertyrr1r classmethoddictr rCrrrrarfr rkr rnr[rer|__static_attributes__rr9r$r'r'5s ]## %[ AsKFC$J  d#89  (#$sCx.S$ )-""%" " h ""H)-&&%& & h &&P444 4 h 4$;;9 ; h ;.)- (=(=4(= % (= h (=^)- *=*=9*= % *= h *=X0 X'0  h0 r9r')'rr collectionsrcollections.abcrrrr itertoolsrtypingr r r rWr r langchain_core.documentsrlangchain_core.retrieversrrlangchain_core.runnablesrlangchain_core.runnables.configrrlangchain_core.runnables.utilsrrpydanticrtyping_extensionsrrrr%r'rr9r$rs #BB .B3G%& CL Cx HQKhsAv.>8A;"[ [ r9