bstraehle commited on
Commit
a426ca0
1 Parent(s): 0d56e9d

Update rag_llamaindex.py

Browse files
Files changed (1) hide show
  1. rag_llamaindex.py +2 -3
rag_llamaindex.py CHANGED
@@ -120,7 +120,6 @@ class LlamaIndexRAG(BaseRAG):
120
 
121
  completion = query_engine.query(prompt)
122
 
123
- print("111 " + str(service_context.callback_manager.token_counter))
124
- print("222 " + str(service_context.callback_manager.token_counter.total_embedding_token_count))
125
-
126
  return completion
 
120
 
121
  completion = query_engine.query(prompt)
122
 
123
+ print("111 " + str(service_context.callback_manager))
124
+
 
125
  return completion