mothership tasks: resolve the pill by task id (jsonb text spacing broke the exact-key match)

Author: Sg312

apps/sim/lib/mothership/tasks/wake.test.ts

@@ -1,7 +1,13 @@
 /**
  * @vitest-environment node
  */
-import { dbChainMock, queueTableRows, resetDbChainMock, schemaMock } from '@sim/testing'
+import {
+  dbChainMock,
+  dbChainMockFns,
+  queueTableRows,
+  resetDbChainMock,
+  schemaMock,
+} from '@sim/testing'
 import { beforeEach, describe, expect, it, vi } from 'vitest'
 
 const {
@@ -47,7 +53,7 @@ vi.mock('@/lib/workspaces/permissions/utils', () => ({
   checkWorkspaceAccess: mockCheckWorkspaceAccess,
 }))
 
-import { runWakeTurn, validateWake } from './wake'
+import { resolveTaskPill, runWakeTurn, validateWake } from './wake'
 
 const WAKE = {
   taskId: '22222222-2222-4222-8222-222222222222',
@@ -120,6 +126,44 @@ describe('copilot task wake', () => {
     )
   })
 
+  it('resolves the pill in the arming message before the wake turn lands', async () => {
+    queueTableRows(schemaMock.copilotMessages, [
+      {
+        id: 'row-1',
+        content: {
+          id: 'assistant-1',
+          role: 'assistant',
+          contentBlocks: [
+            { type: 'text', content: 'armed' },
+            {
+              type: 'task',
+              task: {
+                taskId: WAKE.taskId,
+                kind: 'timer',
+                target: {},
+                note: 'n',
+                status: 'pending',
+              },
+            },
+          ],
+        },
+      },
+    ])
+    await resolveTaskPill('chat-1', WAKE.taskId, 'completed', 'Timer elapsed')
+    expect(dbChainMockFns.update).toHaveBeenCalledWith(schemaMock.copilotMessages)
+    const [setArg] = dbChainMockFns.set.mock.calls[0] as [
+      {
+        content: {
+          contentBlocks: Array<{ type: string; task?: { status?: string; summary?: string } }>
+        }
+      },
+    ]
+    expect(setArg.content.contentBlocks[1]?.task).toMatchObject({
+      status: 'completed',
+      summary: 'Timer elapsed',
+    })
+  })
+
   it('skips the turn when another stream holds the chat', async () => {
     mockAcquirePendingChatStream.mockResolvedValue(false)
     await runWakeTurn(WAKE)

apps/sim/lib/mothership/tasks/wake.ts

@@ -58,7 +58,9 @@ export async function resolveTaskPill(
     .where(
       and(
         eq(copilotMessages.chatId, chatId),
-        sql`${copilotMessages.content}::text LIKE ${`%"taskId":"${taskId}"%`}`
+        // The id alone is the filter — jsonb text puts a space after every colon, so a
+        // `"taskId":"…"` pattern never matches; the block walk below is the real check.
+        sql`${copilotMessages.content}::text LIKE ${`%${taskId}%`}`
       )
     )
   for (const row of rows) {