@athenaintel/react 0.42.2 → 0.48.0

package/dist/chat-ui/thread/chat-thread-shell.cjs

@@ -1,19 +1,46 @@
 "use strict";
 Object.defineProperty(exports, Symbol.toStringTag, { value: "Module" });
 const jsxRuntime = require("react/jsx-runtime");
-const react$1 = require("@assistant-ui/react");
+const react = require("@assistant-ui/react");
 const lucideReact = require("lucide-react");
-const react = require("react");
+const require$$0 = require("react");
 const messageComponents = require("../messages/message-components.cjs");
+const answerLabel = require("../messages/answer-label.cjs");
 const messageActionsConfig = require("../messages/message-actions-config.cjs");
 const userEditComposer = require("../messages/user-edit-composer.cjs");
 const runningIndicator = require("../messages/running-indicator.cjs");
+const handoffArrivalDivider = require("../messages/handoff-arrival-divider.cjs");
 const threadChrome = require("../messages/thread-chrome.cjs");
+const messages = require("../../agent-runtime/src/messages.cjs");
+const withArrivalDivider = /* @__PURE__ */ (() => {
+  const wrapped = /* @__PURE__ */ new WeakMap();
+  return (Message) => {
+    const existing = wrapped.get(Message);
+    if (existing) return existing;
+    const WithArrivalDivider = () => /* @__PURE__ */ jsxRuntime.jsxs(jsxRuntime.Fragment, { children: [
+      /* @__PURE__ */ jsxRuntime.jsx(Message, {}),
+      /* @__PURE__ */ jsxRuntime.jsx(handoffArrivalDivider.HandoffArrivalAfterMessage, {})
+    ] });
+    wrapped.set(Message, WithArrivalDivider);
+    return WithArrivalDivider;
+  };
+})();
+const ROW_KEY_SEPARATOR = "\0";
+const ThreadRows = ({ components }) => {
+  const joinedKeys = react.useAuiState(
+    (s) => s.thread.messages.map((message) => messages.deepAgentRowKey(message)).join(ROW_KEY_SEPARATOR)
+  );
+  const rowKeys = require$$0.useMemo(
+    () => joinedKeys === "" ? [] : joinedKeys.split(ROW_KEY_SEPARATOR),
+    [joinedKeys]
+  );
+  return /* @__PURE__ */ jsxRuntime.jsx(jsxRuntime.Fragment, { children: rowKeys.map((key, index) => /* @__PURE__ */ jsxRuntime.jsx(react.ThreadPrimitive.MessageByIndex, { index, components }, key)) });
+};
 const DEFAULT_VIEWPORT_CLASS = "relative flex flex-1 flex-col overflow-y-auto scroll-smooth";
 const DEFAULT_CONTENT_CLASS = "w-full flex-1 px-3 pt-3";
 const DEFAULT_VIEWPORT_FOOTER_CLASS = "sticky bottom-0";
-const DefaultScrollToBottom = () => /* @__PURE__ */ jsxRuntime.jsx(react$1.AuiIf, { condition: (state) => !state.thread.isEmpty, children: /* @__PURE__ */ jsxRuntime.jsx("div", { className: "flex justify-center pb-0.5", children: /* @__PURE__ */ jsxRuntime.jsx(
-  react$1.ThreadPrimitive.ScrollToBottom,
+const DefaultScrollToBottom = () => /* @__PURE__ */ jsxRuntime.jsx(react.AuiIf, { condition: (state) => !state.thread.isEmpty, children: /* @__PURE__ */ jsxRuntime.jsx("div", { className: "flex justify-center pb-0.5", children: /* @__PURE__ */ jsxRuntime.jsx(
+  react.ThreadPrimitive.ScrollToBottom,
   {
     "aria-label": "Scroll to bottom",
     title: "Scroll to bottom",
@@ -41,24 +68,26 @@ function ChatThreadShell({
   scrollToBottom = /* @__PURE__ */ jsxRuntime.jsx(DefaultScrollToBottom, {}),
   viewportFooter,
   viewportFooterClassName,
-  readOnly = false
+  readOnly = false,
+  arrivalThreadId = null,
+  viewerUserId = null
 }) {
   const {
     UserMessage,
     AssistantMessage,
     UserEditComposer: UserEditComposerComponent = userEditComposer.UserEditComposer
   } = components;
-  const messageComponents2 = react.useMemo(
+  const messageComponents2 = require$$0.useMemo(
     () => ({
-      UserMessage,
-      AssistantMessage,
+      UserMessage: withArrivalDivider(UserMessage),
+      AssistantMessage: withArrivalDivider(AssistantMessage),
       UserEditComposer: UserEditComposerComponent
     }),
     [UserMessage, AssistantMessage, UserEditComposerComponent]
   );
-  const messages = /* @__PURE__ */ jsxRuntime.jsx(react$1.ThreadPrimitive.Messages, { components: messageComponents2 });
+  const messages2 = /* @__PURE__ */ jsxRuntime.jsx(handoffArrivalDivider.HandoffArrivalScope, { threadId: arrivalThreadId, children: /* @__PURE__ */ jsxRuntime.jsx(answerLabel.AnswerLabelScope, { viewerUserId, children: /* @__PURE__ */ jsxRuntime.jsx(ThreadRows, { components: messageComponents2 }) }) });
   return /* @__PURE__ */ jsxRuntime.jsx(messageActionsConfig.MessageActionsReadOnlyProvider, { readOnly, children: /* @__PURE__ */ jsxRuntime.jsxs(
-    react$1.ThreadPrimitive.Root,
+    react.ThreadPrimitive.Root,
     {
       className: panelClassName ?? threadChrome.THREAD_PANEL_CLASS,
       style: { "--thread-max-width": maxWidth },
@@ -67,7 +96,7 @@ function ChatThreadShell({
         header,
         banners,
         /* @__PURE__ */ jsxRuntime.jsxs(
-          react$1.ThreadPrimitive.Viewport,
+          react.ThreadPrimitive.Viewport,
           {
             turnAnchor,
             className: viewportClassName ?? DEFAULT_VIEWPORT_CLASS,
@@ -75,13 +104,13 @@ function ChatThreadShell({
             children: [
               viewportOverlay,
               /* @__PURE__ */ jsxRuntime.jsx("div", { className: "flex flex-1 flex-col items-center", children: /* @__PURE__ */ jsxRuntime.jsxs("div", { className: contentClassName ?? DEFAULT_CONTENT_CLASS, children: [
-                welcome != null ? /* @__PURE__ */ jsxRuntime.jsx(react$1.ThreadPrimitive.Empty, { children: welcome }) : null,
+                welcome != null ? /* @__PURE__ */ jsxRuntime.jsx(react.ThreadPrimitive.Empty, { children: welcome }) : null,
                 beforeMessages,
-                renderMessages ? renderMessages(messages) : messages,
+                renderMessages ? renderMessages(messages2) : messages2,
                 runningIndicator$1
               ] }) }),
               /* @__PURE__ */ jsxRuntime.jsxs(
-                react$1.ThreadPrimitive.ViewportFooter,
+                react.ThreadPrimitive.ViewportFooter,
                 {
                   className: viewportFooterClassName ?? DEFAULT_VIEWPORT_FOOTER_CLASS,
                   children: [

package/dist/chat-ui/thread/chat-thread-shell.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"chat-thread-shell.cjs","sources":["../../../src/chat-ui/thread/chat-thread-shell.tsx"],"sourcesContent":["import { AuiIf, ThreadPrimitive } from \"@assistant-ui/react\";\nimport { ChevronDownIcon } from \"lucide-react\";\nimport type { CSSProperties, FC, ReactNode } from \"react\";\nimport { useMemo } from \"react\";\nimport {\n  THREAD_PANEL_CLASS,\n  THREAD_SCROLL_TO_BOTTOM_CLASS,\n} from \"../messages/message-components\";\nimport { MessageActionsReadOnlyProvider } from \"../messages/message-actions-config\";\nimport { UserEditComposer } from \"../messages/user-edit-composer\";\nimport { ThreadRunningIndicator } from \"../messages/running-indicator\";\n\nexport interface ChatThreadMessageComponents {\n  UserMessage: FC;\n  AssistantMessage: FC;\n  /**\n   * Rendered in place of a user message while it is being edited (the action\n   * bar's Edit control). Defaults to the shared UserEditComposer.\n   */\n  UserEditComposer?: FC;\n}\n\nconst DEFAULT_VIEWPORT_CLASS =\n  \"relative flex flex-1 flex-col overflow-y-auto scroll-smooth\";\nconst DEFAULT_CONTENT_CLASS = \"w-full flex-1 px-3 pt-3\";\nconst DEFAULT_VIEWPORT_FOOTER_CLASS = \"sticky bottom-0\";\n\nconst DefaultScrollToBottom: FC = () => (\n  <AuiIf condition={(state) => !state.thread.isEmpty}>\n    <div className=\"flex justify-center pb-0.5\">\n      <ThreadPrimitive.ScrollToBottom\n        aria-label=\"Scroll to bottom\"\n        title=\"Scroll to bottom\"\n        className={THREAD_SCROLL_TO_BOTTOM_CLASS}\n      >\n        <ChevronDownIcon className=\"size-3.5\" />\n      </ThreadPrimitive.ScrollToBottom>\n    </div>\n  </AuiIf>\n);\n\nexport interface ChatThreadShellProps {\n  /**\n   * Rendered as the first child of the panel — for absolutely-positioned\n   * scrims that must cover the whole panel (AthenaChat's thread-loading\n   * overlay). The shell does not force a positioning context; hosts include\n   * `relative` in `panelClassName` when the scrim anchors to the panel.\n   */\n  overlay?: ReactNode;\n  /**\n   * Rendered above the viewport, inside the panel: header rows, slide-over\n   * panels, and anything else the host docks at the top.\n   */\n  header?: ReactNode;\n  /** Status/approval banners rendered between the header and the viewport. */\n  banners?: ReactNode;\n  /**\n   * Nodes rendered inside the scroll viewport before the message column —\n   * absolutely/sticky-positioned overlays (thread-switch skeletons,\n   * scroll-to-top controls) and invisible observers (read receipts).\n   */\n  viewportOverlay?: ReactNode;\n  /** Empty-thread state, rendered via ThreadPrimitive.Empty when provided. */\n  welcome?: ReactNode;\n  /**\n   * Rendered inside the message column right above the messages — the\n   * windowed wire's history edge (`StatewireHistoryEdge`): the \"N earlier\n   * messages\" row that pages older history in and fetches cut tool outputs.\n   */\n  beforeMessages?: ReactNode;\n  /** Message renderers for ThreadPrimitive.Messages. */\n  components: ChatThreadMessageComponents;\n  /**\n   * Wraps the messages element — for hosts that provide message-scoped\n   * context (e.g. activity-grouping providers). Must render its argument.\n   */\n  renderMessages?: (messages: ReactNode) => ReactNode;\n  /**\n   * Rendered between the viewport and the composer: connection banners,\n   * queued-message trays.\n   */\n  belowViewport?: ReactNode;\n  /** The composer dock (or a read-only notice replacing it). */\n  composer?: ReactNode;\n  /**\n   * Replaces the docked-panel card chrome — for hosts that render a\n   * full-page column (AthenaChat's `aui-*` surface) instead of the\n   * side-panel/task-pane card.\n   */\n  panelClassName?: string;\n  /**\n   * Value of the `--thread-max-width` variable read by width-capped\n   * message/welcome/footer columns. Panels fill their width (the default,\n   * \"100%\"); full-page hosts cap the column (AthenaChat defaults to 44rem).\n   */\n  maxWidth?: string;\n  /**\n   * Anchors the latest turn to the top of the viewport instead of pinning\n   * scroll to the bottom (AthenaChat). Omitted = the primitive's default.\n   */\n  turnAnchor?: \"top\" | \"bottom\";\n  /** Replaces the scroll-viewport chrome (padding, overflow tuning). */\n  viewportClassName?: string;\n  /**\n   * Replaces the message-column chrome. Hosts whose welcome screen centers\n   * vertically make this a flex column so the welcome node can grow.\n   */\n  contentClassName?: string;\n  /**\n   * Thread-level \"Thinking...\" indicator rendered after the messages. Hosts\n   * whose message components render their own per-message indicators pass\n   * null so the two never double up.\n   */\n  runningIndicator?: ReactNode;\n  /**\n   * Replaces the default centered scroll-to-bottom pill inside the sticky\n   * viewport footer (AthenaChat floats a circular button above its composer).\n   */\n  scrollToBottom?: ReactNode;\n  /**\n   * Content docked inside the sticky viewport footer, after the\n   * scroll-to-bottom control — for hosts whose banners + composer float over\n   * the scrolling messages (AthenaChat) instead of sitting below the\n   * viewport via `belowViewport`/`composer`.\n   */\n  viewportFooter?: ReactNode;\n  /** Replaces the sticky viewport-footer chrome. */\n  viewportFooterClassName?: string;\n  /**\n   * Marks the transcript read-only: mutating message actions (Edit,\n   * Regenerate) hide while Copy stays available. For threads whose server\n   * rejects dispatch (e.g. legacy pre-cutover conversations) — the runtime's\n   * static capability flags cannot express a per-thread fence.\n   */\n  readOnly?: boolean;\n}\n\n/**\n * The shared thread layout: panel chrome, scroll viewport with welcome +\n * messages + running indicator, the sticky scroll-to-bottom control, and the\n * host-supplied composer dock.\n *\n * Hosts express their genuine divergences through the slots above; the\n * layout, scroll behaviour, and empty/running states stay identical across\n * surfaces so fixes land once.\n */\nexport function ChatThreadShell({\n  overlay,\n  header,\n  banners,\n  viewportOverlay,\n  welcome,\n  beforeMessages,\n  components,\n  renderMessages,\n  belowViewport,\n  composer,\n  panelClassName,\n  maxWidth = \"100%\",\n  turnAnchor,\n  viewportClassName,\n  contentClassName,\n  runningIndicator = <ThreadRunningIndicator />,\n  scrollToBottom = <DefaultScrollToBottom />,\n  viewportFooter,\n  viewportFooterClassName,\n  readOnly = false,\n}: ChatThreadShellProps) {\n  const {\n    UserMessage,\n    AssistantMessage,\n    UserEditComposer: UserEditComposerComponent = UserEditComposer,\n  } = components;\n  // ThreadPrimitive.Messages resolves per-message renderers by component\n  // identity, so keep the merged map referentially stable across renders.\n  const messageComponents = useMemo(\n    () => ({\n      UserMessage,\n      AssistantMessage,\n      UserEditComposer: UserEditComposerComponent,\n    }),\n    [UserMessage, AssistantMessage, UserEditComposerComponent],\n  );\n  const messages = <ThreadPrimitive.Messages components={messageComponents} />;\n  return (\n    <MessageActionsReadOnlyProvider readOnly={readOnly}>\n      <ThreadPrimitive.Root\n        className={panelClassName ?? THREAD_PANEL_CLASS}\n        style={{ \"--thread-max-width\": maxWidth } as CSSProperties}\n      >\n        {overlay}\n        {header}\n        {banners}\n        <ThreadPrimitive.Viewport\n          turnAnchor={turnAnchor}\n          className={viewportClassName ?? DEFAULT_VIEWPORT_CLASS}\n          data-aui-thread-viewport=\"\"\n        >\n          {viewportOverlay}\n          {/* No min-h-0 here: the column's automatic minimum height is what\n            lets it extend past one screen, keeping the sticky footer's flow\n            position after the full message list. Capping it detaches the\n            footer (scroll pill, docked composers) as soon as the thread\n            scrolls — proven in a live-browser probe of both structures. */}\n          <div className=\"flex flex-1 flex-col items-center\">\n            <div className={contentClassName ?? DEFAULT_CONTENT_CLASS}>\n              {welcome != null ? (\n                <ThreadPrimitive.Empty>{welcome}</ThreadPrimitive.Empty>\n              ) : null}\n              {beforeMessages}\n              {renderMessages ? renderMessages(messages) : messages}\n              {runningIndicator}\n            </div>\n          </div>\n          <ThreadPrimitive.ViewportFooter\n            className={viewportFooterClassName ?? DEFAULT_VIEWPORT_FOOTER_CLASS}\n          >\n            {scrollToBottom}\n            {viewportFooter}\n          </ThreadPrimitive.ViewportFooter>\n        </ThreadPrimitive.Viewport>\n        {belowViewport}\n        {composer}\n      </ThreadPrimitive.Root>\n    </MessageActionsReadOnlyProvider>\n  );\n}\n"],"names":["jsx","AuiIf","ThreadPrimitive","THREAD_SCROLL_TO_BOTTOM_CLASS","ChevronDownIcon","runningIndicator","ThreadRunningIndicator","UserEditComposer","messageComponents","useMemo","MessageActionsReadOnlyProvider","jsxs","THREAD_PANEL_CLASS"],"mappings":";;;;;;;;;;;AAsBA,MAAM,yBACJ;AACF,MAAM,wBAAwB;AAC9B,MAAM,gCAAgC;AAEtC,MAAM,wBAA4B,MAChCA,2BAAAA,IAACC,QAAAA,OAAA,EAAM,WAAW,CAAC,UAAU,CAAC,MAAM,OAAO,SACzC,UAAAD,+BAAC,OAAA,EAAI,WAAU,8BACb,UAAAA,2BAAAA;AAAAA,EAACE,QAAAA,gBAAgB;AAAA,EAAhB;AAAA,IACC,cAAW;AAAA,IACX,OAAM;AAAA,IACN,WAAWC,kBAAAA;AAAAA,IAEX,UAAAH,2BAAAA,IAACI,YAAAA,iBAAA,EAAgB,WAAU,WAAA,CAAW;AAAA,EAAA;AACxC,EAAA,CACF,EAAA,CACF;AA4GK,SAAS,gBAAgB;AAAA,EAC9B;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA,WAAW;AAAA,EACX;AAAA,EACA;AAAA,EACA;AAAA,EAAA,kBACAC,oDAAoBC,iBAAAA,wBAAA,EAAuB;AAAA,EAC3C,gDAAkB,uBAAA,EAAsB;AAAA,EACxC;AAAA,EACA;AAAA,EACA,WAAW;AACb,GAAyB;AACvB,QAAM;AAAA,IACJ;AAAA,IACA;AAAA,IACA,kBAAkB,4BAA4BC,iBAAAA;AAAAA,EAAA,IAC5C;AAGJ,QAAMC,qBAAoBC,MAAAA;AAAAA,IACxB,OAAO;AAAA,MACL;AAAA,MACA;AAAA,MACA,kBAAkB;AAAA,IAAA;AAAA,IAEpB,CAAC,aAAa,kBAAkB,yBAAyB;AAAA,EAAA;AAE3D,QAAM,WAAWT,2BAAAA,IAACE,QAAAA,gBAAgB,UAAhB,EAAyB,YAAYM,oBAAmB;AAC1E,SACER,2BAAAA,IAACU,qBAAAA,kCAA+B,UAC9B,UAAAC,2BAAAA;AAAAA,IAACT,QAAAA,gBAAgB;AAAA,IAAhB;AAAA,MACC,WAAW,kBAAkBU,aAAAA;AAAAA,MAC7B,OAAO,EAAE,sBAAsB,SAAA;AAAA,MAE9B,UAAA;AAAA,QAAA;AAAA,QACA;AAAA,QACA;AAAA,QACDD,2BAAAA;AAAAA,UAACT,QAAAA,gBAAgB;AAAA,UAAhB;AAAA,YACC;AAAA,YACA,WAAW,qBAAqB;AAAA,YAChC,4BAAyB;AAAA,YAExB,UAAA;AAAA,cAAA;AAAA,cAMDF,2BAAAA,IAAC,SAAI,WAAU,qCACb,0CAAC,OAAA,EAAI,WAAW,oBAAoB,uBACjC,UAAA;AAAA,gBAAA,WAAW,OACVA,2BAAAA,IAACE,QAAAA,gBAAgB,OAAhB,EAAuB,mBAAQ,IAC9B;AAAA,gBACH;AAAA,gBACA,iBAAiB,eAAe,QAAQ,IAAI;AAAA,gBAC5CG;AAAAA,cAAA,EAAA,CACH,EAAA,CACF;AAAA,cACAM,2BAAAA;AAAAA,gBAACT,QAAAA,gBAAgB;AAAA,gBAAhB;AAAA,kBACC,WAAW,2BAA2B;AAAA,kBAErC,UAAA;AAAA,oBAAA;AAAA,oBACA;AAAA,kBAAA;AAAA,gBAAA;AAAA,cAAA;AAAA,YACH;AAAA,UAAA;AAAA,QAAA;AAAA,QAED;AAAA,QACA;AAAA,MAAA;AAAA,IAAA;AAAA,EAAA,GAEL;AAEJ;;"}
+{"version":3,"file":"chat-thread-shell.cjs","sources":["../../../src/chat-ui/thread/chat-thread-shell.tsx"],"sourcesContent":["import { AuiIf, ThreadPrimitive, useAuiState } from \"@assistant-ui/react\";\nimport { deepAgentRowKey } from \"@athenaintel/agent-runtime\";\nimport { ChevronDownIcon } from \"lucide-react\";\nimport type { ComponentProps, CSSProperties, FC, ReactNode } from \"react\";\nimport { useMemo } from \"react\";\nimport {\n  THREAD_PANEL_CLASS,\n  THREAD_SCROLL_TO_BOTTOM_CLASS,\n} from \"../messages/message-components\";\nimport { AnswerLabelScope } from \"../messages/answer-label\";\nimport { MessageActionsReadOnlyProvider } from \"../messages/message-actions-config\";\nimport { UserEditComposer } from \"../messages/user-edit-composer\";\nimport { ThreadRunningIndicator } from \"../messages/running-indicator\";\nimport {\n  HandoffArrivalAfterMessage,\n  HandoffArrivalScope,\n} from \"../messages/handoff-arrival-divider\";\n\nexport interface ChatThreadMessageComponents {\n  UserMessage: FC;\n  AssistantMessage: FC;\n  /**\n   * Rendered in place of a user message while it is being edited (the action\n   * bar's Edit control). Defaults to the shared UserEditComposer.\n   */\n  UserEditComposer?: FC;\n}\n\n// One wrapper per message renderer, so a renderer keeps its identity across\n// shells and renders: a row renders its message through the component it is\n// handed, and a new component identity would remount every row.\nconst withArrivalDivider = (() => {\n  const wrapped = new WeakMap<FC, FC>();\n  return (Message: FC): FC => {\n    const existing = wrapped.get(Message);\n    if (existing) return existing;\n    const WithArrivalDivider: FC = () => (\n      <>\n        <Message />\n        <HandoffArrivalAfterMessage />\n      </>\n    );\n    wrapped.set(Message, WithArrivalDivider);\n    return WithArrivalDivider;\n  };\n})();\n\ntype ThreadRowComponents = ComponentProps<typeof ThreadPrimitive.MessageByIndex>[\"components\"];\n\nconst ROW_KEY_SEPARATOR = \"\\u0000\";\n\n/**\n * The thread's messages, one row each under its row key (`deepAgentRowKey`).\n *\n * `ThreadPrimitive.Messages` keys a row by its message id. A page of older\n * history that folds into the first loaded turn (a long run's earlier steps)\n * gives that turn a new id on every page, so keyed by id the turn's row\n * remounted once per page: it flashed through its entrance animation, closed\n * what the person had opened in it and took the browser's scroll anchor with\n * it. The row key stays with the row the turn continues, so the page lands in\n * the mounted row. A message from another runtime keys by its id, as before.\n */\nconst ThreadRows: FC<{ components: ThreadRowComponents }> = ({ components }) => {\n  const joinedKeys = useAuiState((s) =>\n    s.thread.messages.map((message) => deepAgentRowKey(message)).join(ROW_KEY_SEPARATOR),\n  );\n  const rowKeys = useMemo(\n    () => (joinedKeys === \"\" ? [] : joinedKeys.split(ROW_KEY_SEPARATOR)),\n    [joinedKeys],\n  );\n  return (\n    <>\n      {rowKeys.map((key, index) => (\n        <ThreadPrimitive.MessageByIndex key={key} index={index} components={components} />\n      ))}\n    </>\n  );\n};\n\nconst DEFAULT_VIEWPORT_CLASS =\n  \"relative flex flex-1 flex-col overflow-y-auto scroll-smooth\";\nconst DEFAULT_CONTENT_CLASS = \"w-full flex-1 px-3 pt-3\";\nconst DEFAULT_VIEWPORT_FOOTER_CLASS = \"sticky bottom-0\";\n\nconst DefaultScrollToBottom: FC = () => (\n  <AuiIf condition={(state) => !state.thread.isEmpty}>\n    <div className=\"flex justify-center pb-0.5\">\n      <ThreadPrimitive.ScrollToBottom\n        aria-label=\"Scroll to bottom\"\n        title=\"Scroll to bottom\"\n        className={THREAD_SCROLL_TO_BOTTOM_CLASS}\n      >\n        <ChevronDownIcon className=\"size-3.5\" />\n      </ThreadPrimitive.ScrollToBottom>\n    </div>\n  </AuiIf>\n);\n\nexport interface ChatThreadShellProps {\n  /**\n   * Rendered as the first child of the panel — for absolutely-positioned\n   * scrims that must cover the whole panel (AthenaChat's thread-loading\n   * overlay). The shell does not force a positioning context; hosts include\n   * `relative` in `panelClassName` when the scrim anchors to the panel.\n   */\n  overlay?: ReactNode;\n  /**\n   * Rendered above the viewport, inside the panel: header rows, slide-over\n   * panels, and anything else the host docks at the top.\n   */\n  header?: ReactNode;\n  /** Status/approval banners rendered between the header and the viewport. */\n  banners?: ReactNode;\n  /**\n   * Nodes rendered inside the scroll viewport before the message column —\n   * absolutely/sticky-positioned overlays (thread-switch skeletons,\n   * scroll-to-top controls) and invisible observers (read receipts).\n   */\n  viewportOverlay?: ReactNode;\n  /** Empty-thread state, rendered via ThreadPrimitive.Empty when provided. */\n  welcome?: ReactNode;\n  /**\n   * Rendered inside the message column right above the messages — the\n   * windowed wire's history edge (`StatewireHistoryEdge`): the \"N earlier\n   * messages\" row that pages older history in and fetches cut tool outputs.\n   */\n  beforeMessages?: ReactNode;\n  /** Message renderers for the thread's rows. */\n  components: ChatThreadMessageComponents;\n  /**\n   * Wraps the messages element — for hosts that provide message-scoped\n   * context (e.g. activity-grouping providers). Must render its argument.\n   */\n  renderMessages?: (messages: ReactNode) => ReactNode;\n  /**\n   * Rendered between the viewport and the composer: connection banners,\n   * queued-message trays.\n   */\n  belowViewport?: ReactNode;\n  /** The composer dock (or a read-only notice replacing it). */\n  composer?: ReactNode;\n  /**\n   * Replaces the docked-panel card chrome — for hosts that render a\n   * full-page column (AthenaChat's `aui-*` surface) instead of the\n   * side-panel/task-pane card.\n   */\n  panelClassName?: string;\n  /**\n   * Value of the `--thread-max-width` variable read by width-capped\n   * message/welcome/footer columns. Panels fill their width (the default,\n   * \"100%\"); full-page hosts cap the column (AthenaChat defaults to 44rem).\n   */\n  maxWidth?: string;\n  /**\n   * Anchors the latest turn to the top of the viewport instead of pinning\n   * scroll to the bottom (AthenaChat). Omitted = the primitive's default.\n   */\n  turnAnchor?: \"top\" | \"bottom\";\n  /** Replaces the scroll-viewport chrome (padding, overflow tuning). */\n  viewportClassName?: string;\n  /**\n   * Replaces the message-column chrome. Hosts whose welcome screen centers\n   * vertically make this a flex column so the welcome node can grow.\n   */\n  contentClassName?: string;\n  /**\n   * Thread-level \"Thinking...\" indicator rendered after the messages. Hosts\n   * whose message components render their own per-message indicators pass\n   * null so the two never double up.\n   */\n  runningIndicator?: ReactNode;\n  /**\n   * Replaces the default centered scroll-to-bottom pill inside the sticky\n   * viewport footer (AthenaChat floats a circular button above its composer).\n   */\n  scrollToBottom?: ReactNode;\n  /**\n   * Content docked inside the sticky viewport footer, after the\n   * scroll-to-bottom control — for hosts whose banners + composer float over\n   * the scrolling messages (AthenaChat) instead of sitting below the\n   * viewport via `belowViewport`/`composer`.\n   */\n  viewportFooter?: ReactNode;\n  /** Replaces the sticky viewport-footer chrome. */\n  viewportFooterClassName?: string;\n  /**\n   * Marks the transcript read-only: mutating message actions (Edit,\n   * Regenerate) hide while Copy stays available. For threads whose server\n   * rejects dispatch (e.g. legacy pre-cutover conversations) — the runtime's\n   * static capability flags cannot express a per-thread fence.\n   */\n  readOnly?: boolean;\n  /**\n   * The backend id of the thread on screen. When this surface picked the\n   * conversation up from another one (`recordHandoffArrival`), the transcript\n   * shows \"Continued from …\" after the last message that was already there.\n   */\n  arrivalThreadId?: string | null;\n  /**\n   * The signed-in viewer's Athena user id (`user_…`), for the answer labels\n   * (`AnswerLabel`): their own answers read \"for you\", and an answer for\n   * another person is labelled even in a 1:1 chat. Omitted, answers name the\n   * person by name and are labelled only once the thread holds several agents\n   * or several people.\n   */\n  viewerUserId?: string | null;\n}\n\n/**\n * The shared thread layout: panel chrome, scroll viewport with welcome +\n * messages + running indicator, the sticky scroll-to-bottom control, and the\n * host-supplied composer dock.\n *\n * Hosts express their genuine divergences through the slots above; the\n * layout, scroll behaviour, and empty/running states stay identical across\n * surfaces so fixes land once.\n */\nexport function ChatThreadShell({\n  overlay,\n  header,\n  banners,\n  viewportOverlay,\n  welcome,\n  beforeMessages,\n  components,\n  renderMessages,\n  belowViewport,\n  composer,\n  panelClassName,\n  maxWidth = \"100%\",\n  turnAnchor,\n  viewportClassName,\n  contentClassName,\n  runningIndicator = <ThreadRunningIndicator />,\n  scrollToBottom = <DefaultScrollToBottom />,\n  viewportFooter,\n  viewportFooterClassName,\n  readOnly = false,\n  arrivalThreadId = null,\n  viewerUserId = null,\n}: ChatThreadShellProps) {\n  const {\n    UserMessage,\n    AssistantMessage,\n    UserEditComposer: UserEditComposerComponent = UserEditComposer,\n  } = components;\n  // A row resolves its renderer by component identity, so keep the merged map\n  // referentially stable across renders.\n  const messageComponents = useMemo(\n    () => ({\n      UserMessage: withArrivalDivider(UserMessage),\n      AssistantMessage: withArrivalDivider(AssistantMessage),\n      UserEditComposer: UserEditComposerComponent,\n    }),\n    [UserMessage, AssistantMessage, UserEditComposerComponent],\n  );\n  const messages = (\n    <HandoffArrivalScope threadId={arrivalThreadId}>\n      <AnswerLabelScope viewerUserId={viewerUserId}>\n        <ThreadRows components={messageComponents} />\n      </AnswerLabelScope>\n    </HandoffArrivalScope>\n  );\n  return (\n    <MessageActionsReadOnlyProvider readOnly={readOnly}>\n      <ThreadPrimitive.Root\n        className={panelClassName ?? THREAD_PANEL_CLASS}\n        style={{ \"--thread-max-width\": maxWidth } as CSSProperties}\n      >\n        {overlay}\n        {header}\n        {banners}\n        <ThreadPrimitive.Viewport\n          turnAnchor={turnAnchor}\n          className={viewportClassName ?? DEFAULT_VIEWPORT_CLASS}\n          data-aui-thread-viewport=\"\"\n        >\n          {viewportOverlay}\n          {/* No min-h-0 here: the column's automatic minimum height is what\n            lets it extend past one screen, keeping the sticky footer's flow\n            position after the full message list. Capping it detaches the\n            footer (scroll pill, docked composers) as soon as the thread\n            scrolls — proven in a live-browser probe of both structures. */}\n          <div className=\"flex flex-1 flex-col items-center\">\n            <div className={contentClassName ?? DEFAULT_CONTENT_CLASS}>\n              {welcome != null ? (\n                <ThreadPrimitive.Empty>{welcome}</ThreadPrimitive.Empty>\n              ) : null}\n              {beforeMessages}\n              {renderMessages ? renderMessages(messages) : messages}\n              {runningIndicator}\n            </div>\n          </div>\n          <ThreadPrimitive.ViewportFooter\n            className={viewportFooterClassName ?? DEFAULT_VIEWPORT_FOOTER_CLASS}\n          >\n            {scrollToBottom}\n            {viewportFooter}\n          </ThreadPrimitive.ViewportFooter>\n        </ThreadPrimitive.Viewport>\n        {belowViewport}\n        {composer}\n      </ThreadPrimitive.Root>\n    </MessageActionsReadOnlyProvider>\n  );\n}\n"],"names":["jsxs","Fragment","jsx","HandoffArrivalAfterMessage","useAuiState","deepAgentRowKey","useMemo","ThreadPrimitive","AuiIf","THREAD_SCROLL_TO_BOTTOM_CLASS","ChevronDownIcon","runningIndicator","ThreadRunningIndicator","UserEditComposer","messageComponents","messages","HandoffArrivalScope","AnswerLabelScope","MessageActionsReadOnlyProvider","THREAD_PANEL_CLASS"],"mappings":";;;;;;;;;;;;;;AA+BA,MAAM,qBAAsB,uBAAM;AAChC,QAAM,8BAAc,QAAA;AACpB,SAAO,CAAC,YAAoB;AAC1B,UAAM,WAAW,QAAQ,IAAI,OAAO;AACpC,QAAI,SAAU,QAAO;AACrB,UAAM,qBAAyB,MAC7BA,2BAAAA,KAAAC,WAAAA,UAAA,EACE,UAAA;AAAA,MAAAC,2BAAAA,IAAC,SAAA,EAAQ;AAAA,qCACRC,sBAAAA,4BAAA,CAAA,CAA2B;AAAA,IAAA,GAC9B;AAEF,YAAQ,IAAI,SAAS,kBAAkB;AACvC,WAAO;AAAA,EACT;AACF,GAAA;AAIA,MAAM,oBAAoB;AAa1B,MAAM,aAAsD,CAAC,EAAE,iBAAiB;AAC9E,QAAM,aAAaC,MAAAA;AAAAA,IAAY,CAAC,MAC9B,EAAE,OAAO,SAAS,IAAI,CAAC,YAAYC,yBAAgB,OAAO,CAAC,EAAE,KAAK,iBAAiB;AAAA,EAAA;AAErF,QAAM,UAAUC,WAAAA;AAAAA,IACd,MAAO,eAAe,KAAK,CAAA,IAAK,WAAW,MAAM,iBAAiB;AAAA,IAClE,CAAC,UAAU;AAAA,EAAA;AAEb,SACEJ,2BAAAA,IAAAD,WAAAA,UAAA,EACG,UAAA,QAAQ,IAAI,CAAC,KAAK,UACjBC,2BAAAA,IAACK,MAAAA,gBAAgB,gBAAhB,EAAyC,OAAc,cAAnB,GAA2C,CACjF,GACH;AAEJ;AAEA,MAAM,yBACJ;AACF,MAAM,wBAAwB;AAC9B,MAAM,gCAAgC;AAEtC,MAAM,wBAA4B,MAChCL,2BAAAA,IAACM,MAAAA,OAAA,EAAM,WAAW,CAAC,UAAU,CAAC,MAAM,OAAO,SACzC,UAAAN,+BAAC,OAAA,EAAI,WAAU,8BACb,UAAAA,2BAAAA;AAAAA,EAACK,MAAAA,gBAAgB;AAAA,EAAhB;AAAA,IACC,cAAW;AAAA,IACX,OAAM;AAAA,IACN,WAAWE,kBAAAA;AAAAA,IAEX,UAAAP,2BAAAA,IAACQ,YAAAA,iBAAA,EAAgB,WAAU,WAAA,CAAW;AAAA,EAAA;AACxC,EAAA,CACF,EAAA,CACF;AA0HK,SAAS,gBAAgB;AAAA,EAC9B;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA,WAAW;AAAA,EACX;AAAA,EACA;AAAA,EACA;AAAA,EAAA,kBACAC,oDAAoBC,iBAAAA,wBAAA,EAAuB;AAAA,EAC3C,gDAAkB,uBAAA,EAAsB;AAAA,EACxC;AAAA,EACA;AAAA,EACA,WAAW;AAAA,EACX,kBAAkB;AAAA,EAClB,eAAe;AACjB,GAAyB;AACvB,QAAM;AAAA,IACJ;AAAA,IACA;AAAA,IACA,kBAAkB,4BAA4BC,iBAAAA;AAAAA,EAAA,IAC5C;AAGJ,QAAMC,qBAAoBR,WAAAA;AAAAA,IACxB,OAAO;AAAA,MACL,aAAa,mBAAmB,WAAW;AAAA,MAC3C,kBAAkB,mBAAmB,gBAAgB;AAAA,MACrD,kBAAkB;AAAA,IAAA;AAAA,IAEpB,CAAC,aAAa,kBAAkB,yBAAyB;AAAA,EAAA;AAE3D,QAAMS,YACJb,2BAAAA,IAACc,2CAAA,EAAoB,UAAU,iBAC7B,UAAAd,2BAAAA,IAACe,YAAAA,kBAAA,EAAiB,cAChB,UAAAf,2BAAAA,IAAC,YAAA,EAAW,YAAYY,mBAAA,CAAmB,GAC7C,GACF;AAEF,SACEZ,2BAAAA,IAACgB,qBAAAA,kCAA+B,UAC9B,UAAAlB,2BAAAA;AAAAA,IAACO,MAAAA,gBAAgB;AAAA,IAAhB;AAAA,MACC,WAAW,kBAAkBY,aAAAA;AAAAA,MAC7B,OAAO,EAAE,sBAAsB,SAAA;AAAA,MAE9B,UAAA;AAAA,QAAA;AAAA,QACA;AAAA,QACA;AAAA,QACDnB,2BAAAA;AAAAA,UAACO,MAAAA,gBAAgB;AAAA,UAAhB;AAAA,YACC;AAAA,YACA,WAAW,qBAAqB;AAAA,YAChC,4BAAyB;AAAA,YAExB,UAAA;AAAA,cAAA;AAAA,cAMDL,2BAAAA,IAAC,SAAI,WAAU,qCACb,0CAAC,OAAA,EAAI,WAAW,oBAAoB,uBACjC,UAAA;AAAA,gBAAA,WAAW,OACVA,2BAAAA,IAACK,MAAAA,gBAAgB,OAAhB,EAAuB,mBAAQ,IAC9B;AAAA,gBACH;AAAA,gBACA,iBAAiB,eAAeQ,SAAQ,IAAIA;AAAA,gBAC5CJ;AAAAA,cAAA,EAAA,CACH,EAAA,CACF;AAAA,cACAX,2BAAAA;AAAAA,gBAACO,MAAAA,gBAAgB;AAAA,gBAAhB;AAAA,kBACC,WAAW,2BAA2B;AAAA,kBAErC,UAAA;AAAA,oBAAA;AAAA,oBACA;AAAA,kBAAA;AAAA,gBAAA;AAAA,cAAA;AAAA,YACH;AAAA,UAAA;AAAA,QAAA;AAAA,QAED;AAAA,QACA;AAAA,MAAA;AAAA,IAAA;AAAA,EAAA,GAEL;AAEJ;;"}

package/dist/chat-ui/thread/chat-thread-shell.d.ts

@@ -37,7 +37,7 @@ export interface ChatThreadShellProps {
      * messages" row that pages older history in and fetches cut tool outputs.
      */
     beforeMessages?: ReactNode;
-    /** Message renderers for ThreadPrimitive.Messages. */
+    /** Message renderers for the thread's rows. */
     components: ChatThreadMessageComponents;
     /**
      * Wraps the messages element — for hosts that provide message-scoped
@@ -102,6 +102,20 @@ export interface ChatThreadShellProps {
      * static capability flags cannot express a per-thread fence.
      */
     readOnly?: boolean;
+    /**
+     * The backend id of the thread on screen. When this surface picked the
+     * conversation up from another one (`recordHandoffArrival`), the transcript
+     * shows "Continued from …" after the last message that was already there.
+     */
+    arrivalThreadId?: string | null;
+    /**
+     * The signed-in viewer's Athena user id (`user_…`), for the answer labels
+     * (`AnswerLabel`): their own answers read "for you", and an answer for
+     * another person is labelled even in a 1:1 chat. Omitted, answers name the
+     * person by name and are labelled only once the thread holds several agents
+     * or several people.
+     */
+    viewerUserId?: string | null;
 }
 /**
  * The shared thread layout: panel chrome, scroll viewport with welcome +
@@ -112,4 +126,4 @@ export interface ChatThreadShellProps {
  * layout, scroll behaviour, and empty/running states stay identical across
  * surfaces so fixes land once.
  */
-export declare function ChatThreadShell({ overlay, header, banners, viewportOverlay, welcome, beforeMessages, components, renderMessages, belowViewport, composer, panelClassName, maxWidth, turnAnchor, viewportClassName, contentClassName, runningIndicator, scrollToBottom, viewportFooter, viewportFooterClassName, readOnly, }: ChatThreadShellProps): import("react").JSX.Element;
+export declare function ChatThreadShell({ overlay, header, banners, viewportOverlay, welcome, beforeMessages, components, renderMessages, belowViewport, composer, panelClassName, maxWidth, turnAnchor, viewportClassName, contentClassName, runningIndicator, scrollToBottom, viewportFooter, viewportFooterClassName, readOnly, arrivalThreadId, viewerUserId, }: ChatThreadShellProps): import("react").JSX.Element;

package/dist/chat-ui/thread/chat-thread-shell.js

@@ -1,12 +1,39 @@
-import { jsx, jsxs } from "react/jsx-runtime";
-import { ThreadPrimitive, AuiIf } from "@assistant-ui/react";
+import { jsx, jsxs, Fragment } from "react/jsx-runtime";
+import { ThreadPrimitive, AuiIf, useAuiState } from "@assistant-ui/react";
 import { ChevronDownIcon } from "lucide-react";
 import { useMemo } from "react";
 import { THREAD_SCROLL_TO_BOTTOM_CLASS } from "../messages/message-components.js";
+import { AnswerLabelScope } from "../messages/answer-label.js";
 import { MessageActionsReadOnlyProvider } from "../messages/message-actions-config.js";
 import { UserEditComposer } from "../messages/user-edit-composer.js";
 import { ThreadRunningIndicator } from "../messages/running-indicator.js";
+import { HandoffArrivalScope, HandoffArrivalAfterMessage } from "../messages/handoff-arrival-divider.js";
 import { THREAD_PANEL_CLASS } from "../messages/thread-chrome.js";
+import { deepAgentRowKey } from "../../agent-runtime/src/messages.js";
+const withArrivalDivider = /* @__PURE__ */ (() => {
+  const wrapped = /* @__PURE__ */ new WeakMap();
+  return (Message) => {
+    const existing = wrapped.get(Message);
+    if (existing) return existing;
+    const WithArrivalDivider = () => /* @__PURE__ */ jsxs(Fragment, { children: [
+      /* @__PURE__ */ jsx(Message, {}),
+      /* @__PURE__ */ jsx(HandoffArrivalAfterMessage, {})
+    ] });
+    wrapped.set(Message, WithArrivalDivider);
+    return WithArrivalDivider;
+  };
+})();
+const ROW_KEY_SEPARATOR = "\0";
+const ThreadRows = ({ components }) => {
+  const joinedKeys = useAuiState(
+    (s) => s.thread.messages.map((message) => deepAgentRowKey(message)).join(ROW_KEY_SEPARATOR)
+  );
+  const rowKeys = useMemo(
+    () => joinedKeys === "" ? [] : joinedKeys.split(ROW_KEY_SEPARATOR),
+    [joinedKeys]
+  );
+  return /* @__PURE__ */ jsx(Fragment, { children: rowKeys.map((key, index) => /* @__PURE__ */ jsx(ThreadPrimitive.MessageByIndex, { index, components }, key)) });
+};
 const DEFAULT_VIEWPORT_CLASS = "relative flex flex-1 flex-col overflow-y-auto scroll-smooth";
 const DEFAULT_CONTENT_CLASS = "w-full flex-1 px-3 pt-3";
 const DEFAULT_VIEWPORT_FOOTER_CLASS = "sticky bottom-0";
@@ -39,7 +66,9 @@ function ChatThreadShell({
   scrollToBottom = /* @__PURE__ */ jsx(DefaultScrollToBottom, {}),
   viewportFooter,
   viewportFooterClassName,
-  readOnly = false
+  readOnly = false,
+  arrivalThreadId = null,
+  viewerUserId = null
 }) {
   const {
     UserMessage,
@@ -48,13 +77,13 @@ function ChatThreadShell({
   } = components;
   const messageComponents = useMemo(
     () => ({
-      UserMessage,
-      AssistantMessage,
+      UserMessage: withArrivalDivider(UserMessage),
+      AssistantMessage: withArrivalDivider(AssistantMessage),
       UserEditComposer: UserEditComposerComponent
     }),
     [UserMessage, AssistantMessage, UserEditComposerComponent]
   );
-  const messages = /* @__PURE__ */ jsx(ThreadPrimitive.Messages, { components: messageComponents });
+  const messages = /* @__PURE__ */ jsx(HandoffArrivalScope, { threadId: arrivalThreadId, children: /* @__PURE__ */ jsx(AnswerLabelScope, { viewerUserId, children: /* @__PURE__ */ jsx(ThreadRows, { components: messageComponents }) }) });
   return /* @__PURE__ */ jsx(MessageActionsReadOnlyProvider, { readOnly, children: /* @__PURE__ */ jsxs(
     ThreadPrimitive.Root,
     {

package/dist/chat-ui/thread/chat-thread-shell.js.map

@@ -1 +1 @@
-{"version":3,"file":"chat-thread-shell.js","sources":["../../../src/chat-ui/thread/chat-thread-shell.tsx"],"sourcesContent":["import { AuiIf, ThreadPrimitive } from \"@assistant-ui/react\";\nimport { ChevronDownIcon } from \"lucide-react\";\nimport type { CSSProperties, FC, ReactNode } from \"react\";\nimport { useMemo } from \"react\";\nimport {\n  THREAD_PANEL_CLASS,\n  THREAD_SCROLL_TO_BOTTOM_CLASS,\n} from \"../messages/message-components\";\nimport { MessageActionsReadOnlyProvider } from \"../messages/message-actions-config\";\nimport { UserEditComposer } from \"../messages/user-edit-composer\";\nimport { ThreadRunningIndicator } from \"../messages/running-indicator\";\n\nexport interface ChatThreadMessageComponents {\n  UserMessage: FC;\n  AssistantMessage: FC;\n  /**\n   * Rendered in place of a user message while it is being edited (the action\n   * bar's Edit control). Defaults to the shared UserEditComposer.\n   */\n  UserEditComposer?: FC;\n}\n\nconst DEFAULT_VIEWPORT_CLASS =\n  \"relative flex flex-1 flex-col overflow-y-auto scroll-smooth\";\nconst DEFAULT_CONTENT_CLASS = \"w-full flex-1 px-3 pt-3\";\nconst DEFAULT_VIEWPORT_FOOTER_CLASS = \"sticky bottom-0\";\n\nconst DefaultScrollToBottom: FC = () => (\n  <AuiIf condition={(state) => !state.thread.isEmpty}>\n    <div className=\"flex justify-center pb-0.5\">\n      <ThreadPrimitive.ScrollToBottom\n        aria-label=\"Scroll to bottom\"\n        title=\"Scroll to bottom\"\n        className={THREAD_SCROLL_TO_BOTTOM_CLASS}\n      >\n        <ChevronDownIcon className=\"size-3.5\" />\n      </ThreadPrimitive.ScrollToBottom>\n    </div>\n  </AuiIf>\n);\n\nexport interface ChatThreadShellProps {\n  /**\n   * Rendered as the first child of the panel — for absolutely-positioned\n   * scrims that must cover the whole panel (AthenaChat's thread-loading\n   * overlay). The shell does not force a positioning context; hosts include\n   * `relative` in `panelClassName` when the scrim anchors to the panel.\n   */\n  overlay?: ReactNode;\n  /**\n   * Rendered above the viewport, inside the panel: header rows, slide-over\n   * panels, and anything else the host docks at the top.\n   */\n  header?: ReactNode;\n  /** Status/approval banners rendered between the header and the viewport. */\n  banners?: ReactNode;\n  /**\n   * Nodes rendered inside the scroll viewport before the message column —\n   * absolutely/sticky-positioned overlays (thread-switch skeletons,\n   * scroll-to-top controls) and invisible observers (read receipts).\n   */\n  viewportOverlay?: ReactNode;\n  /** Empty-thread state, rendered via ThreadPrimitive.Empty when provided. */\n  welcome?: ReactNode;\n  /**\n   * Rendered inside the message column right above the messages — the\n   * windowed wire's history edge (`StatewireHistoryEdge`): the \"N earlier\n   * messages\" row that pages older history in and fetches cut tool outputs.\n   */\n  beforeMessages?: ReactNode;\n  /** Message renderers for ThreadPrimitive.Messages. */\n  components: ChatThreadMessageComponents;\n  /**\n   * Wraps the messages element — for hosts that provide message-scoped\n   * context (e.g. activity-grouping providers). Must render its argument.\n   */\n  renderMessages?: (messages: ReactNode) => ReactNode;\n  /**\n   * Rendered between the viewport and the composer: connection banners,\n   * queued-message trays.\n   */\n  belowViewport?: ReactNode;\n  /** The composer dock (or a read-only notice replacing it). */\n  composer?: ReactNode;\n  /**\n   * Replaces the docked-panel card chrome — for hosts that render a\n   * full-page column (AthenaChat's `aui-*` surface) instead of the\n   * side-panel/task-pane card.\n   */\n  panelClassName?: string;\n  /**\n   * Value of the `--thread-max-width` variable read by width-capped\n   * message/welcome/footer columns. Panels fill their width (the default,\n   * \"100%\"); full-page hosts cap the column (AthenaChat defaults to 44rem).\n   */\n  maxWidth?: string;\n  /**\n   * Anchors the latest turn to the top of the viewport instead of pinning\n   * scroll to the bottom (AthenaChat). Omitted = the primitive's default.\n   */\n  turnAnchor?: \"top\" | \"bottom\";\n  /** Replaces the scroll-viewport chrome (padding, overflow tuning). */\n  viewportClassName?: string;\n  /**\n   * Replaces the message-column chrome. Hosts whose welcome screen centers\n   * vertically make this a flex column so the welcome node can grow.\n   */\n  contentClassName?: string;\n  /**\n   * Thread-level \"Thinking...\" indicator rendered after the messages. Hosts\n   * whose message components render their own per-message indicators pass\n   * null so the two never double up.\n   */\n  runningIndicator?: ReactNode;\n  /**\n   * Replaces the default centered scroll-to-bottom pill inside the sticky\n   * viewport footer (AthenaChat floats a circular button above its composer).\n   */\n  scrollToBottom?: ReactNode;\n  /**\n   * Content docked inside the sticky viewport footer, after the\n   * scroll-to-bottom control — for hosts whose banners + composer float over\n   * the scrolling messages (AthenaChat) instead of sitting below the\n   * viewport via `belowViewport`/`composer`.\n   */\n  viewportFooter?: ReactNode;\n  /** Replaces the sticky viewport-footer chrome. */\n  viewportFooterClassName?: string;\n  /**\n   * Marks the transcript read-only: mutating message actions (Edit,\n   * Regenerate) hide while Copy stays available. For threads whose server\n   * rejects dispatch (e.g. legacy pre-cutover conversations) — the runtime's\n   * static capability flags cannot express a per-thread fence.\n   */\n  readOnly?: boolean;\n}\n\n/**\n * The shared thread layout: panel chrome, scroll viewport with welcome +\n * messages + running indicator, the sticky scroll-to-bottom control, and the\n * host-supplied composer dock.\n *\n * Hosts express their genuine divergences through the slots above; the\n * layout, scroll behaviour, and empty/running states stay identical across\n * surfaces so fixes land once.\n */\nexport function ChatThreadShell({\n  overlay,\n  header,\n  banners,\n  viewportOverlay,\n  welcome,\n  beforeMessages,\n  components,\n  renderMessages,\n  belowViewport,\n  composer,\n  panelClassName,\n  maxWidth = \"100%\",\n  turnAnchor,\n  viewportClassName,\n  contentClassName,\n  runningIndicator = <ThreadRunningIndicator />,\n  scrollToBottom = <DefaultScrollToBottom />,\n  viewportFooter,\n  viewportFooterClassName,\n  readOnly = false,\n}: ChatThreadShellProps) {\n  const {\n    UserMessage,\n    AssistantMessage,\n    UserEditComposer: UserEditComposerComponent = UserEditComposer,\n  } = components;\n  // ThreadPrimitive.Messages resolves per-message renderers by component\n  // identity, so keep the merged map referentially stable across renders.\n  const messageComponents = useMemo(\n    () => ({\n      UserMessage,\n      AssistantMessage,\n      UserEditComposer: UserEditComposerComponent,\n    }),\n    [UserMessage, AssistantMessage, UserEditComposerComponent],\n  );\n  const messages = <ThreadPrimitive.Messages components={messageComponents} />;\n  return (\n    <MessageActionsReadOnlyProvider readOnly={readOnly}>\n      <ThreadPrimitive.Root\n        className={panelClassName ?? THREAD_PANEL_CLASS}\n        style={{ \"--thread-max-width\": maxWidth } as CSSProperties}\n      >\n        {overlay}\n        {header}\n        {banners}\n        <ThreadPrimitive.Viewport\n          turnAnchor={turnAnchor}\n          className={viewportClassName ?? DEFAULT_VIEWPORT_CLASS}\n          data-aui-thread-viewport=\"\"\n        >\n          {viewportOverlay}\n          {/* No min-h-0 here: the column's automatic minimum height is what\n            lets it extend past one screen, keeping the sticky footer's flow\n            position after the full message list. Capping it detaches the\n            footer (scroll pill, docked composers) as soon as the thread\n            scrolls — proven in a live-browser probe of both structures. */}\n          <div className=\"flex flex-1 flex-col items-center\">\n            <div className={contentClassName ?? DEFAULT_CONTENT_CLASS}>\n              {welcome != null ? (\n                <ThreadPrimitive.Empty>{welcome}</ThreadPrimitive.Empty>\n              ) : null}\n              {beforeMessages}\n              {renderMessages ? renderMessages(messages) : messages}\n              {runningIndicator}\n            </div>\n          </div>\n          <ThreadPrimitive.ViewportFooter\n            className={viewportFooterClassName ?? DEFAULT_VIEWPORT_FOOTER_CLASS}\n          >\n            {scrollToBottom}\n            {viewportFooter}\n          </ThreadPrimitive.ViewportFooter>\n        </ThreadPrimitive.Viewport>\n        {belowViewport}\n        {composer}\n      </ThreadPrimitive.Root>\n    </MessageActionsReadOnlyProvider>\n  );\n}\n"],"names":[],"mappings":";;;;;;;;;AAsBA,MAAM,yBACJ;AACF,MAAM,wBAAwB;AAC9B,MAAM,gCAAgC;AAEtC,MAAM,wBAA4B,MAChC,oBAAC,OAAA,EAAM,WAAW,CAAC,UAAU,CAAC,MAAM,OAAO,SACzC,UAAA,oBAAC,OAAA,EAAI,WAAU,8BACb,UAAA;AAAA,EAAC,gBAAgB;AAAA,EAAhB;AAAA,IACC,cAAW;AAAA,IACX,OAAM;AAAA,IACN,WAAW;AAAA,IAEX,UAAA,oBAAC,iBAAA,EAAgB,WAAU,WAAA,CAAW;AAAA,EAAA;AACxC,EAAA,CACF,EAAA,CACF;AA4GK,SAAS,gBAAgB;AAAA,EAC9B;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA,WAAW;AAAA,EACX;AAAA,EACA;AAAA,EACA;AAAA,EACA,uCAAoB,wBAAA,EAAuB;AAAA,EAC3C,qCAAkB,uBAAA,EAAsB;AAAA,EACxC;AAAA,EACA;AAAA,EACA,WAAW;AACb,GAAyB;AACvB,QAAM;AAAA,IACJ;AAAA,IACA;AAAA,IACA,kBAAkB,4BAA4B;AAAA,EAAA,IAC5C;AAGJ,QAAM,oBAAoB;AAAA,IACxB,OAAO;AAAA,MACL;AAAA,MACA;AAAA,MACA,kBAAkB;AAAA,IAAA;AAAA,IAEpB,CAAC,aAAa,kBAAkB,yBAAyB;AAAA,EAAA;AAE3D,QAAM,WAAW,oBAAC,gBAAgB,UAAhB,EAAyB,YAAY,mBAAmB;AAC1E,SACE,oBAAC,kCAA+B,UAC9B,UAAA;AAAA,IAAC,gBAAgB;AAAA,IAAhB;AAAA,MACC,WAAW,kBAAkB;AAAA,MAC7B,OAAO,EAAE,sBAAsB,SAAA;AAAA,MAE9B,UAAA;AAAA,QAAA;AAAA,QACA;AAAA,QACA;AAAA,QACD;AAAA,UAAC,gBAAgB;AAAA,UAAhB;AAAA,YACC;AAAA,YACA,WAAW,qBAAqB;AAAA,YAChC,4BAAyB;AAAA,YAExB,UAAA;AAAA,cAAA;AAAA,cAMD,oBAAC,SAAI,WAAU,qCACb,+BAAC,OAAA,EAAI,WAAW,oBAAoB,uBACjC,UAAA;AAAA,gBAAA,WAAW,OACV,oBAAC,gBAAgB,OAAhB,EAAuB,mBAAQ,IAC9B;AAAA,gBACH;AAAA,gBACA,iBAAiB,eAAe,QAAQ,IAAI;AAAA,gBAC5C;AAAA,cAAA,EAAA,CACH,EAAA,CACF;AAAA,cACA;AAAA,gBAAC,gBAAgB;AAAA,gBAAhB;AAAA,kBACC,WAAW,2BAA2B;AAAA,kBAErC,UAAA;AAAA,oBAAA;AAAA,oBACA;AAAA,kBAAA;AAAA,gBAAA;AAAA,cAAA;AAAA,YACH;AAAA,UAAA;AAAA,QAAA;AAAA,QAED;AAAA,QACA;AAAA,MAAA;AAAA,IAAA;AAAA,EAAA,GAEL;AAEJ;"}
+{"version":3,"file":"chat-thread-shell.js","sources":["../../../src/chat-ui/thread/chat-thread-shell.tsx"],"sourcesContent":["import { AuiIf, ThreadPrimitive, useAuiState } from \"@assistant-ui/react\";\nimport { deepAgentRowKey } from \"@athenaintel/agent-runtime\";\nimport { ChevronDownIcon } from \"lucide-react\";\nimport type { ComponentProps, CSSProperties, FC, ReactNode } from \"react\";\nimport { useMemo } from \"react\";\nimport {\n  THREAD_PANEL_CLASS,\n  THREAD_SCROLL_TO_BOTTOM_CLASS,\n} from \"../messages/message-components\";\nimport { AnswerLabelScope } from \"../messages/answer-label\";\nimport { MessageActionsReadOnlyProvider } from \"../messages/message-actions-config\";\nimport { UserEditComposer } from \"../messages/user-edit-composer\";\nimport { ThreadRunningIndicator } from \"../messages/running-indicator\";\nimport {\n  HandoffArrivalAfterMessage,\n  HandoffArrivalScope,\n} from \"../messages/handoff-arrival-divider\";\n\nexport interface ChatThreadMessageComponents {\n  UserMessage: FC;\n  AssistantMessage: FC;\n  /**\n   * Rendered in place of a user message while it is being edited (the action\n   * bar's Edit control). Defaults to the shared UserEditComposer.\n   */\n  UserEditComposer?: FC;\n}\n\n// One wrapper per message renderer, so a renderer keeps its identity across\n// shells and renders: a row renders its message through the component it is\n// handed, and a new component identity would remount every row.\nconst withArrivalDivider = (() => {\n  const wrapped = new WeakMap<FC, FC>();\n  return (Message: FC): FC => {\n    const existing = wrapped.get(Message);\n    if (existing) return existing;\n    const WithArrivalDivider: FC = () => (\n      <>\n        <Message />\n        <HandoffArrivalAfterMessage />\n      </>\n    );\n    wrapped.set(Message, WithArrivalDivider);\n    return WithArrivalDivider;\n  };\n})();\n\ntype ThreadRowComponents = ComponentProps<typeof ThreadPrimitive.MessageByIndex>[\"components\"];\n\nconst ROW_KEY_SEPARATOR = \"\\u0000\";\n\n/**\n * The thread's messages, one row each under its row key (`deepAgentRowKey`).\n *\n * `ThreadPrimitive.Messages` keys a row by its message id. A page of older\n * history that folds into the first loaded turn (a long run's earlier steps)\n * gives that turn a new id on every page, so keyed by id the turn's row\n * remounted once per page: it flashed through its entrance animation, closed\n * what the person had opened in it and took the browser's scroll anchor with\n * it. The row key stays with the row the turn continues, so the page lands in\n * the mounted row. A message from another runtime keys by its id, as before.\n */\nconst ThreadRows: FC<{ components: ThreadRowComponents }> = ({ components }) => {\n  const joinedKeys = useAuiState((s) =>\n    s.thread.messages.map((message) => deepAgentRowKey(message)).join(ROW_KEY_SEPARATOR),\n  );\n  const rowKeys = useMemo(\n    () => (joinedKeys === \"\" ? [] : joinedKeys.split(ROW_KEY_SEPARATOR)),\n    [joinedKeys],\n  );\n  return (\n    <>\n      {rowKeys.map((key, index) => (\n        <ThreadPrimitive.MessageByIndex key={key} index={index} components={components} />\n      ))}\n    </>\n  );\n};\n\nconst DEFAULT_VIEWPORT_CLASS =\n  \"relative flex flex-1 flex-col overflow-y-auto scroll-smooth\";\nconst DEFAULT_CONTENT_CLASS = \"w-full flex-1 px-3 pt-3\";\nconst DEFAULT_VIEWPORT_FOOTER_CLASS = \"sticky bottom-0\";\n\nconst DefaultScrollToBottom: FC = () => (\n  <AuiIf condition={(state) => !state.thread.isEmpty}>\n    <div className=\"flex justify-center pb-0.5\">\n      <ThreadPrimitive.ScrollToBottom\n        aria-label=\"Scroll to bottom\"\n        title=\"Scroll to bottom\"\n        className={THREAD_SCROLL_TO_BOTTOM_CLASS}\n      >\n        <ChevronDownIcon className=\"size-3.5\" />\n      </ThreadPrimitive.ScrollToBottom>\n    </div>\n  </AuiIf>\n);\n\nexport interface ChatThreadShellProps {\n  /**\n   * Rendered as the first child of the panel — for absolutely-positioned\n   * scrims that must cover the whole panel (AthenaChat's thread-loading\n   * overlay). The shell does not force a positioning context; hosts include\n   * `relative` in `panelClassName` when the scrim anchors to the panel.\n   */\n  overlay?: ReactNode;\n  /**\n   * Rendered above the viewport, inside the panel: header rows, slide-over\n   * panels, and anything else the host docks at the top.\n   */\n  header?: ReactNode;\n  /** Status/approval banners rendered between the header and the viewport. */\n  banners?: ReactNode;\n  /**\n   * Nodes rendered inside the scroll viewport before the message column —\n   * absolutely/sticky-positioned overlays (thread-switch skeletons,\n   * scroll-to-top controls) and invisible observers (read receipts).\n   */\n  viewportOverlay?: ReactNode;\n  /** Empty-thread state, rendered via ThreadPrimitive.Empty when provided. */\n  welcome?: ReactNode;\n  /**\n   * Rendered inside the message column right above the messages — the\n   * windowed wire's history edge (`StatewireHistoryEdge`): the \"N earlier\n   * messages\" row that pages older history in and fetches cut tool outputs.\n   */\n  beforeMessages?: ReactNode;\n  /** Message renderers for the thread's rows. */\n  components: ChatThreadMessageComponents;\n  /**\n   * Wraps the messages element — for hosts that provide message-scoped\n   * context (e.g. activity-grouping providers). Must render its argument.\n   */\n  renderMessages?: (messages: ReactNode) => ReactNode;\n  /**\n   * Rendered between the viewport and the composer: connection banners,\n   * queued-message trays.\n   */\n  belowViewport?: ReactNode;\n  /** The composer dock (or a read-only notice replacing it). */\n  composer?: ReactNode;\n  /**\n   * Replaces the docked-panel card chrome — for hosts that render a\n   * full-page column (AthenaChat's `aui-*` surface) instead of the\n   * side-panel/task-pane card.\n   */\n  panelClassName?: string;\n  /**\n   * Value of the `--thread-max-width` variable read by width-capped\n   * message/welcome/footer columns. Panels fill their width (the default,\n   * \"100%\"); full-page hosts cap the column (AthenaChat defaults to 44rem).\n   */\n  maxWidth?: string;\n  /**\n   * Anchors the latest turn to the top of the viewport instead of pinning\n   * scroll to the bottom (AthenaChat). Omitted = the primitive's default.\n   */\n  turnAnchor?: \"top\" | \"bottom\";\n  /** Replaces the scroll-viewport chrome (padding, overflow tuning). */\n  viewportClassName?: string;\n  /**\n   * Replaces the message-column chrome. Hosts whose welcome screen centers\n   * vertically make this a flex column so the welcome node can grow.\n   */\n  contentClassName?: string;\n  /**\n   * Thread-level \"Thinking...\" indicator rendered after the messages. Hosts\n   * whose message components render their own per-message indicators pass\n   * null so the two never double up.\n   */\n  runningIndicator?: ReactNode;\n  /**\n   * Replaces the default centered scroll-to-bottom pill inside the sticky\n   * viewport footer (AthenaChat floats a circular button above its composer).\n   */\n  scrollToBottom?: ReactNode;\n  /**\n   * Content docked inside the sticky viewport footer, after the\n   * scroll-to-bottom control — for hosts whose banners + composer float over\n   * the scrolling messages (AthenaChat) instead of sitting below the\n   * viewport via `belowViewport`/`composer`.\n   */\n  viewportFooter?: ReactNode;\n  /** Replaces the sticky viewport-footer chrome. */\n  viewportFooterClassName?: string;\n  /**\n   * Marks the transcript read-only: mutating message actions (Edit,\n   * Regenerate) hide while Copy stays available. For threads whose server\n   * rejects dispatch (e.g. legacy pre-cutover conversations) — the runtime's\n   * static capability flags cannot express a per-thread fence.\n   */\n  readOnly?: boolean;\n  /**\n   * The backend id of the thread on screen. When this surface picked the\n   * conversation up from another one (`recordHandoffArrival`), the transcript\n   * shows \"Continued from …\" after the last message that was already there.\n   */\n  arrivalThreadId?: string | null;\n  /**\n   * The signed-in viewer's Athena user id (`user_…`), for the answer labels\n   * (`AnswerLabel`): their own answers read \"for you\", and an answer for\n   * another person is labelled even in a 1:1 chat. Omitted, answers name the\n   * person by name and are labelled only once the thread holds several agents\n   * or several people.\n   */\n  viewerUserId?: string | null;\n}\n\n/**\n * The shared thread layout: panel chrome, scroll viewport with welcome +\n * messages + running indicator, the sticky scroll-to-bottom control, and the\n * host-supplied composer dock.\n *\n * Hosts express their genuine divergences through the slots above; the\n * layout, scroll behaviour, and empty/running states stay identical across\n * surfaces so fixes land once.\n */\nexport function ChatThreadShell({\n  overlay,\n  header,\n  banners,\n  viewportOverlay,\n  welcome,\n  beforeMessages,\n  components,\n  renderMessages,\n  belowViewport,\n  composer,\n  panelClassName,\n  maxWidth = \"100%\",\n  turnAnchor,\n  viewportClassName,\n  contentClassName,\n  runningIndicator = <ThreadRunningIndicator />,\n  scrollToBottom = <DefaultScrollToBottom />,\n  viewportFooter,\n  viewportFooterClassName,\n  readOnly = false,\n  arrivalThreadId = null,\n  viewerUserId = null,\n}: ChatThreadShellProps) {\n  const {\n    UserMessage,\n    AssistantMessage,\n    UserEditComposer: UserEditComposerComponent = UserEditComposer,\n  } = components;\n  // A row resolves its renderer by component identity, so keep the merged map\n  // referentially stable across renders.\n  const messageComponents = useMemo(\n    () => ({\n      UserMessage: withArrivalDivider(UserMessage),\n      AssistantMessage: withArrivalDivider(AssistantMessage),\n      UserEditComposer: UserEditComposerComponent,\n    }),\n    [UserMessage, AssistantMessage, UserEditComposerComponent],\n  );\n  const messages = (\n    <HandoffArrivalScope threadId={arrivalThreadId}>\n      <AnswerLabelScope viewerUserId={viewerUserId}>\n        <ThreadRows components={messageComponents} />\n      </AnswerLabelScope>\n    </HandoffArrivalScope>\n  );\n  return (\n    <MessageActionsReadOnlyProvider readOnly={readOnly}>\n      <ThreadPrimitive.Root\n        className={panelClassName ?? THREAD_PANEL_CLASS}\n        style={{ \"--thread-max-width\": maxWidth } as CSSProperties}\n      >\n        {overlay}\n        {header}\n        {banners}\n        <ThreadPrimitive.Viewport\n          turnAnchor={turnAnchor}\n          className={viewportClassName ?? DEFAULT_VIEWPORT_CLASS}\n          data-aui-thread-viewport=\"\"\n        >\n          {viewportOverlay}\n          {/* No min-h-0 here: the column's automatic minimum height is what\n            lets it extend past one screen, keeping the sticky footer's flow\n            position after the full message list. Capping it detaches the\n            footer (scroll pill, docked composers) as soon as the thread\n            scrolls — proven in a live-browser probe of both structures. */}\n          <div className=\"flex flex-1 flex-col items-center\">\n            <div className={contentClassName ?? DEFAULT_CONTENT_CLASS}>\n              {welcome != null ? (\n                <ThreadPrimitive.Empty>{welcome}</ThreadPrimitive.Empty>\n              ) : null}\n              {beforeMessages}\n              {renderMessages ? renderMessages(messages) : messages}\n              {runningIndicator}\n            </div>\n          </div>\n          <ThreadPrimitive.ViewportFooter\n            className={viewportFooterClassName ?? DEFAULT_VIEWPORT_FOOTER_CLASS}\n          >\n            {scrollToBottom}\n            {viewportFooter}\n          </ThreadPrimitive.ViewportFooter>\n        </ThreadPrimitive.Viewport>\n        {belowViewport}\n        {composer}\n      </ThreadPrimitive.Root>\n    </MessageActionsReadOnlyProvider>\n  );\n}\n"],"names":[],"mappings":";;;;;;;;;;;;AA+BA,MAAM,qBAAsB,uBAAM;AAChC,QAAM,8BAAc,QAAA;AACpB,SAAO,CAAC,YAAoB;AAC1B,UAAM,WAAW,QAAQ,IAAI,OAAO;AACpC,QAAI,SAAU,QAAO;AACrB,UAAM,qBAAyB,MAC7B,qBAAA,UAAA,EACE,UAAA;AAAA,MAAA,oBAAC,SAAA,EAAQ;AAAA,0BACR,4BAAA,CAAA,CAA2B;AAAA,IAAA,GAC9B;AAEF,YAAQ,IAAI,SAAS,kBAAkB;AACvC,WAAO;AAAA,EACT;AACF,GAAA;AAIA,MAAM,oBAAoB;AAa1B,MAAM,aAAsD,CAAC,EAAE,iBAAiB;AAC9E,QAAM,aAAa;AAAA,IAAY,CAAC,MAC9B,EAAE,OAAO,SAAS,IAAI,CAAC,YAAY,gBAAgB,OAAO,CAAC,EAAE,KAAK,iBAAiB;AAAA,EAAA;AAErF,QAAM,UAAU;AAAA,IACd,MAAO,eAAe,KAAK,CAAA,IAAK,WAAW,MAAM,iBAAiB;AAAA,IAClE,CAAC,UAAU;AAAA,EAAA;AAEb,SACE,oBAAA,UAAA,EACG,UAAA,QAAQ,IAAI,CAAC,KAAK,UACjB,oBAAC,gBAAgB,gBAAhB,EAAyC,OAAc,cAAnB,GAA2C,CACjF,GACH;AAEJ;AAEA,MAAM,yBACJ;AACF,MAAM,wBAAwB;AAC9B,MAAM,gCAAgC;AAEtC,MAAM,wBAA4B,MAChC,oBAAC,OAAA,EAAM,WAAW,CAAC,UAAU,CAAC,MAAM,OAAO,SACzC,UAAA,oBAAC,OAAA,EAAI,WAAU,8BACb,UAAA;AAAA,EAAC,gBAAgB;AAAA,EAAhB;AAAA,IACC,cAAW;AAAA,IACX,OAAM;AAAA,IACN,WAAW;AAAA,IAEX,UAAA,oBAAC,iBAAA,EAAgB,WAAU,WAAA,CAAW;AAAA,EAAA;AACxC,EAAA,CACF,EAAA,CACF;AA0HK,SAAS,gBAAgB;AAAA,EAC9B;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA,WAAW;AAAA,EACX;AAAA,EACA;AAAA,EACA;AAAA,EACA,uCAAoB,wBAAA,EAAuB;AAAA,EAC3C,qCAAkB,uBAAA,EAAsB;AAAA,EACxC;AAAA,EACA;AAAA,EACA,WAAW;AAAA,EACX,kBAAkB;AAAA,EAClB,eAAe;AACjB,GAAyB;AACvB,QAAM;AAAA,IACJ;AAAA,IACA;AAAA,IACA,kBAAkB,4BAA4B;AAAA,EAAA,IAC5C;AAGJ,QAAM,oBAAoB;AAAA,IACxB,OAAO;AAAA,MACL,aAAa,mBAAmB,WAAW;AAAA,MAC3C,kBAAkB,mBAAmB,gBAAgB;AAAA,MACrD,kBAAkB;AAAA,IAAA;AAAA,IAEpB,CAAC,aAAa,kBAAkB,yBAAyB;AAAA,EAAA;AAE3D,QAAM,WACJ,oBAAC,qBAAA,EAAoB,UAAU,iBAC7B,UAAA,oBAAC,kBAAA,EAAiB,cAChB,UAAA,oBAAC,YAAA,EAAW,YAAY,kBAAA,CAAmB,GAC7C,GACF;AAEF,SACE,oBAAC,kCAA+B,UAC9B,UAAA;AAAA,IAAC,gBAAgB;AAAA,IAAhB;AAAA,MACC,WAAW,kBAAkB;AAAA,MAC7B,OAAO,EAAE,sBAAsB,SAAA;AAAA,MAE9B,UAAA;AAAA,QAAA;AAAA,QACA;AAAA,QACA;AAAA,QACD;AAAA,UAAC,gBAAgB;AAAA,UAAhB;AAAA,YACC;AAAA,YACA,WAAW,qBAAqB;AAAA,YAChC,4BAAyB;AAAA,YAExB,UAAA;AAAA,cAAA;AAAA,cAMD,oBAAC,SAAI,WAAU,qCACb,+BAAC,OAAA,EAAI,WAAW,oBAAoB,uBACjC,UAAA;AAAA,gBAAA,WAAW,OACV,oBAAC,gBAAgB,OAAhB,EAAuB,mBAAQ,IAC9B;AAAA,gBACH;AAAA,gBACA,iBAAiB,eAAe,QAAQ,IAAI;AAAA,gBAC5C;AAAA,cAAA,EAAA,CACH,EAAA,CACF;AAAA,cACA;AAAA,gBAAC,gBAAgB;AAAA,gBAAhB;AAAA,kBACC,WAAW,2BAA2B;AAAA,kBAErC,UAAA;AAAA,oBAAA;AAAA,oBACA;AAAA,kBAAA;AAAA,gBAAA;AAAA,cAAA;AAAA,YACH;AAAA,UAAA;AAAA,QAAA;AAAA,QAED;AAAA,QACA;AAAA,MAAA;AAAA,IAAA;AAAA,EAAA,GAEL;AAEJ;"}

package/dist/chat-ui/thread/composer-send-or-stop.cjs

@@ -3,13 +3,14 @@ Object.defineProperty(exports, Symbol.toStringTag, { value: "Module" });
 const jsxRuntime = require("react/jsx-runtime");
 const athenaDiagnostics = require("../../diagnostics/athena-diagnostics.cjs");
 const react = require("@assistant-ui/react");
-const react$1 = require("react");
+const require$$0 = require("react");
 const composerAction = require("../composer/composer-action.cjs");
 const messageComponents = require("../messages/message-components.cjs");
 const ComposerSendOrStop = ({
   composerRef,
   hasDraft,
   liveVoice = false,
+  liveSession = null,
   disabled = false,
   sendBlockedReason = null,
   sendDisabledTitle,
@@ -25,10 +26,10 @@ const ComposerSendOrStop = ({
       return ((_a = state.optional.thread) == null ? void 0 : _a.isRunning) ?? false;
     }
   );
-  const action = liveVoice && hasDraft ? "send" : composerAction.resolveComposerAction({ hasDraft, isRunning });
+  const action = liveVoice && hasDraft ? "send" : !hasDraft && liveSession !== null ? "stop" : composerAction.resolveComposerAction({ hasDraft, isRunning });
   const SendButton = (components == null ? void 0 : components.SendButton) ?? messageComponents.ComposerSendButton;
   const StopButton = (components == null ? void 0 : components.StopButton) ?? messageComponents.ComposerStopButton;
-  const cancelRun = react$1.useCallback(() => {
+  const cancelRun = require$$0.useCallback(() => {
     try {
       aui.thread.cancelRun();
       return true;
@@ -40,29 +41,31 @@ const ComposerSendOrStop = ({
       return false;
     }
   }, [aui]);
-  if (action === "stop") {
-    return /* @__PURE__ */ jsxRuntime.jsx(
-      StopButton,
-      {
-        onStop: () => {
-          if (onStop) {
-            onStop({ cancelRun });
-          } else {
-            cancelRun();
-          }
-        },
-        disabled,
-        title: disabled && stopDisabledTitle ? stopDisabledTitle : void 0
-      }
-    );
-  }
+  const stopRun = () => isRunning ? cancelRun() : true;
+  const stopEverything = () => {
+    liveSession == null ? void 0 : liveSession.end();
+    if (onStop) {
+      onStop({ cancelRun: stopRun });
+    } else {
+      stopRun();
+    }
+  };
+  const stopControl = /* @__PURE__ */ jsxRuntime.jsx(
+    StopButton,
+    {
+      onStop: stopEverything,
+      disabled,
+      title: disabled && stopDisabledTitle ? stopDisabledTitle : liveSession ? "Stop" : void 0
+    }
+  );
+  if (action === "stop") return stopControl;
   const handleSend = async () => {
     var _a;
     await (onBeforeSend == null ? void 0 : onBeforeSend({ action }));
     await ((_a = composerRef.current) == null ? void 0 : _a.submit());
   };
   const sendTitle = disabled && sendDisabledTitle ? sendDisabledTitle : sendBlockedReason ?? void 0;
-  return /* @__PURE__ */ jsxRuntime.jsx(
+  const sendControl = /* @__PURE__ */ jsxRuntime.jsx(
     SendButton,
     {
       action,
@@ -71,6 +74,13 @@ const ComposerSendOrStop = ({
       title: sendTitle
     }
   );
+  if (liveSession !== null) {
+    return /* @__PURE__ */ jsxRuntime.jsxs(jsxRuntime.Fragment, { children: [
+      stopControl,
+      sendControl
+    ] });
+  }
+  return sendControl;
 };
 exports.ComposerSendOrStop = ComposerSendOrStop;
 //# sourceMappingURL=composer-send-or-stop.cjs.map

package/dist/chat-ui/thread/composer-send-or-stop.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"composer-send-or-stop.cjs","sources":["../../../src/chat-ui/thread/composer-send-or-stop.tsx"],"sourcesContent":["import { athenaDiagnostics } from \"../../diagnostics/athena-diagnostics\";\nimport { useAui, useAuiState } from \"@assistant-ui/react\";\nimport type { FC, RefObject } from \"react\";\nimport { useCallback } from \"react\";\nimport {\n  resolveComposerAction,\n  type ComposerAction,\n} from \"../composer/composer-action\";\nimport {\n  ComposerSendButton,\n  ComposerStopButton,\n  type ComposerSendButtonProps,\n  type ComposerStopButtonProps,\n} from \"../messages/message-components\";\n\n/**\n * The slice of a composer handle this control needs: something whose draft\n * can be submitted. `TiptapComposerHandle` satisfies it; hosts with their own\n * editor plumbing (the SDK's AthenaChat) expose a compatible handle.\n */\nexport interface ComposerSubmitHandle {\n  submit: () => void | Promise<void>;\n}\n\nexport interface ComposerStopControls {\n  /**\n   * Cancels the active run via the runtime. Returns false when cancellation\n   * threw (e.g. the run already settled), so hosts can report whether their\n   * own teardown still needs to stop backend work.\n   */\n  cancelRun: () => boolean;\n}\n\n/**\n * Control renderers, for hosts with their own published theming contract\n * (AthenaChat's `aui-*` class system). Defaults render the shared buttons.\n */\nexport interface ComposerSendOrStopComponents {\n  SendButton?: FC<ComposerSendButtonProps>;\n  StopButton?: FC<ComposerStopButtonProps>;\n}\n\nexport interface ComposerSendOrStopProps {\n  composerRef: RefObject<ComposerSubmitHandle | null>;\n  hasDraft: boolean;\n  /** A plain draft enters the live voice conversation instead of the run queue. */\n  liveVoice?: boolean;\n  /** Disables both send and stop (e.g. while a conversation restores). */\n  disabled?: boolean;\n  /**\n   * Blocks sending (not stopping) with this reason as the tooltip — e.g. a\n   * broken host tunnel that would make tool calls silently fail.\n   */\n  sendBlockedReason?: string | null;\n  /** Tooltip for the send/queue control while `disabled`. */\n  sendDisabledTitle?: string;\n  /** Tooltip for the stop control while `disabled`. */\n  stopDisabledTitle?: string;\n  /**\n   * Host work to run before the draft submits (per-turn bookkeeping).\n   * `action` distinguishes a fresh send from queueing into an active run.\n   */\n  onBeforeSend?: (context: {\n    action: Extract<ComposerAction, \"send\" | \"queue\">;\n  }) => void | Promise<void>;\n  /**\n   * Full custom stop sequence. Hosts that must tear down in-flight host-side\n   * work sequence it around `controls.cancelRun()`. Default: cancel the run.\n   */\n  onStop?: (controls: ComposerStopControls) => void;\n  components?: ComposerSendOrStopComponents;\n}\n\n/**\n * The shared send/queue/stop control: sends the composer draft, queues it as\n * a follow-up while a run is in flight, or stops the run when there is\n * nothing to queue.\n */\nexport const ComposerSendOrStop: FC<ComposerSendOrStopProps> = ({\n  composerRef,\n  hasDraft,\n  liveVoice = false,\n  disabled = false,\n  sendBlockedReason = null,\n  sendDisabledTitle,\n  stopDisabledTitle,\n  onBeforeSend,\n  onStop,\n  components,\n}) => {\n  const aui = useAui();\n  const isRunning = useAuiState(\n    (state) => state.optional.thread?.isRunning ?? false,\n  );\n  const action = liveVoice && hasDraft ? 'send' : resolveComposerAction({ hasDraft, isRunning });\n  const SendButton = components?.SendButton ?? ComposerSendButton;\n  const StopButton = components?.StopButton ?? ComposerStopButton;\n\n  const cancelRun = useCallback((): boolean => {\n    try {\n      aui.thread.cancelRun();\n      return true;\n    } catch (error) {\n      athenaDiagnostics.emit(\"composer.cancel.error\", {\n        level: \"warn\",\n        data: { error: error instanceof Error ? error.message : String(error) },\n      });\n      return false;\n    }\n  }, [aui]);\n\n  if (action === \"stop\") {\n    return (\n      <StopButton\n        onStop={() => {\n          if (onStop) {\n            onStop({ cancelRun });\n          } else {\n            cancelRun();\n          }\n        }}\n        disabled={disabled}\n        title={disabled && stopDisabledTitle ? stopDisabledTitle : undefined}\n      />\n    );\n  }\n\n  const handleSend = async () => {\n    await onBeforeSend?.({ action });\n    await composerRef.current?.submit();\n  };\n  const sendTitle =\n    disabled && sendDisabledTitle\n      ? sendDisabledTitle\n      : (sendBlockedReason ?? undefined);\n  return (\n    <SendButton\n      action={action}\n      onSend={() => void handleSend()}\n      disabled={disabled || !hasDraft || sendBlockedReason != null}\n      title={sendTitle}\n    />\n  );\n};\n"],"names":["useAui","useAuiState","resolveComposerAction","ComposerSendButton","ComposerStopButton","useCallback","athenaDiagnostics","jsx"],"mappings":";;;;;;;;AA8EO,MAAM,qBAAkD,CAAC;AAAA,EAC9D;AAAA,EACA;AAAA,EACA,YAAY;AAAA,EACZ,WAAW;AAAA,EACX,oBAAoB;AAAA,EACpB;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF,MAAM;AACJ,QAAM,MAAMA,MAAAA,OAAA;AACZ,QAAM,YAAYC,MAAAA;AAAAA,IAChB,CAAC,UAAA;;AAAU,0BAAM,SAAS,WAAf,mBAAuB,cAAa;AAAA;AAAA,EAAA;AAEjD,QAAM,SAAS,aAAa,WAAW,SAASC,eAAAA,sBAAsB,EAAE,UAAU,WAAW;AAC7F,QAAM,cAAa,yCAAY,eAAcC,kBAAAA;AAC7C,QAAM,cAAa,yCAAY,eAAcC,kBAAAA;AAE7C,QAAM,YAAYC,QAAAA,YAAY,MAAe;AAC3C,QAAI;AACF,UAAI,OAAO,UAAA;AACX,aAAO;AAAA,IACT,SAAS,OAAO;AACdC,wBAAAA,kBAAkB,KAAK,yBAAyB;AAAA,QAC9C,OAAO;AAAA,QACP,MAAM,EAAE,OAAO,iBAAiB,QAAQ,MAAM,UAAU,OAAO,KAAK,EAAA;AAAA,MAAE,CACvE;AACD,aAAO;AAAA,IACT;AAAA,EACF,GAAG,CAAC,GAAG,CAAC;AAER,MAAI,WAAW,QAAQ;AACrB,WACEC,2BAAAA;AAAAA,MAAC;AAAA,MAAA;AAAA,QACC,QAAQ,MAAM;AACZ,cAAI,QAAQ;AACV,mBAAO,EAAE,WAAW;AAAA,UACtB,OAAO;AACL,sBAAA;AAAA,UACF;AAAA,QACF;AAAA,QACA;AAAA,QACA,OAAO,YAAY,oBAAoB,oBAAoB;AAAA,MAAA;AAAA,IAAA;AAAA,EAGjE;AAEA,QAAM,aAAa,YAAY;;AAC7B,WAAM,6CAAe,EAAE;AACvB,YAAM,iBAAY,YAAZ,mBAAqB;AAAA,EAC7B;AACA,QAAM,YACJ,YAAY,oBACR,oBACC,qBAAqB;AAC5B,SACEA,2BAAAA;AAAAA,IAAC;AAAA,IAAA;AAAA,MACC;AAAA,MACA,QAAQ,MAAM,KAAK,WAAA;AAAA,MACnB,UAAU,YAAY,CAAC,YAAY,qBAAqB;AAAA,MACxD,OAAO;AAAA,IAAA;AAAA,EAAA;AAGb;;"}
+{"version":3,"file":"composer-send-or-stop.cjs","sources":["../../../src/chat-ui/thread/composer-send-or-stop.tsx"],"sourcesContent":["import { athenaDiagnostics } from \"../../diagnostics/athena-diagnostics\";\nimport { useAui, useAuiState } from \"@assistant-ui/react\";\nimport type { FC, RefObject } from \"react\";\nimport { useCallback } from \"react\";\nimport {\n  resolveComposerAction,\n  type ComposerAction,\n} from \"../composer/composer-action\";\nimport {\n  ComposerSendButton,\n  ComposerStopButton,\n  type ComposerSendButtonProps,\n  type ComposerStopButtonProps,\n} from \"../messages/message-components\";\n\n/**\n * The slice of a composer handle this control needs: something whose draft\n * can be submitted. `TiptapComposerHandle` satisfies it; hosts with their own\n * editor plumbing (the SDK's AthenaChat) expose a compatible handle.\n */\nexport interface ComposerSubmitHandle {\n  submit: () => void | Promise<void>;\n}\n\nexport interface ComposerStopControls {\n  /**\n   * Cancels the active run via the runtime. Returns false when cancellation\n   * threw (e.g. the run already settled), so hosts can report whether their\n   * own teardown still needs to stop backend work.\n   */\n  cancelRun: () => boolean;\n}\n\n/**\n * Control renderers, for hosts with their own published theming contract\n * (AthenaChat's `aui-*` class system). Defaults render the shared buttons.\n */\nexport interface ComposerSendOrStopComponents {\n  SendButton?: FC<ComposerSendButtonProps>;\n  StopButton?: FC<ComposerStopButtonProps>;\n}\n\nexport interface ComposerSendOrStopProps {\n  composerRef: RefObject<ComposerSubmitHandle | null>;\n  hasDraft: boolean;\n  /** A plain draft enters the live voice conversation instead of the run queue. */\n  liveVoice?: boolean;\n  /**\n   * A live session (voice, connecting or live) the Stop control also ends.\n   * With it, an empty draft shows Stop even when no run is in flight, and\n   * Stop ends the session AND cancels any run: one control stops everything.\n   */\n  liveSession?: { end: () => void } | null;\n  /** Disables both send and stop (e.g. while a conversation restores). */\n  disabled?: boolean;\n  /**\n   * Blocks sending (not stopping) with this reason as the tooltip — e.g. a\n   * broken host tunnel that would make tool calls silently fail.\n   */\n  sendBlockedReason?: string | null;\n  /** Tooltip for the send/queue control while `disabled`. */\n  sendDisabledTitle?: string;\n  /** Tooltip for the stop control while `disabled`. */\n  stopDisabledTitle?: string;\n  /**\n   * Host work to run before the draft submits (per-turn bookkeeping).\n   * `action` distinguishes a fresh send from queueing into an active run.\n   */\n  onBeforeSend?: (context: {\n    action: Extract<ComposerAction, \"send\" | \"queue\">;\n  }) => void | Promise<void>;\n  /**\n   * Full custom stop sequence. Hosts that must tear down in-flight host-side\n   * work sequence it around `controls.cancelRun()`. Default: cancel the run.\n   */\n  onStop?: (controls: ComposerStopControls) => void;\n  components?: ComposerSendOrStopComponents;\n}\n\n/**\n * The shared send/queue/stop control: sends the composer draft, queues it as\n * a follow-up while a run is in flight, or stops the run when there is\n * nothing to queue.\n */\nexport const ComposerSendOrStop: FC<ComposerSendOrStopProps> = ({\n  composerRef,\n  hasDraft,\n  liveVoice = false,\n  liveSession = null,\n  disabled = false,\n  sendBlockedReason = null,\n  sendDisabledTitle,\n  stopDisabledTitle,\n  onBeforeSend,\n  onStop,\n  components,\n}) => {\n  const aui = useAui();\n  const isRunning = useAuiState(\n    (state) => state.optional.thread?.isRunning ?? false,\n  );\n  // A draft sends (into the live voice conversation, or as a run) or queues\n  // behind a run; only an empty draft turns the control into Stop, and a\n  // live session alone is enough for that.\n  const action: ComposerAction =\n    liveVoice && hasDraft\n      ? \"send\"\n      : !hasDraft && liveSession !== null\n        ? \"stop\"\n        : resolveComposerAction({ hasDraft, isRunning });\n  const SendButton = components?.SendButton ?? ComposerSendButton;\n  const StopButton = components?.StopButton ?? ComposerStopButton;\n\n  const cancelRun = useCallback((): boolean => {\n    try {\n      aui.thread.cancelRun();\n      return true;\n    } catch (error) {\n      athenaDiagnostics.emit(\"composer.cancel.error\", {\n        level: \"warn\",\n        data: { error: error instanceof Error ? error.message : String(error) },\n      });\n      return false;\n    }\n  }, [aui]);\n\n  // With only a session to end, a cancel would find no run to stop.\n  const stopRun = (): boolean => (isRunning ? cancelRun() : true);\n  const stopEverything = () => {\n    liveSession?.end();\n    if (onStop) {\n      onStop({ cancelRun: stopRun });\n    } else {\n      stopRun();\n    }\n  };\n  const stopControl = (\n    <StopButton\n      onStop={stopEverything}\n      disabled={disabled}\n      title={\n        disabled && stopDisabledTitle\n          ? stopDisabledTitle\n          : liveSession\n            ? \"Stop\"\n            : undefined\n      }\n    />\n  );\n\n  if (action === \"stop\") return stopControl;\n\n  const handleSend = async () => {\n    await onBeforeSend?.({ action });\n    await composerRef.current?.submit();\n  };\n  const sendTitle =\n    disabled && sendDisabledTitle\n      ? sendDisabledTitle\n      : (sendBlockedReason ?? undefined);\n  const sendControl = (\n    <SendButton\n      action={action}\n      onSend={() => void handleSend()}\n      disabled={disabled || !hasDraft || sendBlockedReason != null}\n      title={sendTitle}\n    />\n  );\n  // A live session keeps its Stop beside Send while a draft is typed, so the\n  // call (and any run) can end without sending or discarding the words; the\n  // draft stays in the composer.\n  if (liveSession !== null) {\n    return (\n      <>\n        {stopControl}\n        {sendControl}\n      </>\n    );\n  }\n  return sendControl;\n};\n"],"names":["useAui","useAuiState","resolveComposerAction","ComposerSendButton","ComposerStopButton","useCallback","athenaDiagnostics","jsx","jsxs","Fragment"],"mappings":";;;;;;;;AAoFO,MAAM,qBAAkD,CAAC;AAAA,EAC9D;AAAA,EACA;AAAA,EACA,YAAY;AAAA,EACZ,cAAc;AAAA,EACd,WAAW;AAAA,EACX,oBAAoB;AAAA,EACpB;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF,MAAM;AACJ,QAAM,MAAMA,MAAAA,OAAA;AACZ,QAAM,YAAYC,MAAAA;AAAAA,IAChB,CAAC,UAAA;;AAAU,0BAAM,SAAS,WAAf,mBAAuB,cAAa;AAAA;AAAA,EAAA;AAKjD,QAAM,SACJ,aAAa,WACT,SACA,CAAC,YAAY,gBAAgB,OAC3B,SACAC,eAAAA,sBAAsB,EAAE,UAAU,WAAW;AACrD,QAAM,cAAa,yCAAY,eAAcC,kBAAAA;AAC7C,QAAM,cAAa,yCAAY,eAAcC,kBAAAA;AAE7C,QAAM,YAAYC,WAAAA,YAAY,MAAe;AAC3C,QAAI;AACF,UAAI,OAAO,UAAA;AACX,aAAO;AAAA,IACT,SAAS,OAAO;AACdC,wBAAAA,kBAAkB,KAAK,yBAAyB;AAAA,QAC9C,OAAO;AAAA,QACP,MAAM,EAAE,OAAO,iBAAiB,QAAQ,MAAM,UAAU,OAAO,KAAK,EAAA;AAAA,MAAE,CACvE;AACD,aAAO;AAAA,IACT;AAAA,EACF,GAAG,CAAC,GAAG,CAAC;AAGR,QAAM,UAAU,MAAgB,YAAY,UAAA,IAAc;AAC1D,QAAM,iBAAiB,MAAM;AAC3B,+CAAa;AACb,QAAI,QAAQ;AACV,aAAO,EAAE,WAAW,SAAS;AAAA,IAC/B,OAAO;AACL,cAAA;AAAA,IACF;AAAA,EACF;AACA,QAAM,cACJC,2BAAAA;AAAAA,IAAC;AAAA,IAAA;AAAA,MACC,QAAQ;AAAA,MACR;AAAA,MACA,OACE,YAAY,oBACR,oBACA,cACE,SACA;AAAA,IAAA;AAAA,EAAA;AAKZ,MAAI,WAAW,OAAQ,QAAO;AAE9B,QAAM,aAAa,YAAY;;AAC7B,WAAM,6CAAe,EAAE;AACvB,YAAM,iBAAY,YAAZ,mBAAqB;AAAA,EAC7B;AACA,QAAM,YACJ,YAAY,oBACR,oBACC,qBAAqB;AAC5B,QAAM,cACJA,2BAAAA;AAAAA,IAAC;AAAA,IAAA;AAAA,MACC;AAAA,MACA,QAAQ,MAAM,KAAK,WAAA;AAAA,MACnB,UAAU,YAAY,CAAC,YAAY,qBAAqB;AAAA,MACxD,OAAO;AAAA,IAAA;AAAA,EAAA;AAMX,MAAI,gBAAgB,MAAM;AACxB,WACEC,2BAAAA,KAAAC,qBAAA,EACG,UAAA;AAAA,MAAA;AAAA,MACA;AAAA,IAAA,GACH;AAAA,EAEJ;AACA,SAAO;AACT;;"}

package/dist/chat-ui/thread/composer-send-or-stop.d.ts

@@ -30,6 +30,14 @@ export interface ComposerSendOrStopProps {
     hasDraft: boolean;
     /** A plain draft enters the live voice conversation instead of the run queue. */
     liveVoice?: boolean;
+    /**
+     * A live session (voice, connecting or live) the Stop control also ends.
+     * With it, an empty draft shows Stop even when no run is in flight, and
+     * Stop ends the session AND cancels any run: one control stops everything.
+     */
+    liveSession?: {
+        end: () => void;
+    } | null;
     /** Disables both send and stop (e.g. while a conversation restores). */
     disabled?: boolean;
     /**

package/dist/chat-ui/thread/composer-send-or-stop.js

@@ -1,4 +1,4 @@
-import { jsx } from "react/jsx-runtime";
+import { jsxs, Fragment, jsx } from "react/jsx-runtime";
 import { athenaDiagnostics } from "../../diagnostics/athena-diagnostics.js";
 import { useAui, useAuiState } from "@assistant-ui/react";
 import { useCallback } from "react";
@@ -8,6 +8,7 @@ const ComposerSendOrStop = ({
   composerRef,
   hasDraft,
   liveVoice = false,
+  liveSession = null,
   disabled = false,
   sendBlockedReason = null,
   sendDisabledTitle,
@@ -23,7 +24,7 @@ const ComposerSendOrStop = ({
       return ((_a = state.optional.thread) == null ? void 0 : _a.isRunning) ?? false;
     }
   );
-  const action = liveVoice && hasDraft ? "send" : resolveComposerAction({ hasDraft, isRunning });
+  const action = liveVoice && hasDraft ? "send" : !hasDraft && liveSession !== null ? "stop" : resolveComposerAction({ hasDraft, isRunning });
   const SendButton = (components == null ? void 0 : components.SendButton) ?? ComposerSendButton;
   const StopButton = (components == null ? void 0 : components.StopButton) ?? ComposerStopButton;
   const cancelRun = useCallback(() => {
@@ -38,29 +39,31 @@ const ComposerSendOrStop = ({
       return false;
     }
   }, [aui]);
-  if (action === "stop") {
-    return /* @__PURE__ */ jsx(
-      StopButton,
-      {
-        onStop: () => {
-          if (onStop) {
-            onStop({ cancelRun });
-          } else {
-            cancelRun();
-          }
-        },
-        disabled,
-        title: disabled && stopDisabledTitle ? stopDisabledTitle : void 0
-      }
-    );
-  }
+  const stopRun = () => isRunning ? cancelRun() : true;
+  const stopEverything = () => {
+    liveSession == null ? void 0 : liveSession.end();
+    if (onStop) {
+      onStop({ cancelRun: stopRun });
+    } else {
+      stopRun();
+    }
+  };
+  const stopControl = /* @__PURE__ */ jsx(
+    StopButton,
+    {
+      onStop: stopEverything,
+      disabled,
+      title: disabled && stopDisabledTitle ? stopDisabledTitle : liveSession ? "Stop" : void 0
+    }
+  );
+  if (action === "stop") return stopControl;
   const handleSend = async () => {
     var _a;
     await (onBeforeSend == null ? void 0 : onBeforeSend({ action }));
     await ((_a = composerRef.current) == null ? void 0 : _a.submit());
   };
   const sendTitle = disabled && sendDisabledTitle ? sendDisabledTitle : sendBlockedReason ?? void 0;
-  return /* @__PURE__ */ jsx(
+  const sendControl = /* @__PURE__ */ jsx(
     SendButton,
     {
       action,
@@ -69,6 +72,13 @@ const ComposerSendOrStop = ({
       title: sendTitle
     }
   );
+  if (liveSession !== null) {
+    return /* @__PURE__ */ jsxs(Fragment, { children: [
+      stopControl,
+      sendControl
+    ] });
+  }
+  return sendControl;
 };
 export {
   ComposerSendOrStop

package/dist/chat-ui/thread/composer-send-or-stop.js.map

@@ -1 +1 @@
-{"version":3,"file":"composer-send-or-stop.js","sources":["../../../src/chat-ui/thread/composer-send-or-stop.tsx"],"sourcesContent":["import { athenaDiagnostics } from \"../../diagnostics/athena-diagnostics\";\nimport { useAui, useAuiState } from \"@assistant-ui/react\";\nimport type { FC, RefObject } from \"react\";\nimport { useCallback } from \"react\";\nimport {\n  resolveComposerAction,\n  type ComposerAction,\n} from \"../composer/composer-action\";\nimport {\n  ComposerSendButton,\n  ComposerStopButton,\n  type ComposerSendButtonProps,\n  type ComposerStopButtonProps,\n} from \"../messages/message-components\";\n\n/**\n * The slice of a composer handle this control needs: something whose draft\n * can be submitted. `TiptapComposerHandle` satisfies it; hosts with their own\n * editor plumbing (the SDK's AthenaChat) expose a compatible handle.\n */\nexport interface ComposerSubmitHandle {\n  submit: () => void | Promise<void>;\n}\n\nexport interface ComposerStopControls {\n  /**\n   * Cancels the active run via the runtime. Returns false when cancellation\n   * threw (e.g. the run already settled), so hosts can report whether their\n   * own teardown still needs to stop backend work.\n   */\n  cancelRun: () => boolean;\n}\n\n/**\n * Control renderers, for hosts with their own published theming contract\n * (AthenaChat's `aui-*` class system). Defaults render the shared buttons.\n */\nexport interface ComposerSendOrStopComponents {\n  SendButton?: FC<ComposerSendButtonProps>;\n  StopButton?: FC<ComposerStopButtonProps>;\n}\n\nexport interface ComposerSendOrStopProps {\n  composerRef: RefObject<ComposerSubmitHandle | null>;\n  hasDraft: boolean;\n  /** A plain draft enters the live voice conversation instead of the run queue. */\n  liveVoice?: boolean;\n  /** Disables both send and stop (e.g. while a conversation restores). */\n  disabled?: boolean;\n  /**\n   * Blocks sending (not stopping) with this reason as the tooltip — e.g. a\n   * broken host tunnel that would make tool calls silently fail.\n   */\n  sendBlockedReason?: string | null;\n  /** Tooltip for the send/queue control while `disabled`. */\n  sendDisabledTitle?: string;\n  /** Tooltip for the stop control while `disabled`. */\n  stopDisabledTitle?: string;\n  /**\n   * Host work to run before the draft submits (per-turn bookkeeping).\n   * `action` distinguishes a fresh send from queueing into an active run.\n   */\n  onBeforeSend?: (context: {\n    action: Extract<ComposerAction, \"send\" | \"queue\">;\n  }) => void | Promise<void>;\n  /**\n   * Full custom stop sequence. Hosts that must tear down in-flight host-side\n   * work sequence it around `controls.cancelRun()`. Default: cancel the run.\n   */\n  onStop?: (controls: ComposerStopControls) => void;\n  components?: ComposerSendOrStopComponents;\n}\n\n/**\n * The shared send/queue/stop control: sends the composer draft, queues it as\n * a follow-up while a run is in flight, or stops the run when there is\n * nothing to queue.\n */\nexport const ComposerSendOrStop: FC<ComposerSendOrStopProps> = ({\n  composerRef,\n  hasDraft,\n  liveVoice = false,\n  disabled = false,\n  sendBlockedReason = null,\n  sendDisabledTitle,\n  stopDisabledTitle,\n  onBeforeSend,\n  onStop,\n  components,\n}) => {\n  const aui = useAui();\n  const isRunning = useAuiState(\n    (state) => state.optional.thread?.isRunning ?? false,\n  );\n  const action = liveVoice && hasDraft ? 'send' : resolveComposerAction({ hasDraft, isRunning });\n  const SendButton = components?.SendButton ?? ComposerSendButton;\n  const StopButton = components?.StopButton ?? ComposerStopButton;\n\n  const cancelRun = useCallback((): boolean => {\n    try {\n      aui.thread.cancelRun();\n      return true;\n    } catch (error) {\n      athenaDiagnostics.emit(\"composer.cancel.error\", {\n        level: \"warn\",\n        data: { error: error instanceof Error ? error.message : String(error) },\n      });\n      return false;\n    }\n  }, [aui]);\n\n  if (action === \"stop\") {\n    return (\n      <StopButton\n        onStop={() => {\n          if (onStop) {\n            onStop({ cancelRun });\n          } else {\n            cancelRun();\n          }\n        }}\n        disabled={disabled}\n        title={disabled && stopDisabledTitle ? stopDisabledTitle : undefined}\n      />\n    );\n  }\n\n  const handleSend = async () => {\n    await onBeforeSend?.({ action });\n    await composerRef.current?.submit();\n  };\n  const sendTitle =\n    disabled && sendDisabledTitle\n      ? sendDisabledTitle\n      : (sendBlockedReason ?? undefined);\n  return (\n    <SendButton\n      action={action}\n      onSend={() => void handleSend()}\n      disabled={disabled || !hasDraft || sendBlockedReason != null}\n      title={sendTitle}\n    />\n  );\n};\n"],"names":[],"mappings":";;;;;;AA8EO,MAAM,qBAAkD,CAAC;AAAA,EAC9D;AAAA,EACA;AAAA,EACA,YAAY;AAAA,EACZ,WAAW;AAAA,EACX,oBAAoB;AAAA,EACpB;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF,MAAM;AACJ,QAAM,MAAM,OAAA;AACZ,QAAM,YAAY;AAAA,IAChB,CAAC,UAAA;;AAAU,0BAAM,SAAS,WAAf,mBAAuB,cAAa;AAAA;AAAA,EAAA;AAEjD,QAAM,SAAS,aAAa,WAAW,SAAS,sBAAsB,EAAE,UAAU,WAAW;AAC7F,QAAM,cAAa,yCAAY,eAAc;AAC7C,QAAM,cAAa,yCAAY,eAAc;AAE7C,QAAM,YAAY,YAAY,MAAe;AAC3C,QAAI;AACF,UAAI,OAAO,UAAA;AACX,aAAO;AAAA,IACT,SAAS,OAAO;AACd,wBAAkB,KAAK,yBAAyB;AAAA,QAC9C,OAAO;AAAA,QACP,MAAM,EAAE,OAAO,iBAAiB,QAAQ,MAAM,UAAU,OAAO,KAAK,EAAA;AAAA,MAAE,CACvE;AACD,aAAO;AAAA,IACT;AAAA,EACF,GAAG,CAAC,GAAG,CAAC;AAER,MAAI,WAAW,QAAQ;AACrB,WACE;AAAA,MAAC;AAAA,MAAA;AAAA,QACC,QAAQ,MAAM;AACZ,cAAI,QAAQ;AACV,mBAAO,EAAE,WAAW;AAAA,UACtB,OAAO;AACL,sBAAA;AAAA,UACF;AAAA,QACF;AAAA,QACA;AAAA,QACA,OAAO,YAAY,oBAAoB,oBAAoB;AAAA,MAAA;AAAA,IAAA;AAAA,EAGjE;AAEA,QAAM,aAAa,YAAY;;AAC7B,WAAM,6CAAe,EAAE;AACvB,YAAM,iBAAY,YAAZ,mBAAqB;AAAA,EAC7B;AACA,QAAM,YACJ,YAAY,oBACR,oBACC,qBAAqB;AAC5B,SACE;AAAA,IAAC;AAAA,IAAA;AAAA,MACC;AAAA,MACA,QAAQ,MAAM,KAAK,WAAA;AAAA,MACnB,UAAU,YAAY,CAAC,YAAY,qBAAqB;AAAA,MACxD,OAAO;AAAA,IAAA;AAAA,EAAA;AAGb;"}
+{"version":3,"file":"composer-send-or-stop.js","sources":["../../../src/chat-ui/thread/composer-send-or-stop.tsx"],"sourcesContent":["import { athenaDiagnostics } from \"../../diagnostics/athena-diagnostics\";\nimport { useAui, useAuiState } from \"@assistant-ui/react\";\nimport type { FC, RefObject } from \"react\";\nimport { useCallback } from \"react\";\nimport {\n  resolveComposerAction,\n  type ComposerAction,\n} from \"../composer/composer-action\";\nimport {\n  ComposerSendButton,\n  ComposerStopButton,\n  type ComposerSendButtonProps,\n  type ComposerStopButtonProps,\n} from \"../messages/message-components\";\n\n/**\n * The slice of a composer handle this control needs: something whose draft\n * can be submitted. `TiptapComposerHandle` satisfies it; hosts with their own\n * editor plumbing (the SDK's AthenaChat) expose a compatible handle.\n */\nexport interface ComposerSubmitHandle {\n  submit: () => void | Promise<void>;\n}\n\nexport interface ComposerStopControls {\n  /**\n   * Cancels the active run via the runtime. Returns false when cancellation\n   * threw (e.g. the run already settled), so hosts can report whether their\n   * own teardown still needs to stop backend work.\n   */\n  cancelRun: () => boolean;\n}\n\n/**\n * Control renderers, for hosts with their own published theming contract\n * (AthenaChat's `aui-*` class system). Defaults render the shared buttons.\n */\nexport interface ComposerSendOrStopComponents {\n  SendButton?: FC<ComposerSendButtonProps>;\n  StopButton?: FC<ComposerStopButtonProps>;\n}\n\nexport interface ComposerSendOrStopProps {\n  composerRef: RefObject<ComposerSubmitHandle | null>;\n  hasDraft: boolean;\n  /** A plain draft enters the live voice conversation instead of the run queue. */\n  liveVoice?: boolean;\n  /**\n   * A live session (voice, connecting or live) the Stop control also ends.\n   * With it, an empty draft shows Stop even when no run is in flight, and\n   * Stop ends the session AND cancels any run: one control stops everything.\n   */\n  liveSession?: { end: () => void } | null;\n  /** Disables both send and stop (e.g. while a conversation restores). */\n  disabled?: boolean;\n  /**\n   * Blocks sending (not stopping) with this reason as the tooltip — e.g. a\n   * broken host tunnel that would make tool calls silently fail.\n   */\n  sendBlockedReason?: string | null;\n  /** Tooltip for the send/queue control while `disabled`. */\n  sendDisabledTitle?: string;\n  /** Tooltip for the stop control while `disabled`. */\n  stopDisabledTitle?: string;\n  /**\n   * Host work to run before the draft submits (per-turn bookkeeping).\n   * `action` distinguishes a fresh send from queueing into an active run.\n   */\n  onBeforeSend?: (context: {\n    action: Extract<ComposerAction, \"send\" | \"queue\">;\n  }) => void | Promise<void>;\n  /**\n   * Full custom stop sequence. Hosts that must tear down in-flight host-side\n   * work sequence it around `controls.cancelRun()`. Default: cancel the run.\n   */\n  onStop?: (controls: ComposerStopControls) => void;\n  components?: ComposerSendOrStopComponents;\n}\n\n/**\n * The shared send/queue/stop control: sends the composer draft, queues it as\n * a follow-up while a run is in flight, or stops the run when there is\n * nothing to queue.\n */\nexport const ComposerSendOrStop: FC<ComposerSendOrStopProps> = ({\n  composerRef,\n  hasDraft,\n  liveVoice = false,\n  liveSession = null,\n  disabled = false,\n  sendBlockedReason = null,\n  sendDisabledTitle,\n  stopDisabledTitle,\n  onBeforeSend,\n  onStop,\n  components,\n}) => {\n  const aui = useAui();\n  const isRunning = useAuiState(\n    (state) => state.optional.thread?.isRunning ?? false,\n  );\n  // A draft sends (into the live voice conversation, or as a run) or queues\n  // behind a run; only an empty draft turns the control into Stop, and a\n  // live session alone is enough for that.\n  const action: ComposerAction =\n    liveVoice && hasDraft\n      ? \"send\"\n      : !hasDraft && liveSession !== null\n        ? \"stop\"\n        : resolveComposerAction({ hasDraft, isRunning });\n  const SendButton = components?.SendButton ?? ComposerSendButton;\n  const StopButton = components?.StopButton ?? ComposerStopButton;\n\n  const cancelRun = useCallback((): boolean => {\n    try {\n      aui.thread.cancelRun();\n      return true;\n    } catch (error) {\n      athenaDiagnostics.emit(\"composer.cancel.error\", {\n        level: \"warn\",\n        data: { error: error instanceof Error ? error.message : String(error) },\n      });\n      return false;\n    }\n  }, [aui]);\n\n  // With only a session to end, a cancel would find no run to stop.\n  const stopRun = (): boolean => (isRunning ? cancelRun() : true);\n  const stopEverything = () => {\n    liveSession?.end();\n    if (onStop) {\n      onStop({ cancelRun: stopRun });\n    } else {\n      stopRun();\n    }\n  };\n  const stopControl = (\n    <StopButton\n      onStop={stopEverything}\n      disabled={disabled}\n      title={\n        disabled && stopDisabledTitle\n          ? stopDisabledTitle\n          : liveSession\n            ? \"Stop\"\n            : undefined\n      }\n    />\n  );\n\n  if (action === \"stop\") return stopControl;\n\n  const handleSend = async () => {\n    await onBeforeSend?.({ action });\n    await composerRef.current?.submit();\n  };\n  const sendTitle =\n    disabled && sendDisabledTitle\n      ? sendDisabledTitle\n      : (sendBlockedReason ?? undefined);\n  const sendControl = (\n    <SendButton\n      action={action}\n      onSend={() => void handleSend()}\n      disabled={disabled || !hasDraft || sendBlockedReason != null}\n      title={sendTitle}\n    />\n  );\n  // A live session keeps its Stop beside Send while a draft is typed, so the\n  // call (and any run) can end without sending or discarding the words; the\n  // draft stays in the composer.\n  if (liveSession !== null) {\n    return (\n      <>\n        {stopControl}\n        {sendControl}\n      </>\n    );\n  }\n  return sendControl;\n};\n"],"names":[],"mappings":";;;;;;AAoFO,MAAM,qBAAkD,CAAC;AAAA,EAC9D;AAAA,EACA;AAAA,EACA,YAAY;AAAA,EACZ,cAAc;AAAA,EACd,WAAW;AAAA,EACX,oBAAoB;AAAA,EACpB;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF,MAAM;AACJ,QAAM,MAAM,OAAA;AACZ,QAAM,YAAY;AAAA,IAChB,CAAC,UAAA;;AAAU,0BAAM,SAAS,WAAf,mBAAuB,cAAa;AAAA;AAAA,EAAA;AAKjD,QAAM,SACJ,aAAa,WACT,SACA,CAAC,YAAY,gBAAgB,OAC3B,SACA,sBAAsB,EAAE,UAAU,WAAW;AACrD,QAAM,cAAa,yCAAY,eAAc;AAC7C,QAAM,cAAa,yCAAY,eAAc;AAE7C,QAAM,YAAY,YAAY,MAAe;AAC3C,QAAI;AACF,UAAI,OAAO,UAAA;AACX,aAAO;AAAA,IACT,SAAS,OAAO;AACd,wBAAkB,KAAK,yBAAyB;AAAA,QAC9C,OAAO;AAAA,QACP,MAAM,EAAE,OAAO,iBAAiB,QAAQ,MAAM,UAAU,OAAO,KAAK,EAAA;AAAA,MAAE,CACvE;AACD,aAAO;AAAA,IACT;AAAA,EACF,GAAG,CAAC,GAAG,CAAC;AAGR,QAAM,UAAU,MAAgB,YAAY,UAAA,IAAc;AAC1D,QAAM,iBAAiB,MAAM;AAC3B,+CAAa;AACb,QAAI,QAAQ;AACV,aAAO,EAAE,WAAW,SAAS;AAAA,IAC/B,OAAO;AACL,cAAA;AAAA,IACF;AAAA,EACF;AACA,QAAM,cACJ;AAAA,IAAC;AAAA,IAAA;AAAA,MACC,QAAQ;AAAA,MACR;AAAA,MACA,OACE,YAAY,oBACR,oBACA,cACE,SACA;AAAA,IAAA;AAAA,EAAA;AAKZ,MAAI,WAAW,OAAQ,QAAO;AAE9B,QAAM,aAAa,YAAY;;AAC7B,WAAM,6CAAe,EAAE;AACvB,YAAM,iBAAY,YAAZ,mBAAqB;AAAA,EAC7B;AACA,QAAM,YACJ,YAAY,oBACR,oBACC,qBAAqB;AAC5B,QAAM,cACJ;AAAA,IAAC;AAAA,IAAA;AAAA,MACC;AAAA,MACA,QAAQ,MAAM,KAAK,WAAA;AAAA,MACnB,UAAU,YAAY,CAAC,YAAY,qBAAqB;AAAA,MACxD,OAAO;AAAA,IAAA;AAAA,EAAA;AAMX,MAAI,gBAAgB,MAAM;AACxB,WACE,qBAAA,UAAA,EACG,UAAA;AAAA,MAAA;AAAA,MACA;AAAA,IAAA,GACH;AAAA,EAEJ;AACA,SAAO;AACT;"}