Update rag.py

rag.py

@@ -101,7 +101,6 @@ def llm_chain(config, openai_api_key, prompt):
     
     with get_openai_callback() as cb:
         completion = llm_chain.generate([{"question": prompt}])
-        print(cb)
     
     return completion, llm_chain, cb
 
@@ -120,6 +119,5 @@ def rag_chain(config, openai_api_key, rag_option, prompt):
     
     with get_openai_callback() as cb:
         completion = rag_chain({"query": prompt})
-        print(cb)
 
     return completion, rag_chain, cb