o `+ i%@sdZddlmZddlmZmZmZddlmZddl m Z m Z ddl m Z ddlmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZddlmZm Z m!Z!ddl"m#Z$GdddeZ%e$dddddddde dede&de'de'de'dee d e d!ed"efd#d$Z(dddddd%de deede'dee&dee d e d!ed"efd&d'Z)ddd(dddddddd)d* de d+eed,eed-e'd.e'd/eed0ee d1ee dee&dee d e d2e*d!ed"efd3d4Z+ddd5d6ddddd7de d+eed8eede'd9e'd:ee dee&dee d e d!ed"efd;d<Z,ed=d>d?d@ A  dEde dBe'dee&dee d!ed"ef dCdDZ-dS)FzLoad question answering chains.)Mapping)AnyOptionalProtocol) deprecated)BaseCallbackManager Callbacks)BaseLanguageModel)BasePromptTemplate)ReduceDocumentsChain)BaseCombineDocumentsChain)MapReduceDocumentsChain)MapRerankDocumentsChain)RefineDocumentsChain)StuffDocumentsChain)LLMChain)map_reduce_promptrefine_prompts stuff_prompt)PROMPTc@s&eZdZdZdededefddZdS)LoadingCallablez2Interface for loading the combine documents chain.llmkwargsreturncKsdS)z-Callable to load the combine documents chain.N)selfrrrru/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/langchain/chains/question_answering/chain.py__call__szLoadingCallable.__call__N)__name__ __module__ __qualname____doc__r rr rrrrrrsrFcontextZscoreZanswerN)promptverbosedocument_variable_namerank_key answer_keycallback_manager callbacksrr#r$r%r&r'r(r)rrc Ks.t|||||d} td| |||||d|S)Nrr#r$r(r)) llm_chainr&r'r%r$r(r)rr) rr#r$r%r&r'r(r)rr+rrr_load_map_rerank_chain's" r,)r#r%r$r(r)c Ks<|ptj|}t|||||d}td|||||d|S)Nr*r+r%r$r(r)r)rZPROMPT_SELECTOR get_promptrr) rr#r%r$r(r)rZ_promptr+rrr_load_stuff_chainEs" r/Z summariesi ) question_promptcombine_promptcombine_document_variable_name!map_reduce_document_variable_namecollapse_prompt reduce_llm collapse_llmr$r(r) token_maxr0r1r2r3r4r5r6r7c  Ks|ptj|} |ptj|}t|| || | d}|p|}t|||| | d}t|||| | d}|dur@d}|dur?d}t|n|pC|}tt|||| | d||| d}t||| |d}td||||| | 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_chainr7r$)r+r%reduce_documents_chainr$r(r)r) rQUESTION_PROMPT_SELECTORr.ZCOMBINE_PROMPT_SELECTORrr ValueErrorr r )rr0r1r2r3r4r5r6r$r(r)r7r_question_promptZ_combine_promptZ map_chainZ _reduce_llmZ reduce_chainr8Zcollapse_chainmsgZ _collapse_llmr9rrr_load_map_reduce_chainbs| r>Z context_strZexisting_answer)r0 refine_promptr%initial_response_name refine_llmr$r(r)r?r@rAc Ksj|ptj|} |ptj|} t|| |||d} |p|} t| | |||d}td| ||||||d| S)Nr*)Zinitial_llm_chainZrefine_llm_chainr%r@r$r(r)r)rr:r.ZREFINE_PROMPT_SELECTORrr)rr0r?r%r@rAr$r(r)rr<Z_refine_promptZ initial_chainZ _refine_llmZ refine_chainrrr_load_refine_chains> rBz0.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 chain_typecKsLttttd}||vrd|d|}t||||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. )rDZ map_reduceZrefineZ map_rerankzGot unsupported chain type: z. Should be one of )r$r()r/r>rBr,keysr;)rrEr$r(rZloader_mappingr=rrr load_qa_chains&#rG)rDNN).r!collections.abcrtypingrrrZlangchain_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>rBrGrrrrs>               !        W    .