Spaces:

bstraehle
/

rag

Running

bstraehle commited on Jan 7

Commit

3462636

•

1 Parent(s): e497043

Update rag_llamaindex.py

Files changed (1) hide show

rag_llamaindex.py CHANGED Viewed

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

         )
         completion = query_engine.query(prompt)
+        callback = self.get_callback(
             service_context.callback_manager.handlers[0])
         return completion, callback