Update rag_llamaindex.py

rag_llamaindex.py

@@ -62,15 +62,13 @@ class LlamaIndexRAG(BaseRAG):
         return CallbackManager([token_counter])
 
     def get_callback(self, token_counter):
-        return ("Embedding Tokens: " +
-                str(token_counter.total_embedding_token_count) + "\n" +
-                "LLM Prompt Tokens: " +
+        return ("Tokens Used: " +
+                str(token_counter.total_llm_token_count) + "\n" +
+                "Prompt Tokens: " +
                 str(token_counter.prompt_llm_token_count) + "\n" +
-                "LLM Completion Tokens: " +
-                str(token_counter.completion_llm_token_count)  + "\n" +
-                "Total LLM Token Count: " +
-                str(token_counter.total_llm_token_count))
-        
+                "Completion Tokens: " +
+                str(token_counter.completion_llm_token_count))
+
     def get_llm(self, config):
         return OpenAI(
             model = config["model_name"],