@athenaintel/react 0.9.15 → 0.9.16

package/dist/index.cjs

@@ -58875,14 +58875,20 @@ const useAssetPanelStore = create()(
         const existing = s.tabs.find((t) => t.id === assetId);
         if (existing) {
           const tabs = meta ? s.tabs.map(
-            (t) => t.id === assetId ? { ...t, name: meta.name ?? t.name, type: meta.type ?? t.type } : t
+            (t) => t.id === assetId ? {
+              ...t,
+              name: meta.name ?? t.name,
+              type: meta.type ?? t.type,
+              embedSearchParams: meta.embedSearchParams ?? t.embedSearchParams
+            } : t
           ) : s.tabs;
           return { isOpen: true, tabs, activeTabId: assetId };
         }
         const newTab = {
           id: assetId,
           name: (meta == null ? void 0 : meta.name) ?? null,
-          type: (meta == null ? void 0 : meta.type) ?? "unknown"
+          type: (meta == null ? void 0 : meta.type) ?? "unknown",
+          embedSearchParams: (meta == null ? void 0 : meta.embedSearchParams) ?? null
         };
         return { isOpen: true, tabs: [...s.tabs, newTab], activeTabId: assetId };
       }),
@@ -58943,13 +58949,43 @@ const ASSET_TYPE_ALIASES = {
   super_document: "document"
 };
 const isAthenaSpacesPath = (url) => url.pathname.replace(/\/+$/, "") === "/dashboard/spaces";
-const normalizeAssetType = (value) => {
+const normalizeAssetType$1 = (value) => {
   if (!value) {
     return null;
   }
   const normalizedValue = value.trim().replace(/^athena\//, "").toLowerCase().replace(/-/g, "_");
   return ASSET_TYPE_ALIASES[normalizedValue] ?? null;
 };
+const isLikelyAssetId = (value) => !!value && /^(asset|thread|project|collection)_[a-z0-9-]+$/i.test(value.trim());
+const sanitizeDisplayName = (value) => {
+  if (!value) {
+    return null;
+  }
+  const trimmedValue = value.trim();
+  if (!trimmedValue) {
+    return null;
+  }
+  if (trimmedValue.startsWith('"') && trimmedValue.endsWith('"') || trimmedValue.startsWith("'") && trimmedValue.endsWith("'")) {
+    return trimmedValue.slice(1, -1).trim() || null;
+  }
+  return trimmedValue;
+};
+const getCitationDisplayName = ({
+  assetName,
+  linkText
+}) => {
+  const sanitizedAssetName = sanitizeDisplayName(assetName);
+  const sanitizedLinkText = sanitizeDisplayName(linkText);
+  const genericLinkTextPattern = /^(this|that|the)\s+asset$/i;
+  if (sanitizedAssetName && !isLikelyAssetId(sanitizedAssetName)) {
+    return sanitizedAssetName;
+  }
+  if (sanitizedLinkText && !genericLinkTextPattern.test(sanitizedLinkText)) {
+    return sanitizedLinkText;
+  }
+  return sanitizedAssetName ?? sanitizedLinkText;
+};
+const getEmbedSearchParams = (url) => Object.fromEntries(url.searchParams.entries());
 const getOrigin = (value) => {
   if (!value) {
     return null;
@@ -59031,7 +59067,11 @@ const parseAthenaCitationLink = ({
     url,
     assetId,
     assetIds,
-    assetType: normalizeAssetType(url.searchParams.get("asset_type") ?? url.searchParams.get("type"))
+    assetType: normalizeAssetType$1(
+      url.searchParams.get("asset_type") ?? url.searchParams.get("assetType") ?? url.searchParams.get("type")
+    ),
+    assetName: sanitizeDisplayName(url.searchParams.get("name")),
+    embedSearchParams: getEmbedSearchParams(url)
   };
 };
 const isAthenaCitationUrl = ({
@@ -59079,12 +59119,19 @@ const useAthenaLinkClickHandler = () => {
       const citationMetadata = citationLink ? {
         assetId: citationLink.assetId,
         assetIds: citationLink.assetIds,
-        assetType: citationLink.assetType
+        assetType: citationLink.assetType,
+        assetName: citationLink.assetName,
+        embedSearchParams: citationLink.embedSearchParams
       } : null;
       const openInAssetPanel = citationLink ? () => {
+        const displayName = getCitationDisplayName({
+          assetName: citationLink.assetName,
+          linkText: trimmedLinkText
+        });
         openAsset(citationLink.assetId, {
-          name: trimmedLinkText ?? void 0,
-          type: citationLink.assetType ?? void 0
+          name: displayName ?? void 0,
+          type: citationLink.assetType ?? void 0,
+          embedSearchParams: citationLink.embedSearchParams
         });
       } : void 0;
       const context2 = {
@@ -63524,6 +63571,12 @@ function extractAssetId(result) {
   if (typeof id === "string" && id.startsWith("asset_")) return id;
   return null;
 }
+function extractAssetTitle(result) {
+  const data = normalizeResult(result);
+  if (!data) return null;
+  const title = data.assetTitle ?? data.asset_title ?? data.title ?? data.name;
+  return typeof title === "string" && title.trim().length > 0 ? title.trim() : null;
+}
 function resetAssetAutoOpen() {
   useAssetPanelStore.getState().resetAutoOpen();
 }
@@ -64243,6 +64296,7 @@ const OpenAssetToolUIImpl = ({
   const typedArgs = args;
   const argsAssetId = (typedArgs == null ? void 0 : typedArgs.asset_id) ?? (typedArgs == null ? void 0 : typedArgs.assetId) ?? "";
   const resultAssetId = extractAssetId(result);
+  const resultAssetTitle = extractAssetTitle(result);
   const assetId = resultAssetId ?? (argsAssetId.startsWith("asset_") ? argsAssetId : null);
   const isRunning = (status == null ? void 0 : status.type) === "running";
   const isComplete = (status == null ? void 0 : status.type) === "complete" || !status;
@@ -64254,10 +64308,10 @@ const OpenAssetToolUIImpl = ({
     if (isComplete && !isCancelled && assetId && !wasCompleteAtMount.current) {
       const store = useAssetPanelStore.getState();
       if (store.markAutoOpened(assetId)) {
-        store.openAsset(assetId);
+        store.openAsset(assetId, { name: resultAssetTitle ?? void 0 });
       }
     }
-  }, [isComplete, isCancelled, assetId]);
+  }, [isComplete, isCancelled, assetId, resultAssetTitle]);
   return /* @__PURE__ */ jsxRuntime.jsx(
     ToolCard,
     {
@@ -64273,7 +64327,7 @@ const OpenAssetToolUIImpl = ({
         "button",
         {
           type: "button",
-          onClick: () => openAsset(assetId),
+          onClick: () => openAsset(assetId, { name: resultAssetTitle ?? void 0 }),
           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(ExternalLink, { className: "size-3" }),
@@ -65198,14 +65252,58 @@ const AthenaUserMessage = ({
     children: /* @__PURE__ */ jsxRuntime.jsx("div", { className: "aui-user-message-content wrap-break-word rounded-2xl bg-muted px-4 py-2.5 text-foreground", children: /* @__PURE__ */ jsxRuntime.jsx(MessagePrimitiveParts, { components: { Text: TextComponent } }) })
   }
 );
+const EMBED_ASSET_TYPE_ALIASES = {
+  document: "document",
+  image: "document",
+  notebook: "notebook",
+  pdf: "document",
+  powerpoint_deck: "presentation",
+  presentation: "presentation",
+  puck_presentation: "presentation",
+  sheet: "spreadsheet",
+  spreadsheet: "spreadsheet",
+  super_document: "document"
+};
 const embedCache = /* @__PURE__ */ new Map();
+const normalizeAssetType = (value) => {
+  if (!value) {
+    return null;
+  }
+  const normalizedValue = value.trim().replace(/^athena\//, "").toLowerCase().replace(/-/g, "_");
+  return EMBED_ASSET_TYPE_ALIASES[normalizedValue] ?? null;
+};
+const appendEmbedSearchParams = ({
+  embedUrl,
+  embedSearchParams
+}) => {
+  if (!embedSearchParams || Object.keys(embedSearchParams).length === 0) {
+    return embedUrl;
+  }
+  const url = new URL(embedUrl);
+  for (const [key, value] of Object.entries(embedSearchParams)) {
+    if (!value) {
+      continue;
+    }
+    url.searchParams.set(key, value);
+  }
+  return url.toString();
+};
 function useAssetEmbed(assetId, options = {
   backendUrl: ""
 }) {
-  const { readOnly = false, expiresInSeconds = 60 * 60 * 24 * 30, backendUrl, apiKey, token } = options;
+  const {
+    readOnly = false,
+    expiresInSeconds = 60 * 60 * 24 * 30,
+    backendUrl,
+    apiKey,
+    token,
+    embedSearchParams
+  } = options;
   const [embedUrl, setEmbedUrl] = React.useState(null);
   const [isLoading, setIsLoading] = React.useState(false);
   const [error2, setError] = React.useState(null);
+  const [assetTitle, setAssetTitle] = React.useState(null);
+  const [assetType, setAssetType] = React.useState(null);
   const abortRef = React.useRef(null);
   React.useEffect(() => {
     var _a2;
@@ -65213,14 +65311,18 @@ function useAssetEmbed(assetId, options = {
       setEmbedUrl(null);
       setIsLoading(false);
       setError(null);
+      setAssetTitle(null);
+      setAssetType(null);
       return;
     }
-    const cacheKey = `${assetId}:${readOnly}:${token ?? apiKey ?? "anon"}`;
+    const cacheKey = `${assetId}:${readOnly}:${token ?? apiKey ?? "anon"}:${JSON.stringify(embedSearchParams ?? {})}`;
     const cached = embedCache.get(cacheKey);
     if (cached && cached.expiresAt > Date.now() / 1e3) {
       setEmbedUrl(cached.url);
       setIsLoading(false);
       setError(null);
+      setAssetTitle(cached.assetTitle);
+      setAssetType(cached.assetType);
       return;
     }
     const apiBaseUrl = backendUrl.replace(/\/api\/assistant-ui\/?$/, "");
@@ -65228,11 +65330,14 @@ function useAssetEmbed(assetId, options = {
     (_a2 = abortRef.current) == null ? void 0 : _a2.abort();
     const controller = new AbortController();
     abortRef.current = controller;
+    setEmbedUrl(null);
     setIsLoading(true);
     setError(null);
+    setAssetTitle(null);
+    setAssetType(null);
     const headers = { "Content-Type": "application/json" };
     if (token) {
-      headers["Authorization"] = `Bearer ${token}`;
+      headers.Authorization = `Bearer ${token}`;
     } else if (apiKey) {
       headers["X-API-KEY"] = apiKey;
     }
@@ -65252,17 +65357,31 @@ function useAssetEmbed(assetId, options = {
       }
       return res.json();
     }).then((data) => {
-      embedCache.set(`${assetId}:${readOnly}:${token ?? apiKey ?? "anon"}`, { url: data.embed_url, expiresAt: data.expires_at });
-      setEmbedUrl(data.embed_url);
+      const resolvedAssetType = normalizeAssetType(data.athena_converted_type) ?? normalizeAssetType(data.athena_original_type);
+      const resolvedEmbedUrl = appendEmbedSearchParams({
+        embedUrl: data.embed_url,
+        embedSearchParams
+      });
+      embedCache.set(cacheKey, {
+        url: resolvedEmbedUrl,
+        expiresAt: data.expires_at,
+        assetTitle: data.title ?? null,
+        assetType: resolvedAssetType
+      });
+      setEmbedUrl(resolvedEmbedUrl);
+      setAssetTitle(data.title ?? null);
+      setAssetType(resolvedAssetType);
       setIsLoading(false);
     }).catch((err) => {
-      if (err.name === "AbortError") return;
+      if (err.name === "AbortError") {
+        return;
+      }
       setError(err.message);
       setIsLoading(false);
     });
     return () => controller.abort();
-  }, [assetId, readOnly, expiresInSeconds, backendUrl, apiKey, token]);
-  return { embedUrl, isLoading, error: error2 };
+  }, [assetId, readOnly, expiresInSeconds, backendUrl, apiKey, token, embedSearchParams]);
+  return { embedUrl, isLoading, error: error2, assetTitle, assetType };
 }
 const ASSET_TYPE_CONFIG = {
   presentation: { icon: Presentation, label: "Presentation" },
@@ -65272,13 +65391,28 @@ const ASSET_TYPE_CONFIG = {
   unknown: { icon: File$1, label: "Asset" }
 };
 const AssetIframe = React.memo(
-  ({ tabId, tabName }) => {
+  ({ tabId, tabName, tabType, embedSearchParams }) => {
     const { backendUrl, apiKey, token } = useAthenaConfig();
-    const { embedUrl, isLoading, error: error2 } = useAssetEmbed(tabId, {
+    const { embedUrl, isLoading, error: error2, assetTitle, assetType } = useAssetEmbed(tabId, {
       backendUrl,
       apiKey,
-      token
+      token,
+      embedSearchParams
     });
+    React.useEffect(() => {
+      if (!assetTitle && !assetType) {
+        return;
+      }
+      const nextName = assetTitle ?? tabName;
+      const nextType = assetType ?? tabType;
+      if (nextName === tabName && nextType === tabType) {
+        return;
+      }
+      useAssetPanelStore.getState().openAsset(tabId, {
+        name: nextName ?? void 0,
+        type: nextType
+      });
+    }, [assetTitle, assetType, tabId, tabName, tabType]);
     if (isLoading) {
       return /* @__PURE__ */ jsxRuntime.jsx("div", { className: "flex h-full items-center justify-center", children: /* @__PURE__ */ jsxRuntime.jsxs("div", { className: "text-center", children: [
         /* @__PURE__ */ jsxRuntime.jsx(LoaderCircle, { className: "mx-auto size-6 animate-spin text-muted-foreground" }),
@@ -65363,7 +65497,15 @@ const PanelContent = ({
                     }
                   )
                 ] }),
-                /* @__PURE__ */ jsxRuntime.jsx("div", { className: "flex-1 overflow-hidden", children: /* @__PURE__ */ jsxRuntime.jsx(AssetIframe, { tabId: tab.id, tabName: tab.name }) })
+                /* @__PURE__ */ jsxRuntime.jsx("div", { className: "flex-1 overflow-hidden", children: /* @__PURE__ */ jsxRuntime.jsx(
+                  AssetIframe,
+                  {
+                    tabId: tab.id,
+                    tabName: tab.name,
+                    tabType: tab.type,
+                    embedSearchParams: tab.embedSearchParams
+                  }
+                ) })
               ]
             },
             tab.id
@@ -65373,7 +65515,15 @@ const PanelContent = ({
           "div",
           {
             className: isActive2 ? "h-full w-full" : "hidden",
-            children: /* @__PURE__ */ jsxRuntime.jsx(AssetIframe, { tabId: tab.id, tabName: tab.name })
+            children: /* @__PURE__ */ jsxRuntime.jsx(
+              AssetIframe,
+              {
+                tabId: tab.id,
+                tabName: tab.name,
+                tabType: tab.type,
+                embedSearchParams: tab.embedSearchParams
+              }
+            )
           },
           tab.id
         );