o `+ i.@sJdZddlmZddlZddlmZddlmZmZddl m Z ddl m Z m Z mZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZmZmZddlmZddlm Z ddl!m"Z"ddl#m$Z$ddl%m&Z&ddl'm(Z(e ddddGdddeZ)e ddddGddde)Z*e ddddGddde)Z+dS)7Chain for question-answering against a vector database.) annotationsN)abstractmethod)AnyOptional) 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: )rselfr)n/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/langchain/chains/retrieval_qa/base.py input_keys7szBaseRetrievalQA.input_keyscCs|jg}|jr g|d}|S)z-Output keys. :meta private: source_documents)r!r#)r(Z _output_keysr)r)r* output_keys?s zBaseRetrievalQA.output_keysNllmr promptOptional[PromptTemplate] callbacksr llm_chain_kwargsOptional[dict]kwargsrc KsZ|pt|}td |||d|pi}tdgdd}t|d||d} |d | |d|S) zInitialize from LLM.)r.r/r1Z page_contentzContext: {page_content})Zinput_variablestemplatecontext) llm_chainZdocument_variable_namedocument_promptr1)rr1Nr))rZ get_promptrr r) clsr.r/r1r2r4Z_promptr7r8rr)r)r*from_llmJs0 zBaseRetrievalQA.from_llmstuff chain_typechain_type_kwargscKs.|pi}t|fd|i|}|dd|i|S)zLoad chain from chain type.r<rNr)r)r9r.r<r=r4Z_chain_type_kwargsrr)r)r*from_chain_typels zBaseRetrievalQA.from_chain_typequestion run_managerr list[Document]cCdSz,Get documents to do question answering over.Nr)r(r?r@r)r)r* _get_docs}szBaseRetrievalQA._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'] r@r@Zinput_documentsr?r1r,) r get_noop_managerrinspect signaturerE parametersrrun get_childr#r!r(rFr@Z _run_managerr?Zaccepts_run_managerdocsZanswerr)r)r*_calls    zBaseRetrievalQA._callrcsdSrCr)rDr)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)rIr@rJNrKr,) rrLrrMrNrUrOrZarunrQr#r!rRr)r)r*_acalls     zBaseRetrievalQA._acall)r%r&)NNN) r.r r/r0r1r r2r3r4rr%r)r;N) r.r r<rr=r3r4rr%rr?rr@r r%rA)N)rFrGr@rHr%rGr?rr@rr%rA)rFrGr@rVr%rG)__name__ __module__ __qualname____doc____annotations__rr!r#rZ model_configpropertyr+r- classmethodr:r>rrErTrUrWr)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 retrieverr?rr@r r%rAcCs|jj|d|idS) Get docs.r1config)rcZinvokerQrDr)r)r*rEs zRetrievalQA._get_docsrcs |jj|d|idIdHS)rdr1reN)rcZainvokerQrDr)r)r*rUs   zRetrievalQA._aget_docscCrB)Return the chain type.Z retrieval_qar)r'r)r)r* _chain_type'zRetrievalQA._chain_typeNrXrYr%r) rZr[r\r]rrcr^rErUr_rhr)r)r)r*ras  -   rac@seZdZUdZedddZded< dZded< d Zd ed < ee d Z d ed< e dde d#ddZ d$ddZd%ddZed&d d!Zd"S)' VectorDBQArT vectorstore)rbaliasrintk similarityr search_type)default_factoryrG search_kwargsbefore)modevaluesdictr%rcCs0d|vr|d}|dvrd|d}t||S)zValidate search type.rr)rqmmrsearch_type of not allowed.) ValueError)r9rwrrmsgr)r)r*validate_search_typeBs  zVectorDBQA.validate_search_typer?r@r rAcCsj|jdkr|jj|fd|ji|j}|S|jdkr*|jj|fd|ji|j}|Sd|jd}t|)rdrqrpryrzr{)rrrlZsimilarity_searchrprtZmax_marginal_relevance_searchr|)r(r?r@rSr}r)r)r*rEMs(  zVectorDBQA._get_docsrcsd}t|)rdz!VectorDBQA does not support async)NotImplementedError)r(r?r@r}r)r)r*rUeszVectorDBQA._aget_docscCrB)rgZ vector_db_qar)r'r)r)r*rhorizVectorDBQA._chain_typeN)rwrxr%rrXrYrj)rZr[r\r]rrlr^rprrrxrtrr`r~rErUr_rhr)r)r)r*rk-s"      rk),r] __future__rrMabcrtypingrrZlangchain_core._apirZlangchain_core.callbacksrr r Zlangchain_core.documentsr Zlangchain_core.language_modelsr Zlangchain_core.promptsr Zlangchain_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_promptrrrarkr)r)r)r*sL               0 N