a ~g@sdZddlmZddlmZmZmZmZddlm Z ddl m Z ddl m Z ddlmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZmZmZddl m!Z"GdddeZ#e"ddddfdddddddddddZ$ej!ej%dd fddddd!dd"d#d$d%Z&ej'ej(ej%ddd d d d d&f ddddddd'd(d(d!d)dd*d+ d,d-Z)ej*ej+ej%d.d/d d fddddddd(d!dd0d1 d2d3Z,e d4d5d6d7d=ddd!dd9d:d;d<Z-d S)>z,Load question answering with sources chains.) annotations)AnyMappingOptionalProtocol) deprecated)BaseLanguageModel)BasePromptTemplate)BaseCombineDocumentsChain)MapReduceDocumentsChain)MapRerankDocumentsChain)ReduceDocumentsChain)RefineDocumentsChain)StuffDocumentsChain)LLMChain)map_reduce_promptrefine_prompts stuff_prompt)PROMPTc@s"eZdZdZddddddZdS) LoadingCallablez2Interface for loading the combine documents chain.rrr )llmkwargsreturncKsdS)z-Callable to load the combine documents chain.N)selfrrrrm/var/www/html/emsaiapi.evdpl.com/venv/lib/python3.9/site-packages/langchain/chains/qa_with_sources/loading.py__call__szLoadingCallable.__call__N)__name__ __module__ __qualname____doc__rrrrrrsrFcontextscoreanswerrr boolstrrr )rpromptverbosedocument_variable_namerank_key answer_keyrrcKs&t|||d}tf||||d|S)Nrr&r') llm_chainr)r*r()rr )rr&r'r(r)r*rr,rrr_load_map_rerank_chain%s r-Z summariesNzOptional[bool]r)rr&document_promptr(r'rrcKs&t|||d}tf||||d|S)Nr+r,r(r.r')rr)rr&r.r(r'rr,rrr_load_stuff_chain8sr0i zOptional[BasePromptTemplate]zOptional[BaseLanguageModel]intr ) rquestion_promptcombine_promptr.combine_document_variable_name!map_reduce_document_variable_namecollapse_prompt reduce_llm collapse_llmr' token_maxrrc Kst||| d} |p|} t| || d}t|||| d}|durRd}|durrtdn |pX|}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.)combine_documents_chaincollapse_documents_chainr9r')r,reduce_documents_chainr(r')rr ValueErrorr r )rr2r3r.r4r5r6r7r8r'r9rZ map_chainZ _reduce_llm reduce_chainr:collapse_chainZ _collapse_llmr<rrr_load_map_reduce_chainJsN r@Z context_strZexisting_answerr) rr2 refine_promptr.r(initial_response_name refine_llmr'rrc Ks@t|||d} |p|} t| ||d} tf| | ||||d|S)Nr+)initial_llm_chainrefine_llm_chainr(rBr.r')rr) rr2rAr.r(rBrCr'rZ initial_chainZ _refine_llmZ refine_chainrrr_load_refine_chains rFz0.2.13z1.0aDThis function is deprecated. Refer to this guide on retrieval and question answering with sources: https://python.langchain.com/docs/how_to/qa_sources/ See also 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 )sinceremovalmessagestuffr )r chain_typer'rrcKsJttttd}||vr.td|d|||}||fd|i|S)aLoad a question answering with sources 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", "refine" and "map_rerank". verbose: Whether chains should be run in verbose mode or not. Note that this applies to all chains that make up the final chain. Returns: A chain to use for question answering with sources. )rJ map_reducerefine map_rerankzGot unsupported chain type: z. Should be one of r')r0r@rFr-r=keys)rrKr'rZloader_mapping_funcrrrload_qa_with_sources_chains"rQ)rJN).r __future__rtypingrrrrlangchain_core._apirlangchain_core.language_modelsrlangchain_core.promptsr 'langchain.chains.combine_documents.baser -langchain.chains.combine_documents.map_reducer -langchain.chains.combine_documents.map_rerankr )langchain.chains.combine_documents.reducer )langchain.chains.combine_documents.refiner(langchain.chains.combine_documents.stuffrlangchain.chains.llmrZ langchain.chains.qa_with_sourcesrrrZ5langchain.chains.question_answering.map_rerank_promptrZMAP_RERANK_PROMPTrr-ZEXAMPLE_PROMPTr0ZQUESTION_PROMPTZCOMBINE_PROMPTr@ZDEFAULT_TEXT_QA_PROMPTZDEFAULT_REFINE_PROMPTrFrQrrrrsj             (:"