openai-llm-rag

Build error

bstraehle commited on Jan 7, 2024

Commit

708911f

1 Parent(s): da2a4b4

Update rag_llamaindex.py

Files changed (1) hide show

rag_llamaindex.py CHANGED Viewed

@@ -124,6 +124,6 @@ class LlamaIndexRAG(BaseRAG):
         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

         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.total_llm_token_count))
         return completion