Update rag.py

rag.py

@@ -115,8 +115,7 @@ 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}, include_run_info = True)
+    completion = rag_chain({"query": prompt})
     print("###" + str(completion))
-    print("###" + str(completion["run_info"]))
     
     return completion, rag_chain