Update rag_langchain.py
Browse files- 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["
|
28 |
RAG_CHAIN_PROMPT = PromptTemplate(
|
29 |
input_variables = ["context", "question"],
|
30 |
-
template = os.environ["
|
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 = []
|