fix(chat): drain rows for a column-header Add-to-chat selection

Cursor: a column-header selection spans every row of the chosen columns, but
Add-to-chat built rowIds from the loaded page only. Extend the drain (already
used for gutter select-all) to column selections — load up to the cap and take
all rows — so the chip references as many rows as it can carry.

Author: mzxchandra

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

@@ -3718,17 +3718,22 @@ export function TableGrid({
 
   const addToChat = useAddToChat()
   const handleAddSelectionToChat = useCallback(async () => {
-    // A gutter select-all only has the loaded page in `contextMenuRowIds`; drain
-    // up to the cap so the chip references as many of the selected rows as it can
-    // carry (bounded by MAX_TABLE_SELECTION_ROWS) instead of a silent
-    // loaded-only subset — mirroring how the copy path loads before writing.
+    // A gutter select-all (filtered) or a column-header selection (every row of
+    // the chosen columns) covers rows beyond the loaded page that
+    // `contextMenuRowIds` reflects; drain up to the cap so the chip references as
+    // many rows as it can carry (bounded by MAX_TABLE_SELECTION_ROWS) instead of a
+    // silent loaded-only subset — mirroring how the copy path loads before writing.
     let sourceRowIds = contextMenuRowIds
-    if (contextMenuIsSelectAll) {
+    if (contextMenuIsSelectAll || isColumnSelectionRef.current) {
       try {
         const { rows: loaded } = await ensureRowsLoadedUpToRef.current(MAX_TABLE_SELECTION_ROWS)
-        const drained = loaded
-          .filter((row) => rowSelectionIncludes(rowSelectionRef.current, row.id))
-          .map((row) => row.id)
+        // A column selection spans all rows; a gutter select-all filters by the
+        // (exclusion-aware) row selection.
+        const drained = (
+          contextMenuIsSelectAll
+            ? loaded.filter((row) => rowSelectionIncludes(rowSelectionRef.current, row.id))
+            : loaded
+        ).map((row) => row.id)
         if (drained.length > 0) sourceRowIds = drained
       } catch {
         // Fall back to the already-loaded rows if the drain fails.