@athenaintel/react 0.19.0 → 0.21.0

package/dist/chat/AthenaChat.js

@@ -0,0 +1,753 @@
+import { jsxs, jsx, Fragment } from "react/jsx-runtime";
+import { TiptapText } from "./tiptap-text.js";
+import { AthenaChatErrorBoundary } from "./AthenaChatErrorBoundary.js";
+import { StatewireApprovalCard } from "./StatewireApprovalCard.js";
+import { StatewireQueuedMessages } from "./StatewireQueuedMessages.js";
+import { StatewireConnectionBanner, StatewireErrorBanner } from "./statewire-banners.js";
+import { ToolFallback } from "../tools/tool-fallback.js";
+import { AppendDocumentToolUI } from "../tools/append-document-tool-ui.js";
+import { ReadAssetToolUI } from "../tools/read-asset-tool-ui.js";
+import { TOOL_UI_REGISTRY } from "../tools/tool-uis.js";
+import { SuperGroupingCard, SuperGroupingFinalText } from "./super-grouping/SuperGroupingCard.js";
+import { NestedPtcComponentsProvider, useNestedPtcComponents, NestedPtcMessages } from "../tools/NestedPtcMessages.js";
+import { TooltipIconButton } from "./tooltip-icon-button.js";
+import { Button } from "../ui/button.js";
+import { Collapsible, CollapsibleTrigger, CollapsibleContent } from "../ui/collapsible.js";
+import { ChatComposerDock } from "../chat-ui/thread/chat-composer-dock.js";
+import { ChatThreadShell } from "../chat-ui/thread/chat-thread-shell.js";
+import { StatewireHistoryEdge } from "../chat-ui/thread/statewire-history-edge.js";
+import { ComposerSendOrStop } from "../chat-ui/thread/composer-send-or-stop.js";
+import { TiptapComposer } from "../chat-ui/composer/tiptap-composer.js";
+import { MentionAuthCtx } from "../chat-ui/mentions/suggestions/data/api-client.js";
+import { ActionBarPrimitive, AuiIf, ActionBarMorePrimitive, MessagePrimitive, useAuiState, ErrorPrimitive, useAui, ThreadPrimitive } from "@assistant-ui/react";
+import { CheckIcon, CopyIcon, MoreHorizontalIcon, DownloadIcon, ExternalLinkIcon, SearchIcon, PresentationIcon, FileTextIcon, BrainIcon, ChevronRightIcon, AlertCircleIcon, RefreshCwIcon, ArrowDownIcon, ArrowUpIcon, SquareIcon, ListPlusIcon, ArrowRightIcon } from "lucide-react";
+import { useMemo, useContext, useState, useRef, useEffect, createContext, useCallback } from "react";
+import { useAthenaThreadId, useAthenaStatewireThreadId } from "../threads/thread-id-context.js";
+import { FileUploadButton } from "../composer/FileUploadButton.js";
+import { ComposerQuotePreview } from "../composer/ComposerQuotePreview.js";
+import { ComposerAttachmentPreview } from "../composer/ComposerAttachmentPreview.js";
+import { ComposerDropZone } from "../composer/ComposerDropZone.js";
+import { QuoteProvider, useQuoteFromPostMessage, useQuote } from "../composer/QuoteContext.js";
+import { AttachmentProvider, useAttachments } from "../composer/AttachmentContext.js";
+import { buildComposedMessage } from "../composer/build-composed-message.js";
+import { useAttachFiles } from "../composer/useAttachFiles.js";
+import { useAthenaConfig } from "../provider/AthenaContext.js";
+import { createAthenaSpacesUrl } from "../lib/athena-urls.js";
+import { isRecord } from "../lib/utils.js";
+import { useSendMessage } from "../composer/useSendMessage.js";
+import { getToolkitDisplayInfo } from "../toolkits/index.js";
+import { runActivityKey, runActivityLabel, RUN_ACTIVITY_STALL_SECONDS } from "./run-activity.js";
+import { cn } from "cnfast";
+function QuotePostMessageBridge() {
+  useQuoteFromPostMessage();
+  return null;
+}
+const DEFAULT_SUGGESTIONS = [
+  {
+    icon: SearchIcon,
+    title: "Research a topic and draft a report",
+    prompt: "Search the web for the latest developments in enterprise AI adoption, then create a document summarizing the key trends, market data, and strategic implications."
+  },
+  {
+    icon: SearchIcon,
+    title: "Compare recent AI model launches",
+    prompt: "Search the web for the latest major AI model launches from OpenAI, Anthropic, and Google, then summarize the release dates, standout capabilities, and practical tradeoffs in a concise comparison table."
+  },
+  {
+    icon: PresentationIcon,
+    title: "Build a presentation from scratch",
+    prompt: "Create a 5-slide investor update presentation covering market opportunity, product traction, key metrics, competitive positioning, and next steps."
+  },
+  {
+    icon: FileTextIcon,
+    title: "Set up a structured spreadsheet",
+    prompt: "Create a project tracking spreadsheet with columns for task name, owner, priority, status, start date, due date, and completion percentage."
+  }
+];
+const AthenaChatDefaultComponentsContext = createContext(null);
+const AthenaReasoningTextComponentContext = createContext(null);
+const useAthenaChatDefaultComponents = () => {
+  const value = useContext(AthenaChatDefaultComponentsContext);
+  if (!value) {
+    throw new Error("Athena default chat components must be used within AthenaChat.");
+  }
+  return value;
+};
+const AthenaDefaultAssistantMessage = () => {
+  const { toolUIs, TextComponent, ReasoningComponent, EmptyComponent, ActionBarComponent, groupToolCalls } = useAthenaChatDefaultComponents();
+  return /* @__PURE__ */ jsx(
+    AthenaAssistantMessage,
+    {
+      toolUIs,
+      TextComponent,
+      ReasoningComponent,
+      EmptyComponent,
+      ActionBarComponent,
+      groupToolCalls
+    }
+  );
+};
+const AthenaDefaultUserMessage = () => {
+  const { TextComponent } = useAthenaChatDefaultComponents();
+  return /* @__PURE__ */ jsx(AthenaUserMessage, { TextComponent });
+};
+const getReasoningTokensFromMetadata = (metadata) => {
+  if (!isRecord(metadata)) {
+    return void 0;
+  }
+  const customMetadata = metadata.custom;
+  if (!isRecord(customMetadata)) {
+    return void 0;
+  }
+  const athenaMetadata = customMetadata._athena;
+  if (!isRecord(athenaMetadata)) {
+    return void 0;
+  }
+  const reasoningTokens = athenaMetadata.reasoningTokens;
+  return typeof reasoningTokens === "number" && Number.isFinite(reasoningTokens) ? reasoningTokens : void 0;
+};
+const formatReasoningTokens = (reasoningTokens) => `${reasoningTokens.toLocaleString("en-US")} ${reasoningTokens === 1 ? "token" : "tokens"}`;
+const withNestedPtcMessages = (ToolUI) => {
+  const ToolUIWithNestedPtcMessages = (props) => {
+    const nestedComponents = useNestedPtcComponents(ToolFallback);
+    return /* @__PURE__ */ jsxs(Fragment, { children: [
+      /* @__PURE__ */ jsx(ToolUI, { ...props }),
+      /* @__PURE__ */ jsx(NestedPtcMessages, { components: nestedComponents })
+    ] });
+  };
+  return ToolUIWithNestedPtcMessages;
+};
+const SuggestionCard = ({
+  suggestion,
+  index
+}) => {
+  const Icon = suggestion.icon;
+  const sendMessage = useSendMessage();
+  const handleClick = useCallback(() => {
+    void sendMessage(suggestion.prompt);
+  }, [sendMessage, suggestion.prompt]);
+  return /* @__PURE__ */ jsxs(
+    "button",
+    {
+      type: "button",
+      onClick: handleClick,
+      className: `fade-in slide-in-from-bottom-2 animate-in fill-mode-both group flex items-start gap-3 rounded-xl border border-border/60 bg-background p-3.5 text-left shadow-sm transition-all duration-150 hover:border-border hover:bg-muted/40 hover:shadow-md`,
+      style: { animationDelay: `${100 + index * 75}ms` },
+      children: [
+        /* @__PURE__ */ jsx("div", { className: "flex size-8 shrink-0 items-center justify-center rounded-lg bg-muted/60 text-muted-foreground transition-colors group-hover:bg-primary/10 group-hover:text-primary", children: /* @__PURE__ */ jsx(Icon, { className: "size-4" }) }),
+        /* @__PURE__ */ jsx("div", { className: "min-w-0 flex-1", children: /* @__PURE__ */ jsx("span", { className: "text-[13px] font-medium leading-snug text-foreground", children: suggestion.title }) }),
+        /* @__PURE__ */ jsx(ArrowRightIcon, { className: "mt-0.5 size-3.5 shrink-0 text-muted-foreground/0 transition-all group-hover:text-muted-foreground" })
+      ]
+    }
+  );
+};
+const AthenaChatWelcome = ({ welcomeMessage, welcomeSubtext, welcomeSuggestions }) => /* @__PURE__ */ jsx("div", { className: "aui-thread-welcome-root mx-auto my-auto flex w-full max-w-(--thread-max-width) grow flex-col", children: /* @__PURE__ */ jsx("div", { className: "aui-thread-welcome-center flex w-full grow flex-col items-center justify-center", children: /* @__PURE__ */ jsxs("div", { className: "aui-thread-welcome-message flex size-full flex-col justify-center px-4", children: [
+  /* @__PURE__ */ jsx("h1", { className: "aui-thread-welcome-message-inner fade-in slide-in-from-bottom-1 animate-in fill-mode-both font-semibold text-2xl duration-200", children: welcomeMessage }),
+  welcomeSubtext && /* @__PURE__ */ jsx("p", { className: "aui-thread-welcome-message-inner fade-in slide-in-from-bottom-1 animate-in fill-mode-both text-muted-foreground text-lg delay-75 duration-200", children: welcomeSubtext }),
+  welcomeSuggestions.length > 0 && /* @__PURE__ */ jsx("div", { className: "mt-6 grid grid-cols-1 gap-2.5 sm:grid-cols-2", children: welcomeSuggestions.map((s, i) => /* @__PURE__ */ jsx(SuggestionCard, { suggestion: s, index: i }, s.title)) })
+] }) }) });
+const AthenaChat = ({
+  className,
+  welcomeMessage = "Hello there!",
+  welcomeSubtext = "What can I help you with?",
+  maxWidth = "44rem",
+  turnAnchor = "bottom",
+  toolUIs,
+  mentionTools,
+  welcomeSuggestions = DEFAULT_SUGGESTIONS,
+  components,
+  groupToolCalls = false
+}) => {
+  var _a, _b, _c;
+  const athenaConfig = useAthenaConfig();
+  const mentionAuth = useMemo(
+    () => ({
+      backendUrl: athenaConfig.backendUrl,
+      appUrl: athenaConfig.appUrl,
+      apiKey: athenaConfig.apiKey,
+      credentials: "include",
+      get accessToken() {
+        return athenaConfig.token ?? void 0;
+      }
+    }),
+    [athenaConfig]
+  );
+  const providerMentionTools = useMemo(() => {
+    const uniqueToolkitIds = Array.from(new Set(athenaConfig.enabledToolkits));
+    return uniqueToolkitIds.map((id) => {
+      const toolkit = getToolkitDisplayInfo(id);
+      return {
+        id: toolkit.id,
+        name: toolkit.name,
+        description: toolkit.description,
+        icon: toolkit.icon,
+        type: "toolkit"
+      };
+    });
+  }, [athenaConfig.enabledToolkits]);
+  const tools = mentionTools ?? providerMentionTools;
+  const mergedToolUIs = useMemo(
+    () => ({
+      append_markdown_to_athena_document: AppendDocumentToolUI,
+      read_full_asset: ReadAssetToolUI,
+      read_asset: ReadAssetToolUI,
+      ...TOOL_UI_REGISTRY,
+      ...toolUIs
+    }),
+    [toolUIs]
+  );
+  const textComponent = ((_a = components == null ? void 0 : components.messageParts) == null ? void 0 : _a.Text) ?? TiptapText;
+  const reasoningComponent = (_b = components == null ? void 0 : components.messageParts) == null ? void 0 : _b.Reasoning;
+  const emptyComponent = ((_c = components == null ? void 0 : components.messageParts) == null ? void 0 : _c.Empty) ?? AthenaAssistantMessageEmpty;
+  const actionBarComponent = (components == null ? void 0 : components.AssistantActionBar) ?? AthenaAssistantActionBar;
+  const defaultComponentsContextValue = useMemo(
+    () => ({
+      toolUIs: mergedToolUIs,
+      TextComponent: textComponent,
+      ReasoningComponent: reasoningComponent,
+      EmptyComponent: emptyComponent,
+      ActionBarComponent: actionBarComponent,
+      groupToolCalls
+    }),
+    [actionBarComponent, emptyComponent, groupToolCalls, mergedToolUIs, reasoningComponent, textComponent]
+  );
+  const AssistantMessageComponent = (components == null ? void 0 : components.AssistantMessage) ?? AthenaDefaultAssistantMessage;
+  const UserMessageComponent = (components == null ? void 0 : components.UserMessage) ?? AthenaDefaultUserMessage;
+  const messageComponents = useMemo(
+    () => ({
+      UserMessage: UserMessageComponent,
+      AssistantMessage: AssistantMessageComponent
+    }),
+    [AssistantMessageComponent, UserMessageComponent]
+  );
+  return /* @__PURE__ */ jsx(AthenaChatDefaultComponentsContext.Provider, { value: defaultComponentsContextValue, children: /* @__PURE__ */ jsx(AthenaChatErrorBoundary, { children: /* @__PURE__ */ jsx(
+    ChatThreadShell,
+    {
+      overlay: /* @__PURE__ */ jsx(ThreadLoadingOverlay, {}),
+      panelClassName: cn(
+        "aui-root aui-thread-root @container relative flex h-full flex-col bg-background",
+        className
+      ),
+      maxWidth,
+      turnAnchor,
+      viewportClassName: "aui-thread-viewport relative flex flex-1 flex-col overflow-x-auto overflow-y-scroll scroll-smooth px-4 pt-4",
+      contentClassName: "flex w-full flex-1 flex-col",
+      viewportOverlay: /* @__PURE__ */ jsx(ThreadScrollToTop, {}),
+      welcome: /* @__PURE__ */ jsx(
+        AthenaChatWelcome,
+        {
+          welcomeMessage,
+          welcomeSubtext,
+          welcomeSuggestions
+        }
+      ),
+      beforeMessages: /* @__PURE__ */ jsx(StatewireHistoryEdge, { className: "aui-thread-history-edge" }),
+      components: messageComponents,
+      runningIndicator: null,
+      scrollToBottom: /* @__PURE__ */ jsx(ThreadScrollToBottom, {}),
+      viewportFooterClassName: "aui-thread-viewport-footer sticky bottom-0 mx-auto mt-auto flex w-full max-w-(--thread-max-width) flex-col gap-4 overflow-visible rounded-t-3xl bg-background pb-4 md:pb-6",
+      viewportFooter: /* @__PURE__ */ jsxs(Fragment, { children: [
+        /* @__PURE__ */ jsx(StatewireConnectionBanner, {}),
+        /* @__PURE__ */ jsx(StatewireErrorBanner, {}),
+        /* @__PURE__ */ jsx(StatewireApprovalCard, {}),
+        /* @__PURE__ */ jsx(StatewireQueuedMessages, {}),
+        /* @__PURE__ */ jsx(AttachmentProvider, { children: /* @__PURE__ */ jsxs(QuoteProvider, { children: [
+          /* @__PURE__ */ jsx(QuotePostMessageBridge, {}),
+          /* @__PURE__ */ jsx(MentionAuthCtx.Provider, { value: mentionAuth, children: /* @__PURE__ */ jsx(AthenaChatComposer, { tools }) })
+        ] }) })
+      ] })
+    }
+  ) }) });
+};
+const ThreadLoadingOverlay = () => {
+  const remoteId = useAthenaThreadId();
+  const [timedOut, setTimedOut] = useState(false);
+  useEffect(() => {
+    if (!remoteId) return;
+    setTimedOut(false);
+    const timer = setTimeout(() => setTimedOut(true), 15e3);
+    return () => clearTimeout(timer);
+  }, [remoteId]);
+  if (!remoteId || timedOut) return null;
+  return /* @__PURE__ */ jsx(AuiIf, { condition: (s) => s.thread.isEmpty, children: /* @__PURE__ */ jsx("div", { className: "absolute inset-0 z-10 flex items-center justify-center bg-background/80", children: /* @__PURE__ */ jsx("div", { className: "size-5 animate-spin rounded-full border-2 border-muted-foreground border-t-transparent" }) }) });
+};
+const THREAD_VIEWPORT_ATTR = "data-aui-thread-viewport";
+const useScrollToTop = () => {
+  const [isAtTop, setIsAtTop] = useState(true);
+  const [buttonEl, setButtonEl] = useState(null);
+  const containerRef = useCallback((node) => {
+    setButtonEl(node);
+  }, []);
+  const getViewport = useCallback(() => {
+    if (!buttonEl) return null;
+    return buttonEl.closest(`[${THREAD_VIEWPORT_ATTR}]`);
+  }, [buttonEl]);
+  useEffect(() => {
+    const viewport = getViewport();
+    if (!viewport) return;
+    const handleScroll = () => {
+      setIsAtTop(viewport.scrollTop <= 50);
+    };
+    handleScroll();
+    viewport.addEventListener("scroll", handleScroll, { passive: true });
+    return () => viewport.removeEventListener("scroll", handleScroll);
+  }, [getViewport]);
+  const handleClick = useCallback(() => {
+    const viewport = getViewport();
+    if (!viewport) return;
+    viewport.scrollTo({ top: 0, behavior: "smooth" });
+  }, [getViewport]);
+  return { isAtTop, containerRef, handleClick };
+};
+const ThreadScrollToTop = () => {
+  const { isAtTop, containerRef, handleClick } = useScrollToTop();
+  const isEmpty = useAuiState((s) => s.thread.isEmpty);
+  if (isEmpty) return null;
+  return /* @__PURE__ */ jsx("div", { className: "sticky top-0 z-10 mx-auto flex h-0 w-full max-w-(--thread-max-width) justify-center", children: /* @__PURE__ */ jsx(
+    TooltipIconButton,
+    {
+      ref: containerRef,
+      tooltip: "Scroll to top",
+      variant: "outline",
+      className: cn("aui-thread-scroll-to-top absolute top-2 rounded-full p-4 dark:bg-background dark:hover:bg-accent", isAtTop && "invisible"),
+      onClick: handleClick,
+      children: /* @__PURE__ */ jsx(ArrowUpIcon, {})
+    }
+  ) });
+};
+const ThreadScrollToBottom = () => /* @__PURE__ */ jsx(ThreadPrimitive.ScrollToBottom, { asChild: true, children: /* @__PURE__ */ jsx(
+  TooltipIconButton,
+  {
+    tooltip: "Scroll to bottom",
+    variant: "outline",
+    className: "aui-thread-scroll-to-bottom absolute -top-12 z-10 self-center rounded-full p-4 disabled:invisible dark:bg-background dark:hover:bg-accent",
+    children: /* @__PURE__ */ jsx(ArrowDownIcon, {})
+  }
+) });
+const AthenaComposerSendButton = ({
+  action,
+  onSend,
+  disabled,
+  title
+}) => {
+  const label = action === "queue" ? "Queue message" : "Send message";
+  return /* @__PURE__ */ jsx(
+    TooltipIconButton,
+    {
+      tooltip: title ?? label,
+      side: "bottom",
+      type: "button",
+      variant: "default",
+      size: "icon",
+      className: "aui-composer-send size-8 rounded-full",
+      "aria-label": label,
+      onClick: onSend,
+      disabled,
+      children: action === "queue" ? /* @__PURE__ */ jsx(ListPlusIcon, { className: "aui-composer-send-icon size-4" }) : /* @__PURE__ */ jsx(ArrowUpIcon, { className: "aui-composer-send-icon size-4" })
+    }
+  );
+};
+const AthenaComposerStopButton = ({ onStop, disabled, title }) => /* @__PURE__ */ jsx(
+  Button,
+  {
+    type: "button",
+    variant: "default",
+    size: "icon",
+    className: "aui-composer-cancel size-8 rounded-full",
+    "aria-label": "Stop generating",
+    title,
+    onClick: onStop,
+    disabled,
+    children: /* @__PURE__ */ jsx(SquareIcon, { className: "aui-composer-cancel-icon size-3 fill-current" })
+  }
+);
+const ATHENA_COMPOSER_COMPONENTS = {
+  SendButton: AthenaComposerSendButton,
+  StopButton: AthenaComposerStopButton
+};
+const ATHENA_CHAT_MENTION_CATEGORIES = [
+  "favorites",
+  "assets",
+  "projects",
+  "folders"
+];
+const AthenaChatComposer = ({ tools }) => {
+  const aui = useAui();
+  const threadId = useAthenaStatewireThreadId();
+  const { appUrl } = useAthenaConfig();
+  const { attachments, clearAttachments, isUploading } = useAttachments();
+  const { quote, clearQuote } = useQuote();
+  const { attachFiles } = useAttachFiles();
+  const composerRef = useRef(null);
+  const [hasText, setHasText] = useState(false);
+  const rootCategories = useMemo(() => {
+    const categories = [...ATHENA_CHAT_MENTION_CATEGORIES];
+    if (tools.some((tool) => tool.type === "toolkit")) categories.push("toolkits");
+    if (tools.some((tool) => tool.type === "tool")) categories.push("tools");
+    return categories;
+  }, [tools]);
+  const attachmentsRef = useRef(attachments);
+  attachmentsRef.current = attachments;
+  const quoteRef = useRef(quote);
+  quoteRef.current = quote;
+  const hasExtras = attachments.length > 0 || quote !== null;
+  const isUploadingRef = useRef(isUploading);
+  isUploadingRef.current = isUploading;
+  const uploadWaitersRef = useRef([]);
+  useEffect(() => {
+    if (isUploading) return;
+    const waiters = uploadWaitersRef.current;
+    uploadWaitersRef.current = [];
+    for (const resolve of waiters) resolve();
+  }, [isUploading]);
+  const waitForHostUploads = useCallback(() => {
+    if (!isUploadingRef.current) return;
+    return new Promise((resolve) => {
+      uploadWaitersRef.current.push(resolve);
+    });
+  }, []);
+  const sendComposed = useCallback(
+    (userText) => {
+      const message = buildComposedMessage({
+        attachments: attachmentsRef.current,
+        quote: quoteRef.current,
+        userText,
+        appUrl
+      });
+      if (!message) return false;
+      aui.composer.setText(message);
+      aui.composer.send({ steer: false });
+      clearAttachments();
+      clearQuote();
+      return true;
+    },
+    [appUrl, aui, clearAttachments, clearQuote]
+  );
+  const interceptSubmit = useCallback(
+    (text) => attachmentsRef.current.length > 0 || quoteRef.current !== null ? sendComposed(text) : false,
+    [sendComposed]
+  );
+  const fileIntake = useMemo(
+    () => ({
+      onPaste: (files) => {
+        void attachFiles(files, "paste");
+      }
+    }),
+    [attachFiles]
+  );
+  return /* @__PURE__ */ jsx(ComposerDropZone, { children: /* @__PURE__ */ jsx(
+    ChatComposerDock,
+    {
+      className: "",
+      surfaceClassName: "aui-composer-root relative flex w-full flex-col rounded-2xl border border-input bg-background px-1 pt-2 outline-none transition-shadow focus-within:border-ring focus-within:ring-2 focus-within:ring-ring/20",
+      editorRowClassName: "px-2.5",
+      toolbarClassName: "aui-composer-action-wrapper relative mx-2 mb-2 flex items-center",
+      beforeEditor: /* @__PURE__ */ jsxs(Fragment, { children: [
+        /* @__PURE__ */ jsx(ComposerQuotePreview, {}),
+        /* @__PURE__ */ jsx(ComposerAttachmentPreview, {})
+      ] }),
+      editor: /* @__PURE__ */ jsx(
+        TiptapComposer,
+        {
+          ref: composerRef,
+          threadId,
+          placeholder: "Send a message...",
+          tools,
+          rootCategories,
+          onDraftStateChange: setHasText,
+          waitForHostUploads,
+          interceptSubmit,
+          fileIntake
+        }
+      ),
+      toolbarStart: /* @__PURE__ */ jsx("div", { className: "flex items-center gap-1", children: /* @__PURE__ */ jsx(FileUploadButton, {}) }),
+      toolbarEnd: /* @__PURE__ */ jsx("div", { className: "flex items-center gap-1", children: /* @__PURE__ */ jsx(
+        ComposerSendOrStop,
+        {
+          composerRef,
+          hasDraft: hasText || hasExtras,
+          sendBlockedReason: isUploading ? "Upload in progress..." : null,
+          components: ATHENA_COMPOSER_COMPONENTS
+        }
+      ) })
+    }
+  ) });
+};
+const MessageError = () => /* @__PURE__ */ jsx(MessagePrimitive.Error, { children: /* @__PURE__ */ jsxs(ErrorPrimitive.Root, { className: "aui-message-error-root mt-2 flex items-start gap-2 rounded-md border border-destructive bg-destructive/10 p-3 text-destructive text-sm dark:bg-destructive/5 dark:text-red-200", children: [
+  /* @__PURE__ */ jsx(AlertCircleIcon, { className: "mt-0.5 size-4 shrink-0" }),
+  /* @__PURE__ */ jsx(ErrorPrimitive.Message, { className: "aui-message-error-message grow" }),
+  /* @__PURE__ */ jsx(ActionBarPrimitive.Reload, { asChild: true, children: /* @__PURE__ */ jsx(
+    Button,
+    {
+      variant: "ghost",
+      size: "icon",
+      className: "size-6 shrink-0 text-destructive hover:text-destructive dark:text-red-200 dark:hover:text-red-100",
+      children: /* @__PURE__ */ jsx(RefreshCwIcon, { className: "size-3.5" })
+    }
+  ) })
+] }) });
+const AthenaAssistantMessageEmpty = ({ status }) => {
+  if ((status == null ? void 0 : status.type) !== "running") return null;
+  return /* @__PURE__ */ jsx("span", { className: "shimmer text-[13px] text-muted-foreground", children: "Thinking..." });
+};
+const AthenaMessageEmptyIndicator = ({
+  EmptyComponent
+}) => {
+  const hasParts = useAuiState((s) => s.message.parts.length > 0);
+  const isRunning = useAuiState((s) => {
+    var _a;
+    return ((_a = s.message.status) == null ? void 0 : _a.type) === "running";
+  });
+  if (hasParts || !isRunning) return null;
+  return /* @__PURE__ */ jsx(EmptyComponent, { status: { type: "running" } });
+};
+const AthenaMessageActivityIndicator = () => {
+  const isRunning = useAuiState((s) => {
+    var _a;
+    return ((_a = s.message.status) == null ? void 0 : _a.type) === "running";
+  });
+  const activityKey = useAuiState((s) => runActivityKey(s.message.parts));
+  const [seconds, setSeconds] = useState(0);
+  useEffect(() => {
+    setSeconds(0);
+    if (!isRunning || activityKey === null) return;
+    const startedAt = Date.now();
+    const timer = setInterval(() => {
+      setSeconds(Math.floor((Date.now() - startedAt) / 1e3));
+    }, 1e3);
+    return () => clearInterval(timer);
+  }, [isRunning, activityKey]);
+  if (!isRunning || activityKey === null) return null;
+  const stalled = seconds >= RUN_ACTIVITY_STALL_SECONDS;
+  return /* @__PURE__ */ jsxs("div", { className: "aui-run-activity mt-2 flex items-center gap-2 px-1 text-[12px] text-muted-foreground", children: [
+    /* @__PURE__ */ jsxs("span", { className: "flex items-center gap-1", "aria-hidden": true, children: [
+      /* @__PURE__ */ jsx("span", { className: "size-1.5 animate-bounce rounded-full bg-current [animation-delay:0ms]" }),
+      /* @__PURE__ */ jsx("span", { className: "size-1.5 animate-bounce rounded-full bg-current [animation-delay:150ms]" }),
+      /* @__PURE__ */ jsx("span", { className: "size-1.5 animate-bounce rounded-full bg-current [animation-delay:300ms]" })
+    ] }),
+    /* @__PURE__ */ jsx("span", { "aria-hidden": true, className: stalled ? "text-amber-600 dark:text-amber-400" : void 0, children: runActivityLabel(seconds) }),
+    /* @__PURE__ */ jsx("span", { "aria-live": "polite", className: "sr-only", children: stalled ? "Assistant is still working; nothing new for over thirty seconds." : "Assistant is working." })
+  ] });
+};
+const AthenaReasoningPart = ({
+  text,
+  status,
+  TextComponent
+}) => {
+  const inheritedTextComponent = useContext(AthenaReasoningTextComponentContext);
+  const EffectiveTextComponent = TextComponent ?? inheritedTextComponent ?? TiptapText;
+  const isMessageRunning = useAuiState((state) => {
+    var _a;
+    return ((_a = state.message.status) == null ? void 0 : _a.type) === "running";
+  });
+  const reasoningTokens = useAuiState(
+    (state) => getReasoningTokensFromMetadata(state.message.metadata)
+  );
+  const isRunning = isMessageRunning || (status == null ? void 0 : status.type) === "running";
+  const [isCollapsed, setIsCollapsed] = useState(!isRunning);
+  const userOverrodeOpenStateRef = useRef(false);
+  const hasText = text.trim().length > 0;
+  const reasoningTextProps = {
+    type: "text",
+    text,
+    status: isRunning ? { type: "running" } : status
+  };
+  useEffect(() => {
+    if (userOverrodeOpenStateRef.current) {
+      return;
+    }
+    setIsCollapsed(!isRunning);
+  }, [isRunning]);
+  if (!hasText && !isRunning) {
+    return null;
+  }
+  return /* @__PURE__ */ jsxs(
+    Collapsible,
+    {
+      open: !isCollapsed,
+      onOpenChange: (open) => {
+        userOverrodeOpenStateRef.current = true;
+        setIsCollapsed(!open);
+      },
+      className: "aui-assistant-reasoning-root mb-3 rounded-2xl border border-border/60 bg-muted/30 px-3 py-2",
+      children: [
+        /* @__PURE__ */ jsx(CollapsibleTrigger, { asChild: true, children: /* @__PURE__ */ jsxs(
+          "button",
+          {
+            type: "button",
+            className: "flex w-full items-center justify-between gap-3 text-left",
+            "aria-label": isRunning ? "Toggle reasoning while streaming" : "Toggle reasoning",
+            children: [
+              /* @__PURE__ */ jsxs("span", { className: "flex min-w-0 items-center gap-2 text-[13px] font-medium text-muted-foreground", children: [
+                /* @__PURE__ */ jsx(BrainIcon, { className: "size-4 shrink-0" }),
+                /* @__PURE__ */ jsx("span", { children: isRunning ? "Reasoning..." : "Reasoning" })
+              ] }),
+              /* @__PURE__ */ jsxs("span", { className: "flex shrink-0 items-center gap-2", children: [
+                reasoningTokens !== void 0 && /* @__PURE__ */ jsx("span", { className: "rounded-full border border-border/60 bg-background px-2 py-0.5 text-[11px] font-medium text-muted-foreground", children: formatReasoningTokens(reasoningTokens) }),
+                /* @__PURE__ */ jsx(
+                  "span",
+                  {
+                    className: cn(
+                      "flex size-5 items-center justify-center text-muted-foreground transition-transform",
+                      isCollapsed ? "rotate-0" : "rotate-90"
+                    ),
+                    "aria-hidden": "true",
+                    children: /* @__PURE__ */ jsx(ChevronRightIcon, { className: "size-4" })
+                  }
+                )
+              ] })
+            ]
+          }
+        ) }),
+        /* @__PURE__ */ jsx(CollapsibleContent, { className: "pt-3", children: isRunning && !hasText ? /* @__PURE__ */ jsx("span", { className: "shimmer text-[13px] text-muted-foreground", children: "Analyzing..." }) : /* @__PURE__ */ jsx("div", { className: "aui-assistant-reasoning-body text-[13px] leading-relaxed", children: /* @__PURE__ */ jsx(EffectiveTextComponent, { ...reasoningTextProps }) }) })
+      ]
+    }
+  );
+};
+const noToolGrouping = () => null;
+const AthenaAssistantMessage = ({
+  toolUIs,
+  TextComponent = TiptapText,
+  ReasoningComponent,
+  EmptyComponent = AthenaAssistantMessageEmpty,
+  ActionBarComponent = AthenaAssistantActionBar,
+  groupToolCalls = false
+}) => {
+  const effectiveReasoningComponent = ReasoningComponent ?? AthenaReasoningPart;
+  const toolUIsWithNestedMessages = useMemo(() => {
+    const wrappedToolUIs = {};
+    for (const [toolName, ToolUI] of Object.entries(toolUIs)) {
+      wrappedToolUIs[toolName] = withNestedPtcMessages(ToolUI);
+    }
+    return wrappedToolUIs;
+  }, [toolUIs]);
+  const nestedPtcComponents = useMemo(
+    () => ({
+      Text: TextComponent,
+      Reasoning: effectiveReasoningComponent,
+      tools: {
+        Fallback: ToolFallback,
+        by_name: toolUIsWithNestedMessages
+      },
+      Empty: EmptyComponent
+    }),
+    [EmptyComponent, TextComponent, effectiveReasoningComponent, toolUIsWithNestedMessages]
+  );
+  return /* @__PURE__ */ jsxs(
+    MessagePrimitive.Root,
+    {
+      className: "aui-assistant-message-root fade-in slide-in-from-bottom-1 relative mx-auto w-full max-w-(--thread-max-width) animate-in py-3 duration-150",
+      "data-role": "assistant",
+      children: [
+        /* @__PURE__ */ jsx(AthenaReasoningTextComponentContext.Provider, { value: TextComponent, children: /* @__PURE__ */ jsxs("div", { className: "aui-assistant-message-content wrap-break-word px-2 text-foreground leading-relaxed", children: [
+          /* @__PURE__ */ jsx(NestedPtcComponentsProvider, { components: nestedPtcComponents, children: groupToolCalls ? /* @__PURE__ */ jsxs("div", { className: "flex flex-col space-y-2", children: [
+            /* @__PURE__ */ jsx(
+              SuperGroupingCard,
+              {
+                toolUIs: toolUIsWithNestedMessages,
+                TextComponent,
+                ReasoningComponent: effectiveReasoningComponent
+              }
+            ),
+            /* @__PURE__ */ jsx(SuperGroupingFinalText, { TextComponent }),
+            /* @__PURE__ */ jsx(AthenaMessageActivityIndicator, {})
+          ] }) : /* @__PURE__ */ jsxs(Fragment, { children: [
+            /* @__PURE__ */ jsx(AthenaMessageEmptyIndicator, { EmptyComponent }),
+            /* @__PURE__ */ jsx(MessagePrimitive.GroupedParts, { groupBy: noToolGrouping, children: ({ part }) => {
+              var _a;
+              switch (part.type) {
+                case "tool-call": {
+                  const ToolUI = toolUIsWithNestedMessages[part.toolName];
+                  const toolProps = part;
+                  return ToolUI ? /* @__PURE__ */ jsx(ToolUI, { ...toolProps }) : /* @__PURE__ */ jsx(ToolFallback, { ...toolProps });
+                }
+                case "reasoning": {
+                  const ReasoningRenderer = effectiveReasoningComponent;
+                  return /* @__PURE__ */ jsx(ReasoningRenderer, { ...part });
+                }
+                case "text": {
+                  const textPart = part;
+                  if (textPart.text === "" && ((_a = textPart.status) == null ? void 0 : _a.type) === "running") {
+                    return /* @__PURE__ */ jsx(EmptyComponent, { status: textPart.status });
+                  }
+                  return /* @__PURE__ */ jsx(TextComponent, { ...textPart });
+                }
+                default:
+                  return null;
+              }
+            } }),
+            /* @__PURE__ */ jsx(AthenaMessageActivityIndicator, {})
+          ] }) }),
+          /* @__PURE__ */ jsx(MessageError, {})
+        ] }) }),
+        /* @__PURE__ */ jsx("div", { className: "aui-assistant-message-footer mt-1 ml-2 flex", children: /* @__PURE__ */ jsx(ActionBarComponent, {}) })
+      ]
+    }
+  );
+};
+const AthenaAssistantActionBar = ({ className }) => {
+  const threadId = useAthenaThreadId();
+  const { appUrl } = useAthenaConfig();
+  return /* @__PURE__ */ jsxs(
+    ActionBarPrimitive.Root,
+    {
+      hideWhenRunning: true,
+      autohide: "not-last",
+      className: cn("aui-assistant-action-bar-root -ml-1 flex gap-1 text-muted-foreground", className),
+      children: [
+        /* @__PURE__ */ jsx(ActionBarPrimitive.Copy, { asChild: true, children: /* @__PURE__ */ jsxs(TooltipIconButton, { tooltip: "Copy", children: [
+          /* @__PURE__ */ jsx(AuiIf, { condition: (s) => s.message.isCopied, children: /* @__PURE__ */ jsx(CheckIcon, {}) }),
+          /* @__PURE__ */ jsx(AuiIf, { condition: (s) => !s.message.isCopied, children: /* @__PURE__ */ jsx(CopyIcon, {}) })
+        ] }) }),
+        /* @__PURE__ */ jsxs(ActionBarMorePrimitive.Root, { children: [
+          /* @__PURE__ */ jsx(ActionBarMorePrimitive.Trigger, { asChild: true, children: /* @__PURE__ */ jsx(TooltipIconButton, { tooltip: "More", className: "data-[state=open]:bg-accent", children: /* @__PURE__ */ jsx(MoreHorizontalIcon, {}) }) }),
+          /* @__PURE__ */ jsxs(
+            ActionBarMorePrimitive.Content,
+            {
+              side: "bottom",
+              align: "start",
+              className: "aui-action-bar-more-content z-50 min-w-32 overflow-hidden rounded-md border bg-popover p-1 text-popover-foreground shadow-md",
+              children: [
+                /* @__PURE__ */ jsx(ActionBarPrimitive.ExportMarkdown, { asChild: true, children: /* @__PURE__ */ jsxs(ActionBarMorePrimitive.Item, { className: "aui-action-bar-more-item flex cursor-pointer select-none items-center gap-2 rounded-sm px-2 py-1.5 text-sm outline-none hover:bg-accent hover:text-accent-foreground focus:bg-accent focus:text-accent-foreground", children: [
+                  /* @__PURE__ */ jsx(DownloadIcon, { className: "size-4" }),
+                  "Export as Markdown"
+                ] }) }),
+                threadId && /* @__PURE__ */ jsxs(
+                  ActionBarMorePrimitive.Item,
+                  {
+                    className: "aui-action-bar-more-item flex cursor-pointer select-none items-center gap-2 rounded-sm px-2 py-1.5 text-sm outline-none hover:bg-accent hover:text-accent-foreground focus:bg-accent focus:text-accent-foreground",
+                    onClick: () => {
+                      window.open(createAthenaSpacesUrl({ appUrl, sessionId: threadId }), "_blank");
+                    },
+                    children: [
+                      /* @__PURE__ */ jsx(ExternalLinkIcon, { className: "size-4" }),
+                      "Open in Athena"
+                    ]
+                  }
+                )
+              ]
+            }
+          )
+        ] })
+      ]
+    }
+  );
+};
+const AthenaUserMessage = ({
+  TextComponent = TiptapText
+}) => /* @__PURE__ */ jsx(
+  MessagePrimitive.Root,
+  {
+    className: "aui-user-message-root fade-in slide-in-from-bottom-1 mx-auto flex w-full max-w-(--thread-max-width) animate-in justify-end px-2 py-3 duration-150",
+    "data-role": "user",
+    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(MessagePrimitive.Parts, { components: { Text: TextComponent } }) })
+  }
+);
+export {
+  AthenaAssistantActionBar,
+  AthenaAssistantMessage,
+  AthenaChat,
+  AthenaReasoningPart,
+  AthenaUserMessage
+};
+//# sourceMappingURL=AthenaChat.js.map