3fi,JddlZddlmZmZddlmZddlmZGddeZy)N)#AsyncCallbackManagerForRetrieverRunCallbackManagerForRetrieverRun)Document) BaseRetrieverceZdZUdZeeed< dededee fdZ dede dee fdZ dededee fdZ dede dee fd Zy ) MergerRetrieverz9Retriever that merges the results of multiple retrievers. retrieversquery run_managerreturnc&|j||S)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 relevant documents. )merge_documentsselfr r s k/var/www/auto_recruiter/arenv/lib/python3.12/site-packages/langchain_classic/retrievers/merger_retriever.py_get_relevant_documentsz'MergerRetriever._get_relevant_documentss ##E;77cBK|j||d{S7w)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 relevant documents. N)amerge_documentsrs r_aget_relevant_documentsz(MergerRetriever._aget_relevant_documents#s! **5+>>>>s c t|jDcgc]/\}}|j|d|jd|dzi1}}}g}t t t |d}t|D]E}t|j|dD](\}} |t | ks|j| |*G|Scc}}w) zMerge the results of the retrievers. Args: query: The query to search for. run_manager: The callback handler to use. Returns: A list of merged documents. callbacks retriever_configrdefaultFstrict) enumerater invoke get_childmaxmaplenrangezipappend) rr r i retrieverretriever_docsmerged_documentsmax_docs _retrieverdocs rrzMergerRetriever.merge_documents5s(!*$// :  9   #[%:%:ZAw;O%PQ    s3/;x 4A#&tu#U 4 Cs3x<$++CF3 4 4  ! s4CcdKtjfdt|jDd{}g}t t t |d}t|D]E}t|j|dD](\}}|t |ks|j||*G|S7vw)zAsynchronously merge the results of the retrievers. Args: query: The query to search for. run_manager: The callback handler to use. Returns: A list of merged documents. c 3xK|]1\}}|jdjd|dzi3yw)rrrrN)ainvoker#).0r*r+r r s r z3MergerRetriever.amerge_documents..fsQ !Ay !!')>)>APQE7?S)TU"s7:NrrFr) asynciogatherr!r r$r%r&r'r(r)) rr r r,r-r.r*r/r0s `` rrz MergerRetriever.amerge_documentsVs '~~ %.doo$>   s3/;x 4A#&tu#U 4 Cs3x<$++CF3 4 4  % s3B0B.AB0B0N)__name__ __module__ __qualname____doc__listr__annotations__strrrrrrrrrrrr sC]##(884 8 h 8$??9 ? h ?$  4  h  B! ! 9!  h ! rr) r6langchain_core.callbacksrrlangchain_core.documentsrlangchain_core.retrieversrrr?rrrCs#.3l ml r