@copilotkit/runtime 1.50.0-beta.1 → 1.50.0-beta.2

package/package.json

@@ -9,7 +9,7 @@
   "publishConfig": {
     "access": "public"
   },
-  "version": "1.50.0-beta.1",
+  "version": "1.50.0-beta.2",
   "sideEffects": false,
   "main": "./dist/index.js",
   "module": "./dist/index.mjs",
@@ -47,8 +47,8 @@
   },
   "dependencies": {
     "@anthropic-ai/sdk": "^0.57.0",
-    "@copilotkitnext/agent": "0.0.22",
-    "@copilotkitnext/runtime": "0.0.22",
+    "@copilotkitnext/agent": "0.0.23",
+    "@copilotkitnext/runtime": "0.0.23",
     "@graphql-yoga/plugin-defer-stream": "^3.3.1",
     "@langchain/aws": "^0.1.9",
     "@langchain/community": "^0.3.29",
@@ -74,7 +74,7 @@
     "rxjs": "7.8.1",
     "type-graphql": "2.0.0-rc.1",
     "zod": "^3.23.3",
-    "@copilotkit/shared": "1.50.0-beta.1"
+    "@copilotkit/shared": "1.50.0-beta.2"
   },
   "peerDependencies": {
     "@ag-ui/client": ">=0.0.39",
@@ -83,6 +83,11 @@
     "@ag-ui/langgraph": ">=0.0.18",
     "@ag-ui/proto": ">=0.0.39"
   },
+  "peerDependenciesMeta": {
+    "@ag-ui/langgraph": {
+      "optional": true
+    }
+  },
   "keywords": [
     "copilotkit",
     "copilot",

package/src/graphql/message-conversion/agui-to-gql.test.ts

@@ -1,4 +1,4 @@
-import { describe, test, expect, vi } from "vitest";
+import { describe, test, expect } from "@jest/globals";
 import * as gql from "../types/converted/index";
 import agui from "@copilotkit/shared";
 import {
@@ -587,7 +587,7 @@ describe("agui-to-gql", () => {
     });
 
     test("should verify render function receives correct props including name", () => {
-      const mockRender = vi.fn(
+      const mockRender = jest.fn(
         (props) => `Rendered: ${props.name} with args: ${JSON.stringify(props.args)}`,
       );
       const aguiMessage: agui.Message = {

package/src/graphql/message-conversion/gql-to-agui.test.ts

@@ -1,4 +1,4 @@
-import { describe, test, expect, vi } from "vitest";
+import { describe, test, expect } from "@jest/globals";
 import * as gql from "../types/converted/index";
 import { MessageStatusCode } from "../types/message-status.type";
 import {
@@ -370,7 +370,7 @@ describe("message-conversion", () => {
         status: { code: MessageStatusCode.Pending },
       });
 
-      const mockRender = vi.fn();
+      const mockRender = jest.fn();
       const mockRenderAndWaitForResponse = (props: any) => "Test Render With Response";
 
       const actions = {
@@ -414,7 +414,7 @@ describe("message-conversion", () => {
         status: { code: MessageStatusCode.Pending },
       });
 
-      const mockRender = vi.fn();
+      const mockRender = jest.fn();
       const actions = {
         testAction: {
           name: "testAction",
@@ -445,7 +445,7 @@ describe("message-conversion", () => {
         status: { code: MessageStatusCode.Success },
       });
 
-      const mockRender = vi.fn();
+      const mockRender = jest.fn();
       const actions = {
         testAction: {
           name: "testAction",
@@ -483,7 +483,7 @@ describe("message-conversion", () => {
         actionName: "testAction",
       });
 
-      const mockRender = vi.fn();
+      const mockRender = jest.fn();
       const actions = {
         testAction: {
           name: "testAction",
@@ -517,7 +517,7 @@ describe("message-conversion", () => {
         status: { code: MessageStatusCode.Pending },
       });
 
-      const mockRender = vi.fn();
+      const mockRender = jest.fn();
       const actions = {
         testAction: {
           name: "testAction",
@@ -627,7 +627,7 @@ describe("message-conversion", () => {
         role: gql.Role.assistant,
       });
 
-      const mockRender = vi.fn();
+      const mockRender = jest.fn();
       const coAgentStateRenders = {
         testAgent: {
           name: "testAgent",
@@ -740,7 +740,7 @@ describe("message-conversion", () => {
         role: gql.Role.assistant,
       });
 
-      const mockRender = vi.fn();
+      const mockRender = jest.fn();
       const coAgentStateRenders = {
         testAgent: {
           name: "testAgent",
@@ -853,7 +853,7 @@ describe("message-conversion", () => {
       const actions = {
         testAction: {
           name: "testAction",
-          render: vi.fn((props) => `Rendered: ${props.args.test}`),
+          render: jest.fn((props) => `Rendered: ${props.args.test}`),
         },
       };
 
@@ -889,7 +889,7 @@ describe("message-conversion", () => {
       const actions = {
         "*": {
           name: "*",
-          render: vi.fn((props) => `Wildcard rendered: ${props.args.test}`),
+          render: jest.fn((props) => `Wildcard rendered: ${props.args.test}`),
         },
       };
 
@@ -922,7 +922,7 @@ describe("message-conversion", () => {
     });
 
     test("should pass tool name to wildcard action render function", () => {
-      const mockRender = vi.fn(
+      const mockRender = jest.fn(
         (props) => `Wildcard rendered: ${props.name} with args: ${JSON.stringify(props.args)}`,
       );
       const actions = {
@@ -954,7 +954,9 @@ describe("message-conversion", () => {
     });
 
     test("should pass tool name to regular action render function", () => {
-      const mockRender = vi.fn((props) => `Regular action rendered: ${JSON.stringify(props.args)}`);
+      const mockRender = jest.fn(
+        (props) => `Regular action rendered: ${JSON.stringify(props.args)}`,
+      );
       const actions = {
         testAction: {
           name: "testAction",
@@ -992,11 +994,11 @@ describe("message-conversion", () => {
       const actions = {
         specificAction: {
           name: "specificAction",
-          render: vi.fn((props) => "Specific action rendered"),
+          render: jest.fn((props) => "Specific action rendered"),
         },
         "*": {
           name: "*",
-          render: vi.fn((props) => "Wildcard action rendered"),
+          render: jest.fn((props) => "Wildcard action rendered"),
         },
       };
 
@@ -1032,7 +1034,7 @@ describe("message-conversion", () => {
       const actions = {
         otherAction: {
           name: "otherAction",
-          render: vi.fn(),
+          render: jest.fn(),
         },
       };
 
@@ -1095,7 +1097,7 @@ describe("message-conversion", () => {
       const actions = {
         "*": {
           name: "*",
-          render: vi.fn((props) => `Result: ${props.result}`),
+          render: jest.fn((props) => `Result: ${props.result}`),
         },
       };
 
@@ -1133,7 +1135,7 @@ describe("message-conversion", () => {
       const actions = {
         "*": {
           name: "*",
-          render: vi.fn((props) => `Status: ${props.status}`),
+          render: jest.fn((props) => `Status: ${props.status}`),
         },
       };
 
@@ -1170,7 +1172,7 @@ describe("message-conversion", () => {
       const actions = {
         "*": {
           name: "*",
-          render: vi.fn((props) => `Status: ${props.status}`),
+          render: jest.fn((props) => `Status: ${props.status}`),
         },
       };
 
@@ -1207,7 +1209,7 @@ describe("message-conversion", () => {
       const actions = {
         "*": {
           name: "*",
-          render: vi.fn((props) => `Status: ${props.status}`),
+          render: jest.fn((props) => `Status: ${props.status}`),
         },
       };
 
@@ -1244,7 +1246,7 @@ describe("message-conversion", () => {
       const actions = {
         "*": {
           name: "*",
-          render: vi.fn((props) => `Status: ${props.status}`),
+          render: jest.fn((props) => `Status: ${props.status}`),
         },
       };
 
@@ -1281,7 +1283,7 @@ describe("message-conversion", () => {
       const actions = {
         "*": {
           name: "*",
-          render: vi.fn((props) => `Args: ${JSON.stringify(props.args)}`),
+          render: jest.fn((props) => `Args: ${JSON.stringify(props.args)}`),
         },
       };
 
@@ -1317,7 +1319,7 @@ describe("message-conversion", () => {
       const actions = {
         "*": {
           name: "*",
-          render: vi.fn((props) => `Args: ${JSON.stringify(props.args)}`),
+          render: jest.fn((props) => `Args: ${JSON.stringify(props.args)}`),
         },
       };
 
@@ -1353,7 +1355,7 @@ describe("message-conversion", () => {
       const actions = {
         "*": {
           name: "*",
-          render: vi.fn((props) => `Complex: ${JSON.stringify(props.args)}`),
+          render: jest.fn((props) => `Complex: ${JSON.stringify(props.args)}`),
         },
       };
 
@@ -1401,11 +1403,11 @@ describe("message-conversion", () => {
       const actions = {
         wildcard1: {
           name: "*",
-          render: vi.fn((props) => "First wildcard"),
+          render: jest.fn((props) => "First wildcard"),
         },
         wildcard2: {
           name: "*",
-          render: vi.fn((props) => "Second wildcard"),
+          render: jest.fn((props) => "Second wildcard"),
         },
       };
 
@@ -1441,7 +1443,7 @@ describe("message-conversion", () => {
       const actions = {
         "*": {
           name: "*",
-          render: vi.fn((props) => `Result: ${JSON.stringify(props.result)}`),
+          render: jest.fn((props) => `Result: ${JSON.stringify(props.result)}`),
         },
       };
 
@@ -1478,7 +1480,7 @@ describe("message-conversion", () => {
       const actions = {
         "*": {
           name: "*",
-          render: vi.fn((props) => `Result: ${JSON.stringify(props.result)}`),
+          render: jest.fn((props) => `Result: ${JSON.stringify(props.result)}`),
         },
       };
 
@@ -1515,7 +1517,7 @@ describe("message-conversion", () => {
       const actions = {
         "*": {
           name: "*",
-          render: vi.fn((props) => `Result: ${JSON.stringify(props.result)}`),
+          render: jest.fn((props) => `Result: ${JSON.stringify(props.result)}`),
         },
       };
 

package/src/graphql/message-conversion/roundtrip-conversion.test.ts

@@ -1,4 +1,4 @@
-import { describe, test, expect, vi } from "vitest";
+import { describe, test, expect } from "@jest/globals";
 import * as gql from "../types/converted/index";
 import agui from "@copilotkit/shared";
 import { aguiToGQL } from "./agui-to-gql";
@@ -131,7 +131,7 @@ describe("roundtrip message conversion", () => {
   });
 
   test("action execution with render function roundtrip", () => {
-    const mockRender = vi.fn();
+    const mockRender = jest.fn();
     const aguiMsg: agui.Message = {
       id: "assistant-1",
       role: "assistant",
@@ -211,7 +211,7 @@ describe("roundtrip message conversion", () => {
   });
 
   test("wild card action roundtrip conversion", () => {
-    const mockRender = vi.fn((props) => `Wildcard rendered: ${props.args.test}`);
+    const mockRender = jest.fn((props) => `Wildcard rendered: ${props.args.test}`);
     const aguiMsg: agui.Message = {
       id: "assistant-wildcard-1",
       role: "assistant",
@@ -256,7 +256,7 @@ describe("roundtrip message conversion", () => {
   });
 
   test("wild card action with specific action priority roundtrip", () => {
-    const mockRender = vi.fn((props) => `Specific action rendered: ${props.args.test}`);
+    const mockRender = jest.fn((props) => `Specific action rendered: ${props.args.test}`);
     const aguiMsg: agui.Message = {
       id: "assistant-priority-1",
       role: "assistant",
@@ -313,7 +313,7 @@ describe("roundtrip message conversion", () => {
     const actions: Record<string, any> = {
       "*": {
         name: "*",
-        render: vi.fn((props) => `GQL wildcard rendered: ${props.args.test}`),
+        render: jest.fn((props) => `GQL wildcard rendered: ${props.args.test}`),
       },
     };
 
@@ -379,7 +379,7 @@ describe("roundtrip message conversion", () => {
   });
 
   test("roundtrip conversion with action execution and result parsing", () => {
-    const mockRender = vi.fn((props) => `Rendered: ${JSON.stringify(props.result)}`);
+    const mockRender = jest.fn((props) => `Rendered: ${JSON.stringify(props.result)}`);
 
     // Create action execution message
     const actionExecMsg = new gql.ActionExecutionMessage({
@@ -439,7 +439,7 @@ describe("roundtrip message conversion", () => {
   });
 
   test("roundtrip conversion verifies correct property distribution for regular actions", () => {
-    const mockRender = vi.fn((props) => `Regular action: ${JSON.stringify(props.args)}`);
+    const mockRender = jest.fn((props) => `Regular action: ${JSON.stringify(props.args)}`);
 
     const actionExecMsg = new gql.ActionExecutionMessage({
       id: "regular-action-test",
@@ -481,7 +481,7 @@ describe("roundtrip message conversion", () => {
   });
 
   test("roundtrip conversion verifies correct property distribution for wildcard actions", () => {
-    const mockRender = vi.fn(
+    const mockRender = jest.fn(
       (props) => `Wildcard action: ${props.name} with ${JSON.stringify(props.args)}`,
     );
 

package/src/lib/integrations/nextjs/app-router.ts

@@ -24,7 +24,9 @@ export function copilotRuntimeNextJSAppRouterEndpoint(options: CreateCopilotRunt
   logger.debug("Creating NextJS App Router endpoint");
 
   const serviceAdapter = options.serviceAdapter;
-  options.runtime.handleServiceAdapter(serviceAdapter);
+  if (serviceAdapter) {
+    options.runtime.handleServiceAdapter(serviceAdapter);
+  }
 
   const copilotRoute = createCopilotEndpointSingleRoute({
     runtime: options.runtime.instance,

package/src/lib/integrations/node-http/index.ts

@@ -55,7 +55,9 @@ export function copilotRuntimeNodeHttpEndpoint(options: CreateCopilotRuntimeServ
   logger.debug("Creating Node HTTP endpoint");
 
   const serviceAdapter = options.serviceAdapter;
-  options.runtime.handleServiceAdapter(serviceAdapter);
+  if (serviceAdapter) {
+    options.runtime.handleServiceAdapter(serviceAdapter);
+  }
 
   const honoApp = createCopilotEndpointSingleRoute({
     runtime: options.runtime.instance,

package/src/lib/integrations/shared.ts

@@ -34,8 +34,8 @@ export type GraphQLContext = YogaInitialContext & {
 };
 
 export interface CreateCopilotRuntimeServerOptions {
-  runtime: CopilotRuntime;
-  serviceAdapter: CopilotServiceAdapter;
+  runtime: CopilotRuntime<any>;
+  serviceAdapter?: CopilotServiceAdapter;
   endpoint: string;
   baseUrl?: string;
   cloud?: CopilotCloudOptions;

package/src/lib/runtime/copilot-runtime.ts

@@ -312,8 +312,8 @@ interface CopilotRuntimeConstructorParams<T extends Parameter[] | [] = []>
 /**
  * Central runtime object passed to all request handlers.
  */
-export class CopilotRuntime {
-  params?: CopilotRuntimeConstructorParams;
+export class CopilotRuntime<const T extends Parameter[] | [] = []> {
+  params?: CopilotRuntimeConstructorParams<T>;
   private observability?: CopilotObservabilityConfig;
   // Cache MCP tools per endpoint to avoid re-fetching repeatedly
   private mcpToolsCache: Map<string, BasicAgentConfiguration["tools"]> = new Map();
@@ -321,7 +321,7 @@ export class CopilotRuntime {
   private _instance: CopilotRuntimeVNext;
 
   constructor(
-    params?: CopilotRuntimeConstructorParams & PartialBy<CopilotRuntimeOptions, "agents">,
+    params?: CopilotRuntimeConstructorParams<T> & PartialBy<CopilotRuntimeOptions, "agents">,
   ) {
     const agents = params?.agents ?? {};
     this.runtimeArgs = {
@@ -345,7 +345,9 @@ export class CopilotRuntime {
     return this._instance;
   }
 
-  private assignEndpointsToAgents(endpoints: CopilotRuntimeConstructorParams["remoteEndpoints"]) {
+  private assignEndpointsToAgents(
+    endpoints: CopilotRuntimeConstructorParams<T>["remoteEndpoints"],
+  ) {
     return endpoints.reduce((acc, endpoint) => {
       if (resolveEndpointType(endpoint) == EndpointType.LangGraphPlatform) {
         let lgAgents = {};
@@ -393,7 +395,7 @@ export class CopilotRuntime {
           });
         }
 
-        if (this.params.actions?.length) {
+        if (this.params.actions) {
           const mcpTools = await this.getToolsFromMCP();
           agentsList = this.assignToolsToAgents(agents, [
             ...this.getToolsFromActions(this.params.actions),
@@ -454,7 +456,7 @@ export class CopilotRuntime {
   }
 
   private createOnBeforeRequestHandler(
-    params?: CopilotRuntimeConstructorParams & PartialBy<CopilotRuntimeOptions, "agents">,
+    params?: CopilotRuntimeConstructorParams<T> & PartialBy<CopilotRuntimeOptions, "agents">,
   ) {
     return async (hookParams: BeforeRequestMiddlewareFnParameters[0]) => {
       // TODO: get public api key and run with expected data
@@ -492,7 +494,7 @@ export class CopilotRuntime {
   }
 
   private createOnAfterRequestHandler(
-    params?: CopilotRuntimeConstructorParams & PartialBy<CopilotRuntimeOptions, "agents">,
+    params?: CopilotRuntimeConstructorParams<T> & PartialBy<CopilotRuntimeOptions, "agents">,
   ) {
     return async (hookParams: AfterRequestMiddlewareFnParameters[0]) => {
       // TODO: get public api key and run with expected data