@athenaintel/react 0.57.2 → 0.59.0

package/dist/chat/AthenaChat.js

@@ -2,6 +2,7 @@ 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 { statewireTypedInterrupts } from "./typed-interrupts/StatewireTypedInterruptCard.js";
 import { AthenaPlanStrip } from "./AthenaPlanStrip.js";
 import { StatewireQueuedMessages } from "./StatewireQueuedMessages.js";
 import { StatewireConnectionBanner, StatewireErrorBanner } from "./statewire-banners.js";
@@ -30,6 +31,7 @@ import { ChatComposerDock } from "../chat-ui/thread/chat-composer-dock.js";
 import { ChatThreadShell } from "../chat-ui/thread/chat-thread-shell.js";
 import { HandoffCursorReveal } from "../chat-ui/thread/handoff-cursor.js";
 import { StatewireHistoryEdge } from "../chat-ui/thread/statewire-history-edge.js";
+import { DeepAgentLaneFollowScope } from "../subagents/lane-follow.js";
 import { ThreadLoadingOverlay } from "../chat-ui/thread/thread-loading-overlay.js";
 import { ComposerSendOrStop } from "../chat-ui/thread/composer-send-or-stop.js";
 import { resolveComposerAction } from "../chat-ui/composer/composer-action.js";
@@ -43,6 +45,14 @@ import { useAthenaDictationEngine } from "../dictation/use-athena-dictation-engi
 import { TiptapComposer } from "../chat-ui/composer/tiptap-composer.js";
 import { MentionAuthCtx } from "../chat-ui/mentions/suggestions/data/api-client.js";
 import { useAccountIdentity } from "../chat-ui/mentions/use-account-workspace-id.js";
+import { readDeepAgentNamedAgents, keepDeepAgentNamedAgentsForIdentity } from "../agent-runtime/src/named-agents.js";
+import { stageComposerNamedAgents } from "../chat-ui/composer/composer-named-agents.js";
+import { useCatalogQueryClient, useAthenaViewerIsAdmin } from "../catalog/runtime-catalog.js";
+import { usePendingAttachedReferences, stageThreadAttachedReferencesForSend, clearAttachedReferences, keepAttachedReferencesForIdentity } from "../composer/attached-references.js";
+import { MessageAttachedReferences, ComposerAttachedReferences } from "../composer/ComposerAttachedReferences.js";
+import { useAttachedReferencesComposerProps } from "../composer/useAttachedReferencesComposerProps.js";
+import { ComposerNamedAgents } from "../composer/ComposerNamedAgents.js";
+import { useAthenaChatAgentsMention } from "./chat-agent-mentions.js";
 import { AnswerLabel } from "../chat-ui/messages/answer-label.js";
 import { useMessageArrivalAnimation, MESSAGE_ARRIVAL_ANIMATION_CLASS } from "../chat-ui/messages/arrival-animation.js";
 import { MessageImagePart, MessageImageGallery } from "../chat-ui/messages/message-images.js";
@@ -53,6 +63,7 @@ import { useAthenaRunLive } from "./use-athena-run-live.js";
 import { DEFAULT_WELCOME_SUGGESTIONS, AthenaChatWelcomeSuggestions, AthenaChatWelcome, welcomeEntrance, welcomeComposerDelayMs } from "./athena-chat-welcome.js";
 import { AthenaComposerSendButton, ATHENA_COMPOSER_COMPONENTS, AthenaComposerMentionButton } from "./athena-composer-controls.js";
 import { useComposerKeyboardHint, ComposerKeyboardHint } from "./composer-keyboard-hint.js";
+import { FollowUpSuggestions } from "./follow-up-suggestions.js";
 import { usePrefersReducedMotion } from "../lib/use-prefers-reduced-motion.js";
 import { useStatewireSessionLane } from "../runtime/statewire-session-lane.js";
 import { AthenaAuthContext } from "../auth/AthenaAuthContext.js";
@@ -70,6 +81,7 @@ import { QuoteProvider, useQuoteFromPostMessage, useQuote } from "../composer/Qu
 import { AttachmentProvider, useAttachments } from "../composer/AttachmentContext.js";
 import { buildComposedSend, holdComposedSend } from "../composer/build-composed-message.js";
 import { useAttachFiles } from "../composer/useAttachFiles.js";
+import { AthenaChatComposerActionsContext } from "./composer-actions.js";
 import { useAthenaConfig } from "../provider/AthenaContext.js";
 import { createAthenaSpacesUrl } from "../lib/athena-urls.js";
 import { isRecord } from "../lib/utils.js";
@@ -77,8 +89,8 @@ import { getToolkitDisplayInfo } from "../toolkits/index.js";
 import { runActivityKey, runActivityVerb, isToolCallInFlight, runActivityLabel, runProgressKey, runElapsedLabel, isRunActivityStalled } from "./run-activity.js";
 import { readMessageCreatedAt, parseMessageTime, isSameCalendarDay, formatDayLabel, formatMessageTime, formatMessageDateTime } from "./message-time.js";
 import { ThreadScrollToBottom } from "./thread-scroll-to-bottom.js";
+import { AthenaTurnChangesSlot, AthenaTurnChangesContext } from "./turn-changes.js";
 import { QueryClientProvider } from "@tanstack/react-query";
-import { useCatalogQueryClient } from "../catalog/runtime-catalog.js";
 import { isStatewireChannelLive } from "../runtime/statewire-lifecycle.js";
 import { useComposerChatVoice, ChatVoiceButton, ChatVoiceStatus, ChatVoiceContinueOffer } from "./chat-voice-controls.js";
 import { TASK_TOOL_NAME } from "../agent-runtime/src/tool-names.js";
@@ -87,6 +99,7 @@ import { readSoleRuntimeMessageKind, readRuntimeMessageFromSourceMessages } from
 import { isGithubWakeMessage } from "../agent-runtime/src/github-wake-event.js";
 import { isSubscriptionWakeMessage } from "../agent-runtime/src/subscription-wake-event.js";
 import { useDeepAgentConnection, readRunLive } from "../agent-runtime/src/thread-status.js";
+import { autoComposerSendOptions } from "../agent-runtime/src/run-send.js";
 import { useChatOutboxReachable, useDeepAgentChatOutboxContext } from "../agent-runtime/src/use-deep-agent-chat-outbox.js";
 import { CHAT_EDIT_UNREACHABLE_LABEL } from "../agent-runtime/src/chat-outbox.js";
 import { cn } from "cnfast";
@@ -174,17 +187,21 @@ const AthenaChatContent = ({
   confirmDecision,
   toolUIs,
   mentionTools,
+  mentionCategories = ATHENA_CHAT_MENTION_CATEGORIES,
   welcomeSuggestions = DEFAULT_WELCOME_SUGGESTIONS,
   components,
   groupToolCalls = true,
   composer: composerSlots,
   dictation = true,
   voice = true,
-  continueByVoice = null
+  continueByVoice = null,
+  turnChanges = null,
+  followUpSuggestions = false
 }) => {
   var _a, _b, _c, _d, _e, _f;
   const athenaConfig = useAthenaConfig();
   const authState = useContext(AthenaAuthContext);
+  const namesAgents = mentionCategories.includes("agents");
   const statewireThreadId = useContext(AthenaStatewireThreadIdContext);
   const startNewSession = (_a = useAthenaStatewireThreadList()) == null ? void 0 : _a.newThread;
   const mentionCredentials = useMemo(
@@ -213,6 +230,13 @@ const AthenaChatContent = ({
   );
   const account = useAccountIdentity(mentionCredentials, { tokenGetter: athenaConfig.getToken });
   const accountWorkspaceId = account.workspaceId;
+  const accountUserId = account.userId;
+  useEffect(() => {
+    if (namesAgents && accountUserId) keepDeepAgentNamedAgentsForIdentity(accountUserId);
+  }, [accountUserId, namesAgents]);
+  useEffect(() => {
+    if (accountUserId) keepAttachedReferencesForIdentity(accountUserId);
+  }, [accountUserId]);
   const mentionAuth = useMemo(
     () => ({
       ...mentionCredentials,
@@ -272,6 +296,21 @@ const AthenaChatContent = ({
       userTextComponent
     ]
   );
+  const turnChangesEnabled = (turnChanges == null ? void 0 : turnChanges.enabled) === true;
+  const turnChangesScope = turnChanges == null ? void 0 : turnChanges.reviewScope;
+  const turnChangesRead = turnChanges == null ? void 0 : turnChanges.read;
+  const turnChangesView = turnChanges == null ? void 0 : turnChanges.onView;
+  const turnChangesUndo = turnChanges == null ? void 0 : turnChanges.onUndo;
+  const turnChangesValue = useMemo(
+    () => turnChangesEnabled && turnChangesRead && turnChangesView && turnChangesUndo ? {
+      enabled: true,
+      ...turnChangesScope !== void 0 && { reviewScope: turnChangesScope },
+      read: turnChangesRead,
+      onView: turnChangesView,
+      onUndo: turnChangesUndo
+    } : null,
+    [turnChangesEnabled, turnChangesScope, turnChangesRead, turnChangesView, turnChangesUndo]
+  );
   const AssistantMessageComponent = (components == null ? void 0 : components.AssistantMessage) ?? AthenaDefaultAssistantMessage;
   const UserMessageComponent = (components == null ? void 0 : components.UserMessage) ?? AthenaDefaultUserMessage;
   const messageComponents = useMemo(
@@ -281,7 +320,7 @@ const AthenaChatContent = ({
     }),
     [AssistantMessageComponent, UserMessageComponent]
   );
-  return /* @__PURE__ */ jsx(AthenaChatDefaultComponentsContext.Provider, { value: defaultComponentsContextValue, children: /* @__PURE__ */ jsx(AthenaChatErrorBoundary, { children: /* @__PURE__ */ jsx(StatewireApprovalWaitProvider, { children: /* @__PURE__ */ jsx(ToolFallbackAppearanceProvider, { appearance: "thread", children: /* @__PURE__ */ jsx(MentionAuthCtx.Provider, { value: mentionAuth, children: /* @__PURE__ */ jsx(
+  return /* @__PURE__ */ jsx(AthenaChatDefaultComponentsContext.Provider, { value: defaultComponentsContextValue, children: /* @__PURE__ */ jsx(AthenaTurnChangesContext.Provider, { value: turnChangesValue, children: /* @__PURE__ */ jsx(AthenaChatErrorBoundary, { children: /* @__PURE__ */ jsx(StatewireApprovalWaitProvider, { children: /* @__PURE__ */ jsx(DeepAgentLaneFollowScope, { children: /* @__PURE__ */ jsx(ToolFallbackAppearanceProvider, { appearance: "thread", children: /* @__PURE__ */ jsx(MentionAuthCtx.Provider, { value: mentionAuth, children: /* @__PURE__ */ jsx(
     ChatThreadShell,
     {
       overlay: /* @__PURE__ */ jsxs(Fragment, { children: [
@@ -319,7 +358,13 @@ const AthenaChatContent = ({
         /* @__PURE__ */ jsx(StatewireConnectionBanner, {}),
         /* @__PURE__ */ jsx(StatewireErrorBanner, { onStartNewSession: startNewSession }),
         /* @__PURE__ */ jsx(StatewireRunRefusalNotice, { onStartNewSession: startNewSession }),
-        /* @__PURE__ */ jsx(StatewireApprovalCard, { confirmDecision }),
+        /* @__PURE__ */ jsx(
+          StatewireApprovalCard,
+          {
+            confirmDecision,
+            typedInterrupts: statewireTypedInterrupts
+          }
+        ),
         plan === void 0 ? /* @__PURE__ */ jsx(AthenaPlanStrip, { className: "mx-3" }) : plan,
         /* @__PURE__ */ jsx(StatewireQueuedMessages, {}),
         /* @__PURE__ */ jsx(AttachmentProvider, { children: /* @__PURE__ */ jsxs(QuoteProvider, { children: [
@@ -328,17 +373,19 @@ const AthenaChatContent = ({
             AthenaChatComposer,
             {
               tools,
+              mentionCategories,
               slots: composerSlots,
               dictation,
               voice: readOnly ? false : voice,
               continueByVoice: readOnly ? null : continueByVoice,
-              welcomeSuggestionCount: welcomeSuggestions.length
+              welcomeSuggestionCount: welcomeSuggestions.length,
+              followUpSuggestions: followUpSuggestions && !readOnly
             }
           )
         ] }) })
       ] })
     }
-  ) }) }) }) }, statewireThreadId ?? void 0) });
+  ) }) }) }) }) }, statewireThreadId ?? void 0) }) });
 };
 const THREAD_VIEWPORT_ATTR = "data-aui-thread-viewport";
 const useScrollToTop = () => {
@@ -397,12 +444,35 @@ const ATHENA_CHAT_MENTION_CATEGORIES = [
   "projects",
   "folders"
 ];
-const AthenaChatComposer = ({ tools, slots, dictation, voice: voiceOptions, continueByVoice, welcomeSuggestionCount }) => {
+const ATHENA_CHAT_MENTION_ORDER = [
+  "favorites",
+  "assets",
+  "projects",
+  "browse_profiles",
+  "folders",
+  "toolkits",
+  "tools",
+  "team",
+  "agents",
+  "drives"
+];
+const AthenaChatComposer = ({
+  tools,
+  mentionCategories,
+  slots,
+  dictation,
+  voice: voiceOptions,
+  continueByVoice,
+  welcomeSuggestionCount,
+  followUpSuggestions
+}) => {
   const aui = useAui();
   const threadId = useAthenaStatewireThreadId();
   const { appUrl } = useAthenaConfig();
   const { attachments, clearAttachments, isUploading, notice: attachNotice, setNotice: setAttachNotice } = useAttachments();
   const { quote, clearQuote } = useQuote();
+  const attachedReferences = usePendingAttachedReferences(threadId);
+  const attachedReferencesComposerProps = useAttachedReferencesComposerProps(threadId);
   const { attachFiles } = useAttachFiles();
   const sessionLane = useStatewireSessionLane();
   const composerRef = useRef(null);
@@ -442,9 +512,9 @@ const AthenaChatComposer = ({ tools, slots, dictation, voice: voiceOptions, cont
   const appendSpokenText = useCallback(
     (spoken) => {
       if (!spoken) return;
-      const composer = composerRef.current;
-      if (composer == null ? void 0 : composer.appendText) {
-        composer.appendText(spoken);
+      const composer2 = composerRef.current;
+      if (composer2 == null ? void 0 : composer2.appendText) {
+        composer2.appendText(spoken);
         return;
       }
       aui.composer.setText(composeDictationDraft(aui.composer.getState().text ?? "", spoken));
@@ -466,16 +536,74 @@ const AthenaChatComposer = ({ tools, slots, dictation, voice: voiceOptions, cont
     if (!dictationEnabled && dictationActive) cancelDictation();
   }, [cancelDictation, dictationActive, dictationEnabled]);
   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 offered = new Set(mentionCategories);
+    if (tools.some((tool) => tool.type === "toolkit")) offered.add("toolkits");
+    if (tools.some((tool) => tool.type === "tool")) offered.add("tools");
+    return ATHENA_CHAT_MENTION_ORDER.filter((category) => offered.has(category));
+  }, [mentionCategories, tools]);
+  const namesAgents = rootCategories.includes("agents");
+  const viewerIsAdmin = useAthenaViewerIsAdmin({
+    enabled: namesAgents || rootCategories.includes("team")
+  });
+  const sayNamedAgentsLimit = useCallback((message) => {
+    queueMicrotask(() => setComposerNotice(message));
+  }, []);
+  const chatAgents = useAthenaChatAgentsMention({
+    threadId,
+    enabled: namesAgents,
+    isAdmin: viewerIsAdmin,
+    onLimit: sayNamedAgentsLimit
+  });
+  const focusComposer = useCallback(() => {
+    var _a, _b;
+    return (_b = (_a = composerRef.current) == null ? void 0 : _a.focus) == null ? void 0 : _b.call(_a);
+  }, []);
+  const filePickerRef = useRef(null);
+  const composerActions = useMemo(
+    () => ({
+      openFilePicker: () => {
+        var _a;
+        if (needsConnectionReason) return;
+        (_a = filePickerRef.current) == null ? void 0 : _a.click();
+      },
+      openMentionMenu: () => {
+        var _a, _b;
+        return (_b = (_a = composerRef.current) == null ? void 0 : _a.openMentionMenu) == null ? void 0 : _b.call(_a);
+      },
+      needsConnectionReason
+    }),
+    [needsConnectionReason]
+  );
+  const pickFollowUp = useCallback(
+    (text) => {
+      var _a;
+      const composer2 = composerRef.current;
+      if (composer2 == null ? void 0 : composer2.appendText) composer2.appendText(text);
+      else aui.composer.setText(text);
+      (_a = composer2 == null ? void 0 : composer2.focus) == null ? void 0 : _a.call(composer2);
+    },
+    [aui]
+  );
   const attachmentsRef = useRef(attachments);
   attachmentsRef.current = attachments;
   const quoteRef = useRef(quote);
   quoteRef.current = quote;
+  const attachedReferencesRef = useRef(attachedReferences);
+  attachedReferencesRef.current = attachedReferences;
   const hasExtras = attachments.length > 0 || quote !== null;
+  const sendStateRef = useRef({ isRunning, liveVoice: false });
+  sendStateRef.current = { isRunning, liveVoice: voice.phase === "live" && !hasExtras };
+  const onBeforeSend = slots == null ? void 0 : slots.onBeforeSend;
+  const beforeSubmit = useCallback(
+    async ({ text }) => {
+      if (!onBeforeSend) return;
+      const { isRunning: running, liveVoice } = sendStateRef.current;
+      const action = liveVoice ? "send" : resolveComposerAction({ hasDraft: true, isRunning: running });
+      if (action === "stop") return;
+      await onBeforeSend({ action, text });
+    },
+    [onBeforeSend]
+  );
   const isUploadingRef = useRef(isUploading);
   isUploadingRef.current = isUploading;
   const uploadWaitersRef = useRef([]);
@@ -501,13 +629,23 @@ const AthenaChatComposer = ({ tools, slots, dictation, voice: voiceOptions, cont
       });
       if (!send) return false;
       prepareComposerSend({ threadId, ...send });
+      if (namesAgents) {
+        stageComposerNamedAgents({
+          threadId,
+          text: send.text,
+          agents: readDeepAgentNamedAgents(threadId)
+        });
+      }
+      const references = stageThreadAttachedReferencesForSend({ threadId, text: send.text });
       aui.composer.setText(send.text);
-      aui.composer.send({ steer: false });
+      aui.composer.send(
+        autoComposerSendOptions(aui.thread().getState().isRunning && references.length === 0)
+      );
       clearAttachments();
       clearQuote();
       return true;
     },
-    [appUrl, aui, clearAttachments, clearQuote, threadId]
+    [appUrl, aui, clearAttachments, clearQuote, namesAgents, threadId]
   );
   const continueByVoiceRef = useRef(continueByVoice);
   continueByVoiceRef.current = continueByVoice;
@@ -522,7 +660,7 @@ const AthenaChatComposer = ({ tools, slots, dictation, voice: voiceOptions, cont
   }, [onKeyboardHintSubmitted, settleContinueByVoice]);
   const interceptSubmit = useCallback(
     async (text, draftMarkdown) => {
-      if (attachmentsRef.current.length > 0 || quoteRef.current !== null) {
+      if (attachmentsRef.current.length > 0 || quoteRef.current !== null || attachedReferencesRef.current.length > 0 && text.length > 0) {
         return sendComposed({ text, draftMarkdown });
       }
       return voice.phase === "live" && text.length > 0 ? voice.sendUserText(text) : false;
@@ -530,14 +668,18 @@ const AthenaChatComposer = ({ tools, slots, dictation, voice: voiceOptions, cont
     [sendComposed, voice.phase, voice.sendUserText]
   );
   const hasPendingExtras = useCallback(
-    () => attachmentsRef.current.length > 0 || quoteRef.current !== null,
-    []
+    () => {
+      var _a, _b, _c;
+      return attachmentsRef.current.length > 0 || quoteRef.current !== null || attachedReferencesRef.current.length > 0 && Boolean(((_c = (_b = (_a = composerRef.current) == null ? void 0 : _a.captureSend) == null ? void 0 : _b.call(_a)) == null ? void 0 : _c.text) ?? aui.composer.getState().text);
+    },
+    [aui]
   );
   const holdPendingExtras = useCallback(
     ({
       outbox,
       send,
-      attachments: attachments2
+      attachments: attachments2,
+      addressedAgents
     }) => {
       const held = holdComposedSend({
         outbox,
@@ -545,14 +687,19 @@ const AthenaChatComposer = ({ tools, slots, dictation, voice: voiceOptions, cont
         carried: attachments2,
         quote: quoteRef.current,
         send,
-        appUrl
+        appUrl,
+        // The agents it names ride its message.
+        ...addressedAgents && { addressedAgents },
+        // So do the references attached to it.
+        attachedReferences: attachedReferencesRef.current
       });
       if (!held) return false;
       clearAttachments();
       clearQuote();
+      clearAttachedReferences(threadId);
       return true;
     },
-    [appUrl, clearAttachments, clearQuote]
+    [appUrl, clearAttachments, clearQuote, threadId]
   );
   const fileIntake = useMemo(
     () => ({
@@ -575,12 +722,12 @@ const AthenaChatComposer = ({ tools, slots, dictation, voice: voiceOptions, cont
       text: aui.composer.getState().text ?? "",
       draftMarkdown: aui.composer.getState().text ?? ""
     };
-    const extras = attachmentsRef.current.length > 0 || quoteRef.current !== null;
+    const extras = attachmentsRef.current.length > 0 || quoteRef.current !== null || attachedReferencesRef.current.length > 0 && send.text.length > 0;
     if (!send.text && !extras) return;
     const action = resolveComposerAction({ hasDraft: true, isRunning });
     if (action === "stop") return;
     try {
-      await ((_c = slots == null ? void 0 : slots.onBeforeSend) == null ? void 0 : _c.call(slots, { action }));
+      await ((_c = slots == null ? void 0 : slots.onBeforeSend) == null ? void 0 : _c.call(slots, { action, text: send.text }));
     } catch (error2) {
       athenaDiagnostics.emit("composer.dictation", {
         level: "warn",
@@ -591,27 +738,47 @@ const AthenaChatComposer = ({ tools, slots, dictation, voice: voiceOptions, cont
       });
       return;
     }
+    const namedAgents = namesAgents ? readDeepAgentNamedAgents(threadId) : [];
     settleContinueByVoice();
     if (extras) {
       sendComposed(send);
       return;
     }
     prepareComposerSend({ threadId, ...send });
+    if (namesAgents) stageComposerNamedAgents({ threadId, text: send.text, agents: namedAgents });
     aui.composer.setText(send.text);
-    aui.composer.send({ steer: false });
+    aui.composer.send(autoComposerSendOptions(aui.thread().getState().isRunning));
   }, [
     appendSpokenText,
     aui,
     dictationCtl,
     isRunning,
+    namesAgents,
     sendComposed,
     sessionLane,
     settleContinueByVoice,
     slots,
     threadId
   ]);
-  return /* @__PURE__ */ jsxs(ComposerDropZone, { onDragActiveChange: setDragActive, children: [
+  const composer = /* @__PURE__ */ jsxs(ComposerDropZone, { onDragActiveChange: setDragActive, children: [
     /* @__PURE__ */ jsx("audio", { ref: audioRef, autoPlay: true, playsInline: true }),
+    /* @__PURE__ */ jsx(
+      "input",
+      {
+        ref: filePickerRef,
+        type: "file",
+        multiple: true,
+        className: "hidden",
+        "aria-hidden": true,
+        tabIndex: -1,
+        onChange: (event) => {
+          const input = event.currentTarget;
+          const files = input.files ? Array.from(input.files) : [];
+          input.value = "";
+          if (files.length > 0) void attachFiles(files, "paperclip");
+        }
+      }
+    ),
     /* @__PURE__ */ jsx(
       "div",
       {
@@ -623,6 +790,15 @@ const AthenaChatComposer = ({ tools, slots, dictation, voice: voiceOptions, cont
           {
             appearance: "paper",
             dragActive,
+            aboveSurface: /* @__PURE__ */ jsx(
+              FollowUpSuggestions,
+              {
+                enabled: followUpSuggestions,
+                threadId,
+                hasDraft: hasText || hasExtras,
+                onPick: pickFollowUp
+              }
+            ),
             beforeEditor: /* @__PURE__ */ jsxs(Fragment, { children: [
               slots == null ? void 0 : slots.aboveEditor,
               /* @__PURE__ */ jsx(
@@ -643,6 +819,8 @@ const AthenaChatComposer = ({ tools, slots, dictation, voice: voiceOptions, cont
                   disabledReason: needsConnectionReason
                 }
               ),
+              namesAgents ? /* @__PURE__ */ jsx(ComposerNamedAgents, { threadId, onEmptied: focusComposer }) : null,
+              /* @__PURE__ */ jsx(ComposerAttachedReferences, { threadId, onEmptied: focusComposer }),
               /* @__PURE__ */ jsx(ComposerQuotePreview, {}),
               /* @__PURE__ */ jsx(ComposerAttachmentPreview, {})
             ] }),
@@ -652,19 +830,24 @@ const AthenaChatComposer = ({ tools, slots, dictation, voice: voiceOptions, cont
                 {
                   ref: composerRef,
                   threadId,
+                  ...attachedReferencesComposerProps,
                   appearance: "paper",
                   placeholder: (runLive ? slots == null ? void 0 : slots.runningPlaceholder : void 0) ?? (slots == null ? void 0 : slots.placeholder) ?? DEFAULT_COMPOSER_PLACEHOLDER,
                   tools,
                   rootCategories,
+                  chatAgents,
+                  viewerIsAdmin,
                   onDraftStateChange,
                   mentionSearchUnavailable: needsConnectionReason ? NEEDS_CONNECTION : null,
                   waitForHostUploads,
                   interceptSubmit,
+                  beforeSubmit: onBeforeSend ? beforeSubmit : void 0,
                   hasPendingExtras,
                   holdPendingExtras,
                   onAttachmentError: setComposerNotice,
                   fileIntake,
-                  onSubmitted: handleComposerSubmitted
+                  onSubmitted: handleComposerSubmitted,
+                  slashCommands: slots == null ? void 0 : slots.slashCommands
                 }
               ) }),
               notice && /* @__PURE__ */ jsx("p", { className: "aui-composer-notice px-1.5 pb-1 text-destructive text-xs", role: "alert", children: notice }),
@@ -679,81 +862,89 @@ const AthenaChatComposer = ({ tools, slots, dictation, voice: voiceOptions, cont
               ),
               dictationEnabled && dictationCtl.status === DICTATION_STATUSES.error && dictationCtl.error && /* @__PURE__ */ jsx("p", { className: "aui-dictation-error px-1.5 pb-1 text-destructive text-xs", role: "alert", children: dictationCtl.error.message })
             ] }),
-            toolbarStart: /* @__PURE__ */ jsx("div", { className: "flex items-center gap-1", children: recording ? /* @__PURE__ */ jsx(DictationCancelButton, { onClick: dictationCtl.cancel }) : /* @__PURE__ */ jsxs(Fragment, { children: [
-              /* @__PURE__ */ jsx(
-                FileUploadButton,
-                {
-                  className: "aui-composer-attach size-8 rounded-full p-0 disabled:opacity-30",
-                  disabledReason: needsConnectionReason
-                }
-              ),
-              /* @__PURE__ */ jsx(
-                AthenaComposerMentionButton,
-                {
-                  onClick: () => {
-                    var _a, _b;
-                    return (_b = (_a = composerRef.current) == null ? void 0 : _a.openMentionMenu) == null ? void 0 : _b.call(_a);
-                  },
-                  disabledReason: needsConnectionReason
-                }
-              ),
-              /* @__PURE__ */ jsx(AthenaChatModeMenu, {}),
-              slots == null ? void 0 : slots.toolbarStart
-            ] }) }),
-            toolbarEnd: /* @__PURE__ */ jsx("div", { className: "flex items-center gap-1", children: recording ? /* @__PURE__ */ jsxs(Fragment, { children: [
-              /* @__PURE__ */ jsx(
-                DictationStopButton,
-                {
-                  onClick: () => void stopDictation(),
-                  disabled: dictationCtl.status === DICTATION_STATUSES.stopping
-                }
-              ),
-              /* @__PURE__ */ jsx(
-                AthenaComposerSendButton,
-                {
-                  action: isRunning ? "queue" : "send",
-                  onSend: () => void sendDictation(),
-                  disabled: dictationCtl.status !== DICTATION_STATUSES.listening || isUploading,
-                  title: isUploading ? "Upload in progress..." : void 0
-                }
-              )
-            ] }) : /* @__PURE__ */ jsxs(Fragment, { children: [
-              slots == null ? void 0 : slots.toolbarEnd,
-              showMic && /* @__PURE__ */ jsx(
-                DictationMicButton,
-                {
-                  onClick: dictationCtl.start,
-                  disabled: isUploading,
-                  disabledReason: needsConnectionReason
-                }
-              ),
-              voiceEnabled && /* @__PURE__ */ jsx(
-                ChatVoiceButton,
-                {
-                  voice,
-                  disabled: isUploading || dictationCtl.isActive || !isStatewireChannelLive(connection),
-                  disabledReason: needsConnectionReason
-                }
-              ),
-              /* @__PURE__ */ jsx(
-                ComposerSendOrStop,
-                {
-                  composerRef,
-                  hasDraft: hasText || hasExtras,
-                  liveVoice: voice.phase === "live" && !hasExtras,
-                  liveSession: voiceActive ? voice : null,
-                  sendBlockedReason: isUploading ? "Upload in progress..." : null,
-                  onBeforeSend: slots == null ? void 0 : slots.onBeforeSend,
-                  components: ATHENA_COMPOSER_COMPONENTS
-                }
-              )
-            ] }) }),
+            toolbarStart: (
+              // `min-w-0 flex-wrap`: when the chips are wider than the composer (a narrow pane, or a
+              // host's own chips), they wrap onto another row instead of pushing voice and Send past
+              // its edge. A toolbar that fits lays out as before.
+              /* @__PURE__ */ jsx("div", { className: "flex min-w-0 flex-wrap items-center gap-1", children: recording ? /* @__PURE__ */ jsx(DictationCancelButton, { onClick: dictationCtl.cancel }) : /* @__PURE__ */ jsxs(Fragment, { children: [
+                (slots == null ? void 0 : slots.attachButton) === false ? null : /* @__PURE__ */ jsx(
+                  FileUploadButton,
+                  {
+                    className: "aui-composer-attach size-8 rounded-full p-0 disabled:opacity-30",
+                    disabledReason: needsConnectionReason
+                  }
+                ),
+                (slots == null ? void 0 : slots.mentionButton) === false ? null : /* @__PURE__ */ jsx(
+                  AthenaComposerMentionButton,
+                  {
+                    onClick: () => {
+                      var _a, _b;
+                      return (_b = (_a = composerRef.current) == null ? void 0 : _a.openMentionMenu) == null ? void 0 : _b.call(_a);
+                    },
+                    disabledReason: needsConnectionReason
+                  }
+                ),
+                /* @__PURE__ */ jsx(AthenaChatModeMenu, {}),
+                slots == null ? void 0 : slots.toolbarStart
+              ] }) })
+            ),
+            toolbarEnd: (
+              // Never shrinks, and sits at the last row's end once the chips wrap.
+              /* @__PURE__ */ jsx("div", { className: "flex shrink-0 items-center gap-1 self-end", children: recording ? /* @__PURE__ */ jsxs(Fragment, { children: [
+                /* @__PURE__ */ jsx(
+                  DictationStopButton,
+                  {
+                    onClick: () => void stopDictation(),
+                    disabled: dictationCtl.status === DICTATION_STATUSES.stopping
+                  }
+                ),
+                /* @__PURE__ */ jsx(
+                  AthenaComposerSendButton,
+                  {
+                    action: isRunning ? "queue" : "send",
+                    onSend: () => void sendDictation(),
+                    disabled: dictationCtl.status !== DICTATION_STATUSES.listening || isUploading,
+                    title: isUploading ? "Upload in progress..." : void 0
+                  }
+                )
+              ] }) : /* @__PURE__ */ jsxs(Fragment, { children: [
+                slots == null ? void 0 : slots.toolbarEnd,
+                showMic && /* @__PURE__ */ jsx(
+                  DictationMicButton,
+                  {
+                    onClick: dictationCtl.start,
+                    disabled: isUploading,
+                    disabledReason: needsConnectionReason
+                  }
+                ),
+                voiceEnabled && /* @__PURE__ */ jsx(
+                  ChatVoiceButton,
+                  {
+                    voice,
+                    disabled: isUploading || dictationCtl.isActive || !isStatewireChannelLive(connection),
+                    disabledReason: needsConnectionReason
+                  }
+                ),
+                /* @__PURE__ */ jsx(
+                  ComposerSendOrStop,
+                  {
+                    composerRef,
+                    hasDraft: hasText || hasExtras,
+                    liveVoice: voice.phase === "live" && !hasExtras,
+                    liveSession: voiceActive ? voice : null,
+                    sendBlockedReason: isUploading ? "Upload in progress..." : null,
+                    components: ATHENA_COMPOSER_COMPONENTS
+                  }
+                )
+              ] }) })
+            ),
             belowSurface: /* @__PURE__ */ jsx(ComposerKeyboardHint, { visible: keyboardHint.visible })
           }
         )
       }
     )
   ] });
+  return /* @__PURE__ */ jsx(AthenaChatComposerActionsContext.Provider, { value: composerActions, children: composer });
 };
 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" }),
@@ -1144,6 +1335,7 @@ const AthenaAssistantMessage = ({
               ] }) }),
               /* @__PURE__ */ jsx(MessageError, {})
             ] }) }),
+            /* @__PURE__ */ jsx(AthenaTurnChangesSlot, { className: "mx-2" }),
             /* @__PURE__ */ jsxs("div", { className: "aui-assistant-message-footer mt-1 ml-2 flex items-center gap-1", children: [
               /* @__PURE__ */ jsx(BranchPicker, {}),
               /* @__PURE__ */ jsx(ActionBarComponent, {}),
@@ -1311,6 +1503,7 @@ const AthenaUserMessage = ({
         /* @__PURE__ */ jsx(MessageDaySeparator, {}),
         /* @__PURE__ */ jsxs("div", { className: "flex min-w-0 max-w-[75%] flex-col items-end gap-1", children: [
           /* @__PURE__ */ jsx(InboundChannelLine, {}),
+          /* @__PURE__ */ jsx(MessageAttachedReferences, {}),
           /* @__PURE__ */ jsxs("div", { className: "aui-user-message-content wrap-break-word min-w-0 max-w-full rounded-2xl bg-muted/70 px-4 py-2.5 text-foreground", children: [
             /* @__PURE__ */ jsx(InboundChannelSubject, {}),
             /* @__PURE__ */ jsx(MessageImageGallery, { children: /* @__PURE__ */ jsx(MessagePrimitive.Parts, { components: partComponents }) })