bstraehle commited on
Commit
f70e2b2
·
1 Parent(s): 431e4ec

Update rag_langchain.py

Browse files
Files changed (1) hide show
  1. rag_langchain.py +2 -2
rag_langchain.py CHANGED
@@ -23,10 +23,10 @@ class LangChainRAG(BaseRAG):
23
  YOUTUBE_DIR = "/data/yt"
24
 
25
  LLM_CHAIN_PROMPT = PromptTemplate(
26
- input_variables = ["question"],
27
  template = os.environ["LLM_TEMPLATE"])
28
  RAG_CHAIN_PROMPT = PromptTemplate(
29
- input_variables = ["context", "question"],
30
  template = os.environ["RAG_TEMPLATE"])
31
 
32
  def load_documents(self):
 
23
  YOUTUBE_DIR = "/data/yt"
24
 
25
  LLM_CHAIN_PROMPT = PromptTemplate(
26
+ input_variables = ["query_str"],
27
  template = os.environ["LLM_TEMPLATE"])
28
  RAG_CHAIN_PROMPT = PromptTemplate(
29
+ input_variables = ["context_str", "query_str"],
30
  template = os.environ["RAG_TEMPLATE"])
31
 
32
  def load_documents(self):