Update rag_llamaindex.py

rag_llamaindex.py

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