Update rag_llamaindex.py

rag_llamaindex.py

@@ -129,7 +129,7 @@ class LlamaIndexRAG(BaseRAG):
         )
 
         completion = query_engine.query(prompt)
-        callback = get_callback(
+        callback = self.get_callback(
             service_context.callback_manager.handlers[0])
         
         return completion, callback