fix(mothership): resync chat caches after an SSE reconnect gap

task_status events are transient and never replayed, so any window with no open connection can drop a create, rename, delete, or completion. The chat hook reconnected silently and reconciled nothing, leaving list and detail caches stale until an unrelated action refreshed them.

Resync on reconnect, on the first open of a re-subscription, and on a first open that only succeeded after an error, matching the pattern useMcpToolsEvents already uses for the same gap.

Author: waleedlatif1

apps/sim/hooks/use-mothership-chat-events.test.ts

@@ -14,7 +14,10 @@ vi.mock('@/lib/browser-agent/transport', () => ({ suspendBrowserScope }))
 vi.mock('@/lib/terminal/transport', () => ({ suspendTerminalScope }))
 
 import { mothershipChatKeys } from '@/hooks/queries/mothership-chats'
-import { handleMothershipChatStatusEvent } from '@/hooks/use-mothership-chat-events'
+import {
+  handleMothershipChatStatusEvent,
+  resyncMothershipChatCaches,
+} from '@/hooks/use-mothership-chat-events'
 
 describe('handleMothershipChatStatusEvent', () => {
   const queryClient = {
@@ -419,3 +422,25 @@ describe('handleMothershipChatStatusEvent', () => {
     expect(queryClient.removeQueries).not.toHaveBeenCalled()
   })
 })
+
+describe('resyncMothershipChatCaches', () => {
+  const queryClient = {
+    invalidateQueries: vi.fn().mockResolvedValue(undefined),
+  } satisfies Pick<QueryClient, 'invalidateQueries'>
+
+  beforeEach(() => {
+    vi.clearAllMocks()
+  })
+
+  it('invalidates the workspace lists and every chat detail', () => {
+    resyncMothershipChatCaches(queryClient, 'ws-1')
+
+    expect(queryClient.invalidateQueries).toHaveBeenCalledTimes(2)
+    expect(queryClient.invalidateQueries).toHaveBeenCalledWith({
+      queryKey: mothershipChatKeys.workspaceLists('ws-1'),
+    })
+    expect(queryClient.invalidateQueries).toHaveBeenCalledWith({
+      queryKey: mothershipChatKeys.details(),
+    })
+  })
+})

apps/sim/hooks/use-mothership-chat-events.ts

@@ -9,6 +9,9 @@ import { type MothershipChatHistory, mothershipChatKeys } from '@/hooks/queries/
 
 const logger = createLogger('MothershipChatEvents')
 
+/** Workspaces this process has subscribed to before, so a re-subscribe can be told from a first one. */
+const everSubscribed = new Set<string>()
+
 const CHAT_STATUS_TYPES = ['started', 'completed', 'created', 'deleted', 'renamed'] as const
 type ChatStatusEventType = (typeof CHAT_STATUS_TYPES)[number]
 const CHAT_STATUS_TYPE_SET = new Set<string>(CHAT_STATUS_TYPES)
@@ -128,6 +131,22 @@ export function handleMothershipChatStatusEvent(
   }
 }
 
+/**
+ * Re-syncs chat caches after a gap in the event stream.
+ *
+ * `task_status` events are transient — nothing replays what was published while
+ * no connection was open — so any reconnect may have missed a create, rename,
+ * delete, or completion. Invalidating the workspace lists and every chat detail
+ * reconciles from the server; only queries that are currently mounted refetch.
+ */
+export function resyncMothershipChatCaches(
+  queryClient: Pick<QueryClient, 'invalidateQueries'>,
+  workspaceId: string
+): void {
+  queryClient.invalidateQueries({ queryKey: mothershipChatKeys.workspaceLists(workspaceId) })
+  queryClient.invalidateQueries({ queryKey: mothershipChatKeys.details() })
+}
+
 /**
  * Subscribes to chat status SSE events and invalidates chat caches on changes.
  * The SSE event name remains `task_status` for wire compatibility.
@@ -154,7 +173,28 @@ export function useMothershipChatEvents(workspaceId: string | undefined) {
       )
     })
 
+    // `onopen` fires on the initial connect and on every auto-reconnect. Re-sync
+    // whenever a gap could have swallowed an event: on any reconnect, on the
+    // first open of a RE-subscription (switching workspace away and back tears
+    // the connection down, and the list/detail queries remount inside their
+    // stale times so they do not refetch on their own), and on a first open that
+    // only succeeded after an error (the initial queries may have failed during
+    // that gap and will not retry themselves). Skip only a clean first
+    // subscription — those queries fetch fresh on their own initial mount.
+    const isResubscribe = everSubscribed.has(workspaceId)
+    everSubscribed.add(workspaceId)
+    let opened = false
+    let erroredBeforeOpen = false
+
+    eventSource.onopen = () => {
+      if (opened || isResubscribe || erroredBeforeOpen) {
+        resyncMothershipChatCaches(queryClient, workspaceId)
+      }
+      opened = true
+    }
+
     eventSource.onerror = () => {
+      if (!opened) erroredBeforeOpen = true
       logger.warn(`SSE connection error for workspace ${workspaceId}`)
     }