@copilotkit/runtime 1.8.4-next.2 → 1.8.4-next.3

package/dist/lib/integrations/node-http/index.mjs

@@ -1,6 +1,6 @@
 import {
   copilotRuntimeNodeHttpEndpoint
-} from "../../../chunk-ZWY6JDFI.mjs";
+} from "../../../chunk-FEE2BWEL.mjs";
 import "../../../chunk-FZJAYGIR.mjs";
 import "../../../chunk-5BIEM2UU.mjs";
 import "../../../chunk-RTFJTJMA.mjs";

package/package.json

@@ -9,7 +9,7 @@
   "publishConfig": {
     "access": "public"
   },
-  "version": "1.8.4-next.2",
+  "version": "1.8.4-next.3",
   "sideEffects": false,
   "main": "./dist/index.js",
   "module": "./dist/index.mjs",
@@ -59,7 +59,7 @@
     "rxjs": "^7.8.1",
     "type-graphql": "2.0.0-rc.1",
     "zod": "^3.23.3",
-    "@copilotkit/shared": "1.8.4-next.2"
+    "@copilotkit/shared": "1.8.4-next.3"
   },
   "keywords": [
     "copilotkit",

package/src/lib/runtime/remote-lg-action.ts

@@ -68,7 +68,11 @@ type LangGraphPlatformMessage =
   | LangGraphPlatformResultMessage
   | BaseLangGraphPlatformMessage;
 
-type SchemaKeys = { input: string[] | null; output: string[] | null } | null;
+type SchemaKeys = {
+  input: string[] | null;
+  output: string[] | null;
+  config: string[] | null;
+} | null;
 
 let activeInterruptEvent = false;
 
@@ -202,17 +206,19 @@ async function streamEvents(controller: ReadableStreamDefaultController, args: E
   }
   const assistantId = retrievedAssistant.assistant_id;
 
-  if (configurable) {
-    await client.assistants.update(assistantId, { config: { configurable } });
-  }
   const graphInfo = await client.assistants.getGraph(assistantId);
   const graphSchema = await client.assistants.getSchemas(assistantId);
   const schemaKeys = getSchemaKeys(graphSchema);
+  if (configurable) {
+    const filteredConfigurable = schemaKeys?.config
+      ? filterObjectBySchemaKeys(configurable, schemaKeys?.config)
+      : configurable;
+    await client.assistants.update(assistantId, { config: { configurable: filteredConfigurable } });
+  }
+
   // Do not input keys that are not part of the input schema
   if (payload.input && schemaKeys?.input) {
-    payload.input = Object.fromEntries(
-      Object.entries(payload.input).filter(([key]) => schemaKeys.input.includes(key)),
-    );
+    payload.input = filterObjectBySchemaKeys(payload.input, schemaKeys.input);
   }
 
   let streamingStateExtractor = new StreamingStateExtractor([]);
@@ -468,9 +474,7 @@ function getStateSyncEvent({
 
   // Do not emit state keys that are not part of the output schema
   if (schemaKeys?.output) {
-    state = Object.fromEntries(
-      Object.entries(state).filter(([key]) => schemaKeys.output.includes(key)),
-    );
+    state = filterObjectBySchemaKeys(state, schemaKeys.output);
   }
 
   return (
@@ -764,8 +768,12 @@ function copilotkitMessagesToLangChain(messages: Message[]): LangGraphPlatformMe
 
 function getSchemaKeys(graphSchema: GraphSchema): SchemaKeys {
   const CONSTANT_KEYS = ["messages", "copilotkit"];
+  let configSchema = null;
+  if (graphSchema.config_schema.properties) {
+    configSchema = Object.keys(graphSchema.config_schema.properties);
+  }
   if (!graphSchema.input_schema.properties || !graphSchema.output_schema.properties) {
-    return null;
+    return configSchema;
   }
   const inputSchema = Object.keys(graphSchema.input_schema.properties);
   const outputSchema = Object.keys(graphSchema.output_schema.properties);
@@ -773,5 +781,10 @@ function getSchemaKeys(graphSchema: GraphSchema): SchemaKeys {
   return {
     input: inputSchema && inputSchema.length ? [...inputSchema, ...CONSTANT_KEYS] : null,
     output: outputSchema && outputSchema.length ? [...outputSchema, ...CONSTANT_KEYS] : null,
+    config: configSchema,
   };
 }
+
+function filterObjectBySchemaKeys(obj: Record<string, any>, schemaKeys: string[]) {
+  return Object.fromEntries(Object.entries(obj).filter(([key]) => schemaKeys.includes(key)));
+}