o Zh.@sZdZddlmZddlZddlZddlmZddlmZm Z m Z m Z ddl m Z ddlmZmZmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZmZmZddl m!Z!ddl"m#Z#ddl$m%Z%ddl&m'Z'ddl(m)Z)ddl*m+Z+e ddddGddde!Z,e ddddGddde,Z-e ddddGddde,Z.dS)7Chain for question-answering against a vector database.) annotationsN)abstractmethod)AnyDictListOptional) deprecated)AsyncCallbackManagerForChainRunCallbackManagerForChainRun Callbacks)Document)BaseLanguageModel)PromptTemplate) BaseRetriever) VectorStore) ConfigDictFieldmodel_validator)Chain)BaseCombineDocumentsChain)StuffDocumentsChain)LLMChain load_qa_chain)PROMPT_SELECTORz0.2.13z1.0zThis class is deprecated. Use the `create_retrieval_chain` constructor instead. See migration guide here: https://python.langchain.com/docs/versions/migrating_chains/retrieval_qa/)ZsinceZremovalmessagec@seZdZUdZded< dZded<dZded<d Zd ed < ed d d dZ e d8ddZ e d8ddZ e   d9d:d d!Ze " d;dd?d0d1Zed@d3d4Z d>dAd6d7ZdS)BBaseRetrievalQAz)Base class for question-answering chains.rcombine_documents_chainquerystr input_keyresult output_keyFboolreturn_source_documentsTZforbid)Zpopulate_by_nameZarbitrary_types_allowedextrareturn List[str]cCs|jgS)z,Input keys. :meta private: )r!selfr+Y/var/www/html/lang_env/lib/python3.10/site-packages/langchain/chains/retrieval_qa/base.py input_keys8szBaseRetrievalQA.input_keyscCs|jg}|jr |dg}|S)z-Output keys. :meta private: source_documents)r#r%)r*Z _output_keysr+r+r, output_keys@s zBaseRetrievalQA.output_keysNllmrpromptOptional[PromptTemplate] callbacksr llm_chain_kwargsOptional[dict]kwargsrc KsZ|pt|}td |||d|pi}tdgdd}t|d||d} |d | |d|S) zInitialize from LLM.)r0r1r3Z page_contentzContext: {page_content})Zinput_variablestemplatecontext) llm_chainZdocument_variable_namedocument_promptr3)rr3Nr+)rZ get_promptrrr) clsr0r1r3r4r6Z_promptr9r:rr+r+r,from_llmKs* zBaseRetrievalQA.from_llmstuff chain_typechain_type_kwargscKs.|pi}t|fd|i|}|dd|i|S)zLoad chain from chain type.r>rNr+r)r;r0r>r?r6Z_chain_type_kwargsrr+r+r,from_chain_typeis zBaseRetrievalQA.from_chain_typequestion run_managerr List[Document]cCdSz,Get documents to do question answering over.Nr+r*rArBr+r+r, _get_docsxszBaseRetrievalQA._get_docsinputsDict[str, Any]$Optional[CallbackManagerForChainRun]cCsz|pt}||j}dt|jjv}|r|j||d}n||}|jj||| d}|j r8|j |d|iS|j |iS)hRun get_relevant_text and llm on input query. If chain has 'return_source_documents' as 'True', returns the retrieved documents as well under the key 'source_documents'. Example: .. code-block:: python res = indexqa({'query': 'This is my query'}) answer, docs = res['result'], res['source_documents'] rBrBZinput_documentsrAr3r.) r get_noop_managerr!inspect signaturerG parametersrrun get_childr%r#r*rHrBZ _run_managerrAZaccepts_run_managerdocsZanswerr+r+r,_calls     zBaseRetrievalQA._callr csdSrEr+rFr+r+r, _aget_docsszBaseRetrievalQA._aget_docs)Optional[AsyncCallbackManagerForChainRun]cs|pt}||j}dt|jjv}|r"|j||dIdH}n||IdH}|jj||| dIdH}|j rB|j |d|iS|j |iS)rKrBrLNrMr.) r rNr!rOrPrWrQrZarunrSr%r#rTr+r+r,_acalls     zBaseRetrievalQA._acall)r'r()NNN) r0rr1r2r3r r4r5r6rr'r)r=N) r0rr>r r?r5r6rr'rrAr rBr r'rC)N)rHrIrBrJr'rIrAr rBr r'rC)rHrIrBrXr'rI)__name__ __module__ __qualname____doc____annotations__r!r#r%rZ model_configpropertyr-r/ classmethodr<r@rrGrVrWrYr+r+r+r,rsD          "  rz0.1.17c@sFeZdZUdZeddZded<dd d ZdddZe dddZ dS) RetrievalQAaChain for question-answering against an index. This class is deprecated. See below for an example implementation using `create_retrieval_chain`: .. code-block:: python from langchain.chains import create_retrieval_chain from langchain.chains.combine_documents import create_stuff_documents_chain from langchain_core.prompts import ChatPromptTemplate from langchain_openai import ChatOpenAI retriever = ... # Your retriever llm = ChatOpenAI() system_prompt = ( "Use the given context to answer the question. " "If you don't know the answer, say you don't know. " "Use three sentence maximum and keep the answer concise. " "Context: {context}" ) prompt = ChatPromptTemplate.from_messages( [ ("system", system_prompt), ("human", "{input}"), ] ) question_answer_chain = create_stuff_documents_chain(llm, prompt) chain = create_retrieval_chain(retriever, question_answer_chain) chain.invoke({"input": query}) Example: .. code-block:: python from langchain_community.llms import OpenAI from langchain.chains import RetrievalQA from langchain_community.vectorstores import FAISS from langchain_core.vectorstores import VectorStoreRetriever retriever = VectorStoreRetriever(vectorstore=FAISS(...)) retrievalQA = RetrievalQA.from_llm(llm=OpenAI(), retriever=retriever) T)excluder retrieverrAr rBr r'rCcCs|jj|d|idS) Get docs.r3config)reZinvokerSrFr+r+r,rGs zRetrievalQA._get_docsr cs |jj|d|idIdHS)rfr3rgN)reZainvokerSrFr+r+r,rWs  zRetrievalQA._aget_docscCrD)Return the chain type.Z retrieval_qar+r)r+r+r, _chain_typezRetrievalQA._chain_typeNrZr[r'r ) r\r]r^r_rrer`rGrWrarjr+r+r+r,rcs  -   rcc@seZdZUdZedddZded< dZded< d Zd ed < ee d Z d ed< e dde d%ddZ e dde d%ddZd&ddZd'd d!Zed(d"d#Zd$S)) VectorDBQArT vectorstore)rdaliasrintk similarityr search_type)default_factoryrI search_kwargsbefore)modevaluesrr'rcCstd|S)NzR`VectorDBQA` is deprecated - please use `from langchain.chains import RetrievalQA`)warningswarn)r;ryr+r+r,raise_deprecation9szVectorDBQA.raise_deprecationcCs,d|vr|d}|dvrtd|d|S)zValidate search type.rt)rsmmrsearch_type of not allowed.) ValueError)r;ryrtr+r+r,validate_search_typeBs zVectorDBQA.validate_search_typerArBr rCcCsf|jdkr|jj|fd|ji|j}|S|jdkr*|jj|fd|ji|j}|Std|jd)rfrsrrr}r~r)rtrnZsimilarity_searchrrrvZmax_marginal_relevance_searchr)r*rArBrUr+r+r,rGLs&  zVectorDBQA._get_docsr cs td)rfz!VectorDBQA does not support async)NotImplementedErrorrFr+r+r,rW_szVectorDBQA._aget_docscCrD)riZ vector_db_qar+r)r+r+r,rjhrkzVectorDBQA._chain_typeN)ryrr'rrZr[rl)r\r]r^r_rrnr`rrrtdictrvrrbr|rrGrWrarjr+r+r+r,rm$s(       rm)/r_ __future__rrOrzabcrtypingrrrrZlangchain_core._apir Zlangchain_core.callbacksr r r Zlangchain_core.documentsr Zlangchain_core.language_modelsrZlangchain_core.promptsrZlangchain_core.retrieversrZlangchain_core.vectorstoresrZpydanticrrrZlangchain.chains.baserZ'langchain.chains.combine_documents.baserZ(langchain.chains.combine_documents.stuffrZlangchain.chains.llmrZ#langchain.chains.question_answeringrZ0langchain.chains.question_answering.stuff_promptrrrcrmr+r+r+r,sN               ( L