@smartspace/chat-ui 1.14.2-main.fb46d29 → 1.14.4-main.0af780f

package/dist/index.js

@@ -1,12 +1,12 @@
 import MuiButton from '@mui/material/Button';
 import IconButton from '@mui/material/IconButton';
-import { Loader2, Check, X, Paperclip, ArrowBigUp, Minimize2, AlertTriangle, FileImage, FileVideo, FileAudio, FileArchive, FileCode, FileSpreadsheet, Presentation, FileText, ChevronUp, ExternalLink, Copy, Download } from 'lucide-react';
+import { Loader2, Check, X, Paperclip, Square, ArrowBigUp, Minimize2, AlertTriangle, FileImage, FileVideo, FileAudio, FileArchive, FileCode, FileSpreadsheet, Presentation, FileText, ChevronUp, Download, Copy, Globe, ShieldAlert } from 'lucide-react';
 import * as React9 from 'react';
 import { createContext, forwardRef, useImperativeHandle, useRef, useState, useEffect, useMemo, useCallback, createElement, useContext } from 'react';
 import { createPortal } from 'react-dom';
+import { jsx, jsxs, Fragment } from 'react/jsx-runtime';
 import { useQuery, queryOptions, useQueryClient, useMutation, skipToken } from '@tanstack/react-query';
 import { toast } from 'sonner';
-import { jsx, jsxs, Fragment } from 'react/jsx-runtime';
 import { Editor, rootCtx, defaultValueCtx, editorViewOptionsCtx, editorViewCtx, serializerCtx, parserCtx, SchemaReady, nodeViewCtx, markViewCtx, schemaCtx, prosePluginsCtx, nodesCtx } from '@milkdown/core';
 import { history } from '@milkdown/kit/plugin/history';
 import { clipboard } from '@milkdown/plugin-clipboard';
@@ -221,6 +221,131 @@ var saveFile = (blob, fileName) => {
   a.click();
 };
 
+// src/shared/utils/threadId.ts
+var NEW_THREAD_ID = "__new__";
+function createThreadId() {
+  const cryptoObj = globalThis?.crypto;
+  return typeof cryptoObj?.randomUUID === "function" ? cryptoObj.randomUUID() : `${Date.now()}-${Math.random().toString(16).slice(2)}`;
+}
+var DRAFT_THREAD_PREFIX = "draft-";
+var DRAFT_THREAD_STORAGE_KEY = "ss:draftThreadIds";
+var draftThreadIds = /* @__PURE__ */ new Set();
+var hydratedFromStorage = false;
+function isBrowserStorageAvailable() {
+  try {
+    return typeof window !== "undefined" && !!window.sessionStorage;
+  } catch {
+    return false;
+  }
+}
+function hydrateFromStorageOnce() {
+  if (hydratedFromStorage) return;
+  hydratedFromStorage = true;
+  if (!isBrowserStorageAvailable()) return;
+  try {
+    const raw2 = window.sessionStorage.getItem(DRAFT_THREAD_STORAGE_KEY);
+    if (!raw2) return;
+    const parsed = JSON.parse(raw2);
+    if (!Array.isArray(parsed)) return;
+    for (const id of parsed) {
+      if (typeof id === "string" && id) draftThreadIds.add(id);
+    }
+  } catch {
+  }
+}
+function persistToStorage() {
+  if (!isBrowserStorageAvailable()) return;
+  try {
+    window.sessionStorage.setItem(
+      DRAFT_THREAD_STORAGE_KEY,
+      JSON.stringify(Array.from(draftThreadIds))
+    );
+  } catch {
+  }
+}
+function markDraftThreadId(threadId) {
+  if (!threadId) return;
+  hydrateFromStorageOnce();
+  draftThreadIds.add(threadId);
+  persistToStorage();
+}
+function unmarkDraftThreadId(threadId) {
+  if (!threadId) return;
+  hydrateFromStorageOnce();
+  draftThreadIds.delete(threadId);
+  persistToStorage();
+}
+function isDraftThreadId(threadId) {
+  if (!threadId) return false;
+  if (threadId.startsWith(DRAFT_THREAD_PREFIX)) return true;
+  hydrateFromStorageOnce();
+  return draftThreadIds.has(threadId);
+}
+function createDraftThreadId() {
+  const cryptoObj = globalThis?.crypto;
+  return typeof cryptoObj?.randomUUID === "function" ? cryptoObj.randomUUID() : `${Date.now()}-${Math.random().toString(16).slice(2)}`;
+}
+
+// src/domains/flowruns/queryKeys.ts
+var flowRunsKeys = {
+  all: ["flowruns"],
+  variables: (flowRunId) => [...flowRunsKeys.all, "variables", { flowRunId }],
+  mutations: () => [...flowRunsKeys.all, "mutations"],
+  updateVariable: (flowRunId, variableName) => [
+    ...flowRunsKeys.mutations(),
+    "updateVariable",
+    { flowRunId, variableName }
+  ],
+  cancel: () => [...flowRunsKeys.mutations(), "cancel"]
+};
+
+// src/domains/flowruns/mutations.ts
+function useCancelFlowRun() {
+  const service = useChatService();
+  return useMutation({
+    mutationKey: flowRunsKeys.cancel(),
+    mutationFn: async (flowRunId) => {
+      if (isDraftThreadId(flowRunId)) return;
+      await service.cancelFlowRun?.(flowRunId);
+    },
+    onError: (error) => {
+      console.error("Failed to cancel flow run:", error);
+      toast.error("Failed to stop the run");
+    }
+  });
+}
+function useUpdateFlowRunVariable() {
+  const qc = useQueryClient();
+  const service = useChatService();
+  return useMutation({
+    mutationKey: flowRunsKeys.updateVariable("", ""),
+    mutationFn: async ({
+      flowRunId,
+      variableName,
+      value
+    }) => {
+      if (isDraftThreadId(flowRunId)) return;
+      await service.updateFlowRunVariable({
+        flowRunId,
+        name: variableName,
+        value
+      });
+    },
+    onSuccess: (_data, variables) => {
+      if (variables?.flowRunId) {
+        qc.invalidateQueries({
+          queryKey: flowRunsKeys.variables(variables.flowRunId)
+        });
+      }
+    },
+    onError: (error) => {
+      console.error("Failed to update variable:", error);
+      toast.error("Failed to update variable");
+      throw error;
+    }
+  });
+}
+
 // ../../node_modules/.pnpm/@milkdown+exception@7.20.0/node_modules/@milkdown/exception/lib/index.js
 var ErrorCode = /* @__PURE__ */ (function(ErrorCode2) {
   ErrorCode2["docTypeError"] = "docTypeError";
@@ -1827,116 +1952,6 @@ var Button = React9.forwardRef(
   }
 );
 Button.displayName = "Button";
-
-// src/shared/utils/threadId.ts
-var NEW_THREAD_ID = "__new__";
-function createThreadId() {
-  const cryptoObj = globalThis?.crypto;
-  return typeof cryptoObj?.randomUUID === "function" ? cryptoObj.randomUUID() : `${Date.now()}-${Math.random().toString(16).slice(2)}`;
-}
-var DRAFT_THREAD_PREFIX = "draft-";
-var DRAFT_THREAD_STORAGE_KEY = "ss:draftThreadIds";
-var draftThreadIds = /* @__PURE__ */ new Set();
-var hydratedFromStorage = false;
-function isBrowserStorageAvailable() {
-  try {
-    return typeof window !== "undefined" && !!window.sessionStorage;
-  } catch {
-    return false;
-  }
-}
-function hydrateFromStorageOnce() {
-  if (hydratedFromStorage) return;
-  hydratedFromStorage = true;
-  if (!isBrowserStorageAvailable()) return;
-  try {
-    const raw2 = window.sessionStorage.getItem(DRAFT_THREAD_STORAGE_KEY);
-    if (!raw2) return;
-    const parsed = JSON.parse(raw2);
-    if (!Array.isArray(parsed)) return;
-    for (const id of parsed) {
-      if (typeof id === "string" && id) draftThreadIds.add(id);
-    }
-  } catch {
-  }
-}
-function persistToStorage() {
-  if (!isBrowserStorageAvailable()) return;
-  try {
-    window.sessionStorage.setItem(
-      DRAFT_THREAD_STORAGE_KEY,
-      JSON.stringify(Array.from(draftThreadIds))
-    );
-  } catch {
-  }
-}
-function markDraftThreadId(threadId) {
-  if (!threadId) return;
-  hydrateFromStorageOnce();
-  draftThreadIds.add(threadId);
-  persistToStorage();
-}
-function unmarkDraftThreadId(threadId) {
-  if (!threadId) return;
-  hydrateFromStorageOnce();
-  draftThreadIds.delete(threadId);
-  persistToStorage();
-}
-function isDraftThreadId(threadId) {
-  if (!threadId) return false;
-  if (threadId.startsWith(DRAFT_THREAD_PREFIX)) return true;
-  hydrateFromStorageOnce();
-  return draftThreadIds.has(threadId);
-}
-function createDraftThreadId() {
-  const cryptoObj = globalThis?.crypto;
-  return typeof cryptoObj?.randomUUID === "function" ? cryptoObj.randomUUID() : `${Date.now()}-${Math.random().toString(16).slice(2)}`;
-}
-
-// src/domains/flowruns/queryKeys.ts
-var flowRunsKeys = {
-  all: ["flowruns"],
-  variables: (flowRunId) => [...flowRunsKeys.all, "variables", { flowRunId }],
-  mutations: () => [...flowRunsKeys.all, "mutations"],
-  updateVariable: (flowRunId, variableName) => [
-    ...flowRunsKeys.mutations(),
-    "updateVariable",
-    { flowRunId, variableName }
-  ]
-};
-
-// src/domains/flowruns/mutations.ts
-function useUpdateFlowRunVariable() {
-  const qc = useQueryClient();
-  const service = useChatService();
-  return useMutation({
-    mutationKey: flowRunsKeys.updateVariable("", ""),
-    mutationFn: async ({
-      flowRunId,
-      variableName,
-      value
-    }) => {
-      if (isDraftThreadId(flowRunId)) return;
-      await service.updateFlowRunVariable({
-        flowRunId,
-        name: variableName,
-        value
-      });
-    },
-    onSuccess: (_data, variables) => {
-      if (variables?.flowRunId) {
-        qc.invalidateQueries({
-          queryKey: flowRunsKeys.variables(variables.flowRunId)
-        });
-      }
-    },
-    onError: (error) => {
-      console.error("Failed to update variable:", error);
-      toast.error("Failed to update variable");
-      throw error;
-    }
-  });
-}
 function useFlowRunVariables(flowRunId) {
   const service = useChatService();
   return useQuery({
@@ -4119,6 +4134,25 @@ function MessageComposer({
     workspaceId,
     threadId: isDraftThread ? void 0 : threadId
   });
+  const chatService = useChatService();
+  const {
+    mutate: cancelMutate,
+    isPending: cancelPending,
+    isSuccess: cancelAccepted,
+    isError: cancelErrored,
+    reset: cancelReset
+  } = useCancelFlowRun();
+  const canStop = isSending && !!chatService.cancelFlowRun;
+  const stopping = cancelPending || cancelAccepted;
+  const handleStopRun = () => {
+    if (stopping) return;
+    cancelMutate(threadId);
+  };
+  useEffect(() => {
+    if (!isSending && (cancelAccepted || cancelErrored)) {
+      cancelReset();
+    }
+  }, [isSending, cancelAccepted, cancelErrored, cancelReset]);
   useEffect(() => {
     if (typeof window === "undefined") return;
     window.__ssDownloadFile = (id) => getFileBlobUrl(id);
@@ -4391,7 +4425,21 @@ function MessageComposer({
                 )
               }
             ),
-            /* @__PURE__ */ jsx(
+            canStop ? /* @__PURE__ */ jsx(
+              IconButton,
+              {
+                onClick: handleStopRun,
+                className: "h-10 w-10 rounded-full self-end",
+                disabled: stopping,
+                "aria-label": stopping ? "Stopping" : "Stop",
+                children: /* @__PURE__ */ jsx(
+                  Square,
+                  {
+                    className: `h-4 w-4 fill-current ${stopping ? "animate-pulse" : ""}`
+                  }
+                )
+              }
+            ) : /* @__PURE__ */ jsx(
               IconButton,
               {
                 onClick: handleSendMessageAndClear,
@@ -4485,7 +4533,21 @@ function MessageComposer({
                           )
                         }
                       ),
-                      /* @__PURE__ */ jsx(
+                      canStop ? /* @__PURE__ */ jsx(
+                        IconButton,
+                        {
+                          onClick: handleStopRun,
+                          className: "h-10 w-10 rounded-full",
+                          disabled: stopping,
+                          "aria-label": stopping ? "Stopping" : "Stop",
+                          children: /* @__PURE__ */ jsx(
+                            Square,
+                            {
+                              className: `h-4 w-4 fill-current ${stopping ? "animate-pulse" : ""}`
+                            }
+                          )
+                        }
+                      ) : /* @__PURE__ */ jsx(
                         IconButton,
                         {
                           onClick: handleSendMessageAndClear,
@@ -4514,7 +4576,24 @@ function MessageComposer({
                 children: /* @__PURE__ */ jsx(Paperclip, { className: "h-5 w-5", strokeWidth: 2 })
               }
             ) }),
-            /* @__PURE__ */ jsx(
+            canStop ? /* @__PURE__ */ jsx(
+              Button,
+              {
+                onClick: handleStopRun,
+                className: `text-xs h-7 px-4 py-1 ${stopping ? "opacity-50 cursor-not-allowed" : ""}`,
+                disabled: stopping,
+                "aria-label": stopping ? "Stopping" : "Stop",
+                children: /* @__PURE__ */ jsxs("span", { className: "flex items-center gap-1.5", children: [
+                  /* @__PURE__ */ jsx(
+                    Square,
+                    {
+                      className: `h-3 w-3 fill-current ${stopping ? "animate-pulse" : ""}`
+                    }
+                  ),
+                  stopping ? "Stopping\u2026" : "Stop"
+                ] })
+              }
+            ) : /* @__PURE__ */ jsx(
               Button,
               {
                 onClick: handleSendMessageAndClear,
@@ -19773,17 +19852,12 @@ function getSafeUrl(url) {
     return null;
   }
 }
-function getDisplayName(source) {
-  if (source.file?.name) return source.file.name;
-  if (source.url) {
-    try {
-      const parsed = new URL(source.url);
-      return parsed.hostname + (parsed.pathname !== "/" ? parsed.pathname : "");
-    } catch {
-      return source.url;
-    }
+function getHost(url) {
+  try {
+    return new URL(url).hostname.replace(/^www\./, "");
+  } catch {
+    return url;
   }
-  return `Source ${source.index}`;
 }
 function isFileSource(s2) {
   return s2.sourceType === "File" /* File */ && !!s2.file;
@@ -19791,15 +19865,82 @@ function isFileSource(s2) {
 function isLinkSource(s2) {
   return (s2.sourceType === "URL" /* URL */ || s2.sourceType === "WebExternal" /* WebExternal */) && !!s2.url;
 }
+function UnverifiedBadge({
+  attribution,
+  compact = false
+}) {
+  if (attribution !== "unsupported" /* Unsupported */) return null;
+  return /* @__PURE__ */ jsxs(
+    "span",
+    {
+      title: "This citation couldn't be verified against the source text \u2014 double-check it.",
+      className: cn(
+        "inline-flex items-center gap-0.5 rounded text-[10px] font-medium text-amber-600 dark:text-amber-400 flex-shrink-0",
+        !compact && "px-1 py-px bg-amber-500/10"
+      ),
+      children: [
+        /* @__PURE__ */ jsx(ShieldAlert, { className: "h-3 w-3" }),
+        !compact && "Unverified"
+      ]
+    }
+  );
+}
+function IndexChip({ index }) {
+  return /* @__PURE__ */ jsx("span", { className: "flex h-4 min-w-[16px] flex-shrink-0 items-center justify-center rounded-full bg-muted px-1 text-[10px] font-medium tabular-nums text-muted-foreground", children: index });
+}
+function SourceFavicon({ host }) {
+  const [failed, setFailed] = useState(false);
+  if (failed) {
+    return /* @__PURE__ */ jsx(Globe, { className: "h-4 w-4 flex-shrink-0 text-muted-foreground" });
+  }
+  return /* @__PURE__ */ jsx(
+    "img",
+    {
+      src: `https://www.google.com/s2/favicons?domain=${encodeURIComponent(
+        host
+      )}&sz=64`,
+      alt: "",
+      width: 16,
+      height: 16,
+      loading: "lazy",
+      onError: () => setFailed(true),
+      className: "h-4 w-4 flex-shrink-0 rounded-sm"
+    }
+  );
+}
+function CitedQuote({ text: text6 }) {
+  if (!text6) return null;
+  return /* @__PURE__ */ jsxs("p", { className: "m-0 text-xs leading-snug text-muted-foreground", children: [
+    "\u201C",
+    text6,
+    "\u201D"
+  ] });
+}
+function SectionLabel({ children: children2 }) {
+  return /* @__PURE__ */ jsx("p", { className: "m-0 px-0.5 pb-1 text-[10px] font-medium uppercase tracking-wider text-muted-foreground", children: children2 });
+}
+var pillClasses = "inline-flex max-w-full items-center gap-1.5 rounded-md border border-border/60 bg-background px-2 py-1 text-xs font-medium text-foreground transition-colors hover:border-border hover:bg-muted/40";
+var expandedCardClasses = "w-full rounded-lg border border-border/60 bg-background p-2.5 flex flex-col gap-1.5";
 function ChatMessageSources({
   sources
 }) {
   const { workspaceId, threadId } = useChatContext();
   const { downloadFileMutation } = useFileMutations({ workspaceId, threadId });
   const [isExpanded, setIsExpanded] = useState(true);
+  const [openIndexes, setOpenIndexes] = useState(
+    /* @__PURE__ */ new Set()
+  );
+  const toggleOpen = (index) => setOpenIndexes((prev) => {
+    const next2 = new Set(prev);
+    if (next2.has(index)) next2.delete(index);
+    else next2.add(index);
+    return next2;
+  });
   const displaySources = (sources ?? []).filter(
     (s2) => isFileSource(s2) || isLinkSource(s2)
   );
+  const urlSources = displaySources.filter((s2) => !isFileSource(s2));
+  const fileSources = displaySources.filter(isFileSource);
   if (displaySources.length === 0) return null;
   return /* @__PURE__ */ jsxs("div", { className: "mt-4 rounded-lg border border-border bg-muted/30 overflow-hidden", children: [
     /* @__PURE__ */ jsxs(
@@ -19825,72 +19966,148 @@ function ChatMessageSources({
         ]
       }
     ),
-    isExpanded && /* @__PURE__ */ jsx("div", { className: "bg-background px-3 py-0.5", children: /* @__PURE__ */ jsx(
-      "ul",
-      {
-        className: "list-none space-y-1 m-0 p-0",
-        style: { paddingLeft: 0, marginLeft: 0 },
-        children: displaySources.map((source) => {
-          const displayName = getDisplayName(source);
-          if (isFileSource(source)) {
-            const Icon = getFileIcon3(source.file.name);
+    isExpanded && /* @__PURE__ */ jsxs("div", { className: "flex flex-col gap-2.5 p-2.5", children: [
+      urlSources.length > 0 && /* @__PURE__ */ jsxs("div", { children: [
+        /* @__PURE__ */ jsx(SectionLabel, { children: "URLs" }),
+        /* @__PURE__ */ jsx("div", { className: "flex flex-wrap gap-1.5", children: urlSources.map((source) => {
+          const host = getHost(source.url ?? "");
+          const safeHref = source.url ? getSafeUrl(source.url) : null;
+          if (!openIndexes.has(source.index)) {
             return /* @__PURE__ */ jsxs(
-              "li",
+              "button",
               {
-                className: "list-none text-sm text-foreground m-0 p-0 flex items-center gap-1.5",
-                style: { paddingLeft: 0, marginLeft: 0 },
+                type: "button",
+                "aria-expanded": false,
+                title: source.url ?? host,
+                onClick: () => toggleOpen(source.index),
+                className: pillClasses,
                 children: [
-                  /* @__PURE__ */ jsxs("span", { children: [
-                    `(${source.index})`,
-                    " : "
-                  ] }),
-                  Icon && /* @__PURE__ */ jsx(Icon, { className: "h-3.5 w-3.5 text-muted-foreground flex-shrink-0" }),
+                  /* @__PURE__ */ jsx(SourceFavicon, { host }),
+                  /* @__PURE__ */ jsx("span", { className: "min-w-0 truncate", children: host }),
+                  /* @__PURE__ */ jsx(
+                    UnverifiedBadge,
+                    {
+                      attribution: source.attribution,
+                      compact: true
+                    }
+                  ),
+                  /* @__PURE__ */ jsx(IndexChip, { index: source.index })
+                ]
+              },
+              `url-${source.index}`
+            );
+          }
+          return /* @__PURE__ */ jsxs(
+            "div",
+            {
+              className: expandedCardClasses,
+              children: [
+                /* @__PURE__ */ jsxs("div", { className: "flex w-full items-center gap-2", children: [
+                  /* @__PURE__ */ jsx(SourceFavicon, { host }),
+                  safeHref ? /* @__PURE__ */ jsx(
+                    "a",
+                    {
+                      href: safeHref,
+                      target: "_blank",
+                      rel: "noreferrer",
+                      className: "min-w-0 truncate text-sm font-medium text-foreground no-underline hover:underline underline-offset-2",
+                      children: host
+                    }
+                  ) : /* @__PURE__ */ jsx("span", { className: "min-w-0 truncate text-sm font-medium text-foreground", children: host }),
+                  /* @__PURE__ */ jsx(UnverifiedBadge, { attribution: source.attribution }),
+                  /* @__PURE__ */ jsx(IndexChip, { index: source.index }),
                   /* @__PURE__ */ jsx(
                     "button",
                     {
                       type: "button",
-                      title: displayName,
-                      onClick: () => downloadFileMutation.mutate(source.file),
-                      className: "text-foreground hover:bg-muted/50 rounded px-1 py-0.5 cursor-pointer disabled:opacity-50 disabled:cursor-not-allowed",
-                      disabled: downloadFileMutation.isPending,
-                      children: displayName
+                      "aria-expanded": true,
+                      "aria-label": "Collapse source",
+                      onClick: () => toggleOpen(source.index),
+                      className: "ml-auto rounded p-0.5 text-muted-foreground transition-colors hover:bg-muted/60 hover:text-foreground",
+                      children: /* @__PURE__ */ jsx(ChevronUp, { className: "h-3.5 w-3.5" })
                     }
                   )
+                ] }),
+                source.url && /* @__PURE__ */ jsx("p", { className: "m-0 truncate text-xs text-muted-foreground/80", children: source.url }),
+                /* @__PURE__ */ jsx(CitedQuote, { text: source.citedText })
+              ]
+            },
+            `url-${source.index}`
+          );
+        }) })
+      ] }),
+      fileSources.length > 0 && /* @__PURE__ */ jsxs("div", { children: [
+        /* @__PURE__ */ jsx(SectionLabel, { children: "Files" }),
+        /* @__PURE__ */ jsx("div", { className: "flex flex-wrap gap-1.5", children: fileSources.map((source) => {
+          const Icon = getFileIcon3(source.file.name);
+          if (!openIndexes.has(source.index)) {
+            return /* @__PURE__ */ jsxs(
+              "button",
+              {
+                type: "button",
+                "aria-expanded": false,
+                title: source.file.name,
+                onClick: () => toggleOpen(source.index),
+                className: pillClasses,
+                children: [
+                  /* @__PURE__ */ jsx(Icon, { className: "h-4 w-4 flex-shrink-0 text-muted-foreground" }),
+                  /* @__PURE__ */ jsx("span", { className: "min-w-0 truncate", children: source.file.name }),
+                  /* @__PURE__ */ jsx(
+                    UnverifiedBadge,
+                    {
+                      attribution: source.attribution,
+                      compact: true
+                    }
+                  ),
+                  /* @__PURE__ */ jsx(IndexChip, { index: source.index })
                 ]
               },
               `${source.file.id}-${source.index}`
             );
           }
-          const safeHref = source.url ? getSafeUrl(source.url) : null;
           return /* @__PURE__ */ jsxs(
-            "li",
+            "div",
             {
-              className: "list-none text-sm text-foreground m-0 p-0 flex items-center gap-1.5",
-              style: { paddingLeft: 0, marginLeft: 0 },
+              className: expandedCardClasses,
               children: [
-                /* @__PURE__ */ jsxs("span", { children: [
-                  `(${source.index})`,
-                  " : "
+                /* @__PURE__ */ jsxs("div", { className: "flex w-full items-center gap-2", children: [
+                  /* @__PURE__ */ jsx(Icon, { className: "h-4 w-4 flex-shrink-0 text-muted-foreground" }),
+                  /* @__PURE__ */ jsx("span", { className: "min-w-0 truncate text-sm font-medium text-foreground", children: source.file.name }),
+                  /* @__PURE__ */ jsx(UnverifiedBadge, { attribution: source.attribution }),
+                  /* @__PURE__ */ jsx(IndexChip, { index: source.index }),
+                  /* @__PURE__ */ jsx(
+                    "button",
+                    {
+                      type: "button",
+                      "aria-expanded": true,
+                      "aria-label": "Collapse source",
+                      onClick: () => toggleOpen(source.index),
+                      className: "ml-auto rounded p-0.5 text-muted-foreground transition-colors hover:bg-muted/60 hover:text-foreground",
+                      children: /* @__PURE__ */ jsx(ChevronUp, { className: "h-3.5 w-3.5" })
+                    }
+                  )
                 ] }),
-                /* @__PURE__ */ jsx(ExternalLink, { className: "h-3.5 w-3.5 text-muted-foreground flex-shrink-0" }),
-                safeHref ? /* @__PURE__ */ jsx(
-                  "a",
+                /* @__PURE__ */ jsx(CitedQuote, { text: source.citedText }),
+                /* @__PURE__ */ jsxs(
+                  "button",
                   {
-                    title: source.url ?? displayName,
-                    href: safeHref,
-                    target: "_blank",
-                    rel: "noreferrer",
-                    className: "text-foreground hover:bg-muted/50 rounded px-1 py-0.5 underline underline-offset-2",
-                    children: displayName
+                    type: "button",
+                    onClick: () => downloadFileMutation.mutate(source.file),
+                    disabled: downloadFileMutation.isPending,
+                    className: "inline-flex w-fit items-center gap-1 rounded-md border border-border/60 px-2 py-1 text-xs font-medium text-foreground transition-colors hover:bg-muted/40 disabled:cursor-not-allowed disabled:opacity-50",
+                    children: [
+                      /* @__PURE__ */ jsx(Download, { className: "h-3 w-3" }),
+                      "Download"
+                    ]
                   }
-                ) : /* @__PURE__ */ jsx("span", { className: "text-foreground rounded px-1 py-0.5", children: displayName })
+                )
               ]
             },
-            `url-${source.index}`
+            `${source.file.id}-${source.index}`
           );
-        })
-      }
-    ) })
+        }) })
+      ] })
+    ] })
   ] });
 }
 var MessageBubble = (props) => {
@@ -20037,7 +20254,14 @@ function pushContent(items, value) {
     return;
   }
   if (Array.isArray(value)) {
-    items.push(...value);
+    const looksLikeContent = value.every(
+      (it) => it != null && typeof it === "object" && ("text" in it || "image" in it)
+    );
+    if (looksLikeContent) {
+      items.push(...value);
+    } else {
+      items.push({ text: "```json\n" + JSON.stringify(value, null, 2) + "\n```" });
+    }
     return;
   }
   if (typeof value === "object") {
@@ -20112,25 +20336,28 @@ var MessageItem = ({
   let groupOpen = false;
   let keyCounter = 0;
   let lastStatusNode = null;
+  const groupHasAnything = () => groupContent.length > 0 || groupFiles.length > 0 || groupSources.length > 0;
   const flush = (nextType) => {
-    bubbles.push(
-      /* @__PURE__ */ jsx(
-        MessageBubble,
-        {
-          createdBy: lastCreatedBy,
-          createdByUserId: lastCreatedByUserId,
-          createdAt: lastCreatedAt,
-          type: groupType,
-          content: groupContent,
-          files: groupFiles,
-          sources: groupSources,
-          chatbotName,
-          userOutput: null,
-          userInput: null
-        },
-        `bubble-${message.id ?? "msg"}-${keyCounter++}`
-      )
-    );
+    if (groupHasAnything()) {
+      bubbles.push(
+        /* @__PURE__ */ jsx(
+          MessageBubble,
+          {
+            createdBy: lastCreatedBy,
+            createdByUserId: lastCreatedByUserId,
+            createdAt: lastCreatedAt,
+            type: groupType,
+            content: groupContent,
+            files: groupFiles,
+            sources: groupSources,
+            chatbotName,
+            userOutput: null,
+            userInput: null
+          },
+          `bubble-${message.id ?? "msg"}-${keyCounter++}`
+        )
+      );
+    }
     groupContent = [];
     groupFiles = [];
     groupSources = [];
@@ -20163,6 +20390,9 @@ var MessageItem = ({
       case "prompt":
       case "response":
       case "content": {
+        if (v.value === "") {
+          continue;
+        }
         lastStatusNode = null;
         if (groupContent.length > 0) flush(v.type);
         if (v.value == null) {
@@ -20217,7 +20447,7 @@ var MessageItem = ({
       }
       case "sources": {
         groupSources = coerceSources(v.value);
-        groupOpen = true;
+        if (groupSources.length > 0) groupOpen = true;
         break;
       }
       default: {
@@ -20231,7 +20461,7 @@ var MessageItem = ({
     lastCreatedBy = v.createdBy;
     lastCreatedByUserId = v.createdByUserId;
   }
-  if (groupOpen) {
+  if (groupOpen && groupHasAnything()) {
     bubbles.push(
       /* @__PURE__ */ jsx(
         MessageBubble,