Update rag.py

rag.py

@@ -100,6 +100,9 @@ def llm_chain(config, openai_api_key, prompt):
                          prompt = LLM_CHAIN_PROMPT)
     
     completion = llm_chain.generate([{"question": prompt}])
+
+    with get_openai_callback() as cb:
+        print(cb)
     
     return completion, llm_chain