Ah#)*SrSSKrSSKJr SSKJrJrJr SSKJ r SSK J r J r J r JrJr SSKJrJr SSKJr SS KJrJr SS KJr SS KJrJr SS KJrJr SS K J!r! SSK"J#r# \"S5r$\"S\S9r%S\\$S\ \$/\%4S\\$4Sjr&"SS\5r'g)zo Ensemble retriever that ensemble the results of multiple retrievers by using weighted Reciprocal Rank Fusion N) defaultdict)HashableIterableIterator)chain)AnyCallableOptionalTypeVarcast)#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 U/var/www/html/shao/venv/lib/python3.13/site-packages/langchain/retrievers/ensemble.py unique_by_keyr&%s6 5D QKA $ HHQKGs;;c \rSrSr%Sr\\\S'\\\S'Sr \ \S'Sr \ \ \S'\S \\4S j5r\"S S 9\S \\ \4S \4Sj55r\SS\ S\ \S\S \\4Sjj5r\SS\ S\ \S\S \\4Sjj5rS\ S\S \\4SjrS\ S\S \\4SjrSS.S\ S\S\ \S \\4SjjrSS.S\ S\S\ \S \\4Sjjr S\\\S \\4Sjr!Sr"g)EnsembleRetriever6a9Retriever 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. Default is 60. 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.c3F# UHoRHo"v M M g7fN) config_specs).0 retrieverspecs r% 1EnsembleRetriever.config_specs..Ns ' "1Y>T>TdD>TD/s!)rr*)selfs r%r2EnsembleRetriever.config_specsKs"'' "&//'   before)modevaluescfURS5(d[US5nSU- /U-US'U$)Nr+r*)getlen)clsr= n_retrieverss r% set_weightsEnsembleRetriever.set_weightsRs>zz)$$vl34L!"\!1 2\ AF9  r:inputconfigkwargsc SSKJn [U5nURUR S5SUR SS5UR S/5UR UR S05UR S9nUR"SU4S UR S 5=(d UR50UD6nURXUS 9nUR"U40UD6 U$![anURU5 eSnAff=f) Nr)CallbackManager callbacksverboseFtagsmetadatarLinheritable_tags local_tagsinheritable_metadatalocal_metadatanamerun_name run_managerrG) langchain_core.callbacksrJr configurer@rMrNon_retriever_startget_name rank_fusionon_retriever_end Exceptionon_retriever_error) r8rFrGrHrJcallback_managerrWresultr#s r%invokeEnsembleRetriever.invokeZs  =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)AsyncCallbackManagerrKrLFrMrNrOrTrUrV) rXrerrYr@rMrNrZr[ arank_fusionr]r^r_) r8rFrGrHrer`rWrar#s r%ainvokeEnsembleRetriever.ainvokesB 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*queryrWc$URX5$)z Get the relevant documents for a given query. Args: query: The query to search for. Returns: A list of reranked documents. )r\r8rirWs r%_get_relevant_documents)EnsembleRetriever._get_relevant_documentss"33r:c@# URX5IShvN $N7f)z Asynchronously get the relevant documents for a given query. Args: query: The query to search for. Returns: A list of reranked documents. N)rfrks r%_aget_relevant_documents*EnsembleRetriever._aget_relevant_documentss"&&u::::s )rGc[UR5VVs/sH2upEURU[UUR SUS-3S9S95PM4 nnn[ [ U55HEnXdVs/sH1n[U[5(a[[[U5S9OUPM3 snXd'MG URU5$s snnfs snf)z Retrieve the results of the retrievers and use rank_fusion_func to get the final result. Args: query: The query to search for. Returns: A list of reranked documents. retriever_r?tagrK page_content) enumerater*rbr get_childrangerA isinstancestrrr weighted_reciprocal_rankr8rirWrGir4retriever_docsdocs r%r\EnsembleRetriever.rank_fusions6!*$// :  !;    )33*QUG8L3M !;   s>*+A*,!,C:DC9M9Md3n5SVV,!N ,,,^<<'  !s 8C17Cc# [R"[UR5VVs/sH2upEUR U[ UUR SUS-3S9S95PM4 snn6IShvN n[[U55H7nXdVs/sH#n[U[5(d [US9OUPM% snXd'M9 URU5$s snnfNjs snf7f)z Asynchronously retrieve the results of the retrievers and use rank_fusion_func to get the final result. Args: query: The query to search for. Returns: A list of reranked documents. rrr?rsruNrv) asynciogatherrxr*rgrryrzrAr{rr}r~s r%rfEnsembleRetriever.arank_fusions& '~~%.doo$> %?LA!! "-"7"7jQ*+A*,!,C3=S(2K2Kc*QTT,!N ,,,^<<)   !s.)C8C #C+C,!C )C6!CC doc_listsc^^[U5[TR5:wa Sn[U5e[[5m[ UTR5Hdup4[ USS9HQupVTTRc UROURTR==XETR-- - ss'MS Mf [R"U5n[[UU4Sj5SUU4SjS9$)ay Perform 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: list: The final aggregated list of items sorted by their weighted RRF scores in descending order. zTRc UR$URTR$r1r.rwrN)rr8s r%.Fs2{{*$$3dkk23r:Tcr>TTRcUR$URTR$r1r)r rrf_scorer8s r%rrMs1I$(KK$7  =@\\$++=Vr:)reverser)rAr+ ValueErrorrfloatziprxr.rwrNr-r from_iterablesortedr&) r8rmsgdoc_listweightrankrall_docsrs ` @r%r}*EnsembleRetriever.weighted_reciprocal_rank s y>S. .PCS/ !'2%&8 #It|| < H&xq9  ;;.(( \\$++6  dff}- .:!=&&y1    r:r1)#__name__ __module__ __qualname____firstlineno____doc__listr__annotations__rr-intr.r r|propertyrr2r classmethoddictrrDrrrrbrgrrlr ror\rfr}__static_attributes__rr:r%r(r(6s ]## %[ AsK FHSM   d#89  (#c3hC$ ,0""(" " h ""H,0&&(& & h &&P444 4 h 4&;;9 ; h ;0,0 &=&=4&= ( &= h &=Z,0 (=(=9(= ( (= h (=T0 X'0  h0 r:r()(rr collectionsrcollections.abcrrr itertoolsrtypingrr r r r rXr rlangchain_core.documentsrlangchain_core.retrieversrrlangchain_core.runnablesrlangchain_core.runnables.configrrlangchain_core.runnables.utilsrrpydanticrtyping_extensionsrrrr&r(rr:r%rs #88.B3G%& CL Cx HQKhsAv.>8A;"Z Z r: