@copilotkit/react-core 1.51.4 → 1.51.5-next.1

package/dist/v2/index.cjs

@@ -0,0 +1,17 @@
+
+
+var _copilotkitnext_core = require("@copilotkitnext/core");
+Object.keys(_copilotkitnext_core).forEach(function (k) {
+  if (k !== 'default' && !Object.prototype.hasOwnProperty.call(exports, k)) Object.defineProperty(exports, k, {
+    enumerable: true,
+    get: function () { return _copilotkitnext_core[k]; }
+  });
+});
+
+var _copilotkitnext_react = require("@copilotkitnext/react");
+Object.keys(_copilotkitnext_react).forEach(function (k) {
+  if (k !== 'default' && !Object.prototype.hasOwnProperty.call(exports, k)) Object.defineProperty(exports, k, {
+    enumerable: true,
+    get: function () { return _copilotkitnext_react[k]; }
+  });
+});

package/dist/v2/index.css

@@ -1,4 +1,3 @@
-@import "@copilotkitnext/react/styles.css";
 
-/* src/v2/index.css */
-/*# sourceMappingURL=index.css.map */
+
+/*# sourceMappingURL=index.css.map*/

package/dist/v2/index.css.map

@@ -1 +1 @@
-{"version":3,"sources":[],"sourcesContent":[],"mappings":"","names":[]}
+{"version":3,"file":"index.css","names":[],"sources":["../../src/v2/index.css"],"sourcesContent":["@import \"@copilotkitnext/react/styles.css\";"],"mappings":""}

package/dist/v2/index.d.cts

@@ -0,0 +1,2 @@
+export * from "@copilotkitnext/react";
+export * from "@copilotkitnext/core";

package/dist/v2/index.d.mts

@@ -0,0 +1,2 @@
+export * from "@copilotkitnext/react";
+export * from "@copilotkitnext/core";

package/dist/v2/index.mjs

@@ -1,6 +1,5 @@
-import "../chunk-SKC7AJIV.mjs";
+export * from "@copilotkitnext/react"
 
-// src/v2/index.ts
-export * from "@copilotkitnext/core";
-export * from "@copilotkitnext/react";
-//# sourceMappingURL=index.mjs.map
+export * from "@copilotkitnext/core"
+
+export {  };

package/dist/v2/index.umd.js

@@ -1,2 +1,24 @@
-!function(e,t){"object"==typeof exports&&"undefined"!=typeof module?t(exports,require("@copilotkitnext/core"),require("@copilotkitnext/react")):"function"==typeof define&&define.amd?define(["exports","@copilotkitnext/core","@copilotkitnext/react"],t):t((e="undefined"!=typeof globalThis?globalThis:e||self).CopilotKitReactCoreV2={},e.CopilotKitNextCore,e.CopilotKitNextReact)}(this,function(e,t,o){"use strict";!function(e,t){void 0===t&&(t={});var o=t.insertAt;if("undefined"!=typeof document){var i=document.head||document.getElementsByTagName("head")[0],n=document.createElement("style");n.type="text/css","top"===o&&i.firstChild?i.insertBefore(n,i.firstChild):i.appendChild(n),n.styleSheet?n.styleSheet.cssText=e:n.appendChild(document.createTextNode(e))}}('@import "@copilotkitnext/react/styles.css";'),Object.keys(t).forEach(function(o){"default"===o||Object.prototype.hasOwnProperty.call(e,o)||Object.defineProperty(e,o,{enumerable:!0,get:function(){return t[o]}})}),Object.keys(o).forEach(function(t){"default"===t||Object.prototype.hasOwnProperty.call(e,t)||Object.defineProperty(e,t,{enumerable:!0,get:function(){return o[t]}})})});
-//# sourceMappingURL=index.umd.js.map
+(function(factory) {
+  
+  typeof define === 'function' && define.amd ? define([], factory) :
+  factory();
+})(function() {
+
+
+var _copilotkitnext_core = require("@copilotkitnext/core");
+Object.keys(_copilotkitnext_core).forEach(function (k) {
+  if (k !== 'default' && !Object.prototype.hasOwnProperty.call(exports, k)) Object.defineProperty(exports, k, {
+    enumerable: true,
+    get: function () { return _copilotkitnext_core[k]; }
+  });
+});
+
+var _copilotkitnext_react = require("@copilotkitnext/react");
+Object.keys(_copilotkitnext_react).forEach(function (k) {
+  if (k !== 'default' && !Object.prototype.hasOwnProperty.call(exports, k)) Object.defineProperty(exports, k, {
+    enumerable: true,
+    get: function () { return _copilotkitnext_react[k]; }
+  });
+});
+
+});

package/package.json

@@ -6,31 +6,31 @@
     "type": "git",
     "url": "https://github.com/CopilotKit/CopilotKit.git"
   },
+  "type": "module",
   "publishConfig": {
     "access": "public"
   },
-  "version": "1.51.4",
+  "version": "1.51.5-next.1",
   "sideEffects": [
     "**/*.css"
   ],
-  "main": "./dist/index.js",
+  "main": "./dist/index.cjs",
   "module": "./dist/index.mjs",
   "exports": {
     ".": {
       "import": "./dist/index.mjs",
-      "require": "./dist/index.js",
-      "types": "./dist/index.d.ts"
+      "require": "./dist/index.cjs"
     },
     "./v2": {
       "import": "./dist/v2/index.mjs",
-      "require": "./dist/v2/index.js",
-      "types": "./dist/v2/index.d.ts"
+      "require": "./dist/v2/index.cjs"
     },
+    "./package.json": "./package.json",
     "./v2/styles.css": "./dist/v2/index.css"
   },
   "unpkg": "./dist/index.umd.js",
   "jsdelivr": "./dist/index.umd.js",
-  "types": "./dist/index.d.ts",
+  "types": "./dist/index.d.cts",
   "license": "MIT",
   "peerDependencies": {
     "react": "^18 || ^19 || ^19.0.0-rc",
@@ -40,31 +40,28 @@
   "devDependencies": {
     "@testing-library/react": "^16.3.0",
     "@testing-library/react-hooks": "^8.0.1",
-    "@types/jest": "^29.5.4",
     "@types/react": "^19.1.0",
     "@types/react-dom": "^19.0.2",
     "eslint": "^8.56.0",
-    "identity-obj-proxy": "^3.0.0",
-    "jest": "^29.6.4",
-    "jest-environment-jsdom": "^30.0.2",
+    "jsdom": "^26.1.0",
     "react": "^19.1.0",
     "react-dom": "^19.2.0",
-    "ts-jest": "^29.1.1",
-    "tsup": "^6.7.0",
+    "vitest": "^3.2.4",
+    "tsdown": "^0.20.3",
     "typescript": "^5.2.3",
     "zod": ">=3.0.0",
     "eslint-config-custom": "1.4.8",
     "tsconfig": "1.4.8"
   },
   "dependencies": {
-    "@ag-ui/client": "^0.0.43",
+    "@ag-ui/client": "^0.0.45",
     "@scarf/scarf": "^1.3.0",
     "react-markdown": "^8.0.7",
     "untruncate-json": "^0.0.1",
-    "@copilotkit/runtime-client-gql": "1.51.4",
-    "@copilotkit/shared": "1.51.4",
-    "@copilotkitnext/core": "1.51.4",
-    "@copilotkitnext/react": "1.51.4"
+    "@copilotkit/runtime-client-gql": "1.51.5-next.1",
+    "@copilotkit/shared": "1.51.5-next.1",
+    "@copilotkitnext/core": "1.51.5-next.1",
+    "@copilotkitnext/react": "1.51.5-next.1"
   },
   "keywords": [
     "copilotkit",
@@ -79,11 +76,14 @@
     "textarea"
   ],
   "scripts": {
-    "build": "tsup --clean && rollup -c rollup.config.mjs",
-    "dev": "tsup --watch",
-    "test": "jest",
+    "build": "tsdown",
+    "dev": "tsdown --watch",
+    "test": "vitest run",
+    "test:watch": "vitest",
     "check-types": "tsc --noEmit",
     "link:global": "pnpm link --global",
-    "unlink:global": "pnpm unlink --global"
+    "unlink:global": "pnpm unlink --global",
+    "publint": "publint .",
+    "attw": "attw --pack . --profile node16"
   }
 }

package/src/components/copilot-provider/__tests__/copilotkit-error.test.tsx

@@ -1,11 +1,11 @@
-/// <reference types="jest" />
+import { vi, type Mock } from "vitest";
 import { CopilotErrorEvent, CopilotErrorHandler } from "@copilotkit/shared";
 
 describe("CopilotKit onError types", () => {
-  let mockError: jest.Mock;
+  let mockError: Mock;
 
   beforeEach(() => {
-    mockError = jest.fn();
+    mockError = vi.fn();
   });
 
   describe("onError type checking", () => {

package/src/hooks/__tests__/use-coagent-config.test.ts

@@ -1,13 +1,14 @@
+import { vi } from "vitest";
 import { renderHook, waitFor } from "@testing-library/react";
 import { useCoAgent } from "../use-coagent";
 import type { AgentSubscriber } from "@ag-ui/client";
 
 // Mock functions for @copilotkitnext/react
-const mockSetState = jest.fn();
-const mockRunAgent = jest.fn();
-const mockAbortRun = jest.fn();
-const mockSubscribe = jest.fn();
-const mockSetProperties = jest.fn();
+const mockSetState = vi.fn();
+const mockRunAgent = vi.fn();
+const mockAbortRun = vi.fn();
+const mockSubscribe = vi.fn();
+const mockSetProperties = vi.fn();
 
 // Store the last subscriber for triggering events
 let lastSubscriber: AgentSubscriber | null = null;
@@ -23,14 +24,14 @@ const mockAgent = {
   subscribe: mockSubscribe.mockImplementation((subscriber: AgentSubscriber) => {
     lastSubscriber = subscriber;
     return {
-      unsubscribe: jest.fn(),
+      unsubscribe: vi.fn(),
     };
   }),
 };
 
-jest.mock("@copilotkitnext/react", () => ({
-  useAgent: jest.fn(() => ({ agent: mockAgent })),
-  useCopilotKit: jest.fn(() => ({
+vi.mock("@copilotkitnext/react", () => ({
+  useAgent: vi.fn(() => ({ agent: mockAgent })),
+  useCopilotKit: vi.fn(() => ({
     copilotkit: {
       setProperties: mockSetProperties,
     },
@@ -38,26 +39,26 @@ jest.mock("@copilotkitnext/react", () => ({
 }));
 
 // Mock other dependencies
-const mockAppendMessage = jest.fn();
-const mockRunChatCompletion = jest.fn();
+const mockAppendMessage = vi.fn();
+const mockRunChatCompletion = vi.fn();
 
-jest.mock("../use-copilot-chat_internal", () => ({
+vi.mock("../use-copilot-chat_internal", () => ({
   useCopilotChat: () => ({
     appendMessage: mockAppendMessage,
     runChatCompletion: mockRunChatCompletion,
   }),
 }));
 
-jest.mock("../use-copilot-runtime-client", () => ({
+vi.mock("../use-copilot-runtime-client", () => ({
   useCopilotRuntimeClient: () => ({
-    loadAgentState: jest.fn().mockResolvedValue({
+    loadAgentState: vi.fn().mockResolvedValue({
       data: { loadAgentState: { state: "{}", threadExists: false } },
       error: null,
     }),
   }),
 }));
 
-jest.mock("../../context", () => ({
+vi.mock("../../context", () => ({
   useCopilotContext: () => ({
     availableAgents: [],
     coagentStates: {},
@@ -75,26 +76,26 @@ jest.mock("../../context", () => ({
   }),
 }));
 
-jest.mock("../../components/toast/toast-provider", () => ({
+vi.mock("../../components/toast/toast-provider", () => ({
   useToast: () => ({
-    setBannerError: jest.fn(),
+    setBannerError: vi.fn(),
   }),
 }));
 
-jest.mock("../../components/error-boundary/error-utils", () => ({
+vi.mock("../../components/error-boundary/error-utils", () => ({
   useAsyncCallback: (fn: any) => fn,
 }));
 
-jest.mock("../../components/copilot-provider/copilot-messages", () => ({
+vi.mock("../../components/copilot-provider/copilot-messages", () => ({
   useMessagesTap: () => ({
-    getMessagesFromTap: jest.fn(() => []),
-    updateTapMessages: jest.fn(),
+    getMessagesFromTap: vi.fn(() => []),
+    updateTapMessages: vi.fn(),
   }),
 }));
 
 describe("useCoAgent config synchronization", () => {
   beforeEach(() => {
-    jest.clearAllMocks();
+    vi.clearAllMocks();
     lastSubscriber = null;
   });
 

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

@@ -1,3 +1,4 @@
+import { vi } from "vitest";
 import React from "react";
 import { act, render, screen, waitFor, within } from "@testing-library/react";
 import { useCoAgentStateRender } from "../use-coagent-state-render";
@@ -10,6 +11,7 @@ import {
 } from "../../context";
 import type { Claim } from "../use-coagent-state-render-bridge.helpers";
 import { createTestCopilotContext } from "../../test-helpers/copilot-context";
+import { useRenderCustomMessages } from "@copilotkitnext/react";
 
 type TestMessage = {
   id: string;
@@ -27,50 +29,50 @@ const mockAgent = {
   messages: [] as TestMessage[],
   state: {},
   isRunning: true,
-  subscribe: jest.fn(),
-  setMessages: jest.fn(),
-  setState: jest.fn(),
-  addMessage: jest.fn(),
-  abortRun: jest.fn(),
-  runAgent: jest.fn(),
+  subscribe: vi.fn(),
+  setMessages: vi.fn(),
+  setState: vi.fn(),
+  addMessage: vi.fn(),
+  abortRun: vi.fn(),
+  runAgent: vi.fn(),
 };
 
 let lastSubscriber: TestAgentSubscriber | null = null;
 
-jest.mock("@copilotkitnext/react", () => ({
-  useAgent: jest.fn(() => ({ agent: mockAgent })),
-  useCopilotKit: jest.fn(() => ({
+vi.mock("@copilotkitnext/react", () => ({
+  useAgent: vi.fn(() => ({ agent: mockAgent })),
+  useCopilotKit: vi.fn(() => ({
     copilotkit: {
-      connectAgent: jest.fn(),
-      getRunIdForMessage: jest.fn(),
-      runAgent: jest.fn(),
-      clearSuggestions: jest.fn(),
-      addSuggestionsConfig: jest.fn(),
-      reloadSuggestions: jest.fn(),
+      connectAgent: vi.fn(),
+      getRunIdForMessage: vi.fn(),
+      runAgent: vi.fn(),
+      clearSuggestions: vi.fn(),
+      addSuggestionsConfig: vi.fn(),
+      reloadSuggestions: vi.fn(),
     },
   })),
-  useCopilotChatConfiguration: jest.fn(() => ({ agentId: "test-agent" })),
-  useRenderCustomMessages: jest.fn(() => undefined),
-  useSuggestions: jest.fn(() => ({ suggestions: [], isLoading: false })),
+  useCopilotChatConfiguration: vi.fn(() => ({ agentId: "test-agent" })),
+  useRenderCustomMessages: vi.fn(() => undefined),
+  useSuggestions: vi.fn(() => ({ suggestions: [], isLoading: false })),
 }));
 
-jest.mock("../../components/toast/toast-provider", () => ({
+vi.mock("../../components/toast/toast-provider", () => ({
   useToast: () => ({
-    setBannerError: jest.fn(),
-    addToast: jest.fn(),
+    setBannerError: vi.fn(),
+    addToast: vi.fn(),
   }),
 }));
 
-jest.mock("../../components/error-boundary/error-utils", () => ({
+vi.mock("../../components/error-boundary/error-utils", () => ({
   useAsyncCallback: <T extends (...args: unknown[]) => unknown>(fn: T) => fn,
 }));
 
-jest.mock("../use-langgraph-interrupt-render", () => ({
-  useLangGraphInterruptRender: jest.fn(() => null),
+vi.mock("../use-langgraph-interrupt-render", () => ({
+  useLangGraphInterruptRender: vi.fn(() => null),
 }));
 
-jest.mock("../use-lazy-tool-renderer", () => ({
-  useLazyToolRenderer: jest.fn(() => () => null),
+vi.mock("../use-lazy-tool-renderer", () => ({
+  useLazyToolRenderer: vi.fn(() => () => null),
 }));
 
 function TestHarness({ snapshot }: { snapshot: string }) {
@@ -247,7 +249,7 @@ describe("useCoAgentStateRender", () => {
     mockAgent.subscribe.mockImplementation(
       (subscriber: TestAgentSubscriber) => {
         lastSubscriber = subscriber;
-        return { unsubscribe: jest.fn() };
+        return { unsubscribe: vi.fn() };
       },
     );
   });
@@ -474,10 +476,7 @@ describe("useCoAgentStateRender", () => {
   });
 
   it("falls back to legacy renderer when renderCustomMessages throws", async () => {
-    const { useRenderCustomMessages } = jest.requireMock(
-      "@copilotkitnext/react",
-    );
-    useRenderCustomMessages.mockImplementationOnce(() => () => {
+    vi.mocked(useRenderCustomMessages).mockImplementationOnce(() => () => {
       throw new Error("boom");
     });
 
@@ -507,11 +506,10 @@ describe("useCoAgentStateRender", () => {
   });
 
   it("prefers legacy renderer over renderCustomMessages when both exist", async () => {
-    const { useRenderCustomMessages } = jest.requireMock(
-      "@copilotkitnext/react",
+    const renderCustomSpy = vi.fn(() => null);
+    vi.mocked(useRenderCustomMessages).mockImplementationOnce(
+      () => renderCustomSpy,
     );
-    const renderCustomSpy = jest.fn(() => null);
-    useRenderCustomMessages.mockImplementationOnce(() => renderCustomSpy);
 
     const copilotContextValue = createTestCopilotContext({
       threadId: "thread-1",

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

@@ -1,3 +1,4 @@
+import { vi, type Mock } from "vitest";
 import React, { type ReactNode } from "react";
 import { render, renderHook, waitFor } from "@testing-library/react";
 import { useCoAgentStateRender } from "../use-coagent-state-render";
@@ -10,10 +11,10 @@ import {
 import { CopilotKitAgentDiscoveryError, randomId } from "@copilotkit/shared";
 import { createTestCopilotContext } from "../../test-helpers/copilot-context";
 
-const addToast = jest.fn();
-const setBannerError = jest.fn();
+const addToast = vi.fn();
+const setBannerError = vi.fn();
 
-jest.mock("../../components/toast/toast-provider", () => ({
+vi.mock("../../components/toast/toast-provider", () => ({
   useToast: () => ({
     addToast,
     setBannerError,
@@ -61,9 +62,9 @@ describe("useCoAgentStateRender (hook behaviors)", () => {
   let idCounter = 0;
 
   beforeEach(() => {
-    jest.clearAllMocks();
+    vi.clearAllMocks();
     idCounter = 0;
-    (randomId as jest.Mock).mockImplementation(
+    (randomId as Mock).mockImplementation(
       () => `test-random-id-${++idCounter}`,
     );
   });
@@ -78,7 +79,7 @@ describe("useCoAgentStateRender (hook behaviors)", () => {
       }),
     );
 
-    const renderFn = jest.fn(() => null);
+    const renderFn = vi.fn(() => null);
 
     const { result } = renderHook(
       () =>
@@ -109,10 +110,10 @@ describe("useCoAgentStateRender (hook behaviors)", () => {
       }),
     );
 
-    const handlerOne = jest.fn();
-    const handlerTwo = jest.fn();
-    const renderOne = jest.fn(() => null);
-    const renderTwo = jest.fn(() => null);
+    const handlerOne = vi.fn();
+    const handlerTwo = vi.fn();
+    const renderOne = vi.fn(() => null);
+    const renderTwo = vi.fn(() => null);
 
     const { result, rerender } = renderHook(
       ({ handler, renderFn }) =>
@@ -151,8 +152,8 @@ describe("useCoAgentStateRender (hook behaviors)", () => {
   it("re-registers when dependencies change", async () => {
     const wrapper = createWrapper(createTestCopilotContext());
 
-    const handlerOne = jest.fn();
-    const handlerTwo = jest.fn();
+    const handlerOne = vi.fn();
+    const handlerTwo = vi.fn();
 
     const { result, rerender } = renderHook(
       ({ deps, handler }) =>
@@ -233,14 +234,14 @@ describe("useCoAgentStateRender (hook behaviors)", () => {
             action={{
               name: "agent-dup",
               nodeName: "node-x",
-              handler: jest.fn(),
+              handler: vi.fn(),
             }}
           />
           <HookUser
             action={{
               name: "agent-dup",
               nodeName: "node-x",
-              handler: jest.fn(),
+              handler: vi.fn(),
             }}
           />
         </>
@@ -278,14 +279,14 @@ describe("useCoAgentStateRender (hook behaviors)", () => {
             action={{
               name: "agent-ok",
               nodeName: "node-a",
-              handler: jest.fn(),
+              handler: vi.fn(),
             }}
           />
           <HookUser
             action={{
               name: "agent-ok",
               nodeName: "node-b",
-              handler: jest.fn(),
+              handler: vi.fn(),
             }}
           />
         </>
@@ -317,7 +318,7 @@ describe("useCoAgentStateRender (hook behaviors)", () => {
       () =>
         useHarness({
           name: "missing-agent",
-          handler: jest.fn(),
+          handler: vi.fn(),
         }),
       { wrapper },
     );
@@ -343,7 +344,7 @@ describe("useCoAgentStateRender (hook behaviors)", () => {
       () =>
         useHarness({
           name: "agent-present",
-          handler: jest.fn(),
+          handler: vi.fn(),
         }),
       { wrapper },
     );

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

@@ -0,0 +1,72 @@
+import { vi } from "vitest";
+import React from "react";
+import { render, waitFor } from "@testing-library/react";
+import { useFrontendTool } from "../use-frontend-tool";
+
+// Track what gets passed to the v2 hook
+let lastV2ToolCall: any = null;
+
+vi.mock("@copilotkitnext/react", () => ({
+  useFrontendTool: vi.fn((tool: any) => {
+    lastV2ToolCall = tool;
+  }),
+}));
+
+describe("useFrontendTool available passthrough (v1 → v2)", () => {
+  beforeEach(() => {
+    lastV2ToolCall = null;
+  });
+
+  it("converts available: 'disabled' to false for v2", () => {
+    const Component: React.FC = () => {
+      useFrontendTool({
+        name: "testTool",
+        description: "Test tool",
+        available: "disabled",
+        parameters: [],
+        handler: async () => "done",
+      });
+      return null;
+    };
+
+    render(<Component />);
+
+    expect(lastV2ToolCall).toBeDefined();
+    expect(lastV2ToolCall.available).toBe(false);
+  });
+
+  it("converts available: 'enabled' to true for v2", () => {
+    const Component: React.FC = () => {
+      useFrontendTool({
+        name: "testTool",
+        description: "Test tool",
+        available: "enabled",
+        parameters: [],
+        handler: async () => "done",
+      });
+      return null;
+    };
+
+    render(<Component />);
+
+    expect(lastV2ToolCall).toBeDefined();
+    expect(lastV2ToolCall.available).toBe(true);
+  });
+
+  it("does not include available when not specified", () => {
+    const Component: React.FC = () => {
+      useFrontendTool({
+        name: "testTool",
+        description: "Test tool",
+        parameters: [],
+        handler: async () => "done",
+      });
+      return null;
+    };
+
+    render(<Component />);
+
+    expect(lastV2ToolCall).toBeDefined();
+    expect(lastV2ToolCall.available).toBeUndefined();
+  });
+});

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

@@ -1,14 +1,16 @@
+import { vi } from "vitest";
 import React, { useEffect } from "react";
 import { render, waitFor } from "@testing-library/react";
 import { ToolCallStatus } from "@copilotkitnext/core";
 import { useFrontendTool } from "../use-frontend-tool";
+import * as copilotkitNextReact from "@copilotkitnext/react";
 
-jest.mock("@copilotkitnext/react", () => {
+vi.mock("@copilotkitnext/react", () => {
   let currentRender: any = null;
   const listeners = new Set<() => void>();
 
   return {
-    useFrontendTool: jest.fn((tool: { render?: any }) => {
+    useFrontendTool: vi.fn((tool: { render?: any }) => {
       React.useEffect(() => {
         currentRender = tool.render ?? null;
         listeners.forEach((listener) => listener());
@@ -22,7 +24,7 @@ jest.mock("@copilotkitnext/react", () => {
   };
 });
 
-const toolRenderModule = jest.requireMock("@copilotkitnext/react") as {
+const toolRenderModule = copilotkitNextReact as unknown as {
   __getCurrentRender: () => any;
   __subscribeRender: (listener: () => void) => () => void;
 };
@@ -53,8 +55,8 @@ function RunActionButton({
   onMount,
   onUnmount,
 }: {
-  onMount: jest.Mock;
-  onUnmount: jest.Mock;
+  onMount: () => void;
+  onUnmount: () => void;
 }) {
   useEffect(() => {
     onMount();
@@ -66,8 +68,8 @@ function RunActionButton({
 
 describe("useFrontendTool dependency changes", () => {
   it("should not remount rendered tool UI when deps change", async () => {
-    const mounted = jest.fn();
-    const unmounted = jest.fn();
+    const mounted = vi.fn();
+    const unmounted = vi.fn();
 
     const ToolUser = ({ version }: { version: number }) => {
       useFrontendTool(