Update rag_langchain.py

rag_langchain.py

@@ -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