a ag@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)selfrrrrv/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain/chains/qa_with_sources/loading.py__call__szLoadingCallable.__call__N)__name__ __module__ __qualname____doc__rrrrrrsrFcontextZscoreZanswerrr 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_chain8sr.i 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_chainZcollapse_documents_chainr7r%)r*reduce_documents_chainr&r%)rr ValueErrorr r )rr0r1r,r2r3r4r5r6r%r7rZ map_chainZ _reduce_llmZ reduce_chainr8Zcollapse_chainZ _collapse_llmr9rrr_load_map_reduce_chainJsN r;Z context_strZexisting_answerr) rr0 refine_promptr,r&initial_response_name refine_llmr%rrc Ks@t|||d} |p|} t| ||d} tf| | ||||d|S)Nr))Zinitial_llm_chainZrefine_llm_chainr&r=r,r%)rr) rr0r<r,r&r=r>r%rZ initial_chainZ _refine_llmZ refine_chainrrr_load_refine_chains r?z0.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 )ZsinceZremovalmessagestuffr )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. )rAZ map_reduceZrefineZ map_rerankzGot unsupported chain type: z. Should be one of r%)r.r;r?r+r:keys)rrBr%rZloader_mappingZ_funcrrrload_qa_with_sources_chains"rD)rAN).r __future__rtypingrrrrZlangchain_core._apirZlangchain_core.language_modelsrZlangchain_core.promptsr Z'langchain.chains.combine_documents.baser Z-langchain.chains.combine_documents.map_reducer Z-langchain.chains.combine_documents.map_rerankr Z)langchain.chains.combine_documents.reducer Z)langchain.chains.combine_documents.refinerZ(langchain.chains.combine_documents.stuffrZlangchain.chains.llmrZ langchain.chains.qa_with_sourcesrrrZ5langchain.chains.question_answering.map_rerank_promptrZMAP_RERANK_PROMPTrr+ZEXAMPLE_PROMPTr.ZQUESTION_PROMPTZCOMBINE_PROMPTr;ZDEFAULT_TEXT_QA_PROMPTZDEFAULT_REFINE_PROMPTr?rDrrrrsj             (:"