@athenaintel/react 0.28.1 → 0.30.0

package/dist/chat/AthenaChat.js

@@ -4,13 +4,15 @@ import { AthenaChatErrorBoundary } from "./AthenaChatErrorBoundary.js";
 import { StatewireApprovalCard } from "./StatewireApprovalCard.js";
 import { StatewireQueuedMessages } from "./StatewireQueuedMessages.js";
 import { StatewireConnectionBanner, StatewireErrorBanner } from "./statewire-banners.js";
+import { StatewireRunRefusalNotice } from "./StatewireRunRefusalNotice.js";
 import { WaitingOnYouBanner } from "./WaitingOnYouBanner.js";
 import { WakeMarkerMessage } from "../chat-ui/messages/wake-marker.js";
 import { ToolFallback, ToolFallbackAppearanceProvider } 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 { SuperGroupingFinalText } from "./super-grouping/SuperGroupingCard.js";
+import { SuperGroupingTranscript } from "./super-grouping/SuperGroupingTranscript.js";
 import { NestedPtcComponentsProvider, useNestedPtcComponents, NestedPtcMessages } from "../tools/NestedPtcMessages.js";
 import { TooltipIconButton } from "./tooltip-icon-button.js";
 import { Button } from "../ui/button.js";
@@ -21,6 +23,7 @@ import { StatewireHistoryEdge } from "../chat-ui/thread/statewire-history-edge.j
 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";
+import { prepareComposerSend } from "../chat-ui/composer/composer-message.js";
 import { athenaDiagnostics } from "../diagnostics/athena-diagnostics.js";
 import { useDictation } from "../dictation/use-dictation.js";
 import { DictationStrip } from "../dictation/DictationStrip.js";
@@ -29,7 +32,7 @@ import { composeDictationDraft, DICTATION_STATUSES } from "../dictation/dictatio
 import { useAthenaDictationEngine } from "../dictation/use-athena-dictation-engine.js";
 import { TiptapComposer } from "../chat-ui/composer/tiptap-composer.js";
 import { MentionAuthCtx } from "../chat-ui/mentions/suggestions/data/api-client.js";
-import { ACTION_BAR_REVEAL_CLASS, BranchPicker } from "../chat-ui/messages/message-components.js";
+import { ACTION_BAR_REVEAL_CLASS, BranchPicker, UserMessageText } from "../chat-ui/messages/message-components.js";
 import { useMessageActionsReadOnly } from "../chat-ui/messages/message-actions-config.js";
 import { useAthenaRunLive } from "./use-athena-run-live.js";
 import { DEFAULT_WELCOME_SUGGESTIONS, AthenaChatWelcomeSuggestions, AthenaChatWelcome, welcomeEntrance, welcomeComposerDelayMs } from "./athena-chat-welcome.js";
@@ -40,26 +43,31 @@ import { useAuiState, ActionBarPrimitive, ActionBarMorePrimitive, MessagePrimiti
 import { MoreHorizontalIcon, DownloadIcon, ExternalLinkIcon, BrainIcon, ChevronRightIcon, CheckIcon, CopyIcon, RefreshCwIcon, AlertCircleIcon, ArrowUpIcon, PencilIcon } from "lucide-react";
 import { useMemo, useContext, useState, useRef, useEffect, createContext, useCallback } from "react";
 import { useAthenaThreadId, AthenaStatewireThreadIdContext, useAthenaStatewireThreadId } from "../threads/thread-id-context.js";
+import { useAthenaStatewireThreadList } from "../threads/statewire-thread-list.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 { buildComposedSend } 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 { getToolkitDisplayInfo } from "../toolkits/index.js";
-import { runActivityKey, runActivityVerb, RUN_ACTIVITY_COUNTER_SECONDS, runActivityLabel, RUN_ACTIVITY_STALL_SECONDS } from "./run-activity.js";
+import { runActivityKey, runActivityVerb, runActivityLabel, runElapsedLabel, RUN_ACTIVITY_STALL_SECONDS } from "./run-activity.js";
 import { readMessageCreatedAt, parseMessageTime, isSameCalendarDay, formatDayLabel, formatMessageTime, formatMessageDateTime } from "./message-time.js";
 import { ThreadScrollToBottom } from "./thread-scroll-to-bottom.js";
-import { isGithubWakeMessage } from "../packages/agent-runtime/src/github-wake-event.js";
+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 { cn } from "cnfast";
+const DEFAULT_COMPOSER_PLACEHOLDER = "Type @ to tag an asset";
 function QuotePostMessageBridge() {
-  useQuoteFromPostMessage();
-  return null;
+  const marker = useRef(null);
+  useQuoteFromPostMessage({ scope: marker });
+  return /* @__PURE__ */ jsx("span", { ref: marker, hidden: true });
 }
 const AthenaChatDefaultComponentsContext = createContext(null);
 const AthenaReasoningTextComponentContext = createContext(null);
@@ -94,8 +102,8 @@ const AthenaDefaultAssistantMessage = () => {
   );
 };
 const AthenaDefaultUserMessage = () => {
-  const { TextComponent } = useAthenaChatDefaultComponents();
-  return /* @__PURE__ */ jsx(AthenaUserMessage, { TextComponent });
+  const { UserTextComponent } = useAthenaChatDefaultComponents();
+  return /* @__PURE__ */ jsx(AthenaUserMessage, { TextComponent: UserTextComponent });
 };
 const getReasoningTokensFromMetadata = (metadata) => {
   if (!isRecord(metadata)) {
@@ -138,9 +146,10 @@ const AthenaChat = ({
   composer: composerSlots,
   dictation = true
 }) => {
-  var _a, _b, _c;
+  var _a, _b, _c, _d, _e;
   const athenaConfig = useAthenaConfig();
   const statewireThreadId = useContext(AthenaStatewireThreadIdContext);
+  const startNewSession = (_a = useAthenaStatewireThreadList()) == null ? void 0 : _a.newThread;
   const mentionAuth = useMemo(
     () => ({
       backendUrl: athenaConfig.backendUrl,
@@ -185,15 +194,17 @@ const AthenaChat = ({
     }),
     [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 textComponent = ((_b = components == null ? void 0 : components.messageParts) == null ? void 0 : _b.Text) ?? TiptapText;
+  const userTextComponent = ((_c = components == null ? void 0 : components.messageParts) == null ? void 0 : _c.Text) ?? UserMessageText;
+  const reasoningComponent = (_d = components == null ? void 0 : components.messageParts) == null ? void 0 : _d.Reasoning;
+  const emptyComponent = ((_e = components == null ? void 0 : components.messageParts) == null ? void 0 : _e.Empty) ?? AthenaAssistantMessageEmpty;
   const actionBarComponent = (components == null ? void 0 : components.AssistantActionBar) ?? AthenaAssistantActionBar;
   const avatarComponent = components == null ? void 0 : components.AssistantAvatar;
   const defaultComponentsContextValue = useMemo(
     () => ({
       toolUIs: mergedToolUIs,
       TextComponent: textComponent,
+      UserTextComponent: userTextComponent,
       ReasoningComponent: reasoningComponent,
       EmptyComponent: emptyComponent,
       ActionBarComponent: actionBarComponent,
@@ -207,7 +218,8 @@ const AthenaChat = ({
       groupToolCalls,
       mergedToolUIs,
       reasoningComponent,
-      textComponent
+      textComponent,
+      userTextComponent
     ]
   );
   const AssistantMessageComponent = (components == null ? void 0 : components.AssistantMessage) ?? AthenaDefaultAssistantMessage;
@@ -243,7 +255,8 @@ const AthenaChat = ({
       viewportFooter: /* @__PURE__ */ jsxs(Fragment, { children: [
         welcomeSuggestions.length > 0 && /* @__PURE__ */ jsx(ThreadPrimitive.Empty, { children: /* @__PURE__ */ jsx(AthenaChatWelcomeSuggestions, { suggestions: welcomeSuggestions }) }),
         /* @__PURE__ */ jsx(StatewireConnectionBanner, {}),
-        /* @__PURE__ */ jsx(StatewireErrorBanner, {}),
+        /* @__PURE__ */ jsx(StatewireErrorBanner, { onStartNewSession: startNewSession }),
+        /* @__PURE__ */ jsx(StatewireRunRefusalNotice, { onStartNewSession: startNewSession }),
         /* @__PURE__ */ jsx(StatewireApprovalCard, {}),
         /* @__PURE__ */ jsx(MentionAuthCtx.Provider, { value: mentionAuth, children: /* @__PURE__ */ jsx(StatewireQueuedMessages, {}) }),
         /* @__PURE__ */ jsx(AttachmentProvider, { children: /* @__PURE__ */ jsxs(QuoteProvider, { children: [
@@ -325,6 +338,11 @@ const AthenaChatComposer = ({ tools, slots, dictation, welcomeSuggestionCount })
     var _a;
     return ((_a = state.optional.thread) == null ? void 0 : _a.isRunning) ?? false;
   });
+  const connection = useDeepAgentConnection();
+  const runLive = useAuiState((state) => {
+    const thread = state.optional.thread;
+    return thread ? readRunLive(thread, connection) : false;
+  });
   const [dragActive, setDragActive] = useState(false);
   const keyboardHint = useComposerKeyboardHint();
   const reducedMotion = usePrefersReducedMotion();
@@ -339,6 +357,11 @@ const AthenaChatComposer = ({ tools, slots, dictation, welcomeSuggestionCount })
   const appendSpokenText = useCallback(
     (spoken) => {
       if (!spoken) return;
+      const composer = composerRef.current;
+      if (composer == null ? void 0 : composer.appendText) {
+        composer.appendText(spoken);
+        return;
+      }
       aui.composer.setText(composeDictationDraft(aui.composer.getState().text ?? "", spoken));
     },
     [aui]
@@ -384,24 +407,25 @@ const AthenaChatComposer = ({ tools, slots, dictation, welcomeSuggestionCount })
     });
   }, []);
   const sendComposed = useCallback(
-    (userText) => {
-      const message = buildComposedMessage({
+    (editorSend) => {
+      const send = buildComposedSend({
         attachments: attachmentsRef.current,
         quote: quoteRef.current,
-        userText,
+        send: editorSend,
         appUrl
       });
-      if (!message) return false;
-      aui.composer.setText(message);
+      if (!send) return false;
+      prepareComposerSend({ threadId, ...send });
+      aui.composer.setText(send.text);
       aui.composer.send({ steer: false });
       clearAttachments();
       clearQuote();
       return true;
     },
-    [appUrl, aui, clearAttachments, clearQuote]
+    [appUrl, aui, clearAttachments, clearQuote, threadId]
   );
   const interceptSubmit = useCallback(
-    (text) => attachmentsRef.current.length > 0 || quoteRef.current !== null ? sendComposed(text) : false,
+    (text, draftMarkdown) => attachmentsRef.current.length > 0 || quoteRef.current !== null ? sendComposed({ text, draftMarkdown }) : false,
     [sendComposed]
   );
   const fileIntake = useMemo(
@@ -416,20 +440,20 @@ const AthenaChatComposer = ({ tools, slots, dictation, welcomeSuggestionCount })
     appendSpokenText((await dictationCtl.stop()).text);
   }, [appendSpokenText, dictationCtl]);
   const sendDictation = useCallback(async () => {
-    var _a;
+    var _a, _b, _c;
     const { text: spoken, error } = await dictationCtl.stop();
-    const text = composeDictationDraft(aui.composer.getState().text ?? "", spoken);
-    if (error) {
-      aui.composer.setText(text);
-      return;
-    }
+    appendSpokenText(spoken);
+    if (error) return;
+    const send = ((_b = (_a = composerRef.current) == null ? void 0 : _a.captureSend) == null ? void 0 : _b.call(_a)) ?? {
+      text: aui.composer.getState().text ?? "",
+      draftMarkdown: aui.composer.getState().text ?? ""
+    };
     const extras = attachmentsRef.current.length > 0 || quoteRef.current !== null;
-    if (!text && !extras) return;
-    aui.composer.setText(text);
+    if (!send.text && !extras) return;
     const action = resolveComposerAction({ hasDraft: true, isRunning });
     if (action === "stop") return;
     try {
-      await ((_a = slots == null ? void 0 : slots.onBeforeSend) == null ? void 0 : _a.call(slots, { action }));
+      await ((_c = slots == null ? void 0 : slots.onBeforeSend) == null ? void 0 : _c.call(slots, { action }));
     } catch (error2) {
       athenaDiagnostics.emit("composer.dictation", {
         level: "warn",
@@ -441,11 +465,13 @@ const AthenaChatComposer = ({ tools, slots, dictation, welcomeSuggestionCount })
       return;
     }
     if (extras) {
-      sendComposed(text);
+      sendComposed(send);
       return;
     }
+    prepareComposerSend({ threadId, ...send });
+    aui.composer.setText(send.text);
     aui.composer.send({ steer: false });
-  }, [aui, dictationCtl, isRunning, sendComposed, slots]);
+  }, [appendSpokenText, aui, dictationCtl, isRunning, sendComposed, slots, threadId]);
   return /* @__PURE__ */ jsx(ComposerDropZone, { onDragActiveChange: setDragActive, children: /* @__PURE__ */ jsx(
     "div",
     {
@@ -469,7 +495,7 @@ const AthenaChatComposer = ({ tools, slots, dictation, welcomeSuggestionCount })
                 ref: composerRef,
                 threadId,
                 appearance: "paper",
-                placeholder: "Type @ to tag an asset",
+                placeholder: (runLive ? slots == null ? void 0 : slots.runningPlaceholder : void 0) ?? (slots == null ? void 0 : slots.placeholder) ?? DEFAULT_COMPOSER_PLACEHOLDER,
                 tools,
                 rootCategories,
                 onDraftStateChange: setHasText,
@@ -553,7 +579,7 @@ const MessageError = () => /* @__PURE__ */ jsx(MessagePrimitive.Error, { childre
     }
   ) })
 ] }) });
-const RunStatusLine = ({ verb, resetKey, counterOnly = false, className }) => {
+function useSecondsSince(resetKey) {
   const [seconds, setSeconds] = useState(0);
   useEffect(() => {
     setSeconds(0);
@@ -563,8 +589,11 @@ const RunStatusLine = ({ verb, resetKey, counterOnly = false, className }) => {
     }, 1e3);
     return () => clearInterval(timer);
   }, [resetKey]);
+  return seconds;
+}
+const RunStatusLine = ({ verb, resetKey, className }) => {
+  const seconds = useSecondsSince(resetKey);
   const stalled = seconds >= RUN_ACTIVITY_STALL_SECONDS;
-  if (counterOnly && seconds < RUN_ACTIVITY_COUNTER_SECONDS) return null;
   return /* @__PURE__ */ jsxs(
     "div",
     {
@@ -625,13 +654,22 @@ const AthenaMessageActivityIndicator = () => {
 };
 const AthenaGroupedRunMeta = () => {
   const activityKey = useAuiState((s) => runActivityKey(s.message.parts));
-  return /* @__PURE__ */ jsx(
-    RunStatusLine,
+  const seconds = useSecondsSince(activityKey ?? "thinking");
+  const stalled = seconds >= RUN_ACTIVITY_STALL_SECONDS;
+  if (!stalled) return null;
+  return /* @__PURE__ */ jsxs(
+    "span",
     {
-      verb: "Working",
-      resetKey: activityKey ?? "thinking",
-      counterOnly: true,
-      className: "mt-1 px-0"
+      "data-slot": "run-stalled",
+      "data-stalled": "true",
+      className: "tabular-nums text-amber-600 dark:text-amber-400",
+      children: [
+        /* @__PURE__ */ jsxs("span", { "aria-hidden": "true", children: [
+          "Nothing new for ",
+          runElapsedLabel(seconds)
+        ] }),
+        /* @__PURE__ */ jsx("span", { "aria-live": "polite", className: "sr-only", children: "Assistant is still working; nothing new for over thirty seconds." })
+      ]
     }
   );
 };
@@ -816,7 +854,7 @@ const AthenaAssistantMessage = ({
             /* @__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,
+                  SuperGroupingTranscript,
                   {
                     toolUIs: toolUIsWithNestedMessages,
                     TextComponent,
@@ -957,12 +995,14 @@ const AthenaAssistantActionBar = ({ className }) => {
   );
 };
 const AthenaUserMessage = ({
-  TextComponent = TiptapText
+  TextComponent = UserMessageText
 }) => {
-  const isWake = useAuiState((s) => {
-    var _a;
-    return isGithubWakeMessage((_a = s.message.metadata) == null ? void 0 : _a.custom);
-  });
+  const isWake = useAuiState(
+    (s) => {
+      var _a, _b;
+      return isGithubWakeMessage((_a = s.message.metadata) == null ? void 0 : _a.custom) || isSubscriptionWakeMessage((_b = s.message.metadata) == null ? void 0 : _b.custom);
+    }
+  );
   if (isWake) {
     return /* @__PURE__ */ jsxs(
       MessagePrimitive.Root,