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

package/dist/index.mjs

@@ -1,18 +1,18 @@
 import {
   CustomEventNames,
   LangGraphAgent
-} from "./chunk-6FLO2QTT.mjs";
+} from "./chunk-5SG4WWXH.mjs";
 import {
   config,
   copilotRuntimeNextJSAppRouterEndpoint,
   copilotRuntimeNextJSPagesRouterEndpoint
-} from "./chunk-6AD7VT26.mjs";
+} from "./chunk-JWPSIGSA.mjs";
 import {
   copilotRuntimeNestEndpoint
-} from "./chunk-K4JQLKXK.mjs";
+} from "./chunk-4TLMVLU4.mjs";
 import {
   copilotRuntimeNodeExpressEndpoint
-} from "./chunk-JZWHLURE.mjs";
+} from "./chunk-KYCDL2KX.mjs";
 import {
   CopilotRuntime,
   addCustomHeaderPlugin,
@@ -28,12 +28,7 @@ import {
   getCommonConfig,
   langGraphPlatformEndpoint,
   resolveEndpointType
-} from "./chunk-RCIYK6YJ.mjs";
-import {
-  GuardrailsValidationFailureResponse,
-  MessageStreamInterruptedResponse,
-  UnknownErrorResponse
-} from "./chunk-5BIEM2UU.mjs";
+} from "./chunk-WIXS6EG7.mjs";
 import {
   AnthropicAdapter,
   BedrockAdapter,
@@ -48,6 +43,11 @@ import {
   RemoteChain,
   UnifyAdapter
 } from "./chunk-IIXJVVTV.mjs";
+import {
+  GuardrailsValidationFailureResponse,
+  MessageStreamInterruptedResponse,
+  UnknownErrorResponse
+} from "./chunk-5BIEM2UU.mjs";
 import "./chunk-SHBDMA63.mjs";
 import "./chunk-2OZAGFV3.mjs";
 import "./chunk-FHD4JECV.mjs";

package/dist/lib/index.d.ts

@@ -6,7 +6,7 @@ export { copilotRuntimeNodeExpressEndpoint } from './integrations/node-express/i
 export { copilotRuntimeNestEndpoint } from './integrations/nest/index.js';
 import * as rxjs from 'rxjs';
 import { TextMessageStartEvent, TextMessageContentEvent, TextMessageEndEvent, ToolCallStartEvent, ToolCallArgsEvent, ToolCallEndEvent, RunAgentInput } from '@ag-ui/client';
-import { LangGraphAgent as LangGraphAgent$1, LangGraphAgentConfig, ProcessedEvents } from '@ag-ui/langgraph';
+import { LangGraphAgent as LangGraphAgent$1, LangGraphAgentConfig, ProcessedEvents, SchemaKeys } from '@ag-ui/langgraph';
 import { Message } from '@langchain/langgraph-sdk/dist/types.messages';
 import 'openai';
 import '../langserve-4a5c9217.js';
@@ -41,6 +41,7 @@ declare class LangGraphAgent extends LangGraphAgent$1 {
     dispatchEvent(event: ProcessedEvents): boolean;
     run(input: RunAgentInput): rxjs.Observable<any>;
     langGraphDefaultMergeState(state: State, messages: Message[], tools: any): State;
+    getSchemaKeys(): Promise<SchemaKeys>;
 }
 
 export { CustomEventNames, LangGraphAgent, PredictStateTool, State, TextMessageEvents, ToolCallEvents };

package/dist/lib/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",
@@ -3100,7 +3100,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) {
@@ -3143,7 +3142,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;
@@ -3201,18 +3200,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
@@ -3274,12 +3269,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
@@ -3298,23 +3304,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")
@@ -3329,8 +3320,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;
@@ -3401,9 +3392,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,
@@ -3735,6 +3726,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");
@@ -7255,6 +7264,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: