@athenaintel/react 0.57.2 → 0.60.0

package/dist/tools/studio-screenshot-tool-uis.cjs

@@ -0,0 +1,262 @@
+"use strict";
+Object.defineProperty(exports, Symbol.toStringTag, { value: "Module" });
+const jsxRuntime = require("react/jsx-runtime");
+const reactQuery = require("@tanstack/react-query");
+const lucideReact = require("lucide-react");
+const require$$0 = require("react");
+const assetPanelStore = require("../assets/asset-panel-store.cjs");
+const useAthenaRestQuery = require("../lib/use-athena-rest-query.cjs");
+const AthenaContext = require("../provider/AthenaContext.cjs");
+const credentialRenewal = require("../runtime/credential-renewal.cjs");
+const primitives = require("./primitives.cjs");
+function record(value) {
+  return value && typeof value === "object" && !Array.isArray(value) ? value : void 0;
+}
+function parseResultString(result) {
+  try {
+    const parsed = JSON.parse(result);
+    return typeof parsed === "string" ? [] : [parsed];
+  } catch {
+  }
+  const items = [];
+  for (const line of result.split("\n")) {
+    const trimmed = line.trim();
+    if (!trimmed) continue;
+    try {
+      const parsed = JSON.parse(trimmed);
+      if (typeof parsed !== "string") items.push(parsed);
+    } catch {
+    }
+  }
+  return items;
+}
+function resultObjects(value) {
+  const found = [];
+  const stack = [value];
+  let budget = 2e3;
+  while (stack.length > 0 && budget > 0) {
+    budget -= 1;
+    const item = stack.pop();
+    if (typeof item === "string") {
+      stack.push(...parseResultString(item));
+      continue;
+    }
+    if (Array.isArray(item)) {
+      for (let index = item.length - 1; index >= 0; index -= 1) stack.push(item[index]);
+      continue;
+    }
+    const fields = record(item);
+    if (!fields) continue;
+    found.push(fields);
+    for (const nested of Object.values(fields)) {
+      if (nested && typeof nested === "object") stack.push(nested);
+      else if (typeof nested === "string" && (nested.startsWith("{") || nested.startsWith("["))) stack.push(nested);
+    }
+  }
+  return found;
+}
+function readStudioScreenshot(...sources) {
+  const shot = {
+    presignedUrl: null,
+    s3Key: null,
+    text: null,
+    sheetId: null,
+    sheetName: null,
+    range: null,
+    pageCount: null,
+    savedAssetId: null
+  };
+  for (const source of sources) {
+    for (const fields of resultObjects(source)) {
+      const imageSource = fields.type === "image" ? record(fields.source) : void 0;
+      if (imageSource) {
+        if (shot.presignedUrl === null && typeof imageSource.presigned_url === "string") {
+          shot.presignedUrl = imageSource.presigned_url;
+        }
+        if (shot.s3Key === null && imageSource.type === "s3_key" && typeof imageSource.data === "string") {
+          shot.s3Key = imageSource.data;
+        }
+      }
+      if (shot.presignedUrl === null && typeof fields.presigned_url === "string") shot.presignedUrl = fields.presigned_url;
+      if (shot.text === null && fields.type === "text" && typeof fields.text === "string") shot.text = fields.text;
+      if (shot.sheetId === null && typeof fields.sheet_id === "number") shot.sheetId = fields.sheet_id;
+      if (shot.sheetName === null && typeof fields.sheet_name === "string") shot.sheetName = fields.sheet_name;
+      if (shot.range === null && typeof fields.range === "string") shot.range = fields.range;
+      if (shot.pageCount === null && typeof fields.page_count === "number") shot.pageCount = fields.page_count;
+      if (shot.savedAssetId === null && typeof fields.saved_asset_id === "string" && fields.saved_asset_id !== "") {
+        shot.savedAssetId = fields.saved_asset_id;
+      }
+    }
+  }
+  return shot;
+}
+const FRESH_URL_STALE_MS = 10 * 60 * 1e3;
+function useFreshPresignedUrlQuery({
+  s3Key,
+  fallbackUrl
+}) {
+  const config = require$$0.useContext(AthenaContext.AthenaContext);
+  const baseUrl = config ? config.backendUrl.replace(/\/api\/assistant-ui\/?$/, "") : null;
+  const identity = useAthenaRestQuery.useAthenaQueryIdentity({
+    apiKey: config == null ? void 0 : config.apiKey,
+    token: config == null ? void 0 : config.token,
+    getToken: config == null ? void 0 : config.getToken
+  });
+  const reader = require$$0.useId();
+  const query = reactQuery.useQuery({
+    queryKey: ["athena", "screenshot-presigned-url", baseUrl, ...identity, s3Key, reader],
+    enabled: s3Key !== null && config !== null,
+    staleTime: FRESH_URL_STALE_MS,
+    gcTime: 0,
+    refetchOnWindowFocus: false,
+    retry: false,
+    queryFn: async ({ signal }) => {
+      if (!config || baseUrl === null) return null;
+      const response = await credentialRenewal.sendWithCredentialRenewal(
+        config,
+        (headers) => fetch(`${baseUrl}/api/asset/screenshot-presigned-url`, {
+          method: "POST",
+          headers: { "Content-Type": "application/json", ...headers },
+          body: JSON.stringify({ screenshot_key: s3Key }),
+          signal
+        })
+      );
+      if (!response.ok) return null;
+      const data = await response.json();
+      return typeof data.presigned_url === "string" ? data.presigned_url : null;
+    }
+  });
+  return query.data ?? fallbackUrl;
+}
+function useScreenshotUrl({
+  s3Key,
+  fallbackUrl
+}) {
+  return useFreshPresignedUrlQuery({ s3Key, fallbackUrl });
+}
+function ScreenshotQueryScope({ children }) {
+  const queryClient = useAthenaRestQuery.useAthenaQueryClient();
+  return /* @__PURE__ */ jsxRuntime.jsx(reactQuery.QueryClientProvider, { client: queryClient, children });
+}
+function createStudioScreenshotToolUI(spec) {
+  const Card = ({ toolName, args, result, artifact, status }) => {
+    var _a;
+    const typedArgs = args;
+    const shot = require$$0.useMemo(() => readStudioScreenshot(result, artifact), [result, artifact]);
+    const imageUrl = useScreenshotUrl({ s3Key: shot.s3Key, fallbackUrl: shot.presignedUrl });
+    const openAsset = assetPanelStore.useAssetPanelStore((s) => s.openAsset);
+    const assetId = typeof (typedArgs == null ? void 0 : typedArgs.asset_id) === "string" && typedArgs.asset_id !== "" ? typedArgs.asset_id : null;
+    const label = spec.label({ args: typedArgs, shot });
+    const detail = ((_a = spec.detail) == null ? void 0 : _a.call(spec, { args: typedArgs, shot })) ?? "";
+    const isRunning = (status == null ? void 0 : status.type) === "running";
+    const isComplete = (status == null ? void 0 : status.type) === "complete" || !status;
+    const hasScreenshot = isComplete && (shot.presignedUrl !== null || shot.s3Key !== null);
+    const failed = isComplete && !hasScreenshot;
+    const errorMsg = (status == null ? void 0 : status.type) === "incomplete" ? String(status.error ?? status.reason ?? "Capture failed") : null;
+    const title = isRunning ? `Capturing screenshot of ${label}...` : failed || errorMsg ? `Failed to capture screenshot of ${label}` : `Screenshot captured for ${label}${detail ? ` ${detail}` : ""}`;
+    const open = assetId ? spec.open({ args: typedArgs, shot }) : null;
+    return /* @__PURE__ */ jsxRuntime.jsxs(
+      primitives.ToolCard,
+      {
+        icon: spec.icon,
+        status: failed ? "incomplete" : (status == null ? void 0 : status.type) ?? "complete",
+        title: primitives.truncate(title, 90),
+        toolName,
+        args: typedArgs,
+        result,
+        error: errorMsg ?? (failed ? spec.failureNote : null),
+        children: [
+          hasScreenshot && imageUrl ? /* @__PURE__ */ jsxRuntime.jsxs("div", { "data-slot": "studio-screenshot", className: "border-t border-border/40 p-3", children: [
+            /* @__PURE__ */ jsxRuntime.jsx("div", { className: "overflow-hidden rounded-lg border border-border/50 bg-muted/20", children: /* @__PURE__ */ jsxRuntime.jsx(
+              "img",
+              {
+                src: imageUrl,
+                alt: `Screenshot of ${label}`,
+                className: spec.imageClassName,
+                loading: "lazy"
+              }
+            ) }),
+            shot.savedAssetId ? /* @__PURE__ */ jsxRuntime.jsxs("div", { className: "mt-2 flex items-center gap-2 text-[11px] text-muted-foreground", children: [
+              /* @__PURE__ */ jsxRuntime.jsx("span", { children: "Saved as asset:" }),
+              /* @__PURE__ */ jsxRuntime.jsx(
+                "button",
+                {
+                  type: "button",
+                  onClick: () => openAsset(shot.savedAssetId),
+                  className: "rounded font-mono text-foreground/80 hover:underline",
+                  children: primitives.truncate(shot.savedAssetId, 28)
+                }
+              )
+            ] }) : null
+          ] }) : null,
+          hasScreenshot && open && assetId ? /* @__PURE__ */ jsxRuntime.jsx("div", { className: "flex flex-wrap gap-2 border-t border-border/40 px-4 py-2", children: /* @__PURE__ */ jsxRuntime.jsxs(
+            "button",
+            {
+              type: "button",
+              onClick: () => openAsset(assetId, open.options),
+              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" }),
+                open.label
+              ]
+            }
+          ) }) : null
+        ]
+      }
+    );
+  };
+  const Scoped = (props) => /* @__PURE__ */ jsxRuntime.jsx(ScreenshotQueryScope, { children: /* @__PURE__ */ jsxRuntime.jsx(Card, { ...props }) });
+  return Scoped;
+}
+function sheetLabel({ args, shot }) {
+  const name = shot.sheetName ?? (typeof (args == null ? void 0 : args.sheet_name) === "string" ? args.sheet_name : null);
+  const id = shot.sheetId ?? (typeof (args == null ? void 0 : args.sheet_id) === "number" ? args.sheet_id : null);
+  if (name) return `"${name}"`;
+  if (id !== null) return `sheet ${id}`;
+  return "active sheet";
+}
+function sheetRange({ args, shot }) {
+  return shot.range ?? (typeof (args == null ? void 0 : args.cell_range) === "string" ? args.cell_range : null);
+}
+const CaptureSheetScreenshotToolUIImpl = createStudioScreenshotToolUI({
+  icon: lucideReact.CameraIcon,
+  label: sheetLabel,
+  detail: (params) => {
+    const range = sheetRange(params);
+    return range ? `(${range})` : "";
+  },
+  imageClassName: "w-full object-contain",
+  failureNote: "The screenshot could not be captured. Check that the workbook exists and the sheet or range is valid.",
+  open: (params) => {
+    const range = sheetRange(params);
+    return {
+      label: range ? `Open ${range}` : "Open spreadsheet",
+      options: { type: "spreadsheet" }
+    };
+  }
+});
+const CaptureSheetScreenshotToolUI = require$$0.memo(
+  CaptureSheetScreenshotToolUIImpl
+);
+CaptureSheetScreenshotToolUI.displayName = "CaptureSheetScreenshotToolUI";
+const PAGE_IMAGE_CLASS = "aspect-[8.5/11] w-full object-contain";
+const CapturePageScreenshotToolUIImpl = createStudioScreenshotToolUI({
+  icon: lucideReact.FileTextIcon,
+  label: ({ args, shot }) => {
+    const page = typeof (args == null ? void 0 : args.page_number) === "number" ? args.page_number : 1;
+    return shot.pageCount !== null ? `page ${page} of ${shot.pageCount}` : `page ${page}`;
+  },
+  imageClassName: PAGE_IMAGE_CLASS,
+  failureNote: "The screenshot could not be captured. Check that the document exists and the page number is within range.",
+  open: () => ({ label: "Open document", options: { type: "document" } })
+});
+const CapturePageScreenshotToolUI = require$$0.memo(
+  CapturePageScreenshotToolUIImpl
+);
+CapturePageScreenshotToolUI.displayName = "CapturePageScreenshotToolUI";
+exports.CapturePageScreenshotToolUI = CapturePageScreenshotToolUI;
+exports.CaptureSheetScreenshotToolUI = CaptureSheetScreenshotToolUI;
+exports.ScreenshotQueryScope = ScreenshotQueryScope;
+exports.readStudioScreenshot = readStudioScreenshot;
+exports.useScreenshotUrl = useScreenshotUrl;
+//# sourceMappingURL=studio-screenshot-tool-uis.cjs.map

package/dist/tools/studio-screenshot-tool-uis.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"studio-screenshot-tool-uis.cjs","sources":["../../src/tools/studio-screenshot-tool-uis.tsx"],"sourcesContent":["\"use client\";\n\n/**\n * Screenshots the studio tools take of what they edit, as Athena on the web\n * shows them: a workbook's sheet (`capture_sheet_screenshot`) and a Word\n * document's page (`capture_page_screenshot`), each with what was captured\n * (\"Screenshot captured for \"Q1 Plan\" (A1:F40)\"), the picture, the asset it\n * was saved as, and a button that opens the asset.\n *\n * A screenshot's result carries a presigned URL that expires and the image's\n * storage key, which does not. The picture is shown from a fresh URL agora\n * signs for that key (`POST /api/asset/screenshot-presigned-url`, which signs\n * only objects it resolves and the caller may read), read once per key with\n * TanStack Query, and from the result's own URL meanwhile or when that read\n * fails, so a session reopened later still shows its screenshots. A card\n * rendered outside `AthenaProvider` has no backend to ask and shows the\n * result's own URL.\n */\nimport type { ToolCallMessagePartComponent } from \"@assistant-ui/react\";\nimport { QueryClientProvider, useQuery } from \"@tanstack/react-query\";\nimport { CameraIcon, ExternalLinkIcon, FileTextIcon } from \"lucide-react\";\nimport { type ReactNode, memo, useContext, useId, useMemo } from \"react\";\nimport { type AssetPanelState, useAssetPanelStore } from \"../assets/asset-panel-store\";\nimport { useAthenaQueryClient, useAthenaQueryIdentity } from \"../lib/use-athena-rest-query\";\nimport { AthenaContext } from \"../provider/AthenaContext\";\nimport { sendWithCredentialRenewal } from \"../runtime/credential-renewal\";\nimport { ToolCard, type ToolStatus, truncate } from \"./primitives\";\n\n// ─── Reading a screenshot result ─────────────────────────────────────\n\nfunction record(value: unknown): Record<string, unknown> | undefined {\n  return value && typeof value === \"object\" && !Array.isArray(value)\n    ? (value as Record<string, unknown>)\n    : undefined;\n}\n\n/**\n * A result string's content items: one JSON value, or the backend's\n * newline-delimited serialization of a multi-block list (an image block and a\n * note, `{…}\\n{…}`), which a plain `JSON.parse` rejects. A bare string carries\n * no image source.\n */\nfunction parseResultString(result: string): unknown[] {\n  try {\n    const parsed: unknown = JSON.parse(result);\n    return typeof parsed === \"string\" ? [] : [parsed];\n  } catch {\n    // Newline-delimited below.\n  }\n  const items: unknown[] = [];\n  for (const line of result.split(\"\\n\")) {\n    const trimmed = line.trim();\n    if (!trimmed) continue;\n    try {\n      const parsed: unknown = JSON.parse(trimmed);\n      if (typeof parsed !== \"string\") items.push(parsed);\n    } catch {\n      // A plain-text line holds nothing to read.\n    }\n  }\n  return items;\n}\n\n/** Every object a result holds, depth first, through JSON strings and `result` wrappers. */\nfunction resultObjects(value: unknown): Record<string, unknown>[] {\n  const found: Record<string, unknown>[] = [];\n  const stack: unknown[] = [value];\n  let budget = 2_000;\n  while (stack.length > 0 && budget > 0) {\n    budget -= 1;\n    const item = stack.pop();\n    if (typeof item === \"string\") {\n      stack.push(...parseResultString(item));\n      continue;\n    }\n    if (Array.isArray(item)) {\n      for (let index = item.length - 1; index >= 0; index -= 1) stack.push(item[index]);\n      continue;\n    }\n    const fields = record(item);\n    if (!fields) continue;\n    found.push(fields);\n    for (const nested of Object.values(fields)) {\n      if (nested && typeof nested === \"object\") stack.push(nested);\n      else if (typeof nested === \"string\" && (nested.startsWith(\"{\") || nested.startsWith(\"[\"))) stack.push(nested);\n    }\n  }\n  return found;\n}\n\n/** What a screenshot result says about its picture and what it captured. */\nexport interface StudioScreenshot {\n  /** The result's own presigned URL, which expires. */\n  presignedUrl: string | null;\n  /** The image's storage key (`source.data` of an `s3_key` image source), which does not. */\n  s3Key: string | null;\n  /** The note the tool returned beside the picture. */\n  text: string | null;\n  sheetId: number | null;\n  sheetName: string | null;\n  range: string | null;\n  pageCount: number | null;\n  /** The asset the screenshot was saved as (`save_as_asset`). */\n  savedAssetId: string | null;\n}\n\n/** Reads a screenshot tool's result (and artifact): the first value of each field found. */\nexport function readStudioScreenshot(...sources: unknown[]): StudioScreenshot {\n  const shot: StudioScreenshot = {\n    presignedUrl: null,\n    s3Key: null,\n    text: null,\n    sheetId: null,\n    sheetName: null,\n    range: null,\n    pageCount: null,\n    savedAssetId: null,\n  };\n  for (const source of sources) {\n    for (const fields of resultObjects(source)) {\n      const imageSource = fields.type === \"image\" ? record(fields.source) : undefined;\n      if (imageSource) {\n        if (shot.presignedUrl === null && typeof imageSource.presigned_url === \"string\") {\n          shot.presignedUrl = imageSource.presigned_url;\n        }\n        if (shot.s3Key === null && imageSource.type === \"s3_key\" && typeof imageSource.data === \"string\") {\n          shot.s3Key = imageSource.data;\n        }\n      }\n      if (shot.presignedUrl === null && typeof fields.presigned_url === \"string\") shot.presignedUrl = fields.presigned_url;\n      if (shot.text === null && fields.type === \"text\" && typeof fields.text === \"string\") shot.text = fields.text;\n      if (shot.sheetId === null && typeof fields.sheet_id === \"number\") shot.sheetId = fields.sheet_id;\n      if (shot.sheetName === null && typeof fields.sheet_name === \"string\") shot.sheetName = fields.sheet_name;\n      if (shot.range === null && typeof fields.range === \"string\") shot.range = fields.range;\n      if (shot.pageCount === null && typeof fields.page_count === \"number\") shot.pageCount = fields.page_count;\n      if (shot.savedAssetId === null && typeof fields.saved_asset_id === \"string\" && fields.saved_asset_id !== \"\") {\n        shot.savedAssetId = fields.saved_asset_id;\n      }\n    }\n  }\n  return shot;\n}\n\n// ─── A fresh URL for the picture ─────────────────────────────────────\n\n/** How long a URL agora signed for a key is used before it is asked again. */\nconst FRESH_URL_STALE_MS = 10 * 60 * 1000;\n\nfunction useFreshPresignedUrlQuery({\n  s3Key,\n  fallbackUrl,\n}: {\n  s3Key: string | null;\n  fallbackUrl: string | null;\n}): string | null {\n  const config = useContext(AthenaContext);\n  const baseUrl = config ? config.backendUrl.replace(/\\/api\\/assistant-ui\\/?$/, \"\") : null;\n  // A signed URL is a bearer grant to the picture, so no card is handed one\n  // another read: the key carries the signed-in principal and a fingerprint\n  // of the credential (never the credential itself) and this card's own id,\n  // and the entry goes when the card unmounts. Each card asks agora, which\n  // runs the read gate for whoever asks, after an account switch too, and\n  // behind a host `getToken` whose account the key cannot see.\n  const identity = useAthenaQueryIdentity({\n    apiKey: config?.apiKey,\n    token: config?.token,\n    getToken: config?.getToken,\n  });\n  const reader = useId();\n  const query = useQuery({\n    queryKey: [\"athena\", \"screenshot-presigned-url\", baseUrl, ...identity, s3Key, reader],\n    enabled: s3Key !== null && config !== null,\n    staleTime: FRESH_URL_STALE_MS,\n    gcTime: 0,\n    refetchOnWindowFocus: false,\n    retry: false,\n    queryFn: async ({ signal }): Promise<string | null> => {\n      if (!config || baseUrl === null) return null;\n      const response = await sendWithCredentialRenewal(config, (headers) =>\n        fetch(`${baseUrl}/api/asset/screenshot-presigned-url`, {\n          method: \"POST\",\n          headers: { \"Content-Type\": \"application/json\", ...headers },\n          body: JSON.stringify({ screenshot_key: s3Key }),\n          signal,\n        }),\n      );\n      if (!response.ok) return null;\n      const data = (await response.json()) as { presigned_url?: unknown };\n      return typeof data.presigned_url === \"string\" ? data.presigned_url : null;\n    },\n  });\n  return query.data ?? fallbackUrl;\n}\n\n/**\n * The URL to show a screenshot from: one agora signs afresh for its storage\n * key, else the result's own. Needs a `QueryClient` in context: wrap the card\n * in `ScreenshotQueryScope`, which brings one when the host has none.\n */\nexport function useScreenshotUrl({\n  s3Key,\n  fallbackUrl,\n}: {\n  s3Key: string | null;\n  fallbackUrl: string | null;\n}): string | null {\n  return useFreshPresignedUrlQuery({ s3Key, fallbackUrl });\n}\n\n/** The host's `QueryClient` when one is in context, else one of the card's own. */\nexport function ScreenshotQueryScope({ children }: { children: ReactNode }) {\n  const queryClient = useAthenaQueryClient();\n  return <QueryClientProvider client={queryClient}>{children}</QueryClientProvider>;\n}\n\n// ─── The cards ───────────────────────────────────────────────────────\n\ntype OpenAssetOptions = Parameters<AssetPanelState[\"openAsset\"]>[1];\n\ninterface StudioScreenshotSpec {\n  /** What was captured, in the card's words (\"“Q1 Plan”\", \"page 2 of 9\"). */\n  label: (params: { args: Record<string, unknown> | undefined; shot: StudioScreenshot }) => string;\n  /** The completed title's tail after the label (\"(A1:F40)\"); empty for none. */\n  detail?: (params: { args: Record<string, unknown> | undefined; shot: StudioScreenshot }) => string;\n  icon: typeof CameraIcon;\n  /** The picture's box: a page keeps a page's proportions while it loads. */\n  imageClassName: string;\n  /** What a failed capture says under its title. */\n  failureNote: string;\n  /** How the asset opens, and the button's words. */\n  open: (params: { args: Record<string, unknown> | undefined; shot: StudioScreenshot }) => {\n    label: string;\n    options: OpenAssetOptions;\n  };\n}\n\nfunction createStudioScreenshotToolUI(spec: StudioScreenshotSpec): ToolCallMessagePartComponent {\n  const Card: ToolCallMessagePartComponent = ({ toolName, args, result, artifact, status }) => {\n    const typedArgs = args as Record<string, unknown> | undefined;\n    const shot = useMemo(() => readStudioScreenshot(result, artifact), [result, artifact]);\n    const imageUrl = useScreenshotUrl({ s3Key: shot.s3Key, fallbackUrl: shot.presignedUrl });\n    const openAsset = useAssetPanelStore((s) => s.openAsset);\n    const assetId = typeof typedArgs?.asset_id === \"string\" && typedArgs.asset_id !== \"\" ? typedArgs.asset_id : null;\n    const label = spec.label({ args: typedArgs, shot });\n    const detail = spec.detail?.({ args: typedArgs, shot }) ?? \"\";\n    const isRunning = status?.type === \"running\";\n    const isComplete = status?.type === \"complete\" || !status;\n    const hasScreenshot = isComplete && (shot.presignedUrl !== null || shot.s3Key !== null);\n    const failed = isComplete && !hasScreenshot;\n    const errorMsg = status?.type === \"incomplete\" ? String(status.error ?? status.reason ?? \"Capture failed\") : null;\n    const title = isRunning\n      ? `Capturing screenshot of ${label}...`\n      : failed || errorMsg\n        ? `Failed to capture screenshot of ${label}`\n        : `Screenshot captured for ${label}${detail ? ` ${detail}` : \"\"}`;\n    const open = assetId ? spec.open({ args: typedArgs, shot }) : null;\n\n    return (\n      <ToolCard\n        icon={spec.icon}\n        status={(failed ? \"incomplete\" : (status?.type ?? \"complete\")) as ToolStatus}\n        title={truncate(title, 90)}\n        toolName={toolName}\n        args={typedArgs}\n        result={result}\n        error={errorMsg ?? (failed ? spec.failureNote : null)}\n      >\n        {hasScreenshot && imageUrl ? (\n          <div data-slot=\"studio-screenshot\" className=\"border-t border-border/40 p-3\">\n            <div className=\"overflow-hidden rounded-lg border border-border/50 bg-muted/20\">\n              <img\n                src={imageUrl}\n                alt={`Screenshot of ${label}`}\n                className={spec.imageClassName}\n                loading=\"lazy\"\n              />\n            </div>\n            {shot.savedAssetId ? (\n              <div className=\"mt-2 flex items-center gap-2 text-[11px] text-muted-foreground\">\n                <span>Saved as asset:</span>\n                <button\n                  type=\"button\"\n                  onClick={() => openAsset(shot.savedAssetId as string)}\n                  className=\"rounded font-mono text-foreground/80 hover:underline\"\n                >\n                  {truncate(shot.savedAssetId, 28)}\n                </button>\n              </div>\n            ) : null}\n          </div>\n        ) : null}\n        {hasScreenshot && open && assetId ? (\n          <div className=\"flex flex-wrap gap-2 border-t border-border/40 px-4 py-2\">\n            <button\n              type=\"button\"\n              onClick={() => openAsset(assetId, open.options)}\n              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\"\n            >\n              <ExternalLinkIcon className=\"size-3\" />\n              {open.label}\n            </button>\n          </div>\n        ) : null}\n      </ToolCard>\n    );\n  };\n  const Scoped: ToolCallMessagePartComponent = (props) => (\n    <ScreenshotQueryScope>\n      <Card {...props} />\n    </ScreenshotQueryScope>\n  );\n  return Scoped;\n}\n\n/** \"“Q1 Plan”\", \"sheet 3\", or \"active sheet\", as Athena on the web names the captured sheet. */\nfunction sheetLabel({ args, shot }: { args: Record<string, unknown> | undefined; shot: StudioScreenshot }): string {\n  const name = shot.sheetName ?? (typeof args?.sheet_name === \"string\" ? args.sheet_name : null);\n  const id = shot.sheetId ?? (typeof args?.sheet_id === \"number\" ? args.sheet_id : null);\n  if (name) return `\"${name}\"`;\n  if (id !== null) return `sheet ${id}`;\n  return \"active sheet\";\n}\n\nfunction sheetRange({ args, shot }: { args: Record<string, unknown> | undefined; shot: StudioScreenshot }): string | null {\n  return shot.range ?? (typeof args?.cell_range === \"string\" ? args.cell_range : null);\n}\n\nconst CaptureSheetScreenshotToolUIImpl = createStudioScreenshotToolUI({\n  icon: CameraIcon,\n  label: sheetLabel,\n  detail: (params) => {\n    const range = sheetRange(params);\n    return range ? `(${range})` : \"\";\n  },\n  imageClassName: \"w-full object-contain\",\n  failureNote: \"The screenshot could not be captured. Check that the workbook exists and the sheet or range is valid.\",\n  open: (params) => {\n    const range = sheetRange(params);\n    return {\n      label: range ? `Open ${range}` : \"Open spreadsheet\",\n      options: { type: \"spreadsheet\" },\n    };\n  },\n});\n\n/** `capture_sheet_screenshot`: a workbook sheet's screenshot. */\nexport const CaptureSheetScreenshotToolUI = memo(\n  CaptureSheetScreenshotToolUIImpl,\n) as unknown as ToolCallMessagePartComponent;\nCaptureSheetScreenshotToolUI.displayName = \"CaptureSheetScreenshotToolUI\";\n\n/** A Word page's proportions (US letter portrait), so the picture's box holds still while it loads. */\nconst PAGE_IMAGE_CLASS = \"aspect-[8.5/11] w-full object-contain\";\n\nconst CapturePageScreenshotToolUIImpl = createStudioScreenshotToolUI({\n  icon: FileTextIcon,\n  label: ({ args, shot }) => {\n    const page = typeof args?.page_number === \"number\" ? args.page_number : 1;\n    return shot.pageCount !== null ? `page ${page} of ${shot.pageCount}` : `page ${page}`;\n  },\n  imageClassName: PAGE_IMAGE_CLASS,\n  failureNote: \"The screenshot could not be captured. Check that the document exists and the page number is within range.\",\n  open: () => ({ label: \"Open document\", options: { type: \"document\" } }),\n});\n\n/** `capture_page_screenshot`: a Word document page's screenshot. */\nexport const CapturePageScreenshotToolUI = memo(\n  CapturePageScreenshotToolUIImpl,\n) as unknown as ToolCallMessagePartComponent;\nCapturePageScreenshotToolUI.displayName = \"CapturePageScreenshotToolUI\";\n"],"names":["useContext","AthenaContext","useAthenaQueryIdentity","useId","useQuery","sendWithCredentialRenewal","useAthenaQueryClient","jsx","QueryClientProvider","useMemo","useAssetPanelStore","jsxs","ToolCard","truncate","ExternalLinkIcon","CameraIcon","memo","FileTextIcon"],"mappings":";;;;;;;;;;;AA8BA,SAAS,OAAO,OAAqD;AACnE,SAAO,SAAS,OAAO,UAAU,YAAY,CAAC,MAAM,QAAQ,KAAK,IAC5D,QACD;AACN;AAQA,SAAS,kBAAkB,QAA2B;AACpD,MAAI;AACF,UAAM,SAAkB,KAAK,MAAM,MAAM;AACzC,WAAO,OAAO,WAAW,WAAW,CAAA,IAAK,CAAC,MAAM;AAAA,EAClD,QAAQ;AAAA,EAER;AACA,QAAM,QAAmB,CAAA;AACzB,aAAW,QAAQ,OAAO,MAAM,IAAI,GAAG;AACrC,UAAM,UAAU,KAAK,KAAA;AACrB,QAAI,CAAC,QAAS;AACd,QAAI;AACF,YAAM,SAAkB,KAAK,MAAM,OAAO;AAC1C,UAAI,OAAO,WAAW,SAAU,OAAM,KAAK,MAAM;AAAA,IACnD,QAAQ;AAAA,IAER;AAAA,EACF;AACA,SAAO;AACT;AAGA,SAAS,cAAc,OAA2C;AAChE,QAAM,QAAmC,CAAA;AACzC,QAAM,QAAmB,CAAC,KAAK;AAC/B,MAAI,SAAS;AACb,SAAO,MAAM,SAAS,KAAK,SAAS,GAAG;AACrC,cAAU;AACV,UAAM,OAAO,MAAM,IAAA;AACnB,QAAI,OAAO,SAAS,UAAU;AAC5B,YAAM,KAAK,GAAG,kBAAkB,IAAI,CAAC;AACrC;AAAA,IACF;AACA,QAAI,MAAM,QAAQ,IAAI,GAAG;AACvB,eAAS,QAAQ,KAAK,SAAS,GAAG,SAAS,GAAG,SAAS,EAAG,OAAM,KAAK,KAAK,KAAK,CAAC;AAChF;AAAA,IACF;AACA,UAAM,SAAS,OAAO,IAAI;AAC1B,QAAI,CAAC,OAAQ;AACb,UAAM,KAAK,MAAM;AACjB,eAAW,UAAU,OAAO,OAAO,MAAM,GAAG;AAC1C,UAAI,UAAU,OAAO,WAAW,SAAU,OAAM,KAAK,MAAM;AAAA,eAClD,OAAO,WAAW,aAAa,OAAO,WAAW,GAAG,KAAK,OAAO,WAAW,GAAG,GAAI,OAAM,KAAK,MAAM;AAAA,IAC9G;AAAA,EACF;AACA,SAAO;AACT;AAmBO,SAAS,wBAAwB,SAAsC;AAC5E,QAAM,OAAyB;AAAA,IAC7B,cAAc;AAAA,IACd,OAAO;AAAA,IACP,MAAM;AAAA,IACN,SAAS;AAAA,IACT,WAAW;AAAA,IACX,OAAO;AAAA,IACP,WAAW;AAAA,IACX,cAAc;AAAA,EAAA;AAEhB,aAAW,UAAU,SAAS;AAC5B,eAAW,UAAU,cAAc,MAAM,GAAG;AAC1C,YAAM,cAAc,OAAO,SAAS,UAAU,OAAO,OAAO,MAAM,IAAI;AACtE,UAAI,aAAa;AACf,YAAI,KAAK,iBAAiB,QAAQ,OAAO,YAAY,kBAAkB,UAAU;AAC/E,eAAK,eAAe,YAAY;AAAA,QAClC;AACA,YAAI,KAAK,UAAU,QAAQ,YAAY,SAAS,YAAY,OAAO,YAAY,SAAS,UAAU;AAChG,eAAK,QAAQ,YAAY;AAAA,QAC3B;AAAA,MACF;AACA,UAAI,KAAK,iBAAiB,QAAQ,OAAO,OAAO,kBAAkB,SAAU,MAAK,eAAe,OAAO;AACvG,UAAI,KAAK,SAAS,QAAQ,OAAO,SAAS,UAAU,OAAO,OAAO,SAAS,SAAU,MAAK,OAAO,OAAO;AACxG,UAAI,KAAK,YAAY,QAAQ,OAAO,OAAO,aAAa,SAAU,MAAK,UAAU,OAAO;AACxF,UAAI,KAAK,cAAc,QAAQ,OAAO,OAAO,eAAe,SAAU,MAAK,YAAY,OAAO;AAC9F,UAAI,KAAK,UAAU,QAAQ,OAAO,OAAO,UAAU,SAAU,MAAK,QAAQ,OAAO;AACjF,UAAI,KAAK,cAAc,QAAQ,OAAO,OAAO,eAAe,SAAU,MAAK,YAAY,OAAO;AAC9F,UAAI,KAAK,iBAAiB,QAAQ,OAAO,OAAO,mBAAmB,YAAY,OAAO,mBAAmB,IAAI;AAC3G,aAAK,eAAe,OAAO;AAAA,MAC7B;AAAA,IACF;AAAA,EACF;AACA,SAAO;AACT;AAKA,MAAM,qBAAqB,KAAK,KAAK;AAErC,SAAS,0BAA0B;AAAA,EACjC;AAAA,EACA;AACF,GAGkB;AAChB,QAAM,SAASA,WAAAA,WAAWC,2BAAa;AACvC,QAAM,UAAU,SAAS,OAAO,WAAW,QAAQ,2BAA2B,EAAE,IAAI;AAOpF,QAAM,WAAWC,mBAAAA,uBAAuB;AAAA,IACtC,QAAQ,iCAAQ;AAAA,IAChB,OAAO,iCAAQ;AAAA,IACf,UAAU,iCAAQ;AAAA,EAAA,CACnB;AACD,QAAM,SAASC,WAAAA,MAAA;AACf,QAAM,QAAQC,WAAAA,SAAS;AAAA,IACrB,UAAU,CAAC,UAAU,4BAA4B,SAAS,GAAG,UAAU,OAAO,MAAM;AAAA,IACpF,SAAS,UAAU,QAAQ,WAAW;AAAA,IACtC,WAAW;AAAA,IACX,QAAQ;AAAA,IACR,sBAAsB;AAAA,IACtB,OAAO;AAAA,IACP,SAAS,OAAO,EAAE,aAAqC;AACrD,UAAI,CAAC,UAAU,YAAY,KAAM,QAAO;AACxC,YAAM,WAAW,MAAMC,kBAAAA;AAAAA,QAA0B;AAAA,QAAQ,CAAC,YACxD,MAAM,GAAG,OAAO,uCAAuC;AAAA,UACrD,QAAQ;AAAA,UACR,SAAS,EAAE,gBAAgB,oBAAoB,GAAG,QAAA;AAAA,UAClD,MAAM,KAAK,UAAU,EAAE,gBAAgB,OAAO;AAAA,UAC9C;AAAA,QAAA,CACD;AAAA,MAAA;AAEH,UAAI,CAAC,SAAS,GAAI,QAAO;AACzB,YAAM,OAAQ,MAAM,SAAS,KAAA;AAC7B,aAAO,OAAO,KAAK,kBAAkB,WAAW,KAAK,gBAAgB;AAAA,IACvE;AAAA,EAAA,CACD;AACD,SAAO,MAAM,QAAQ;AACvB;AAOO,SAAS,iBAAiB;AAAA,EAC/B;AAAA,EACA;AACF,GAGkB;AAChB,SAAO,0BAA0B,EAAE,OAAO,aAAa;AACzD;AAGO,SAAS,qBAAqB,EAAE,YAAqC;AAC1E,QAAM,cAAcC,mBAAAA,qBAAA;AACpB,SAAOC,2BAAAA,IAACC,WAAAA,qBAAA,EAAoB,QAAQ,aAAc,SAAA,CAAS;AAC7D;AAuBA,SAAS,6BAA6B,MAA0D;AAC9F,QAAM,OAAqC,CAAC,EAAE,UAAU,MAAM,QAAQ,UAAU,aAAa;;AAC3F,UAAM,YAAY;AAClB,UAAM,OAAOC,mBAAQ,MAAM,qBAAqB,QAAQ,QAAQ,GAAG,CAAC,QAAQ,QAAQ,CAAC;AACrF,UAAM,WAAW,iBAAiB,EAAE,OAAO,KAAK,OAAO,aAAa,KAAK,cAAc;AACvF,UAAM,YAAYC,gBAAAA,mBAAmB,CAAC,MAAM,EAAE,SAAS;AACvD,UAAM,UAAU,QAAO,uCAAW,cAAa,YAAY,UAAU,aAAa,KAAK,UAAU,WAAW;AAC5G,UAAM,QAAQ,KAAK,MAAM,EAAE,MAAM,WAAW,MAAM;AAClD,UAAM,WAAS,UAAK,WAAL,8BAAc,EAAE,MAAM,WAAW,KAAA,OAAW;AAC3D,UAAM,aAAY,iCAAQ,UAAS;AACnC,UAAM,cAAa,iCAAQ,UAAS,cAAc,CAAC;AACnD,UAAM,gBAAgB,eAAe,KAAK,iBAAiB,QAAQ,KAAK,UAAU;AAClF,UAAM,SAAS,cAAc,CAAC;AAC9B,UAAM,YAAW,iCAAQ,UAAS,eAAe,OAAO,OAAO,SAAS,OAAO,UAAU,gBAAgB,IAAI;AAC7G,UAAM,QAAQ,YACV,2BAA2B,KAAK,QAChC,UAAU,WACR,mCAAmC,KAAK,KACxC,2BAA2B,KAAK,GAAG,SAAS,IAAI,MAAM,KAAK,EAAE;AACnE,UAAM,OAAO,UAAU,KAAK,KAAK,EAAE,MAAM,WAAW,KAAA,CAAM,IAAI;AAE9D,WACEC,2BAAAA;AAAAA,MAACC,WAAAA;AAAAA,MAAA;AAAA,QACC,MAAM,KAAK;AAAA,QACX,QAAS,SAAS,gBAAgB,iCAAQ,SAAQ;AAAA,QAClD,OAAOC,WAAAA,SAAS,OAAO,EAAE;AAAA,QACzB;AAAA,QACA,MAAM;AAAA,QACN;AAAA,QACA,OAAO,aAAa,SAAS,KAAK,cAAc;AAAA,QAE/C,UAAA;AAAA,UAAA,iBAAiB,WAChBF,2BAAAA,KAAC,OAAA,EAAI,aAAU,qBAAoB,WAAU,iCAC3C,UAAA;AAAA,YAAAJ,2BAAAA,IAAC,OAAA,EAAI,WAAU,kEACb,UAAAA,2BAAAA;AAAAA,cAAC;AAAA,cAAA;AAAA,gBACC,KAAK;AAAA,gBACL,KAAK,iBAAiB,KAAK;AAAA,gBAC3B,WAAW,KAAK;AAAA,gBAChB,SAAQ;AAAA,cAAA;AAAA,YAAA,GAEZ;AAAA,YACC,KAAK,eACJI,gCAAC,OAAA,EAAI,WAAU,kEACb,UAAA;AAAA,cAAAJ,2BAAAA,IAAC,UAAK,UAAA,kBAAA,CAAe;AAAA,cACrBA,2BAAAA;AAAAA,gBAAC;AAAA,gBAAA;AAAA,kBACC,MAAK;AAAA,kBACL,SAAS,MAAM,UAAU,KAAK,YAAsB;AAAA,kBACpD,WAAU;AAAA,kBAET,UAAAM,WAAAA,SAAS,KAAK,cAAc,EAAE;AAAA,gBAAA;AAAA,cAAA;AAAA,YACjC,EAAA,CACF,IACE;AAAA,UAAA,EAAA,CACN,IACE;AAAA,UACH,iBAAiB,QAAQ,UACxBN,2BAAAA,IAAC,OAAA,EAAI,WAAU,4DACb,UAAAI,2BAAAA;AAAAA,YAAC;AAAA,YAAA;AAAA,cACC,MAAK;AAAA,cACL,SAAS,MAAM,UAAU,SAAS,KAAK,OAAO;AAAA,cAC9C,WAAU;AAAA,cAEV,UAAA;AAAA,gBAAAJ,2BAAAA,IAACO,YAAAA,kBAAA,EAAiB,WAAU,SAAA,CAAS;AAAA,gBACpC,KAAK;AAAA,cAAA;AAAA,YAAA;AAAA,UAAA,GAEV,IACE;AAAA,QAAA;AAAA,MAAA;AAAA,IAAA;AAAA,EAGV;AACA,QAAM,SAAuC,CAAC,UAC5CP,2BAAAA,IAAC,wBACC,UAAAA,2BAAAA,IAAC,MAAA,EAAM,GAAG,MAAA,CAAO,EAAA,CACnB;AAEF,SAAO;AACT;AAGA,SAAS,WAAW,EAAE,MAAM,QAAuF;AACjH,QAAM,OAAO,KAAK,cAAc,QAAO,6BAAM,gBAAe,WAAW,KAAK,aAAa;AACzF,QAAM,KAAK,KAAK,YAAY,QAAO,6BAAM,cAAa,WAAW,KAAK,WAAW;AACjF,MAAI,KAAM,QAAO,IAAI,IAAI;AACzB,MAAI,OAAO,KAAM,QAAO,SAAS,EAAE;AACnC,SAAO;AACT;AAEA,SAAS,WAAW,EAAE,MAAM,QAA8F;AACxH,SAAO,KAAK,UAAU,QAAO,6BAAM,gBAAe,WAAW,KAAK,aAAa;AACjF;AAEA,MAAM,mCAAmC,6BAA6B;AAAA,EACpE,MAAMQ,YAAAA;AAAAA,EACN,OAAO;AAAA,EACP,QAAQ,CAAC,WAAW;AAClB,UAAM,QAAQ,WAAW,MAAM;AAC/B,WAAO,QAAQ,IAAI,KAAK,MAAM;AAAA,EAChC;AAAA,EACA,gBAAgB;AAAA,EAChB,aAAa;AAAA,EACb,MAAM,CAAC,WAAW;AAChB,UAAM,QAAQ,WAAW,MAAM;AAC/B,WAAO;AAAA,MACL,OAAO,QAAQ,QAAQ,KAAK,KAAK;AAAA,MACjC,SAAS,EAAE,MAAM,cAAA;AAAA,IAAc;AAAA,EAEnC;AACF,CAAC;AAGM,MAAM,+BAA+BC,WAAAA;AAAAA,EAC1C;AACF;AACA,6BAA6B,cAAc;AAG3C,MAAM,mBAAmB;AAEzB,MAAM,kCAAkC,6BAA6B;AAAA,EACnE,MAAMC,YAAAA;AAAAA,EACN,OAAO,CAAC,EAAE,MAAM,WAAW;AACzB,UAAM,OAAO,QAAO,6BAAM,iBAAgB,WAAW,KAAK,cAAc;AACxE,WAAO,KAAK,cAAc,OAAO,QAAQ,IAAI,OAAO,KAAK,SAAS,KAAK,QAAQ,IAAI;AAAA,EACrF;AAAA,EACA,gBAAgB;AAAA,EAChB,aAAa;AAAA,EACb,MAAM,OAAO,EAAE,OAAO,iBAAiB,SAAS,EAAE,MAAM,aAAW;AACrE,CAAC;AAGM,MAAM,8BAA8BD,WAAAA;AAAAA,EACzC;AACF;AACA,4BAA4B,cAAc;;;;;;"}

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

@@ -0,0 +1,52 @@
+/**
+ * Screenshots the studio tools take of what they edit, as Athena on the web
+ * shows them: a workbook's sheet (`capture_sheet_screenshot`) and a Word
+ * document's page (`capture_page_screenshot`), each with what was captured
+ * ("Screenshot captured for "Q1 Plan" (A1:F40)"), the picture, the asset it
+ * was saved as, and a button that opens the asset.
+ *
+ * A screenshot's result carries a presigned URL that expires and the image's
+ * storage key, which does not. The picture is shown from a fresh URL agora
+ * signs for that key (`POST /api/asset/screenshot-presigned-url`, which signs
+ * only objects it resolves and the caller may read), read once per key with
+ * TanStack Query, and from the result's own URL meanwhile or when that read
+ * fails, so a session reopened later still shows its screenshots. A card
+ * rendered outside `AthenaProvider` has no backend to ask and shows the
+ * result's own URL.
+ */
+import type { ToolCallMessagePartComponent } from "@assistant-ui/react";
+import { type ReactNode } from "react";
+/** What a screenshot result says about its picture and what it captured. */
+export interface StudioScreenshot {
+    /** The result's own presigned URL, which expires. */
+    presignedUrl: string | null;
+    /** The image's storage key (`source.data` of an `s3_key` image source), which does not. */
+    s3Key: string | null;
+    /** The note the tool returned beside the picture. */
+    text: string | null;
+    sheetId: number | null;
+    sheetName: string | null;
+    range: string | null;
+    pageCount: number | null;
+    /** The asset the screenshot was saved as (`save_as_asset`). */
+    savedAssetId: string | null;
+}
+/** Reads a screenshot tool's result (and artifact): the first value of each field found. */
+export declare function readStudioScreenshot(...sources: unknown[]): StudioScreenshot;
+/**
+ * The URL to show a screenshot from: one agora signs afresh for its storage
+ * key, else the result's own. Needs a `QueryClient` in context: wrap the card
+ * in `ScreenshotQueryScope`, which brings one when the host has none.
+ */
+export declare function useScreenshotUrl({ s3Key, fallbackUrl, }: {
+    s3Key: string | null;
+    fallbackUrl: string | null;
+}): string | null;
+/** The host's `QueryClient` when one is in context, else one of the card's own. */
+export declare function ScreenshotQueryScope({ children }: {
+    children: ReactNode;
+}): import("react").JSX.Element;
+/** `capture_sheet_screenshot`: a workbook sheet's screenshot. */
+export declare const CaptureSheetScreenshotToolUI: ToolCallMessagePartComponent;
+/** `capture_page_screenshot`: a Word document page's screenshot. */
+export declare const CapturePageScreenshotToolUI: ToolCallMessagePartComponent;

package/dist/tools/studio-screenshot-tool-uis.js

@@ -0,0 +1,262 @@
+import { jsx, jsxs } from "react/jsx-runtime";
+import { QueryClientProvider, useQuery } from "@tanstack/react-query";
+import { ExternalLinkIcon, CameraIcon, FileTextIcon } from "lucide-react";
+import { memo, useMemo, useContext, useId } from "react";
+import { useAssetPanelStore } from "../assets/asset-panel-store.js";
+import { useAthenaQueryClient, useAthenaQueryIdentity } from "../lib/use-athena-rest-query.js";
+import { AthenaContext } from "../provider/AthenaContext.js";
+import { sendWithCredentialRenewal } from "../runtime/credential-renewal.js";
+import { ToolCard, truncate } from "./primitives.js";
+function record(value) {
+  return value && typeof value === "object" && !Array.isArray(value) ? value : void 0;
+}
+function parseResultString(result) {
+  try {
+    const parsed = JSON.parse(result);
+    return typeof parsed === "string" ? [] : [parsed];
+  } catch {
+  }
+  const items = [];
+  for (const line of result.split("\n")) {
+    const trimmed = line.trim();
+    if (!trimmed) continue;
+    try {
+      const parsed = JSON.parse(trimmed);
+      if (typeof parsed !== "string") items.push(parsed);
+    } catch {
+    }
+  }
+  return items;
+}
+function resultObjects(value) {
+  const found = [];
+  const stack = [value];
+  let budget = 2e3;
+  while (stack.length > 0 && budget > 0) {
+    budget -= 1;
+    const item = stack.pop();
+    if (typeof item === "string") {
+      stack.push(...parseResultString(item));
+      continue;
+    }
+    if (Array.isArray(item)) {
+      for (let index = item.length - 1; index >= 0; index -= 1) stack.push(item[index]);
+      continue;
+    }
+    const fields = record(item);
+    if (!fields) continue;
+    found.push(fields);
+    for (const nested of Object.values(fields)) {
+      if (nested && typeof nested === "object") stack.push(nested);
+      else if (typeof nested === "string" && (nested.startsWith("{") || nested.startsWith("["))) stack.push(nested);
+    }
+  }
+  return found;
+}
+function readStudioScreenshot(...sources) {
+  const shot = {
+    presignedUrl: null,
+    s3Key: null,
+    text: null,
+    sheetId: null,
+    sheetName: null,
+    range: null,
+    pageCount: null,
+    savedAssetId: null
+  };
+  for (const source of sources) {
+    for (const fields of resultObjects(source)) {
+      const imageSource = fields.type === "image" ? record(fields.source) : void 0;
+      if (imageSource) {
+        if (shot.presignedUrl === null && typeof imageSource.presigned_url === "string") {
+          shot.presignedUrl = imageSource.presigned_url;
+        }
+        if (shot.s3Key === null && imageSource.type === "s3_key" && typeof imageSource.data === "string") {
+          shot.s3Key = imageSource.data;
+        }
+      }
+      if (shot.presignedUrl === null && typeof fields.presigned_url === "string") shot.presignedUrl = fields.presigned_url;
+      if (shot.text === null && fields.type === "text" && typeof fields.text === "string") shot.text = fields.text;
+      if (shot.sheetId === null && typeof fields.sheet_id === "number") shot.sheetId = fields.sheet_id;
+      if (shot.sheetName === null && typeof fields.sheet_name === "string") shot.sheetName = fields.sheet_name;
+      if (shot.range === null && typeof fields.range === "string") shot.range = fields.range;
+      if (shot.pageCount === null && typeof fields.page_count === "number") shot.pageCount = fields.page_count;
+      if (shot.savedAssetId === null && typeof fields.saved_asset_id === "string" && fields.saved_asset_id !== "") {
+        shot.savedAssetId = fields.saved_asset_id;
+      }
+    }
+  }
+  return shot;
+}
+const FRESH_URL_STALE_MS = 10 * 60 * 1e3;
+function useFreshPresignedUrlQuery({
+  s3Key,
+  fallbackUrl
+}) {
+  const config = useContext(AthenaContext);
+  const baseUrl = config ? config.backendUrl.replace(/\/api\/assistant-ui\/?$/, "") : null;
+  const identity = useAthenaQueryIdentity({
+    apiKey: config == null ? void 0 : config.apiKey,
+    token: config == null ? void 0 : config.token,
+    getToken: config == null ? void 0 : config.getToken
+  });
+  const reader = useId();
+  const query = useQuery({
+    queryKey: ["athena", "screenshot-presigned-url", baseUrl, ...identity, s3Key, reader],
+    enabled: s3Key !== null && config !== null,
+    staleTime: FRESH_URL_STALE_MS,
+    gcTime: 0,
+    refetchOnWindowFocus: false,
+    retry: false,
+    queryFn: async ({ signal }) => {
+      if (!config || baseUrl === null) return null;
+      const response = await sendWithCredentialRenewal(
+        config,
+        (headers) => fetch(`${baseUrl}/api/asset/screenshot-presigned-url`, {
+          method: "POST",
+          headers: { "Content-Type": "application/json", ...headers },
+          body: JSON.stringify({ screenshot_key: s3Key }),
+          signal
+        })
+      );
+      if (!response.ok) return null;
+      const data = await response.json();
+      return typeof data.presigned_url === "string" ? data.presigned_url : null;
+    }
+  });
+  return query.data ?? fallbackUrl;
+}
+function useScreenshotUrl({
+  s3Key,
+  fallbackUrl
+}) {
+  return useFreshPresignedUrlQuery({ s3Key, fallbackUrl });
+}
+function ScreenshotQueryScope({ children }) {
+  const queryClient = useAthenaQueryClient();
+  return /* @__PURE__ */ jsx(QueryClientProvider, { client: queryClient, children });
+}
+function createStudioScreenshotToolUI(spec) {
+  const Card = ({ toolName, args, result, artifact, status }) => {
+    var _a;
+    const typedArgs = args;
+    const shot = useMemo(() => readStudioScreenshot(result, artifact), [result, artifact]);
+    const imageUrl = useScreenshotUrl({ s3Key: shot.s3Key, fallbackUrl: shot.presignedUrl });
+    const openAsset = useAssetPanelStore((s) => s.openAsset);
+    const assetId = typeof (typedArgs == null ? void 0 : typedArgs.asset_id) === "string" && typedArgs.asset_id !== "" ? typedArgs.asset_id : null;
+    const label = spec.label({ args: typedArgs, shot });
+    const detail = ((_a = spec.detail) == null ? void 0 : _a.call(spec, { args: typedArgs, shot })) ?? "";
+    const isRunning = (status == null ? void 0 : status.type) === "running";
+    const isComplete = (status == null ? void 0 : status.type) === "complete" || !status;
+    const hasScreenshot = isComplete && (shot.presignedUrl !== null || shot.s3Key !== null);
+    const failed = isComplete && !hasScreenshot;
+    const errorMsg = (status == null ? void 0 : status.type) === "incomplete" ? String(status.error ?? status.reason ?? "Capture failed") : null;
+    const title = isRunning ? `Capturing screenshot of ${label}...` : failed || errorMsg ? `Failed to capture screenshot of ${label}` : `Screenshot captured for ${label}${detail ? ` ${detail}` : ""}`;
+    const open = assetId ? spec.open({ args: typedArgs, shot }) : null;
+    return /* @__PURE__ */ jsxs(
+      ToolCard,
+      {
+        icon: spec.icon,
+        status: failed ? "incomplete" : (status == null ? void 0 : status.type) ?? "complete",
+        title: truncate(title, 90),
+        toolName,
+        args: typedArgs,
+        result,
+        error: errorMsg ?? (failed ? spec.failureNote : null),
+        children: [
+          hasScreenshot && imageUrl ? /* @__PURE__ */ jsxs("div", { "data-slot": "studio-screenshot", className: "border-t border-border/40 p-3", children: [
+            /* @__PURE__ */ jsx("div", { className: "overflow-hidden rounded-lg border border-border/50 bg-muted/20", children: /* @__PURE__ */ jsx(
+              "img",
+              {
+                src: imageUrl,
+                alt: `Screenshot of ${label}`,
+                className: spec.imageClassName,
+                loading: "lazy"
+              }
+            ) }),
+            shot.savedAssetId ? /* @__PURE__ */ jsxs("div", { className: "mt-2 flex items-center gap-2 text-[11px] text-muted-foreground", children: [
+              /* @__PURE__ */ jsx("span", { children: "Saved as asset:" }),
+              /* @__PURE__ */ jsx(
+                "button",
+                {
+                  type: "button",
+                  onClick: () => openAsset(shot.savedAssetId),
+                  className: "rounded font-mono text-foreground/80 hover:underline",
+                  children: truncate(shot.savedAssetId, 28)
+                }
+              )
+            ] }) : null
+          ] }) : null,
+          hasScreenshot && open && assetId ? /* @__PURE__ */ jsx("div", { className: "flex flex-wrap gap-2 border-t border-border/40 px-4 py-2", children: /* @__PURE__ */ jsxs(
+            "button",
+            {
+              type: "button",
+              onClick: () => openAsset(assetId, open.options),
+              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" }),
+                open.label
+              ]
+            }
+          ) }) : null
+        ]
+      }
+    );
+  };
+  const Scoped = (props) => /* @__PURE__ */ jsx(ScreenshotQueryScope, { children: /* @__PURE__ */ jsx(Card, { ...props }) });
+  return Scoped;
+}
+function sheetLabel({ args, shot }) {
+  const name = shot.sheetName ?? (typeof (args == null ? void 0 : args.sheet_name) === "string" ? args.sheet_name : null);
+  const id = shot.sheetId ?? (typeof (args == null ? void 0 : args.sheet_id) === "number" ? args.sheet_id : null);
+  if (name) return `"${name}"`;
+  if (id !== null) return `sheet ${id}`;
+  return "active sheet";
+}
+function sheetRange({ args, shot }) {
+  return shot.range ?? (typeof (args == null ? void 0 : args.cell_range) === "string" ? args.cell_range : null);
+}
+const CaptureSheetScreenshotToolUIImpl = createStudioScreenshotToolUI({
+  icon: CameraIcon,
+  label: sheetLabel,
+  detail: (params) => {
+    const range = sheetRange(params);
+    return range ? `(${range})` : "";
+  },
+  imageClassName: "w-full object-contain",
+  failureNote: "The screenshot could not be captured. Check that the workbook exists and the sheet or range is valid.",
+  open: (params) => {
+    const range = sheetRange(params);
+    return {
+      label: range ? `Open ${range}` : "Open spreadsheet",
+      options: { type: "spreadsheet" }
+    };
+  }
+});
+const CaptureSheetScreenshotToolUI = memo(
+  CaptureSheetScreenshotToolUIImpl
+);
+CaptureSheetScreenshotToolUI.displayName = "CaptureSheetScreenshotToolUI";
+const PAGE_IMAGE_CLASS = "aspect-[8.5/11] w-full object-contain";
+const CapturePageScreenshotToolUIImpl = createStudioScreenshotToolUI({
+  icon: FileTextIcon,
+  label: ({ args, shot }) => {
+    const page = typeof (args == null ? void 0 : args.page_number) === "number" ? args.page_number : 1;
+    return shot.pageCount !== null ? `page ${page} of ${shot.pageCount}` : `page ${page}`;
+  },
+  imageClassName: PAGE_IMAGE_CLASS,
+  failureNote: "The screenshot could not be captured. Check that the document exists and the page number is within range.",
+  open: () => ({ label: "Open document", options: { type: "document" } })
+});
+const CapturePageScreenshotToolUI = memo(
+  CapturePageScreenshotToolUIImpl
+);
+CapturePageScreenshotToolUI.displayName = "CapturePageScreenshotToolUI";
+export {
+  CapturePageScreenshotToolUI,
+  CaptureSheetScreenshotToolUI,
+  ScreenshotQueryScope,
+  readStudioScreenshot,
+  useScreenshotUrl
+};
+//# sourceMappingURL=studio-screenshot-tool-uis.js.map