Update rag.py

rag.py

@@ -96,8 +96,7 @@ def get_llm(config, openai_api_key):
 
 def llm_chain(config, openai_api_key, prompt):
     llm_chain = LLMChain(llm = get_llm(config, openai_api_key), 
-                         prompt = LLM_CHAIN_PROMPT, 
-                         verbose = False)
+                         prompt = LLM_CHAIN_PROMPT)
     
     completion = llm_chain.generate([{"question": prompt}])
     
@@ -112,14 +111,12 @@ def rag_chain(config, openai_api_key, rag_option, prompt):
         db = document_retrieval_mongodb()
 
     rag_chain = RetrievalQA.from_chain_type(llm, 
-                                            chain_type_kwargs = {"prompt": RAG_CHAIN_PROMPT,
-                                                                 "verbose": True}, 
-                                            retriever = db.as_retriever(search_kwargs = {"k": config["k"],
-                                                                                         "verbose": True}), 
-                                            return_source_documents = True,
-                                            verbose = False)
+                                            chain_type_kwargs = {"prompt": RAG_CHAIN_PROMPT}, 
+                                            retriever = db.as_retriever(search_kwargs = {"k": config["k"]}), 
+                                            return_source_documents = True)
     
-    completion = rag_chain({"query": prompt})
+    completion = rag_chain({"query": prompt}, include_run_info = true)
     print("###" + str(completion))
+    print("###" + str(completion["run_info"]))
     
     return completion, rag_chain