feat: Implement agent message received handling and update agent status logic

Author: Killea

src/db/crud.py

@@ -1593,6 +1593,13 @@ async def agent_msg_wait(db: aiosqlite.Connection, agent_id: str, token: str) ->
     return await _set_agent_activity(db, agent_id, "msg_wait", touch_heartbeat=True)
 
 
+async def agent_msg_received(db: aiosqlite.Connection, agent_id: str) -> bool:
+    """Mark that agent has received a message from msg_wait and is now processing.
+    This transitions the agent from Listening → Working on the frontend.
+    """
+    return await _set_agent_activity(db, agent_id, "msg_received", touch_heartbeat=False)
+
+
 async def agent_msg_post(db: aiosqlite.Connection, agent_id: str) -> bool:
     return await _set_agent_activity(db, agent_id, "msg_post", touch_heartbeat=False)
 

src/static/js/shared-agent-status.js

@@ -4,29 +4,31 @@
    *
    * States (priority order):
    *   Listening — SSE connected + last_activity is msg_wait              (⏳)
-   *   Working   — SSE connected + NOT msg_wait + activity within 60s     (⚡)
-   *   Idle      — SSE connected + NOT msg_wait + activity older than 60s (🔌)
+   *   Working   — SSE connected + last_activity is msg_received/msg_post  (⚡)
+   *               (no timeout — agent may be on a long task)
+   *   Idle      — SSE connected + never entered message loop              (🔌)
    *               (SSE open but agent stopped responding — may be processing
    *               a long task or is a stale connection)
    *   Offline   — no SSE + heartbeat expired                             (⚫)
    *
    * For stdio agents (no SSE): fall back to heartbeat-based detection.
    * stdio agents will show Listening when is_online=true, Offline otherwise.
    */
-  const WORKING_TIMEOUT_S = 60;
-
   function getAgentState(agent) {
     if (!agent) return "Offline";
 
     if (agent.is_sse_connected) {
       if (agent.last_activity === "msg_wait") return "Listening";
-      // Check how long ago the last activity was
-      const lastActivityTime = agent.last_activity_time ? new Date(agent.last_activity_time) : null;
-      if (lastActivityTime) {
-        const elapsedS = (Date.now() - lastActivityTime.getTime()) / 1000;
-        if (elapsedS > WORKING_TIMEOUT_S) return "Idle";
+      // msg_received: msg_wait just delivered a message, agent is processing.
+      // msg_post: agent just posted a reply, still in working cycle.
+      // No timeout — agent may be doing a long task (editing code, reasoning, etc.)
+      // SSE disconnect is the only reliable signal that work has truly stopped.
+      if (agent.last_activity === "msg_received" || agent.last_activity === "msg_post") {
+        return "Working";
       }
-      return "Working";
+      // Any other activity (registered, heartbeat, resume, etc.) with SSE open:
+      // agent is connected but hasn't entered the message loop yet → Idle.
+      return "Idle";
     }
 
     // stdio or truly offline — rely on heartbeat

src/tools/dispatch.py

@@ -848,6 +848,7 @@ async def _poll():
                         # Exit wait state only when returning a message to caller.
                         if agent_id:
                             await crud.thread_wait_exit(db, thread_id, agent_id)
+                            await crud.agent_msg_received(db, agent_id)
                         return filtered
                     # Messages were present but not targeted at this waiter.
                     # Move the local cursor forward to avoid polling the same
@@ -857,6 +858,7 @@ async def _poll():
                     # No for_agent filter: any message wakes this waiter.
                     if agent_id:
                         await crud.thread_wait_exit(db, thread_id, agent_id)
+                        await crud.agent_msg_received(db, agent_id)
                     return msgs
 
             now = asyncio.get_event_loop().time()