diff --git a/chatbot_ros/chatbot_ros/states/llama_state.py b/chatbot_ros/chatbot_ros/states/llama_state.py index c484c2e..4f33b91 100644 --- 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()