@copilotkit/runtime 1.4.0-lgc-alpha3.0 → 1.4.0-pre-1-4-0.10

package/dist/index.js

@@ -44,7 +44,7 @@ var require_package = __commonJS({
       publishConfig: {
         access: "public"
       },
-      version: "1.4.0-lgc-alpha3.0",
+      version: "1.4.0-pre-1-4-0.10",
       sideEffects: false,
       main: "./dist/index.js",
       module: "./dist/index.mjs",
@@ -1685,7 +1685,7 @@ async function streamEvents(controller, args) {
   let streamingStateExtractor = new StreamingStateExtractor([]);
   let prevNodeName = null;
   let emitIntermediateStateUntilEnd = null;
-  let shouldExit = null;
+  let shouldExit = false;
   let externalRunId = null;
   const streamResponse2 = client.runs.stream(threadId, assistantId, {
     input: streamInput,
@@ -1696,6 +1696,7 @@ async function streamEvents(controller, args) {
   });
   const emit = /* @__PURE__ */ __name((message) => controller.enqueue(new TextEncoder().encode(message)), "emit");
   let latestStateValues = {};
+  let updatedState = state;
   try {
     for await (const chunk of streamResponse2) {
       if (![
@@ -1718,28 +1719,29 @@ async function streamEvents(controller, args) {
       const runId = event.metadata.run_id;
       externalRunId = runId;
       const metadata = event.metadata;
-      shouldExit = shouldExit != null ? shouldExit : eventType === LangGraphEventTypes.OnCustomEvent && event.name === CustomEventNames.CopilotKitExit;
+      shouldExit = shouldExit || eventType === LangGraphEventTypes.OnCustomEvent && event.name === CustomEventNames.CopilotKitExit;
       const emitIntermediateState = metadata["copilotkit:emit-intermediate-state"];
       const manuallyEmitIntermediateState = eventType === LangGraphEventTypes.OnCustomEvent && event.name === CustomEventNames.CopilotKitManuallyEmitIntermediateState;
       if (graphInfo["nodes"].some((node) => node.id === currentNodeName)) {
         nodeName = currentNodeName;
+        if (eventType === LangGraphEventTypes.OnChainStart || eventType === LangGraphEventTypes.OnChainEnd) {
+          updatedState = latestStateValues;
+        }
       }
       if (!nodeName) {
         continue;
       }
       if (manuallyEmitIntermediateState) {
-        if (eventType === LangGraphEventTypes.OnChainEnd) {
-          state = event.data.output;
-          emit(getStateSyncEvent({
-            threadId,
-            runId,
-            agentName: agent.name,
-            nodeName,
-            state: event.data.output,
-            running: true,
-            active: true
-          }));
-        }
+        updatedState = event.data;
+        emit(getStateSyncEvent({
+          threadId,
+          runId,
+          agentName: agent.name,
+          nodeName,
+          state: updatedState,
+          running: true,
+          active: true
+        }));
         continue;
       }
       if (emitIntermediateState && emitIntermediateStateUntilEnd == null) {
@@ -1748,7 +1750,6 @@ async function streamEvents(controller, args) {
       if (emitIntermediateState && eventType === LangGraphEventTypes.OnChatModelStart) {
         streamingStateExtractor = new StreamingStateExtractor(emitIntermediateState);
       }
-      let updatedState = latestStateValues;
       if (emitIntermediateState && eventType === LangGraphEventTypes.OnChatModelStream) {
         streamingStateExtractor.bufferToolCalls(event);
       }
@@ -1829,10 +1830,10 @@ var StreamingStateExtractor = /* @__PURE__ */ __name(class StreamingStateExtract
   bufferToolCalls(event) {
     if (event.data.chunk.tool_call_chunks.length > 0) {
       const chunk = event.data.chunk.tool_call_chunks[0];
-      if (chunk.name !== null) {
+      if (chunk.name !== null && chunk.name !== void 0) {
         this.currentToolCall = chunk.name;
         this.toolCallBuffer[this.currentToolCall] = chunk.args;
-      } else if (this.currentToolCall !== null) {
+      } else if (this.currentToolCall !== null && this.currentToolCall !== void 0) {
         this.toolCallBuffer[this.currentToolCall] += chunk.args;
       }
     }
@@ -1908,12 +1909,10 @@ function langGraphDefaultMergeState(state, messages, actions, agentName) {
       mergedMessages.push(message);
     } else {
       for (let i = 0; i < mergedMessages.length; i++) {
-        if (mergedMessages[i].id === message.id) {
-          if ("tool_calls" in message) {
-            if (("tool_calls" in mergedMessages[i] || "additional_kwargs" in mergedMessages[i]) && mergedMessages[i].content) {
-              message.tool_calls = mergedMessages[i]["tool_calls"];
-              message.additional_kwargs = mergedMessages[i].additional_kwargs;
-            }
+        if (mergedMessages[i].id === message.id && message.role === "assistant") {
+          if (("tool_calls" in mergedMessages[i] || "additional_kwargs" in mergedMessages[i]) && mergedMessages[i].content) {
+            message.tool_calls = mergedMessages[i]["tool_calls"];
+            message.additional_kwargs = mergedMessages[i].additional_kwargs;
           }
           mergedMessages[i] = message;
         }
@@ -1965,34 +1964,15 @@ function langGraphDefaultMergeState(state, messages, actions, agentName) {
     }
     correctedMessages.push(currentMessage);
   }
-  return deepMerge(state, {
+  return {
+    ...state,
     messages: correctedMessages,
     copilotkit: {
       actions
     }
-  });
-}
-__name(langGraphDefaultMergeState, "langGraphDefaultMergeState");
-function deepMerge(obj1, obj2) {
-  let result = {
-    ...obj1
   };
-  for (let key in obj2) {
-    if (typeof obj2[key] === "object" && !Array.isArray(obj2[key])) {
-      if (obj1[key]) {
-        result[key] = deepMerge(obj1[key], obj2[key]);
-      } else {
-        result[key] = {
-          ...obj2[key]
-        };
-      }
-    } else {
-      result[key] = obj2[key];
-    }
-  }
-  return result;
 }
-__name(deepMerge, "deepMerge");
+__name(langGraphDefaultMergeState, "langGraphDefaultMergeState");
 function formatMessages(messages) {
   return messages.map((message) => {
     if (message.isTextMessage() && message.role === "assistant") {