Update rag_langchain.py

rag_langchain.py

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