Update rag_langchain.py

rag_langchain.py

@@ -23,11 +23,11 @@ class LangChainRAG(BaseRAG):
     YOUTUBE_DIR = "/data/yt"
 
     LLM_CHAIN_PROMPT = PromptTemplate(
-        input_variables = ["query_str"], 
+        input_variables = ["question"], 
         template = os.environ["LLM_TEMPLATE"])
     RAG_CHAIN_PROMPT = PromptTemplate(
-        input_variables = ["context_str", "query_str"], 
-        template = os.environ["RAG_TEMPLATE_2"])
+        input_variables = ["context", "question"], 
+        template = os.environ["RAG_TEMPLATE"])
 
     def load_documents(self):
         docs = []
@@ -114,7 +114,7 @@ class LangChainRAG(BaseRAG):
         )
     
         with get_openai_callback() as callback:
-            completion = llm_chain.generate([{"query_str": prompt}])
+            completion = llm_chain.generate([{"question": prompt}])
     
         return completion, llm_chain, callback
 
@@ -130,6 +130,6 @@ class LangChainRAG(BaseRAG):
         )
     
         with get_openai_callback() as callback:
-            completion = rag_chain({"query_str": prompt})
+            completion = rag_chain({"query": prompt})
 
         return completion, rag_chain, callback