a ~g%@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)selfrrrrn/var/www/html/emsaiapi.evdpl.com/venv/lib/python3.9/site-packages/langchain/chains/question_answering/chain.py__call__szLoadingCallable.__call__N)__name__ __module__ __qualname____doc__r rr rrrrrrsrFcontextscoreanswerN) 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" r1 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_chaincollapse_documents_chainr;r&)r-r'reduce_documents_chainr&r*r+) rQUESTION_PROMPT_SELECTORr0ZCOMBINE_PROMPT_SELECTORrr ValueErrorr r )rr4r5r6r7r8r9r:r&r*r+r;r_question_promptZ_combine_prompt map_chain _reduce_llm reduce_chainr<collapse_chain _collapse_llmr>rrr_load_map_reduce_chain]sz rG context_strexisting_answer) rr4 refine_promptr'initial_response_name refine_llmr&r*r+rrc  Ksj|ptj|} |ptj|} t|| |||d} |p8|} t| | |||d}tf| ||||||d| S)Nr,)initial_llm_chainrefine_llm_chainr'rKr&r*r+)rr?r0ZREFINE_PROMPT_SELECTORrr)rr4rJr'rKrLr&r*r+rrAZ_refine_prompt initial_chain _refine_llm refine_chainrrr_load_refine_chains>  rRz0.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)sinceremovalmessagestuff)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. )rV map_reducerefine map_rerankzGot unsupported chain type: z. Should be one of )r&r*)r1rGrRr.r@keys)rrWr&r*rloader_mappingrrr load_qa_chains$#r])Nr"NNN) NNr2r"NNNNNNr3)NNrHrINNNN)rVNN)-r!typingrrrrlangchain_core._apirlangchain_core.callbacksrrlangchain_core.language_modelsr langchain_core.promptsr langchain.chainsr 'langchain.chains.combine_documents.baser -langchain.chains.combine_documents.map_reducer -langchain.chains.combine_documents.map_rerankr)langchain.chains.combine_documents.refiner(langchain.chains.combine_documents.stuffrlangchain.chains.llmrZ#langchain.chains.question_answeringrrr5langchain.chains.question_answering.map_rerank_promptrMAP_RERANK_PROMPTrboolstrr.r1intrGrRr]rrrrs               T -