@'h dZddlmZddlmZddlmZmZmZddl m Z ddl m Z ddl mZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZddlmZmZm Z ddl!m"Z#GddeZ$e#ddddf d%dZ%e jDe jLddf d&dZ'ejPejRejLdddddddf d'dZ*ejVejXejLddddf d(dZ-e d d!d"# d) d*d$Z.y)+z,Load question answering with sources chains.) annotations)Mapping)AnyOptionalProtocol) deprecated)BaseLanguageModel)BasePromptTemplate)BaseCombineDocumentsChain)MapReduceDocumentsChain)MapRerankDocumentsChain)ReduceDocumentsChain)RefineDocumentsChain)StuffDocumentsChain)LLMChain)map_reduce_promptrefine_prompts stuff_prompt)PROMPTc$eZdZdZ ddZy)LoadingCallablez2Interface for loading the combine documents chain.c y)z-Callable to load the combine documents chain.N)selfllmkwargss o/home/kushmeetdev/Regenta/Chatbot/venv/lib/python3.12/site-packages/langchain/chains/qa_with_sources/loading.py__call__zLoadingCallable.__call__ sN)rr rrreturnr )__name__ __module__ __qualname____doc__rrrrrrs"<<$<03< "t|||}td||||d|S)Nrpromptverbose) llm_chainrank_key answer_keydocument_variable_namer)rr )rr*r+r/r-r.rr,s r_load_map_rerank_chainr0&s:SAI " 5     r summariesNc >t|||}td||||d|S)Nr)r,r/document_promptr+r)rr)rr*r4r/r+rr,s r_load_stuff_chainr59s:SAI  5'     ri c  t||| } |xs|} t| || }t|||| }|d}|+td|xs|}tt||| ||}t||| | }t d| ||| d| S)Nr)r3zfcollapse_llm provided, but collapse_prompt was not: please provide one or stop providing collapse_llm.)r,r/r4)combine_documents_chaincollapse_documents_chain token_maxr+)r,reduce_documents_chainr/r+r)rr ValueErrorrr )rquestion_promptcombine_promptr4combine_document_variable_name!map_reduce_document_variable_namecollapse_prompt reduce_llm collapse_llmr+r9r map_chain _reduce_llm reduce_chainr7collapse_chain _collapse_llmr:s r_load_map_reduce_chainrHKsS'JI#K NGTL1='    #>  %+ ,!& $B+ 2 7!/  # 5@     r context_strexisting_answerc jt|||} |xs|} t| ||} td| | ||||d|S)Nr))initial_llm_chainrefine_llm_chainr/initial_response_namer4r+r)rr) rr< refine_promptr4r/rN refine_llmr+r initial_chain _refine_llm refine_chains r_load_refine_chainrTsX_gNM#K M7SL  '%53'    rz0.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 )sinceremovalmessagec ttttd}||vrt d|d|j ||}||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. )stuff map_reducerefine map_rerankzGot unsupported chain type: z. Should be one of r+)r5rHrTr0r;keys)r chain_typer+rloader_mapping_funcs rload_qa_with_sources_chainraspD#,$, 5N '*:,7 . 3 3 56 8  ,J7E  0g 0 00r)rr r*r r+boolr/strr-rcr.rcrrr r )rr r*r r4r r/rcr+Optional[bool]rrr r)rr r<r r=r r4r r>rcr?rcr@zOptional[BasePromptTemplate]rAOptional[BaseLanguageModel]rBrer+rdr9intrrr r )rr r<r rOr r4r r/rcrNrcrPrer+rdrrr r)rYN) rr r^rcr+rdrrr r )/r$ __future__rcollections.abcrtypingrrrlangchain_core._apirlangchain_core.language_modelsr langchain_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.llmr langchain.chains.qa_with_sourcesrrr5langchain.chains.question_answering.map_rerank_promptrMAP_RERANK_PROMPTrr0EXAMPLE_PROMPTr5QUESTION_PROMPTCOMBINE_PROMPTrHDEFAULT_TEXT_QA_PROMPTDEFAULT_REFINE_PROMPTrTrarrrr|s2"#***<5MQQJJH)