Spaces:
Build error
Build error
Update rag_langchain.py
Browse files- 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 = ["
|
27 |
template = os.environ["LLM_TEMPLATE"])
|
28 |
RAG_CHAIN_PROMPT = PromptTemplate(
|
29 |
-
input_variables = ["
|
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):
|