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 = []
|