@athenaintel/react 0.28.0 → 0.30.0

package/dist/tools/tool-uis.js

@@ -2,6 +2,8 @@ import { jsx, jsxs, Fragment } from "react/jsx-runtime";
 import { athenaDiagnostics } from "../diagnostics/athena-diagnostics.js";
 import { memo, useMemo, useCallback, useEffect, useState, useRef } from "react";
 import { GlobeIcon, FilePlusIcon, MailIcon, BookOpenIcon, PresentationIcon, BarChart3Icon, DatabaseIcon, TablePropertiesIcon, TableIcon, FolderOpenIcon, ExternalLinkIcon, AlertTriangleIcon, CodeIcon, TerminalIcon, ImageIcon, PlayIcon, SheetIcon, CameraIcon, ScissorsIcon, SearchIcon, Maximize2Icon, XIcon, FileTextIcon } from "lucide-react";
+import { rangePreviewFromRows } from "../sheet-range/range-preview.js";
+import { RangePreviewTable } from "../sheet-range/range-preview-table.js";
 import { normalizeResult, truncate, ToolCard, ExpandableSection } from "./primitives.js";
 import { SearchFilesToolUI, EditFileToolUI, WriteFileToolUI, ListFilesToolUI, ReadFileToolUI } from "./file-tool-uis.js";
 import { WriteTodosToolUI } from "./write-todos-tool-ui.js";
@@ -2010,47 +2012,27 @@ const RunSqlToolUI = memo(
   RunSqlToolUIImpl
 );
 RunSqlToolUI.displayName = "RunSqlToolUI";
-function parseA1Range(range) {
-  const stripped = range.replace(/^(?:'[^']+'|[^'!]+)!/, "");
-  const match = stripped.match(/^([A-Z]+)(\d+)/i);
-  if (!match) return null;
-  return { startCol: match[1].toUpperCase(), startRow: parseInt(match[2], 10) };
-}
-function colNumToLetter(n) {
-  let result = "";
-  let num = n;
-  while (num > 0) {
-    num--;
-    result = String.fromCharCode(65 + num % 26) + result;
-    num = Math.floor(num / 26);
-  }
-  return result;
-}
-function parseCellsToGrid(cells) {
-  let parsed = null;
+function parseSheetCellRows(cells) {
+  let parsed = cells;
   if (typeof cells === "string") {
     try {
       parsed = JSON.parse(cells);
     } catch {
       return null;
     }
-  } else if (Array.isArray(cells)) {
-    parsed = cells;
-  }
-  if (!parsed || !Array.isArray(parsed)) return null;
-  return parsed.map((row) => {
-    if (!Array.isArray(row)) return [];
-    return row.map((cell) => {
-      if (cell == null) return "";
-      if (typeof cell === "object" && cell !== null) {
-        const c = cell;
-        if (c.formula) return c.formula;
-        if (c.value !== void 0 && c.value !== null) return String(c.value);
-        return "";
-      }
-      return String(cell);
-    });
-  });
+  }
+  if (!Array.isArray(parsed)) return null;
+  return parsed.map((row) => Array.isArray(row) ? row : null);
+}
+function sheetCellText(cell) {
+  if (cell == null) return "";
+  if (typeof cell === "object") {
+    const c = cell;
+    if (c.formula) return c.formula;
+    if (c.value !== void 0 && c.value !== null) return String(c.value);
+    return "";
+  }
+  return String(cell);
 }
 const MAX_SHEET_DISPLAY_ROWS = 30;
 const MAX_LISTED_DV_VIOLATIONS = 3;
@@ -2087,23 +2069,19 @@ const UpdateSheetRangeToolUIImpl = ({
   const isFailure = isComplete && (data == null ? void 0 : data.success) === false;
   const errorMsg = isFailure ? buildSheetUpdateFailureMessage(data, range) : (status == null ? void 0 : status.type) === "incomplete" ? status.error : null;
   const grid = useMemo(
-    () => parseCellsToGrid(typedArgs == null ? void 0 : typedArgs.cells),
+    () => parseSheetCellRows(typedArgs == null ? void 0 : typedArgs.cells),
     [typedArgs == null ? void 0 : typedArgs.cells]
   );
-  const rangeInfo = useMemo(() => parseA1Range(range), [range]);
-  const { headers, displayRows, totalRows } = useMemo(() => {
-    if (!grid || grid.length === 0) {
-      return { headers: [], displayRows: [], totalRows: 0 };
-    }
-    const maxCols = Math.max(...grid.map((r) => r.length));
-    const startColNum = rangeInfo ? rangeInfo.startCol.split("").reduce((acc, c) => acc * 26 + c.charCodeAt(0) - 64, 0) : 1;
-    const hdrs = Array.from(
-      { length: maxCols },
-      (_, i) => colNumToLetter(startColNum + i)
-    );
-    const capped = grid.slice(0, MAX_SHEET_DISPLAY_ROWS);
-    return { headers: hdrs, displayRows: capped, totalRows: grid.length };
-  }, [grid, rangeInfo]);
+  const totalRows = (grid == null ? void 0 : grid.length) ?? 0;
+  const preview = useMemo(
+    () => grid ? rangePreviewFromRows({
+      anchor: range,
+      rows: grid,
+      formatCell: sheetCellText,
+      maxRows: MAX_SHEET_DISPLAY_ROWS
+    }) : null,
+    [grid, range]
+  );
   const openAsset = useAssetPanelStore((s) => s.openAsset);
   const effectiveAssetId = resultAssetId ?? (assetId.startsWith("asset_") ? assetId : null);
   const title = isRunning ? explanation ? `Updating ${range}: ${explanation}` : `Updating sheet range ${range}...` : isFailure ? `Update to ${range || "sheet"} rejected` : explanation ? `Updated ${range}: ${explanation}` : `Updated sheet range ${range}`;
@@ -2120,55 +2098,19 @@ const UpdateSheetRangeToolUIImpl = ({
       badge: isComplete && success && grid ? `${totalRows} ${totalRows === 1 ? "row" : "rows"}` : void 0,
       error: errorMsg,
       children: [
-        headers.length > 0 && displayRows.length > 0 && /* @__PURE__ */ jsxs(
+        preview && /* @__PURE__ */ jsx(
           ExpandableSection,
           {
             label: `Show data (${totalRows} ${totalRows === 1 ? "row" : "rows"})`,
             defaultOpen: totalRows <= 10,
-            children: [
-              /* @__PURE__ */ jsx("div", { className: "overflow-auto max-h-72", children: /* @__PURE__ */ jsxs("table", { className: "w-full text-[11px]", children: [
-                /* @__PURE__ */ jsx("thead", { className: "sticky top-0 bg-muted/60 backdrop-blur", children: /* @__PURE__ */ jsxs("tr", { children: [
-                  /* @__PURE__ */ jsx("th", { className: "w-8 px-1.5 py-1 text-right font-medium text-muted-foreground/60", children: "#" }),
-                  headers.map((h) => /* @__PURE__ */ jsx(
-                    "th",
-                    {
-                      className: "whitespace-nowrap px-2 py-1 text-center font-medium text-muted-foreground",
-                      children: h
-                    },
-                    h
-                  ))
-                ] }) }),
-                /* @__PURE__ */ jsx("tbody", { children: displayRows.map((row, i) => {
-                  const rowNum = ((rangeInfo == null ? void 0 : rangeInfo.startRow) ?? 1) + i;
-                  return /* @__PURE__ */ jsxs("tr", { className: "border-t border-border/30 hover:bg-muted/20", children: [
-                    /* @__PURE__ */ jsx("td", { className: "w-8 px-1.5 py-0.5 text-right tabular-nums text-muted-foreground/60", children: rowNum }),
-                    headers.map((h, j) => {
-                      const val = row[j] ?? "";
-                      const isFormula = typeof val === "string" && val.startsWith("=");
-                      return /* @__PURE__ */ jsx(
-                        "td",
-                        {
-                          className: cn(
-                            "max-w-[180px] truncate whitespace-nowrap px-2 py-0.5 text-foreground/80",
-                            isFormula && "font-mono text-blue-600"
-                          ),
-                          title: val,
-                          children: val || /* @__PURE__ */ jsx("span", { className: "text-muted-foreground/30", children: "—" })
-                        },
-                        h
-                      );
-                    })
-                  ] }, i);
-                }) })
-              ] }) }),
-              totalRows > MAX_SHEET_DISPLAY_ROWS && /* @__PURE__ */ jsxs("div", { className: "px-2 py-1 text-[10px] text-muted-foreground", children: [
-                "Showing first ",
-                MAX_SHEET_DISPLAY_ROWS,
-                " of ",
-                totalRows,
-                " rows"
-              ] })
-            ]
+            children: /* @__PURE__ */ jsx(
+              RangePreviewTable,
+              {
+                preview,
+                maxRows: MAX_SHEET_DISPLAY_ROWS,
+                caption: range ? `Values written to ${range}` : "Values written"
+              }
+            )
           }
         ),
         effectiveAssetId && isComplete && !errorMsg && /* @__PURE__ */ jsx("div", { className: "border-t border-border/40 px-4 py-2", children: /* @__PURE__ */ jsxs(
@@ -2698,13 +2640,13 @@ function useFreshPresignedUrl({
       try {
         const config2 = configRef.current;
         const baseUrl = config2.backendUrl.replace(/\/api\/assistant-ui\/?$/, "");
-        const url = `${baseUrl}/api/asset/fetch-attachment-presigned-url`;
+        const url = `${baseUrl}/api/asset/screenshot-presigned-url`;
         const resp = await sendWithCredentialRenewal(
           config2,
           (headers) => fetch(url, {
             method: "POST",
             headers: { "Content-Type": "application/json", ...headers },
-            body: JSON.stringify({ s3_key: s3Key })
+            body: JSON.stringify({ screenshot_key: s3Key })
           })
         );
         if (!cancelled && resp.ok) {