@copilotkit/runtime 1.5.0-tyler-reset-chat.0 → 1.5.1-custom-tag-pre.0

package/dist/{chunk-K67A6XOJ.mjs → chunk-3ECBC2K2.mjs}

@@ -1,7 +1,7 @@
 import {
   RemoteChain,
   streamLangChainResponse
-} from "./chunk-OKQVDDJ2.mjs";
+} from "./chunk-CLGKEUOA.mjs";
 import {
   GuardrailsValidationFailureResponse,
   MessageStreamInterruptedResponse,
@@ -14,7 +14,7 @@ import {
   AgentStateMessage,
   ResultMessage,
   TextMessage
-} from "./chunk-B74M7FXG.mjs";
+} from "./chunk-RFF5IIZJ.mjs";
 import {
   BaseMessageInput
 } from "./chunk-D2WLFQS6.mjs";
@@ -37,7 +37,7 @@ var require_package = __commonJS({
       publishConfig: {
         access: "public"
       },
-      version: "1.5.0-tyler-reset-chat.0",
+      version: "1.5.1-custom-tag-pre.0",
       sideEffects: false,
       main: "./dist/index.js",
       module: "./dist/index.mjs",
@@ -122,7 +122,7 @@ import { buildSchemaSync } from "type-graphql";
 
 // src/graphql/resolvers/copilot.resolver.ts
 import { Arg, Ctx, Mutation, Query, Resolver } from "type-graphql";
-import { ReplaySubject as ReplaySubject3, Subject, finalize, firstValueFrom as firstValueFrom2, shareReplay, skipWhile, take, takeWhile, tap } from "rxjs";
+import { ReplaySubject as ReplaySubject3, Subject, filter, finalize, firstValueFrom as firstValueFrom2, shareReplay, skipWhile, take, takeWhile, tap } from "rxjs";
 
 // src/graphql/inputs/generate-copilot-response.input.ts
 import { Field as Field9, InputType as InputType9 } from "type-graphql";
@@ -139,12 +139,6 @@ var MessageRole;
   MessageRole2["system"] = "system";
   MessageRole2["tool"] = "tool";
 })(MessageRole || (MessageRole = {}));
-var ActionExecutionScope;
-(function(ActionExecutionScope2) {
-  ActionExecutionScope2["server"] = "server";
-  ActionExecutionScope2["client"] = "client";
-  ActionExecutionScope2["passThrough"] = "passThrough";
-})(ActionExecutionScope || (ActionExecutionScope = {}));
 var CopilotRequestType;
 (function(CopilotRequestType2) {
   CopilotRequestType2["Chat"] = "Chat";
@@ -163,10 +157,6 @@ registerEnumType(MessageRole, {
   name: "MessageRole",
   description: "The role of the message"
 });
-registerEnumType(ActionExecutionScope, {
-  name: "ActionExecutionScope",
-  description: "The scope of the action"
-});
 registerEnumType(CopilotRequestType, {
   name: "CopilotRequestType",
   description: "The type of Copilot request"
@@ -229,6 +219,7 @@ MessageInput = _ts_decorate([
 ], MessageInput);
 var TextMessageInput = class {
   content;
+  parentMessageId;
   role;
 };
 __name(TextMessageInput, "TextMessageInput");
@@ -236,6 +227,12 @@ _ts_decorate([
   Field(() => String),
   _ts_metadata("design:type", String)
 ], TextMessageInput.prototype, "content", void 0);
+_ts_decorate([
+  Field(() => String, {
+    nullable: true
+  }),
+  _ts_metadata("design:type", String)
+], TextMessageInput.prototype, "parentMessageId", void 0);
 _ts_decorate([
   Field(() => MessageRole),
   _ts_metadata("design:type", typeof MessageRole === "undefined" ? Object : MessageRole)
@@ -246,6 +243,7 @@ TextMessageInput = _ts_decorate([
 var ActionExecutionMessageInput = class {
   name;
   arguments;
+  parentMessageId;
   scope;
 };
 __name(ActionExecutionMessageInput, "ActionExecutionMessageInput");
@@ -258,8 +256,17 @@ _ts_decorate([
   _ts_metadata("design:type", String)
 ], ActionExecutionMessageInput.prototype, "arguments", void 0);
 _ts_decorate([
-  Field(() => ActionExecutionScope),
-  _ts_metadata("design:type", typeof ActionExecutionScope === "undefined" ? Object : ActionExecutionScope)
+  Field(() => String, {
+    nullable: true
+  }),
+  _ts_metadata("design:type", String)
+], ActionExecutionMessageInput.prototype, "parentMessageId", void 0);
+_ts_decorate([
+  Field(() => String, {
+    nullable: true,
+    deprecationReason: "This field will be removed in a future version"
+  }),
+  _ts_metadata("design:type", typeof String === "undefined" ? Object : String)
 ], ActionExecutionMessageInput.prototype, "scope", void 0);
 ActionExecutionMessageInput = _ts_decorate([
   InputType()
@@ -267,6 +274,7 @@ ActionExecutionMessageInput = _ts_decorate([
 var ResultMessageInput = class {
   actionExecutionId;
   actionName;
+  parentMessageId;
   result;
 };
 __name(ResultMessageInput, "ResultMessageInput");
@@ -278,6 +286,12 @@ _ts_decorate([
   Field(() => String),
   _ts_metadata("design:type", String)
 ], ResultMessageInput.prototype, "actionName", void 0);
+_ts_decorate([
+  Field(() => String, {
+    nullable: true
+  }),
+  _ts_metadata("design:type", String)
+], ResultMessageInput.prototype, "parentMessageId", void 0);
 _ts_decorate([
   Field(() => String),
   _ts_metadata("design:type", String)
@@ -541,6 +555,7 @@ var ForwardedParametersInput = class {
   stop;
   toolChoice;
   toolChoiceFunctionName;
+  temperature;
 };
 __name(ForwardedParametersInput, "ForwardedParametersInput");
 _ts_decorate6([
@@ -575,6 +590,12 @@ _ts_decorate6([
   }),
   _ts_metadata6("design:type", String)
 ], ForwardedParametersInput.prototype, "toolChoiceFunctionName", void 0);
+_ts_decorate6([
+  Field6(() => Number, {
+    nullable: true
+  }),
+  _ts_metadata6("design:type", Number)
+], ForwardedParametersInput.prototype, "temperature", void 0);
 ForwardedParametersInput = _ts_decorate6([
   InputType6()
 ], ForwardedParametersInput);
@@ -893,6 +914,7 @@ BaseMessageOutput = _ts_decorate11([
 var TextMessageOutput = class {
   role;
   content;
+  parentMessageId;
 };
 __name(TextMessageOutput, "TextMessageOutput");
 _ts_decorate11([
@@ -905,6 +927,12 @@ _ts_decorate11([
   ]),
   _ts_metadata11("design:type", Array)
 ], TextMessageOutput.prototype, "content", void 0);
+_ts_decorate11([
+  Field11(() => String, {
+    nullable: true
+  }),
+  _ts_metadata11("design:type", String)
+], TextMessageOutput.prototype, "parentMessageId", void 0);
 TextMessageOutput = _ts_decorate11([
   ObjectType2({
     implements: BaseMessageOutput
@@ -914,6 +942,7 @@ var ActionExecutionMessageOutput = class {
   name;
   scope;
   arguments;
+  parentMessageId;
 };
 __name(ActionExecutionMessageOutput, "ActionExecutionMessageOutput");
 _ts_decorate11([
@@ -921,8 +950,11 @@ _ts_decorate11([
   _ts_metadata11("design:type", String)
 ], ActionExecutionMessageOutput.prototype, "name", void 0);
 _ts_decorate11([
-  Field11(() => ActionExecutionScope),
-  _ts_metadata11("design:type", typeof ActionExecutionScope === "undefined" ? Object : ActionExecutionScope)
+  Field11(() => String, {
+    nullable: true,
+    deprecationReason: "This field will be removed in a future version"
+  }),
+  _ts_metadata11("design:type", String)
 ], ActionExecutionMessageOutput.prototype, "scope", void 0);
 _ts_decorate11([
   Field11(() => [
@@ -930,6 +962,12 @@ _ts_decorate11([
   ]),
   _ts_metadata11("design:type", Array)
 ], ActionExecutionMessageOutput.prototype, "arguments", void 0);
+_ts_decorate11([
+  Field11(() => String, {
+    nullable: true
+  }),
+  _ts_metadata11("design:type", String)
+], ActionExecutionMessageOutput.prototype, "parentMessageId", void 0);
 ActionExecutionMessageOutput = _ts_decorate11([
   ObjectType2({
     implements: BaseMessageOutput
@@ -1041,8 +1079,8 @@ CopilotResponse = _ts_decorate11([
 import { Repeater } from "graphql-yoga";
 
 // src/service-adapters/events.ts
-import { randomId as randomId3 } from "@copilotkit/shared";
-import { of, concat, map, scan as scan2, concatMap, ReplaySubject as ReplaySubject2, firstValueFrom, from as from2 } from "rxjs";
+import { randomId as randomId2 } from "@copilotkit/shared";
+import { of, concat, scan as scan2, concatMap, ReplaySubject as ReplaySubject2, firstValueFrom, from as from2 } from "rxjs";
 
 // src/lib/telemetry-client.ts
 import { TelemetryClient } from "@copilotkit/shared";
@@ -1090,163 +1128,167 @@ var RemoteLangGraphEventSource = class {
     }
     return shouldEmitToolCalls === toolCallName;
   }
+  getCurrentContent(event) {
+    var _a, _b, _c, _d, _e;
+    const content = ((_c = (_b = (_a = event.data) == null ? void 0 : _a.chunk) == null ? void 0 : _b.kwargs) == null ? void 0 : _c.content) ?? ((_e = (_d = event.data) == null ? void 0 : _d.chunk) == null ? void 0 : _e.content);
+    if (!content) {
+      const toolCallChunks = this.getCurrentToolCallChunks(event) ?? [];
+      for (const chunk of toolCallChunks) {
+        if (chunk.args) {
+          return chunk.args;
+        }
+      }
+    }
+    if (typeof content === "string") {
+      return content;
+    } else if (Array.isArray(content) && content.length > 0) {
+      return content[0].text;
+    }
+    return null;
+  }
+  getCurrentMessageId(event) {
+    var _a, _b, _c, _d, _e;
+    return ((_c = (_b = (_a = event.data) == null ? void 0 : _a.chunk) == null ? void 0 : _b.kwargs) == null ? void 0 : _c.id) ?? ((_e = (_d = event.data) == null ? void 0 : _d.chunk) == null ? void 0 : _e.id);
+  }
+  getCurrentToolCallChunks(event) {
+    var _a, _b, _c, _d, _e;
+    return ((_c = (_b = (_a = event.data) == null ? void 0 : _a.chunk) == null ? void 0 : _b.kwargs) == null ? void 0 : _c.tool_call_chunks) ?? ((_e = (_d = event.data) == null ? void 0 : _d.chunk) == null ? void 0 : _e.tool_call_chunks);
+  }
+  getResponseMetadata(event) {
+    var _a, _b, _c, _d, _e;
+    return ((_c = (_b = (_a = event.data) == null ? void 0 : _a.chunk) == null ? void 0 : _b.kwargs) == null ? void 0 : _c.response_metadata) ?? ((_e = (_d = event.data) == null ? void 0 : _d.chunk) == null ? void 0 : _e.response_metadata);
+  }
   processLangGraphEvents() {
     let lastEventWithState = null;
     return this.eventStream$.pipe(scan((acc, event) => {
-      var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n, _o, _p, _q;
       if (event.event === LangGraphEventTypes.OnChatModelStream) {
-        const content = ((_c = (_b = (_a = event.data) == null ? void 0 : _a.chunk) == null ? void 0 : _b.kwargs) == null ? void 0 : _c.content) ?? ((_e = (_d = event.data) == null ? void 0 : _d.chunk) == null ? void 0 : _e.content);
-        if (typeof content === "string") {
-          acc.content = content;
-        } else if (Array.isArray(content) && content.length > 0) {
-          acc.content = content[0].text;
-        } else {
-          acc.content = null;
-        }
-        const toolCallChunks = (
-          // @ts-expect-error -- LangGraph Platform implementation stores data outside of kwargs
-          ((_h = (_g = (_f = event.data) == null ? void 0 : _f.chunk) == null ? void 0 : _g.kwargs) == null ? void 0 : _h.tool_call_chunks) ?? ((_j = (_i = event.data) == null ? void 0 : _i.chunk) == null ? void 0 : _j.tool_call_chunks)
-        );
-        const toolCallMessageId = ((_m = (_l = (_k = event.data) == null ? void 0 : _k.chunk) == null ? void 0 : _l.kwargs) == null ? void 0 : _m.id) ?? ((_o = (_n = event.data) == null ? void 0 : _n.chunk) == null ? void 0 : _o.id);
-        if (toolCallChunks && toolCallChunks.length > 0) {
-          acc.prevToolCallMessageId = acc.toolCallMessageId;
-          acc.toolCallMessageId = toolCallMessageId;
-          if ((_p = toolCallChunks[0]) == null ? void 0 : _p.name) {
-            acc.toolCallName = toolCallChunks[0].name;
-          }
-          if ((_q = toolCallChunks[0]) == null ? void 0 : _q.id) {
-            acc.toolCallId = toolCallChunks[0].id;
-          }
-          acc.prevMessageId = acc.messageId;
-          acc.messageId = toolCallMessageId;
-        } else if (acc.content && acc.content != "") {
-          acc.prevMessageId = acc.messageId;
-          acc.messageId = toolCallMessageId;
-        } else {
-          acc.prevToolCallMessageId = acc.toolCallMessageId;
-          acc.prevMessageId = acc.messageId;
-        }
-      } else {
-        acc.prevToolCallMessageId = acc.toolCallMessageId;
-        acc.toolCallMessageId = null;
-        acc.prevMessageId = acc.messageId;
-        acc.messageId = null;
-        acc.toolCallName = null;
+        const prevMessageId = acc.lastMessageId;
+        acc.currentContent = this.getCurrentContent(event);
+        acc.lastMessageId = this.getCurrentMessageId(event) ?? acc.lastMessageId;
+        const toolCallChunks = this.getCurrentToolCallChunks(event) ?? [];
+        const responseMetadata = this.getResponseMetadata(event);
+        acc.isToolCallStart = toolCallChunks.some((chunk) => chunk.name && chunk.id);
+        acc.isMessageStart = prevMessageId !== acc.lastMessageId && !acc.isToolCallStart;
+        acc.isToolCall = toolCallChunks && toolCallChunks.length > 0;
+        acc.isToolCallEnd = (responseMetadata == null ? void 0 : responseMetadata.finish_reason) === "tool_calls";
+        acc.isMessageEnd = (responseMetadata == null ? void 0 : responseMetadata.finish_reason) === "stop";
+        ({ name: acc.lastToolCallName, id: acc.lastToolCallId } = toolCallChunks.find((chunk) => chunk.name && chunk.id) ?? {
+          name: acc.lastToolCallName,
+          id: acc.lastToolCallId
+        });
       }
       acc.event = event;
       lastEventWithState = acc;
       return acc;
     }, {
       event: null,
-      toolCallId: null,
-      toolCallMessageId: null,
-      prevToolCallMessageId: null,
-      messageId: null,
-      toolCallName: null,
-      prevMessageId: null,
-      content: null
-    }), mergeMap((eventWithState) => {
-      var _a, _b, _c, _d, _e, _f, _g, _h, _i;
+      isMessageStart: false,
+      isMessageEnd: false,
+      isToolCallStart: false,
+      isToolCallEnd: false,
+      isToolCall: false,
+      lastMessageId: null,
+      lastToolCallId: null,
+      lastToolCallName: null,
+      currentContent: null
+    }), mergeMap((acc) => {
       const events = [];
       let shouldEmitMessages = true;
-      let shouldEmitToolCalls = false;
-      if (eventWithState.event.event == LangGraphEventTypes.OnChatModelStream) {
-        if ("copilotkit:emit-tool-calls" in (eventWithState.event.metadata || {})) {
-          shouldEmitToolCalls = eventWithState.event.metadata["copilotkit:emit-tool-calls"];
+      let shouldEmitToolCalls = true;
+      if (acc.event.event == LangGraphEventTypes.OnChatModelStream) {
+        if ("copilotkit:emit-tool-calls" in (acc.event.metadata || {})) {
+          shouldEmitToolCalls = acc.event.metadata["copilotkit:emit-tool-calls"];
         }
-        if ("copilotkit:emit-messages" in (eventWithState.event.metadata || {})) {
-          shouldEmitMessages = eventWithState.event.metadata["copilotkit:emit-messages"];
+        if ("copilotkit:emit-messages" in (acc.event.metadata || {})) {
+          shouldEmitMessages = acc.event.metadata["copilotkit:emit-messages"];
         }
       }
-      if (eventWithState.prevToolCallMessageId !== null && eventWithState.prevToolCallMessageId !== eventWithState.toolCallMessageId && this.shouldEmitToolCall(shouldEmitToolCalls, eventWithState.toolCallName)) {
+      const responseMetadata = this.getResponseMetadata(acc.event);
+      if ((responseMetadata == null ? void 0 : responseMetadata.finish_reason) === "tool_calls" && this.shouldEmitToolCall(shouldEmitToolCalls, acc.lastToolCallName)) {
         events.push({
-          type: RuntimeEventTypes.ActionExecutionEnd
+          type: RuntimeEventTypes.ActionExecutionEnd,
+          actionExecutionId: acc.lastToolCallId
         });
       }
-      if (eventWithState.prevMessageId !== null && eventWithState.prevMessageId !== eventWithState.messageId && shouldEmitMessages) {
+      if ((responseMetadata == null ? void 0 : responseMetadata.finish_reason) === "stop" && shouldEmitMessages) {
         events.push({
-          type: RuntimeEventTypes.TextMessageEnd
+          type: RuntimeEventTypes.TextMessageEnd,
+          messageId: acc.lastMessageId
         });
       }
-      switch (eventWithState.event.event) {
+      switch (acc.event.event) {
         case LangGraphEventTypes.OnCustomEvent:
-          if (eventWithState.event.name === CustomEventNames.CopilotKitManuallyEmitMessage) {
+          if (acc.event.name === CustomEventNames.CopilotKitManuallyEmitMessage) {
             events.push({
               type: RuntimeEventTypes.TextMessageStart,
-              messageId: eventWithState.event.data.message_id
+              messageId: acc.event.data.message_id
             });
             events.push({
               type: RuntimeEventTypes.TextMessageContent,
-              content: eventWithState.event.data.message
+              messageId: acc.event.data.message_id,
+              content: acc.event.data.message
             });
             events.push({
-              type: RuntimeEventTypes.TextMessageEnd
+              type: RuntimeEventTypes.TextMessageEnd,
+              messageId: acc.event.data.message_id
             });
-          } else if (eventWithState.event.name === CustomEventNames.CopilotKitManuallyEmitToolCall) {
+          } else if (acc.event.name === CustomEventNames.CopilotKitManuallyEmitToolCall) {
             events.push({
               type: RuntimeEventTypes.ActionExecutionStart,
-              actionExecutionId: eventWithState.event.data.id,
-              actionName: eventWithState.event.data.name
+              actionExecutionId: acc.event.data.id,
+              actionName: acc.event.data.name,
+              parentMessageId: acc.event.data.id
             });
             events.push({
               type: RuntimeEventTypes.ActionExecutionArgs,
-              args: JSON.stringify(eventWithState.event.data.args)
+              actionExecutionId: acc.event.data.id,
+              args: JSON.stringify(acc.event.data.args)
             });
             events.push({
-              type: RuntimeEventTypes.ActionExecutionEnd
+              type: RuntimeEventTypes.ActionExecutionEnd,
+              actionExecutionId: acc.event.data.id
             });
           }
           break;
         case LangGraphEventTypes.OnCopilotKitStateSync:
           events.push({
             type: RuntimeEventTypes.AgentStateMessage,
-            threadId: eventWithState.event.thread_id,
-            role: eventWithState.event.role,
-            agentName: eventWithState.event.agent_name,
-            nodeName: eventWithState.event.node_name,
-            runId: eventWithState.event.run_id,
-            active: eventWithState.event.active,
-            state: JSON.stringify(eventWithState.event.state),
-            running: eventWithState.event.running
+            threadId: acc.event.thread_id,
+            role: acc.event.role,
+            agentName: acc.event.agent_name,
+            nodeName: acc.event.node_name,
+            runId: acc.event.run_id,
+            active: acc.event.active,
+            state: JSON.stringify(acc.event.state),
+            running: acc.event.running
           });
           break;
-        case LangGraphEventTypes.OnToolEnd:
-          break;
         case LangGraphEventTypes.OnChatModelStream:
-          if (eventWithState.toolCallMessageId !== null && eventWithState.prevToolCallMessageId !== eventWithState.toolCallMessageId) {
-            if (this.shouldEmitToolCall(shouldEmitToolCalls, eventWithState.toolCallName)) {
-              events.push({
-                type: RuntimeEventTypes.ActionExecutionStart,
-                actionExecutionId: eventWithState.toolCallMessageId,
-                actionName: eventWithState.toolCallName,
-                scope: "client"
-              });
-            }
-          } else if (eventWithState.messageId !== null && eventWithState.prevMessageId !== eventWithState.messageId) {
-            if (shouldEmitMessages) {
-              events.push({
-                type: RuntimeEventTypes.TextMessageStart,
-                messageId: eventWithState.messageId
-              });
-            }
+          if (acc.isToolCallStart && this.shouldEmitToolCall(shouldEmitToolCalls, acc.lastToolCallName)) {
+            events.push({
+              type: RuntimeEventTypes.ActionExecutionStart,
+              actionExecutionId: acc.lastToolCallId,
+              actionName: acc.lastToolCallName,
+              parentMessageId: acc.lastMessageId
+            });
+          } else if (acc.isMessageStart && shouldEmitMessages) {
+            events.push({
+              type: RuntimeEventTypes.TextMessageStart,
+              messageId: acc.lastMessageId
+            });
           }
-          const args = ((_e = (_d = (_c = (_b = (_a = eventWithState.event.data) == null ? void 0 : _a.chunk) == null ? void 0 : _b.kwargs) == null ? void 0 : _c.tool_call_chunks) == null ? void 0 : _d[0]) == null ? void 0 : _e.args) ?? // @ts-expect-error -- sdf
-          ((_i = (_h = (_g = (_f = eventWithState.event.data) == null ? void 0 : _f.chunk) == null ? void 0 : _g.tool_call_chunks) == null ? void 0 : _h[0]) == null ? void 0 : _i.args);
-          const content = eventWithState.content;
-          if (args) {
-            if (this.shouldEmitToolCall(shouldEmitToolCalls, eventWithState.toolCallName)) {
-              events.push({
-                type: RuntimeEventTypes.ActionExecutionArgs,
-                args
-              });
-            }
-          } else if (eventWithState.messageId !== null && content) {
-            if (shouldEmitMessages) {
-              events.push({
-                type: RuntimeEventTypes.TextMessageContent,
-                content
-              });
-            }
+          if (acc.isToolCall && acc.currentContent && this.shouldEmitToolCall(shouldEmitToolCalls, acc.lastToolCallName)) {
+            events.push({
+              type: RuntimeEventTypes.ActionExecutionArgs,
+              actionExecutionId: acc.lastToolCallId,
+              args: acc.currentContent
+            });
+          } else if (!acc.isToolCall && acc.currentContent && shouldEmitMessages) {
+            events.push({
+              type: RuntimeEventTypes.TextMessageContent,
+              messageId: acc.lastMessageId,
+              content: acc.currentContent
+            });
           }
           break;
       }
@@ -1254,14 +1296,16 @@ var RemoteLangGraphEventSource = class {
     }), catchError((error) => {
       console.error(error);
       const events = [];
-      if (lastEventWithState == null ? void 0 : lastEventWithState.messageId) {
+      if ((lastEventWithState == null ? void 0 : lastEventWithState.lastMessageId) && !lastEventWithState.isToolCall) {
         events.push({
-          type: RuntimeEventTypes.TextMessageEnd
+          type: RuntimeEventTypes.TextMessageEnd,
+          messageId: lastEventWithState.lastMessageId
         });
       }
-      if (lastEventWithState == null ? void 0 : lastEventWithState.toolCallMessageId) {
+      if (lastEventWithState == null ? void 0 : lastEventWithState.lastToolCallId) {
         events.push({
-          type: RuntimeEventTypes.ActionExecutionEnd
+          type: RuntimeEventTypes.ActionExecutionEnd,
+          actionExecutionId: lastEventWithState.lastToolCallId
         });
       }
       const messageId = randomId();
@@ -1271,10 +1315,12 @@ var RemoteLangGraphEventSource = class {
       });
       events.push({
         type: RuntimeEventTypes.TextMessageContent,
+        messageId,
         content: "\u274C An error occurred. Please try again."
       });
       events.push({
-        type: RuntimeEventTypes.TextMessageEnd
+        type: RuntimeEventTypes.TextMessageEnd,
+        messageId
       });
       return events;
     }));
@@ -1313,8 +1359,6 @@ async function streamEvents(controller, args) {
   if (initialThreadId && initialThreadId.startsWith("ck-")) {
     initialThreadId = initialThreadId.substring(3);
   }
-  const assistants = await client.assistants.search();
-  const retrievedAssistant = assistants.find((a) => a.name === name);
   const threadId = initialThreadId ?? randomUUID();
   if (initialThreadId === threadId) {
     await client.threads.get(threadId);
@@ -1334,7 +1378,7 @@ async function streamEvents(controller, args) {
   const mode = wasInitiatedWithExistingThread && nodeName != "__end__" ? "continue" : "start";
   let formattedMessages = [];
   try {
-    formattedMessages = formatMessages(messages);
+    formattedMessages = copilotkitMessagesToLangChain(messages);
   } catch (e) {
     logger2.error(e, `Error event thrown: ${e.message}`);
   }
@@ -1345,8 +1389,16 @@ async function streamEvents(controller, args) {
       asNode: nodeName
     });
   }
-  const assistantId = initialAssistantId ?? (retrievedAssistant == null ? void 0 : retrievedAssistant.assistant_id);
-  if (!assistantId) {
+  let streamInfo = {
+    hashedLgcKey: createHash("sha256").update(langsmithApiKey).digest("hex")
+  };
+  const assistants = await client.assistants.search();
+  const retrievedAssistant = assistants.find((a) => a.name === name || a.assistant_id === initialAssistantId);
+  if (!retrievedAssistant) {
+    telemetry_client_default.capture("oss.runtime.agent_execution_stream_errored", {
+      ...streamInfo,
+      error: `Found no assistants for given information, while ${assistants.length} assistants exists`
+    });
     console.error(`
       No agent found for the agent name specified in CopilotKit provider
       Please check your available agents or provide an agent ID in the LangGraph Platform endpoint definition.
@@ -1356,6 +1408,7 @@ async function streamEvents(controller, args) {
       `);
     throw new Error("No agent id found");
   }
+  const assistantId = retrievedAssistant.assistant_id;
   const graphInfo = await client.assistants.getGraph(assistantId);
   const streamInput = mode === "start" ? state : null;
   let streamingStateExtractor = new StreamingStateExtractor([]);
@@ -1374,9 +1427,6 @@ async function streamEvents(controller, args) {
   let latestStateValues = {};
   let updatedState = state;
   let manuallyEmittedState = null;
-  let streamInfo = {
-    hashedLgcKey: createHash("sha256").update(langsmithApiKey).digest("hex")
-  };
   try {
     telemetry_client_default.capture("oss.runtime.agent_execution_stream_started", {
       hashedLgcKey: streamInfo.hashedLgcKey
@@ -1389,7 +1439,6 @@ async function streamEvents(controller, args) {
       ].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") {
@@ -1482,21 +1531,34 @@ async function streamEvents(controller, args) {
       nodeName: isEndNode ? "__end__" : nodeName,
       state: state.values,
       running: !shouldExit,
-      active: false
+      active: false,
+      includeMessages: true
     }));
     return Promise.resolve();
   } catch (e) {
+    logger2.error(e);
+    telemetry_client_default.capture("oss.runtime.agent_execution_stream_errored", {
+      ...streamInfo,
+      error: e.message
+    });
     return Promise.resolve();
   }
 }
 __name(streamEvents, "streamEvents");
-function getStateSyncEvent({ threadId, runId, agentName, nodeName, state, running, active }) {
-  const stateWithoutMessages = Object.keys(state).reduce((acc, key) => {
-    if (key !== "messages") {
-      acc[key] = state[key];
-    }
-    return acc;
-  }, {});
+function getStateSyncEvent({ threadId, runId, agentName, nodeName, state, running, active, includeMessages = false }) {
+  if (!includeMessages) {
+    state = Object.keys(state).reduce((acc, key) => {
+      if (key !== "messages") {
+        acc[key] = state[key];
+      }
+      return acc;
+    }, {});
+  } else {
+    state = {
+      ...state,
+      messages: langchainMessagesToCopilotKit(state.messages || [])
+    };
+  }
   return JSON.stringify({
     event: LangGraphEventTypes.OnCopilotKitStateSync,
     thread_id: threadId,
@@ -1504,7 +1566,7 @@ function getStateSyncEvent({ threadId, runId, agentName, nodeName, state, runnin
     agent_name: agentName,
     node_name: nodeName,
     active,
-    state: stateWithoutMessages,
+    state,
     running,
     role: "assistant"
   }) + "\n";
@@ -1580,134 +1642,157 @@ function langGraphDefaultMergeState(state, messages, actions, agentName) {
   if (messages.length > 0 && "role" in messages[0] && messages[0].role === "system") {
     messages = messages.slice(1);
   }
-  const mergedMessages = state.messages || [];
-  const existingMessageIds = new Set(mergedMessages.map((message) => message.id));
-  const existingToolCallResults = /* @__PURE__ */ new Set();
-  for (const message of mergedMessages) {
-    if ("tool_call_id" in message) {
-      existingToolCallResults.add(message.tool_call_id);
+  const existingMessages = state.messages || [];
+  const existingMessageIds = new Set(existingMessages.map((message) => message.id));
+  const newMessages = messages.filter((message) => !existingMessageIds.has(message.id));
+  return {
+    ...state,
+    messages: newMessages,
+    copilotkit: {
+      actions
+    }
+  };
+}
+__name(langGraphDefaultMergeState, "langGraphDefaultMergeState");
+function langchainMessagesToCopilotKit(messages) {
+  const result = [];
+  const tool_call_names = {};
+  for (const message of messages) {
+    if (message.type === "ai") {
+      for (const tool_call of message.tool_calls) {
+        tool_call_names[tool_call.id] = tool_call.name;
+      }
     }
   }
   for (const message of messages) {
-    if ("tool_calls" in message && message.tool_calls.length > 0 && message.tool_calls[0].name === agentName) {
-      continue;
+    let content = message.content;
+    if (content instanceof Array) {
+      content = content[0];
     }
-    if ("name" in message && message.name === agentName) {
-      continue;
+    if (content instanceof Object) {
+      content = content.text;
     }
-    if (!existingMessageIds.has(message.id)) {
-      if ("tool_call_id" in message && existingToolCallResults.has(message.tool_call_id)) {
-        console.warn("Warning: Duplicate tool call result, skipping:", message.tool_call_id);
-        continue;
-      }
-      mergedMessages.push(message);
-    } else {
-      for (let i = 0; i < mergedMessages.length; i++) {
-        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;
+    if (message.type === "human") {
+      result.push({
+        role: "user",
+        content,
+        id: message.id
+      });
+    } else if (message.type === "system") {
+      result.push({
+        role: "system",
+        content,
+        id: message.id
+      });
+    } else if (message.type === "ai") {
+      if (message.tool_calls && message.tool_calls.length > 0) {
+        for (const tool_call of message.tool_calls) {
+          result.push({
+            id: tool_call.id,
+            name: tool_call.name,
+            arguments: tool_call.args,
+            parentMessageId: message.id
+          });
         }
+      } else {
+        result.push({
+          role: "assistant",
+          content,
+          id: message.id,
+          parentMessageId: message.id
+        });
       }
+    } else if (message.type === "tool") {
+      const actionName = tool_call_names[message.tool_call_id] || message.name || "";
+      result.push({
+        actionExecutionId: message.tool_call_id,
+        actionName,
+        result: content,
+        id: message.id
+      });
     }
   }
-  for (let i = 0; i < mergedMessages.length - 1; i++) {
-    const currentMessage = mergedMessages[i];
-    const nextMessage = mergedMessages[i + 1];
-    if ("tool_calls" in currentMessage && currentMessage.tool_calls.length > 0 && "tool_call_id" in nextMessage) {
-      nextMessage.tool_call_id = currentMessage.tool_calls[0].id;
+  const resultsDict = {};
+  for (const msg of result) {
+    if (msg.actionExecutionId) {
+      resultsDict[msg.actionExecutionId] = msg;
     }
   }
-  const correctedMessages = [];
-  for (let i = 0; i < mergedMessages.length; i++) {
-    const currentMessage = mergedMessages[i];
-    const nextMessage = mergedMessages[i + 1] || null;
-    const prevMessage = mergedMessages[i - 1] || null;
-    if ("tool_calls" in currentMessage && currentMessage.tool_calls.length > 0) {
-      if (!nextMessage) {
-        console.warn("No next message to auto-correct tool call, skipping:", currentMessage.tool_calls[0].id);
-        continue;
-      }
-      if (!("tool_call_id" in nextMessage) || nextMessage.tool_call_id !== currentMessage.tool_calls[0].id) {
-        const toolMessage = mergedMessages.find((m) => "tool_call_id" in m && m.tool_call_id === currentMessage.tool_calls[0].id);
-        if (toolMessage) {
-          console.warn("Auto-corrected tool call alignment issue:", currentMessage.tool_calls[0].id);
-          correctedMessages.push(currentMessage, toolMessage);
-          continue;
-        } else {
-          console.warn("No corresponding tool call result found for tool call, skipping:", currentMessage.tool_calls[0].id);
-          continue;
-        }
-      }
-      correctedMessages.push(currentMessage);
-      continue;
+  const reorderedResult = [];
+  for (const msg of result) {
+    if (!("actionExecutionId" in msg)) {
+      reorderedResult.push(msg);
     }
-    if ("tool_call_id" in currentMessage) {
-      if (!prevMessage || !("tool_calls" in prevMessage)) {
-        console.warn("No previous tool call, skipping tool call result:", currentMessage.id);
-        continue;
-      }
-      if (prevMessage.tool_calls && prevMessage.tool_calls[0].id !== currentMessage.tool_call_id) {
-        console.warn("Tool call id is incorrect, skipping tool call result:", currentMessage.id);
-        continue;
+    if ("arguments" in msg) {
+      const msgId = msg.id;
+      if (msgId in resultsDict) {
+        reorderedResult.push(resultsDict[msgId]);
       }
-      correctedMessages.push(currentMessage);
-      continue;
     }
-    correctedMessages.push(currentMessage);
   }
-  return {
-    ...state,
-    messages: correctedMessages,
-    copilotkit: {
-      actions
-    }
-  };
+  return reorderedResult;
 }
-__name(langGraphDefaultMergeState, "langGraphDefaultMergeState");
-function formatMessages(messages) {
-  return messages.map((message) => {
-    if (message.isTextMessage() && message.role === "assistant") {
-      return message;
-    }
-    if (message.isTextMessage() && message.role === "system") {
-      return message;
-    }
-    if (message.isTextMessage() && message.role === "user") {
-      return message;
+__name(langchainMessagesToCopilotKit, "langchainMessagesToCopilotKit");
+function copilotkitMessagesToLangChain(messages) {
+  const result = [];
+  const processedActionExecutions = /* @__PURE__ */ new Set();
+  for (const message of messages) {
+    if (message.isTextMessage()) {
+      if (message.role === "user") {
+        result.push({
+          ...message,
+          role: MessageRole.user
+        });
+      } else if (message.role === "system") {
+        result.push({
+          ...message,
+          role: MessageRole.system
+        });
+      } else if (message.role === "assistant") {
+        result.push({
+          ...message,
+          role: MessageRole.assistant
+        });
+      }
+      continue;
     }
     if (message.isActionExecutionMessage()) {
-      const toolCall = {
-        name: message.name,
-        args: message.arguments,
-        id: message.id
-      };
-      return {
-        type: message.type,
+      const messageId = message.parentMessageId ?? message.id;
+      if (processedActionExecutions.has(messageId)) {
+        continue;
+      }
+      processedActionExecutions.add(messageId);
+      const relatedActionExecutions = messages.filter((m) => m.isActionExecutionMessage() && (m.parentMessageId && m.parentMessageId === messageId || m.id === messageId));
+      const tool_calls = relatedActionExecutions.map((m) => ({
+        name: m.name,
+        args: m.arguments,
+        id: m.id
+      }));
+      result.push({
+        id: messageId,
+        type: "ActionExecutionMessage",
         content: "",
-        tool_calls: [
-          toolCall
-        ],
-        role: MessageRole.assistant,
-        id: message.id
-      };
+        tool_calls,
+        role: MessageRole.assistant
+      });
+      continue;
     }
     if (message.isResultMessage()) {
-      return {
+      result.push({
         type: message.type,
         content: message.result,
         id: message.id,
         tool_call_id: message.actionExecutionId,
         name: message.actionName,
         role: MessageRole.tool
-      };
+      });
+      continue;
     }
     throw new Error(`Unknown message type ${message.type}`);
-  });
+  }
+  return result;
 }
-__name(formatMessages, "formatMessages");
+__name(copilotkitMessagesToLangChain, "copilotkitMessagesToLangChain");
 
 // src/lib/runtime/remote-action-constructors.ts
 function constructLGCRemoteAction({ endpoint, graphqlContext, logger: logger2, messages, agentStates }) {
@@ -1717,7 +1802,7 @@ function constructLGCRemoteAction({ endpoint, graphqlContext, logger: logger2, m
     parameters: [],
     handler: async (_args) => {
     },
-    langGraphAgentHandler: async ({ name, actionInputsWithoutAgents, threadId, nodeName }) => {
+    langGraphAgentHandler: async ({ name, actionInputsWithoutAgents, threadId, nodeName, additionalMessages = [] }) => {
       var _a;
       logger2.debug({
         actionName: agent.name
@@ -1743,7 +1828,10 @@ function constructLGCRemoteAction({ endpoint, graphqlContext, logger: logger2, m
           agent,
           threadId,
           nodeName,
-          messages,
+          messages: [
+            ...messages,
+            ...additionalMessages
+          ],
           state,
           properties: graphqlContext.properties,
           actions: actionInputsWithoutAgents.map((action) => ({
@@ -1771,6 +1859,7 @@ function constructLGCRemoteAction({ endpoint, graphqlContext, logger: logger2, m
 }
 __name(constructLGCRemoteAction, "constructLGCRemoteAction");
 function constructRemoteActions({ json, url, onBeforeRequest, graphqlContext, logger: logger2, messages, agentStates }) {
+  const totalAgents = Array.isArray(json["agents"]) ? json["agents"].length : 0;
   const actions = json["actions"].map((action) => ({
     name: action.name,
     description: action.description,
@@ -1784,7 +1873,7 @@ function constructRemoteActions({ json, url, onBeforeRequest, graphqlContext, lo
       telemetry_client_default.capture("oss.runtime.remote_action_executed", {
         agentExecution: false,
         type: "self-hosted",
-        agentsAmount: json["agents"].length
+        agentsAmount: totalAgents
       });
       try {
         const response = await fetch(`${url}/actions/execute`, {
@@ -1819,13 +1908,13 @@ function constructRemoteActions({ json, url, onBeforeRequest, graphqlContext, lo
       }
     }
   }));
-  const agents = json["agents"].map((agent) => ({
+  const agents = totalAgents ? json["agents"].map((agent) => ({
     name: agent.name,
     description: agent.description,
     parameters: [],
     handler: async (_args) => {
     },
-    langGraphAgentHandler: async ({ name, actionInputsWithoutAgents, threadId, nodeName }) => {
+    langGraphAgentHandler: async ({ name, actionInputsWithoutAgents, threadId, nodeName, additionalMessages = [] }) => {
       var _a;
       logger2.debug({
         actionName: agent.name
@@ -1850,7 +1939,10 @@ function constructRemoteActions({ json, url, onBeforeRequest, graphqlContext, lo
           name,
           threadId,
           nodeName,
-          messages,
+          messages: [
+            ...messages,
+            ...additionalMessages
+          ],
           state,
           properties: graphqlContext.properties,
           actions: actionInputsWithoutAgents.map((action) => ({
@@ -1872,7 +1964,7 @@ function constructRemoteActions({ json, url, onBeforeRequest, graphqlContext, lo
       streamResponse(response.body, eventSource.eventStream$);
       return eventSource.processLangGraphEvents();
     }
-  }));
+  })) : [];
   return [
     ...actions,
     ...agents
@@ -2050,7 +2142,7 @@ __name(setupRemoteActions, "setupRemoteActions");
 import { createHash as createHash3 } from "crypto";
 
 // src/lib/runtime/copilot-runtime.ts
-import { actionParametersToJsonSchema, randomId as randomId2 } from "@copilotkit/shared";
+import { actionParametersToJsonSchema } from "@copilotkit/shared";
 
 // src/service-adapters/conversion.ts
 import { plainToInstance } from "class-transformer";
@@ -2062,7 +2154,8 @@ function convertGqlInputToMessages(inputMessages) {
         id: message.id,
         createdAt: message.createdAt,
         role: message.textMessage.role,
-        content: message.textMessage.content
+        content: message.textMessage.content,
+        parentMessageId: message.textMessage.parentMessageId
       }));
     } else if (message.actionExecutionMessage) {
       messages.push(plainToInstance(ActionExecutionMessage, {
@@ -2070,7 +2163,7 @@ function convertGqlInputToMessages(inputMessages) {
         createdAt: message.createdAt,
         name: message.actionExecutionMessage.name,
         arguments: JSON.parse(message.actionExecutionMessage.arguments),
-        scope: message.actionExecutionMessage.scope
+        parentMessageId: message.actionExecutionMessage.parentMessageId
       }));
     } else if (message.resultMessage) {
       messages.push(plainToInstance(ResultMessage, {
@@ -2181,13 +2274,7 @@ var CopilotRuntime = class {
     } catch (error) {
       console.error("Error getting response:", error);
       eventSource.sendErrorMessageToChat();
-      return {
-        threadId: threadId || randomId2(),
-        runId: runId || randomId2(),
-        eventSource,
-        serverSideActions: [],
-        actionInputsWithoutAgents: []
-      };
+      throw error;
     }
   }
   async processAgentRequest(request) {
@@ -2373,6 +2460,7 @@ __name(getRuntimeInstanceTelemetryInfo, "getRuntimeInstanceTelemetryInfo");
 var telemetry_client_default = telemetryClient;
 
 // src/service-adapters/events.ts
+import { plainToInstance as plainToInstance2 } from "class-transformer";
 var RuntimeEventTypes;
 (function(RuntimeEventTypes2) {
   RuntimeEventTypes2["TextMessageStart"] = "TextMessageStart";
@@ -2388,52 +2476,74 @@ var RuntimeEventSubject = class extends ReplaySubject2 {
   constructor() {
     super();
   }
-  sendTextMessageStart(messageId) {
+  sendTextMessageStart({ messageId, parentMessageId }) {
     this.next({
       type: "TextMessageStart",
-      messageId
+      messageId,
+      parentMessageId
     });
   }
-  sendTextMessageContent(content) {
+  sendTextMessageContent({ messageId, content }) {
     this.next({
       type: "TextMessageContent",
-      content
+      content,
+      messageId
     });
   }
-  sendTextMessageEnd() {
+  sendTextMessageEnd({ messageId }) {
     this.next({
-      type: "TextMessageEnd"
+      type: "TextMessageEnd",
+      messageId
     });
   }
   sendTextMessage(messageId, content) {
-    this.sendTextMessageStart(messageId);
-    this.sendTextMessageContent(content);
-    this.sendTextMessageEnd();
+    this.sendTextMessageStart({
+      messageId
+    });
+    this.sendTextMessageContent({
+      messageId,
+      content
+    });
+    this.sendTextMessageEnd({
+      messageId
+    });
   }
-  sendActionExecutionStart(actionExecutionId, actionName) {
+  sendActionExecutionStart({ actionExecutionId, actionName, parentMessageId }) {
     this.next({
       type: "ActionExecutionStart",
       actionExecutionId,
-      actionName
+      actionName,
+      parentMessageId
     });
   }
-  sendActionExecutionArgs(args) {
+  sendActionExecutionArgs({ actionExecutionId, args }) {
     this.next({
       type: "ActionExecutionArgs",
-      args
+      args,
+      actionExecutionId
     });
   }
-  sendActionExecutionEnd() {
+  sendActionExecutionEnd({ actionExecutionId }) {
     this.next({
-      type: "ActionExecutionEnd"
+      type: "ActionExecutionEnd",
+      actionExecutionId
     });
   }
-  sendActionExecution(actionExecutionId, toolName, args) {
-    this.sendActionExecutionStart(actionExecutionId, toolName);
-    this.sendActionExecutionArgs(args);
-    this.sendActionExecutionEnd();
+  sendActionExecution({ actionExecutionId, actionName, args, parentMessageId }) {
+    this.sendActionExecutionStart({
+      actionExecutionId,
+      actionName,
+      parentMessageId
+    });
+    this.sendActionExecutionArgs({
+      actionExecutionId,
+      args
+    });
+    this.sendActionExecutionEnd({
+      actionExecutionId
+    });
   }
-  sendActionExecutionResult(actionExecutionId, actionName, result) {
+  sendActionExecutionResult({ actionExecutionId, actionName, result }) {
     this.next({
       type: "ActionExecutionResult",
       actionName,
@@ -2441,7 +2551,7 @@ var RuntimeEventSubject = class extends ReplaySubject2 {
       result
     });
   }
-  sendAgentStateMessage(threadId, agentName, nodeName, runId, active, role, state, running) {
+  sendAgentStateMessage({ threadId, agentName, nodeName, runId, active, role, state, running }) {
     this.next({
       type: "AgentStateMessage",
       threadId,
@@ -2462,14 +2572,14 @@ var RuntimeEventSource = class {
   async stream(callback) {
     this.callback = callback;
   }
-  sendErrorMessageToChat() {
-    const errorMessage = "\u274C An error occurred. Please try again.";
+  sendErrorMessageToChat(message = "An error occurred. Please try again.") {
+    const errorMessage = `\u274C ${message}`;
     if (!this.callback) {
       this.stream(async (eventStream$) => {
-        eventStream$.sendTextMessage(randomId3(), errorMessage);
+        eventStream$.sendTextMessage(randomId2(), errorMessage);
       });
     } else {
-      this.eventStream$.sendTextMessage(randomId3(), errorMessage);
+      this.eventStream$.sendTextMessage(randomId2(), errorMessage);
     }
   }
   processRuntimeEvents({ serverSideActions, guardrailsResult$, actionInputsWithoutAgents }) {
@@ -2478,27 +2588,19 @@ var RuntimeEventSource = class {
       this.sendErrorMessageToChat();
     });
     return this.eventStream$.pipe(
-      // mark tools for server side execution
-      map((event) => {
-        if (event.type === "ActionExecutionStart") {
-          if (event.scope !== "passThrough") {
-            event.scope = serverSideActions.find((action) => action.name === event.actionName) ? "server" : "client";
-          }
-        }
-        return event;
-      }),
       // track state
       scan2((acc, event) => {
         acc = {
           ...acc
         };
         if (event.type === "ActionExecutionStart") {
-          acc.callActionServerSide = event.scope === "server";
+          acc.callActionServerSide = serverSideActions.find((action) => action.name === event.actionName) !== void 0;
           acc.args = "";
           acc.actionExecutionId = event.actionExecutionId;
           if (acc.callActionServerSide) {
             acc.action = serverSideActions.find((action) => action.name === event.actionName);
           }
+          acc.actionExecutionParentMessageId = event.parentMessageId;
         } else if (event.type === "ActionExecutionArgs") {
           acc.args += event.args;
         }
@@ -2509,12 +2611,13 @@ var RuntimeEventSource = class {
         callActionServerSide: false,
         args: "",
         actionExecutionId: null,
-        action: null
+        action: null,
+        actionExecutionParentMessageId: null
       }),
       concatMap((eventWithState) => {
         if (eventWithState.event.type === "ActionExecutionEnd" && eventWithState.callActionServerSide) {
           const toolCallEventStream$ = new RuntimeEventSubject();
-          executeAction(toolCallEventStream$, guardrailsResult$ ? guardrailsResult$ : null, eventWithState.action, eventWithState.args, eventWithState.actionExecutionId, actionInputsWithoutAgents).catch((error) => {
+          executeAction(toolCallEventStream$, guardrailsResult$ ? guardrailsResult$ : null, eventWithState.action, eventWithState.args, eventWithState.actionExecutionParentMessageId, eventWithState.actionExecutionId, actionInputsWithoutAgents).catch((error) => {
             console.error(error);
           });
           telemetry_client_default.capture("oss.runtime.server_action_executed", {});
@@ -2527,7 +2630,7 @@ var RuntimeEventSource = class {
   }
 };
 __name(RuntimeEventSource, "RuntimeEventSource");
-async function executeAction(eventStream$, guardrailsResult$, action, actionArguments, actionExecutionId, actionInputsWithoutAgents) {
+async function executeAction(eventStream$, guardrailsResult$, action, actionArguments, actionExecutionParentMessageId, actionExecutionId, actionInputsWithoutAgents) {
   var _a;
   if (guardrailsResult$) {
     const { status } = await firstValueFrom(guardrailsResult$);
@@ -2547,10 +2650,33 @@ async function executeAction(eventStream$, guardrailsResult$, action, actionArgu
     }
   }
   if (isLangGraphAgentAction(action)) {
-    eventStream$.sendActionExecutionResult(actionExecutionId, action.name, `${action.name} agent started`);
+    const result = `${action.name} agent started`;
+    const agentExecution = plainToInstance2(ActionExecutionMessage, {
+      id: actionExecutionId,
+      createdAt: /* @__PURE__ */ new Date(),
+      name: action.name,
+      arguments: JSON.parse(actionArguments),
+      parentMessageId: actionExecutionParentMessageId ?? actionExecutionId
+    });
+    const agentExecutionResult = plainToInstance2(ResultMessage, {
+      id: "result-" + actionExecutionId,
+      createdAt: /* @__PURE__ */ new Date(),
+      actionExecutionId,
+      actionName: action.name,
+      result
+    });
+    eventStream$.sendActionExecutionResult({
+      actionExecutionId,
+      actionName: action.name,
+      result
+    });
     const stream = await action.langGraphAgentHandler({
       name: action.name,
-      actionInputsWithoutAgents
+      actionInputsWithoutAgents,
+      additionalMessages: [
+        agentExecution,
+        agentExecutionResult
+      ]
     });
     from2(stream).subscribe({
       next: (event) => eventStream$.next(event),
@@ -2573,9 +2699,9 @@ __name(executeAction, "executeAction");
 
 // src/graphql/resolvers/copilot.resolver.ts
 import { GraphQLJSONObject } from "graphql-scalars";
-import { plainToInstance as plainToInstance2 } from "class-transformer";
+import { plainToInstance as plainToInstance3 } from "class-transformer";
 import { GraphQLError } from "graphql";
-import { randomId as randomId4 } from "@copilotkit/shared";
+import { randomId as randomId3 } from "@copilotkit/shared";
 function _ts_decorate12(decorators, target, key, desc) {
   var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
   if (typeof Reflect === "object" && typeof Reflect.decorate === "function")
@@ -2692,7 +2818,7 @@ var CopilotResolver = class {
       rejectOutputMessagesPromise = reject;
     });
     logger2.debug("Processing");
-    const { eventSource, threadId = randomId4(), runId, serverSideActions, actionInputsWithoutAgents } = await copilotRuntime.processRuntimeRequest({
+    const { eventSource, threadId = randomId3(), runId, serverSideActions, actionInputsWithoutAgents } = await copilotRuntime.processRuntimeRequest({
       serviceAdapter,
       messages: data.messages,
       actions: data.frontend.actions.filter((action) => action.available !== ActionInputAvailability.disabled),
@@ -2736,8 +2862,8 @@ var CopilotResolver = class {
                   reason: `Interrupted due to Guardrails validation failure. Reason: ${result.reason}`
                 });
                 outputMessages = [
-                  plainToInstance2(TextMessage, {
-                    id: randomId4(),
+                  plainToInstance3(TextMessage, {
+                    id: randomId3(),
                     createdAt: /* @__PURE__ */ new Date(),
                     content: result.reason,
                     role: MessageRole.assistant
@@ -2785,12 +2911,15 @@ var CopilotResolver = class {
                   // skip until this message start event
                   skipWhile((e) => e !== event),
                   // take until the message end event
-                  takeWhile((e) => e.type != RuntimeEventTypes.TextMessageEnd)
+                  takeWhile((e) => !(e.type === RuntimeEventTypes.TextMessageEnd && e.messageId == event.messageId)),
+                  // filter out any other message events or message ids
+                  filter((e) => e.type == RuntimeEventTypes.TextMessageContent && e.messageId == event.messageId)
                 );
                 const streamingTextStatus = new Subject();
                 const messageId = event.messageId;
                 pushMessage({
                   id: messageId,
+                  parentMessageId: event.parentMessageId,
                   status: firstValueFrom2(streamingTextStatus),
                   createdAt: /* @__PURE__ */ new Date(),
                   role: MessageRole.assistant,
@@ -2803,7 +2932,7 @@ var CopilotResolver = class {
                         reason,
                         messageId: messageId2
                       }, "Text streaming interrupted");
-                      streamingTextStatus.next(plainToInstance2(FailedMessageStatus, {
+                      streamingTextStatus.next(plainToInstance3(FailedMessageStatus, {
                         reason
                       }));
                       responseStatus$.next(new MessageStreamInterruptedResponse({
@@ -2836,7 +2965,7 @@ var CopilotResolver = class {
                         streamingTextStatus.next(new SuccessMessageStatus());
                         stopStreamingText();
                         textSubscription == null ? void 0 : textSubscription.unsubscribe();
-                        outputMessages.push(plainToInstance2(TextMessage, {
+                        outputMessages.push(plainToInstance3(TextMessage, {
                           id: messageId,
                           createdAt: /* @__PURE__ */ new Date(),
                           content: textChunks.join(""),
@@ -2849,14 +2978,20 @@ var CopilotResolver = class {
                 break;
               case RuntimeEventTypes.ActionExecutionStart:
                 logger2.debug("Action execution start event received");
-                const actionExecutionArgumentStream = eventStream.pipe(skipWhile((e) => e !== event), takeWhile((e) => e.type != RuntimeEventTypes.ActionExecutionEnd));
+                const actionExecutionArgumentStream = eventStream.pipe(
+                  skipWhile((e) => e !== event),
+                  // take until the action execution end event
+                  takeWhile((e) => !(e.type === RuntimeEventTypes.ActionExecutionEnd && e.actionExecutionId == event.actionExecutionId)),
+                  // filter out any other action execution events or action execution ids
+                  filter((e) => e.type == RuntimeEventTypes.ActionExecutionArgs && e.actionExecutionId == event.actionExecutionId)
+                );
                 const streamingArgumentsStatus = new Subject();
                 pushMessage({
                   id: event.actionExecutionId,
+                  parentMessageId: event.parentMessageId,
                   status: firstValueFrom2(streamingArgumentsStatus),
                   createdAt: /* @__PURE__ */ new Date(),
                   name: event.actionName,
-                  scope: event.scope,
                   arguments: new Repeater(async (pushArgumentsChunk, stopStreamingArguments) => {
                     logger2.debug("Action execution argument stream created");
                     const argumentChunks = [];
@@ -2872,7 +3007,7 @@ var CopilotResolver = class {
                         logger2.error({
                           err
                         }, "Error in action execution argument stream");
-                        streamingArgumentsStatus.next(plainToInstance2(FailedMessageStatus, {
+                        streamingArgumentsStatus.next(plainToInstance3(FailedMessageStatus, {
                           reason: "An unknown error has occurred in the action execution argument stream"
                         }));
                         stopStreamingArguments();
@@ -2883,11 +3018,10 @@ var CopilotResolver = class {
                         streamingArgumentsStatus.next(new SuccessMessageStatus());
                         stopStreamingArguments();
                         actionExecutionArgumentSubscription == null ? void 0 : actionExecutionArgumentSubscription.unsubscribe();
-                        outputMessages.push(plainToInstance2(ActionExecutionMessage, {
+                        outputMessages.push(plainToInstance3(ActionExecutionMessage, {
                           id: event.actionExecutionId,
                           createdAt: /* @__PURE__ */ new Date(),
                           name: event.actionName,
-                          scope: event.scope,
                           arguments: argumentChunks.join("")
                         }));
                       }
@@ -2900,15 +3034,15 @@ var CopilotResolver = class {
                   result: event.result
                 }, "Action execution result event received");
                 pushMessage({
-                  id: randomId4(),
+                  id: "result-" + event.actionExecutionId,
                   status: new SuccessMessageStatus(),
                   createdAt: /* @__PURE__ */ new Date(),
                   actionExecutionId: event.actionExecutionId,
                   actionName: event.actionName,
                   result: event.result
                 });
-                outputMessages.push(plainToInstance2(ResultMessage, {
-                  id: randomId4(),
+                outputMessages.push(plainToInstance3(ResultMessage, {
+                  id: "result-" + event.actionExecutionId,
                   createdAt: /* @__PURE__ */ new Date(),
                   actionExecutionId: event.actionExecutionId,
                   actionName: event.actionName,
@@ -2920,7 +3054,7 @@ var CopilotResolver = class {
                   event
                 }, "Agent message event received");
                 pushMessage({
-                  id: randomId4(),
+                  id: randomId3(),
                   status: new SuccessMessageStatus(),
                   threadId: event.threadId,
                   agentName: event.agentName,
@@ -2932,8 +3066,8 @@ var CopilotResolver = class {
                   role: MessageRole.assistant,
                   createdAt: /* @__PURE__ */ new Date()
                 });
-                outputMessages.push(plainToInstance2(AgentStateMessage, {
-                  id: randomId4(),
+                outputMessages.push(plainToInstance3(AgentStateMessage, {
+                  id: randomId3(),
                   threadId: event.threadId,
                   agentName: event.agentName,
                   nodeName: event.nodeName,
@@ -3144,4 +3278,4 @@ export {
   getCommonConfig,
   copilotRuntimeNodeHttpEndpoint
 };
-//# sourceMappingURL=chunk-K67A6XOJ.mjs.map
+//# sourceMappingURL=chunk-3ECBC2K2.mjs.map