@copilotkit/react-core 1.51.4-next.7 → 1.51.4-next.8

package/src/hooks/__tests__/use-coagent-state-render.e2e.test.tsx

@@ -3,7 +3,11 @@ import { act, render, screen, waitFor, within } from "@testing-library/react";
 import { useCoAgentStateRender } from "../use-coagent-state-render";
 import { CoAgentStateRenderBridge } from "../use-coagent-state-render-bridge";
 import { useCopilotChatInternal } from "../use-copilot-chat_internal";
-import { CoAgentStateRendersProvider, CopilotContext, useCoAgentStateRenders } from "../../context";
+import {
+  CoAgentStateRendersProvider,
+  CopilotContext,
+  useCoAgentStateRenders,
+} from "../../context";
 import type { Claim } from "../use-coagent-state-render-bridge.helpers";
 import { createTestCopilotContext } from "../../test-helpers/copilot-context";
 
@@ -72,7 +76,9 @@ jest.mock("../use-lazy-tool-renderer", () => ({
 function TestHarness({ snapshot }: { snapshot: string }) {
   useCoAgentStateRender<{ current_step?: string }>({
     name: "test-agent",
-    render: ({ state }) => <div data-testid="state">{state.current_step ?? "none"}</div>,
+    render: ({ state }) => (
+      <div data-testid="state">{state.current_step ?? "none"}</div>
+    ),
   });
 
   return (
@@ -89,10 +95,18 @@ function TestHarness({ snapshot }: { snapshot: string }) {
   );
 }
 
-function SnapshotHarness({ snapshot, message }: { snapshot: string; message: TestMessage }) {
+function SnapshotHarness({
+  snapshot,
+  message,
+}: {
+  snapshot: string;
+  message: TestMessage;
+}) {
   useCoAgentStateRender<{ current_step?: string }>({
     name: "test-agent",
-    render: ({ state }) => <div data-testid="state">{state.current_step ?? "none"}</div>,
+    render: ({ state }) => (
+      <div data-testid="state">{state.current_step ?? "none"}</div>
+    ),
   });
 
   return (
@@ -109,10 +123,16 @@ function SnapshotHarness({ snapshot, message }: { snapshot: string; message: Tes
   );
 }
 
-function LiveStateHarness({ message }: { message: Pick<TestMessage, "id" | "role"> }) {
+function LiveStateHarness({
+  message,
+}: {
+  message: Pick<TestMessage, "id" | "role">;
+}) {
   useCoAgentStateRender<{ current_step?: string }>({
     name: "test-agent",
-    render: ({ state }) => <div data-testid="state">{state.current_step ?? "none"}</div>,
+    render: ({ state }) => (
+      <div data-testid="state">{state.current_step ?? "none"}</div>
+    ),
   });
 
   return (
@@ -132,7 +152,9 @@ function LiveStateHarness({ message }: { message: Pick<TestMessage, "id" | "role
 function NonFirstMessageHarness({ snapshot }: { snapshot: string }) {
   useCoAgentStateRender<{ current_step?: string }>({
     name: "test-agent",
-    render: ({ state }) => <div data-testid="state">{state.current_step ?? "none"}</div>,
+    render: ({ state }) => (
+      <div data-testid="state">{state.current_step ?? "none"}</div>
+    ),
   });
 
   return (
@@ -162,7 +184,9 @@ function MultiRunHarness({
 }) {
   useCoAgentStateRender<{ current_step?: string }>({
     name: "test-agent",
-    render: ({ state }) => <div data-testid="state">{state.current_step ?? "none"}</div>,
+    render: ({ state }) => (
+      <div data-testid="state">{state.current_step ?? "none"}</div>
+    ),
   });
 
   return (
@@ -179,7 +203,11 @@ function MultiRunHarness({
   );
 }
 
-function ClaimsObserver({ onChange }: { onChange: (claims: Record<string, Claim>) => void }) {
+function ClaimsObserver({
+  onChange,
+}: {
+  onChange: (claims: Record<string, Claim>) => void;
+}) {
   const { claimsRef } = useCoAgentStateRenders();
   React.useEffect(() => {
     onChange(claimsRef.current as Record<string, Claim>);
@@ -190,7 +218,9 @@ function ClaimsObserver({ onChange }: { onChange: (claims: Record<string, Claim>
 function ChatHarness({ tick }: { tick: number }) {
   useCoAgentStateRender<{ current_step?: string }>({
     name: "test-agent",
-    render: ({ state }) => <div data-testid="state">{state.current_step ?? "none"}</div>,
+    render: ({ state }) => (
+      <div data-testid="state">{state.current_step ?? "none"}</div>
+    ),
   });
 
   const { messages } = useCopilotChatInternal();
@@ -214,10 +244,12 @@ describe("useCoAgentStateRender", () => {
     lastSubscriber = null;
     mockAgent.state = {};
     mockAgent.messages = [];
-    mockAgent.subscribe.mockImplementation((subscriber: TestAgentSubscriber) => {
-      lastSubscriber = subscriber;
-      return { unsubscribe: jest.fn() };
-    });
+    mockAgent.subscribe.mockImplementation(
+      (subscriber: TestAgentSubscriber) => {
+        lastSubscriber = subscriber;
+        return { unsubscribe: jest.fn() };
+      },
+    );
   });
 
   it("re-renders when state snapshots change", async () => {
@@ -226,7 +258,9 @@ describe("useCoAgentStateRender", () => {
     const { rerender } = render(
       <CopilotContext.Provider value={copilotContextValue}>
         <CoAgentStateRendersProvider>
-          <TestHarness snapshot={JSON.stringify({ current_step: "Processing..." })} />
+          <TestHarness
+            snapshot={JSON.stringify({ current_step: "Processing..." })}
+          />
         </CoAgentStateRendersProvider>
       </CopilotContext.Provider>,
     );
@@ -238,7 +272,9 @@ describe("useCoAgentStateRender", () => {
     rerender(
       <CopilotContext.Provider value={copilotContextValue}>
         <CoAgentStateRendersProvider>
-          <TestHarness snapshot={JSON.stringify({ current_step: "Thinking..." })} />
+          <TestHarness
+            snapshot={JSON.stringify({ current_step: "Thinking..." })}
+          />
         </CoAgentStateRendersProvider>
       </CopilotContext.Provider>,
     );
@@ -347,7 +383,8 @@ describe("useCoAgentStateRender", () => {
       </CopilotContext.Provider>,
     );
 
-    const placeholderTestId = "message-coagent-state-render-test-agent-pending:msg-user-1";
+    const placeholderTestId =
+      "message-coagent-state-render-test-agent-pending:msg-user-1";
     expect(screen.queryByTestId(placeholderTestId)).toBeNull();
 
     mockAgent.isRunning = true;
@@ -411,7 +448,9 @@ describe("useCoAgentStateRender", () => {
 
     await waitFor(() => {
       expect(
-        screen.queryByTestId("message-coagent-state-render-test-agent-pending:msg-user-1"),
+        screen.queryByTestId(
+          "message-coagent-state-render-test-agent-pending:msg-user-1",
+        ),
       ).toBeNull();
     });
   });
@@ -422,7 +461,9 @@ describe("useCoAgentStateRender", () => {
     render(
       <CopilotContext.Provider value={copilotContextValue}>
         <CoAgentStateRendersProvider>
-          <NonFirstMessageHarness snapshot={JSON.stringify({ current_step: "Processing..." })} />
+          <NonFirstMessageHarness
+            snapshot={JSON.stringify({ current_step: "Processing..." })}
+          />
         </CoAgentStateRendersProvider>
       </CopilotContext.Provider>,
     );
@@ -433,7 +474,9 @@ describe("useCoAgentStateRender", () => {
   });
 
   it("falls back to legacy renderer when renderCustomMessages throws", async () => {
-    const { useRenderCustomMessages } = jest.requireMock("@copilotkitnext/react");
+    const { useRenderCustomMessages } = jest.requireMock(
+      "@copilotkitnext/react",
+    );
     useRenderCustomMessages.mockImplementationOnce(() => () => {
       throw new Error("boom");
     });
@@ -464,7 +507,9 @@ describe("useCoAgentStateRender", () => {
   });
 
   it("prefers legacy renderer over renderCustomMessages when both exist", async () => {
-    const { useRenderCustomMessages } = jest.requireMock("@copilotkitnext/react");
+    const { useRenderCustomMessages } = jest.requireMock(
+      "@copilotkitnext/react",
+    );
     const renderCustomSpy = jest.fn(() => null);
     useRenderCustomMessages.mockImplementationOnce(() => renderCustomSpy);
 
@@ -544,8 +589,18 @@ describe("useCoAgentStateRender", () => {
     render(
       <CopilotContext.Provider value={copilotContextValue}>
         <CoAgentStateRendersProvider>
-          <MultiRunHarness snapshot={snapshot} runId="run-1" messageId="msg-1" messageIndex={0} />
-          <MultiRunHarness snapshot={snapshot} runId="run-2" messageId="msg-2" messageIndex={1} />
+          <MultiRunHarness
+            snapshot={snapshot}
+            runId="run-1"
+            messageId="msg-1"
+            messageIndex={0}
+          />
+          <MultiRunHarness
+            snapshot={snapshot}
+            runId="run-2"
+            messageId="msg-2"
+            messageIndex={1}
+          />
         </CoAgentStateRendersProvider>
       </CopilotContext.Provider>,
     );
@@ -566,8 +621,18 @@ describe("useCoAgentStateRender", () => {
     render(
       <CopilotContext.Provider value={copilotContextValue}>
         <CoAgentStateRendersProvider>
-          <MultiRunHarness snapshot={snapshot} runId="run-1" messageId="msg-1" messageIndex={0} />
-          <MultiRunHarness snapshot={snapshot} runId="run-1" messageId="msg-2" messageIndex={1} />
+          <MultiRunHarness
+            snapshot={snapshot}
+            runId="run-1"
+            messageId="msg-1"
+            messageIndex={0}
+          />
+          <MultiRunHarness
+            snapshot={snapshot}
+            runId="run-1"
+            messageId="msg-2"
+            messageIndex={1}
+          />
         </CoAgentStateRendersProvider>
       </CopilotContext.Provider>,
     );
@@ -590,8 +655,18 @@ describe("useCoAgentStateRender", () => {
       <CopilotContext.Provider value={copilotContextValue}>
         <CoAgentStateRendersProvider>
           <ClaimsObserver onChange={(claims) => (latestClaims = claims)} />
-          <MultiRunHarness snapshot={snapshot} runId="run-1" messageId="msg-1" messageIndex={0} />
-          <MultiRunHarness snapshot={snapshot} runId="run-1" messageId="msg-2" messageIndex={1} />
+          <MultiRunHarness
+            snapshot={snapshot}
+            runId="run-1"
+            messageId="msg-1"
+            messageIndex={0}
+          />
+          <MultiRunHarness
+            snapshot={snapshot}
+            runId="run-1"
+            messageId="msg-2"
+            messageIndex={1}
+          />
         </CoAgentStateRendersProvider>
       </CopilotContext.Provider>,
     );
@@ -613,28 +688,49 @@ describe("useCoAgentStateRender", () => {
       <CopilotContext.Provider value={copilotContextValue}>
         <CoAgentStateRendersProvider>
           <ClaimsObserver onChange={(claims) => (latestClaims = claims)} />
-          <MultiRunHarness snapshot={snapshot} runId="run-1" messageId="msg-1" messageIndex={0} />
+          <MultiRunHarness
+            snapshot={snapshot}
+            runId="run-1"
+            messageId="msg-1"
+            messageIndex={0}
+          />
         </CoAgentStateRendersProvider>
       </CopilotContext.Provider>,
     );
 
     await waitFor(() => {
-      expect(latestClaims["msg-1"]?.stateSnapshot?.current_step).toBe("Processing...");
+      expect(latestClaims["msg-1"]?.stateSnapshot?.current_step).toBe(
+        "Processing...",
+      );
     });
 
     rerender(
       <CopilotContext.Provider value={copilotContextValue}>
         <CoAgentStateRendersProvider>
           <ClaimsObserver onChange={(claims) => (latestClaims = claims)} />
-          <MultiRunHarness snapshot={snapshot} runId="run-1" messageId="msg-1" messageIndex={0} />
-          <MultiRunHarness snapshot={snapshot} runId="run-2" messageId="msg-2" messageIndex={1} />
+          <MultiRunHarness
+            snapshot={snapshot}
+            runId="run-1"
+            messageId="msg-1"
+            messageIndex={0}
+          />
+          <MultiRunHarness
+            snapshot={snapshot}
+            runId="run-2"
+            messageId="msg-2"
+            messageIndex={1}
+          />
         </CoAgentStateRendersProvider>
       </CopilotContext.Provider>,
     );
 
     await waitFor(() => {
-      expect(latestClaims["msg-1"]?.stateSnapshot?.current_step).toBe("Processing...");
-      expect(latestClaims["msg-2"]?.stateSnapshot?.current_step).toBe("Processing...");
+      expect(latestClaims["msg-1"]?.stateSnapshot?.current_step).toBe(
+        "Processing...",
+      );
+      expect(latestClaims["msg-2"]?.stateSnapshot?.current_step).toBe(
+        "Processing...",
+      );
     });
   });
 
@@ -660,7 +756,9 @@ describe("useCoAgentStateRender", () => {
     );
 
     await waitFor(() => {
-      expect(latestClaims["msg-1"]?.stateSnapshot?.current_step).toBe("Processing...");
+      expect(latestClaims["msg-1"]?.stateSnapshot?.current_step).toBe(
+        "Processing...",
+      );
     });
 
     rerender(
@@ -684,8 +782,12 @@ describe("useCoAgentStateRender", () => {
     );
 
     await waitFor(() => {
-      expect(latestClaims["msg-1"]?.stateSnapshot?.current_step).toBe("Processing...");
-      expect(latestClaims["msg-2"]?.stateSnapshot?.current_step).toBe("Finalizing...");
+      expect(latestClaims["msg-1"]?.stateSnapshot?.current_step).toBe(
+        "Processing...",
+      );
+      expect(latestClaims["msg-2"]?.stateSnapshot?.current_step).toBe(
+        "Finalizing...",
+      );
     });
   });
 
@@ -704,7 +806,12 @@ describe("useCoAgentStateRender", () => {
       <CopilotContext.Provider value={copilotContextValue}>
         <CoAgentStateRendersProvider>
           <ClaimsObserver onChange={(claims) => (latestClaims = claims)} />
-          <MultiRunHarness snapshot={snapshot} runId="run-1" messageId="msg-1" messageIndex={0} />
+          <MultiRunHarness
+            snapshot={snapshot}
+            runId="run-1"
+            messageId="msg-1"
+            messageIndex={0}
+          />
         </CoAgentStateRendersProvider>
       </CopilotContext.Provider>,
     );
@@ -739,7 +846,9 @@ describe("useCoAgentStateRender", () => {
     );
 
     await waitFor(() => {
-      expect(latestClaims["msg-assistant-1"]?.stateSnapshot?.current_step).toBe("First");
+      expect(latestClaims["msg-assistant-1"]?.stateSnapshot?.current_step).toBe(
+        "First",
+      );
     });
 
     mockAgent.messages = [
@@ -759,7 +868,9 @@ describe("useCoAgentStateRender", () => {
     );
 
     await waitFor(() => {
-      expect(latestClaims["msg-assistant-1"]?.stateSnapshot?.current_step).toBe("First");
+      expect(latestClaims["msg-assistant-1"]?.stateSnapshot?.current_step).toBe(
+        "First",
+      );
     });
   });
 
@@ -883,7 +994,9 @@ describe("useCoAgentStateRender", () => {
 
     await waitFor(() => {
       const message = screen.getByTestId("message-msg-assistant-1");
-      expect(within(message).getByTestId("state").textContent).toBe("First run");
+      expect(within(message).getByTestId("state").textContent).toBe(
+        "First run",
+      );
     });
 
     mockAgent.messages = [
@@ -912,7 +1025,12 @@ describe("useCoAgentStateRender", () => {
       { id: "msg-user-1", role: "user", content: "Hi" },
       { id: "msg-assistant-1", role: "assistant", content: "" },
       { id: "msg-user-2", role: "user", content: "Next" },
-      { id: "msg-assistant-2", role: "assistant", content: "", state: "{\"current_step\":\"Second run\"}" },
+      {
+        id: "msg-assistant-2",
+        role: "assistant",
+        content: "",
+        state: '{"current_step":"Second run"}',
+      },
     ];
 
     rerender(
@@ -925,7 +1043,9 @@ describe("useCoAgentStateRender", () => {
 
     await waitFor(() => {
       const message = screen.getByTestId("message-msg-assistant-2");
-      expect(within(message).getByTestId("state").textContent).toBe("Second run");
+      expect(within(message).getByTestId("state").textContent).toBe(
+        "Second run",
+      );
     });
   });
 
@@ -952,7 +1072,9 @@ describe("useCoAgentStateRender", () => {
 
     await waitFor(() => {
       const message = screen.getByTestId("message-msg-assistant-1");
-      expect(within(message).getByTestId("state").textContent).toBe("First run");
+      expect(within(message).getByTestId("state").textContent).toBe(
+        "First run",
+      );
     });
 
     mockAgent.messages = [
@@ -985,7 +1107,9 @@ describe("useCoAgentStateRender", () => {
 
     await waitFor(() => {
       const message = screen.getByTestId("message-msg-assistant-2");
-      expect(within(message).getByTestId("state").textContent).toBe("Second run");
+      expect(within(message).getByTestId("state").textContent).toBe(
+        "Second run",
+      );
     });
   });
 
@@ -1005,7 +1129,7 @@ describe("useCoAgentStateRender", () => {
         id: "msg-assistant-1",
         role: "assistant",
         content: "",
-        state: "{\"current_step\":\"Processing...\"}",
+        state: '{"current_step":"Processing..."}',
       },
     ];
     mockAgent.isRunning = true;
@@ -1021,7 +1145,9 @@ describe("useCoAgentStateRender", () => {
 
     await waitFor(() => {
       const message = screen.getByTestId("message-msg-assistant-1");
-      expect(within(message).getByTestId("state").textContent).toBe("Processing...");
+      expect(within(message).getByTestId("state").textContent).toBe(
+        "Processing...",
+      );
     });
 
     mockAgent.messages = [
@@ -1058,7 +1184,9 @@ describe("useCoAgentStateRender", () => {
     render(
       <CopilotContext.Provider value={copilotContextValue}>
         <CoAgentStateRendersProvider>
-          <LiveStateHarness message={{ id: "msg-live-empty", role: "assistant" }} />
+          <LiveStateHarness
+            message={{ id: "msg-live-empty", role: "assistant" }}
+          />
         </CoAgentStateRendersProvider>
       </CopilotContext.Provider>,
     );

package/src/hooks/__tests__/use-coagent-state-render.test.tsx

@@ -20,7 +20,9 @@ jest.mock("../../components/toast/toast-provider", () => ({
   }),
 }));
 
-function createWrapper(copilotContextValue: ReturnType<typeof createTestCopilotContext>) {
+function createWrapper(
+  copilotContextValue: ReturnType<typeof createTestCopilotContext>,
+) {
   return function Wrapper({ children }: { children: ReactNode }) {
     return (
       <CopilotContext.Provider value={copilotContextValue}>
@@ -30,7 +32,10 @@ function createWrapper(copilotContextValue: ReturnType<typeof createTestCopilotC
   };
 }
 
-function useHarness<T>(action: Parameters<typeof useCoAgentStateRender<T>>[0], deps?: unknown[]) {
+function useHarness<T>(
+  action: Parameters<typeof useCoAgentStateRender<T>>[0],
+  deps?: unknown[],
+) {
   useCoAgentStateRender(action, deps);
   return useCoAgentStateRenders();
 }
@@ -58,11 +63,15 @@ describe("useCoAgentStateRender (hook behaviors)", () => {
   beforeEach(() => {
     jest.clearAllMocks();
     idCounter = 0;
-    (randomId as jest.Mock).mockImplementation(() => `test-random-id-${++idCounter}`);
+    (randomId as jest.Mock).mockImplementation(
+      () => `test-random-id-${++idCounter}`,
+    );
   });
 
   it("registers state render and writes to the render cache", async () => {
-    const chatComponentsCache = { current: { actions: {}, coAgentStateRenders: {} } };
+    const chatComponentsCache = {
+      current: { actions: {}, coAgentStateRenders: {} },
+    };
     const wrapper = createWrapper(
       createTestCopilotContext({
         chatComponentsCache,
@@ -85,11 +94,15 @@ describe("useCoAgentStateRender (hook behaviors)", () => {
       expect(Object.keys(result.current.coAgentStateRenders)).toHaveLength(1);
     });
 
-    expect(chatComponentsCache.current.coAgentStateRenders["agent-a-node-1"]).toBe(renderFn);
+    expect(
+      chatComponentsCache.current.coAgentStateRenders["agent-a-node-1"],
+    ).toBe(renderFn);
   });
 
   it("mutates handler + cache in place when dependencies are omitted", async () => {
-    const chatComponentsCache = { current: { actions: {}, coAgentStateRenders: {} } };
+    const chatComponentsCache = {
+      current: { actions: {}, coAgentStateRenders: {} },
+    };
     const wrapper = createWrapper(
       createTestCopilotContext({
         chatComponentsCache,
@@ -122,13 +135,17 @@ describe("useCoAgentStateRender (hook behaviors)", () => {
     const [id, initialRender] = getSingleEntry(initialRenders);
 
     expect(initialRender.handler).toBe(handlerOne);
-    expect(chatComponentsCache.current.coAgentStateRenders["agent-b-global"]).toBe(renderOne);
+    expect(
+      chatComponentsCache.current.coAgentStateRenders["agent-b-global"],
+    ).toBe(renderOne);
 
     rerender({ handler: handlerTwo, renderFn: renderTwo });
 
     expect(result.current.coAgentStateRenders).toBe(initialRenders);
     expect(result.current.coAgentStateRenders[id].handler).toBe(handlerTwo);
-    expect(chatComponentsCache.current.coAgentStateRenders["agent-b-global"]).toBe(renderTwo);
+    expect(
+      chatComponentsCache.current.coAgentStateRenders["agent-b-global"],
+    ).toBe(renderTwo);
   });
 
   it("re-registers when dependencies change", async () => {
@@ -169,7 +186,9 @@ describe("useCoAgentStateRender (hook behaviors)", () => {
   });
 
   it("re-registers when string render changes", async () => {
-    const chatComponentsCache = { current: { actions: {}, coAgentStateRenders: {} } };
+    const chatComponentsCache = {
+      current: { actions: {}, coAgentStateRenders: {} },
+    };
     const wrapper = createWrapper(
       createTestCopilotContext({
         chatComponentsCache,
@@ -199,7 +218,9 @@ describe("useCoAgentStateRender (hook behaviors)", () => {
       expect(result.current.coAgentStateRenders).not.toBe(initialRenders);
     });
 
-    expect(chatComponentsCache.current.coAgentStateRenders["agent-d-global"]).toBe("Step 2");
+    expect(
+      chatComponentsCache.current.coAgentStateRenders["agent-d-global"],
+    ).toBe("Step 2");
   });
 
   it("warns when duplicate registrations target the same agent + node", async () => {

package/src/hooks/__tests__/use-frontend-tool-remount.e2e.test.tsx

@@ -49,7 +49,13 @@ function ToolRenderHost() {
   );
 }
 
-function RunActionButton({ onMount, onUnmount }: { onMount: jest.Mock; onUnmount: jest.Mock }) {
+function RunActionButton({
+  onMount,
+  onUnmount,
+}: {
+  onMount: jest.Mock;
+  onUnmount: jest.Mock;
+}) {
   useEffect(() => {
     onMount();
     return () => onUnmount();
@@ -68,7 +74,9 @@ describe("useFrontendTool dependency changes", () => {
         {
           name: "actionOne",
           description: "Execute action one",
-          render: () => <RunActionButton onMount={mounted} onUnmount={unmounted} />,
+          render: () => (
+            <RunActionButton onMount={mounted} onUnmount={unmounted} />
+          ),
         },
         [version],
       );