Update rag.py

rag.py

@@ -116,7 +116,7 @@ def rag_chain(config, openai_api_key, rag_option, prompt):
                                             return_source_documents = True,
                                             verbose = False)
     
-    completion = rag_chain({"query": prompt})
+    completion = rag_chain({"query": prompt}, include_run_info: True)
     print("###" + completion)
     
     return completion, rag_chain