@athenaintel/react 0.9.16 → 0.9.17

package/dist/index.d.ts

@@ -51,7 +51,6 @@ export declare interface AssetPanelState {
     openAsset: (assetId: string, meta?: {
         name?: string;
         type?: AssetType;
-        embedSearchParams?: Record<string, string> | null;
     }) => void;
     closeTab: (assetId: string) => void;
     setActiveTab: (assetId: string) => void;
@@ -68,7 +67,6 @@ export declare interface AssetTab {
     id: string;
     name: string | null;
     type: AssetType;
-    embedSearchParams: Record<string, string> | null;
 }
 
 export declare type AssetType = 'presentation' | 'spreadsheet' | 'document' | 'notebook' | 'unknown';
@@ -148,8 +146,6 @@ export declare interface AthenaCitationLinkMetadata {
     assetId: string;
     assetIds: string[];
     assetType: AssetType | null;
-    assetName: string | null;
-    embedSearchParams: Record<string, string>;
 }
 
 /**
@@ -590,8 +586,6 @@ export declare interface ParsedAthenaCitationLink {
     assetId: string;
     assetIds: string[];
     assetType: AssetType | null;
-    assetName: string | null;
-    embedSearchParams: Record<string, string>;
 }
 
 declare interface QuoteContextValue {
@@ -1024,15 +1018,12 @@ export declare function useAssetEmbed(assetId: string | null, options?: UseAsset
 export declare interface UseAssetEmbedOptions {
     readOnly?: boolean;
     expiresInSeconds?: number;
-    embedSearchParams?: Record<string, string> | null;
 }
 
 export declare interface UseAssetEmbedResult {
     embedUrl: string | null;
     isLoading: boolean;
     error: string | null;
-    assetTitle: string | null;
-    assetType: AssetType | null;
 }
 
 export declare const useAssetPanelStore: UseBoundStore<Omit<StoreApi<AssetPanelState>, "setState" | "persist"> & {

package/dist/index.js

@@ -19849,14 +19849,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(
@@ -19865,10 +19871,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":
@@ -19999,7 +20009,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);
   }
@@ -20022,37 +20038,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;
@@ -20157,7 +20176,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,
@@ -20180,6 +20199,8 @@ const convertLangChainMessage = (message) => {
         isError: message.status === "error"
       };
     }
+    default:
+      return null;
   }
 };
 const convertLangChainToThreadMessages = (messages, isRunning = false, metadata = {}) => {
@@ -58857,20 +58878,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 };
       }),
@@ -58931,43 +58946,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;
@@ -59049,11 +59034,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 = ({
@@ -59101,19 +59082,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 = {
@@ -63553,12 +63527,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();
 }
@@ -64278,7 +64246,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;
@@ -64290,10 +64257,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__ */ jsx(
     ToolCard,
     {
@@ -64309,7 +64276,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__ */ jsx(ExternalLink, { className: "size-3" }),
@@ -65234,58 +65201,14 @@ const AthenaUserMessage = ({
     children: /* @__PURE__ */ jsx("div", { className: "aui-user-message-content wrap-break-word rounded-2xl bg-muted px-4 py-2.5 text-foreground", children: /* @__PURE__ */ 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] = useState(null);
   const [isLoading, setIsLoading] = useState(false);
   const [error2, setError] = useState(null);
-  const [assetTitle, setAssetTitle] = useState(null);
-  const [assetType, setAssetType] = useState(null);
   const abortRef = useRef(null);
   useEffect(() => {
     var _a2;
@@ -65293,18 +65216,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\/?$/, "");
@@ -65312,14 +65231,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;
     }
@@ -65339,31 +65255,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" },
@@ -65373,28 +65275,13 @@ const ASSET_TYPE_CONFIG = {
   unknown: { icon: File$1, label: "Asset" }
 };
 const AssetIframe = 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
     });
-    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__ */ jsx("div", { className: "flex h-full items-center justify-center", children: /* @__PURE__ */ jsxs("div", { className: "text-center", children: [
         /* @__PURE__ */ jsx(LoaderCircle, { className: "mx-auto size-6 animate-spin text-muted-foreground" }),
@@ -65479,15 +65366,7 @@ const PanelContent = ({
                     }
                   )
                 ] }),
-                /* @__PURE__ */ jsx("div", { className: "flex-1 overflow-hidden", children: /* @__PURE__ */ jsx(
-                  AssetIframe,
-                  {
-                    tabId: tab.id,
-                    tabName: tab.name,
-                    tabType: tab.type,
-                    embedSearchParams: tab.embedSearchParams
-                  }
-                ) })
+                /* @__PURE__ */ jsx("div", { className: "flex-1 overflow-hidden", children: /* @__PURE__ */ jsx(AssetIframe, { tabId: tab.id, tabName: tab.name }) })
               ]
             },
             tab.id
@@ -65497,15 +65376,7 @@ const PanelContent = ({
           "div",
           {
             className: isActive2 ? "h-full w-full" : "hidden",
-            children: /* @__PURE__ */ jsx(
-              AssetIframe,
-              {
-                tabId: tab.id,
-                tabName: tab.name,
-                tabType: tab.type,
-                embedSearchParams: tab.embedSearchParams
-              }
-            )
+            children: /* @__PURE__ */ jsx(AssetIframe, { tabId: tab.id, tabName: tab.name })
           },
           tab.id
         );