Spaces:

bstraehle
/

rag

Running

bstraehle commited on Jan 7, 2024

Commit

da2a4b4

1 Parent(s): e18b2ee

Update rag_llamaindex.py

Files changed (1) hide show

rag_llamaindex.py CHANGED Viewed

@@ -123,7 +123,7 @@ class LlamaIndexRAG(BaseRAG):
         print("111 " + str(service_context.callback_manager))
         print("111 " + str(service_context.callback_manager.handlers))
         print("111 " + str(service_context.callback_manager.handlers[0]))
-        token_counter = (TokenCountingHandler) service_context.callback_manager.handlers[0]
         print("222 " + str(token_counter))
         return completion

         print("111 " + str(service_context.callback_manager))
         print("111 " + str(service_context.callback_manager.handlers))
         print("111 " + str(service_context.callback_manager.handlers[0]))
+        token_counter = service_context.callback_manager.handlers[0]
         print("222 " + str(token_counter))
         return completion