a ~g.@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/)sinceremovalmessagec @seZdZUdZded<dZded<dZded<d Zd ed <ed d d dZ e ddddZ e ddddZ e d2dddddddddZe d3ddddddd d!Zedd"d#d$d%d&Zd4d'd(d'd)d*d+Zedd,d#d$d-d.Zd5d'd/d'd)d0d1ZdS)6BaseRetrievalQAz)Base class for question-answering chains.rcombine_documents_chainquerystr input_keyresult output_keyFboolreturn_source_documentsTforbid)populate_by_namearbitrary_types_allowedextraz List[str]returncCs|jgS)z,Input keys. :meta private: )r#selfr0g/var/www/html/emsaiapi.evdpl.com/venv/lib/python3.9/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/ _output_keysr0r0r1 output_keys@s zBaseRetrievalQA.output_keysNrzOptional[PromptTemplate]r zOptional[dict]r)llmprompt callbacksllm_chain_kwargskwargsr-c KsZ|p t|}tf|||d|p"i}tdgdd}t|d||d} |f| |d|S)zInitialize from LLM.)r6r7r8 page_contentzContext: {page_content})input_variablestemplatecontext) llm_chaindocument_variable_namedocument_promptr8)r r8)r get_promptrrr) clsr6r7r8r9r:_promptr?rAr r0r0r1from_llmKs* zBaseRetrievalQA.from_llmstuff)r6 chain_typechain_type_kwargsr:r-cKs.|pi}t|fd|i|}|fd|i|S)zLoad chain from chain type.rGr r)rCr6rGrHr:Z_chain_type_kwargsr r0r0r1from_chain_typeis zBaseRetrievalQA.from_chain_typer List[Document]question run_managerr-cCsdSz,Get documents to do question answering over.Nr0r/rLrMr0r0r1 _get_docsxszBaseRetrievalQA._get_docsDict[str, Any]z$Optional[CallbackManagerForChainRun])inputsrMr-cCs~|p t}||j}dt|jjv}|r<|j||d}n ||}|jj||| d}|j rp|j |d|iS|j |iSdS)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'] rMrMinput_documentsrLr8r3N) r get_noop_managerr#inspect signaturerP parametersr run get_childr'r%r/rRrM _run_managerrLaccepts_run_managerdocsanswerr0r0r1_calls    zBaseRetrievalQA._callr csdSrNr0rOr0r0r1 _aget_docsszBaseRetrievalQA._aget_docsz)Optional[AsyncCallbackManagerForChainRun]cs|p t}||j}dt|jjv}|rB|j||dIdH}n||IdH}|jj||| dIdH}|j r|j |d|iS|j |iSdS)rSrMrTNrUr3) r rWr#rXrYrcrZr arunr\r'r%r]r0r0r1_acalls   zBaseRetrievalQA._acall)NNN)rFN)N)N)__name__ __module__ __qualname____doc____annotations__r#r%r'r model_configpropertyr2r5 classmethodrErIrrPrbrcrer0r0r0r1rs>      " rz0.1.17c@sZeZdZUdZeddZded<dddd d d Zdd dd d dZe ddddZ 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"r rJrKcCs|jj|d|idS) Get docs.r8config)rpinvoker\rOr0r0r1rPs zRetrievalQA._get_docsr cs|jj|d|idIdHS)rqr8rrN)rpainvoker\rOr0r0r1rcs zRetrievalQA._aget_docsr,cCsdS)Return the chain type. retrieval_qar0r.r0r0r1 _chain_typeszRetrievalQA._chain_typeN) rfrgrhrirrprjrPrcrlrxr0r0r0r1rns -  rnc@seZdZUdZedddZded<dZded<d Zd ed <ee d Z d ed<e dde dddddZ e dde dddddZd dddddZd dddddZed d d!d"Zd#S)$ VectorDBQArT vectorstore)roaliasrintk similarityr" search_type)default_factoryrQ search_kwargsbefore)moderr)valuesr-cCstd|S)NzR`VectorDBQA` is deprecated - please use `from langchain.chains import RetrievalQA`)warningswarn)rCrr0r0r1raise_deprecation9szVectorDBQA.raise_deprecationcCs,d|vr(|d}|dvr(td|d|S)zValidate search type.r)rmmrsearch_type of not allowed.) ValueError)rCrrr0r0r1validate_search_typeBs zVectorDBQA.validate_search_typer rJrKcCsf|jdkr(|jj|fd|ji|j}n:|jdkrP|jj|fd|ji|j}ntd|jd|S)rqrr~rrr)rrzsimilarity_searchr~rmax_marginal_relevance_searchr)r/rLrMr`r0r0r1rPLs$  zVectorDBQA._get_docsr cs tddS)rqz!VectorDBQA does not support asyncN)NotImplementedErrorrOr0r0r1rc_szVectorDBQA._aget_docsr,cCsdS)rv vector_db_qar0r.r0r0r1rxhszVectorDBQA._chain_typeN)rfrgrhrirrzrjr~rdictrrrmrrrPrcrlrxr0r0r0r1ry$s    ry)/ri __future__rrXrabcrtypingrrrrlangchain_core._apir langchain_core.callbacksr r r langchain_core.documentsr langchain_core.language_modelsrlangchain_core.promptsrlangchain_core.retrieversrlangchain_core.vectorstoresrpydanticrrrlangchain.chains.baser'langchain.chains.combine_documents.baser(langchain.chains.combine_documents.stuffrlangchain.chains.llmrZ#langchain.chains.question_answeringrZ0langchain.chains.question_answering.stuff_promptrrrnryr0r0r0r1sL               ( L