@athenaintel/react 0.57.2 → 0.59.0

package/dist/tools/tool-uis.cjs

@@ -13,8 +13,18 @@ const miscToolUis = require("./misc-tool-uis.cjs");
 const subagentTaskToolUi = require("./subagent-task-tool-ui.cjs");
 const testRecordingToolUis = require("./test-recording-tool-uis.cjs");
 const assetPanelStore = require("../assets/asset-panel-store.cjs");
-const AthenaContext = require("../provider/AthenaContext.cjs");
-const credentialRenewal = require("../runtime/credential-renewal.cjs");
+const studioScreenshotToolUis = require("./studio-screenshot-tool-uis.cjs");
+const sheetQueryToolUi = require("./sheet-query-tool-ui.cjs");
+const browserToolUis = require("./browser-tool-uis.cjs");
+const computerAssetToolUis = require("./computer-asset-tool-uis.cjs");
+const documentEditToolUis = require("./document-edit-tool-uis.cjs");
+const driveToolUis = require("./drive-tool-uis.cjs");
+const emailCalendarToolUis = require("./email-calendar-tool-uis.cjs");
+const ontologyToolUis = require("./ontology-tool-uis.cjs");
+const orchestrationToolUis = require("./orchestration-tool-uis.cjs");
+const presentationEditToolUis = require("./presentation-edit-tool-uis.cjs");
+const projectToolUis = require("./project-tool-uis.cjs");
+const sandboxToolUis = require("./sandbox-tool-uis.cjs");
 const cnfast = require("cnfast");
 function isSdkToolDebugEnabled() {
   var _a, _b;
@@ -1239,12 +1249,50 @@ function highlightPython(code) {
     return result;
   }).join("\n");
 }
-const RunPythonCodeToolUIImpl = ({
+function readStudioExecutionProgress(result) {
+  const data = primitives.normalizeResult(result);
+  if (!data || data.type !== "studio_execution_progress") return void 0;
+  const phase = data.phase;
+  if (phase !== "preparing" && phase !== "executing" && phase !== "finalizing") return void 0;
+  return {
+    phase,
+    ...typeof data.message === "string" ? { message: data.message } : {},
+    ...typeof data.completed === "number" ? { completed: data.completed } : {},
+    ...typeof data.total === "number" && data.total > 0 ? { total: data.total } : {}
+  };
+}
+function studioExecutionTitle({
+  copy,
+  progress
+}) {
+  if (!progress) return copy.generating;
+  const words = progress.message ?? copy[progress.phase];
+  if (progress.completed === void 0 || progress.total === void 0) return words;
+  const percent = Math.round(Math.min(100, Math.max(0, progress.completed / progress.total * 100)));
+  return `${words} ${percent}%`;
+}
+const STUDIO_ASSET_ID_SOURCE = "asset_[0-9a-fA-F]{8}-[0-9a-fA-F]{4}-[0-9a-fA-F]{4}-[0-9a-fA-F]{4}-[0-9a-fA-F]{12}(?:_[^\"'`\\s),\\]]+)?";
+const STUDIO_TARGET_ASSET_ID_RE = new RegExp(
+  String.raw`\b(?:Document|Workbook|Presentation|load_workbook)\s*\(\s*(?:[A-Za-z_]+\s*=\s*)?(["'\`])(${STUDIO_ASSET_ID_SOURCE})\1`,
+  "g"
+);
+const STUDIO_ANY_ASSET_ID_RE = new RegExp(String.raw`(["'\`])(${STUDIO_ASSET_ID_SOURCE})\1`, "g");
+function distinctAssetIds(code, pattern) {
+  return [...new Set(Array.from(code.matchAll(pattern), (match) => match[2]))];
+}
+function studioCodeAssetId(code) {
+  if (!code) return null;
+  const opened = distinctAssetIds(code, STUDIO_TARGET_ASSET_ID_RE);
+  const candidates = opened.length > 0 ? opened : distinctAssetIds(code, STUDIO_ANY_ASSET_ID_RE);
+  return candidates.length === 1 ? candidates[0] : null;
+}
+function CodeExecutionToolCard({
   toolName,
   args,
   result,
-  status
-}) => {
+  status,
+  spec
+}) {
   var _a;
   const typedArgs = args;
   const code = (typedArgs == null ? void 0 : typedArgs.code) ?? "";
@@ -1252,28 +1300,49 @@ const RunPythonCodeToolUIImpl = ({
   const lines = code ? code.split("\n").length : 0;
   const isRunning = (status == null ? void 0 : status.type) === "running";
   const isComplete = (status == null ? void 0 : status.type) === "complete";
-  const errorMsg = (status == null ? void 0 : status.type) === "incomplete" ? status.error : null;
   const parsed = require$$0.useMemo(
     () => isComplete ? parsePythonResult(result) : null,
     [result, isComplete]
   );
   const openAsset = assetPanelStore.useAssetPanelStore((s) => s.openAsset);
+  const editedAssetId = require$$0.useMemo(
+    () => spec.editedAsset ? spec.editedAsset.assetId(code) : null,
+    [spec.editedAsset, code]
+  );
   const outputText = parsed ? getExecutionTextOutput(parsed) : null;
   const hasOutput = Boolean(outputText);
   const hasError = parsed && (parsed.error || parsed.exception);
   const hasImage = parsed == null ? void 0 : parsed.imagePng;
   const hasPlotly = parsed == null ? void 0 : parsed.plotlyJson;
+  const raised = (parsed == null ? void 0 : parsed.exception) ? `${parsed.exception.name}: ${parsed.exception.value}` : (parsed == null ? void 0 : parsed.error) ? parsed.error.split("\n").find((line) => line.trim() !== "") ?? "Error" : null;
+  const errorMsg = (status == null ? void 0 : status.type) === "incomplete" ? status.error : raised;
+  const cardStatus = raised ? "incomplete" : (status == null ? void 0 : status.type) ?? "complete";
   return /* @__PURE__ */ jsxRuntime.jsxs(
     primitives.ToolCard,
     {
-      icon: lucideReact.CodeIcon,
-      status: (status == null ? void 0 : status.type) ?? "complete",
-      title: isRunning ? "Running code..." : summary || `Executed ${lines} ${lines !== 1 ? "lines" : "line"}`,
+      icon: spec.icon,
+      status: cardStatus,
+      title: isRunning ? spec.runningTitle(result) : summary || `Executed ${lines} ${lines !== 1 ? "lines" : "line"}`,
       toolName,
       args: typedArgs,
       result,
       error: errorMsg,
       children: [
+        spec.editedAsset && editedAssetId && isComplete && !hasError ? /* @__PURE__ */ jsxRuntime.jsx("div", { className: "border-t border-border/40 px-4 py-2", children: /* @__PURE__ */ jsxRuntime.jsxs(
+          "button",
+          {
+            type: "button",
+            onClick: () => {
+              var _a2;
+              return openAsset(editedAssetId, { type: (_a2 = spec.editedAsset) == null ? void 0 : _a2.type });
+            },
+            className: "flex items-center gap-1.5 rounded-md border border-border/60 px-3 py-1.5 text-xs font-medium text-muted-foreground transition-colors hover:bg-muted/50 hover:text-foreground",
+            children: [
+              /* @__PURE__ */ jsxRuntime.jsx(lucideReact.ExternalLinkIcon, { className: "size-3" }),
+              spec.editedAsset.label
+            ]
+          }
+        ) }) : null,
         code && /* @__PURE__ */ jsxRuntime.jsx(primitives.ExpandableSection, { label: "Show code", children: /* @__PURE__ */ jsxRuntime.jsx(SyntaxHighlightedCode, { code }) }),
         isComplete && hasOutput && /* @__PURE__ */ jsxRuntime.jsxs("div", { className: "border-t border-border/40 px-4 py-2.5", children: [
           /* @__PURE__ */ jsxRuntime.jsxs("div", { className: "flex items-center gap-1.5 mb-1", children: [
@@ -1285,7 +1354,7 @@ const RunPythonCodeToolUIImpl = ({
         isComplete && hasError && /* @__PURE__ */ jsxRuntime.jsxs("div", { className: "border-t border-border/40 bg-destructive/10 px-4 py-2.5", children: [
           /* @__PURE__ */ jsxRuntime.jsxs("div", { className: "flex items-center gap-1.5 mb-1", children: [
             /* @__PURE__ */ jsxRuntime.jsx(lucideReact.AlertTriangleIcon, { className: "size-3 text-destructive" }),
-            /* @__PURE__ */ jsxRuntime.jsx("span", { className: "text-[11px] font-medium text-destructive", children: parsed.exception ? `${parsed.exception.name}: ${parsed.exception.value}` : "Error" })
+            /* @__PURE__ */ jsxRuntime.jsx("span", { className: "text-[11px] font-medium text-destructive", children: parsed.exception ? "Traceback" : "Error" })
           ] }),
           /* @__PURE__ */ jsxRuntime.jsx("pre", { className: "max-h-48 overflow-auto whitespace-pre-wrap break-words rounded-md bg-destructive/5 p-2 text-[11px] leading-relaxed font-mono text-destructive/80", children: ((_a = parsed.exception) == null ? void 0 : _a.traceback) ?? parsed.error })
         ] }),
@@ -1323,7 +1392,41 @@ const RunPythonCodeToolUIImpl = ({
       ]
     }
   );
+}
+const RUN_PYTHON_SPEC = { icon: lucideReact.CodeIcon, runningTitle: () => "Running code..." };
+const RunPythonCodeToolUIImpl = (props) => /* @__PURE__ */ jsxRuntime.jsx(CodeExecutionToolCard, { ...props, spec: RUN_PYTHON_SPEC });
+const SPREADSHEET_EXECUTION_COPY = {
+  generating: "Generating spreadsheet code...",
+  preparing: "Preparing Spreadsheet Studio...",
+  executing: "Executing spreadsheet code...",
+  finalizing: "Finalizing spreadsheet changes..."
+};
+const WORD_EXECUTION_COPY = {
+  generating: "Generating document code...",
+  preparing: "Preparing Word Studio...",
+  executing: "Executing document code...",
+  finalizing: "Finalizing document changes..."
 };
+const SPREADSHEET_CODE_SPEC = {
+  icon: lucideReact.SheetIcon,
+  runningTitle: (result) => studioExecutionTitle({ copy: SPREADSHEET_EXECUTION_COPY, progress: readStudioExecutionProgress(result) }),
+  editedAsset: { assetId: studioCodeAssetId, type: "spreadsheet", label: "Open workbook" }
+};
+const WORD_CODE_SPEC = {
+  icon: lucideReact.FileTextIcon,
+  runningTitle: (result) => studioExecutionTitle({ copy: WORD_EXECUTION_COPY, progress: readStudioExecutionProgress(result) }),
+  editedAsset: { assetId: studioCodeAssetId, type: "document", label: "Open document" }
+};
+const ExecuteSpreadsheetCodeToolUIImpl = (props) => /* @__PURE__ */ jsxRuntime.jsx(CodeExecutionToolCard, { ...props, spec: SPREADSHEET_CODE_SPEC });
+const ExecuteWordDocumentCodeToolUIImpl = (props) => /* @__PURE__ */ jsxRuntime.jsx(CodeExecutionToolCard, { ...props, spec: WORD_CODE_SPEC });
+const ExecuteSpreadsheetCodeToolUI = require$$0.memo(
+  ExecuteSpreadsheetCodeToolUIImpl
+);
+ExecuteSpreadsheetCodeToolUI.displayName = "ExecuteSpreadsheetCodeToolUI";
+const ExecuteWordDocumentCodeToolUI = require$$0.memo(
+  ExecuteWordDocumentCodeToolUIImpl
+);
+ExecuteWordDocumentCodeToolUI.displayName = "ExecuteWordDocumentCodeToolUI";
 const RunPythonCodeToolUI = require$$0.memo(
   RunPythonCodeToolUIImpl
 );
@@ -2628,46 +2731,6 @@ function normalizeContentResult(result) {
   }
   return [];
 }
-function useFreshPresignedUrl({
-  s3Key,
-  fallbackUrl
-}) {
-  const [freshUrl, setFreshUrl] = require$$0.useState(null);
-  const config = AthenaContext.useAthenaConfig();
-  const configRef = require$$0.useRef(config);
-  configRef.current = config;
-  require$$0.useEffect(() => {
-    setFreshUrl(null);
-    if (!s3Key) return;
-    let cancelled = false;
-    (async () => {
-      try {
-        const config2 = configRef.current;
-        const baseUrl = config2.backendUrl.replace(/\/api\/assistant-ui\/?$/, "");
-        const url = `${baseUrl}/api/asset/screenshot-presigned-url`;
-        const resp = await credentialRenewal.sendWithCredentialRenewal(
-          config2,
-          (headers) => fetch(url, {
-            method: "POST",
-            headers: { "Content-Type": "application/json", ...headers },
-            body: JSON.stringify({ screenshot_key: s3Key })
-          })
-        );
-        if (!cancelled && resp.ok) {
-          const data = await resp.json();
-          if (!cancelled && data.presigned_url) {
-            setFreshUrl(data.presigned_url);
-          }
-        }
-      } catch {
-      }
-    })();
-    return () => {
-      cancelled = true;
-    };
-  }, [s3Key]);
-  return freshUrl ?? fallbackUrl;
-}
 function getSlideScreenshotResult(result) {
   let text = null;
   let imageUrl = null;
@@ -2711,7 +2774,7 @@ const CaptureSlideScreenshotToolUIImpl = ({
     () => getSlideScreenshotResult(result),
     [result]
   );
-  const freshImageUrl = useFreshPresignedUrl({ s3Key, fallbackUrl: imageUrl });
+  const freshImageUrl = studioScreenshotToolUis.useScreenshotUrl({ s3Key, fallbackUrl: imageUrl });
   const isRunning = (status == null ? void 0 : status.type) === "running";
   const isComplete = (status == null ? void 0 : status.type) === "complete" || !status;
   const isCancelled = (status == null ? void 0 : status.type) === "incomplete" && status.reason === "cancelled";
@@ -2786,8 +2849,9 @@ const CaptureSlideScreenshotToolUIImpl = ({
     }
   );
 };
+const ScopedCaptureSlideScreenshotToolUI = (props) => /* @__PURE__ */ jsxRuntime.jsx(studioScreenshotToolUis.ScreenshotQueryScope, { children: /* @__PURE__ */ jsxRuntime.jsx(CaptureSlideScreenshotToolUIImpl, { ...props }) });
 const CaptureSlideScreenshotToolUI = require$$0.memo(
-  CaptureSlideScreenshotToolUIImpl
+  ScopedCaptureSlideScreenshotToolUI
 );
 CaptureSlideScreenshotToolUI.displayName = "CaptureSlideScreenshotToolUI";
 const OpenBrowserToolUIImpl = ({
@@ -2917,15 +2981,21 @@ const TOOL_UI_REGISTRY = {
   OpenDocument: StudioOpenDocumentToolUI,
   CreateParagraph: StudioCreateParagraphToolUI,
   execute_presentation_code: ExecutePresentationCodeToolUI,
+  // Spreadsheet and Word Studio code runs, as Athena on the web shows them.
+  execute_spreadsheet_code: ExecuteSpreadsheetCodeToolUI,
+  execute_word_document_code: ExecuteWordDocumentCodeToolUI,
   run_python_code: RunPythonCodeToolUI,
   open_asset_in_workspace: OpenAssetToolUI,
   // Media capture
   capture_moment: CaptureMomentToolUI,
   capture_slide_screenshot: CaptureSlideScreenshotToolUI,
+  capture_sheet_screenshot: studioScreenshotToolUis.CaptureSheetScreenshotToolUI,
+  capture_page_screenshot: studioScreenshotToolUis.CapturePageScreenshotToolUI,
   // Spreadsheet toolkit
   bulk_format_sheet_range: BulkFormatSheetRangeToolUI,
   set_row_column_dimensions: SetRowColumnDimensionsToolUI,
   update_sheet_range: UpdateSheetRangeToolUI,
+  query_sheet_range: sheetQueryToolUi.QuerySheetRangeToolUI,
   // Database toolkit
   describe_database: DescribeDatabaseToolUI,
   list_database_tables: ListDatabaseTablesToolUI,
@@ -2949,7 +3019,60 @@ const TOOL_UI_REGISTRY = {
   // UI Testing toolkit: the annotated recording of an agent checking a UI change.
   test_recording_start: testRecordingToolUis.TestRecordingStartToolUI,
   test_recording_mark: testRecordingToolUis.TestRecordingMarkToolUI,
-  test_recording_stop: testRecordingToolUis.TestRecordingStopToolUI
+  test_recording_stop: testRecordingToolUis.TestRecordingStopToolUI,
+  // The cards below draw these tools as Athena on the web does, from the
+  // call's own arguments and result (no fetch but a screenshot's fresh URL).
+  // Deep-agent sandbox: a shell command and its output, a deleted path.
+  execute: sandboxToolUis.ExecuteToolUI,
+  delete: sandboxToolUis.DeleteToolUI,
+  // Code interpreter, and background tasks (coding workers, agent sessions).
+  eval: orchestrationToolUis.EvalToolUI,
+  start_async_task: orchestrationToolUis.AsyncTaskToolUI,
+  check_async_task: orchestrationToolUis.AsyncTaskToolUI,
+  update_async_task: orchestrationToolUis.AsyncTaskToolUI,
+  cancel_async_task: orchestrationToolUis.AsyncTaskToolUI,
+  ask_agent: orchestrationToolUis.AsyncTaskToolUI,
+  // Athena Drive: asset search, a folder's contents, a project's links.
+  search_assets: driveToolUis.SearchAssetsToolUI,
+  list_contents: driveToolUis.ListContentsToolUI,
+  project_links: driveToolUis.ProjectLinksToolUI,
+  // Ontology
+  list_object_types: ontologyToolUis.ListObjectTypesToolUI,
+  search_objects: ontologyToolUis.SearchObjectsToolUI,
+  get_object_links: ontologyToolUis.GetObjectLinksToolUI,
+  // Project manager
+  pm_documents: projectToolUis.PmDocumentsToolUI,
+  get_pm_roadmap: projectToolUis.GetPmRoadmapToolUI,
+  // Hosted browser and its browse sessions
+  computer: browserToolUis.ComputerToolUI,
+  browser_batch: browserToolUis.BrowserBatchToolUI,
+  go_to_url: browserToolUis.GoToUrlToolUI,
+  get_page_text: browserToolUis.GetPageTextToolUI,
+  get_current_url: browserToolUis.GetCurrentUrlToolUI,
+  create_new_browse_session: browserToolUis.CreateNewBrowseSessionToolUI,
+  stop_browser_session: browserToolUis.StopBrowserSessionToolUI,
+  // Computer assets
+  computer_asset_create: computerAssetToolUis.ComputerAssetCreateToolUI,
+  computer_asset_ensure_sandbox: computerAssetToolUis.ComputerAssetEnsureSandboxToolUI,
+  computer_asset_get_status: computerAssetToolUis.ComputerAssetGetStatusToolUI,
+  computer_asset_deploy: computerAssetToolUis.ComputerAssetDeployToolUI,
+  // Athena document edits
+  replace_markdown_in_athena_document: documentEditToolUis.ReplaceDocumentContentToolUI,
+  insert_markdown_before_block: documentEditToolUis.InsertDocumentContentBeforeToolUI,
+  insert_markdown_after_block: documentEditToolUis.InsertDocumentContentAfterToolUI,
+  delete_blocks_from_athena_document: documentEditToolUis.DeleteDocumentBlocksToolUI,
+  clear_athena_document: documentEditToolUis.ClearDocumentToolUI,
+  convert_athena_doc_to_pdf: documentEditToolUis.ConvertDocumentToPdfToolUI,
+  // Email sent from Athena, mail threads and calendar events
+  send_email_from_athena: emailCalendarToolUis.SendAthenaEmailToolUI,
+  unified_email_get_thread: emailCalendarToolUis.EmailThreadToolUI,
+  unified_email_get_message: emailCalendarToolUis.EmailThreadToolUI,
+  unified_search_calendar_events: emailCalendarToolUis.CalendarEventsToolUI,
+  // Deck edits and the contact sheet of several slides
+  capture_slide_screenshots: presentationEditToolUis.CaptureSlideScreenshotsToolUI,
+  manage_slides: presentationEditToolUis.ManageSlidesToolUI,
+  set_slide_notes: presentationEditToolUis.SetSlideNotesToolUI,
+  update_slide_shapes: presentationEditToolUis.UpdateSlideShapesToolUI
 };
 exports.ExpandableSection = primitives.ExpandableSection;
 exports.ToolCard = primitives.ToolCard;
@@ -2968,6 +3091,8 @@ exports.DescribeDatabaseToolUI = DescribeDatabaseToolUI;
 exports.EmailAccountsToolUI = EmailAccountsToolUI;
 exports.EmailSearchToolUI = EmailSearchToolUI;
 exports.ExecutePresentationCodeToolUI = ExecutePresentationCodeToolUI;
+exports.ExecuteSpreadsheetCodeToolUI = ExecuteSpreadsheetCodeToolUI;
+exports.ExecuteWordDocumentCodeToolUI = ExecuteWordDocumentCodeToolUI;
 exports.GetDatabaseTableSchemaToolUI = GetDatabaseTableSchemaToolUI;
 exports.ListDatabaseTablesToolUI = ListDatabaseTablesToolUI;
 exports.OpenAssetToolUI = OpenAssetToolUI;
@@ -2992,5 +3117,8 @@ exports.createAssetToolUI = createAssetToolUI;
 exports.formatEmailDate = formatEmailDate;
 exports.humanizeEmailQuery = humanizeEmailQuery;
 exports.parseEmailAccounts = parseEmailAccounts;
+exports.readStudioExecutionProgress = readStudioExecutionProgress;
 exports.resetAssetAutoOpen = resetAssetAutoOpen;
+exports.studioCodeAssetId = studioCodeAssetId;
+exports.studioExecutionTitle = studioExecutionTitle;
 //# sourceMappingURL=tool-uis.cjs.map