github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-07-04 20:28:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

make prompt a variable in vector db qa (#170)

Browse Source
This commit is contained in:
Harrison Chase 2022-11-21 19:30:40 -08:00 committed by GitHub
parent 4a4dfbfbed
commit 22bd12a097
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
langchain/chains/vector_db_qa/base.py
View File

@ -5,8 +5,9 @@ from pydantic import BaseModel, Extra
from langchain.chains.base import Chain from langchain.chains.base import Chain
from langchain.chains.llm import LLMChain from langchain.chains.llm import LLMChain
from langchain.chains.vector_db_qa.prompt import prompt from langchain.chains.vector_db_qa.prompt import PROMPT
from langchain.llms.base import LLM from langchain.llms.base import LLM
from langchain.prompts import PromptTemplate
from langchain.vectorstores.base import VectorStore from langchain.vectorstores.base import VectorStore
@ -29,6 +30,8 @@ class VectorDBQA(Chain, BaseModel):
"""Vector Database to connect to.""" """Vector Database to connect to."""
k: int = 4 k: int = 4
"""Number of documents to query for.""" """Number of documents to query for."""
prompt: PromptTemplate = PROMPT
"""Prompt to use when questioning the documents."""
input_key: str = "query" #: :meta private: input_key: str = "query" #: :meta private:
output_key: str = "result" #: :meta private: output_key: str = "result" #: :meta private:
@ -56,7 +59,7 @@ class VectorDBQA(Chain, BaseModel):
def _call(self, inputs: Dict[str, str]) -> Dict[str, str]: def _call(self, inputs: Dict[str, str]) -> Dict[str, str]:
question = inputs[self.input_key] question = inputs[self.input_key]
llm_chain = LLMChain(llm=self.llm, prompt=prompt) llm_chain = LLMChain(llm=self.llm, prompt=self.prompt)
docs = self.vectorstore.similarity_search(question, k=self.k) docs = self.vectorstore.similarity_search(question, k=self.k)
contexts = [] contexts = []
for j, doc in enumerate(docs): for j, doc in enumerate(docs):

2
langchain/chains/vector_db_qa/prompt.py
View File

@ -7,6 +7,6 @@ prompt_template = """Use the following pieces of context to answer the question
Question: {question} Question: {question}
Helpful Answer:""" Helpful Answer:"""
prompt = PromptTemplate( PROMPT = PromptTemplate(
template=prompt_template, input_variables=["context", "question"] template=prompt_template, input_variables=["context", "question"]
) )