@copilotkit/runtime 1.6.0-next.7 → 1.6.0-next.8

package/dist/index.mjs

@@ -3,13 +3,13 @@ import {
   config,
   copilotRuntimeNextJSAppRouterEndpoint,
   copilotRuntimeNextJSPagesRouterEndpoint
-} from "./chunk-DV4VS5ER.mjs";
+} from "./chunk-DI2VJM3I.mjs";
 import {
   copilotRuntimeNestEndpoint
-} from "./chunk-LTUE24TH.mjs";
+} from "./chunk-LR7HA4X5.mjs";
 import {
   copilotRuntimeNodeExpressEndpoint
-} from "./chunk-LVJSZOFH.mjs";
+} from "./chunk-CC5KHDCV.mjs";
 import {
   CopilotRuntime,
   addCustomHeaderPlugin,
@@ -21,7 +21,7 @@ import {
   getCommonConfig,
   langGraphPlatformEndpoint,
   resolveEndpointType
-} from "./chunk-SAHKR6PK.mjs";
+} from "./chunk-C7VKUXWC.mjs";
 import {
   AnthropicAdapter,
   EmptyAdapter,

package/dist/lib/index.js

@@ -44,7 +44,7 @@ var require_package = __commonJS({
       publishConfig: {
         access: "public"
       },
-      version: "1.6.0-next.7",
+      version: "1.6.0-next.8",
       sideEffects: false,
       main: "./dist/index.js",
       module: "./dist/index.mjs",
@@ -2283,6 +2283,11 @@ async function streamEvents(controller, args) {
     });
   }
   const graphInfo = await client.assistants.getGraph(assistantId);
+  const graphSchema = await client.assistants.getSchemas(assistantId);
+  const schemaKeys = getSchemaKeys(graphSchema);
+  if (payload.input && schemaKeys.input) {
+    payload.input = Object.fromEntries(Object.entries(payload.input).filter(([key]) => schemaKeys.input.includes(key)));
+  }
   let streamingStateExtractor = new StreamingStateExtractor([]);
   let prevNodeName = null;
   let emitIntermediateStateUntilEnd = null;
@@ -2374,7 +2379,8 @@ async function streamEvents(controller, args) {
           nodeName,
           state: manuallyEmittedState,
           running: true,
-          active: true
+          active: true,
+          schemaKeys
         }));
         continue;
       }
@@ -2406,7 +2412,8 @@ async function streamEvents(controller, args) {
           nodeName,
           state,
           running: true,
-          active: !exitingNode
+          active: !exitingNode,
+          schemaKeys
         }));
       }
       emit(JSON.stringify(event) + "\n");
@@ -2424,7 +2431,8 @@ async function streamEvents(controller, args) {
       state: state.values,
       running: !shouldExit,
       active: false,
-      includeMessages: true
+      includeMessages: true,
+      schemaKeys
     }));
     return Promise.resolve();
   } catch (e) {
@@ -2437,7 +2445,7 @@ async function streamEvents(controller, args) {
   }
 }
 __name(streamEvents, "streamEvents");
-function getStateSyncEvent({ threadId, runId, agentName, nodeName, state, running, active, includeMessages = false }) {
+function getStateSyncEvent({ threadId, runId, agentName, nodeName, state, running, active, includeMessages = false, schemaKeys }) {
   if (!includeMessages) {
     state = Object.keys(state).reduce((acc, key) => {
       if (key !== "messages") {
@@ -2451,6 +2459,9 @@ function getStateSyncEvent({ threadId, runId, agentName, nodeName, state, runnin
       messages: langchainMessagesToCopilotKit(state.messages || [])
     };
   }
+  if (schemaKeys.output) {
+    state = Object.fromEntries(Object.entries(state).filter(([key]) => schemaKeys.output.includes(key)));
+  }
   return JSON.stringify({
     event: LangGraphEventTypes.OnCopilotKitStateSync,
     thread_id: threadId,
@@ -2685,6 +2696,25 @@ function copilotkitMessagesToLangChain(messages) {
   return result;
 }
 __name(copilotkitMessagesToLangChain, "copilotkitMessagesToLangChain");
+function getSchemaKeys(graphSchema) {
+  const CONSTANT_KEYS = [
+    "messages",
+    "copilotkit"
+  ];
+  const inputSchema = Object.keys(graphSchema.input_schema.properties);
+  const outputSchema = Object.keys(graphSchema.output_schema.properties);
+  return {
+    input: inputSchema && inputSchema.length ? [
+      ...inputSchema,
+      ...CONSTANT_KEYS
+    ] : null,
+    output: outputSchema && outputSchema.length ? [
+      ...outputSchema,
+      ...CONSTANT_KEYS
+    ] : null
+  };
+}
+__name(getSchemaKeys, "getSchemaKeys");
 
 // src/lib/runtime/remote-action-constructors.ts
 var import_shared11 = require("@copilotkit/shared");