@agent-native/toolkit 0.198.3 → 0.198.4

package/src/app/chat/AgentKitAssistantChat.spec.tsx

@@ -234,6 +234,18 @@ vi.mock("../agentkit/react/root.js", async () => {
   return {
     AgentKitRoot: (props: any) => {
       chatMocks.rootProps = props;
+      if (chatMocks.useRealRoot) {
+        return React.createElement(actual.AgentKitRoot, {
+          ...props,
+          ...(chatMocks.realComposerController
+            ? {
+                controller: chatMocks.realComposerController,
+                transport: undefined,
+                endpoint: undefined,
+              }
+            : {}),
+        });
+      }
       if (chatMocks.realComposerController) {
         return React.createElement(AgentKitProvider, {
           ...props,
@@ -747,6 +759,29 @@ afterEach(async () => {
 });
 
 describe("AgentKitAssistantChat host behavior", () => {
+  it("shows a retry when chat history fails to load", async () => {
+    const retryHistory = vi.fn();
+    chatMocks.history = {
+      historyLoadFailed: true,
+      isRetryingHistory: false,
+      retryHistory,
+    };
+
+    await mount(baseProps());
+
+    expect(container.querySelector('[role="alert"]')?.textContent).toContain(
+      "agentChat.message.historyUnavailable",
+    );
+    const retry = [...container.querySelectorAll("button")].find(
+      (button) => button.textContent === "agentChat.common.retry",
+    );
+    expect(retry).toBeDefined();
+
+    await act(async () => retry?.click());
+
+    expect(retryHistory).toHaveBeenCalledOnce();
+  });
+
   it("shows Thinking in the transcript while a submitted user message is pending", async () => {
     chatMocks.history = { isSubmissionInFlight: true };
     chatMocks.thread.messages = [
@@ -759,6 +794,8 @@ describe("AgentKitAssistantChat host behavior", () => {
     ];
     await mount(baseProps());
 
+    expect(chatMocks.composerProps.disabled).toBe(false);
+    expect(chatMocks.composerProps.submissionDisabled).toBe(true);
     expect(container.querySelector('[role="status"]')?.textContent).toBe(
       "agentChat.status.thinking",
     );
@@ -847,6 +884,29 @@ describe("AgentKitAssistantChat host behavior", () => {
     return client;
   }
 
+  it("keeps the real editor editable while a submission is in flight", async () => {
+    chatMocks.useRealChat = true;
+    chatMocks.history = { isSubmissionInFlight: true };
+    await useRealComposer();
+    await mount(baseProps({ showModelSelector: false }));
+
+    const editor = container.querySelector<HTMLElement>(
+      '[contenteditable="true"]',
+    );
+    expect(editor).not.toBeNull();
+    expect(chatMocks.composerProps.submissionDisabled).toBe(true);
+
+    await act(async () =>
+      chatMocks.composerProps.composerRef.current.setText("Keep this draft"),
+    );
+    expect(editor?.textContent).toBe("Keep this draft");
+    expect(
+      container.querySelector<HTMLButtonElement>(
+        '[data-agent-composer-slot="send-button"]',
+      )?.disabled,
+    ).toBe(true);
+  });
+
   it.each(["default", "compact"] as const)(
     "clears the real %s composer before startRun resolves and preserves the next draft",
     async (composerLayoutVariant) => {
@@ -855,12 +915,13 @@ describe("AgentKitAssistantChat host behavior", () => {
       chatMocks.useRealChat = true;
       const client = await useRealComposer(startRun);
       try {
-        await mount(
-          baseProps({
-            composerLayoutVariant,
-            showModelSelector: false,
-          }),
-        );
+        const props = baseProps({
+          composerLayoutVariant,
+          showModelSelector: false,
+        });
+        const openThread = vi.spyOn(client, "openThread");
+        chatMocks.useRealRoot = true;
+        await mount(props);
         container.style.width =
           composerLayoutVariant === "compact" ? "320px" : "960px";
         const composer = chatMocks.composerProps.composerRef.current;
@@ -884,6 +945,19 @@ describe("AgentKitAssistantChat host behavior", () => {
         expect(client.getThread(chatMocks.threadId).activeRunIds).toEqual([]);
         expect(editor.textContent).toBe("");
 
+        await act(async () =>
+          root.render(<AgentKitAssistantChat {...props} isNewThread={false} />),
+        );
+        await flush();
+
+        expect(chatMocks.rootProps.load).toBe("manual");
+        expect(openThread).not.toHaveBeenCalled();
+        expect(
+          container
+            .querySelector<HTMLElement>("[contenteditable]")
+            ?.getAttribute("contenteditable"),
+        ).toBe("true");
+
         await act(async () => composer.setText("Next draft"));
         await act(async () => started.resolve({ runId: "run-latency" }));
 
@@ -1585,7 +1659,7 @@ describe("AgentKitAssistantChat host behavior", () => {
     expect(chatMocks.appState.has(stateKey!)).toBe(false);
   });
 
-  it("keeps a queued follow-up queued if its run finishes during provider discovery", async () => {
+  it("preserves queued intent when its run finishes during provider discovery", async () => {
     chatMocks.readiness = {
       canChat: false,
       missing: false,
@@ -1611,12 +1685,27 @@ describe("AgentKitAssistantChat host behavior", () => {
         [],
         [],
         { intent: "queued" },
+        async () => ({}),
       );
     });
 
     expect(chatMocks.control.sendMessage).not.toHaveBeenCalled();
     expect(chatMocks.control.queueMessage).not.toHaveBeenCalled();
 
+    const stateKey = [...chatMocks.appState.keys()].find((key) =>
+      key.startsWith("agentkit-deferred-provider-submissions:"),
+    );
+    expect(chatMocks.appState.get(stateKey!)).toMatchObject({
+      submissions: [
+        {
+          composerOptions: {
+            intent: "queued",
+            queuedWhileRunActive: true,
+          },
+        },
+      ],
+    });
+
     chatMocks.thread.activeRunIds = [];
     chatMocks.thread.runs["analytics-run"].status = "completed";
     chatMocks.readiness = {
@@ -1640,6 +1729,7 @@ describe("AgentKitAssistantChat host behavior", () => {
       }),
     );
     expect(chatMocks.control.sendMessage).not.toHaveBeenCalled();
+    expect(chatMocks.appState.has(stateKey!)).toBe(false);
   });
 
   it("keeps the draft rejected if provider status becomes missing before submit", async () => {
@@ -3338,11 +3428,11 @@ describe("AgentKitAssistantChat host behavior", () => {
     chatMocks.thread.activeRunIds = ["run-1"];
     await mount(baseProps());
 
-    await act(async () => {
-      await chatMocks.resumeProps.onMessageResume({
-        message: "Continue after connecting the integration.",
-      });
+    const resume = chatMocks.resumeProps.onMessageResume({
+      message: "Continue after connecting the integration.",
     });
+    expect(resume).toBeInstanceOf(Promise);
+    await act(async () => resume);
     expect(chatMocks.control.queueMessage).toHaveBeenCalledWith(
       expect.objectContaining({
         text: "Continue after connecting the integration.",
@@ -3382,6 +3472,19 @@ describe("AgentKitAssistantChat host behavior", () => {
     expect(chatMocks.control.sendMessage).not.toHaveBeenCalled();
   });
 
+  it("keeps failed integration prompt submissions resumable", async () => {
+    chatMocks.thread.activeRunIds = ["run-1"];
+    const submissionError = new Error("Temporary send failure");
+    chatMocks.control.queueMessage.mockRejectedValueOnce(submissionError);
+    await mount(baseProps());
+
+    await expect(
+      chatMocks.resumeProps.onMessageResume({
+        message: "Continue after OAuth.",
+      }),
+    ).rejects.toBe(submissionError);
+  });
+
   it("shows the missing-final-response warning from recovered run metadata", async () => {
     chatMocks.thread.messages = [
       {

package/src/app/chat/AgentKitAssistantChat.tsx

@@ -460,6 +460,15 @@ type AgentKitSuggestionSubmitOptions = PromptComposerSubmitOptions & {
   queuedWhileRunActive?: boolean;
 };
 
+function preserveQueuedIntent(
+  preparedOptions: PromptComposerSubmitOptions,
+  originalOptions: PromptComposerSubmitOptions,
+): PromptComposerSubmitOptions {
+  return originalOptions.intent === "queued"
+    ? { ...preparedOptions, intent: "queued" }
+    : preparedOptions;
+}
+
 function captureQueuedRunState(
   options: PromptComposerSubmitOptions,
   runWasActive: boolean,
@@ -594,8 +603,13 @@ export const AgentKitAssistantChat = forwardRef<
       : readAgentKitThreadHandoffSnapshot(
           createAgentKitThreadHandoffKey(props, threadId),
         );
+  const [manualLoadThreadId, setManualLoadThreadId] = useState<string | null>(
+    () => (props.isNewThread ? threadId : null),
+  );
+  const hasManualLoadForCurrentThread =
+    props.isNewThread || manualLoadThreadId === threadId;
   const [threadRestore, setThreadRestore] = useState<ThreadRestoreState>(() =>
-    props.isThreadStateLoading || !props.isNewThread
+    props.isThreadStateLoading || !hasManualLoadForCurrentThread
       ? { status: "loading" }
       : { status: "ready" },
   );
@@ -611,15 +625,25 @@ export const AgentKitAssistantChat = forwardRef<
     setRestoreRetryLoadPhase("idle");
     setRestoreRetryThreadId(null);
   }, []);
+  useEffect(() => {
+    setManualLoadThreadId((current) =>
+      props.isNewThread ? threadId : current === threadId ? current : null,
+    );
+  }, [props.isNewThread, threadId]);
   useEffect(() => {
     setThreadRestore(
-      props.isThreadStateLoading || !props.isNewThread
+      props.isThreadStateLoading || !hasManualLoadForCurrentThread
         ? { status: "loading" }
         : { status: "ready" },
     );
     setRestoreRetryLoadPhase("idle");
     setRestoreRetryThreadId(null);
-  }, [props.isNewThread, props.isThreadStateLoading, threadId]);
+  }, [
+    hasManualLoadForCurrentThread,
+    props.isNewThread,
+    props.isThreadStateLoading,
+    threadId,
+  ]);
   const onThreadRestoreLoadError = useCallback(
     (error: unknown) => {
       const record = asRecord(error);
@@ -989,7 +1013,7 @@ export const AgentKitAssistantChat = forwardRef<
   ]);
   const agentKitLoad =
     props.isThreadStateLoading ||
-    props.isNewThread ||
+    hasManualLoadForCurrentThread ||
     (restoreRetryThreadId === threadId && restoreRetryLoadPhase === "release")
       ? "manual"
       : "auto";
@@ -2204,7 +2228,7 @@ const AgentKitAssistantChatBody = forwardRef<
           try {
             const preparedOptions = prepare ? await prepare() : composerOptions;
             const submittedOptions = captureQueuedRunState(
-              preparedOptions,
+              preserveQueuedIntent(preparedOptions, composerOptions),
               runWasActiveAtSubmit,
             );
             const result = await submit(
@@ -2231,7 +2255,10 @@ const AgentKitAssistantChatBody = forwardRef<
           text,
           files,
           references,
-          captureQueuedRunState(preparedOptions, runWasActiveAtSubmit),
+          captureQueuedRunState(
+            preserveQueuedIntent(preparedOptions, composerOptions),
+            runWasActiveAtSubmit,
+          ),
         );
       } catch (error) {
         dispatchSetupRequiredEvent(error, props.tabId, threadId);
@@ -2552,11 +2579,19 @@ const AgentKitAssistantChatBody = forwardRef<
     [isThreadRunning, submit],
   );
   const resumeIntegrationPrompt = useCallback(
-    (message: string) => {
-      if (props.isActiveComposer === false) return;
-      void send(message).catch((error) => {
+    async (message: string) => {
+      if (props.isActiveComposer === false) {
+        throw new Error("Cannot resume a request in an inactive chat.");
+      }
+      try {
+        const result = await send(message);
+        if (result.status === "rejected") {
+          throw new Error(`Chat resume was rejected: ${result.reason}.`);
+        }
+      } catch (error) {
         dispatchSetupRequiredEvent(error, props.tabId, threadId);
-      });
+        throw error;
+      }
     },
     [props.isActiveComposer, props.tabId, send, threadId],
   );
@@ -2775,15 +2810,19 @@ const AgentKitAssistantChatBody = forwardRef<
             { threadId: targetThreadId, runId, requestId },
             request,
           ) => {
-            if (targetThreadId !== threadId) return;
+            if (targetThreadId !== threadId) {
+              throw new Error(
+                "Cannot resume a connection request in another chat.",
+              );
+            }
             return control.resolveConnectionRequest(runId, requestId, {
               status: "connected",
               message: request.message,
             });
           }}
-          onMessageResume={(request) => {
-            resumeIntegrationPrompt(request.message);
-          }}
+          onMessageResume={(request) =>
+            resumeIntegrationPrompt(request.message)
+          }
         />
       )}
       <RunStuckBanner
@@ -2805,6 +2844,24 @@ const AgentKitAssistantChatBody = forwardRef<
           void sendRecoveryMessage(RECOVERY_CONTINUE_PROMPT, "continue")
         }
       />
+      {history?.historyLoadFailed ? (
+        <div
+          role="alert"
+          className="mx-3 mb-2 flex items-center justify-between gap-3 rounded-md border border-border px-3 py-2 text-xs"
+        >
+          <span className="text-muted-foreground">
+            {t("agentChat.message.historyUnavailable")}
+          </span>
+          <button
+            type="button"
+            onClick={history.retryHistory}
+            disabled={history.isRetryingHistory}
+            className="shrink-0 font-medium text-foreground hover:underline disabled:cursor-not-allowed disabled:opacity-50"
+          >
+            {t("agentChat.common.retry")}
+          </button>
+        </div>
+      ) : null}
       <AgentKitChat
         className={props.className}
         composerProps={{ attachmentsEnabled: fileStorageConfigured }}
@@ -3668,12 +3725,12 @@ function AgentKitComposerSurface({
           disabled={
             (!canChat && !providerSubmissionPending) ||
             props.composerDisabled ||
-            isRestoring ||
-            isSubmissionInFlight
+            isRestoring
           }
           submissionDisabled={
             (!canChat && !providerSubmissionPending) ||
-            props.composerSubmissionDisabled === true
+            props.composerSubmissionDisabled === true ||
+            isSubmissionInFlight
           }
           onDisabledClick={
             props.composerDisabled || !setupMissing
@@ -4128,6 +4185,10 @@ function AgentKitConnectionRequest({
     <McpAgentKitConnectionRequestCard
       provider={value.provider}
       detail={value.detail}
+      reason={value.reason}
+      status={value.status}
+      appId={value.appId}
+      source={value.source}
       target={{ threadId, runId, requestId: value.id }}
       onConnected={() =>
         control.resolveConnectionRequest(runId, value.id, {

package/src/app/chat/AgentPanel.header.spec.ts

@@ -2,16 +2,16 @@
 
 import { readFileSync } from "node:fs";
 
+import {
+  DropdownMenu,
+  DropdownMenuContent,
+  DropdownMenuTrigger,
+} from "@agent-native/toolkit/ui/dropdown-menu";
 import React, { act } from "react";
 import { createRoot, type Root } from "react-dom/client";
 import { MemoryRouter, useLocation } from "react-router";
 import { describe, expect, it, vi } from "vitest";
 
-import {
-  DropdownMenu,
-  DropdownMenuContent,
-  DropdownMenuTrigger,
-} from "../../ui/dropdown-menu.js";
 import {
   AgentChatSurface,
   AgentPanelFullViewMenuItem,
@@ -51,6 +51,13 @@ import {
   preloadAgentChatSurface,
 } from "./AgentSidebar.js";
 
+function readSource(
+  path: string,
+  _options?: "utf8" | { encoding: "utf8" },
+): string {
+  return readFileSync(path, "utf8").replace(/\r\n/g, "\n");
+}
+
 describe("AgentPanel compatibility exports", () => {
   it("preserves the legacy sidebar entry point", () => {
     expect(LegacyAgentSidebar).toBe(AgentSidebar);
@@ -60,7 +67,7 @@ describe("AgentPanel compatibility exports", () => {
   });
 
   it("uses a stable-ref link in the full-view menu item", () => {
-    const source = readFileSync("src/app/chat/AgentPanel.tsx", "utf8").replace(
+    const source = readSource("src/app/chat/AgentPanel.tsx", "utf8").replace(
       /\s+/g,
       " ",
     );
@@ -150,7 +157,7 @@ describe("AgentPanel fullscreen menu", () => {
 
 describe("AgentPanel suggestion placement", () => {
   it("forwards explicit placement and defaults to context chips", () => {
-    const source = readFileSync("src/app/chat/AgentPanel.tsx", {
+    const source = readSource("src/app/chat/AgentPanel.tsx", {
       encoding: "utf8",
     });
 
@@ -323,7 +330,7 @@ describe("AgentPanel header tab visibility", () => {
   });
 
   it("exposes page header composition without moving it into app chrome", () => {
-    const source = readFileSync("src/app/chat/AgentPanel.tsx", "utf8");
+    const source = readSource("src/app/chat/AgentPanel.tsx", "utf8");
 
     expect(source).toContain('data-agent-page-chat-header=""');
     expect(source).toContain("pageHeaderLeadingSlot");
@@ -699,7 +706,7 @@ describe("AgentPanel Integrations link", () => {
   });
 
   it("sits right after Open full view and shares its separator", () => {
-    const source = readFileSync("src/app/chat/AgentPanel.tsx", {
+    const source = readSource("src/app/chat/AgentPanel.tsx", {
       encoding: "utf8",
     });
     const overflowMenu = source.slice(
@@ -820,6 +827,36 @@ describe("AgentSidebar composer focus", () => {
     }
   });
 
+  it("stops retrying when the document is torn down", () => {
+    vi.useFakeTimers();
+    const previousRequestAnimationFrame = window.requestAnimationFrame;
+    const frames: Array<FrameRequestCallback> = [];
+    const panel = document.createElement("div");
+    panel.className = "agent-sidebar-panel";
+    panel.dataset.agentSidebarState = "open";
+    document.body.appendChild(panel);
+
+    window.requestAnimationFrame = ((callback: FrameRequestCallback) => {
+      frames.push(callback);
+      return frames.length;
+    }) as typeof window.requestAnimationFrame;
+
+    try {
+      focusAgentChat();
+      frames[0]!(0);
+      vi.stubGlobal("document", undefined);
+
+      expect(() => vi.advanceTimersByTime(50)).not.toThrow();
+      expect(vi.getTimerCount()).toBe(0);
+    } finally {
+      vi.unstubAllGlobals();
+      window.requestAnimationFrame = previousRequestAnimationFrame;
+      panel.remove();
+      vi.clearAllTimers();
+      vi.useRealTimers();
+    }
+  });
+
   it("focuses a frame-owned composer", () => {
     const previousRequestAnimationFrame = window.requestAnimationFrame;
     const frames: Array<FrameRequestCallback> = [];
@@ -914,10 +951,10 @@ describe("AgentPanel header overflow actions", () => {
   });
 
   it("keeps width and full-view actions out of the icon row", () => {
-    const source = readFileSync("src/app/chat/AgentPanel.tsx", {
+    const source = readSource("src/app/chat/AgentPanel.tsx", {
       encoding: "utf8",
     });
-    const sidebarSource = readFileSync("src/app/chat/AgentSidebar.tsx", {
+    const sidebarSource = readSource("src/app/chat/AgentSidebar.tsx", {
       encoding: "utf8",
     });
     const headerActions = source.slice(
@@ -957,7 +994,7 @@ describe("AgentPanel header overflow actions", () => {
   });
 
   it("keeps the overflow menu scrollable within the viewport", () => {
-    const source = readFileSync("src/app/chat/AgentPanel.tsx", {
+    const source = readSource("src/app/chat/AgentPanel.tsx", {
       encoding: "utf8",
     });
     const overflowMenu = source.slice(
@@ -972,7 +1009,7 @@ describe("AgentPanel header overflow actions", () => {
   });
 
   it("offers sharing from the sidebar overflow for an active chat", () => {
-    const source = readFileSync("src/app/chat/AgentPanel.tsx", {
+    const source = readSource("src/app/chat/AgentPanel.tsx", {
       encoding: "utf8",
     });
     const overflowMenu = source.slice(
@@ -992,7 +1029,7 @@ describe("AgentPanel header overflow actions", () => {
   });
 
   it("keeps chat headers persistent while switching app surfaces", () => {
-    const source = readFileSync("src/app/chat/AgentPanel.tsx", {
+    const source = readSource("src/app/chat/AgentPanel.tsx", {
       encoding: "utf8",
     });
 
@@ -1002,10 +1039,10 @@ describe("AgentPanel header overflow actions", () => {
   });
 
   it("supports a persistent two-state sidebar toggle", () => {
-    const source = readFileSync("src/app/chat/AgentSidebar.tsx", {
+    const source = readSource("src/app/chat/AgentSidebar.tsx", {
       encoding: "utf8",
     });
-    const panelSource = readFileSync("src/app/chat/AgentPanel.tsx", {
+    const panelSource = readSource("src/app/chat/AgentPanel.tsx", {
       encoding: "utf8",
     });
 
@@ -1021,7 +1058,7 @@ describe("AgentPanel header overflow actions", () => {
   });
 
   it("keeps host CLI tabs mounted while chat is active", () => {
-    const source = readFileSync("src/app/chat/AgentPanel.tsx", {
+    const source = readSource("src/app/chat/AgentPanel.tsx", {
       encoding: "utf8",
     });
 
@@ -1035,7 +1072,7 @@ describe("AgentPanel header overflow actions", () => {
   });
 
   it("only shows tabs for the active desktop surface", () => {
-    const source = readFileSync("src/app/chat/AgentPanel.tsx", {
+    const source = readSource("src/app/chat/AgentPanel.tsx", {
       encoding: "utf8",
     });
 
@@ -1046,7 +1083,7 @@ describe("AgentPanel header overflow actions", () => {
 
 describe("AgentSidebar wide drawer layout", () => {
   it("can disable the panel without unmounting the app surface", () => {
-    const source = readFileSync("src/app/chat/AgentSidebar.tsx", {
+    const source = readSource("src/app/chat/AgentSidebar.tsx", {
       encoding: "utf8",
     });
 
@@ -1058,7 +1095,7 @@ describe("AgentSidebar wide drawer layout", () => {
   });
 
   it("does not reserve the drawer placeholder after the panel closes", () => {
-    const source = readFileSync("src/app/chat/AgentSidebar.tsx", {
+    const source = readSource("src/app/chat/AgentSidebar.tsx", {
       encoding: "utf8",
     });
     const placeholderStart = source.indexOf("const drawerPlaceholder");
@@ -1083,7 +1120,7 @@ describe("AgentChatSurface chrome defaults", () => {
   });
 
   it("keeps settings out of every chat surface", () => {
-    const source = readFileSync("src/app/chat/AgentPanel.tsx", {
+    const source = readSource("src/app/chat/AgentPanel.tsx", {
       encoding: "utf8",
     });
 
@@ -1117,7 +1154,7 @@ describe("AgentChatSurface chrome defaults", () => {
 
 describe("AgentPanel stale lazy chunk recovery", () => {
   it("uses the guarded reload path before the panel reset fallback", () => {
-    const source = readFileSync("src/app/chat/AgentPanel.tsx", {
+    const source = readSource("src/app/chat/AgentPanel.tsx", {
       encoding: "utf8",
     });
     const componentDidCatch = source.slice(

package/src/app/chat/AgentSidebar.tsx

@@ -1331,6 +1331,8 @@ export function focusAgentChat() {
 
 function focusAgentChatComposer() {
   const focusComposer = (attempt = 0) => {
+    if (typeof window === "undefined" || typeof document === "undefined")
+      return;
     const panel = document.querySelector(
       ".agent-sidebar-panel[data-agent-sidebar-state='open'], " +
         ".agent-frame-sidebar[data-agent-frame-sidebar-state='open']",

package/src/app/chat/FileStorageSetupPopover.tsx

@@ -8,8 +8,8 @@ import {
 import { IconCloudUpload } from "@tabler/icons-react";
 import { useRef, type RefObject } from "react";
 
-import { BuilderConnectPopover } from "../settings/index.js";
-import { BuilderConnectCard } from "../setup-connections/index.js";
+import { DeferredBuilderConnectPopover as BuilderConnectPopover } from "../settings/deferred-builder-connect-popover.js";
+import { BuilderConnectCard } from "../setup-connections/BuilderConnectCard.js";
 
 type FileStorageSetupPopoverCommonProps = {
   open: boolean;

package/src/app/chat/MultiTabAssistantChat.messageCount.spec.tsx

@@ -1,6 +1,8 @@
 import { describe, expect, it, vi } from "vitest";
 
-import MultiTabSource from "./MultiTabAssistantChat.tsx?raw";
+import rawMultiTabSource from "./MultiTabAssistantChat.tsx?raw";
+
+const MultiTabSource = rawMultiTabSource.replace(/\r\n/g, "\n");
 
 describe("MultiTabAssistantChat message count", () => {
   it("forwards the host callback instead of only counting tabs", () => {

package/src/app/chat/MultiTabAssistantChat.spec.tsx

@@ -210,9 +210,11 @@ const modelCatalogMocks = vi.hoisted(() => ({
   load: null as null | (() => Promise<unknown>),
 }));
 
-vi.mock("@agent-native/core/client/hooks", async (importOriginal) => {
+vi.mock("@agent-native/core/client/use-action", async (importOriginal) => {
   const actual =
-    await importOriginal<typeof import("@agent-native/core/client/hooks")>();
+    await importOriginal<
+      typeof import("@agent-native/core/client/use-action")
+    >();
   return { ...actual, ...actionMocks };
 });
 
@@ -433,6 +435,8 @@ describe("MultiTabAssistantChat postMessage bridge", () => {
   beforeEach(async () => {
     vi.stubGlobal("IS_REACT_ACT_ENVIRONMENT", true);
     resetThreadMocks();
+    actionMocks.callAction.mockReset();
+    actionMocks.callAction.mockResolvedValue(null as never);
     ensureLocalStorage();
     vi.stubGlobal(
       "fetch",
@@ -3345,7 +3349,7 @@ describe("MultiTabAssistantChat page overlay", () => {
     ).not.toBeNull();
   });
 
-  it("reserves the page top bar even when its actions are temporarily empty", async () => {
+  it("reserves one page-header height when its actions are temporarily empty", async () => {
     await act(async () => {
       root.render(
         <MultiTabAssistantChat
@@ -3359,7 +3363,7 @@ describe("MultiTabAssistantChat page overlay", () => {
       "[data-agent-page-chat-topbar]",
     );
     expect(topbar).not.toBeNull();
-    expect(topbar?.className).toContain("pt-14");
+    expect(topbar?.className).toContain("pt-12");
   });
 });