fix(chat): persist the source names a selection chip renders from

Cursor Bugbot: fileName/tableName were mapped into the optimistic message and
accepted by the API schema, but PersistedMessageContext, buildPersistedUserMessage
and toDisplayContexts never carried them. After a reload a file_selection chip
fell back to its label for getDocumentIcon — and the label carries a location
suffix ('notes.md:12-40'), so extension detection broke.

Persists only the two names the display path reads. The rest of the payload
(text, rowIds, columnIds, line numbers) stays unpersisted on purpose: it exists
to resolve the selection server-side at send time, is never re-read when
rendering a past message, and would put a selection-sized blob — up to the 20k
char cap — in every stored message.

Test asserts both halves of that, and was verified to fail with the mapping
removed.

Author: waleedlatif1

apps/sim/lib/copilot/chat/display-message.ts

@@ -130,6 +130,8 @@ function toDisplayContexts(
     ...(c.blockType ? { blockType: c.blockType } : {}),
     ...(c.skillId ? { skillId: c.skillId } : {}),
     ...(c.serverId ? { serverId: c.serverId } : {}),
+    ...(c.fileName ? { fileName: c.fileName } : {}),
+    ...(c.tableName ? { tableName: c.tableName } : {}),
   }))
 }
 

apps/sim/lib/copilot/chat/persisted-message.test.ts

@@ -235,6 +235,47 @@ describe('persisted-message', () => {
     expect(msg.fileAttachments).toBeUndefined()
     expect(msg.contexts).toBeUndefined()
   })
+
+  it('persists the source names a selection chip renders from, but not its payload', () => {
+    const msg = buildPersistedUserMessage({
+      id: 'user-1',
+      content: 'explain this',
+      contexts: [
+        {
+          kind: 'file_selection',
+          label: 'notes.md:12-40',
+          fileId: 'f1',
+          fileName: 'notes.md',
+          // Send-time payload: resolved server-side, never re-read for display.
+          text: 'the exact passage',
+          startLine: 12,
+          endLine: 40,
+        },
+        {
+          kind: 'table_selection',
+          label: 'Sales (2 rows)',
+          tableId: 't1',
+          tableName: 'Sales',
+          rowIds: ['r1', 'r2'],
+        },
+      ],
+    })
+
+    // fileName must survive: the label carries a `:12-40` suffix, so the chip's
+    // icon cannot recover an extension from it after a reload.
+    expect(msg.contexts?.[0]).toEqual({
+      kind: 'file_selection',
+      label: 'notes.md:12-40',
+      fileId: 'f1',
+      fileName: 'notes.md',
+    })
+    expect(msg.contexts?.[1]).toEqual({
+      kind: 'table_selection',
+      label: 'Sales (2 rows)',
+      tableId: 't1',
+      tableName: 'Sales',
+    })
+  })
 })
 
 describe('stripToolResultOutput', () => {

apps/sim/lib/copilot/chat/persisted-message.ts

@@ -78,6 +78,18 @@ interface PersistedMessageContext {
   blockType?: string
   skillId?: string
   serverId?: string
+  /**
+   * Source names for `file_selection` / `table_selection` chips. Persisted
+   * because the rendered chip reads them — the label carries a location suffix
+   * (`notes.md:12-40`), so the file icon cannot derive an extension from it.
+   *
+   * The rest of a selection's payload (`text`, `rowIds`, `columnIds`, line
+   * numbers) is deliberately NOT persisted: it exists to resolve the selection
+   * server-side when the message is sent, is never read when re-rendering a past
+   * message, and would put a selection-sized blob in every stored message.
+   */
+  fileName?: string
+  tableName?: string
 }
 
 export interface PersistedMessage {
@@ -359,6 +371,8 @@ export function buildPersistedUserMessage(params: UserMessageParams): PersistedM
       ...(c.blockType ? { blockType: c.blockType } : {}),
       ...(c.skillId ? { skillId: c.skillId } : {}),
       ...(c.serverId ? { serverId: c.serverId } : {}),
+      ...(c.fileName ? { fileName: c.fileName } : {}),
+      ...(c.tableName ? { tableName: c.tableName } : {}),
     }))
   }
 
@@ -680,6 +694,8 @@ export function normalizeMessage(raw: Record<string, unknown>): PersistedMessage
       ...(c.blockType ? { blockType: c.blockType } : {}),
       ...(c.skillId ? { skillId: c.skillId } : {}),
       ...(c.serverId ? { serverId: c.serverId } : {}),
+      ...(c.fileName ? { fileName: c.fileName } : {}),
+      ...(c.tableName ? { tableName: c.tableName } : {}),
     }))
   }