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

package/package.json

@@ -9,41 +9,40 @@
   "publishConfig": {
     "access": "public"
   },
-  "version": "1.51.5-next.0",
-  "sideEffects": false,
-  "main": "./dist/index.js",
+  "version": "1.51.5-next.1",
+  "sideEffects": [
+    "./dist/index.mjs",
+    "./dist/index.cjs",
+    "./dist/v2/index.mjs",
+    "./dist/v2/index.cjs"
+  ],
+  "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"
+    },
+    "./langgraph": {
+      "import": "./dist/langgraph.mjs",
+      "require": "./dist/langgraph.cjs"
     },
     "./v2": {
       "import": "./dist/v2/index.mjs",
-      "require": "./dist/v2/index.js",
-      "types": "./dist/v2/index.d.ts"
+      "require": "./dist/v2/index.cjs"
     },
-    "./langgraph": {
-      "import": "./dist/langgraph.mjs",
-      "require": "./dist/langgraph.js",
-      "types": "./dist/langgraph.d.ts"
-    }
+    "./package.json": "./package.json"
   },
-  "types": "./dist/index.d.ts",
+  "types": "./dist/index.d.cts",
   "license": "MIT",
   "devDependencies": {
-    "@jest/globals": "^29.7.0",
     "@swc/core": "1.5.28",
-    "@types/jest": "^29.5.12",
     "@types/node": "^18.11.17",
     "@whatwg-node/server": "^0.9.34",
     "eslint": "^8.56.0",
-    "jest": "^29.6.4",
     "nodemon": "^3.1.3",
-    "ts-jest": "^29.1.1",
     "ts-node": "^10.9.2",
-    "tsup": "^6.7.0",
+    "tsdown": "^0.20.3",
     "typescript": "^5.2.3",
     "vitest": "^3.2.4",
     "eslint-config-custom": "1.4.8",
@@ -70,9 +69,9 @@
     "rxjs": "7.8.1",
     "type-graphql": "2.0.0-rc.1",
     "zod": "^3.23.3",
-    "@copilotkit/shared": "1.51.5-next.0",
-    "@copilotkitnext/agent": "1.51.5-next.0",
-    "@copilotkitnext/runtime": "1.51.5-next.0"
+    "@copilotkit/shared": "1.51.5-next.1",
+    "@copilotkitnext/agent": "1.51.5-next.1",
+    "@copilotkitnext/runtime": "1.51.5-next.1"
   },
   "peerDependencies": {
     "@anthropic-ai/sdk": "^0.57.0",
@@ -85,9 +84,9 @@
     "groq-sdk": ">=0.3.0 <1.0.0",
     "langchain": ">=0.3.3",
     "openai": "^4.85.1",
-    "@copilotkit/shared": "1.51.5-next.0",
-    "@copilotkitnext/agent": "1.51.5-next.0",
-    "@copilotkitnext/runtime": "1.51.5-next.0"
+    "@copilotkit/shared": "1.51.5-next.1",
+    "@copilotkitnext/agent": "1.51.5-next.1",
+    "@copilotkitnext/runtime": "1.51.5-next.1"
   },
   "peerDependenciesMeta": {
     "@anthropic-ai/sdk": {
@@ -131,12 +130,15 @@
     "textarea"
   ],
   "scripts": {
-    "build": "pnpm run generate-graphql-schema && tsup",
-    "dev": "tsup --watch --onSuccess \"pnpm run generate-graphql-schema\"",
-    "test": "jest --passWithNoTests",
+    "build": "pnpm run generate-graphql-schema && tsdown",
+    "dev": "tsdown --watch",
+    "test": "vitest run",
+    "test:watch": "vitest",
     "check-types": "tsc --noEmit",
     "generate-graphql-schema": "rm -rf __snapshots__ && ts-node ./scripts/generate-gql-schema.ts",
     "link:global": "pnpm link --global",
-    "unlink:global": "pnpm unlink --global"
+    "unlink:global": "pnpm unlink --global",
+    "publint": "publint .",
+    "attw": "attw --pack . --profile node16"
   }
 }

package/src/graphql/inputs/extensions.input.ts

@@ -5,12 +5,6 @@ import { Field, InputType } from "type-graphql";
  * service adapter or agent framework.
  */
 
-@InputType()
-export class ExtensionsInput {
-  @Field(() => OpenAIApiAssistantAPIInput, { nullable: true })
-  openaiAssistantAPI?: OpenAIApiAssistantAPIInput;
-}
-
 @InputType()
 export class OpenAIApiAssistantAPIInput {
   @Field(() => String, { nullable: true })
@@ -19,3 +13,9 @@ export class OpenAIApiAssistantAPIInput {
   @Field(() => String, { nullable: true })
   threadId?: string;
 }
+
+@InputType()
+export class ExtensionsInput {
+  @Field(() => OpenAIApiAssistantAPIInput, { nullable: true })
+  openaiAssistantAPI?: OpenAIApiAssistantAPIInput;
+}

package/src/graphql/inputs/message.input.ts

@@ -2,26 +2,6 @@ import { Field, InputType } from "type-graphql";
 import { MessageRole } from "../types/enums";
 import { BaseMessageInput } from "../types/base";
 
-// GraphQL does not support union types in inputs, so we need to use
-// optional fields for the different subtypes.
-@InputType()
-export class MessageInput extends BaseMessageInput {
-  @Field(() => TextMessageInput, { nullable: true })
-  textMessage?: TextMessageInput;
-
-  @Field(() => ActionExecutionMessageInput, { nullable: true })
-  actionExecutionMessage?: ActionExecutionMessageInput;
-
-  @Field(() => ResultMessageInput, { nullable: true })
-  resultMessage?: ResultMessageInput;
-
-  @Field(() => AgentStateMessageInput, { nullable: true })
-  agentStateMessage?: AgentStateMessageInput;
-
-  @Field(() => ImageMessageInput, { nullable: true })
-  imageMessage?: ImageMessageInput;
-}
-
 @InputType()
 export class TextMessageInput {
   @Field(() => String)
@@ -108,3 +88,23 @@ export class ImageMessageInput {
   @Field(() => MessageRole)
   role: MessageRole;
 }
+
+// GraphQL does not support union types in inputs, so we need to use
+// optional fields for the different subtypes.
+@InputType()
+export class MessageInput extends BaseMessageInput {
+  @Field(() => TextMessageInput, { nullable: true })
+  textMessage?: TextMessageInput;
+
+  @Field(() => ActionExecutionMessageInput, { nullable: true })
+  actionExecutionMessage?: ActionExecutionMessageInput;
+
+  @Field(() => ResultMessageInput, { nullable: true })
+  resultMessage?: ResultMessageInput;
+
+  @Field(() => AgentStateMessageInput, { nullable: true })
+  agentStateMessage?: AgentStateMessageInput;
+
+  @Field(() => ImageMessageInput, { nullable: true })
+  imageMessage?: ImageMessageInput;
+}

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

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

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

@@ -1,4 +1,4 @@
-import { describe, test, expect } from "@jest/globals";
+import { vi } from "vitest";
 import * as gql from "../types/converted/index";
 import { MessageStatusCode } from "../types/message-status.type";
 import {
@@ -372,7 +372,7 @@ describe("message-conversion", () => {
         status: { code: MessageStatusCode.Pending },
       });
 
-      const mockRender = jest.fn();
+      const mockRender = vi.fn();
       const mockRenderAndWaitForResponse = (props: any) =>
         "Test Render With Response";
 
@@ -417,7 +417,7 @@ describe("message-conversion", () => {
         status: { code: MessageStatusCode.Pending },
       });
 
-      const mockRender = jest.fn();
+      const mockRender = vi.fn();
       const actions = {
         testAction: {
           name: "testAction",
@@ -448,7 +448,7 @@ describe("message-conversion", () => {
         status: { code: MessageStatusCode.Success },
       });
 
-      const mockRender = jest.fn();
+      const mockRender = vi.fn();
       const actions = {
         testAction: {
           name: "testAction",
@@ -486,7 +486,7 @@ describe("message-conversion", () => {
         actionName: "testAction",
       });
 
-      const mockRender = jest.fn();
+      const mockRender = vi.fn();
       const actions = {
         testAction: {
           name: "testAction",
@@ -522,7 +522,7 @@ describe("message-conversion", () => {
         status: { code: MessageStatusCode.Pending },
       });
 
-      const mockRender = jest.fn();
+      const mockRender = vi.fn();
       const actions = {
         testAction: {
           name: "testAction",
@@ -632,7 +632,7 @@ describe("message-conversion", () => {
         role: gql.Role.assistant,
       });
 
-      const mockRender = jest.fn();
+      const mockRender = vi.fn();
       const coAgentStateRenders = {
         testAgent: {
           name: "testAgent",
@@ -745,7 +745,7 @@ describe("message-conversion", () => {
         role: gql.Role.assistant,
       });
 
-      const mockRender = jest.fn();
+      const mockRender = vi.fn();
       const coAgentStateRenders = {
         testAgent: {
           name: "testAgent",
@@ -864,7 +864,7 @@ describe("message-conversion", () => {
       const actions = {
         testAction: {
           name: "testAction",
-          render: jest.fn((props) => `Rendered: ${props.args.test}`),
+          render: vi.fn((props) => `Rendered: ${props.args.test}`),
         },
       };
 
@@ -903,7 +903,7 @@ describe("message-conversion", () => {
       const actions = {
         "*": {
           name: "*",
-          render: jest.fn((props) => `Wildcard rendered: ${props.args.test}`),
+          render: vi.fn((props) => `Wildcard rendered: ${props.args.test}`),
         },
       };
 
@@ -939,7 +939,7 @@ describe("message-conversion", () => {
     });
 
     test("should pass tool name to wildcard action render function", () => {
-      const mockRender = jest.fn(
+      const mockRender = vi.fn(
         (props) =>
           `Wildcard rendered: ${props.name} with args: ${JSON.stringify(props.args)}`,
       );
@@ -975,7 +975,7 @@ describe("message-conversion", () => {
     });
 
     test("should pass tool name to regular action render function", () => {
-      const mockRender = jest.fn(
+      const mockRender = vi.fn(
         (props) => `Regular action rendered: ${JSON.stringify(props.args)}`,
       );
       const actions = {
@@ -1018,11 +1018,11 @@ describe("message-conversion", () => {
       const actions = {
         specificAction: {
           name: "specificAction",
-          render: jest.fn((props) => "Specific action rendered"),
+          render: vi.fn((props) => "Specific action rendered"),
         },
         "*": {
           name: "*",
-          render: jest.fn((props) => "Wildcard action rendered"),
+          render: vi.fn((props) => "Wildcard action rendered"),
         },
       };
 
@@ -1061,7 +1061,7 @@ describe("message-conversion", () => {
       const actions = {
         otherAction: {
           name: "otherAction",
-          render: jest.fn(),
+          render: vi.fn(),
         },
       };
 
@@ -1127,7 +1127,7 @@ describe("message-conversion", () => {
       const actions = {
         "*": {
           name: "*",
-          render: jest.fn((props) => `Result: ${props.result}`),
+          render: vi.fn((props) => `Result: ${props.result}`),
         },
       };
 
@@ -1169,7 +1169,7 @@ describe("message-conversion", () => {
       const actions = {
         "*": {
           name: "*",
-          render: jest.fn((props) => `Status: ${props.status}`),
+          render: vi.fn((props) => `Status: ${props.status}`),
         },
       };
 
@@ -1209,7 +1209,7 @@ describe("message-conversion", () => {
       const actions = {
         "*": {
           name: "*",
-          render: jest.fn((props) => `Status: ${props.status}`),
+          render: vi.fn((props) => `Status: ${props.status}`),
         },
       };
 
@@ -1249,7 +1249,7 @@ describe("message-conversion", () => {
       const actions = {
         "*": {
           name: "*",
-          render: jest.fn((props) => `Status: ${props.status}`),
+          render: vi.fn((props) => `Status: ${props.status}`),
         },
       };
 
@@ -1289,7 +1289,7 @@ describe("message-conversion", () => {
       const actions = {
         "*": {
           name: "*",
-          render: jest.fn((props) => `Status: ${props.status}`),
+          render: vi.fn((props) => `Status: ${props.status}`),
         },
       };
 
@@ -1329,7 +1329,7 @@ describe("message-conversion", () => {
       const actions = {
         "*": {
           name: "*",
-          render: jest.fn((props) => `Args: ${JSON.stringify(props.args)}`),
+          render: vi.fn((props) => `Args: ${JSON.stringify(props.args)}`),
         },
       };
 
@@ -1368,7 +1368,7 @@ describe("message-conversion", () => {
       const actions = {
         "*": {
           name: "*",
-          render: jest.fn((props) => `Args: ${JSON.stringify(props.args)}`),
+          render: vi.fn((props) => `Args: ${JSON.stringify(props.args)}`),
         },
       };
 
@@ -1407,7 +1407,7 @@ describe("message-conversion", () => {
       const actions = {
         "*": {
           name: "*",
-          render: jest.fn((props) => `Complex: ${JSON.stringify(props.args)}`),
+          render: vi.fn((props) => `Complex: ${JSON.stringify(props.args)}`),
         },
       };
 
@@ -1458,11 +1458,11 @@ describe("message-conversion", () => {
       const actions = {
         wildcard1: {
           name: "*",
-          render: jest.fn((props) => "First wildcard"),
+          render: vi.fn((props) => "First wildcard"),
         },
         wildcard2: {
           name: "*",
-          render: jest.fn((props) => "Second wildcard"),
+          render: vi.fn((props) => "Second wildcard"),
         },
       };
 
@@ -1501,7 +1501,7 @@ describe("message-conversion", () => {
       const actions = {
         "*": {
           name: "*",
-          render: jest.fn((props) => `Result: ${JSON.stringify(props.result)}`),
+          render: vi.fn((props) => `Result: ${JSON.stringify(props.result)}`),
         },
       };
 
@@ -1545,7 +1545,7 @@ describe("message-conversion", () => {
       const actions = {
         "*": {
           name: "*",
-          render: jest.fn((props) => `Result: ${JSON.stringify(props.result)}`),
+          render: vi.fn((props) => `Result: ${JSON.stringify(props.result)}`),
         },
       };
 
@@ -1586,7 +1586,7 @@ describe("message-conversion", () => {
       const actions = {
         "*": {
           name: "*",
-          render: jest.fn((props) => `Result: ${JSON.stringify(props.result)}`),
+          render: vi.fn((props) => `Result: ${JSON.stringify(props.result)}`),
         },
       };
 

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

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

package/src/graphql/types/extensions-response.type.ts

@@ -7,12 +7,6 @@ import { Field, ObjectType } from "type-graphql";
  * Next time a request to the runtime is made, the extensions response will be included in the request as input.
  */
 
-@ObjectType()
-export class ExtensionsResponse {
-  @Field(() => OpenAIApiAssistantAPIResponse, { nullable: true })
-  openaiAssistantAPI?: OpenAIApiAssistantAPIResponse;
-}
-
 @ObjectType()
 export class OpenAIApiAssistantAPIResponse {
   @Field(() => String, { nullable: true })
@@ -21,3 +15,9 @@ export class OpenAIApiAssistantAPIResponse {
   @Field(() => String, { nullable: true })
   threadId?: string;
 }
+
+@ObjectType()
+export class ExtensionsResponse {
+  @Field(() => OpenAIApiAssistantAPIResponse, { nullable: true })
+  openaiAssistantAPI?: OpenAIApiAssistantAPIResponse;
+}

package/src/index.ts

@@ -1,4 +1,3 @@
-import "reflect-metadata";
 export * from "./lib";
 export * from "./utils";
 export * from "./service-adapters";

package/src/lib/integrations/shared.ts

@@ -1,16 +1,12 @@
 import { YogaInitialContext } from "graphql-yoga";
 import { buildSchemaSync } from "type-graphql";
 import { CopilotResolver } from "../../graphql/resolvers/copilot.resolver";
-import { useDeferStream } from "@graphql-yoga/plugin-defer-stream";
 import { CopilotRuntime } from "../runtime/copilot-runtime";
 import { CopilotServiceAdapter } from "../../service-adapters";
 import { CopilotCloudOptions } from "../cloud";
 import { LogLevel, createLogger } from "../../lib/logger";
-import { createYoga } from "graphql-yoga";
 import telemetry from "../telemetry-client";
 import { StateResolver } from "../../graphql/resolvers/state.resolver";
-import * as packageJson from "../../../package.json";
-import { CopilotKitError, CopilotKitErrorCode } from "@copilotkit/shared";
 
 /**
  * CORS configuration for CopilotKit endpoints.
@@ -32,13 +28,6 @@ export interface CopilotEndpointCorsConfig {
 
 const logger = createLogger();
 
-export const addCustomHeaderPlugin = {
-  onResponse({ response }) {
-    // Set your custom header; adjust the header name and value as needed
-    response.headers.set("X-CopilotKit-Runtime-Version", packageJson.version);
-  },
-};
-
 type AnyPrimitive = string | boolean | number | null;
 export type CopilotRequestContextProperties = Record<
   string,
@@ -66,24 +55,6 @@ export interface CreateCopilotRuntimeServerOptions {
   cors?: CopilotEndpointCorsConfig;
 }
 
-export async function createContext(
-  initialContext: YogaInitialContext,
-  copilotKitContext: CreateCopilotRuntimeServerOptions,
-  contextLogger: typeof logger,
-  properties: CopilotRequestContextProperties = {},
-): Promise<Partial<GraphQLContext>> {
-  logger.debug({ copilotKitContext }, "Creating GraphQL context");
-  const ctx: GraphQLContext = {
-    ...initialContext,
-    _copilotkit: {
-      ...copilotKitContext,
-    },
-    properties: { ...properties },
-    logger: contextLogger,
-  };
-  return ctx;
-}
-
 export function buildSchema(
   options: {
     emitSchemaFile?: string;
@@ -100,12 +71,6 @@ export function buildSchema(
 
 export type CommonConfig = {
   logging: typeof logger;
-  schema: ReturnType<typeof buildSchema>;
-  plugins: Parameters<typeof createYoga>[0]["plugins"];
-  context: (ctx: YogaInitialContext) => Promise<Partial<GraphQLContext>>;
-  maskedErrors: {
-    maskError: (error: any, message: string, isDev?: boolean) => any;
-  };
 };
 
 export function getCommonConfig(
@@ -120,8 +85,6 @@ export function getCommonConfig(
     component: "getCommonConfig",
   });
 
-  const contextLogger = createLogger({ level: logLevel });
-
   if (options.cloud) {
     telemetry.setCloudConfiguration({
       publicApiKey: options.cloud.publicApiKey,
@@ -143,50 +106,7 @@ export function getCommonConfig(
     },
   });
 
-  // User error codes that should not be logged as server errors
-  const userErrorCodes = [
-    CopilotKitErrorCode.AGENT_NOT_FOUND,
-    CopilotKitErrorCode.API_NOT_FOUND,
-    CopilotKitErrorCode.REMOTE_ENDPOINT_NOT_FOUND,
-    CopilotKitErrorCode.CONFIGURATION_ERROR,
-    CopilotKitErrorCode.MISSING_PUBLIC_API_KEY_ERROR,
-  ];
-
   return {
-    logging: createLogger({ component: "Yoga GraphQL", level: logLevel }),
-    schema: buildSchema(),
-    plugins: [useDeferStream(), addCustomHeaderPlugin],
-    context: (ctx: YogaInitialContext): Promise<Partial<GraphQLContext>> =>
-      createContext(ctx, options, contextLogger, options.properties),
-    // Suppress logging for user configuration errors
-    maskedErrors: {
-      maskError: (error: any, message: string, isDev?: boolean) => {
-        // Check if this is a user configuration error (could be wrapped in GraphQLError)
-        const originalError = error.originalError || error;
-        const extensions = error.extensions;
-        const errorCode = extensions?.code;
-
-        // Suppress logging for user errors based on error code
-        if (errorCode && userErrorCodes.includes(errorCode)) {
-          // Log user configuration errors at debug level instead
-          console.debug("User configuration error:", error.message);
-          return error;
-        }
-
-        // Check if the original error is a user error
-        if (
-          originalError instanceof CopilotKitError &&
-          userErrorCodes.includes(originalError.code)
-        ) {
-          // Log user configuration errors at debug level instead
-          console.debug("User configuration error:", error.message);
-          return error;
-        }
-
-        // For application errors, log normally and mask if needed
-        console.error("Application error:", error);
-        return error;
-      },
-    },
+    logging: createLogger({ component: "CopilotKit Runtime", level: logLevel }),
   };
 }