fix(chat): unique table selection labels + correct file-selection icon

Cursor:
- buildTableSelectionLabel now appends a short deterministic key (selectionKey
  over the row/column ids) so two distinct same-size selections don't share a
  label; guard insertContextChip against an already-selected label so a duplicate
  never inserts an orphan @token with no backing context.
- file_selection chip icon strips the `:line` suffix before getDocumentIcon so it
  reads the real extension (`md`, not `md:12-40`) instead of the generic glyph.

Author: mzxchandra

apps/sim/app/workspace/[workspaceId]/home/components/chat-context-kind-registry/chat-context-kind-registry.tsx

@@ -12,6 +12,7 @@ import {
 } from '@sim/emcn/icons'
 import { AgentSkillsIcon, McpIcon } from '@/components/icons'
 import { getDocumentIcon } from '@/components/icons/document-icons'
+import { fileNameFromSelectionLabel } from '@/lib/copilot/chat/selection-context'
 import type { ChatContextKind, ChatMessageContext } from '@/app/workspace/[workspaceId]/home/types'
 import { getBareIconStyle } from '@/blocks/icon-color'
 import { getBlockRegistry } from '@/blocks/registry'
@@ -89,7 +90,9 @@ export const CHAT_CONTEXT_KIND_REGISTRY: Record<ChatContextKind, ChatContextKind
   file_selection: {
     label: 'File selection',
     renderIcon: ({ context, className }) => {
-      const FileDocIcon = getDocumentIcon('', context.label)
+      // Strip the `:line` suffix so `getDocumentIcon` reads the real extension
+      // (e.g. `md`, not `md:12-40`) and shows the correct file glyph.
+      const FileDocIcon = getDocumentIcon('', fileNameFromSelectionLabel(context.label))
       return <FileDocIcon className={className} />
     },
   },

apps/sim/app/workspace/[workspaceId]/home/components/user-input/components/prompt-editor/use-prompt-editor.ts

@@ -21,6 +21,7 @@ import {
   useMentionTokens,
 } from '@/app/workspace/[workspaceId]/w/[workflowId]/components/panel/components/copilot/components/user-input/hooks'
 import {
+  isContextAlreadySelected,
   restoreSkillTriggerText,
   SKILL_CHIP_TRIGGER,
 } from '@/app/workspace/[workspaceId]/w/[workflowId]/components/panel/components/copilot/components/user-input/utils'
@@ -470,6 +471,13 @@ export function usePromptEditor({
    */
   const insertContextChip = useCallback(
     (context: ChatContext) => {
+      // A chip's `@label` token must be unique — `addContext` dedupes a matching
+      // label, so inserting a token for an already-present context would orphan
+      // it (a second token with no backing context). Skip and just focus.
+      if (isContextAlreadySelected(context, contextManagementRef.current.selectedContexts)) {
+        textareaRef.current?.focus()
+        return
+      }
       const textarea = textareaRef.current
       if (textarea) {
         const currentValue = valueRef.current

apps/sim/app/workspace/[workspaceId]/tables/[tableId]/components/table-grid/table-grid.tsx

@@ -15,6 +15,7 @@ import {
   buildTableSelectionLabel,
   MAX_TABLE_SELECTION_COLUMNS,
   MAX_TABLE_SELECTION_ROWS,
+  selectionKey,
 } from '@/lib/copilot/chat/selection-context'
 import { captureEvent } from '@/lib/posthog/client'
 import type {
@@ -2888,7 +2889,12 @@ export function TableGrid({
               attachSelectionContextToClipboard(e.clipboardData, {
                 kind: 'table_selection',
                 tableId,
-                label: buildTableSelectionLabel(tableNameRef.current, rowIds.length),
+                label: buildTableSelectionLabel(
+                  tableNameRef.current,
+                  rowIds.length,
+                  undefined,
+                  selectionKey(rowIds)
+                ),
                 rowIds,
               })
             }
@@ -2964,7 +2970,8 @@ export function TableGrid({
             label: buildTableSelectionLabel(
               tableNameRef.current,
               rangeRowIds.length,
-              columnIds?.length
+              columnIds?.length,
+              selectionKey([...rangeRowIds, ...(columnIds ?? [])])
             ),
             rowIds: rangeRowIds,
             ...(columnIds ? { columnIds } : {}),
@@ -3745,7 +3752,12 @@ export function TableGrid({
     addToChat({
       kind: 'table_selection',
       tableId,
-      label: buildTableSelectionLabel(tableData?.name ?? 'Table', rowIds.length, columnIds?.length),
+      label: buildTableSelectionLabel(
+        tableData?.name ?? 'Table',
+        rowIds.length,
+        columnIds?.length,
+        selectionKey([...rowIds, ...(columnIds ?? [])])
+      ),
       rowIds,
       ...(columnIds && columnIds.length > 0 ? { columnIds } : {}),
     })

apps/sim/lib/copilot/chat/selection-context.test.ts

@@ -2,7 +2,13 @@
  * @vitest-environment node
  */
 import { describe, expect, it } from 'vitest'
-import { buildFileSelectionLabel, buildTableSelectionLabel } from './selection-context'
+import {
+  buildFileSelectionLabel,
+  buildTableSelectionLabel,
+  fileNameFromSelectionLabel,
+  selectionKey,
+  tableNameFromSelectionLabel,
+} from './selection-context'
 
 describe('buildFileSelectionLabel', () => {
   it('renders a line range', () => {
@@ -32,4 +38,32 @@ describe('buildTableSelectionLabel', () => {
     expect(buildTableSelectionLabel('Sales', 5, 3)).toBe('Sales (5 rows, 3 cols)')
     expect(buildTableSelectionLabel('Sales', 2, 1)).toBe('Sales (2 rows, 1 col)')
   })
+
+  it('appends the disambiguation key when provided', () => {
+    expect(buildTableSelectionLabel('Sales', 2, undefined, 'k3f9')).toBe('Sales (2 rows #k3f9)')
+    expect(buildTableSelectionLabel('Sales', 2, 3, 'k3f9')).toBe('Sales (2 rows, 3 cols #k3f9)')
+  })
+})
+
+describe('selectionKey', () => {
+  it('is deterministic and order-independent', () => {
+    expect(selectionKey(['r1', 'r2', 'r3'])).toBe(selectionKey(['r3', 'r1', 'r2']))
+  })
+
+  it('differs for distinct id sets of the same size', () => {
+    expect(selectionKey(['r1', 'r2'])).not.toBe(selectionKey(['r3', 'r4']))
+  })
+})
+
+describe('selection-label name recovery', () => {
+  it('strips the line suffix from a file selection label', () => {
+    expect(fileNameFromSelectionLabel('notes.md:12-40')).toBe('notes.md')
+    expect(fileNameFromSelectionLabel('notes.md')).toBe('notes.md')
+  })
+
+  it('strips rows/cols/key suffix from a table selection label', () => {
+    expect(tableNameFromSelectionLabel('Sales (2 rows)')).toBe('Sales')
+    expect(tableNameFromSelectionLabel('Sales (2 rows, 3 cols #k3f9)')).toBe('Sales')
+    expect(tableNameFromSelectionLabel('Sales (5 rows #ab12)')).toBe('Sales')
+  })
 })

apps/sim/lib/copilot/chat/selection-context.ts

@@ -51,19 +51,37 @@ export function buildFileSelectionLabel(
 }
 
 /**
- * Builds the chip label for a table selection, e.g. `Sales (5 rows)` or, for a
- * cell range, `Sales (5 rows, 3 cols)`. ASCII-only for the same reason as
- * {@link buildFileSelectionLabel}.
+ * Short, deterministic key for a set of ids — same ids (any order) yield the
+ * same key. Used to disambiguate table-selection labels so two distinct
+ * selections of the same size don't collapse to one chip (chips are keyed by
+ * their `@label`, and a collision would drop the second context).
+ */
+export function selectionKey(ids: string[]): string {
+  const joined = [...ids].sort().join(',')
+  let hash = 0
+  for (let i = 0; i < joined.length; i++) {
+    hash = (Math.imul(hash, 31) + joined.charCodeAt(i)) | 0
+  }
+  return (hash >>> 0).toString(36)
+}
+
+/**
+ * Builds the chip label for a table selection, e.g. `Sales (5 rows · k3f9)` or,
+ * for a cell range, `Sales (5 rows, 3 cols · k3f9)`. The trailing `key` (from
+ * {@link selectionKey}) keeps distinct same-size selections from sharing a
+ * label. ASCII-only for the same reason as {@link buildFileSelectionLabel}.
  */
 export function buildTableSelectionLabel(
   tableName: string,
   rowCount: number,
-  columnCount?: number
+  columnCount?: number,
+  key?: string
 ): string {
   const rows = `${rowCount} ${rowCount === 1 ? 'row' : 'rows'}`
-  if (!columnCount) return `${tableName} (${rows})`
+  const suffix = key ? ` #${key}` : ''
+  if (!columnCount) return `${tableName} (${rows}${suffix})`
   const cols = `${columnCount} ${columnCount === 1 ? 'col' : 'cols'}`
-  return `${tableName} (${rows}, ${cols})`
+  return `${tableName} (${rows}, ${cols}${suffix})`
 }
 
 /**
@@ -82,5 +100,5 @@ export function fileNameFromSelectionLabel(label: string): string {
  * builder for the same reason as {@link fileNameFromSelectionLabel}.
  */
 export function tableNameFromSelectionLabel(label: string): string {
-  return label.replace(/\s*\(\d+ rows?(?:, \d+ cols?)?\)$/, '')
+  return label.replace(/\s*\(\d+ rows?(?:, \d+ cols?)?(?: #[0-9a-z]+)?\)$/, '')
 }