@copilotkit/runtime 1.4.0-pre-1-4-0.0 → 1.4.0-pre-1-4-0.11

package/dist/index.js

@@ -44,7 +44,7 @@ var require_package = __commonJS({
       publishConfig: {
         access: "public"
       },
-      version: "1.4.0-pre-1-4-0.0",
+      version: "1.4.0-pre-1-4-0.11",
       sideEffects: false,
       main: "./dist/index.js",
       module: "./dist/index.mjs",
@@ -1361,6 +1361,7 @@ var CustomEventNames;
   CustomEventNames2["CopilotKitManuallyEmitMessage"] = "copilotkit_manually_emit_message";
   CustomEventNames2["CopilotKitManuallyEmitToolCall"] = "copilotkit_manually_emit_tool_call";
   CustomEventNames2["CopilotKitManuallyEmitIntermediateState"] = "copilotkit_manually_emit_intermediate_state";
+  CustomEventNames2["CopilotKitExit"] = "copilotkit_exit";
 })(CustomEventNames || (CustomEventNames = {}));
 
 // src/agents/langgraph/event-source.ts
@@ -1580,6 +1581,7 @@ var MessageRole;
   MessageRole2["user"] = "user";
   MessageRole2["assistant"] = "assistant";
   MessageRole2["system"] = "system";
+  MessageRole2["tool"] = "tool";
 })(MessageRole || (MessageRole = {}));
 var ActionExecutionScope;
 (function(ActionExecutionScope2) {
@@ -1654,7 +1656,13 @@ async function streamEvents(controller, args) {
   const agentStateValues = agentState.values;
   state.messages = agentStateValues.messages;
   const mode = wasInitiatedWithExistingThread && nodeName != "__end__" ? "continue" : "start";
-  state = langGraphDefaultMergeState(state, formatMessages(messages), actions, name);
+  let formattedMessages = [];
+  try {
+    formattedMessages = formatMessages(messages);
+  } catch (e) {
+    logger2.error(e, `Error event thrown: ${e.message}`);
+  }
+  state = langGraphDefaultMergeState(state, formattedMessages, actions, name);
   if (mode === "continue") {
     await client.threads.updateState(threadId, {
       values: state,
@@ -1677,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,
@@ -1688,125 +1696,104 @@ async function streamEvents(controller, args) {
   });
   const emit = /* @__PURE__ */ __name((message) => controller.enqueue(new TextEncoder().encode(message)), "emit");
   let latestStateValues = {};
-  for await (const chunk of streamResponse2) {
-    if (![
-      "events",
-      "values",
-      "error"
-    ].includes(chunk.event))
-      continue;
-    if (chunk.event === "error") {
-      logger2.error(chunk, `Error event thrown: ${chunk.data.message}`);
-      throw new Error(`Error event thrown: ${chunk.data.message}`);
-    }
-    if (chunk.event === "values") {
-      latestStateValues = chunk.data;
-      continue;
-    }
-    const event = chunk.data;
-    const currentNodeName = event.name;
-    const eventType = event.event;
-    const runId = event.metadata.run_id;
-    externalRunId = runId;
-    const metadata = event.metadata;
-    shouldExit = shouldExit != null ? shouldExit : metadata["copilotkit:exit"];
-    const emitIntermediateState = metadata["copilotkit:emit-intermediate-state"];
-    const forceEmitIntermediateState = metadata["copilotkit:force-emit-intermediate-state"];
-    const manuallyEmitMessage = metadata["copilotkit:manually-emit-messages"];
-    const manuallyEmitToolCall = metadata["copilotkit:manually-emit-tool-calls"];
-    if (graphInfo["nodes"].some((node) => node.id === currentNodeName)) {
-      nodeName = currentNodeName;
-    }
-    if (!nodeName) {
-      continue;
-    }
-    if (forceEmitIntermediateState) {
-      if (eventType === "on_chain_end") {
-        state = event.data.output;
+  let updatedState = state;
+  try {
+    for await (const chunk of streamResponse2) {
+      if (![
+        "events",
+        "values",
+        "error"
+      ].includes(chunk.event))
+        continue;
+      if (chunk.event === "error") {
+        logger2.error(chunk, `Error event thrown: ${chunk.data.message}`);
+        throw new Error(`Error event thrown: ${chunk.data.message}`);
+      }
+      if (chunk.event === "values") {
+        latestStateValues = chunk.data;
+        continue;
+      }
+      const event = chunk.data;
+      const currentNodeName = event.name;
+      const eventType = event.event;
+      const runId = event.metadata.run_id;
+      externalRunId = runId;
+      const metadata = event.metadata;
+      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) {
+        updatedState = event.data;
         emit(getStateSyncEvent({
           threadId,
           runId,
           agentName: agent.name,
           nodeName,
-          state: event.data.output,
+          state: updatedState,
           running: true,
           active: true
         }));
+        continue;
       }
-      continue;
-    }
-    if (manuallyEmitMessage) {
-      if (eventType === "on_chain_end") {
-        state = event.data.output;
-        emit(JSON.stringify({
-          event: "on_copilotkit_emit_message",
-          message: event.data.output,
-          messageId: (0, import_node_crypto.randomUUID)(),
-          role: MessageRole.assistant
-        }) + "\n");
+      if (emitIntermediateState && emitIntermediateStateUntilEnd == null) {
+        emitIntermediateStateUntilEnd = nodeName;
       }
-      continue;
-    }
-    if (manuallyEmitToolCall) {
-      if (eventType === "on_chain_end") {
-        state = event.data.output;
-        emit(JSON.stringify({
-          event: "on_copilotkit_emit_tool_call",
-          name: event.data.output.name,
-          args: event.data.output.args,
-          id: event.data.output.id
-        }) + "\n");
+      if (emitIntermediateState && eventType === LangGraphEventTypes.OnChatModelStart) {
+        streamingStateExtractor = new StreamingStateExtractor(emitIntermediateState);
       }
-      continue;
-    }
-    if (emitIntermediateState && emitIntermediateStateUntilEnd == null) {
-      emitIntermediateStateUntilEnd = nodeName;
-    }
-    if (emitIntermediateState && eventType === "on_chat_model_start") {
-      streamingStateExtractor = new StreamingStateExtractor(emitIntermediateState);
-    }
-    let updatedState = latestStateValues;
-    if (emitIntermediateState && eventType === "on_chat_model_stream") {
-      streamingStateExtractor.bufferToolCalls(event);
-    }
-    if (emitIntermediateStateUntilEnd !== null) {
-      updatedState = {
-        ...updatedState,
-        ...streamingStateExtractor.extractState()
-      };
-    }
-    if (!emitIntermediateState && currentNodeName === emitIntermediateStateUntilEnd && eventType === "on_chain_end") {
-      emitIntermediateStateUntilEnd = null;
-    }
-    const exitingNode = nodeName === currentNodeName && eventType === "on_chain_end";
-    if (JSON.stringify(updatedState) !== JSON.stringify(state) || prevNodeName != nodeName || exitingNode) {
-      state = updatedState;
-      prevNodeName = nodeName;
-      emit(getStateSyncEvent({
-        threadId,
-        runId,
-        agentName: agent.name,
-        nodeName,
-        state,
-        running: true,
-        active: !exitingNode
-      }));
+      if (emitIntermediateState && eventType === LangGraphEventTypes.OnChatModelStream) {
+        streamingStateExtractor.bufferToolCalls(event);
+      }
+      if (emitIntermediateStateUntilEnd !== null) {
+        updatedState = {
+          ...updatedState,
+          ...streamingStateExtractor.extractState()
+        };
+      }
+      if (!emitIntermediateState && currentNodeName === emitIntermediateStateUntilEnd && eventType === LangGraphEventTypes.OnChainEnd) {
+        emitIntermediateStateUntilEnd = null;
+      }
+      const exitingNode = nodeName === currentNodeName && eventType === LangGraphEventTypes.OnChainEnd;
+      if (JSON.stringify(updatedState) !== JSON.stringify(state) || prevNodeName != nodeName || exitingNode) {
+        state = updatedState;
+        prevNodeName = nodeName;
+        emit(getStateSyncEvent({
+          threadId,
+          runId,
+          agentName: agent.name,
+          nodeName,
+          state,
+          running: true,
+          active: !exitingNode
+        }));
+      }
+      emit(JSON.stringify(event) + "\n");
     }
-    emit(JSON.stringify(event) + "\n");
+    state = await client.threads.getState(threadId);
+    const isEndNode = state.next.length === 0;
+    nodeName = Object.keys(state.metadata.writes)[0];
+    emit(getStateSyncEvent({
+      threadId,
+      runId: externalRunId,
+      agentName: agent.name,
+      nodeName: isEndNode ? "__end__" : nodeName,
+      state: state.values,
+      running: !shouldExit,
+      active: false
+    }));
+    return Promise.resolve();
+  } catch (e) {
+    return Promise.resolve();
   }
-  state = await client.threads.getState(threadId);
-  const isEndNode = state.next.length === 0;
-  nodeName = Object.keys(state.metadata.writes)[0];
-  emit(getStateSyncEvent({
-    threadId,
-    runId: externalRunId,
-    agentName: agent.name,
-    nodeName: isEndNode ? "__end__" : nodeName,
-    state: state.values,
-    running: !shouldExit,
-    active: false
-  }));
-  return Promise.resolve();
 }
 __name(streamEvents, "streamEvents");
 function getStateSyncEvent({ threadId, runId, agentName, nodeName, state, running, active }) {
@@ -1817,7 +1804,7 @@ function getStateSyncEvent({ threadId, runId, agentName, nodeName, state, runnin
     return acc;
   }, {});
   return JSON.stringify({
-    event: "on_copilotkit_state_sync",
+    event: LangGraphEventTypes.OnCopilotKitStateSync,
     thread_id: threadId,
     run_id: runId,
     agent_name: agentName,
@@ -1843,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;
       }
     }
@@ -1922,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;
         }
@@ -1979,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") {
@@ -2041,7 +2007,7 @@ function formatMessages(messages) {
         id: message.id,
         tool_call_id: message.actionExecutionId,
         name: message.actionName,
-        role: MessageRole.assistant
+        role: MessageRole.tool
       };
     }
     throw new Error(`Unknown message type ${message.type}`);