fix(enrichment): align readiness input mapping

Author: TheodoreSpeaks

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

@@ -4,7 +4,9 @@ import React from 'react'
 import { Button, Checkbox, cn, handleKeyboardActivation } from '@sim/emcn'
 import { PlayOutline, Square } from '@sim/emcn/icons'
 import type { ActiveDispatch } from '@/lib/api/contracts/tables'
-import type { TableRow as TableRowType, WorkflowGroup } from '@/lib/table'
+import type { ColumnDefinition, TableRow as TableRowType, WorkflowGroup } from '@/lib/table'
+import { mapInputValues } from '@/lib/table/cell-format'
+import { getColumnId } from '@/lib/table/column-keys'
 import { getUnmetGroupDeps } from '@/lib/table/deps'
 import { getEnrichmentReadiness } from '@/enrichments/readiness'
 import { getEnrichment } from '@/enrichments/registry'
@@ -23,6 +25,7 @@ import { type NormalizedSelection, resolveCellExec } from './utils'
 export interface DataRowProps {
   row: TableRowType
   columns: DisplayColumn[]
+  schemaColumns: ColumnDefinition[]
   /** Current workspace id — forwarded to cells so in-workspace resource URLs
    *  render as tagged-resource chips. */
   workspaceId: string
@@ -106,6 +109,7 @@ function dataRowPropsAreEqual(prev: DataRowProps, next: DataRowProps): boolean {
   if (
     prev.row !== next.row ||
     prev.columns !== next.columns ||
+    prev.schemaColumns !== next.schemaColumns ||
     prev.workspaceId !== next.workspaceId ||
     prev.rowIndex !== next.rowIndex ||
     prev.isFirstRow !== next.isFirstRow ||
@@ -152,6 +156,7 @@ function dataRowPropsAreEqual(prev: DataRowProps, next: DataRowProps): boolean {
 export const DataRow = React.memo(function DataRow({
   row,
   columns,
+  schemaColumns,
   workspaceId,
   rowIndex,
   isFirstRow,
@@ -190,7 +195,7 @@ export const DataRow = React.memo(function DataRow({
   const waitingByGroupId = React.useMemo(() => {
     if (workflowGroups.length === 0) return null
     // Deps are stored as column ids; the "Waiting on …" pill shows display names.
-    const nameByColumnId = new Map(columns.map((c) => [c.key, c.name]))
+    const nameByColumnId = new Map(schemaColumns.map((c) => [getColumnId(c), c.name]))
     const map = new Map<string, string[]>()
     for (const group of workflowGroups) {
       const labels = new Set<string>()
@@ -204,12 +209,15 @@ export const DataRow = React.memo(function DataRow({
       if (group.type === 'enrichment' && (group.autoRun !== false || hasAttemptedManualRun)) {
         const enrichment = getEnrichment(group.enrichmentId)
         if (enrichment) {
+          const ownOutputColumns = new Set(group.outputs.map((output) => output.columnName))
+          const enrichmentInputMappings = (group.inputMappings ?? []).filter(
+            (mapping) => !ownOutputColumns.has(mapping.columnName)
+          )
           const inputColumnById = new Map<string, string>()
-          const inputs: Record<string, unknown> = {}
-          for (const mapping of group.inputMappings ?? []) {
+          for (const mapping of enrichmentInputMappings) {
             inputColumnById.set(mapping.inputName, mapping.columnName)
-            inputs[mapping.inputName] = row.data[mapping.columnName]
           }
+          const inputs = mapInputValues(row.data, schemaColumns, enrichmentInputMappings)
           const readiness = getEnrichmentReadiness(enrichment, inputs)
           if (!readiness.ready) {
             const mappedLabels = readiness.missingInputs.flatMap((input) => {
@@ -228,7 +236,7 @@ export const DataRow = React.memo(function DataRow({
       if (labels.size > 0) map.set(group.id, [...labels])
     }
     return map
-  }, [workflowGroups, row, columns])
+  }, [workflowGroups, row, schemaColumns])
   const isMultiCell = sel !== null && (sel.startRow !== sel.endRow || sel.startCol !== sel.endCol)
   const isRowSelected = isRowChecked
   /**

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

@@ -4459,6 +4459,7 @@ export function TableGrid({
                                 key={row.id}
                                 row={row}
                                 columns={displayColumns}
+                                schemaColumns={columns}
                                 workspaceId={workspaceId}
                                 rowIndex={index}
                                 isFirstRow={index === 0}