Update rag.py

rag.py

@@ -115,7 +115,8 @@ def rag_chain(config, openai_api_key, rag_option, prompt):
                                             retriever = db.as_retriever(search_kwargs = {"k": config["k"]}), 
                                             return_source_documents = True)
     
-    completion = rag_chain({"query": prompt})
+    #completion = rag_chain({"query": prompt})
+    completion = rag_chain.generate({"query": prompt})
     print("###" + str(completion))
     
     return completion, rag_chain