fix(chat): preserve custom CLI preference during catalog load (#149)

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>

frontend/src/features/layout/LayoutGrid.chat.test.tsx

@@ -77,6 +77,14 @@ const ptyProfile: AgentProfile = {
   cwdTemplate: "{projectRoot}",
 };
 
+function deferred<T>() {
+  let resolve!: (value: T | PromiseLike<T>) => void;
+  const promise = new Promise<T>((res) => {
+    resolve = res;
+  });
+  return { promise, resolve };
+}
+
 beforeEach(() => {
   window.localStorage.clear();
 });
@@ -191,4 +199,58 @@ describe("LayoutGrid custom agent CLI (#147)", () => {
       expect(screen.getByTestId("custom-agent-chat-view")).toBeTruthy(),
     );
   });
+
+  it("keeps a restored custom CLI mode while the agent/profile catalog is still loading (#149)", async () => {
+    const layout = new MockLayoutGateway();
+    const agent = new MockAgentGateway();
+    const profileGateway = new MockProfileGateway();
+    const terminal = new MockTerminalGateway();
+    const system = new MockSystemGateway();
+    await profileGateway.configureProfiles([structuredProfile]);
+    const created = await agent.createAgent("p1", {
+      name: "Worker",
+      profileId: structuredProfile.id,
+    });
+    const tree = await layout.loadLayout("p1");
+    const leafId = leaves(tree)[0].id;
+    await layout.mutateLayout("p1", {
+      type: "setCellAgent",
+      target: leafId,
+      agent: created.id,
+    });
+    window.localStorage.setItem(`idea.agent-cell-mode.p1.${leafId}`, "custom");
+
+    const agentsLoaded = deferred<void>();
+    const profilesLoaded = deferred<void>();
+    const originalListAgents = agent.listAgents.bind(agent);
+    const originalListProfiles = profileGateway.listProfiles.bind(profileGateway);
+    vi.spyOn(agent, "listAgents").mockImplementation(async (projectId) => {
+      await agentsLoaded.promise;
+      return originalListAgents(projectId);
+    });
+    vi.spyOn(profileGateway, "listProfiles").mockImplementation(async () => {
+      await profilesLoaded.promise;
+      return originalListProfiles();
+    });
+
+    renderGrid({
+      layout,
+      agent,
+      profile: profileGateway,
+      terminal,
+      system,
+    } as unknown as Gateways);
+
+    await waitFor(() => expect(screen.getByTestId("layout-leaf")).toBeTruthy());
+    expect(window.localStorage.getItem(`idea.agent-cell-mode.p1.${leafId}`)).toBe("custom");
+    expect(screen.queryByTestId("custom-agent-chat-view")).toBeNull();
+
+    agentsLoaded.resolve();
+    profilesLoaded.resolve();
+
+    await waitFor(() =>
+      expect(screen.getByTestId("custom-agent-chat-view")).toBeTruthy(),
+    );
+    expect(window.localStorage.getItem(`idea.agent-cell-mode.p1.${leafId}`)).toBe("custom");
+  });
 });