Update rag.py

rag.py

@@ -118,10 +118,11 @@ def rag_chain(config, openai_api_key, rag_option, prompt):
     rag_chain = RetrievalQA.from_chain_type(llm, 
                                             chain_type_kwargs = {"prompt": RAG_CHAIN_PROMPT}, 
                                             retriever = db.as_retriever(search_kwargs = {"k": config["k"]}), 
+                                            include_run_info = True,
                                             return_source_documents = True,
-                                            verbose = False)
+                                            verbose = True)
     
-    completion = rag_chain({"query": prompt, "include_run_info": True})
+    completion = rag_chain({"query": prompt})
     print(completion)
     
     return completion, rag_chain