@copilotkit/runtime 1.51.5-next.0 → 1.51.5-next.1

package/src/lib/runtime/__tests__/copilot-runtime-error.test.ts

@@ -122,7 +122,7 @@ describe("CopilotRuntime onError types", () => {
     };
 
     it("should return true when both onError and publicApiKey are provided", () => {
-      const onError = jest.fn();
+      const onError = vi.fn();
       const result = shouldHandleError(onError, "valid-api-key");
       expect(result).toBe(true);
     });
@@ -133,13 +133,13 @@ describe("CopilotRuntime onError types", () => {
     });
 
     it("should return false when publicApiKey is missing", () => {
-      const onError = jest.fn();
+      const onError = vi.fn();
       const result = shouldHandleError(onError, undefined);
       expect(result).toBe(false);
     });
 
     it("should return false when publicApiKey is empty string", () => {
-      const onError = jest.fn();
+      const onError = vi.fn();
       const result = shouldHandleError(onError, "");
       expect(result).toBe(false);
     });
@@ -175,7 +175,7 @@ describe("CopilotRuntime onError types", () => {
       expect(nonCloudKey).toBe("test-key-123");
 
       // Both should enable error handling when onError is present
-      const onError = jest.fn();
+      const onError = vi.fn();
       expect(shouldHandleError(onError, cloudKey)).toBe(true);
       expect(shouldHandleError(onError, nonCloudKey)).toBe(true);
     });

package/src/lib/runtime/__tests__/mcp-tools-utils.test.ts

@@ -394,7 +394,7 @@ describe("MCP Tools Utils", () => {
     });
 
     it("should handle tool execution correctly", async () => {
-      const mockExecute = jest.fn().mockResolvedValue("mock result");
+      const mockExecute = vi.fn().mockResolvedValue("mock result");
       const mcpTools: Record<string, MCPTool> = {
         testTool: {
           description: "A test tool",

package/src/service-adapters/google/google-genai-adapter.test.ts

@@ -1,22 +1,65 @@
-import { GoogleGenerativeAIAdapter } from "./google-genai-adapter";
 import {
   AIMessage,
   HumanMessage,
   SystemMessage,
 } from "@langchain/core/messages";
+import * as langchainMessages from "@langchain/core/messages";
+
+// Create mock ChatGoogle that captures the filtered messages passed to stream()
+const mockStream = vi.fn();
+const mockBindTools = vi.fn();
+const MockChatGoogle = vi.fn();
+
+// The adapter's chainFn uses lazy require() calls for @langchain/google-gauth
+// and @langchain/core/messages. Vitest's vi.mock cannot intercept CJS require()
+// calls in all environments. Instead, inject mocks directly into Node's require
+// cache so that:
+// 1. require("@langchain/google-gauth") returns our mock ChatGoogle
+// 2. require("@langchain/core/messages") returns the same module instance as
+//    the ESM import above, ensuring instanceof checks work correctly
+beforeAll(() => {
+  mockStream.mockImplementation((messages: any) =>
+    Promise.resolve({ filteredMessages: messages }),
+  );
+  mockBindTools.mockReturnValue({ stream: mockStream });
+  MockChatGoogle.mockImplementation(() => ({ bindTools: mockBindTools }));
+
+  const googleAuthPath = require.resolve("@langchain/google-gauth");
+  require.cache[googleAuthPath] = {
+    id: googleAuthPath,
+    filename: googleAuthPath,
+    loaded: true,
+    exports: { ChatGoogle: MockChatGoogle },
+  } as any;
+
+  const messagesPath = require.resolve("@langchain/core/messages");
+  require.cache[messagesPath] = {
+    id: messagesPath,
+    filename: messagesPath,
+    loaded: true,
+    exports: langchainMessages,
+  } as any;
+});
 
-// Mock ChatGoogle to capture what messages are passed to it
-jest.mock("@langchain/google-gauth", () => ({
-  ChatGoogle: jest.fn().mockImplementation(() => ({
-    bindTools: jest.fn().mockReturnThis(),
-    stream: jest.fn().mockImplementation((messages) => {
-      // Return the messages so we can verify filtering
-      return Promise.resolve({ filteredMessages: messages });
-    }),
-  })),
-}));
+afterAll(() => {
+  const googleAuthPath = require.resolve("@langchain/google-gauth");
+  delete require.cache[googleAuthPath];
+  const messagesPath = require.resolve("@langchain/core/messages");
+  delete require.cache[messagesPath];
+});
+
+import { GoogleGenerativeAIAdapter } from "./google-genai-adapter";
 
 describe("GoogleGenerativeAIAdapter", () => {
+  beforeEach(() => {
+    vi.clearAllMocks();
+    mockStream.mockImplementation((messages: any) =>
+      Promise.resolve({ filteredMessages: messages }),
+    );
+    mockBindTools.mockReturnValue({ stream: mockStream });
+    MockChatGoogle.mockImplementation(() => ({ bindTools: mockBindTools }));
+  });
+
   it("should filter out empty AIMessages to prevent Gemini validation errors", async () => {
     const adapter = new GoogleGenerativeAIAdapter();
 

package/src/v2/index.ts

@@ -1,3 +1,2 @@
-import "reflect-metadata";
 export * from "@copilotkitnext/runtime";
 export * from "@copilotkitnext/agent";

package/tests/global.d.ts

@@ -1,13 +1 @@
-import "@jest/globals";
-
-declare global {
-  const jest: (typeof import("@jest/globals"))["jest"];
-  const expect: (typeof import("@jest/globals"))["expect"];
-  const test: (typeof import("@jest/globals"))["test"];
-  const describe: (typeof import("@jest/globals"))["describe"];
-  const beforeEach: (typeof import("@jest/globals"))["beforeEach"];
-  const afterEach: (typeof import("@jest/globals"))["afterEach"];
-  const beforeAll: (typeof import("@jest/globals"))["beforeAll"];
-  const afterAll: (typeof import("@jest/globals"))["afterAll"];
-  const it: (typeof import("@jest/globals"))["it"];
-}
+/// <reference types="vitest/globals" />

package/tests/service-adapters/anthropic/allowlist-approach.test.ts

@@ -1,9 +1,3 @@
-/**
- * @jest-environment node
- */
-
-import { describe, it, expect } from "@jest/globals";
-
 describe("Anthropic Adapter - Allowlist Approach", () => {
   it("should filter out tool_result messages with no corresponding tool_use ID", () => {
     // Setup test data

package/tests/service-adapters/anthropic/anthropic-adapter.test.ts

@@ -1,22 +1,23 @@
-/**
- * @jest-environment node
- */
-
 import { AnthropicAdapter } from "../../../src/service-adapters/anthropic/anthropic-adapter";
+import {
+  TextMessage,
+  ActionExecutionMessage,
+  ResultMessage,
+} from "../../../src/graphql/types/converted";
 
 // Mock only the Anthropic SDK, not our adapter
-jest.mock("@anthropic-ai/sdk", () => {
+vi.mock("@anthropic-ai/sdk", () => {
   return {
-    default: jest.fn().mockImplementation(() => ({
+    default: vi.fn().mockImplementation(() => ({
       messages: {
-        create: jest.fn(),
+        create: vi.fn(),
       },
     })),
   };
 });
 
 // Mock the message classes
-jest.mock("../../../src/graphql/types/converted", () => {
+vi.mock("../../../src/graphql/types/converted", () => {
   class MockTextMessage {
     content: string;
     role: string;
@@ -102,15 +103,15 @@ jest.mock("../../../src/graphql/types/converted", () => {
 describe("AnthropicAdapter", () => {
   let adapter: AnthropicAdapter;
   let mockEventSource: any;
-  let mockAnthropicCreate: jest.Mock;
+  let mockAnthropicCreate: any;
 
   beforeEach(() => {
-    jest.clearAllMocks();
+    vi.clearAllMocks();
 
     // Create a mock Anthropic instance
     const mockAnthropic = {
       messages: {
-        create: jest.fn(),
+        create: vi.fn(),
       },
     };
 
@@ -121,15 +122,15 @@ describe("AnthropicAdapter", () => {
     mockAnthropicCreate = mockAnthropic.messages.create;
 
     mockEventSource = {
-      stream: jest.fn((callback) => {
+      stream: vi.fn((callback) => {
         const mockStream = {
-          sendTextMessageStart: jest.fn(),
-          sendTextMessageContent: jest.fn(),
-          sendTextMessageEnd: jest.fn(),
-          sendActionExecutionStart: jest.fn(),
-          sendActionExecutionArgs: jest.fn(),
-          sendActionExecutionEnd: jest.fn(),
-          complete: jest.fn(),
+          sendTextMessageStart: vi.fn(),
+          sendTextMessageContent: vi.fn(),
+          sendTextMessageEnd: vi.fn(),
+          sendActionExecutionStart: vi.fn(),
+          sendActionExecutionArgs: vi.fn(),
+          sendActionExecutionEnd: vi.fn(),
+          complete: vi.fn(),
         };
         callback(mockStream);
         return Promise.resolve();
@@ -139,12 +140,6 @@ describe("AnthropicAdapter", () => {
 
   describe("Deduplication Logic", () => {
     it("should filter out duplicate result messages", async () => {
-      const {
-        TextMessage,
-        ActionExecutionMessage,
-        ResultMessage,
-      } = require("../../../src/graphql/types/converted");
-
       const systemMessage = new TextMessage("system", "System message");
       const userMessage = new TextMessage("user", "Set theme to orange");
 
@@ -238,12 +233,6 @@ describe("AnthropicAdapter", () => {
     });
 
     it("should filter out invalid result messages without corresponding tool_use", async () => {
-      const {
-        TextMessage,
-        ActionExecutionMessage,
-        ResultMessage,
-      } = require("../../../src/graphql/types/converted");
-
       const systemMessage = new TextMessage("system", "System message");
 
       // Valid tool execution
@@ -290,12 +279,6 @@ describe("AnthropicAdapter", () => {
 
   describe("Fallback Response Logic", () => {
     it("should generate contextual fallback when Anthropic returns no content", async () => {
-      const {
-        TextMessage,
-        ActionExecutionMessage,
-        ResultMessage,
-      } = require("../../../src/graphql/types/converted");
-
       const systemMessage = new TextMessage("system", "System message");
       const userMessage = new TextMessage("user", "Set theme to orange");
 
@@ -318,10 +301,10 @@ describe("AnthropicAdapter", () => {
       });
 
       const mockStream = {
-        sendTextMessageStart: jest.fn(),
-        sendTextMessageContent: jest.fn(),
-        sendTextMessageEnd: jest.fn(),
-        complete: jest.fn(),
+        sendTextMessageStart: vi.fn(),
+        sendTextMessageContent: vi.fn(),
+        sendTextMessageEnd: vi.fn(),
+        complete: vi.fn(),
       };
 
       mockEventSource.stream.mockImplementation((callback) => {
@@ -347,12 +330,6 @@ describe("AnthropicAdapter", () => {
     });
 
     it("should use generic fallback when no tool result content available", async () => {
-      const {
-        TextMessage,
-        ActionExecutionMessage,
-        ResultMessage,
-      } = require("../../../src/graphql/types/converted");
-
       const systemMessage = new TextMessage("system", "System message");
 
       const toolExecution = new ActionExecutionMessage({
@@ -371,10 +348,10 @@ describe("AnthropicAdapter", () => {
       });
 
       const mockStream = {
-        sendTextMessageStart: jest.fn(),
-        sendTextMessageContent: jest.fn(),
-        sendTextMessageEnd: jest.fn(),
-        complete: jest.fn(),
+        sendTextMessageStart: vi.fn(),
+        sendTextMessageContent: vi.fn(),
+        sendTextMessageEnd: vi.fn(),
+        complete: vi.fn(),
       };
 
       mockEventSource.stream.mockImplementation((callback) => {

package/tests/service-adapters/openai/allowlist-approach.test.ts

@@ -1,9 +1,3 @@
-/**
- * @jest-environment node
- */
-
-import { describe, it, expect } from "@jest/globals";
-
 describe("OpenAI Adapter - Allowlist Approach", () => {
   it("should filter out tool_result messages with no corresponding tool_call ID", () => {
     // Setup test data

package/tests/service-adapters/openai/openai-adapter.test.ts

@@ -1,11 +1,11 @@
 // Mock the modules first
-jest.mock("openai", () => {
+vi.mock("openai", () => {
   function MockOpenAI() {}
   return { default: MockOpenAI };
 });
 
 // Mock the OpenAIAdapter class to avoid the "new OpenAI()" issue
-jest.mock("../../../src/service-adapters/openai/openai-adapter", () => {
+vi.mock("../../../src/service-adapters/openai/openai-adapter", () => {
   class MockOpenAIAdapter {
     _openai: any;
     model: string = "gpt-4o";
@@ -17,7 +17,7 @@ jest.mock("../../../src/service-adapters/openai/openai-adapter", () => {
         beta: {
           chat: {
             completions: {
-              stream: jest.fn(),
+              stream: vi.fn(),
             },
           },
         },
@@ -42,11 +42,8 @@ jest.mock("../../../src/service-adapters/openai/openai-adapter", () => {
   return { OpenAIAdapter: MockOpenAIAdapter };
 });
 
-// Now import the modules
-import { OpenAIAdapter } from "../../../src/service-adapters/openai/openai-adapter";
-
 // Mock the Message classes since they use TypeGraphQL decorators
-jest.mock("../../../src/graphql/types/converted", () => {
+vi.mock("../../../src/graphql/types/converted", () => {
   // Create minimal implementations of the message classes
   class MockTextMessage {
     content: string;
@@ -130,23 +127,31 @@ jest.mock("../../../src/graphql/types/converted", () => {
   };
 });
 
+// Now import the modules (vi.mock is hoisted above imports, so these get the mocked versions)
+import { OpenAIAdapter } from "../../../src/service-adapters/openai/openai-adapter";
+import {
+  TextMessage,
+  ActionExecutionMessage,
+  ResultMessage,
+} from "../../../src/graphql/types/converted";
+
 describe("OpenAIAdapter", () => {
   let adapter: OpenAIAdapter;
   let mockEventSource: any;
 
   beforeEach(() => {
-    jest.clearAllMocks();
+    vi.clearAllMocks();
     adapter = new OpenAIAdapter();
     mockEventSource = {
-      stream: jest.fn((callback) => {
+      stream: vi.fn((callback) => {
         const mockStream = {
-          sendTextMessageStart: jest.fn(),
-          sendTextMessageContent: jest.fn(),
-          sendTextMessageEnd: jest.fn(),
-          sendActionExecutionStart: jest.fn(),
-          sendActionExecutionArgs: jest.fn(),
-          sendActionExecutionEnd: jest.fn(),
-          complete: jest.fn(),
+          sendTextMessageStart: vi.fn(),
+          sendTextMessageContent: vi.fn(),
+          sendTextMessageEnd: vi.fn(),
+          sendActionExecutionStart: vi.fn(),
+          sendActionExecutionArgs: vi.fn(),
+          sendActionExecutionEnd: vi.fn(),
+          complete: vi.fn(),
         };
         callback(mockStream);
       }),
@@ -155,13 +160,6 @@ describe("OpenAIAdapter", () => {
 
   describe("Tool ID handling", () => {
     it("should filter out tool_result messages that don't have corresponding tool_call IDs", async () => {
-      // Import dynamically after mocking
-      const {
-        TextMessage,
-        ActionExecutionMessage,
-        ResultMessage,
-      } = require("../../../src/graphql/types/converted");
-
       // Create messages including one valid pair and one invalid tool_result
       const systemMessage = new TextMessage("system", "System message");
       const userMessage = new TextMessage("user", "User message");
@@ -186,7 +184,7 @@ describe("OpenAIAdapter", () => {
       });
 
       // Spy on the process method to test it's called properly
-      const processSpy = jest.spyOn(adapter, "process");
+      const processSpy = vi.spyOn(adapter, "process");
 
       await adapter.process({
         threadId: "test-thread",
@@ -211,13 +209,6 @@ describe("OpenAIAdapter", () => {
     });
 
     it("should handle duplicate tool IDs by only using each once", async () => {
-      // Import dynamically after mocking
-      const {
-        TextMessage,
-        ActionExecutionMessage,
-        ResultMessage,
-      } = require("../../../src/graphql/types/converted");
-
       // Create messages including duplicate tool results for the same ID
       const systemMessage = new TextMessage("system", "System message");
 
@@ -240,7 +231,7 @@ describe("OpenAIAdapter", () => {
       });
 
       // Spy on the process method to test it's called properly
-      const processSpy = jest.spyOn(adapter, "process");
+      const processSpy = vi.spyOn(adapter, "process");
 
       await adapter.process({
         threadId: "test-thread",
@@ -264,9 +255,6 @@ describe("OpenAIAdapter", () => {
     });
 
     it("should call the stream method on eventSource", async () => {
-      // Import dynamically after mocking
-      const { TextMessage } = require("../../../src/graphql/types/converted");
-
       // Create messages
       const systemMessage = new TextMessage("system", "System message");
       const userMessage = new TextMessage("user", "User message");
@@ -285,9 +273,6 @@ describe("OpenAIAdapter", () => {
     });
 
     it("should return the provided threadId", async () => {
-      // Import dynamically after mocking
-      const { TextMessage } = require("../../../src/graphql/types/converted");
-
       // Create a message
       const systemMessage = new TextMessage("system", "System message");
 

package/tests/setup.vitest.ts

@@ -0,0 +1,8 @@
+// Import reflect-metadata to support TypeGraphQL
+import "reflect-metadata";
+
+import { vi } from "vitest";
+
+// Suppress console output during tests
+vi.spyOn(console, "log").mockImplementation(() => {});
+vi.spyOn(console, "error").mockImplementation(() => {});

package/tests/tsconfig.json

@@ -1,7 +1,7 @@
 {
   "extends": "../tsconfig.json",
   "compilerOptions": {
-    "types": ["node", "jest"],
+    "types": ["node", "vitest/globals"],
     "esModuleInterop": true,
     "noEmit": true,
     "allowJs": true

package/tsconfig.json

@@ -1,7 +1,7 @@
 {
   "extends": "../tsconfig/base.json",
   "compilerOptions": {
-    "types": ["node", "jest"],
+    "types": ["node", "vitest/globals"],
     "lib": ["es2017", "dom"],
     "emitDecoratorMetadata": true,
     "experimentalDecorators": true,

package/tsdown.config.ts

@@ -0,0 +1,37 @@
+import { defineConfig } from "tsdown";
+
+export default defineConfig({
+  entry: ["src/index.ts", "src/v2/index.ts", "src/langgraph.ts"],
+  format: ["esm", "cjs"],
+  dts: true,
+  sourcemap: true,
+  target: "es2022",
+  outDir: "dist",
+  unbundle: true,
+  banner: ({ format, fileName }) => {
+    // tsdown/rolldown reorders bare side-effect imports to the end of the entry chunk,
+    // breaking type-graphql which needs reflect-metadata at load time.
+    // The _virtual/_rolldown/runtime banner propagates to all output files per format,
+    // ensuring reflect-metadata is always the first thing that runs.
+    if (fileName.includes("_virtual/_rolldown/runtime")) {
+      return format === "cjs"
+        ? 'require("reflect-metadata");'
+        : 'import "reflect-metadata";';
+    }
+  },
+  external: [
+    "@ag-ui/langgraph",
+    "@langchain/core",
+    "@langchain/openai",
+    "@langchain/aws",
+    "@langchain/community",
+    "@langchain/google-gauth",
+    "@langchain/langgraph-sdk",
+    "langchain",
+    "@anthropic-ai/sdk",
+    "groq-sdk",
+    "@whatwg-node/fetch",
+    "@whatwg-node/server",
+  ],
+  exports: true,
+});

package/vitest.config.mjs

@@ -0,0 +1,16 @@
+import { defineConfig } from "vitest/config";
+
+export default defineConfig({
+  test: {
+    environment: "node",
+    globals: true,
+    include: [
+      "src/**/*.{test,spec}.ts",
+      "tests/**/*.{test,spec}.ts",
+    ],
+    exclude: ["**/dist/**"],
+    setupFiles: ["./tests/setup.vitest.ts"],
+    reporters: [["default", { summary: false }]],
+    silent: true,
+  },
+});