log total tokens genreated

mgonzs13 · Jul 5, 2024 · c64f881 · c64f881
1 parent 31b1ffb
commit c64f881
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/chatbot_ros/chatbot_ros/states/llama_state.py b/chatbot_ros/chatbot_ros/states/llama_state.py
@@ -63,6 +63,8 @@ def handle_result(
     ) -> str:
 
         self._node.get_logger().info(result.response.text)
+        self._node.get_logger().info(
+            f"Total tokens: {len(result.response.tokens)}")
 
         self._tts_end_event.clear()
         self._tts_end_event.wait()