@athenaintel/react 0.57.1 → 0.59.0

package/dist/tools/tool-uis.d.ts

@@ -36,6 +36,46 @@ export declare const StudioCreateParagraphToolUI: ToolCallMessagePartComponent;
 export declare const StudioOpenDocumentToolUI: ToolCallMessagePartComponent;
 export declare const CreateEmailDraftToolUI: ToolCallMessagePartComponent;
 export declare const CreateNotebookToolUI: ToolCallMessagePartComponent;
+/**
+ * The progress a studio executor reports while it runs (its preliminary
+ * result): the phase it is in and, when it counts, how far along.
+ */
+interface StudioExecutionProgress {
+    phase: "preparing" | "executing" | "finalizing";
+    message?: string;
+    completed?: number;
+    total?: number;
+}
+/** A preliminary `studio_execution_progress` result, or `undefined` for anything else. */
+export declare function readStudioExecutionProgress(result: unknown): StudioExecutionProgress | undefined;
+/** A studio executor's running words, phase by phase, as Athena on the web words them. */
+export interface StudioExecutionCopy {
+    generating: string;
+    preparing: string;
+    executing: string;
+    finalizing: string;
+}
+/** The running title: the executor's own message, else its phase's words, with how far along. */
+export declare function studioExecutionTitle({ copy, progress, }: {
+    copy: StudioExecutionCopy;
+    progress: StudioExecutionProgress | undefined;
+}): string;
+/**
+ * The asset a studio executor's code edits: the one its SDK constructor
+ * opens (`Workbook("asset_…")`), else the one asset id quoted in the code.
+ * Code that opens, or names, several assets gives none: the card cannot tell
+ * which one was written, and offers no button rather than the wrong asset.
+ * The executors take no asset argument: the agent names the asset in its code.
+ */
+export declare function studioCodeAssetId(code: string | undefined): string | null;
+/**
+ * `execute_spreadsheet_code`: the Spreadsheet Studio code a run executed in a
+ * workbook, as Athena on the web shows it: its phase while it runs, then its
+ * summary, code, output or exception, and the workbook it edited.
+ */
+export declare const ExecuteSpreadsheetCodeToolUI: ToolCallMessagePartComponent;
+/** `execute_word_document_code`: the Word Studio code a run executed in a document, likewise. */
+export declare const ExecuteWordDocumentCodeToolUI: ToolCallMessagePartComponent;
 export declare const RunPythonCodeToolUI: ToolCallMessagePartComponent;
 export declare const ExecutePresentationCodeToolUI: ToolCallMessagePartComponent;
 export declare const OpenAssetToolUI: ToolCallMessagePartComponent;

package/dist/tools/tool-uis.js

@@ -1,7 +1,7 @@
 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 { memo, useMemo, useEffect, useCallback, useState, useRef } from "react";
+import { GlobeIcon, FilePlusIcon, MailIcon, BookOpenIcon, PresentationIcon, BarChart3Icon, DatabaseIcon, SheetIcon, FileTextIcon, TablePropertiesIcon, TableIcon, FolderOpenIcon, ExternalLinkIcon, AlertTriangleIcon, CodeIcon, PlayIcon, ScissorsIcon, CameraIcon, TerminalIcon, SearchIcon, ImageIcon, Maximize2Icon, XIcon } 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";
@@ -11,8 +11,18 @@ import { InstallPackageToolUI } from "./misc-tool-uis.js";
 import { SubagentTaskToolUI } from "./subagent-task-tool-ui.js";
 import { TestRecordingStopToolUI, TestRecordingMarkToolUI, TestRecordingStartToolUI } from "./test-recording-tool-uis.js";
 import { useAssetPanelStore } from "../assets/asset-panel-store.js";
-import { useAthenaConfig } from "../provider/AthenaContext.js";
-import { sendWithCredentialRenewal } from "../runtime/credential-renewal.js";
+import { CapturePageScreenshotToolUI, CaptureSheetScreenshotToolUI, ScreenshotQueryScope, useScreenshotUrl } from "./studio-screenshot-tool-uis.js";
+import { QuerySheetRangeToolUI } from "./sheet-query-tool-ui.js";
+import { StopBrowserSessionToolUI, CreateNewBrowseSessionToolUI, GetCurrentUrlToolUI, GetPageTextToolUI, GoToUrlToolUI, BrowserBatchToolUI, ComputerToolUI } from "./browser-tool-uis.js";
+import { ComputerAssetDeployToolUI, ComputerAssetGetStatusToolUI, ComputerAssetEnsureSandboxToolUI, ComputerAssetCreateToolUI } from "./computer-asset-tool-uis.js";
+import { ConvertDocumentToPdfToolUI, ClearDocumentToolUI, DeleteDocumentBlocksToolUI, InsertDocumentContentAfterToolUI, InsertDocumentContentBeforeToolUI, ReplaceDocumentContentToolUI } from "./document-edit-tool-uis.js";
+import { ProjectLinksToolUI, ListContentsToolUI, SearchAssetsToolUI } from "./drive-tool-uis.js";
+import { CalendarEventsToolUI, EmailThreadToolUI, SendAthenaEmailToolUI } from "./email-calendar-tool-uis.js";
+import { GetObjectLinksToolUI, SearchObjectsToolUI, ListObjectTypesToolUI } from "./ontology-tool-uis.js";
+import { AsyncTaskToolUI, EvalToolUI } from "./orchestration-tool-uis.js";
+import { UpdateSlideShapesToolUI, SetSlideNotesToolUI, ManageSlidesToolUI, CaptureSlideScreenshotsToolUI } from "./presentation-edit-tool-uis.js";
+import { GetPmRoadmapToolUI, PmDocumentsToolUI } from "./project-tool-uis.js";
+import { DeleteToolUI, ExecuteToolUI } from "./sandbox-tool-uis.js";
 import { cn } from "cnfast";
 function isSdkToolDebugEnabled() {
   var _a, _b;
@@ -1237,12 +1247,50 @@ function highlightPython(code) {
     return result;
   }).join("\n");
 }
-const RunPythonCodeToolUIImpl = ({
+function readStudioExecutionProgress(result) {
+  const data = 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) ?? "";
@@ -1250,28 +1298,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 = useMemo(
     () => isComplete ? parsePythonResult(result) : null,
     [result, isComplete]
   );
   const openAsset = useAssetPanelStore((s) => s.openAsset);
+  const editedAssetId = 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__ */ jsxs(
     ToolCard,
     {
-      icon: 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__ */ jsx("div", { className: "border-t border-border/40 px-4 py-2", children: /* @__PURE__ */ 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__ */ jsx(ExternalLinkIcon, { className: "size-3" }),
+              spec.editedAsset.label
+            ]
+          }
+        ) }) : null,
         code && /* @__PURE__ */ jsx(ExpandableSection, { label: "Show code", children: /* @__PURE__ */ jsx(SyntaxHighlightedCode, { code }) }),
         isComplete && hasOutput && /* @__PURE__ */ jsxs("div", { className: "border-t border-border/40 px-4 py-2.5", children: [
           /* @__PURE__ */ jsxs("div", { className: "flex items-center gap-1.5 mb-1", children: [
@@ -1283,7 +1352,7 @@ const RunPythonCodeToolUIImpl = ({
         isComplete && hasError && /* @__PURE__ */ jsxs("div", { className: "border-t border-border/40 bg-destructive/10 px-4 py-2.5", children: [
           /* @__PURE__ */ jsxs("div", { className: "flex items-center gap-1.5 mb-1", children: [
             /* @__PURE__ */ jsx(AlertTriangleIcon, { className: "size-3 text-destructive" }),
-            /* @__PURE__ */ jsx("span", { className: "text-[11px] font-medium text-destructive", children: parsed.exception ? `${parsed.exception.name}: ${parsed.exception.value}` : "Error" })
+            /* @__PURE__ */ jsx("span", { className: "text-[11px] font-medium text-destructive", children: parsed.exception ? "Traceback" : "Error" })
           ] }),
           /* @__PURE__ */ 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 })
         ] }),
@@ -1321,7 +1390,41 @@ const RunPythonCodeToolUIImpl = ({
       ]
     }
   );
+}
+const RUN_PYTHON_SPEC = { icon: CodeIcon, runningTitle: () => "Running code..." };
+const RunPythonCodeToolUIImpl = (props) => /* @__PURE__ */ 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: SheetIcon,
+  runningTitle: (result) => studioExecutionTitle({ copy: SPREADSHEET_EXECUTION_COPY, progress: readStudioExecutionProgress(result) }),
+  editedAsset: { assetId: studioCodeAssetId, type: "spreadsheet", label: "Open workbook" }
+};
+const WORD_CODE_SPEC = {
+  icon: FileTextIcon,
+  runningTitle: (result) => studioExecutionTitle({ copy: WORD_EXECUTION_COPY, progress: readStudioExecutionProgress(result) }),
+  editedAsset: { assetId: studioCodeAssetId, type: "document", label: "Open document" }
+};
+const ExecuteSpreadsheetCodeToolUIImpl = (props) => /* @__PURE__ */ jsx(CodeExecutionToolCard, { ...props, spec: SPREADSHEET_CODE_SPEC });
+const ExecuteWordDocumentCodeToolUIImpl = (props) => /* @__PURE__ */ jsx(CodeExecutionToolCard, { ...props, spec: WORD_CODE_SPEC });
+const ExecuteSpreadsheetCodeToolUI = memo(
+  ExecuteSpreadsheetCodeToolUIImpl
+);
+ExecuteSpreadsheetCodeToolUI.displayName = "ExecuteSpreadsheetCodeToolUI";
+const ExecuteWordDocumentCodeToolUI = memo(
+  ExecuteWordDocumentCodeToolUIImpl
+);
+ExecuteWordDocumentCodeToolUI.displayName = "ExecuteWordDocumentCodeToolUI";
 const RunPythonCodeToolUI = memo(
   RunPythonCodeToolUIImpl
 );
@@ -2626,46 +2729,6 @@ function normalizeContentResult(result) {
   }
   return [];
 }
-function useFreshPresignedUrl({
-  s3Key,
-  fallbackUrl
-}) {
-  const [freshUrl, setFreshUrl] = useState(null);
-  const config = useAthenaConfig();
-  const configRef = useRef(config);
-  configRef.current = config;
-  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 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;
@@ -2709,7 +2772,7 @@ const CaptureSlideScreenshotToolUIImpl = ({
     () => getSlideScreenshotResult(result),
     [result]
   );
-  const freshImageUrl = useFreshPresignedUrl({ s3Key, fallbackUrl: imageUrl });
+  const freshImageUrl = 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";
@@ -2784,8 +2847,9 @@ const CaptureSlideScreenshotToolUIImpl = ({
     }
   );
 };
+const ScopedCaptureSlideScreenshotToolUI = (props) => /* @__PURE__ */ jsx(ScreenshotQueryScope, { children: /* @__PURE__ */ jsx(CaptureSlideScreenshotToolUIImpl, { ...props }) });
 const CaptureSlideScreenshotToolUI = memo(
-  CaptureSlideScreenshotToolUIImpl
+  ScopedCaptureSlideScreenshotToolUI
 );
 CaptureSlideScreenshotToolUI.displayName = "CaptureSlideScreenshotToolUI";
 const OpenBrowserToolUIImpl = ({
@@ -2915,15 +2979,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: CaptureSheetScreenshotToolUI,
+  capture_page_screenshot: CapturePageScreenshotToolUI,
   // Spreadsheet toolkit
   bulk_format_sheet_range: BulkFormatSheetRangeToolUI,
   set_row_column_dimensions: SetRowColumnDimensionsToolUI,
   update_sheet_range: UpdateSheetRangeToolUI,
+  query_sheet_range: QuerySheetRangeToolUI,
   // Database toolkit
   describe_database: DescribeDatabaseToolUI,
   list_database_tables: ListDatabaseTablesToolUI,
@@ -2947,7 +3017,60 @@ const TOOL_UI_REGISTRY = {
   // UI Testing toolkit: the annotated recording of an agent checking a UI change.
   test_recording_start: TestRecordingStartToolUI,
   test_recording_mark: TestRecordingMarkToolUI,
-  test_recording_stop: TestRecordingStopToolUI
+  test_recording_stop: 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: ExecuteToolUI,
+  delete: DeleteToolUI,
+  // Code interpreter, and background tasks (coding workers, agent sessions).
+  eval: EvalToolUI,
+  start_async_task: AsyncTaskToolUI,
+  check_async_task: AsyncTaskToolUI,
+  update_async_task: AsyncTaskToolUI,
+  cancel_async_task: AsyncTaskToolUI,
+  ask_agent: AsyncTaskToolUI,
+  // Athena Drive: asset search, a folder's contents, a project's links.
+  search_assets: SearchAssetsToolUI,
+  list_contents: ListContentsToolUI,
+  project_links: ProjectLinksToolUI,
+  // Ontology
+  list_object_types: ListObjectTypesToolUI,
+  search_objects: SearchObjectsToolUI,
+  get_object_links: GetObjectLinksToolUI,
+  // Project manager
+  pm_documents: PmDocumentsToolUI,
+  get_pm_roadmap: GetPmRoadmapToolUI,
+  // Hosted browser and its browse sessions
+  computer: ComputerToolUI,
+  browser_batch: BrowserBatchToolUI,
+  go_to_url: GoToUrlToolUI,
+  get_page_text: GetPageTextToolUI,
+  get_current_url: GetCurrentUrlToolUI,
+  create_new_browse_session: CreateNewBrowseSessionToolUI,
+  stop_browser_session: StopBrowserSessionToolUI,
+  // Computer assets
+  computer_asset_create: ComputerAssetCreateToolUI,
+  computer_asset_ensure_sandbox: ComputerAssetEnsureSandboxToolUI,
+  computer_asset_get_status: ComputerAssetGetStatusToolUI,
+  computer_asset_deploy: ComputerAssetDeployToolUI,
+  // Athena document edits
+  replace_markdown_in_athena_document: ReplaceDocumentContentToolUI,
+  insert_markdown_before_block: InsertDocumentContentBeforeToolUI,
+  insert_markdown_after_block: InsertDocumentContentAfterToolUI,
+  delete_blocks_from_athena_document: DeleteDocumentBlocksToolUI,
+  clear_athena_document: ClearDocumentToolUI,
+  convert_athena_doc_to_pdf: ConvertDocumentToPdfToolUI,
+  // Email sent from Athena, mail threads and calendar events
+  send_email_from_athena: SendAthenaEmailToolUI,
+  unified_email_get_thread: EmailThreadToolUI,
+  unified_email_get_message: EmailThreadToolUI,
+  unified_search_calendar_events: CalendarEventsToolUI,
+  // Deck edits and the contact sheet of several slides
+  capture_slide_screenshots: CaptureSlideScreenshotsToolUI,
+  manage_slides: ManageSlidesToolUI,
+  set_slide_notes: SetSlideNotesToolUI,
+  update_slide_shapes: UpdateSlideShapesToolUI
 };
 export {
   BrowseToolUI,
@@ -2963,6 +3086,8 @@ export {
   EmailAccountsToolUI,
   EmailSearchToolUI,
   ExecutePresentationCodeToolUI,
+  ExecuteSpreadsheetCodeToolUI,
+  ExecuteWordDocumentCodeToolUI,
   ExpandableSection,
   GetDatabaseTableSchemaToolUI,
   ListDatabaseTablesToolUI,
@@ -2990,7 +3115,10 @@ export {
   humanizeEmailQuery,
   normalizeResult,
   parseEmailAccounts,
+  readStudioExecutionProgress,
   resetAssetAutoOpen,
+  studioCodeAssetId,
+  studioExecutionTitle,
   truncate
 };
 //# sourceMappingURL=tool-uis.js.map