@athenaintel/react 0.9.16 → 0.9.17

package/dist/index.cjs

@@ -19867,14 +19867,20 @@ const joinExternalMessages = (messages) => {
               )
             );
         }
+        const { messages: existingMessages, ...toolCallRest } = toolCall;
         const updatedToolCall = {
-          ...toolCall,
+          ...toolCallRest,
           result: output.result,
           artifact: output.artifact,
           isError: output.isError,
-          messages: output.messages
+          ...(output.messages ?? existingMessages) !== void 0 && {
+            messages: output.messages ?? existingMessages
+          }
         };
-        bindExternalStoreMessage(updatedToolCall, [...getExternalStoreMessages(toolCall), output]);
+        bindExternalStoreMessage(updatedToolCall, [
+          ...getExternalStoreMessages(toolCallRest),
+          output
+        ]);
         assistantMessage.content[toolCallIdx] = updatedToolCall;
       } else {
         console.warn(
@@ -19883,10 +19889,14 @@ const joinExternalMessages = (messages) => {
       }
     } else {
       const role = output.role;
-      const content = (typeof output.content === "string" ? [{ type: "text", text: output.content }] : output.content).map((c) => {
+      const rawContent = typeof output.content === "string" ? [{ type: "text", text: output.content }] : Array.isArray(output.content) ? output.content : [];
+      const content = rawContent.flatMap((c) => {
+        if (!c || typeof c !== "object") {
+          return [];
+        }
         const mapped = { ...c };
         bindExternalStoreMessage(mapped, output);
-        return mapped;
+        return [mapped];
       });
       switch (role) {
         case "system":
@@ -20017,7 +20027,13 @@ const convertExternalMessages = (messages, callback, isRunning, metadata) => {
   const callbackResults = [];
   for (const message of messages) {
     const output = callback(message, metadata);
-    const outputs = Array.isArray(output) ? output : [output];
+    const rawOutputs = Array.isArray(output) ? output : [output];
+    const outputs = rawOutputs.filter(
+      (candidate) => candidate != null
+    );
+    if (outputs.length === 0) {
+      continue;
+    }
     const result = { input: message, outputs };
     callbackResults.push(result);
   }
@@ -20040,37 +20056,40 @@ const warnForUnknownMessagePartType = (type) => {
 };
 const contentToParts = (content) => {
   if (typeof content === "string") return [{ type: "text", text: content }];
-  return content.map((part) => {
+  return content.flatMap((part) => {
+    if (!part || typeof part !== "object" || !("type" in part)) {
+      return [];
+    }
     const type = part.type;
     switch (type) {
       case "text":
-        return { type: "text", text: part.text };
+        return [{ type: "text", text: part.text }];
       case "text_delta":
-        return { type: "text", text: part.text };
+        return [{ type: "text", text: part.text }];
       case "image_url":
         if (typeof part.image_url === "string") {
-          return { type: "image", image: part.image_url };
+          return [{ type: "image", image: part.image_url }];
         }
-        return {
+        return [{
           type: "image",
           image: part.image_url.url
-        };
+        }];
       case "thinking":
-        return { type: "reasoning", text: part.thinking };
+        return [{ type: "reasoning", text: part.thinking }];
       case "reasoning":
-        return {
+        return [{
           type: "reasoning",
           text: part.summary.map((s) => s.text).join("\n\n\n")
-        };
+        }];
       case "tool_use":
-        return null;
+        return [];
       case "input_json_delta":
-        return null;
+        return [];
       default:
         warnForUnknownMessagePartType(type);
-        return null;
+        return [];
     }
-  }).filter((a) => a !== null);
+  });
 };
 const getNumberAtPath = (value, path) => {
   let current = value;
@@ -20175,7 +20194,7 @@ const convertLangChainMessage = (message) => {
         (_b = message.additional_kwargs) == null ? void 0 : _b.reasoning,
         ...normalizedContent,
         ...((_c = message.additional_kwargs) == null ? void 0 : _c.tool_outputs) ?? []
-      ].filter((c) => c !== void 0);
+      ].filter((c) => c != null);
       return {
         role: "assistant",
         id: message.id,
@@ -20198,6 +20217,8 @@ const convertLangChainMessage = (message) => {
         isError: message.status === "error"
       };
     }
+    default:
+      return null;
   }
 };
 const convertLangChainToThreadMessages = (messages, isRunning = false, metadata = {}) => {
@@ -58875,20 +58896,14 @@ 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,
-              embedSearchParams: meta.embedSearchParams ?? t.embedSearchParams
-            } : t
+            (t) => t.id === assetId ? { ...t, name: meta.name ?? t.name, type: meta.type ?? t.type } : 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",
-          embedSearchParams: (meta == null ? void 0 : meta.embedSearchParams) ?? null
+          type: (meta == null ? void 0 : meta.type) ?? "unknown"
         };
         return { isOpen: true, tabs: [...s.tabs, newTab], activeTabId: assetId };
       }),
@@ -58949,43 +58964,13 @@ const ASSET_TYPE_ALIASES = {
   super_document: "document"
 };
 const isAthenaSpacesPath = (url) => url.pathname.replace(/\/+$/, "") === "/dashboard/spaces";
-const normalizeAssetType$1 = (value) => {
+const normalizeAssetType = (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;
@@ -59067,11 +59052,7 @@ const parseAthenaCitationLink = ({
     url,
     assetId,
     assetIds,
-    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)
+    assetType: normalizeAssetType(url.searchParams.get("asset_type") ?? url.searchParams.get("type"))
   };
 };
 const isAthenaCitationUrl = ({
@@ -59119,19 +59100,12 @@ const useAthenaLinkClickHandler = () => {
       const citationMetadata = citationLink ? {
         assetId: citationLink.assetId,
         assetIds: citationLink.assetIds,
-        assetType: citationLink.assetType,
-        assetName: citationLink.assetName,
-        embedSearchParams: citationLink.embedSearchParams
+        assetType: citationLink.assetType
       } : null;
       const openInAssetPanel = citationLink ? () => {
-        const displayName = getCitationDisplayName({
-          assetName: citationLink.assetName,
-          linkText: trimmedLinkText
-        });
         openAsset(citationLink.assetId, {
-          name: displayName ?? void 0,
-          type: citationLink.assetType ?? void 0,
-          embedSearchParams: citationLink.embedSearchParams
+          name: trimmedLinkText ?? void 0,
+          type: citationLink.assetType ?? void 0
         });
       } : void 0;
       const context2 = {
@@ -63571,12 +63545,6 @@ 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();
 }
@@ -64296,7 +64264,6 @@ 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;
@@ -64308,10 +64275,10 @@ const OpenAssetToolUIImpl = ({
     if (isComplete && !isCancelled && assetId && !wasCompleteAtMount.current) {
       const store = useAssetPanelStore.getState();
       if (store.markAutoOpened(assetId)) {
-        store.openAsset(assetId, { name: resultAssetTitle ?? void 0 });
+        store.openAsset(assetId);
       }
     }
-  }, [isComplete, isCancelled, assetId, resultAssetTitle]);
+  }, [isComplete, isCancelled, assetId]);
   return /* @__PURE__ */ jsxRuntime.jsx(
     ToolCard,
     {
@@ -64327,7 +64294,7 @@ const OpenAssetToolUIImpl = ({
         "button",
         {
           type: "button",
-          onClick: () => openAsset(assetId, { name: resultAssetTitle ?? void 0 }),
+          onClick: () => openAsset(assetId),
           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" }),
@@ -65252,58 +65219,14 @@ 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,
-    embedSearchParams
-  } = options;
+  const { readOnly = false, expiresInSeconds = 60 * 60 * 24 * 30, backendUrl, apiKey, token } = 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;
@@ -65311,18 +65234,14 @@ function useAssetEmbed(assetId, options = {
       setEmbedUrl(null);
       setIsLoading(false);
       setError(null);
-      setAssetTitle(null);
-      setAssetType(null);
       return;
     }
-    const cacheKey = `${assetId}:${readOnly}:${token ?? apiKey ?? "anon"}:${JSON.stringify(embedSearchParams ?? {})}`;
+    const cacheKey = `${assetId}:${readOnly}:${token ?? apiKey ?? "anon"}`;
     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\/?$/, "");
@@ -65330,14 +65249,11 @@ 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;
     }
@@ -65357,31 +65273,17 @@ function useAssetEmbed(assetId, options = {
       }
       return res.json();
     }).then((data) => {
-      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);
+      embedCache.set(`${assetId}:${readOnly}:${token ?? apiKey ?? "anon"}`, { url: data.embed_url, expiresAt: data.expires_at });
+      setEmbedUrl(data.embed_url);
       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, embedSearchParams]);
-  return { embedUrl, isLoading, error: error2, assetTitle, assetType };
+  }, [assetId, readOnly, expiresInSeconds, backendUrl, apiKey, token]);
+  return { embedUrl, isLoading, error: error2 };
 }
 const ASSET_TYPE_CONFIG = {
   presentation: { icon: Presentation, label: "Presentation" },
@@ -65391,28 +65293,13 @@ const ASSET_TYPE_CONFIG = {
   unknown: { icon: File$1, label: "Asset" }
 };
 const AssetIframe = React.memo(
-  ({ tabId, tabName, tabType, embedSearchParams }) => {
+  ({ tabId, tabName }) => {
     const { backendUrl, apiKey, token } = useAthenaConfig();
-    const { embedUrl, isLoading, error: error2, assetTitle, assetType } = useAssetEmbed(tabId, {
+    const { embedUrl, isLoading, error: error2 } = useAssetEmbed(tabId, {
       backendUrl,
       apiKey,
-      token,
-      embedSearchParams
+      token
     });
-    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" }),
@@ -65497,15 +65384,7 @@ const PanelContent = ({
                     }
                   )
                 ] }),
-                /* @__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
-                  }
-                ) })
+                /* @__PURE__ */ jsxRuntime.jsx("div", { className: "flex-1 overflow-hidden", children: /* @__PURE__ */ jsxRuntime.jsx(AssetIframe, { tabId: tab.id, tabName: tab.name }) })
               ]
             },
             tab.id
@@ -65515,15 +65394,7 @@ const PanelContent = ({
           "div",
           {
             className: isActive2 ? "h-full w-full" : "hidden",
-            children: /* @__PURE__ */ jsxRuntime.jsx(
-              AssetIframe,
-              {
-                tabId: tab.id,
-                tabName: tab.name,
-                tabType: tab.type,
-                embedSearchParams: tab.embedSearchParams
-              }
-            )
+            children: /* @__PURE__ */ jsxRuntime.jsx(AssetIframe, { tabId: tab.id, tabName: tab.name })
           },
           tab.id
         );