openclaw - 💡(How to fix) Fix Feature Request: TUI should show working/busy status when agent is executing tool calls [2 comments, 3 participants]

Fix Plan

To address the issue, we need to update the TUI status bar to reflect the agent's state based on tool call lifecycle events. Here are the steps:

• Update the TUI to listen for tool_call_start and tool_call_end events from the gateway.
• When a tool_call_start event is received, update the status bar to connected | working.
• When a tool_call_end event is received, check if there are any pending tool calls. If not, update the status bar to connected | idle.
• If the agent starts streaming a text reply, update the status bar to connected | responding.

Example Code

import enum

class AgentStatus(enum.Enum):
    IDLE = "connected | idle"
    WORKING = "connected | working"
    RESPONDING = "connected | responding"

class TUI:
    def __init__(self):
        self.status = AgentStatus.IDLE
        self.pending_tool_calls = 0

    def on_tool_call_start(self):
        self.pending_tool_calls += 1
        self.status = AgentStatus.WORKING

    def on_tool_call_end(self):
        self.pending_tool_calls -= 1
        if self.pending_tool_calls == 0:
            self.status = AgentStatus.IDLE

    def on_streaming_text_reply(self):
        self.status = AgentStatus.RESPONDING

    def update_status_bar(self):
        # Update the status bar with the current status
        print(self.status.value)

# Example usage:
tui = TUI()
tui.on_tool_call_start()  # connected | working
tui.on_tool_call_end()   # connected | idle
tui.on_tool_call_start()  # connected | working
tui.on_streaming_text_reply()  # connected | responding
tui.on_tool_call_end()   # connected | idle

Verification

To verify the fix, test the TUI with different scenarios:

• Send a message and verify that the status bar updates to connected | working.
• Wait for the agent to finish processing and verify that the status bar updates to connected | idle.
• Send a message that triggers a tool call and verify that the status bar updates to connected | working and then connected | responding when the agent starts streaming a text reply.

Extra Tips

• Make sure to handle edge cases, such as multiple tool calls in parallel or tool calls that fail.
• Consider adding a timeout to update the status bar to connected | idle if the agent takes too long to respond.