@athenaintel/react 0.41.3 → 0.42.1

package/dist/chat/AthenaChat.js

@@ -12,9 +12,12 @@ 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 { isSdkInlineTool } from "../tools/inline-tools.js";
+import { SubagentTaskToolUI } from "../tools/subagent-task-tool-ui.js";
 import { TOOL_UI_REGISTRY } from "../tools/tool-uis.js";
 import { SuperGroupingFinalText } from "./super-grouping/SuperGroupingCard.js";
 import { SuperGroupingTranscript } from "./super-grouping/SuperGroupingTranscript.js";
+import { useApprovalWaitInScope, StatewireApprovalWaitProvider } from "./use-statewire-approval-request.js";
 import { NestedPtcComponentsProvider, useNestedPtcComponents, NestedPtcMessages } from "../tools/NestedPtcMessages.js";
 import { TooltipIconButton } from "./tooltip-icon-button.js";
 import { Button } from "../ui/button.js";
@@ -42,6 +45,7 @@ import { DEFAULT_WELCOME_SUGGESTIONS, AthenaChatWelcomeSuggestions, AthenaChatWe
 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 { useStatewireSessionLane } from "../runtime/statewire-session-lane.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";
@@ -55,7 +59,7 @@ import { ComposerAttachmentPreview } from "../composer/ComposerAttachmentPreview
 import { ComposerDropZone } from "../composer/ComposerDropZone.js";
 import { QuoteProvider, useQuoteFromPostMessage, useQuote } from "../composer/QuoteContext.js";
 import { AttachmentProvider, useAttachments } from "../composer/AttachmentContext.js";
-import { buildComposedSend } from "../composer/build-composed-message.js";
+import { buildComposedSend, holdComposedSend } 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";
@@ -64,9 +68,13 @@ 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 { TASK_TOOL_NAME } from "../agent-runtime/src/tool-names.js";
+import { useTurnToolCallsAwaitingApproval } from "../agent-runtime/src/subagents.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 { 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";
 const DEFAULT_COMPOSER_PLACEHOLDER = "Type @ to tag an asset";
 function QuotePostMessageBridge() {
@@ -249,7 +257,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(MentionAuthCtx.Provider, { value: mentionAuth, children: /* @__PURE__ */ jsx(
+  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(
     ChatThreadShell,
     {
       overlay: /* @__PURE__ */ jsxs(Fragment, { children: [
@@ -297,7 +305,7 @@ const AthenaChat = ({
         ] }) })
       ] })
     }
-  ) }) }) }, statewireThreadId ?? void 0) });
+  ) }) }) }) }, statewireThreadId ?? void 0) });
 };
 const THREAD_VIEWPORT_ATTR = "data-aui-thread-viewport";
 const useScrollToTop = () => {
@@ -353,11 +361,22 @@ const AthenaChatComposer = ({ tools, slots, dictation, welcomeSuggestionCount })
   const aui = useAui();
   const threadId = useAthenaStatewireThreadId();
   const { appUrl } = useAthenaConfig();
-  const { attachments, clearAttachments, isUploading } = useAttachments();
+  const { attachments, clearAttachments, isUploading, notice: attachNotice, setNotice: setAttachNotice } = useAttachments();
   const { quote, clearQuote } = useQuote();
   const { attachFiles } = useAttachFiles();
+  const sessionLane = useStatewireSessionLane();
   const composerRef = useRef(null);
   const [hasText, setHasText] = useState(false);
+  const [composerNotice, setComposerNotice] = useState(null);
+  const onDraftStateChange = useCallback(
+    (next) => {
+      setHasText(next);
+      setComposerNotice(null);
+      setAttachNotice(null);
+    },
+    [setAttachNotice]
+  );
+  const notice = composerNotice ?? attachNotice;
   const isRunning = useAuiState((state) => {
     var _a;
     return ((_a = state.optional.thread) == null ? void 0 : _a.isRunning) ?? false;
@@ -452,6 +471,31 @@ const AthenaChatComposer = ({ tools, slots, dictation, welcomeSuggestionCount })
     (text, draftMarkdown) => attachmentsRef.current.length > 0 || quoteRef.current !== null ? sendComposed({ text, draftMarkdown }) : false,
     [sendComposed]
   );
+  const hasPendingExtras = useCallback(
+    () => attachmentsRef.current.length > 0 || quoteRef.current !== null,
+    []
+  );
+  const holdPendingExtras = useCallback(
+    ({
+      outbox,
+      send,
+      attachments: attachments2
+    }) => {
+      const held = holdComposedSend({
+        outbox,
+        attachments: attachmentsRef.current,
+        carried: attachments2,
+        quote: quoteRef.current,
+        send,
+        appUrl
+      });
+      if (!held) return false;
+      clearAttachments();
+      clearQuote();
+      return true;
+    },
+    [appUrl, clearAttachments, clearQuote]
+  );
   const fileIntake = useMemo(
     () => ({
       onPaste: (files) => {
@@ -468,6 +512,7 @@ const AthenaChatComposer = ({ tools, slots, dictation, welcomeSuggestionCount })
     const { text: spoken, error } = await dictationCtl.stop();
     appendSpokenText(spoken);
     if (error) return;
+    if (sessionLane == null ? void 0 : sessionLane.isPending()) await sessionLane.whenKnown();
     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 ?? ""
@@ -495,7 +540,7 @@ const AthenaChatComposer = ({ tools, slots, dictation, welcomeSuggestionCount })
     prepareComposerSend({ threadId, ...send });
     aui.composer.setText(send.text);
     aui.composer.send({ steer: false });
-  }, [appendSpokenText, aui, dictationCtl, isRunning, sendComposed, slots, threadId]);
+  }, [appendSpokenText, aui, dictationCtl, isRunning, sendComposed, sessionLane, slots, threadId]);
   return /* @__PURE__ */ jsx(ComposerDropZone, { onDragActiveChange: setDragActive, children: /* @__PURE__ */ jsx(
     "div",
     {
@@ -522,13 +567,17 @@ const AthenaChatComposer = ({ tools, slots, dictation, welcomeSuggestionCount })
                 placeholder: (runLive ? slots == null ? void 0 : slots.runningPlaceholder : void 0) ?? (slots == null ? void 0 : slots.placeholder) ?? DEFAULT_COMPOSER_PLACEHOLDER,
                 tools,
                 rootCategories,
-                onDraftStateChange: setHasText,
+                onDraftStateChange,
                 waitForHostUploads,
                 interceptSubmit,
+                hasPendingExtras,
+                holdPendingExtras,
+                onAttachmentError: setComposerNotice,
                 fileIntake,
                 onSubmitted: keyboardHint.onSubmitted
               }
             ) }),
+            notice && /* @__PURE__ */ jsx("p", { className: "aui-composer-notice px-1.5 pb-1 text-destructive text-xs", role: "alert", children: notice }),
             recording && /* @__PURE__ */ jsx(
               DictationStrip,
               {
@@ -840,6 +889,7 @@ const AthenaReasoningPart = ({
   );
 };
 const noToolGrouping = () => null;
+const isInlineToolOrSubagent = (toolName) => toolName === TASK_TOOL_NAME || isSdkInlineTool(toolName);
 const AthenaAssistantMessage = ({
   toolUIs,
   AvatarComponent,
@@ -857,6 +907,8 @@ const AthenaAssistantMessage = ({
     }
     return wrappedToolUIs;
   }, [toolUIs]);
+  const isInlineTool = toolUIs[TASK_TOOL_NAME] === SubagentTaskToolUI ? isInlineToolOrSubagent : isSdkInlineTool;
+  const waitingToolCallIds = useTurnToolCallsAwaitingApproval(useApprovalWaitInScope().toolCallIds);
   const nestedPtcComponents = useMemo(
     () => ({
       Text: TextComponent,
@@ -894,6 +946,8 @@ const AthenaAssistantMessage = ({
                     toolUIs: toolUIsWithNestedMessages,
                     TextComponent,
                     ReasoningComponent: effectiveReasoningComponent,
+                    isInlineTool,
+                    waitingToolCallIds,
                     renderRunningMeta: renderGroupedRunningMeta
                   }
                 ),
@@ -951,7 +1005,20 @@ const EditMessageButton = () => {
   const allowEdit = useAuiState((s) => s.thread.capabilities.edit);
   const isRunning = useAthenaRunLive();
   const readOnly = useMessageActionsReadOnly();
+  const reachable = useChatOutboxReachable(useDeepAgentChatOutboxContext());
   if (!allowEdit || isRunning || readOnly) return null;
+  if (!reachable) {
+    return /* @__PURE__ */ jsx(
+      TooltipIconButton,
+      {
+        tooltip: CHAT_EDIT_UNREACHABLE_LABEL,
+        "aria-disabled": true,
+        "data-testid": "chat-message-edit-unreachable",
+        className: "cursor-not-allowed opacity-50",
+        children: /* @__PURE__ */ jsx(PencilIcon, {})
+      }
+    );
+  }
   return /* @__PURE__ */ jsx(ActionBarPrimitive.Edit, { asChild: true, children: /* @__PURE__ */ jsx(TooltipIconButton, { tooltip: "Edit", children: /* @__PURE__ */ jsx(PencilIcon, {}) }) });
 };
 const AthenaUserActionBar = () => {