Update rag.py

rag.py

@@ -104,6 +104,9 @@ def llm_chain(config, openai_api_key, prompt):
 
 def rag_chain(config, openai_api_key, rag_option, prompt):
     llm = get_llm(config, openai_api_key)
+
+    with get_openai_callback() as cb:
+        print(cb)
     
     if (rag_option == RAG_CHROMA):
         db = document_retrieval_chroma()