o `+ i` @sDddlZddlmZmZddlmZddlmZGdddeZdS)N)#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_documentsselfrr rq/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/langchain/retrievers/merger_retriever.py_get_relevant_documentss z'MergerRetriever._get_relevant_documentscs|||IdHS)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)Zinvoke get_child.0iZ retrieverrr rr Gsz3MergerRetriever.merge_documents..rdefault) enumeratermaxmaplenrangezipappend r rr Zretriever_docsZmerged_documentsZmax_docsrZ _retrieverdocrrrr 7s    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)Zainvokerrrrr js z3MergerRetriever.amerge_documents..Nrr) asynciogatherr rr!r"r#r$r%r&r'rrrrYs     z MergerRetriever.amerge_documentsN)__name__ __module__ __qualname____doc__listr__annotations__strrrrrrr rrrrrr s@     "r) r*Zlangchain_core.callbacksrrZlangchain_core.documentsrZlangchain_core.retrieversrrrrrrs