@athenaintel/react 0.28.1 → 0.32.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,37 +32,44 @@ 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 { useAccountWorkspaceId } from "../chat-ui/mentions/use-account-workspace-id.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";
 import { AthenaComposerSendButton, ATHENA_COMPOSER_COMPONENTS, AthenaComposerMentionButton } from "./athena-composer-controls.js";
 import { useComposerKeyboardHint, ComposerKeyboardHint } from "./composer-keyboard-hint.js";
 import { usePrefersReducedMotion } from "../lib/use-prefers-reduced-motion.js";
+import { AthenaAuthContext } from "../auth/AthenaAuthContext.js";
 import { useAuiState, ActionBarPrimitive, ActionBarMorePrimitive, MessagePrimitive, ThreadPrimitive, AuiIf, ErrorPrimitive, useAui } from "@assistant-ui/react";
 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 +104,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,28 +148,42 @@ const AthenaChat = ({
   composer: composerSlots,
   dictation = true
 }) => {
-  var _a, _b, _c;
+  var _a, _b, _c, _d, _e, _f;
   const athenaConfig = useAthenaConfig();
+  const authState = useContext(AthenaAuthContext);
   const statewireThreadId = useContext(AthenaStatewireThreadIdContext);
+  const startNewSession = (_a = useAthenaStatewireThreadList()) == null ? void 0 : _a.newThread;
+  const mentionCredentials = useMemo(
+    () => {
+      var _a2;
+      return {
+        backendUrl: athenaConfig.backendUrl,
+        appUrl: athenaConfig.appUrl,
+        apiKey: athenaConfig.apiKey,
+        credentials: "include",
+        userId: (_a2 = authState == null ? void 0 : authState.user) == null ? void 0 : _a2.userId,
+        get accessToken() {
+          return athenaConfig.token ?? void 0;
+        },
+        getAccessToken: async () => {
+          var _a3;
+          return await ((_a3 = athenaConfig.getToken) == null ? void 0 : _a3.call(athenaConfig)) ?? void 0;
+        },
+        renewAccessToken: async () => {
+          var _a3;
+          return await ((_a3 = athenaConfig.getToken) == null ? void 0 : _a3.call(athenaConfig, { force: true })) ?? void 0;
+        }
+      };
+    },
+    [athenaConfig, (_b = authState == null ? void 0 : authState.user) == null ? void 0 : _b.userId]
+  );
+  const accountWorkspaceId = useAccountWorkspaceId(mentionCredentials);
   const mentionAuth = useMemo(
     () => ({
-      backendUrl: athenaConfig.backendUrl,
-      appUrl: athenaConfig.appUrl,
-      apiKey: athenaConfig.apiKey,
-      credentials: "include",
-      get accessToken() {
-        return athenaConfig.token ?? void 0;
-      },
-      getAccessToken: async () => {
-        var _a2;
-        return await ((_a2 = athenaConfig.getToken) == null ? void 0 : _a2.call(athenaConfig)) ?? void 0;
-      },
-      renewAccessToken: async () => {
-        var _a2;
-        return await ((_a2 = athenaConfig.getToken) == null ? void 0 : _a2.call(athenaConfig, { force: true })) ?? void 0;
-      }
+      ...mentionCredentials,
+      workspaceId: mentionCredentials.workspaceId ?? accountWorkspaceId
     }),
-    [athenaConfig]
+    [mentionCredentials, accountWorkspaceId]
   );
   const providerMentionTools = useMemo(() => {
     const uniqueToolkitIds = Array.from(new Set(athenaConfig.enabledToolkits));
@@ -185,15 +209,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 = ((_c = components == null ? void 0 : components.messageParts) == null ? void 0 : _c.Text) ?? TiptapText;
+  const userTextComponent = ((_d = components == null ? void 0 : components.messageParts) == null ? void 0 : _d.Text) ?? UserMessageText;
+  const reasoningComponent = (_e = components == null ? void 0 : components.messageParts) == null ? void 0 : _e.Reasoning;
+  const emptyComponent = ((_f = components == null ? void 0 : components.messageParts) == null ? void 0 : _f.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 +233,8 @@ const AthenaChat = ({
       groupToolCalls,
       mergedToolUIs,
       reasoningComponent,
-      textComponent
+      textComponent,
+      userTextComponent
     ]
   );
   const AssistantMessageComponent = (components == null ? void 0 : components.AssistantMessage) ?? AthenaDefaultAssistantMessage;
@@ -219,7 +246,7 @@ const AthenaChat = ({
     }),
     [AssistantMessageComponent, UserMessageComponent]
   );
-  return /* @__PURE__ */ jsx(AthenaChatDefaultComponentsContext.Provider, { value: defaultComponentsContextValue, children: /* @__PURE__ */ jsx(AthenaChatErrorBoundary, { children: /* @__PURE__ */ jsx(ToolFallbackAppearanceProvider, { appearance: "thread", children: /* @__PURE__ */ jsx(
+  return /* @__PURE__ */ jsx(AthenaChatDefaultComponentsContext.Provider, { value: defaultComponentsContextValue, children: /* @__PURE__ */ jsx(AthenaChatErrorBoundary, { children: /* @__PURE__ */ jsx(ToolFallbackAppearanceProvider, { appearance: "thread", children: /* @__PURE__ */ jsx(MentionAuthCtx.Provider, { value: mentionAuth, children: /* @__PURE__ */ jsx(
     ChatThreadShell,
     {
       overlay: /* @__PURE__ */ jsx(ThreadLoadingOverlay, {}),
@@ -243,12 +270,13 @@ 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(StatewireQueuedMessages, {}),
         /* @__PURE__ */ jsx(AttachmentProvider, { children: /* @__PURE__ */ jsxs(QuoteProvider, { children: [
           /* @__PURE__ */ jsx(QuotePostMessageBridge, {}),
-          /* @__PURE__ */ jsx(MentionAuthCtx.Provider, { value: mentionAuth, children: /* @__PURE__ */ jsx(
+          /* @__PURE__ */ jsx(
             AthenaChatComposer,
             {
               tools,
@@ -256,11 +284,11 @@ const AthenaChat = ({
               dictation,
               welcomeSuggestionCount: welcomeSuggestions.length
             }
-          ) })
+          )
         ] }) })
       ] })
     }
-  ) }) }, statewireThreadId ?? void 0) });
+  ) }) }) }, statewireThreadId ?? void 0) });
 };
 const THREAD_VIEWPORT_ATTR = "data-aui-thread-viewport";
 const useScrollToTop = () => {
@@ -325,6 +353,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 +372,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 +422,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 +455,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 +480,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 +510,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 +594,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 +604,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 +669,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 +869,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 +1010,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,