@athenaintel/react 0.57.1 → 0.57.2

package/dist/chat/AthenaChat.js

@@ -12,6 +12,7 @@ import { WaitingOnYouBanner } from "./WaitingOnYouBanner.js";
 import { DrivingOnChip } from "./DrivingOnChip.js";
 import { OfflineSnapshotBanner } from "../offline/offline-history.js";
 import { WakeMarkerMessage } from "../chat-ui/messages/wake-marker.js";
+import { RuntimeMessageCard } from "../chat-ui/messages/runtime-message.js";
 import { ToolFallback, ToolFallbackAppearanceProvider } from "../tools/tool-fallback.js";
 import { AppendDocumentToolUI } from "../tools/append-document-tool-ui.js";
 import { ReadAssetToolUI } from "../tools/read-asset-tool-ui.js";
@@ -82,6 +83,7 @@ import { isStatewireChannelLive } from "../runtime/statewire-lifecycle.js";
 import { useComposerChatVoice, ChatVoiceButton, ChatVoiceStatus, ChatVoiceContinueOffer } from "./chat-voice-controls.js";
 import { TASK_TOOL_NAME } from "../agent-runtime/src/tool-names.js";
 import { useTurnToolCallsAwaitingApproval } from "../agent-runtime/src/subagents.js";
+import { readSoleRuntimeMessageKind, readRuntimeMessageFromSourceMessages } from "../agent-runtime/src/runtime-messages.js";
 import { isGithubWakeMessage } from "../agent-runtime/src/github-wake-event.js";
 import { isSubscriptionWakeMessage } from "../agent-runtime/src/subscription-wake-event.js";
 import { useDeepAgentConnection, readRunLive } from "../agent-runtime/src/thread-status.js";
@@ -1003,6 +1005,12 @@ const AthenaReasoningPart = ({
 };
 const noToolGrouping = () => null;
 const isInlineToolOrSubagent = (toolName) => toolName === TASK_TOOL_NAME || isSdkInlineTool(toolName);
+const AthenaRuntimeEndingCard = () => {
+  const message = useAuiState((s) => s.message);
+  const runtime = useMemo(() => readRuntimeMessageFromSourceMessages(message), [message]);
+  if (!runtime || runtime.silent) return null;
+  return /* @__PURE__ */ jsx(RuntimeMessageCard, { message: runtime });
+};
 const AthenaAssistantMessage = ({
   toolUIs,
   AvatarComponent,
@@ -1035,6 +1043,43 @@ const AthenaAssistantMessage = ({
     [EmptyComponent, TextComponent, effectiveReasoningComponent, toolUIsWithNestedMessages]
   );
   const arrivedLive = useMessageArrivalAnimation();
+  const soleEndingKind = useAuiState((s) => readSoleRuntimeMessageKind(s.message));
+  const endingSilent = useAuiState(
+    (s) => {
+      var _a;
+      return ((_a = readRuntimeMessageFromSourceMessages(s.message)) == null ? void 0 : _a.silent) === true;
+    }
+  );
+  const chunkedEndingText = useAuiState((s) => {
+    const runtime = readRuntimeMessageFromSourceMessages(s.message);
+    if (!runtime || runtime.isSoleSourceMessage || runtime.kind === "handoff") return void 0;
+    const markerText = runtime.text;
+    const last = s.message.content.at(-1);
+    if (markerText === void 0 || (last == null ? void 0 : last.type) !== "text" || last.text !== markerText) return void 0;
+    const sameText = s.message.content.filter((part) => part.type === "text" && part.text === markerText);
+    return sameText.length === 1 ? markerText : void 0;
+  });
+  const AnswerTextComponent = useMemo(() => {
+    if (chunkedEndingText === void 0) return TextComponent;
+    const Base = TextComponent;
+    return function TextOrRunEnding(props) {
+      return props.text === chunkedEndingText ? /* @__PURE__ */ jsx(AthenaRuntimeEndingCard, {}) : /* @__PURE__ */ jsx(Base, { ...props });
+    };
+  }, [TextComponent, chunkedEndingText]);
+  if (soleEndingKind !== void 0 && soleEndingKind !== "handoff") {
+    if (endingSilent) return null;
+    return /* @__PURE__ */ jsxs(
+      MessagePrimitive.Root,
+      {
+        className: "aui-assistant-message-root aui-run-ending-root mx-auto w-full max-w-(--thread-max-width) px-2 py-1",
+        "data-role": "assistant",
+        children: [
+          /* @__PURE__ */ jsx(MessageDaySeparator, {}),
+          /* @__PURE__ */ jsx(AthenaRuntimeEndingCard, {})
+        ]
+      }
+    );
+  }
   return /* @__PURE__ */ jsxs(
     MessagePrimitive.Root,
     {
@@ -1069,7 +1114,7 @@ const AthenaAssistantMessage = ({
                     renderRunningMeta: renderGroupedRunningMeta
                   }
                 ),
-                /* @__PURE__ */ jsx(SuperGroupingFinalText, { TextComponent })
+                /* @__PURE__ */ jsx(SuperGroupingFinalText, { TextComponent: AnswerTextComponent })
               ] }) : /* @__PURE__ */ jsxs(Fragment, { children: [
                 /* @__PURE__ */ jsx(AthenaMessageEmptyIndicator, { EmptyComponent }),
                 /* @__PURE__ */ jsx(MessagePrimitive.GroupedParts, { groupBy: noToolGrouping, children: ({ part }) => {
@@ -1089,7 +1134,7 @@ const AthenaAssistantMessage = ({
                       if (textPart.text === "" && ((_a = textPart.status) == null ? void 0 : _a.type) === "running") {
                         return /* @__PURE__ */ jsx(EmptyComponent, { status: textPart.status });
                       }
-                      return /* @__PURE__ */ jsx(TextComponent, { ...textPart });
+                      return /* @__PURE__ */ jsx(AnswerTextComponent, { ...textPart });
                     }
                     default:
                       return null;