bstraehle commited on
Commit
0fa9763
·
1 Parent(s): f6fce88

Update rag_langchain.py

Browse files
Files changed (1) hide show
  1. rag_langchain.py +2 -2
rag_langchain.py CHANGED
@@ -24,10 +24,10 @@ class LangChainRAG(BaseRAG):
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):
33
  docs = []
 
24
 
25
  LLM_CHAIN_PROMPT = PromptTemplate(
26
  input_variables = ["question"],
27
+ template = os.environ["TEMPLATE"])
28
  RAG_CHAIN_PROMPT = PromptTemplate(
29
  input_variables = ["context", "question"],
30
+ template = os.environ["LANGCHAIN_TEMPLATE"])
31
 
32
  def load_documents(self):
33
  docs = []