o 粪g@sPddlZddlmZddlmZmZddlmZddlm Z Gddde Z dS)N)List)#AsyncCallbackManagerForRetrieverRunCallbackManagerForRetrieverRun)Document) BaseRetrieverc@seZdZUdZeeed< dededee fddZ dede dee fdd Z dededee fd d Z dede dee fd d ZdS)MergerRetrieverz9Retriever that merges the results of multiple retrievers. retrieversquery run_managerreturncCs|||}|S)z Get the relevant documents for a given query. Args: query: The query to search for. Returns: A list of relevant documents. )merge_documentsselfr r merged_documentsra/var/www/html/chatdoc2/venv/lib/python3.10/site-packages/langchain/retrievers/merger_retriever.py_get_relevant_documentss z'MergerRetriever._get_relevant_documentscs|||IdH}|S)z Asynchronously get the relevant documents for a given query. Args: query: The query to search for. Returns: A list of relevant documents. N)amerge_documentsr rrr_aget_relevant_documents'sz(MergerRetriever._aget_relevant_documentsc srfddt|jD}g}ttt|dd}t|D]}t|j|D]\}}|t|kr5|||q$q|S)z Merge the results of the retrievers. Args: query: The query to search for. Returns: A list of merged documents. c s2g|]\}}|jdd|didqS) callbacks retriever_{}config)invoke get_childformat.0i retrieverr r rr Jsz3MergerRetriever.merge_documents..rdefault) enumeratermaxmaplenrangezipappend rr r retriever_docsrmax_docsrr docrr!rr <s    zMergerRetriever.merge_documentsc stjfddt|jDIdH}g}ttt|dd}t|D]}t|j|D]\}}|t|kr<| ||q+q#|S)z Asynchronously merge the results of the retrievers. Args: query: The query to search for. Returns: A list of merged documents. c 3s6|]\}}|jdd|didVqdS)rrrrN)ainvokerrrr!rr ms z3MergerRetriever.amerge_documents..Nrr#) asynciogatherr%rr&r'r(r)r*r+r,rr!rr^s     z MergerRetriever.amerge_documentsN)__name__ __module__ __qualname____doc__rr__annotations__strrrrrrr rrrrrr s@     "r) r2typingrlangchain_core.callbacksrrlangchain_core.documentsrlangchain_core.retrieversrrrrrrs