@copilotkit/runtime 1.9.2-next.6 → 1.9.2-next.8

package/dist/index.js

@@ -44,7 +44,7 @@ var require_package = __commonJS({
       publishConfig: {
         access: "public"
       },
-      version: "1.9.2-next.6",
+      version: "1.9.2-next.8",
       sideEffects: false,
       main: "./dist/index.js",
       module: "./dist/index.mjs",
@@ -3110,7 +3110,6 @@ async function fetchWithRetry(url, options, logger2) {
 __name(fetchWithRetry, "fetchWithRetry");
 
 // src/lib/runtime/remote-lg-action.ts
-var activeInterruptEvent = false;
 async function execute(args) {
   return new ReadableStream({
     async start(controller) {
@@ -3153,7 +3152,7 @@ async function execute(args) {
 }
 __name(execute, "execute");
 async function streamEvents(controller, args) {
-  var _a, _b, _c, _d, _e, _f, _g, _h;
+  var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k;
   const { deploymentUrl, langsmithApiKey, threadId: argsInitialThreadId, agent, nodeName: initialNodeName, state: initialState, config: explicitConfig, messages, actions, logger: logger2, properties, metaEvents } = args;
   let nodeName = initialNodeName;
   let state = initialState;
@@ -3211,18 +3210,14 @@ async function streamEvents(controller, args) {
     command: void 0
   };
   const lgInterruptMetaEvent = metaEvents == null ? void 0 : metaEvents.find((ev) => ev.name === MetaEventName.LangGraphInterruptEvent);
-  if (activeInterruptEvent && !lgInterruptMetaEvent) {
-    payload.command = {
-      resume: state.messages
-    };
-  }
   if (lgInterruptMetaEvent == null ? void 0 : lgInterruptMetaEvent.response) {
     let response = lgInterruptMetaEvent.response;
     payload.command = {
       resume: (0, import_shared9.parseJson)(response, response)
     };
   }
-  if (mode === "continue" && !activeInterruptEvent) {
+  const interrupts = ((_b = (_a = agentState.tasks) == null ? void 0 : _a[0]) == null ? void 0 : _b.interrupts) ?? [];
+  if (mode === "continue" && !interrupts.length) {
     await client.threads.updateState(threadId, {
       values: state,
       asNode: nodeName
@@ -3284,12 +3279,23 @@ async function streamEvents(controller, args) {
   let emitIntermediateStateUntilEnd = null;
   let shouldExit = false;
   let externalRunId = null;
-  const streamResponse = client.runs.stream(threadId, assistantId, payload);
   const emit = /* @__PURE__ */ __name((message) => controller.enqueue(new TextEncoder().encode(message)), "emit");
+  if ((interrupts == null ? void 0 : interrupts.length) && !((_c = payload.command) == null ? void 0 : _c.resume)) {
+    if (!lgInterruptMetaEvent) {
+      payload.command = {
+        resume: state.messages
+      };
+    } else {
+      interrupts.forEach((interrupt) => {
+        emitInterrupt(interrupt.value, emit);
+      });
+      return Promise.resolve();
+    }
+  }
+  const streamResponse = client.runs.stream(threadId, assistantId, payload);
   let latestStateValues = {};
   let updatedState = state;
   let manuallyEmittedState = null;
-  activeInterruptEvent = false;
   try {
     telemetry_client_default.capture("oss.runtime.agent_execution_stream_started", {
       hashedLgcKey: streamInfo.hashedLgcKey
@@ -3308,23 +3314,8 @@ async function streamEvents(controller, args) {
       const chunk = streamResponseChunk;
       const interruptEvents = chunk.data.__interrupt__;
       if (interruptEvents == null ? void 0 : interruptEvents.length) {
-        activeInterruptEvent = true;
         const interruptValue = interruptEvents == null ? void 0 : interruptEvents[0].value;
-        if (typeof interruptValue != "string" && "__copilotkit_interrupt_value__" in interruptValue) {
-          const evValue = interruptValue.__copilotkit_interrupt_value__;
-          emit(JSON.stringify({
-            event: LangGraphEventTypes.OnCopilotKitInterrupt,
-            data: {
-              value: typeof evValue === "string" ? evValue : JSON.stringify(evValue),
-              messages: langchainMessagesToCopilotKit(interruptValue.__copilotkit_messages__)
-            }
-          }) + "\n");
-        } else {
-          emit(JSON.stringify({
-            event: LangGraphEventTypes.OnInterrupt,
-            value: typeof interruptValue === "string" ? interruptValue : JSON.stringify(interruptValue)
-          }) + "\n");
-        }
+        emitInterrupt(interruptValue, emit);
         continue;
       }
       if (streamResponseChunk.event === "updates")
@@ -3339,8 +3330,8 @@ async function streamEvents(controller, args) {
       const runId = chunkData.metadata.run_id;
       externalRunId = runId;
       const metadata = chunkData.metadata;
-      if (((_b = (_a = chunkData.data) == null ? void 0 : _a.output) == null ? void 0 : _b.model) != null && ((_d = (_c = chunkData.data) == null ? void 0 : _c.output) == null ? void 0 : _d.model) != "") {
-        streamInfo.provider = (_f = (_e = chunkData.data) == null ? void 0 : _e.output) == null ? void 0 : _f.model;
+      if (((_e = (_d = chunkData.data) == null ? void 0 : _d.output) == null ? void 0 : _e.model) != null && ((_g = (_f = chunkData.data) == null ? void 0 : _f.output) == null ? void 0 : _g.model) != "") {
+        streamInfo.provider = (_i = (_h = chunkData.data) == null ? void 0 : _h.output) == null ? void 0 : _i.model;
       }
       if (metadata.langgraph_host != null && metadata.langgraph_host != "") {
         streamInfo.langGraphHost = metadata.langgraph_host;
@@ -3411,9 +3402,9 @@ async function streamEvents(controller, args) {
       emit(JSON.stringify(chunkData) + "\n");
     }
     state = await client.threads.getState(threadId);
-    const interrupts = (_h = (_g = state.tasks) == null ? void 0 : _g[0]) == null ? void 0 : _h.interrupts;
-    nodeName = interrupts ? nodeName : Object.keys(state.metadata.writes)[0];
-    const isEndNode = state.next.length === 0 && !interrupts;
+    const interrupts2 = (_k = (_j = state.tasks) == null ? void 0 : _j[0]) == null ? void 0 : _k.interrupts;
+    nodeName = interrupts2 ? nodeName : Object.keys(state.metadata.writes)[0];
+    const isEndNode = state.next.length === 0 && !interrupts2;
     telemetry_client_default.capture("oss.runtime.agent_execution_stream_ended", streamInfo);
     emit(getStateSyncEvent({
       threadId,
@@ -3745,6 +3736,24 @@ function filterObjectBySchemaKeys(obj, schemaKeys) {
   return Object.fromEntries(Object.entries(obj).filter(([key]) => schemaKeys.includes(key)));
 }
 __name(filterObjectBySchemaKeys, "filterObjectBySchemaKeys");
+function emitInterrupt(interruptValue, emit) {
+  if (typeof interruptValue != "string" && "__copilotkit_interrupt_value__" in interruptValue) {
+    const evValue = interruptValue.__copilotkit_interrupt_value__;
+    emit(JSON.stringify({
+      event: LangGraphEventTypes.OnCopilotKitInterrupt,
+      data: {
+        value: typeof evValue === "string" ? evValue : JSON.stringify(evValue),
+        messages: langchainMessagesToCopilotKit(interruptValue.__copilotkit_messages__)
+      }
+    }) + "\n");
+  } else {
+    emit(JSON.stringify({
+      event: LangGraphEventTypes.OnInterrupt,
+      value: typeof interruptValue === "string" ? interruptValue : JSON.stringify(interruptValue)
+    }) + "\n");
+  }
+}
+__name(emitInterrupt, "emitInterrupt");
 
 // src/lib/runtime/remote-action-constructors.ts
 var import_shared11 = require("@copilotkit/shared");
@@ -7667,6 +7676,23 @@ var LangGraphAgent = class extends import_langgraph.LangGraphAgent {
       }
     };
   }
+  async getSchemaKeys() {
+    const CONSTANT_KEYS = [
+      "copilotkit"
+    ];
+    const schemaKeys = await super.getSchemaKeys();
+    return {
+      config: schemaKeys.config,
+      input: schemaKeys.input ? [
+        ...schemaKeys.input,
+        ...CONSTANT_KEYS
+      ] : null,
+      output: schemaKeys.output ? [
+        ...schemaKeys.output,
+        ...CONSTANT_KEYS
+      ] : null
+    };
+  }
 };
 __name(LangGraphAgent, "LangGraphAgent");
 // Annotate the CommonJS export names for ESM import in node: