a ag%@sdZddlmZmZmZmZddlmZddlm Z m Z ddl m Z ddl mZddlmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZmZm Z ddl!m"Z#GdddeZ$e#ddddddfe ee%e&e&e&ee e eed ddZ'd/e eee&ee%ee e eedddZ(d0e eeeee&e&eeee ee ee%ee e e)eedd d!Z*d1e eeeee&e&ee ee%ee e eed$ d%d&Z+ed'd(d)d*d2e e&ee%ee eed,d-d.Z,dS)3zLoad question answering chains.)AnyMappingOptionalProtocol) deprecated)BaseCallbackManager Callbacks)BaseLanguageModel)BasePromptTemplate)ReduceDocumentsChain)BaseCombineDocumentsChain)MapReduceDocumentsChain)MapRerankDocumentsChain)RefineDocumentsChain)StuffDocumentsChain)LLMChain)map_reduce_promptrefine_prompts stuff_prompt)PROMPTc@s"eZdZdZeeedddZdS)LoadingCallablez2Interface for loading the combine documents chain.)llmkwargsreturncKsdS)z-Callable to load the combine documents chain.N)selfrrrrw/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain/chains/question_answering/chain.py__call__szLoadingCallable.__call__N)__name__ __module__ __qualname____doc__r rr rrrrrrsrFcontextZscoreZanswerN) rpromptverbosedocument_variable_namerank_key answer_keycallback_manager callbacksrrc Ks.t|||||d} tf| |||||d|S)Nrr#r$r(r)) llm_chainr&r'r%r$r()rr) rr#r$r%r&r'r(r)rr+rrr_load_map_rerank_chain$s" r,)rr#r%r$r(r)rrc Ks<|ptj|}t|||||d}tf|||||d|S)Nr*r+r%r$r(r))rZPROMPT_SELECTOR get_promptrr) rr#r%r$r(r)rZ_promptr+rrr_load_stuff_chainAs" r/ summaries )rquestion_promptcombine_promptcombine_document_variable_name!map_reduce_document_variable_namecollapse_prompt reduce_llm collapse_llmr$r(r) token_maxrrc  Ks|ptj|} |ptj|}t|| || | d}|p8|}t|||| | d}t|||| | d}|dur|d}|durtdn&|p|}tt|||| | d||| d}t||| |d}tf||||| | d| S)Nr*r-zfcollapse_llm provided, but collapse_prompt was not: please provide one or stop providing collapse_llm.)r+r%r$r()combine_documents_chainZcollapse_documents_chainr9r$)r+r%reduce_documents_chainr$r(r)) rQUESTION_PROMPT_SELECTORr.ZCOMBINE_PROMPT_SELECTORrr ValueErrorr r )rr2r3r4r5r6r7r8r$r(r)r9r_question_promptZ_combine_promptZ map_chainZ _reduce_llmZ reduce_chainr:Zcollapse_chainZ _collapse_llmr;rrr_load_map_reduce_chain]sz r? context_strexisting_answer) rr2 refine_promptr%initial_response_name refine_llmr$r(r)rrc  Ksj|ptj|} |ptj|} t|| |||d} |p8|} t| | |||d}tf| ||||||d| S)Nr*)Zinitial_llm_chainZrefine_llm_chainr%rCr$r(r))rr<r.ZREFINE_PROMPT_SELECTORrr)rr2rBr%rCrDr$r(r)rr>Z_refine_promptZ initial_chainZ _refine_llmZ refine_chainrrr_load_refine_chains>  rEz0.2.13z1.0a1This class is deprecated. See the following migration guides for replacements based on `chain_type`: stuff: https://python.langchain.com/docs/versions/migrating_chains/stuff_docs_chain map_reduce: https://python.langchain.com/docs/versions/migrating_chains/map_reduce_chain refine: https://python.langchain.com/docs/versions/migrating_chains/refine_chain map_rerank: https://python.langchain.com/docs/versions/migrating_chains/map_rerank_docs_chain See also guides on retrieval and question-answering here: https://python.langchain.com/docs/how_to/#qa-with-rag)ZsinceZremovalmessagestuff)r chain_typer$r(rrcKsHttttd}||vr.td|d||||f||d|S)aLoad question answering chain. Args: llm: Language Model to use in the chain. chain_type: Type of document combining chain to use. Should be one of "stuff", "map_reduce", "map_rerank", and "refine". verbose: Whether chains should be run in verbose mode or not. Note that this applies to all chains that make up the final chain. callback_manager: Callback manager to use for the chain. Returns: A chain to use for question answering. )rGZ map_reduceZrefineZ map_rerankzGot unsupported chain type: z. Should be one of )r$r()r/r?rEr,r=keys)rrHr$r(rZloader_mappingrrr load_qa_chains$#rJ)Nr"NNN) NNr0r"NNNNNNr1)NNr@rANNNN)rGNN)-r!typingrrrrZlangchain_core._apirZlangchain_core.callbacksrrZlangchain_core.language_modelsr Zlangchain_core.promptsr Zlangchain.chainsr Z'langchain.chains.combine_documents.baser Z-langchain.chains.combine_documents.map_reducer Z-langchain.chains.combine_documents.map_rerankrZ)langchain.chains.combine_documents.refinerZ(langchain.chains.combine_documents.stuffrZlangchain.chains.llmrZ#langchain.chains.question_answeringrrrZ5langchain.chains.question_answering.map_rerank_promptrZMAP_RERANK_PROMPTrboolstrr,r/intr?rErJrrrrs               T -