@claritylabs/cl-sdk 3.1.8 → 3.1.10

package/dist/index.js

@@ -11104,6 +11104,19 @@ function isSourceTreeHeaderRow(row) {
 function tableCellText(cell) {
   return cleanText(cell.textExcerpt ?? cell.description ?? cell.title, "");
 }
+function tableCellValueText(cell) {
+  return cleanText(cell.textExcerpt ?? cell.description ?? "", "");
+}
+function tableRowTextFromCells(cells) {
+  const text = cells.map((cell) => {
+    const label = cleanText(cell.title, "");
+    const value = tableCellValueText(cell);
+    if (!value) return label;
+    if (!label || value.toLowerCase() === label.toLowerCase()) return value;
+    return `${label}: ${value}`;
+  }).filter(Boolean).join(" | ");
+  return text ? cleanText(text, text) : void 0;
+}
 function tableRowTextForPrompt(row, cells) {
   return cleanText(
     cells.length ? cells.map(tableCellText).filter(Boolean).join(" | ") : row.textExcerpt ?? row.description ?? row.title,
@@ -11118,6 +11131,17 @@ function isGenericColumnTitle(value) {
   const title = cleanText(value, "");
   return !title || /^(?:column\s+\d+|table cell|value)$/i.test(title);
 }
+function metadataColumnName(metadata) {
+  const value = metadata?.columnName;
+  return typeof value === "string" ? cleanText(value, "") || void 0 : void 0;
+}
+function metadataTableColumnName(metadata) {
+  const table = metadata?.table;
+  if (!table || typeof table !== "object" || Array.isArray(table)) return void 0;
+  if (!("columnName" in table)) return void 0;
+  const value = table.columnName;
+  return typeof value === "string" ? cleanText(value, "") || void 0 : void 0;
+}
 function bboxSummary(node) {
   const box = node.bbox?.[0];
   if (!box) return void 0;
@@ -11261,12 +11285,40 @@ function findCellForContinuation(params) {
   }
   return params.cells[1] ?? params.cells[params.cells.length - 1];
 }
+function columnLabelStartIndex(rows) {
+  const headerIndex = rows.findIndex(
+    ({ row, cells }) => isSourceTreeHeaderRow(row) && cells.length > 1
+  );
+  return headerIndex >= 0 ? headerIndex : 0;
+}
+function metadataWithColumnLabel(metadata, label) {
+  const next = {
+    ...metadata ?? {},
+    columnName: label,
+    visualTableRepairColumnLabel: label
+  };
+  if (metadata?.table && typeof metadata.table === "object" && !Array.isArray(metadata.table)) {
+    next.table = {
+      ...metadata.table,
+      columnName: label
+    };
+  }
+  return next;
+}
+function tableCellDescription(cell, label) {
+  const text = tableCellText(cell);
+  return cleanText(
+    text && text.toLowerCase() !== label.toLowerCase() ? `${label} | ${text}` : label,
+    cell.description
+  );
+}
 function applyVisualTableRepair(sourceTree, repair) {
   if (repair.tables.length === 0) return sourceTree;
   const byId = new Map(sourceTree.map((node) => [node.id, node]));
   const byParent = nodesByParent(sourceTree);
   const updates = /* @__PURE__ */ new Map();
   const removeIds = /* @__PURE__ */ new Set();
+  const rowsToRebuildText = /* @__PURE__ */ new Set();
   const currentNode = (id) => updates.get(id) ?? byId.get(id);
   for (const tableRepair of repair.tables) {
     const table = byId.get(tableRepair.tableNodeId);
@@ -11280,20 +11332,40 @@ function applyVisualTableRepair(sourceTree, repair) {
       if (normalized) columnLabels.set(label.columnIndex, normalized);
     }
     if (columnLabels.size > 0) {
-      for (const { row, cells } of rows) {
+      const firstLabelRowIndex = columnLabelStartIndex(rows);
+      const normalizedLabels = new Set([...columnLabels.values()].map((label) => label.toLowerCase()));
+      for (const [rowIndex, { row, cells }] of rows.entries()) {
         if (removeIds.has(row.id)) continue;
+        if (rowIndex < firstLabelRowIndex) {
+          for (const cell of cells) {
+            const metadataLabel = metadataColumnName(cell.metadata);
+            if (metadataLabel && isGenericColumnTitle(metadataLabel) && normalizedLabels.has(cleanText(cell.title, "").toLowerCase())) {
+              updates.set(cell.id, {
+                ...currentNode(cell.id) ?? cell,
+                title: metadataLabel,
+                description: tableCellDescription(cell, metadataLabel),
+                metadata: metadataWithColumnLabel(cell.metadata, metadataLabel)
+              });
+              rowsToRebuildText.add(row.id);
+            }
+          }
+          continue;
+        }
         for (const [fallbackIndex, cell] of cells.entries()) {
           const columnIndex = tableCellColumnIndex(cell, fallbackIndex);
           const label = columnLabels.get(columnIndex);
-          if (!label || cell.title === label) continue;
+          if (!label) continue;
+          const current = currentNode(cell.id) ?? cell;
+          if (current.title === label && metadataColumnName(current.metadata) === label && (metadataTableColumnName(current.metadata) ?? label) === label) {
+            continue;
+          }
           updates.set(cell.id, {
-            ...currentNode(cell.id) ?? cell,
+            ...current,
             title: label,
-            metadata: {
-              ...cell.metadata ?? {},
-              visualTableRepairColumnLabel: label
-            }
+            description: tableCellDescription(current, label),
+            metadata: metadataWithColumnLabel(current.metadata, label)
           });
+          rowsToRebuildText.add(row.id);
         }
       }
     }
@@ -11334,6 +11406,7 @@ function applyVisualTableRepair(sourceTree, repair) {
               visualTableRepair: "merged_continuation"
             }
           });
+          rowsToRebuildText.add(targetRow.id);
         }
       }
       const nextRowText = appendDistinctText(targetRow.textExcerpt ?? targetRow.description, sourceText);
@@ -11352,6 +11425,21 @@ function applyVisualTableRepair(sourceTree, repair) {
       for (const sourceCell of sourceCells) removeIds.add(sourceCell.id);
     }
   }
+  for (const rowId of rowsToRebuildText) {
+    if (removeIds.has(rowId)) continue;
+    const row = currentNode(rowId);
+    if (!row || row.kind !== "table_row") continue;
+    const cells = (byParent.get(row.id) ?? []).filter((node) => node.kind === "table_cell" && !removeIds.has(node.id)).map((node) => currentNode(node.id) ?? node).sort(
+      (left, right) => tableCellColumnIndex(left, 0) - tableCellColumnIndex(right, 0) || left.order - right.order || left.id.localeCompare(right.id)
+    );
+    const textExcerpt = tableRowTextFromCells(cells);
+    if (!textExcerpt) continue;
+    updates.set(row.id, {
+      ...row,
+      textExcerpt,
+      description: cleanText([row.title, textExcerpt].filter(Boolean).join(" | "), row.description)
+    });
+  }
   if (updates.size === 0 && removeIds.size === 0) return sourceTree;
   const repaired = sourceTree.filter((node) => !removeIds.has(node.id)).map((node) => updates.get(node.id) ?? node);
   return normalizeDocumentSourceTreePaths(normalizeContainerEvidenceFromChildren(repaired));