@athenaintel/react 0.2.0 → 0.3.0

package/dist/index.cjs

@@ -1378,7 +1378,7 @@ const createStoreImpl = (createState) => {
 };
 const createStore = ((createState) => createState ? createStoreImpl(createState) : createStoreImpl);
 const identity = (arg) => arg;
-function useStore(api, selector = identity) {
+function useStore$1(api, selector = identity) {
   const slice2 = React.useSyncExternalStore(
     api.subscribe,
     React.useCallback(() => selector(api.getState()), [api, selector]),
@@ -1389,7 +1389,7 @@ function useStore(api, selector = identity) {
 }
 const createImpl = (createState) => {
   const api = createStore(createState);
-  const useBoundStore = (selector) => useStore(api, selector);
+  const useBoundStore = (selector) => useStore$1(api, selector);
   Object.assign(useBoundStore, api);
   return useBoundStore;
 };
@@ -20749,7 +20749,11 @@ const useAthenaRuntime = (config2) => {
     systemPrompt,
     threadId: threadIdProp
   } = config2;
-  const threadId = React.useMemo(() => threadIdProp ?? crypto.randomUUID(), [threadIdProp]);
+  const generatedIdRef = React.useRef(null);
+  if (generatedIdRef.current === null) {
+    generatedIdRef.current = crypto.randomUUID();
+  }
+  const threadId = threadIdProp ?? generatedIdRef.current;
   const enabledTools = React.useMemo(
     () => Array.from(/* @__PURE__ */ new Set([...tools, ...frontendToolIds])),
     [tools, frontendToolIds]
@@ -24118,8 +24122,9 @@ function AthenaProvider({
   threadId
 }) {
   const frontendToolNames = React.useMemo(() => Object.keys(frontendTools), [frontendTools]);
+  const auiTools = React.useMemo(() => Tools({ toolkit: frontendTools }), [frontendTools]);
   const aui = useAui({
-    tools: Tools({ toolkit: frontendTools })
+    tools: auiTools
   });
   const parentAuthToken = useParentAuth();
   const effectiveToken = tokenProp ?? parentAuthToken;
@@ -59526,6 +59531,31 @@ class Store {
     return this.atom.subscribe(toObserver(observerOrFn));
   }
 }
+function defaultCompare(a, b) {
+  return a === b;
+}
+function useStore(atom, selector, compare = defaultCompare) {
+  const subscribe = React.useCallback(
+    (handleStoreChange) => {
+      if (!atom) {
+        return () => {
+        };
+      }
+      const { unsubscribe } = atom.subscribe(handleStoreChange);
+      return unsubscribe;
+    },
+    [atom]
+  );
+  const boundGetSnapshot = React.useCallback(() => atom == null ? void 0 : atom.get(), [atom]);
+  const selectedSnapshot = withSelectorExports.useSyncExternalStoreWithSelector(
+    subscribe,
+    boundGetSnapshot,
+    boundGetSnapshot,
+    selector,
+    compare
+  );
+  return selectedSnapshot;
+}
 function createScopeRegistry() {
   return { entries: /* @__PURE__ */ new Map() };
 }
@@ -59561,52 +59591,13 @@ function getFilteredItems(cache, scope, query) {
   }
   return results;
 }
-function toolsToMenuItems(tools) {
-  return tools.map((t) => ({
-    id: t.id,
-    name: t.name,
-    type: t.type,
-    fallbackIcon: t.icon ?? (t.type === "toolkit" ? "🧰" : "🔧"),
-    description: t.description,
-    visibleInScopes: /* @__PURE__ */ new Set(["root"])
-  }));
-}
-function useMentionSuggestions(tools) {
-  const store = React.useMemo(() => {
-    const registry = createScopeRegistry();
-    const cache = createItemCache();
-    registerSource(registry, "root", async () => ({
-      items: [],
-      pagination: { hasMore: false }
-    }));
-    const items = toolsToMenuItems(tools);
-    addItems(cache, items, "root");
-    return new Store({ registry, cache });
-  }, []);
-  React.useEffect(() => {
-    const { cache } = store.state;
-    for (const item of cache.items.values()) {
-      item.visibleInScopes.delete("root");
-    }
-    for (const [id, item] of cache.items) {
-      if (item.visibleInScopes.size === 0) {
-        cache.items.delete(id);
-      }
-    }
-    const items = toolsToMenuItems(tools);
-    addItems(cache, items, "root");
-  }, [tools, store]);
-  return store;
-}
-function selectItems(store, scope, query) {
-  const { cache } = store.state;
-  const items = getFilteredItems(cache, scope, query);
-  return { items, isFetching: false, hasMore: false };
-}
 const MentionSuggestionList = React.forwardRef(({ store, query, command: command2, closeMenu }, ref) => {
   const [selectedIndex, setSelectedIndex] = React.useState(0);
   const scrollContainerRef = React.useRef(null);
-  const { items, isFetching } = selectItems(store, "root", query);
+  const { items, isFetching } = useStore(store, (state) => ({
+    items: getFilteredItems(state.cache, "root", query),
+    isFetching: false
+  }));
   React.useEffect(() => {
     setSelectedIndex(0);
   }, [query]);
@@ -59669,7 +59660,10 @@ const MentionSuggestionList = React.forwardRef(({ store, query, command: command
       },
       item.id
     )),
-    isFetching
+    isFetching && /* @__PURE__ */ jsxRuntime.jsxs("div", { className: "flex items-center gap-2 px-2 py-1.5 text-sm text-muted-foreground", children: [
+      /* @__PURE__ */ jsxRuntime.jsx("span", { className: "h-4 w-4 animate-spin rounded-full border-2 border-current border-t-transparent" }),
+      "Loading..."
+    ] })
   ] }) });
 });
 MentionSuggestionList.displayName = "MentionSuggestionList";
@@ -59859,6 +59853,45 @@ const MentionSuggestionsExtension = Extension.create({
     ];
   }
 });
+function toolsToMenuItems(tools) {
+  return tools.map((t) => ({
+    id: t.id,
+    name: t.name,
+    type: t.type,
+    fallbackIcon: t.icon ?? (t.type === "toolkit" ? "🧰" : "🔧"),
+    description: t.description,
+    visibleInScopes: /* @__PURE__ */ new Set(["root"])
+  }));
+}
+function useMentionSuggestions(tools) {
+  const storeRef = React.useRef(null);
+  if (!storeRef.current) {
+    const registry = createScopeRegistry();
+    const cache = createItemCache();
+    registerSource(registry, "root", async () => ({
+      items: [],
+      pagination: { hasMore: false }
+    }));
+    const items = toolsToMenuItems(tools);
+    addItems(cache, items, "root");
+    storeRef.current = new Store({ registry, cache });
+  }
+  const store = storeRef.current;
+  React.useEffect(() => {
+    store.setState((prev) => {
+      const newCache = createItemCache();
+      for (const [id, item] of prev.cache.items) {
+        if (item.visibleInScopes.size > 0 && !item.visibleInScopes.has("root")) {
+          newCache.items.set(id, { ...item, visibleInScopes: new Set(item.visibleInScopes) });
+        }
+      }
+      const items = toolsToMenuItems(tools);
+      addItems(newCache, items, "root");
+      return { ...prev, cache: newCache };
+    });
+  }, [tools, store]);
+  return store;
+}
 const TiptapComposer = ({ tools = [] }) => {
   const composerRuntime = useComposerRuntime();
   const editorRef = React.useRef(null);
@@ -60574,12 +60607,14 @@ function CollapsibleTrigger({ ...props }) {
 function CollapsibleContent({ ...props }) {
   return /* @__PURE__ */ jsxRuntime.jsx(CollapsibleContent$1, { "data-slot": "collapsible-content", ...props });
 }
-const useAssetPanelStore = create((set2) => ({
+const useAssetPanelStore = create((set2, get2) => ({
   isOpen: false,
   tabs: [],
   activeTabId: null,
   viewMode: "tabs",
   isFullscreen: false,
+  autoOpenGeneration: 0,
+  autoOpenedAssets: /* @__PURE__ */ new Map(),
   openAsset: (assetId, meta) => set2((s) => {
     const existing = s.tabs.find((t) => t.id === assetId);
     if (existing) {
@@ -60607,39 +60642,59 @@ const useAssetPanelStore = create((set2) => ({
   setActiveTab: (assetId) => set2({ activeTabId: assetId, viewMode: "tabs" }),
   setViewMode: (mode) => set2({ viewMode: mode }),
   closePanel: () => set2({ isOpen: false, tabs: [], activeTabId: null, viewMode: "tabs", isFullscreen: false }),
-  toggleFullscreen: () => set2((s) => ({ isFullscreen: !s.isFullscreen }))
+  toggleFullscreen: () => set2((s) => ({ isFullscreen: !s.isFullscreen })),
+  resetAutoOpen: () => set2((s) => ({ autoOpenGeneration: s.autoOpenGeneration + 1 })),
+  markAutoOpened: (assetId) => {
+    const state = get2();
+    if (state.autoOpenedAssets.get(assetId) === state.autoOpenGeneration) {
+      return false;
+    }
+    state.autoOpenedAssets.set(assetId, state.autoOpenGeneration);
+    return true;
+  }
 }));
 const ANIMATION_DURATION = 200;
 const TOOL_META = {
-  search: { displayName: "Web Search", icon: Search, describer: (a) => a.query ? `"${a.query}"` : "" },
-  browse: { displayName: "Browse Page", icon: Globe, describer: (a) => a.url ?? "" },
-  search_email: { displayName: "Email Search", icon: Mail, describer: (a) => a.query ? `"${a.query}"` : "" },
-  unified_email_search: { displayName: "Email Search", icon: Mail, describer: (a) => a.query ? `"${a.query}"` : "" },
-  create_email_draft: { displayName: "Create Email Draft", icon: Mail },
-  unified_email_create_draft: { displayName: "Create Email Draft", icon: Mail },
-  edit_email_draft: { displayName: "Edit Email Draft", icon: Mail },
-  unified_edit_email_draft: { displayName: "Edit Email Draft", icon: Mail },
-  search_contacts: { displayName: "Search Contacts", icon: Search },
-  search_calendar_events: { displayName: "Search Calendar", icon: Search },
-  read_full_asset: { displayName: "Read Asset", icon: BookOpen },
-  create_new_document: { displayName: "Create Document", icon: FileText },
-  create_document_from_markdown: { displayName: "Create Document", icon: FileText },
-  append_markdown_to_athena_document: { displayName: "Write to Document", icon: PenLine },
-  replace_markdown_in_athena_document: { displayName: "Update Document", icon: PenLine },
-  create_new_sheet: { displayName: "Create Spreadsheet", icon: ChartColumn },
-  update_sheet_range: { displayName: "Update Spreadsheet", icon: ChartColumn },
-  run_python_code: { displayName: "Run Python", icon: Code },
-  run_sql_query_tool: { displayName: "Run SQL Query", icon: Database },
-  create_database: { displayName: "Create Database", icon: Database },
-  run_database_sql: { displayName: "Run Database Query", icon: Database },
-  computer_asset_exec: { displayName: "Run Command", icon: Monitor },
-  create_new_notebook: { displayName: "Create Notebook", icon: BookOpen },
-  execute_cell: { displayName: "Execute Cell", icon: Code },
-  create_new_aop: { displayName: "Create AOP", icon: Brain },
-  execute_aop: { displayName: "Execute AOP", icon: Brain },
-  list_contents: { displayName: "List Files", icon: Search },
-  search_assets: { displayName: "Search Assets", icon: Search },
-  save_preference_as_memory: { displayName: "Save Preference", icon: Settings }
+  // Search & Browse
+  search: { displayName: "Searching the web", icon: Search, describer: (a) => a.query ? `"${a.query}"` : "" },
+  browse: { displayName: "Reading webpage", icon: Globe, describer: (a) => a.url ?? "" },
+  search_email: { displayName: "Searching emails", icon: Mail, describer: (a) => a.query ? `"${a.query}"` : "" },
+  unified_email_search: { displayName: "Searching emails", icon: Mail, describer: (a) => a.query ? `"${a.query}"` : "" },
+  search_contacts: { displayName: "Looking up contacts", icon: Search },
+  search_calendar_events: { displayName: "Checking calendar", icon: Search },
+  search_assets: { displayName: "Searching files", icon: Search },
+  list_contents: { displayName: "Browsing files", icon: Search },
+  // Email
+  create_email_draft: { displayName: "Drafting email", icon: Mail },
+  unified_email_create_draft: { displayName: "Drafting email", icon: Mail },
+  edit_email_draft: { displayName: "Editing email draft", icon: Mail },
+  unified_edit_email_draft: { displayName: "Editing email draft", icon: Mail },
+  // Documents
+  read_full_asset: { displayName: "Reading document", icon: BookOpen },
+  create_new_document: { displayName: "Creating document", icon: FileText },
+  create_document_from_markdown: { displayName: "Creating document", icon: FileText },
+  append_markdown_to_athena_document: { displayName: "Updating document", icon: PenLine },
+  replace_markdown_in_athena_document: { displayName: "Updating document", icon: PenLine },
+  // Spreadsheets
+  create_new_sheet: { displayName: "Creating spreadsheet", icon: ChartColumn },
+  update_sheet_range: { displayName: "Updating spreadsheet", icon: ChartColumn },
+  // Presentations
+  create_powerpoint_deck: { displayName: "Building presentation", icon: Monitor },
+  execute_presentation_code: { displayName: "Generating slides", icon: Monitor },
+  // Code & Data
+  run_python_code: { displayName: "Running analysis", icon: Code },
+  run_sql_query_tool: { displayName: "Querying data", icon: Database },
+  create_database: { displayName: "Setting up database", icon: Database },
+  run_database_sql: { displayName: "Querying database", icon: Database },
+  computer_asset_exec: { displayName: "Running command", icon: Monitor },
+  // Notebooks
+  create_new_notebook: { displayName: "Creating notebook", icon: BookOpen },
+  execute_cell: { displayName: "Running notebook cell", icon: Code },
+  // Workflows
+  create_new_aop: { displayName: "Creating workflow", icon: Brain },
+  execute_aop: { displayName: "Running workflow", icon: Brain },
+  // Preferences
+  save_preference_as_memory: { displayName: "Saving preference", icon: Settings }
 };
 function getToolMeta(toolName) {
   if (TOOL_META[toolName]) return TOOL_META[toolName];
@@ -60726,10 +60781,8 @@ function extractTitle(argsText, result) {
   }
   return null;
 }
-let assetGeneration$1 = 0;
-const autoOpenedAssets$1 = /* @__PURE__ */ new Map();
 function clearAutoOpenedAssets() {
-  assetGeneration$1++;
+  useAssetPanelStore.getState().resetAutoOpen();
 }
 function ToolFallbackRoot({
   className,
@@ -60764,7 +60817,7 @@ function ToolFallbackRoot({
       open: isOpen,
       onOpenChange: handleOpenChange,
       className: cn(
-        "aui-tool-fallback-root group/tool-fallback-root w-full rounded-xl border border-border/60 bg-background py-2.5 shadow-sm",
+        "aui-tool-fallback-root group/tool-fallback-root my-3 w-full rounded-xl border border-border/60 bg-background py-2.5 shadow-sm",
         className
       ),
       style: {
@@ -60835,16 +60888,16 @@ function ToolFallbackTrigger({
               ),
               children: [
                 /* @__PURE__ */ jsxRuntime.jsx("span", { children: isRunning ? /* @__PURE__ */ jsxRuntime.jsxs(jsxRuntime.Fragment, { children: [
-                  /* @__PURE__ */ jsxRuntime.jsx("b", { children: meta.displayName }),
+                  meta.displayName,
                   "..."
-                ] }) : /* @__PURE__ */ jsxRuntime.jsx("b", { children: meta.displayName }) }),
+                ] }) : meta.displayName }),
                 isRunning && /* @__PURE__ */ jsxRuntime.jsxs(
                   "span",
                   {
                     "aria-hidden": true,
                     className: "shimmer pointer-events-none absolute inset-0 motion-reduce:animate-none",
                     children: [
-                      /* @__PURE__ */ jsxRuntime.jsx("b", { children: meta.displayName }),
+                      meta.displayName,
                       "..."
                     ]
                   }
@@ -60978,17 +61031,19 @@ function AssetToolCard({
   const assetType = toolMetaToAssetType(toolName);
   const wasCompleteAtMount = React.useRef(isComplete);
   React.useEffect(() => {
-    if (isComplete && !isCancelled && assetId && !wasCompleteAtMount.current && autoOpenedAssets$1.get(assetId) !== assetGeneration$1) {
-      autoOpenedAssets$1.set(assetId, assetGeneration$1);
-      useAssetPanelStore.getState().openAsset(assetId, {
-        name: title ?? void 0,
-        type: assetType
-      });
+    if (isComplete && !isCancelled && assetId && !wasCompleteAtMount.current) {
+      const store = useAssetPanelStore.getState();
+      if (store.markAutoOpened(assetId)) {
+        store.openAsset(assetId, {
+          name: title ?? void 0,
+          type: assetType
+        });
+      }
     }
   }, [isComplete, isCancelled, assetId, title, assetType]);
   const success = isComplete && isResultSuccess(result);
   return /* @__PURE__ */ jsxRuntime.jsxs("div", { className: cn(
-    "aui-tool-fallback-root w-full rounded-xl border border-border/60 bg-background py-2.5 shadow-sm",
+    "aui-tool-fallback-root my-3 w-full rounded-xl border border-border/60 bg-background py-2.5 shadow-sm",
     isCancelled && "border-muted-foreground/30 bg-muted/30"
   ), children: [
     /* @__PURE__ */ jsxRuntime.jsxs("div", { className: "flex items-center gap-2.5 px-3 text-sm", children: [
@@ -61005,13 +61060,13 @@ function AssetToolCard({
         }
       ),
       /* @__PURE__ */ jsxRuntime.jsxs("div", { className: "flex min-w-0 grow flex-col items-start gap-0.5 text-left", children: [
-        /* @__PURE__ */ jsxRuntime.jsxs("span", { className: "leading-tight", children: [
-          /* @__PURE__ */ jsxRuntime.jsx("b", { children: meta.displayName }),
+        /* @__PURE__ */ jsxRuntime.jsxs("span", { className: "leading-tight font-medium", children: [
+          meta.displayName,
           isRunning && "..."
         ] }),
         !isRunning && title && /* @__PURE__ */ jsxRuntime.jsx("span", { className: "max-w-full truncate text-xs leading-snug text-muted-foreground", children: title })
       ] }),
-      assetId && isComplete && !isCancelled && /* @__PURE__ */ jsxRuntime.jsxs(
+      assetId && isComplete && !isCancelled && CREATE_ASSET_TOOLS.includes(toolName.toLowerCase()) && /* @__PURE__ */ jsxRuntime.jsxs(
         "button",
         {
           onClick: () => openAsset(assetId, { name: title ?? void 0, type: assetType }),
@@ -61433,11 +61488,15 @@ function normalizeResult(result) {
 function truncate(text2, max2) {
   return text2.length > max2 ? `${text2.slice(0, max2)}...` : text2;
 }
+function formatToolName(name) {
+  return name.replace(/_/g, " ").replace(/\b\w/g, (c) => c.toUpperCase());
+}
 function ToolCard({
   icon: Icon2,
   status,
   title,
   subtitle,
+  toolName,
   badge,
   error: error2,
   children
@@ -61445,7 +61504,7 @@ function ToolCard({
   const isRunning = status === "running";
   const isComplete = status === "complete";
   const isError = status === "incomplete";
-  return /* @__PURE__ */ jsxRuntime.jsxs("div", { className: "my-1.5 w-full overflow-hidden rounded-xl border border-border/60 bg-background shadow-sm", children: [
+  return /* @__PURE__ */ jsxRuntime.jsxs("div", { className: "my-3 w-full overflow-hidden rounded-xl border border-border/60 bg-background shadow-sm", children: [
     /* @__PURE__ */ jsxRuntime.jsxs("div", { className: "flex items-center gap-3 px-4 py-3", children: [
       /* @__PURE__ */ jsxRuntime.jsx(
         "div",
@@ -61461,11 +61520,12 @@ function ToolCard({
       ),
       /* @__PURE__ */ jsxRuntime.jsxs("div", { className: "min-w-0 flex-1", children: [
         /* @__PURE__ */ jsxRuntime.jsxs("div", { className: "flex items-center gap-2", children: [
-          /* @__PURE__ */ jsxRuntime.jsx("span", { className: "text-[13px] font-semibold text-foreground", children: title }),
+          /* @__PURE__ */ jsxRuntime.jsx("span", { className: "text-[13px] font-medium text-foreground", children: title }),
           isComplete && !error2 && /* @__PURE__ */ jsxRuntime.jsx(CircleCheck, { className: "size-3.5 text-emerald-500" })
         ] }),
         subtitle && /* @__PURE__ */ jsxRuntime.jsx("p", { className: "truncate text-[12px] text-muted-foreground", children: subtitle })
       ] }),
+      toolName && /* @__PURE__ */ jsxRuntime.jsx("span", { className: "shrink-0 rounded-md bg-muted/60 px-1.5 py-0.5 text-[10px] font-mono text-muted-foreground", children: formatToolName(toolName) }),
       badge && /* @__PURE__ */ jsxRuntime.jsx("span", { className: "shrink-0 rounded-full bg-muted px-2 py-0.5 text-[10px] font-medium text-muted-foreground", children: badge }),
       isRunning && /* @__PURE__ */ jsxRuntime.jsxs("div", { className: "flex items-center gap-1.5", children: [
         /* @__PURE__ */ jsxRuntime.jsx("div", { className: "h-1.5 w-10 animate-pulse rounded-full bg-blue-100" }),
@@ -61519,6 +61579,7 @@ function parseSearchResults(data) {
   });
 }
 const WebSearchToolUIImpl = ({
+  toolName,
   args,
   result,
   status
@@ -61537,11 +61598,12 @@ const WebSearchToolUIImpl = ({
       status: (status == null ? void 0 : status.type) ?? "complete",
       title: isRunning ? "Searching the web..." : "Web search",
       subtitle: query ? truncate(query, 80) : void 0,
+      toolName,
       badge: isComplete && results.length > 0 ? `${results.length} results` : void 0,
       error: errorMsg,
       children: isComplete && results.length > 0 && /* @__PURE__ */ jsxRuntime.jsx(ExpandableSection, { label: "Show search results", children: /* @__PURE__ */ jsxRuntime.jsx("div", { className: "flex flex-col gap-2.5", children: results.map((r2, i) => /* @__PURE__ */ jsxRuntime.jsxs("div", { className: "flex flex-col gap-0.5", children: [
         /* @__PURE__ */ jsxRuntime.jsxs("div", { className: "flex items-center gap-1.5", children: [
-          r2.url ? /* @__PURE__ */ jsxRuntime.jsx(
+          r2.url && /^https?:\/\//i.test(r2.url) ? /* @__PURE__ */ jsxRuntime.jsx(
             "a",
             {
               href: r2.url,
@@ -61550,7 +61612,7 @@ const WebSearchToolUIImpl = ({
               className: "text-[12px] font-medium text-blue-600 hover:underline",
               children: r2.title || r2.url
             }
-          ) : /* @__PURE__ */ jsxRuntime.jsx("span", { className: "text-[12px] font-medium text-foreground", children: r2.title || "Result" }),
+          ) : /* @__PURE__ */ jsxRuntime.jsx("span", { className: "text-[12px] font-medium text-foreground", children: r2.title || r2.url || "Result" }),
           r2.url && /* @__PURE__ */ jsxRuntime.jsx(ExternalLink, { className: "size-3 shrink-0 text-muted-foreground" })
         ] }),
         r2.url && /* @__PURE__ */ jsxRuntime.jsx("span", { className: "text-[10px] text-muted-foreground", children: truncate(r2.url, 60) }),
@@ -61564,6 +61626,7 @@ const WebSearchToolUI = React.memo(
 );
 WebSearchToolUI.displayName = "WebSearchToolUI";
 const BrowseToolUIImpl = ({
+  toolName,
   args,
   result,
   status
@@ -61592,6 +61655,7 @@ const BrowseToolUIImpl = ({
       status: (status == null ? void 0 : status.type) ?? "complete",
       title: isRunning ? "Browsing page..." : pageTitle ? truncate(pageTitle, 50) : "Browsed page",
       subtitle: displayUrl,
+      toolName,
       error: errorMsg,
       children: isComplete && pageContent && /* @__PURE__ */ jsxRuntime.jsx(ExpandableSection, { label: "Show page content", children: /* @__PURE__ */ jsxRuntime.jsx("pre", { className: "whitespace-pre-wrap break-words text-[11px] leading-relaxed text-muted-foreground", children: truncate(pageContent, 3e3) }) })
     }
@@ -61615,6 +61679,7 @@ function parseEmailResults(data) {
   });
 }
 const EmailSearchToolUIImpl = ({
+  toolName,
   args,
   result,
   status
@@ -61633,6 +61698,7 @@ const EmailSearchToolUIImpl = ({
       status: (status == null ? void 0 : status.type) ?? "complete",
       title: isRunning ? "Searching emails..." : "Email search",
       subtitle: query ? truncate(query, 80) : void 0,
+      toolName,
       badge: isComplete && emails.length > 0 ? `${emails.length} emails` : void 0,
       error: errorMsg,
       children: isComplete && emails.length > 0 && /* @__PURE__ */ jsxRuntime.jsx(ExpandableSection, { label: "Show email results", children: /* @__PURE__ */ jsxRuntime.jsx("div", { className: "flex flex-col divide-y divide-border/30", children: emails.map((email, i) => /* @__PURE__ */ jsxRuntime.jsxs("div", { className: cn("flex flex-col gap-0.5 py-2", i === 0 && "pt-0"), children: [
@@ -61657,16 +61723,15 @@ function extractAssetId(result) {
   if (typeof id === "string" && id.startsWith("asset_")) return id;
   return null;
 }
-let assetGeneration = 0;
-const autoOpenedAssets = /* @__PURE__ */ new Map();
 function resetAssetAutoOpen() {
-  assetGeneration++;
+  useAssetPanelStore.getState().resetAutoOpen();
 }
 function CreateAssetToolUIImpl({
   icon: Icon2,
   assetType,
   runningLabel,
   doneLabel,
+  toolName,
   args,
   result,
   status
@@ -61683,12 +61748,14 @@ function CreateAssetToolUIImpl({
   const openAsset = useAssetPanelStore((s) => s.openAsset);
   const wasCompleteAtMount = React.useRef(isComplete);
   React.useEffect(() => {
-    if (isComplete && !isCancelled && assetId && !wasCompleteAtMount.current && autoOpenedAssets.get(assetId) !== assetGeneration) {
-      autoOpenedAssets.set(assetId, assetGeneration);
-      useAssetPanelStore.getState().openAsset(assetId, {
-        name: createdName || name || void 0,
-        type: assetType
-      });
+    if (isComplete && !isCancelled && assetId && !wasCompleteAtMount.current) {
+      const store = useAssetPanelStore.getState();
+      if (store.markAutoOpened(assetId)) {
+        store.openAsset(assetId, {
+          name: createdName || name || void 0,
+          type: assetType
+        });
+      }
     }
   }, [isComplete, isCancelled, assetId, createdName, name, assetType]);
   const handleOpen = () => {
@@ -61706,6 +61773,7 @@ function CreateAssetToolUIImpl({
       status: (status == null ? void 0 : status.type) ?? "complete",
       title: isRunning ? runningLabel : doneLabel,
       subtitle: createdName || name || void 0,
+      toolName,
       error: errorMsg,
       children: assetId && isComplete && !isCancelled && /* @__PURE__ */ jsxRuntime.jsx("div", { className: "border-t border-border/40 px-4 py-2", children: /* @__PURE__ */ jsxRuntime.jsxs(
         "button",
@@ -61765,6 +61833,7 @@ const CreatePresentationToolUI = React.memo(
 );
 CreatePresentationToolUI.displayName = "CreatePresentationToolUI";
 const CreateEmailDraftToolUIImpl = ({
+  toolName,
   args,
   result,
   status
@@ -61780,6 +61849,7 @@ const CreateEmailDraftToolUIImpl = ({
       icon: Mail,
       status: (status == null ? void 0 : status.type) ?? "complete",
       title: isRunning ? "Creating email draft..." : "Email draft created",
+      toolName,
       subtitle: subject ? truncate(subject, 60) : to ? `To: ${truncate(to, 40)}` : void 0,
       error: errorMsg
     }
@@ -61799,7 +61869,6 @@ const TOOL_UI_REGISTRY = {
   create_new_document: CreateDocumentToolUI,
   create_document_from_markdown: CreateDocumentToolUI,
   create_new_sheet: CreateSheetToolUI,
-  update_sheet_range: CreateSheetToolUI,
   create_powerpoint_deck: CreatePresentationToolUI
 };
 const falsyToString = (value) => typeof value === "boolean" ? `${value}` : value === 0 ? "0" : value;
@@ -62098,7 +62167,8 @@ function useAssetEmbed(assetId, options = {
       setError(null);
       return;
     }
-    const cached = embedCache.get(assetId);
+    const cacheKey = `${assetId}:${readOnly}:${token ?? apiKey ?? "anon"}`;
+    const cached = embedCache.get(cacheKey);
     if (cached && cached.expiresAt > Date.now() / 1e3) {
       setEmbedUrl(cached.url);
       setIsLoading(false);
@@ -62134,7 +62204,7 @@ function useAssetEmbed(assetId, options = {
       }
       return res.json();
     }).then((data) => {
-      embedCache.set(assetId, { url: data.embed_url, expiresAt: data.expires_at });
+      embedCache.set(`${assetId}:${readOnly}:${token ?? apiKey ?? "anon"}`, { url: data.embed_url, expiresAt: data.expires_at });
       setEmbedUrl(data.embed_url);
       setIsLoading(false);
     }).catch((err) => {