Only report user messages

code-asher · code-asher · commit 234bead29565 · 2025-06-11T18:04:30.000-08:00
diff --git a/cli/exp_mcp.go b/cli/exp_mcp.go
@@ -599,12 +599,14 @@ func (s *mcpServer) startWatcher(ctx context.Context, inv *serpent.Invocation) {
 						return
 					}
 				case agentapi.EventMessageUpdate:
-					err := s.queue.Push(reportTask{
-						messageID: ev.Id,
-					})
-					if err != nil {
-						cliui.Warnf(inv.Stderr, "Failed to queue update: %s", err)
-						return
+					if ev.Role == agentapi.RoleUser {
+						err := s.queue.Push(reportTask{
+							messageID: ev.Id,
+						})
+						if err != nil {
+							cliui.Warnf(inv.Stderr, "Failed to queue update: %s", err)
+							return
+						}
 					}
 				}
 			case err := <-errCh:
diff --git a/cli/exp_mcp_test.go b/cli/exp_mcp_test.go
@@ -790,11 +790,12 @@ func TestExpMcpReporter(t *testing.T) {
 			}
 		}
 
-		makeMessageEvent := func(id int64) *codersdk.ServerSentEvent {
+		makeMessageEvent := func(id int64, role agentapi.ConversationRole) *codersdk.ServerSentEvent {
 			return &codersdk.ServerSentEvent{
 				Type: ServerSentEventTypeMessageUpdate,
 				Data: agentapi.EventMessageUpdate{
-					Id: id,
+					Id:   id,
+					Role: role,
 				},
 			}
 		}
@@ -813,7 +814,7 @@ func TestExpMcpReporter(t *testing.T) {
 				return
 			}
 			// Send initial message.
-			send(*makeMessageEvent(0))
+			send(*makeMessageEvent(0, agentapi.RoleAgent))
 			listening <- send
 			<-closed
 		}))
@@ -902,13 +903,17 @@ func TestExpMcpReporter(t *testing.T) {
 					URI:     "https://dev.coder.com",
 				},
 			},
-			// Terminal becomes active again according to the screen watcher, but
-			// message length is the same.  This could be the LLM being active again,
-			// but it could also be the user messing around.  We will prefer not
-			// updating the status so the "working" update here should be skipped.
+			// Terminal becomes active again according to the screen watcher, but no
+			// new user message.  This could be the LLM being active again, but it
+			// could also be the user messing around.  We will prefer not updating the
+			// status so the "working" update here should be skipped.
 			{
 				event: makeStatusEvent(agentapi.StatusRunning),
 			},
+			// Agent messages are ignored.
+			{
+				event: makeMessageEvent(1, agentapi.RoleAgent),
+			},
 			// LLM reports that it failed and URI is blank.
 			{
 				state:   codersdk.WorkspaceAppStatusStateFailure,
@@ -923,10 +928,10 @@ func TestExpMcpReporter(t *testing.T) {
 			{
 				event: makeStatusEvent(agentapi.StatusRunning),
 			},
-			// ... but this time the message length has increased so we know there is
-			// LLM activity.  This time the "working" update will not be skipped.
+			// ... but this time we have a new user message so we know there is LLM
+			// activity.  This time the "working" update will not be skipped.
 			{
-				event: makeMessageEvent(1),
+				event: makeMessageEvent(2, agentapi.RoleUser),
 				expected: &codersdk.WorkspaceAppStatus{
 					State:   codersdk.WorkspaceAppStatusStateWorking,
 					Message: "oops",
