@copilotkit/runtime 1.2.0 → 1.2.2-feat-runtime-remote-actions.0

package/dist/index.js

@@ -12,11 +12,11 @@ var __export = (target, all) => {
   for (var name in all)
     __defProp(target, name, { get: all[name], enumerable: true });
 };
-var __copyProps = (to, from, except, desc) => {
-  if (from && typeof from === "object" || typeof from === "function") {
-    for (let key of __getOwnPropNames(from))
+var __copyProps = (to, from3, except, desc) => {
+  if (from3 && typeof from3 === "object" || typeof from3 === "function") {
+    for (let key of __getOwnPropNames(from3))
       if (!__hasOwnProp.call(to, key) && key !== except)
-        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+        __defProp(to, key, { get: () => from3[key], enumerable: !(desc = __getOwnPropDesc(from3, key)) || desc.enumerable });
   }
   return to;
 };
@@ -44,7 +44,7 @@ var require_package = __commonJS({
       publishConfig: {
         access: "public"
       },
-      version: "1.2.0",
+      version: "1.2.2-feat-runtime-remote-actions.0",
       sideEffects: false,
       main: "./dist/index.js",
       module: "./dist/index.mjs",
@@ -59,7 +59,9 @@ var require_package = __commonJS({
         test: "jest --passWithNoTests",
         "check-types": "tsc --noEmit",
         clean: "rm -rf .turbo && rm -rf node_modules && rm -rf dist && rm -rf .next && rm -rf __snapshots__",
-        "generate-graphql-schema": "rm -rf __snapshots__ && ts-node ./scripts/generate-gql-schema.ts"
+        "generate-graphql-schema": "rm -rf __snapshots__ && ts-node ./scripts/generate-gql-schema.ts",
+        "link:global": "pnpm link --global",
+        "unlink:global": "pnpm unlink --global"
       },
       devDependencies: {
         "@swc/core": "1.5.28",
@@ -142,8 +144,8 @@ __export(src_exports, {
 module.exports = __toCommonJS(src_exports);
 var import_reflect_metadata = require("reflect-metadata");
 
-// src/lib/copilot-runtime.ts
-var import_shared7 = require("@copilotkit/shared");
+// src/lib/runtime/copilot-runtime.ts
+var import_shared8 = require("@copilotkit/shared");
 
 // src/service-adapters/openai/openai-adapter.ts
 var import_openai = __toESM(require("openai"));
@@ -201,6 +203,17 @@ var ResultMessage = class extends BaseMessage {
   result;
 };
 __name(ResultMessage, "ResultMessage");
+var AgentStateMessage = class extends BaseMessage {
+  threadId;
+  agentName;
+  nodeName;
+  runId;
+  active;
+  role;
+  state;
+  running;
+};
+__name(AgentStateMessage, "AgentStateMessage");
 
 // src/service-adapters/openai/utils.ts
 function limitMessagesToTokenCount(messages, tools, model, maxTokens) {
@@ -1270,7 +1283,366 @@ var GroqAdapter = class {
 __name(GroqAdapter, "GroqAdapter");
 
 // src/service-adapters/events.ts
+var import_rxjs2 = require("rxjs");
+
+// src/lib/telemetry-client.ts
+var import_shared7 = require("@copilotkit/shared");
+var packageJson = require_package();
+var telemetryClient = new import_shared7.TelemetryClient({
+  packageName: packageJson.name,
+  packageVersion: packageJson.version
+});
+var telemetry_client_default = telemetryClient;
+
+// src/agents/langgraph/event-source.ts
 var import_rxjs = require("rxjs");
+
+// src/agents/langgraph/events.ts
+var LangGraphEventTypes;
+(function(LangGraphEventTypes2) {
+  LangGraphEventTypes2["OnChainStart"] = "on_chain_start";
+  LangGraphEventTypes2["OnChainStream"] = "on_chain_stream";
+  LangGraphEventTypes2["OnChainEnd"] = "on_chain_end";
+  LangGraphEventTypes2["OnChatModelStart"] = "on_chat_model_start";
+  LangGraphEventTypes2["OnChatModelStream"] = "on_chat_model_stream";
+  LangGraphEventTypes2["OnChatModelEnd"] = "on_chat_model_end";
+  LangGraphEventTypes2["OnToolStart"] = "on_tool_start";
+  LangGraphEventTypes2["OnToolEnd"] = "on_tool_end";
+  LangGraphEventTypes2["OnCopilotKitStateSync"] = "on_copilotkit_state_sync";
+})(LangGraphEventTypes || (LangGraphEventTypes = {}));
+
+// src/agents/langgraph/event-source.ts
+var RemoteLangGraphEventSource = class {
+  eventStream$ = new import_rxjs.ReplaySubject();
+  async streamResponse(response) {
+    const reader = response.body.getReader();
+    const decoder = new TextDecoder();
+    let buffer = [];
+    const eventStream$ = this.eventStream$;
+    function flushBuffer() {
+      const currentBuffer = buffer.join("");
+      if (currentBuffer.trim().length === 0) {
+        return;
+      }
+      const parts = currentBuffer.split("\n");
+      if (parts.length === 0) {
+        return;
+      }
+      const lastPartIsComplete = currentBuffer.endsWith("\n");
+      buffer = [];
+      if (!lastPartIsComplete) {
+        buffer.push(parts.pop());
+      }
+      parts.map((part) => part.trim()).filter((part) => part != "").forEach((part) => {
+        eventStream$.next(JSON.parse(part));
+      });
+    }
+    __name(flushBuffer, "flushBuffer");
+    while (true) {
+      const { done, value } = await reader.read();
+      if (!done) {
+        buffer.push(decoder.decode(value, {
+          stream: true
+        }));
+      }
+      flushBuffer();
+      if (done) {
+        break;
+      }
+    }
+    eventStream$.complete();
+  }
+  processLangGraphEvents() {
+    return this.eventStream$.pipe((0, import_rxjs.scan)((acc, event) => {
+      var _a, _b, _c, _d, _e, _f, _g, _h;
+      if (event.event === LangGraphEventTypes.OnChatModelStream) {
+        if ((_c = (_b = (_a = event.data) == null ? void 0 : _a.chunk) == null ? void 0 : _b.kwargs) == null ? void 0 : _c.tool_call_chunks) {
+          acc.prevToolCallId = acc.toolCallId;
+          acc.toolCallId = (_d = event.data.chunk.kwargs) == null ? void 0 : _d.id;
+          if ((_e = event.data.chunk.kwargs.tool_call_chunks[0]) == null ? void 0 : _e.name) {
+            acc.toolCallName = event.data.chunk.kwargs.tool_call_chunks[0].name;
+          }
+        }
+        acc.prevMessageId = acc.messageId;
+        acc.messageId = (_h = (_g = (_f = event.data) == null ? void 0 : _f.chunk) == null ? void 0 : _g.kwargs) == null ? void 0 : _h.id;
+      } else {
+        acc.prevToolCallId = acc.toolCallId;
+        acc.toolCallId = null;
+        acc.prevMessageId = acc.messageId;
+        acc.messageId = null;
+        acc.toolCallName = null;
+      }
+      acc.event = event;
+      return acc;
+    }, {
+      event: null,
+      toolCallId: null,
+      prevToolCallId: null,
+      messageId: null,
+      toolCallName: null,
+      prevMessageId: null
+    }), (0, import_rxjs.mergeMap)((eventWithState) => {
+      var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j;
+      const events = [];
+      let shouldEmitMessages = false;
+      let shouldEmitToolCalls = false;
+      if (eventWithState.event.event == LangGraphEventTypes.OnChatModelStream) {
+        if ((_a = eventWithState.event.tags) == null ? void 0 : _a.includes("copilotkit:emit-tool-calls")) {
+          shouldEmitToolCalls = true;
+        }
+        if ((_b = eventWithState.event.tags) == null ? void 0 : _b.includes("copilotkit:emit-messages")) {
+          shouldEmitMessages = true;
+        }
+      }
+      if (eventWithState.prevToolCallId !== null && eventWithState.prevToolCallId !== eventWithState.toolCallId && shouldEmitToolCalls) {
+        events.push({
+          type: RuntimeEventTypes.ActionExecutionEnd
+        });
+      }
+      if (eventWithState.prevMessageId !== null && eventWithState.prevMessageId !== eventWithState.messageId && shouldEmitMessages) {
+        events.push({
+          type: RuntimeEventTypes.TextMessageEnd
+        });
+      }
+      switch (eventWithState.event.event) {
+        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
+          });
+          break;
+        case LangGraphEventTypes.OnToolEnd:
+          break;
+        case LangGraphEventTypes.OnChatModelStream:
+          if (eventWithState.toolCallId !== null && eventWithState.prevToolCallId !== eventWithState.toolCallId) {
+            if (shouldEmitToolCalls) {
+              events.push({
+                type: RuntimeEventTypes.ActionExecutionStart,
+                actionExecutionId: eventWithState.toolCallId,
+                actionName: eventWithState.toolCallName,
+                scope: "client"
+              });
+            }
+          } else if (eventWithState.messageId !== null && eventWithState.prevMessageId !== eventWithState.messageId) {
+            if (shouldEmitMessages) {
+              events.push({
+                type: RuntimeEventTypes.TextMessageStart,
+                messageId: eventWithState.messageId
+              });
+            }
+          }
+          const args = (_g = (_f = (_e = (_d = (_c = eventWithState.event.data) == null ? void 0 : _c.chunk) == null ? void 0 : _d.kwargs) == null ? void 0 : _e.tool_call_chunks) == null ? void 0 : _f[0]) == null ? void 0 : _g.args;
+          const content = (_j = (_i = (_h = eventWithState.event.data) == null ? void 0 : _h.chunk) == null ? void 0 : _i.kwargs) == null ? void 0 : _j.content;
+          if (args) {
+            if (shouldEmitToolCalls) {
+              events.push({
+                type: RuntimeEventTypes.ActionExecutionArgs,
+                args
+              });
+            }
+          } else if (eventWithState.messageId !== null && content) {
+            if (shouldEmitMessages) {
+              events.push({
+                type: RuntimeEventTypes.TextMessageContent,
+                content
+              });
+            }
+          }
+          break;
+      }
+      return events;
+    }));
+  }
+};
+__name(RemoteLangGraphEventSource, "RemoteLangGraphEventSource");
+
+// src/lib/runtime/remote-actions.ts
+function isLangGraphAgentAction(action) {
+  if (!action) {
+    return false;
+  }
+  return typeof action.langGraphAgentHandler === "function";
+}
+__name(isLangGraphAgentAction, "isLangGraphAgentAction");
+function createHeaders(onBeforeRequest, graphqlContext) {
+  const headers = {
+    "Content-Type": "application/json"
+  };
+  if (onBeforeRequest) {
+    const { headers: additionalHeaders } = onBeforeRequest({
+      ctx: graphqlContext
+    });
+    if (additionalHeaders) {
+      Object.assign(headers, additionalHeaders);
+    }
+  }
+  return headers;
+}
+__name(createHeaders, "createHeaders");
+async function fetchRemoteInfo({ url, onBeforeRequest, graphqlContext, logger: logger2 }) {
+  logger2.debug({
+    url
+  }, "Fetching actions from url");
+  const headers = createHeaders(onBeforeRequest, graphqlContext);
+  const response = await fetch(`${url}/info`, {
+    method: "POST",
+    headers,
+    body: JSON.stringify({
+      properties: graphqlContext.properties
+    })
+  });
+  if (!response.ok) {
+    logger2.error({
+      url,
+      status: response.status,
+      body: await response.text()
+    }, "Failed to fetch actions from url");
+    return [];
+  }
+  const json = await response.json();
+  logger2.debug({
+    json
+  }, "Fetched actions from url");
+  return json;
+}
+__name(fetchRemoteInfo, "fetchRemoteInfo");
+function constructRemoteActions({ json, url, onBeforeRequest, graphqlContext, logger: logger2, messages, agentStates }) {
+  const actions = json["actions"].map((action) => ({
+    name: action.name,
+    description: action.description,
+    parameters: action.parameters,
+    handler: async (args) => {
+      logger2.debug({
+        actionName: action.name,
+        args
+      }, "Executing remote action");
+      const headers = createHeaders(onBeforeRequest, graphqlContext);
+      telemetry_client_default.capture("oss.runtime.remote_action_executed", {});
+      const response = await fetch(`${url}/actions/execute`, {
+        method: "POST",
+        headers,
+        body: JSON.stringify({
+          name: action.name,
+          arguments: args,
+          properties: graphqlContext.properties
+        })
+      });
+      if (!response.ok) {
+        logger2.error({
+          url,
+          status: response.status,
+          body: await response.text()
+        }, "Failed to execute remote action");
+        return "Failed to execute remote action";
+      }
+      const requestResult = await response.json();
+      const result = requestResult["result"];
+      logger2.debug({
+        actionName: action.name,
+        result
+      }, "Executed remote action");
+      return result;
+    }
+  }));
+  const agents = json["agents"].map((agent) => ({
+    name: agent.name,
+    description: agent.description,
+    parameters: [],
+    handler: async (_args) => {
+    },
+    langGraphAgentHandler: async ({ name, actionInputsWithoutAgents, threadId, nodeName }) => {
+      var _a;
+      logger2.debug({
+        actionName: agent.name
+      }, "Executing remote agent");
+      const headers = createHeaders(onBeforeRequest, graphqlContext);
+      telemetry_client_default.capture("oss.runtime.remote_action_executed", {});
+      let state = {};
+      if (agentStates) {
+        const jsonState = (_a = agentStates.find((state2) => state2.agentName === name)) == null ? void 0 : _a.state;
+        if (jsonState) {
+          state = JSON.parse(jsonState);
+        }
+      }
+      const response = await fetch(`${url}/agents/execute`, {
+        method: "POST",
+        headers,
+        body: JSON.stringify({
+          name,
+          threadId,
+          nodeName,
+          messages,
+          state,
+          properties: graphqlContext.properties,
+          actions: actionInputsWithoutAgents.map((action) => ({
+            name: action.name,
+            description: action.description,
+            parameters: JSON.parse(action.jsonSchema)
+          }))
+        })
+      });
+      if (!response.ok) {
+        logger2.error({
+          url,
+          status: response.status,
+          body: await response.text()
+        }, "Failed to execute remote agent");
+        throw new Error("Failed to execute remote agent");
+      }
+      const eventSource = new RemoteLangGraphEventSource();
+      eventSource.streamResponse(response);
+      return eventSource.processLangGraphEvents();
+    }
+  }));
+  return [
+    ...actions,
+    ...agents
+  ];
+}
+__name(constructRemoteActions, "constructRemoteActions");
+async function setupRemoteActions({ remoteActionDefinitions, graphqlContext, messages, agentStates }) {
+  const logger2 = graphqlContext.logger.child({
+    component: "remote-actions.fetchRemoteActions"
+  });
+  logger2.debug({
+    remoteActionDefinitions
+  }, "Fetching remote actions");
+  const filtered = remoteActionDefinitions.filter((value, index, self) => index === self.findIndex((t) => t.url === value.url));
+  const result = await Promise.all(filtered.map(async (actionDefinition) => {
+    const json = await fetchRemoteInfo({
+      url: actionDefinition.url,
+      onBeforeRequest: actionDefinition.onBeforeRequest,
+      graphqlContext,
+      logger: logger2.child({
+        component: "remote-actions.fetchActionsFromUrl",
+        actionDefinition
+      })
+    });
+    return constructRemoteActions({
+      json,
+      messages,
+      url: actionDefinition.url,
+      onBeforeRequest: actionDefinition.onBeforeRequest,
+      graphqlContext,
+      logger: logger2.child({
+        component: "remote-actions.constructActions",
+        actionDefinition
+      }),
+      agentStates
+    });
+  }));
+  return result.flat();
+}
+__name(setupRemoteActions, "setupRemoteActions");
+
+// src/service-adapters/events.ts
 var RuntimeEventTypes;
 (function(RuntimeEventTypes2) {
   RuntimeEventTypes2["TextMessageStart"] = "TextMessageStart";
@@ -1280,8 +1652,9 @@ var RuntimeEventTypes;
   RuntimeEventTypes2["ActionExecutionArgs"] = "ActionExecutionArgs";
   RuntimeEventTypes2["ActionExecutionEnd"] = "ActionExecutionEnd";
   RuntimeEventTypes2["ActionExecutionResult"] = "ActionExecutionResult";
+  RuntimeEventTypes2["AgentStateMessage"] = "AgentStateMessage";
 })(RuntimeEventTypes || (RuntimeEventTypes = {}));
-var RuntimeEventSubject = class extends import_rxjs.ReplaySubject {
+var RuntimeEventSubject = class extends import_rxjs2.ReplaySubject {
   constructor() {
     super();
   }
@@ -1338,6 +1711,19 @@ var RuntimeEventSubject = class extends import_rxjs.ReplaySubject {
       result
     });
   }
+  sendAgentStateMessage(threadId, agentName, nodeName, runId, active, role, state, running) {
+    this.next({
+      type: "AgentStateMessage",
+      threadId,
+      agentName,
+      nodeName,
+      runId,
+      active,
+      role,
+      state,
+      running
+    });
+  }
 };
 __name(RuntimeEventSubject, "RuntimeEventSubject");
 var RuntimeEventSource = class {
@@ -1346,26 +1732,31 @@ var RuntimeEventSource = class {
   async stream(callback) {
     this.callback = callback;
   }
-  process({ serversideActions, guardrailsResult$ }) {
+  processRuntimeEvents({ serverSideActions, guardrailsResult$, actionInputsWithoutAgents }) {
     this.callback(this.eventStream$).catch((error) => {
       console.error("Error in event source callback", error);
     });
     return this.eventStream$.pipe(
       // mark tools for server side execution
-      (0, import_rxjs.map)((event) => {
+      (0, import_rxjs2.map)((event) => {
         if (event.type === "ActionExecutionStart") {
-          event.scope = serversideActions.find((action) => action.name === event.actionName) ? "server" : "client";
+          if (event.scope !== "passThrough") {
+            event.scope = serverSideActions.find((action) => action.name === event.actionName) ? "server" : "client";
+          }
         }
         return event;
       }),
       // track state
-      (0, import_rxjs.scan)((acc, event) => {
+      (0, import_rxjs2.scan)((acc, event) => {
+        acc = {
+          ...acc
+        };
         if (event.type === "ActionExecutionStart") {
           acc.callActionServerSide = event.scope === "server";
           acc.args = "";
           acc.actionExecutionId = event.actionExecutionId;
           if (acc.callActionServerSide) {
-            acc.action = serversideActions.find((action) => action.name === event.actionName);
+            acc.action = serverSideActions.find((action) => action.name === event.actionName);
           }
         } else if (event.type === "ActionExecutionArgs") {
           acc.args += event.args;
@@ -1379,24 +1770,26 @@ var RuntimeEventSource = class {
         actionExecutionId: null,
         action: null
       }),
-      (0, import_rxjs.concatMap)((eventWithState) => {
+      (0, import_rxjs2.concatMap)((eventWithState) => {
         if (eventWithState.event.type === "ActionExecutionEnd" && eventWithState.callActionServerSide) {
           const toolCallEventStream$ = new RuntimeEventSubject();
-          executeAction(toolCallEventStream$, guardrailsResult$ ? guardrailsResult$ : null, eventWithState.action, eventWithState.args, eventWithState.actionExecutionId).catch((error) => {
+          executeAction(toolCallEventStream$, guardrailsResult$ ? guardrailsResult$ : null, eventWithState.action, eventWithState.args, eventWithState.actionExecutionId, actionInputsWithoutAgents).catch((error) => {
             console.error(error);
           });
-          return (0, import_rxjs.concat)((0, import_rxjs.of)(eventWithState.event), toolCallEventStream$);
+          telemetry_client_default.capture("oss.runtime.server_action_executed", {});
+          return (0, import_rxjs2.concat)((0, import_rxjs2.of)(eventWithState.event), toolCallEventStream$);
         } else {
-          return (0, import_rxjs.of)(eventWithState.event);
+          return (0, import_rxjs2.of)(eventWithState.event);
         }
       })
     );
   }
 };
 __name(RuntimeEventSource, "RuntimeEventSource");
-async function executeAction(eventStream$, guardrailsResult$, action, actionArguments, actionExecutionId) {
+async function executeAction(eventStream$, guardrailsResult$, action, actionArguments, actionExecutionId, actionInputsWithoutAgents) {
+  var _a;
   if (guardrailsResult$) {
-    const { status } = await (0, import_rxjs.firstValueFrom)(guardrailsResult$);
+    const { status } = await (0, import_rxjs2.firstValueFrom)(guardrailsResult$);
     if (status === "denied") {
       eventStream$.complete();
       return;
@@ -1406,15 +1799,28 @@ async function executeAction(eventStream$, guardrailsResult$, action, actionArgu
   if (actionArguments) {
     args = JSON.parse(actionArguments);
   }
-  const result = await action.handler(args);
-  await streamLangChainResponse({
-    result,
-    eventStream$,
-    actionExecution: {
+  if (isLangGraphAgentAction(action)) {
+    eventStream$.sendActionExecutionResult(actionExecutionId, action.name, `${action.name} agent started`);
+    const stream = await action.langGraphAgentHandler({
       name: action.name,
-      id: actionExecutionId
-    }
-  });
+      actionInputsWithoutAgents
+    });
+    (0, import_rxjs2.from)(stream).subscribe({
+      next: (event) => eventStream$.next(event),
+      error: (err) => console.error("Error in stream", err),
+      complete: () => eventStream$.complete()
+    });
+  } else {
+    const result = await ((_a = action.handler) == null ? void 0 : _a.call(action, args));
+    await streamLangChainResponse({
+      result,
+      eventStream$,
+      actionExecution: {
+        name: action.name,
+        id: actionExecutionId
+      }
+    });
+  }
 }
 __name(executeAction, "executeAction");
 
@@ -1446,15 +1852,30 @@ function convertGqlInputToMessages(inputMessages) {
         actionName: message.resultMessage.actionName,
         result: message.resultMessage.result
       }));
+    } else if (message.agentStateMessage) {
+      messages.push((0, import_class_transformer.plainToInstance)(AgentStateMessage, {
+        id: message.id,
+        threadId: message.agentStateMessage.threadId,
+        createdAt: message.createdAt,
+        agentName: message.agentStateMessage.agentName,
+        nodeName: message.agentStateMessage.nodeName,
+        runId: message.agentStateMessage.runId,
+        active: message.agentStateMessage.active,
+        role: message.agentStateMessage.role,
+        state: JSON.parse(message.agentStateMessage.state),
+        running: message.agentStateMessage.running
+      }));
     }
   }
   return messages;
 }
 __name(convertGqlInputToMessages, "convertGqlInputToMessages");
 
-// src/lib/copilot-runtime.ts
+// src/lib/runtime/copilot-runtime.ts
+var import_rxjs3 = require("rxjs");
 var CopilotRuntime = class {
   actions;
+  remoteActionDefinitions;
   langserve = [];
   onBeforeRequest;
   onAfterRequest;
@@ -1465,44 +1886,33 @@ var CopilotRuntime = class {
       const remoteChain = new RemoteChain(chain);
       this.langserve.push(remoteChain.toAction());
     }
+    this.remoteActionDefinitions = (params == null ? void 0 : params.remoteActions) || [];
     this.onBeforeRequest = (_a = params == null ? void 0 : params.middleware) == null ? void 0 : _a.onBeforeRequest;
     this.onAfterRequest = (_b = params == null ? void 0 : params.middleware) == null ? void 0 : _b.onAfterRequest;
   }
-  async process(request) {
+  async processRuntimeRequest(request) {
     var _a;
-    const { serviceAdapter, messages, actions: clientSideActionsInput, threadId, runId, properties, outputMessagesPromise, forwardedParameters, url } = request;
-    const langserveFunctions = [];
-    for (const chainPromise of this.langserve) {
-      try {
-        const chain = await chainPromise;
-        langserveFunctions.push(chain);
-      } catch (error) {
-        console.error("Error loading langserve chain:", error);
-      }
+    const { serviceAdapter, messages: rawMessages, actions: clientSideActionsInput, threadId, runId, outputMessagesPromise, graphqlContext, forwardedParameters, agentSession, url } = request;
+    if (agentSession) {
+      return this.processAgentRequest(request);
     }
-    const configuredActions = typeof this.actions === "function" ? this.actions({
-      properties,
-      url
-    }) : this.actions;
-    const actions = [
-      ...configuredActions,
-      ...langserveFunctions
-    ];
-    const serverSideActionsInput = actions.map((action) => ({
+    const messages = rawMessages.filter((message) => !message.agentStateMessage);
+    const inputMessages = convertGqlInputToMessages(messages);
+    const serverSideActions = await this.getServerSideActions(request);
+    const serverSideActionsInput = serverSideActions.map((action) => ({
       name: action.name,
       description: action.description,
-      jsonSchema: JSON.stringify((0, import_shared7.actionParametersToJsonSchema)(action.parameters))
+      jsonSchema: JSON.stringify((0, import_shared8.actionParametersToJsonSchema)(action.parameters))
     }));
     const actionInputs = flattenToolCallsNoDuplicates([
       ...serverSideActionsInput,
       ...clientSideActionsInput
     ]);
-    const inputMessages = convertGqlInputToMessages(messages);
     await ((_a = this.onBeforeRequest) == null ? void 0 : _a.call(this, {
       threadId,
       runId,
       inputMessages,
-      properties,
+      properties: graphqlContext.properties,
       url
     }));
     try {
@@ -1522,7 +1932,7 @@ var CopilotRuntime = class {
           runId: result.runId,
           inputMessages,
           outputMessages,
-          properties,
+          properties: graphqlContext.properties,
           url
         });
       }).catch((_error) => {
@@ -1531,13 +1941,108 @@ var CopilotRuntime = class {
         threadId: result.threadId,
         runId: result.runId,
         eventSource,
-        actions
+        serverSideActions,
+        actionInputsWithoutAgents: actionInputs.filter((action) => (
+          // TODO-AGENTS: do not exclude ALL server side actions
+          !serverSideActions.find((serverSideAction) => serverSideAction.name == action.name)
+        ))
+      };
+    } catch (error) {
+      console.error("Error getting response:", error);
+      throw error;
+    }
+  }
+  async processAgentRequest(request) {
+    var _a;
+    const { messages: rawMessages, outputMessagesPromise, graphqlContext, agentSession } = request;
+    const { threadId = (0, import_shared8.randomId)(), agentName, nodeName } = agentSession;
+    const serverSideActions = await this.getServerSideActions(request);
+    const messages = convertGqlInputToMessages(rawMessages);
+    const agent = serverSideActions.find((action) => action.name === agentName && isLangGraphAgentAction(action));
+    if (!agent) {
+      throw new Error(`Agent ${agentName} not found`);
+    }
+    const serverSideActionsInput = serverSideActions.filter((action) => !isLangGraphAgentAction(action)).map((action) => ({
+      name: action.name,
+      description: action.description,
+      jsonSchema: JSON.stringify((0, import_shared8.actionParametersToJsonSchema)(action.parameters))
+    }));
+    const actionInputsWithoutAgents = flattenToolCallsNoDuplicates([
+      ...serverSideActionsInput,
+      ...request.actions
+    ]);
+    await ((_a = this.onBeforeRequest) == null ? void 0 : _a.call(this, {
+      threadId,
+      runId: void 0,
+      inputMessages: messages,
+      properties: graphqlContext.properties
+    }));
+    try {
+      const eventSource = new RuntimeEventSource();
+      const stream = await agent.langGraphAgentHandler({
+        name: agentName,
+        threadId,
+        nodeName,
+        actionInputsWithoutAgents
+      });
+      eventSource.stream(async (eventStream$) => {
+        (0, import_rxjs3.from)(stream).subscribe({
+          next: (event) => eventStream$.next(event),
+          error: (err) => console.error("Error in stream", err),
+          complete: () => eventStream$.complete()
+        });
+      });
+      outputMessagesPromise.then((outputMessages) => {
+        var _a2;
+        (_a2 = this.onAfterRequest) == null ? void 0 : _a2.call(this, {
+          threadId,
+          runId: void 0,
+          inputMessages: messages,
+          outputMessages,
+          properties: graphqlContext.properties
+        });
+      }).catch((_error) => {
+      });
+      return {
+        threadId,
+        runId: void 0,
+        eventSource,
+        serverSideActions: [],
+        actionInputsWithoutAgents
       };
     } catch (error) {
       console.error("Error getting response:", error);
       throw error;
     }
   }
+  async getServerSideActions(request) {
+    const { messages: rawMessages, graphqlContext, agentStates, url } = request;
+    const inputMessages = convertGqlInputToMessages(rawMessages);
+    const langserveFunctions = [];
+    for (const chainPromise of this.langserve) {
+      try {
+        const chain = await chainPromise;
+        langserveFunctions.push(chain);
+      } catch (error) {
+        console.error("Error loading langserve chain:", error);
+      }
+    }
+    const remoteActions = await setupRemoteActions({
+      remoteActionDefinitions: this.remoteActionDefinitions,
+      graphqlContext,
+      messages: inputMessages,
+      agentStates
+    });
+    const configuredActions = typeof this.actions === "function" ? this.actions({
+      properties: graphqlContext.properties,
+      url
+    }) : this.actions;
+    return [
+      ...configuredActions,
+      ...langserveFunctions,
+      ...remoteActions
+    ];
+  }
 };
 __name(CopilotRuntime, "CopilotRuntime");
 function flattenToolCallsNoDuplicates(toolsByPriority) {
@@ -1554,14 +2059,14 @@ function flattenToolCallsNoDuplicates(toolsByPriority) {
 __name(flattenToolCallsNoDuplicates, "flattenToolCallsNoDuplicates");
 
 // src/lib/integrations/shared.ts
-var import_type_graphql14 = require("type-graphql");
+var import_type_graphql16 = require("type-graphql");
 
 // src/graphql/resolvers/copilot.resolver.ts
-var import_type_graphql13 = require("type-graphql");
-var import_rxjs2 = require("rxjs");
+var import_type_graphql15 = require("type-graphql");
+var import_rxjs4 = require("rxjs");
 
 // src/graphql/inputs/generate-copilot-response.input.ts
-var import_type_graphql9 = require("type-graphql");
+var import_type_graphql11 = require("type-graphql");
 
 // src/graphql/inputs/message.input.ts
 var import_type_graphql3 = require("type-graphql");
@@ -1578,6 +2083,7 @@ var ActionExecutionScope;
 (function(ActionExecutionScope2) {
   ActionExecutionScope2["server"] = "server";
   ActionExecutionScope2["client"] = "client";
+  ActionExecutionScope2["passThrough"] = "passThrough";
 })(ActionExecutionScope || (ActionExecutionScope = {}));
 var CopilotRequestType;
 (function(CopilotRequestType2) {
@@ -1621,6 +2127,7 @@ var MessageInput = class extends BaseMessage {
   textMessage;
   actionExecutionMessage;
   resultMessage;
+  agentStateMessage;
 };
 __name(MessageInput, "MessageInput");
 _ts_decorate2([
@@ -1641,6 +2148,12 @@ _ts_decorate2([
   }),
   _ts_metadata2("design:type", typeof ResultMessageInput === "undefined" ? Object : ResultMessageInput)
 ], MessageInput.prototype, "resultMessage", void 0);
+_ts_decorate2([
+  (0, import_type_graphql3.Field)(() => AgentStateMessageInput, {
+    nullable: true
+  }),
+  _ts_metadata2("design:type", typeof AgentStateMessageInput === "undefined" ? Object : AgentStateMessageInput)
+], MessageInput.prototype, "agentStateMessage", void 0);
 MessageInput = _ts_decorate2([
   (0, import_type_graphql3.InputType)()
 ], MessageInput);
@@ -1702,6 +2215,52 @@ _ts_decorate2([
 ResultMessageInput = _ts_decorate2([
   (0, import_type_graphql3.InputType)()
 ], ResultMessageInput);
+var AgentStateMessageInput = class {
+  threadId;
+  agentName;
+  role;
+  state;
+  running;
+  nodeName;
+  runId;
+  active;
+};
+__name(AgentStateMessageInput, "AgentStateMessageInput");
+_ts_decorate2([
+  (0, import_type_graphql3.Field)(() => String),
+  _ts_metadata2("design:type", String)
+], AgentStateMessageInput.prototype, "threadId", void 0);
+_ts_decorate2([
+  (0, import_type_graphql3.Field)(() => String),
+  _ts_metadata2("design:type", String)
+], AgentStateMessageInput.prototype, "agentName", void 0);
+_ts_decorate2([
+  (0, import_type_graphql3.Field)(() => MessageRole),
+  _ts_metadata2("design:type", typeof MessageRole === "undefined" ? Object : MessageRole)
+], AgentStateMessageInput.prototype, "role", void 0);
+_ts_decorate2([
+  (0, import_type_graphql3.Field)(() => String),
+  _ts_metadata2("design:type", String)
+], AgentStateMessageInput.prototype, "state", void 0);
+_ts_decorate2([
+  (0, import_type_graphql3.Field)(() => Boolean),
+  _ts_metadata2("design:type", Boolean)
+], AgentStateMessageInput.prototype, "running", void 0);
+_ts_decorate2([
+  (0, import_type_graphql3.Field)(() => String),
+  _ts_metadata2("design:type", String)
+], AgentStateMessageInput.prototype, "nodeName", void 0);
+_ts_decorate2([
+  (0, import_type_graphql3.Field)(() => String),
+  _ts_metadata2("design:type", String)
+], AgentStateMessageInput.prototype, "runId", void 0);
+_ts_decorate2([
+  (0, import_type_graphql3.Field)(() => Boolean),
+  _ts_metadata2("design:type", Boolean)
+], AgentStateMessageInput.prototype, "active", void 0);
+AgentStateMessageInput = _ts_decorate2([
+  (0, import_type_graphql3.InputType)()
+], AgentStateMessageInput);
 
 // src/graphql/inputs/frontend.input.ts
 var import_type_graphql5 = require("type-graphql");
@@ -1943,7 +2502,8 @@ ForwardedParametersInput = _ts_decorate7([
   (0, import_type_graphql8.InputType)()
 ], ForwardedParametersInput);
 
-// src/graphql/inputs/generate-copilot-response.input.ts
+// src/graphql/inputs/agent-session.input.ts
+var import_type_graphql9 = require("type-graphql");
 function _ts_decorate8(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")
@@ -1960,18 +2520,96 @@ function _ts_metadata8(k, v) {
     return Reflect.metadata(k, v);
 }
 __name(_ts_metadata8, "_ts_metadata");
+var AgentSessionInput = class {
+  agentName;
+  threadId;
+  nodeName;
+};
+__name(AgentSessionInput, "AgentSessionInput");
+_ts_decorate8([
+  (0, import_type_graphql9.Field)(() => String),
+  _ts_metadata8("design:type", String)
+], AgentSessionInput.prototype, "agentName", void 0);
+_ts_decorate8([
+  (0, import_type_graphql9.Field)(() => String, {
+    nullable: true
+  }),
+  _ts_metadata8("design:type", String)
+], AgentSessionInput.prototype, "threadId", void 0);
+_ts_decorate8([
+  (0, import_type_graphql9.Field)(() => String, {
+    nullable: true
+  }),
+  _ts_metadata8("design:type", String)
+], AgentSessionInput.prototype, "nodeName", void 0);
+AgentSessionInput = _ts_decorate8([
+  (0, import_type_graphql9.InputType)()
+], AgentSessionInput);
+
+// src/graphql/inputs/agent-state.input.ts
+var import_type_graphql10 = require("type-graphql");
+function _ts_decorate9(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")
+    r = Reflect.decorate(decorators, target, key, desc);
+  else
+    for (var i = decorators.length - 1; i >= 0; i--)
+      if (d = decorators[i])
+        r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+  return c > 3 && r && Object.defineProperty(target, key, r), r;
+}
+__name(_ts_decorate9, "_ts_decorate");
+function _ts_metadata9(k, v) {
+  if (typeof Reflect === "object" && typeof Reflect.metadata === "function")
+    return Reflect.metadata(k, v);
+}
+__name(_ts_metadata9, "_ts_metadata");
+var AgentStateInput = class {
+  agentName;
+  state;
+};
+__name(AgentStateInput, "AgentStateInput");
+_ts_decorate9([
+  (0, import_type_graphql10.Field)(() => String),
+  _ts_metadata9("design:type", String)
+], AgentStateInput.prototype, "agentName", void 0);
+_ts_decorate9([
+  (0, import_type_graphql10.Field)(() => String),
+  _ts_metadata9("design:type", String)
+], AgentStateInput.prototype, "state", void 0);
+AgentStateInput = _ts_decorate9([
+  (0, import_type_graphql10.InputType)()
+], AgentStateInput);
+
+// src/graphql/inputs/generate-copilot-response.input.ts
+function _ts_decorate10(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")
+    r = Reflect.decorate(decorators, target, key, desc);
+  else
+    for (var i = decorators.length - 1; i >= 0; i--)
+      if (d = decorators[i])
+        r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+  return c > 3 && r && Object.defineProperty(target, key, r), r;
+}
+__name(_ts_decorate10, "_ts_decorate");
+function _ts_metadata10(k, v) {
+  if (typeof Reflect === "object" && typeof Reflect.metadata === "function")
+    return Reflect.metadata(k, v);
+}
+__name(_ts_metadata10, "_ts_metadata");
 var GenerateCopilotResponseMetadataInput = class {
   requestType;
 };
 __name(GenerateCopilotResponseMetadataInput, "GenerateCopilotResponseMetadataInput");
-_ts_decorate8([
-  (0, import_type_graphql9.Field)(() => CopilotRequestType, {
+_ts_decorate10([
+  (0, import_type_graphql11.Field)(() => CopilotRequestType, {
     nullable: true
   }),
-  _ts_metadata8("design:type", typeof CopilotRequestType === "undefined" ? Object : CopilotRequestType)
+  _ts_metadata10("design:type", typeof CopilotRequestType === "undefined" ? Object : CopilotRequestType)
 ], GenerateCopilotResponseMetadataInput.prototype, "requestType", void 0);
-GenerateCopilotResponseMetadataInput = _ts_decorate8([
-  (0, import_type_graphql9.InputType)()
+GenerateCopilotResponseMetadataInput = _ts_decorate10([
+  (0, import_type_graphql11.InputType)()
 ], GenerateCopilotResponseMetadataInput);
 var GenerateCopilotResponseInput = class {
   metadata;
@@ -1981,58 +2619,81 @@ var GenerateCopilotResponseInput = class {
   frontend;
   cloud;
   forwardedParameters;
+  agentSession;
+  agentState;
+  agentStates;
 };
 __name(GenerateCopilotResponseInput, "GenerateCopilotResponseInput");
-_ts_decorate8([
-  (0, import_type_graphql9.Field)(() => GenerateCopilotResponseMetadataInput, {
+_ts_decorate10([
+  (0, import_type_graphql11.Field)(() => GenerateCopilotResponseMetadataInput, {
     nullable: false
   }),
-  _ts_metadata8("design:type", typeof GenerateCopilotResponseMetadataInput === "undefined" ? Object : GenerateCopilotResponseMetadataInput)
+  _ts_metadata10("design:type", typeof GenerateCopilotResponseMetadataInput === "undefined" ? Object : GenerateCopilotResponseMetadataInput)
 ], GenerateCopilotResponseInput.prototype, "metadata", void 0);
-_ts_decorate8([
-  (0, import_type_graphql9.Field)(() => String, {
+_ts_decorate10([
+  (0, import_type_graphql11.Field)(() => String, {
     nullable: true
   }),
-  _ts_metadata8("design:type", String)
+  _ts_metadata10("design:type", String)
 ], GenerateCopilotResponseInput.prototype, "threadId", void 0);
-_ts_decorate8([
-  (0, import_type_graphql9.Field)(() => String, {
+_ts_decorate10([
+  (0, import_type_graphql11.Field)(() => String, {
     nullable: true
   }),
-  _ts_metadata8("design:type", String)
+  _ts_metadata10("design:type", String)
 ], GenerateCopilotResponseInput.prototype, "runId", void 0);
-_ts_decorate8([
-  (0, import_type_graphql9.Field)(() => [
+_ts_decorate10([
+  (0, import_type_graphql11.Field)(() => [
     MessageInput
   ]),
-  _ts_metadata8("design:type", Array)
+  _ts_metadata10("design:type", Array)
 ], GenerateCopilotResponseInput.prototype, "messages", void 0);
-_ts_decorate8([
-  (0, import_type_graphql9.Field)(() => FrontendInput),
-  _ts_metadata8("design:type", typeof FrontendInput === "undefined" ? Object : FrontendInput)
+_ts_decorate10([
+  (0, import_type_graphql11.Field)(() => FrontendInput),
+  _ts_metadata10("design:type", typeof FrontendInput === "undefined" ? Object : FrontendInput)
 ], GenerateCopilotResponseInput.prototype, "frontend", void 0);
-_ts_decorate8([
-  (0, import_type_graphql9.Field)(() => CloudInput, {
+_ts_decorate10([
+  (0, import_type_graphql11.Field)(() => CloudInput, {
     nullable: true
   }),
-  _ts_metadata8("design:type", typeof CloudInput === "undefined" ? Object : CloudInput)
+  _ts_metadata10("design:type", typeof CloudInput === "undefined" ? Object : CloudInput)
 ], GenerateCopilotResponseInput.prototype, "cloud", void 0);
-_ts_decorate8([
-  (0, import_type_graphql9.Field)(() => ForwardedParametersInput, {
+_ts_decorate10([
+  (0, import_type_graphql11.Field)(() => ForwardedParametersInput, {
     nullable: true
   }),
-  _ts_metadata8("design:type", typeof ForwardedParametersInput === "undefined" ? Object : ForwardedParametersInput)
+  _ts_metadata10("design:type", typeof ForwardedParametersInput === "undefined" ? Object : ForwardedParametersInput)
 ], GenerateCopilotResponseInput.prototype, "forwardedParameters", void 0);
-GenerateCopilotResponseInput = _ts_decorate8([
-  (0, import_type_graphql9.InputType)()
+_ts_decorate10([
+  (0, import_type_graphql11.Field)(() => AgentSessionInput, {
+    nullable: true
+  }),
+  _ts_metadata10("design:type", typeof AgentSessionInput === "undefined" ? Object : AgentSessionInput)
+], GenerateCopilotResponseInput.prototype, "agentSession", void 0);
+_ts_decorate10([
+  (0, import_type_graphql11.Field)(() => AgentStateInput, {
+    nullable: true
+  }),
+  _ts_metadata10("design:type", typeof AgentStateInput === "undefined" ? Object : AgentStateInput)
+], GenerateCopilotResponseInput.prototype, "agentState", void 0);
+_ts_decorate10([
+  (0, import_type_graphql11.Field)(() => [
+    AgentStateInput
+  ], {
+    nullable: true
+  }),
+  _ts_metadata10("design:type", Array)
+], GenerateCopilotResponseInput.prototype, "agentStates", void 0);
+GenerateCopilotResponseInput = _ts_decorate10([
+  (0, import_type_graphql11.InputType)()
 ], GenerateCopilotResponseInput);
 
 // src/graphql/types/copilot-response.type.ts
-var import_type_graphql12 = require("type-graphql");
+var import_type_graphql14 = require("type-graphql");
 
 // src/graphql/types/message-status.type.ts
-var import_type_graphql10 = require("type-graphql");
-function _ts_decorate9(decorators, target, key, desc) {
+var import_type_graphql12 = require("type-graphql");
+function _ts_decorate11(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")
     r = Reflect.decorate(decorators, target, key, desc);
@@ -2042,58 +2703,58 @@ function _ts_decorate9(decorators, target, key, desc) {
         r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
   return c > 3 && r && Object.defineProperty(target, key, r), r;
 }
-__name(_ts_decorate9, "_ts_decorate");
-function _ts_metadata9(k, v) {
+__name(_ts_decorate11, "_ts_decorate");
+function _ts_metadata11(k, v) {
   if (typeof Reflect === "object" && typeof Reflect.metadata === "function")
     return Reflect.metadata(k, v);
 }
-__name(_ts_metadata9, "_ts_metadata");
+__name(_ts_metadata11, "_ts_metadata");
 var MessageStatusCode;
 (function(MessageStatusCode2) {
   MessageStatusCode2["Pending"] = "pending";
   MessageStatusCode2["Success"] = "success";
   MessageStatusCode2["Failed"] = "failed";
 })(MessageStatusCode || (MessageStatusCode = {}));
-(0, import_type_graphql10.registerEnumType)(MessageStatusCode, {
+(0, import_type_graphql12.registerEnumType)(MessageStatusCode, {
   name: "MessageStatusCode"
 });
 var BaseMessageStatus = /* @__PURE__ */ __name(class BaseMessageStatus2 {
   code;
 }, "BaseMessageStatus");
-_ts_decorate9([
-  (0, import_type_graphql10.Field)(() => MessageStatusCode),
-  _ts_metadata9("design:type", String)
+_ts_decorate11([
+  (0, import_type_graphql12.Field)(() => MessageStatusCode),
+  _ts_metadata11("design:type", String)
 ], BaseMessageStatus.prototype, "code", void 0);
-BaseMessageStatus = _ts_decorate9([
-  (0, import_type_graphql10.ObjectType)()
+BaseMessageStatus = _ts_decorate11([
+  (0, import_type_graphql12.ObjectType)()
 ], BaseMessageStatus);
 var PendingMessageStatus = class extends BaseMessageStatus {
   code = "pending";
 };
 __name(PendingMessageStatus, "PendingMessageStatus");
-PendingMessageStatus = _ts_decorate9([
-  (0, import_type_graphql10.ObjectType)()
+PendingMessageStatus = _ts_decorate11([
+  (0, import_type_graphql12.ObjectType)()
 ], PendingMessageStatus);
 var SuccessMessageStatus = class extends BaseMessageStatus {
   code = "success";
 };
 __name(SuccessMessageStatus, "SuccessMessageStatus");
-SuccessMessageStatus = _ts_decorate9([
-  (0, import_type_graphql10.ObjectType)()
+SuccessMessageStatus = _ts_decorate11([
+  (0, import_type_graphql12.ObjectType)()
 ], SuccessMessageStatus);
 var FailedMessageStatus = class extends BaseMessageStatus {
   code = "failed";
   reason;
 };
 __name(FailedMessageStatus, "FailedMessageStatus");
-_ts_decorate9([
-  (0, import_type_graphql10.Field)(() => String),
-  _ts_metadata9("design:type", String)
+_ts_decorate11([
+  (0, import_type_graphql12.Field)(() => String),
+  _ts_metadata11("design:type", String)
 ], FailedMessageStatus.prototype, "reason", void 0);
-FailedMessageStatus = _ts_decorate9([
-  (0, import_type_graphql10.ObjectType)()
+FailedMessageStatus = _ts_decorate11([
+  (0, import_type_graphql12.ObjectType)()
 ], FailedMessageStatus);
-var MessageStatusUnion = (0, import_type_graphql10.createUnionType)({
+var MessageStatusUnion = (0, import_type_graphql12.createUnionType)({
   name: "MessageStatus",
   types: () => [
     PendingMessageStatus,
@@ -2104,8 +2765,8 @@ var MessageStatusUnion = (0, import_type_graphql10.createUnionType)({
 
 // src/graphql/types/response-status.type.ts
 var import_graphql_scalars = require("graphql-scalars");
-var import_type_graphql11 = require("type-graphql");
-function _ts_decorate10(decorators, target, key, desc) {
+var import_type_graphql13 = require("type-graphql");
+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")
     r = Reflect.decorate(decorators, target, key, desc);
@@ -2115,30 +2776,30 @@ function _ts_decorate10(decorators, target, key, desc) {
         r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
   return c > 3 && r && Object.defineProperty(target, key, r), r;
 }
-__name(_ts_decorate10, "_ts_decorate");
-function _ts_metadata10(k, v) {
+__name(_ts_decorate12, "_ts_decorate");
+function _ts_metadata12(k, v) {
   if (typeof Reflect === "object" && typeof Reflect.metadata === "function")
     return Reflect.metadata(k, v);
 }
-__name(_ts_metadata10, "_ts_metadata");
+__name(_ts_metadata12, "_ts_metadata");
 var ResponseStatusCode;
 (function(ResponseStatusCode2) {
   ResponseStatusCode2["Pending"] = "pending";
   ResponseStatusCode2["Success"] = "success";
   ResponseStatusCode2["Failed"] = "failed";
 })(ResponseStatusCode || (ResponseStatusCode = {}));
-(0, import_type_graphql11.registerEnumType)(ResponseStatusCode, {
+(0, import_type_graphql13.registerEnumType)(ResponseStatusCode, {
   name: "ResponseStatusCode"
 });
 var BaseResponseStatus = /* @__PURE__ */ __name(class BaseResponseStatus2 {
   code;
 }, "BaseResponseStatus");
-_ts_decorate10([
-  (0, import_type_graphql11.Field)(() => ResponseStatusCode),
-  _ts_metadata10("design:type", String)
+_ts_decorate12([
+  (0, import_type_graphql13.Field)(() => ResponseStatusCode),
+  _ts_metadata12("design:type", String)
 ], BaseResponseStatus.prototype, "code", void 0);
-BaseResponseStatus = _ts_decorate10([
-  (0, import_type_graphql11.InterfaceType)({
+BaseResponseStatus = _ts_decorate12([
+  (0, import_type_graphql13.InterfaceType)({
     resolveType(value) {
       if (value.code === "success") {
         return SuccessResponseStatus;
@@ -2150,14 +2811,14 @@ BaseResponseStatus = _ts_decorate10([
       return void 0;
     }
   }),
-  (0, import_type_graphql11.ObjectType)()
+  (0, import_type_graphql13.ObjectType)()
 ], BaseResponseStatus);
 var PendingResponseStatus = class extends BaseResponseStatus {
   code = "pending";
 };
 __name(PendingResponseStatus, "PendingResponseStatus");
-PendingResponseStatus = _ts_decorate10([
-  (0, import_type_graphql11.ObjectType)({
+PendingResponseStatus = _ts_decorate12([
+  (0, import_type_graphql13.ObjectType)({
     implements: BaseResponseStatus
   })
 ], PendingResponseStatus);
@@ -2165,8 +2826,8 @@ var SuccessResponseStatus = class extends BaseResponseStatus {
   code = "success";
 };
 __name(SuccessResponseStatus, "SuccessResponseStatus");
-SuccessResponseStatus = _ts_decorate10([
-  (0, import_type_graphql11.ObjectType)({
+SuccessResponseStatus = _ts_decorate12([
+  (0, import_type_graphql13.ObjectType)({
     implements: BaseResponseStatus
   })
 ], SuccessResponseStatus);
@@ -2176,7 +2837,7 @@ var FailedResponseStatusReason;
   FailedResponseStatusReason2["MESSAGE_STREAM_INTERRUPTED"] = "MESSAGE_STREAM_INTERRUPTED";
   FailedResponseStatusReason2["UNKNOWN_ERROR"] = "UNKNOWN_ERROR";
 })(FailedResponseStatusReason || (FailedResponseStatusReason = {}));
-(0, import_type_graphql11.registerEnumType)(FailedResponseStatusReason, {
+(0, import_type_graphql13.registerEnumType)(FailedResponseStatusReason, {
   name: "FailedResponseStatusReason"
 });
 var FailedResponseStatus = class extends BaseResponseStatus {
@@ -2185,22 +2846,22 @@ var FailedResponseStatus = class extends BaseResponseStatus {
   details = null;
 };
 __name(FailedResponseStatus, "FailedResponseStatus");
-_ts_decorate10([
-  (0, import_type_graphql11.Field)(() => FailedResponseStatusReason),
-  _ts_metadata10("design:type", String)
+_ts_decorate12([
+  (0, import_type_graphql13.Field)(() => FailedResponseStatusReason),
+  _ts_metadata12("design:type", String)
 ], FailedResponseStatus.prototype, "reason", void 0);
-_ts_decorate10([
-  (0, import_type_graphql11.Field)(() => import_graphql_scalars.GraphQLJSON, {
+_ts_decorate12([
+  (0, import_type_graphql13.Field)(() => import_graphql_scalars.GraphQLJSON, {
     nullable: true
   }),
-  _ts_metadata10("design:type", typeof Record === "undefined" ? Object : Record)
+  _ts_metadata12("design:type", typeof Record === "undefined" ? Object : Record)
 ], FailedResponseStatus.prototype, "details", void 0);
-FailedResponseStatus = _ts_decorate10([
-  (0, import_type_graphql11.ObjectType)({
+FailedResponseStatus = _ts_decorate12([
+  (0, import_type_graphql13.ObjectType)({
     implements: BaseResponseStatus
   })
 ], FailedResponseStatus);
-var ResponseStatusUnion = (0, import_type_graphql11.createUnionType)({
+var ResponseStatusUnion = (0, import_type_graphql13.createUnionType)({
   name: "ResponseStatus",
   types: () => [
     PendingResponseStatus,
@@ -2210,7 +2871,7 @@ var ResponseStatusUnion = (0, import_type_graphql11.createUnionType)({
 });
 
 // src/graphql/types/copilot-response.type.ts
-function _ts_decorate11(decorators, target, key, desc) {
+function _ts_decorate13(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")
     r = Reflect.decorate(decorators, target, key, desc);
@@ -2220,31 +2881,31 @@ function _ts_decorate11(decorators, target, key, desc) {
         r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
   return c > 3 && r && Object.defineProperty(target, key, r), r;
 }
-__name(_ts_decorate11, "_ts_decorate");
-function _ts_metadata11(k, v) {
+__name(_ts_decorate13, "_ts_decorate");
+function _ts_metadata13(k, v) {
   if (typeof Reflect === "object" && typeof Reflect.metadata === "function")
     return Reflect.metadata(k, v);
 }
-__name(_ts_metadata11, "_ts_metadata");
+__name(_ts_metadata13, "_ts_metadata");
 var BaseMessageOutput = /* @__PURE__ */ __name(class BaseMessageOutput2 {
   id;
   createdAt;
   status;
 }, "BaseMessageOutput");
-_ts_decorate11([
-  (0, import_type_graphql12.Field)(() => String),
-  _ts_metadata11("design:type", String)
+_ts_decorate13([
+  (0, import_type_graphql14.Field)(() => String),
+  _ts_metadata13("design:type", String)
 ], BaseMessageOutput.prototype, "id", void 0);
-_ts_decorate11([
-  (0, import_type_graphql12.Field)(() => Date),
-  _ts_metadata11("design:type", typeof Date === "undefined" ? Object : Date)
+_ts_decorate13([
+  (0, import_type_graphql14.Field)(() => Date),
+  _ts_metadata13("design:type", typeof Date === "undefined" ? Object : Date)
 ], BaseMessageOutput.prototype, "createdAt", void 0);
-_ts_decorate11([
-  (0, import_type_graphql12.Field)(() => MessageStatusUnion),
-  _ts_metadata11("design:type", Object)
+_ts_decorate13([
+  (0, import_type_graphql14.Field)(() => MessageStatusUnion),
+  _ts_metadata13("design:type", Object)
 ], BaseMessageOutput.prototype, "status", void 0);
-BaseMessageOutput = _ts_decorate11([
-  (0, import_type_graphql12.InterfaceType)({
+BaseMessageOutput = _ts_decorate13([
+  (0, import_type_graphql14.InterfaceType)({
     resolveType(value) {
       if (value.hasOwnProperty("content")) {
         return TextMessageOutput;
@@ -2252,6 +2913,8 @@ BaseMessageOutput = _ts_decorate11([
         return ActionExecutionMessageOutput;
       } else if (value.hasOwnProperty("result")) {
         return ResultMessageOutput;
+      } else if (value.hasOwnProperty("state")) {
+        return AgentStateMessageOutput;
       }
       return void 0;
     }
@@ -2262,18 +2925,18 @@ var TextMessageOutput = class {
   content;
 };
 __name(TextMessageOutput, "TextMessageOutput");
-_ts_decorate11([
-  (0, import_type_graphql12.Field)(() => MessageRole),
-  _ts_metadata11("design:type", typeof MessageRole === "undefined" ? Object : MessageRole)
+_ts_decorate13([
+  (0, import_type_graphql14.Field)(() => MessageRole),
+  _ts_metadata13("design:type", typeof MessageRole === "undefined" ? Object : MessageRole)
 ], TextMessageOutput.prototype, "role", void 0);
-_ts_decorate11([
-  (0, import_type_graphql12.Field)(() => [
+_ts_decorate13([
+  (0, import_type_graphql14.Field)(() => [
     String
   ]),
-  _ts_metadata11("design:type", Array)
+  _ts_metadata13("design:type", Array)
 ], TextMessageOutput.prototype, "content", void 0);
-TextMessageOutput = _ts_decorate11([
-  (0, import_type_graphql12.ObjectType)({
+TextMessageOutput = _ts_decorate13([
+  (0, import_type_graphql14.ObjectType)({
     implements: BaseMessageOutput
   })
 ], TextMessageOutput);
@@ -2283,22 +2946,22 @@ var ActionExecutionMessageOutput = class {
   arguments;
 };
 __name(ActionExecutionMessageOutput, "ActionExecutionMessageOutput");
-_ts_decorate11([
-  (0, import_type_graphql12.Field)(() => String),
-  _ts_metadata11("design:type", String)
+_ts_decorate13([
+  (0, import_type_graphql14.Field)(() => String),
+  _ts_metadata13("design:type", String)
 ], ActionExecutionMessageOutput.prototype, "name", void 0);
-_ts_decorate11([
-  (0, import_type_graphql12.Field)(() => ActionExecutionScope),
-  _ts_metadata11("design:type", typeof ActionExecutionScope === "undefined" ? Object : ActionExecutionScope)
+_ts_decorate13([
+  (0, import_type_graphql14.Field)(() => ActionExecutionScope),
+  _ts_metadata13("design:type", typeof ActionExecutionScope === "undefined" ? Object : ActionExecutionScope)
 ], ActionExecutionMessageOutput.prototype, "scope", void 0);
-_ts_decorate11([
-  (0, import_type_graphql12.Field)(() => [
+_ts_decorate13([
+  (0, import_type_graphql14.Field)(() => [
     String
   ]),
-  _ts_metadata11("design:type", Array)
+  _ts_metadata13("design:type", Array)
 ], ActionExecutionMessageOutput.prototype, "arguments", void 0);
-ActionExecutionMessageOutput = _ts_decorate11([
-  (0, import_type_graphql12.ObjectType)({
+ActionExecutionMessageOutput = _ts_decorate13([
+  (0, import_type_graphql14.ObjectType)({
     implements: BaseMessageOutput
   })
 ], ActionExecutionMessageOutput);
@@ -2308,23 +2971,71 @@ var ResultMessageOutput = class {
   result;
 };
 __name(ResultMessageOutput, "ResultMessageOutput");
-_ts_decorate11([
-  (0, import_type_graphql12.Field)(() => String),
-  _ts_metadata11("design:type", String)
+_ts_decorate13([
+  (0, import_type_graphql14.Field)(() => String),
+  _ts_metadata13("design:type", String)
 ], ResultMessageOutput.prototype, "actionExecutionId", void 0);
-_ts_decorate11([
-  (0, import_type_graphql12.Field)(() => String),
-  _ts_metadata11("design:type", String)
+_ts_decorate13([
+  (0, import_type_graphql14.Field)(() => String),
+  _ts_metadata13("design:type", String)
 ], ResultMessageOutput.prototype, "actionName", void 0);
-_ts_decorate11([
-  (0, import_type_graphql12.Field)(() => String),
-  _ts_metadata11("design:type", String)
+_ts_decorate13([
+  (0, import_type_graphql14.Field)(() => String),
+  _ts_metadata13("design:type", String)
 ], ResultMessageOutput.prototype, "result", void 0);
-ResultMessageOutput = _ts_decorate11([
-  (0, import_type_graphql12.ObjectType)({
+ResultMessageOutput = _ts_decorate13([
+  (0, import_type_graphql14.ObjectType)({
     implements: BaseMessageOutput
   })
 ], ResultMessageOutput);
+var AgentStateMessageOutput = class {
+  threadId;
+  agentName;
+  nodeName;
+  runId;
+  active;
+  role;
+  state;
+  running;
+};
+__name(AgentStateMessageOutput, "AgentStateMessageOutput");
+_ts_decorate13([
+  (0, import_type_graphql14.Field)(() => String),
+  _ts_metadata13("design:type", String)
+], AgentStateMessageOutput.prototype, "threadId", void 0);
+_ts_decorate13([
+  (0, import_type_graphql14.Field)(() => String),
+  _ts_metadata13("design:type", String)
+], AgentStateMessageOutput.prototype, "agentName", void 0);
+_ts_decorate13([
+  (0, import_type_graphql14.Field)(() => String),
+  _ts_metadata13("design:type", String)
+], AgentStateMessageOutput.prototype, "nodeName", void 0);
+_ts_decorate13([
+  (0, import_type_graphql14.Field)(() => String),
+  _ts_metadata13("design:type", String)
+], AgentStateMessageOutput.prototype, "runId", void 0);
+_ts_decorate13([
+  (0, import_type_graphql14.Field)(() => Boolean),
+  _ts_metadata13("design:type", Boolean)
+], AgentStateMessageOutput.prototype, "active", void 0);
+_ts_decorate13([
+  (0, import_type_graphql14.Field)(() => MessageRole),
+  _ts_metadata13("design:type", typeof MessageRole === "undefined" ? Object : MessageRole)
+], AgentStateMessageOutput.prototype, "role", void 0);
+_ts_decorate13([
+  (0, import_type_graphql14.Field)(() => String),
+  _ts_metadata13("design:type", String)
+], AgentStateMessageOutput.prototype, "state", void 0);
+_ts_decorate13([
+  (0, import_type_graphql14.Field)(() => Boolean),
+  _ts_metadata13("design:type", Boolean)
+], AgentStateMessageOutput.prototype, "running", void 0);
+AgentStateMessageOutput = _ts_decorate13([
+  (0, import_type_graphql14.ObjectType)({
+    implements: BaseMessageOutput
+  })
+], AgentStateMessageOutput);
 var CopilotResponse = class {
   threadId;
   status;
@@ -2332,28 +3043,28 @@ var CopilotResponse = class {
   messages;
 };
 __name(CopilotResponse, "CopilotResponse");
-_ts_decorate11([
-  (0, import_type_graphql12.Field)(() => String),
-  _ts_metadata11("design:type", String)
+_ts_decorate13([
+  (0, import_type_graphql14.Field)(() => String),
+  _ts_metadata13("design:type", String)
 ], CopilotResponse.prototype, "threadId", void 0);
-_ts_decorate11([
-  (0, import_type_graphql12.Field)(() => ResponseStatusUnion),
-  _ts_metadata11("design:type", Object)
+_ts_decorate13([
+  (0, import_type_graphql14.Field)(() => ResponseStatusUnion),
+  _ts_metadata13("design:type", Object)
 ], CopilotResponse.prototype, "status", void 0);
-_ts_decorate11([
-  (0, import_type_graphql12.Field)({
+_ts_decorate13([
+  (0, import_type_graphql14.Field)({
     nullable: true
   }),
-  _ts_metadata11("design:type", String)
+  _ts_metadata13("design:type", String)
 ], CopilotResponse.prototype, "runId", void 0);
-_ts_decorate11([
-  (0, import_type_graphql12.Field)(() => [
+_ts_decorate13([
+  (0, import_type_graphql14.Field)(() => [
     BaseMessageOutput
   ]),
-  _ts_metadata11("design:type", Array)
+  _ts_metadata13("design:type", Array)
 ], CopilotResponse.prototype, "messages", void 0);
-CopilotResponse = _ts_decorate11([
-  (0, import_type_graphql12.ObjectType)()
+CopilotResponse = _ts_decorate13([
+  (0, import_type_graphql14.ObjectType)()
 ], CopilotResponse);
 
 // src/graphql/resolvers/copilot.resolver.ts
@@ -2395,18 +3106,9 @@ var UnknownErrorResponse = class extends FailedResponseStatus {
 };
 __name(UnknownErrorResponse, "UnknownErrorResponse");
 
-// src/lib/telemetry-client.ts
-var import_shared8 = require("@copilotkit/shared");
-var packageJson = require_package();
-var telemetryClient = new import_shared8.TelemetryClient({
-  packageName: packageJson.name,
-  packageVersion: packageJson.version
-});
-var telemetry_client_default = telemetryClient;
-
 // src/graphql/resolvers/copilot.resolver.ts
 var import_shared9 = require("@copilotkit/shared");
-function _ts_decorate12(decorators, target, key, desc) {
+function _ts_decorate14(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")
     r = Reflect.decorate(decorators, target, key, desc);
@@ -2416,12 +3118,12 @@ function _ts_decorate12(decorators, target, key, desc) {
         r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
   return c > 3 && r && Object.defineProperty(target, key, r), r;
 }
-__name(_ts_decorate12, "_ts_decorate");
-function _ts_metadata12(k, v) {
+__name(_ts_decorate14, "_ts_decorate");
+function _ts_metadata14(k, v) {
   if (typeof Reflect === "object" && typeof Reflect.metadata === "function")
     return Reflect.metadata(k, v);
 }
-__name(_ts_metadata12, "_ts_metadata");
+__name(_ts_metadata14, "_ts_metadata");
 function _ts_param(paramIndex, decorator) {
   return function(target, key) {
     decorator(target, key, paramIndex);
@@ -2430,7 +3132,6 @@ function _ts_param(paramIndex, decorator) {
 __name(_ts_param, "_ts_param");
 var invokeGuardrails = /* @__PURE__ */ __name(async ({ baseUrl, copilotCloudPublicApiKey, data, onResult, onError }) => {
   var _a;
-  console.log("invokeGuardrails.baseUrl", baseUrl);
   if (data.messages.length && ((_a = data.messages[data.messages.length - 1].textMessage) == null ? void 0 : _a.role) === MessageRole.user) {
     const messages = data.messages.filter((m) => m.textMessage !== void 0 && (m.textMessage.role === MessageRole.user || m.textMessage.role === MessageRole.assistant)).map((m) => ({
       role: m.textMessage.role,
@@ -2476,8 +3177,6 @@ var CopilotResolver = class {
     logger2.debug({
       data
     }, "Generating Copilot response");
-    const copilotRuntime = ctx._copilotkit.runtime;
-    const serviceAdapter = ctx._copilotkit.serviceAdapter;
     if (properties) {
       logger2.debug("Properties provided, merging with context properties");
       ctx.properties = {
@@ -2485,6 +3184,8 @@ var CopilotResolver = class {
         ...properties
       };
     }
+    const copilotRuntime = ctx._copilotkit.runtime;
+    const serviceAdapter = ctx._copilotkit.serviceAdapter;
     let copilotCloudPublicApiKey = null;
     let copilotCloudBaseUrl;
     if (data.cloud) {
@@ -2512,9 +3213,9 @@ var CopilotResolver = class {
       });
     }
     logger2.debug("Setting up subjects");
-    const responseStatus$ = new import_rxjs2.ReplaySubject();
-    const interruptStreaming$ = new import_rxjs2.ReplaySubject();
-    const guardrailsResult$ = new import_rxjs2.ReplaySubject();
+    const responseStatus$ = new import_rxjs4.ReplaySubject();
+    const interruptStreaming$ = new import_rxjs4.ReplaySubject();
+    const guardrailsResult$ = new import_rxjs4.ReplaySubject();
     let outputMessages = [];
     let resolveOutputMessagesPromise;
     let rejectOutputMessagesPromise;
@@ -2523,23 +3224,25 @@ var CopilotResolver = class {
       rejectOutputMessagesPromise = reject;
     });
     logger2.debug("Processing");
-    const { eventSource, threadId = (0, import_shared9.randomId)(), runId, actions } = await copilotRuntime.process({
+    const { eventSource, threadId = (0, import_shared9.randomId)(), runId, serverSideActions, actionInputsWithoutAgents } = await copilotRuntime.processRuntimeRequest({
       serviceAdapter,
       messages: data.messages,
       actions: data.frontend.actions,
       threadId: data.threadId,
       runId: data.runId,
       publicApiKey: void 0,
-      properties: ctx.properties || {},
       outputMessagesPromise,
+      graphqlContext: ctx,
       forwardedParameters: data.forwardedParameters,
+      agentSession: data.agentSession,
+      agentStates: data.agentStates,
       url: data.frontend.url
     });
     logger2.debug("Event source created, creating response");
     const response = {
       threadId,
       runId,
-      status: (0, import_rxjs2.firstValueFrom)(responseStatus$),
+      status: (0, import_rxjs4.firstValueFrom)(responseStatus$),
       messages: new import_graphql_yoga.Repeater(async (pushMessage, stopStreamingMessages) => {
         var _a2, _b2;
         logger2.debug("Messages repeater created");
@@ -2590,14 +3293,18 @@ var CopilotResolver = class {
           });
         }
         let eventStreamSubscription;
-        const eventStream = eventSource.process({
-          serversideActions: actions,
-          guardrailsResult$: ((_b2 = data.cloud) == null ? void 0 : _b2.guardrails) ? guardrailsResult$ : null
+        const eventStream = eventSource.processRuntimeEvents({
+          serverSideActions,
+          guardrailsResult$: ((_b2 = data.cloud) == null ? void 0 : _b2.guardrails) ? guardrailsResult$ : null,
+          actionInputsWithoutAgents: actionInputsWithoutAgents.filter(
+            // TODO-AGENTS: do not exclude ALL server side actions
+            (action) => !serverSideActions.find((serverSideAction) => serverSideAction.name == action.name)
+          )
         }).pipe(
           // shareReplay() ensures that later subscribers will see the whole stream instead of
           // just the events that were emitted after the subscriber was added.
-          (0, import_rxjs2.shareReplay)(),
-          (0, import_rxjs2.finalize)(() => {
+          (0, import_rxjs4.shareReplay)(),
+          (0, import_rxjs4.finalize)(() => {
             logger2.debug("Event stream finalized");
           })
         );
@@ -2608,22 +3315,22 @@ var CopilotResolver = class {
               case RuntimeEventTypes.TextMessageStart:
                 const textMessageContentStream = eventStream.pipe(
                   // skip until this message start event
-                  (0, import_rxjs2.skipWhile)((e) => e !== event),
+                  (0, import_rxjs4.skipWhile)((e) => e !== event),
                   // take until the message end event
-                  (0, import_rxjs2.takeWhile)((e) => e.type != RuntimeEventTypes.TextMessageEnd)
+                  (0, import_rxjs4.takeWhile)((e) => e.type != RuntimeEventTypes.TextMessageEnd)
                 );
-                const streamingTextStatus = new import_rxjs2.Subject();
+                const streamingTextStatus = new import_rxjs4.Subject();
                 const messageId = (0, import_shared9.randomId)();
                 pushMessage({
                   id: messageId,
-                  status: (0, import_rxjs2.firstValueFrom)(streamingTextStatus),
+                  status: (0, import_rxjs4.firstValueFrom)(streamingTextStatus),
                   createdAt: /* @__PURE__ */ new Date(),
                   role: MessageRole.assistant,
                   content: new import_graphql_yoga.Repeater(async (pushTextChunk, stopStreamingText) => {
                     logger2.debug("Text message content repeater created");
                     const textChunks = [];
                     let textSubscription;
-                    interruptStreaming$.pipe((0, import_rxjs2.shareReplay)(), (0, import_rxjs2.take)(1), (0, import_rxjs2.tap)(({ reason, messageId: messageId2 }) => {
+                    interruptStreaming$.pipe((0, import_rxjs4.shareReplay)(), (0, import_rxjs4.take)(1), (0, import_rxjs4.tap)(({ reason, messageId: messageId2 }) => {
                       logger2.debug({
                         reason,
                         messageId: messageId2
@@ -2674,11 +3381,11 @@ var CopilotResolver = class {
                 break;
               case RuntimeEventTypes.ActionExecutionStart:
                 logger2.debug("Action execution start event received");
-                const actionExecutionArgumentStream = eventStream.pipe((0, import_rxjs2.skipWhile)((e) => e !== event), (0, import_rxjs2.takeWhile)((e) => e.type != RuntimeEventTypes.ActionExecutionEnd));
-                const streamingArgumentsStatus = new import_rxjs2.Subject();
+                const actionExecutionArgumentStream = eventStream.pipe((0, import_rxjs4.skipWhile)((e) => e !== event), (0, import_rxjs4.takeWhile)((e) => e.type != RuntimeEventTypes.ActionExecutionEnd));
+                const streamingArgumentsStatus = new import_rxjs4.Subject();
                 pushMessage({
                   id: event.actionExecutionId,
-                  status: (0, import_rxjs2.firstValueFrom)(streamingArgumentsStatus),
+                  status: (0, import_rxjs4.firstValueFrom)(streamingArgumentsStatus),
                   createdAt: /* @__PURE__ */ new Date(),
                   name: event.actionName,
                   scope: event.scope,
@@ -2740,6 +3447,36 @@ var CopilotResolver = class {
                   result: event.result
                 }));
                 break;
+              case RuntimeEventTypes.AgentStateMessage:
+                logger2.debug({
+                  event
+                }, "Agent message event received");
+                pushMessage({
+                  id: (0, import_shared9.randomId)(),
+                  status: new SuccessMessageStatus(),
+                  threadId: event.threadId,
+                  agentName: event.agentName,
+                  nodeName: event.nodeName,
+                  runId: event.runId,
+                  active: event.active,
+                  state: event.state,
+                  running: event.running,
+                  role: MessageRole.assistant,
+                  createdAt: /* @__PURE__ */ new Date()
+                });
+                outputMessages.push((0, import_class_transformer2.plainToInstance)(AgentStateMessage, {
+                  id: (0, import_shared9.randomId)(),
+                  threadId: event.threadId,
+                  agentName: event.agentName,
+                  nodeName: event.nodeName,
+                  runId: event.runId,
+                  active: event.active,
+                  state: event.state,
+                  running: event.running,
+                  role: MessageRole.assistant,
+                  createdAt: /* @__PURE__ */ new Date()
+                }));
+                break;
             }
           },
           error: (err) => {
@@ -2758,7 +3495,7 @@ var CopilotResolver = class {
             logger2.debug("Event stream completed");
             if ((_a3 = data.cloud) == null ? void 0 : _a3.guardrails) {
               logger2.debug("Guardrails is enabled, waiting for guardrails result");
-              await (0, import_rxjs2.firstValueFrom)(guardrailsResult$);
+              await (0, import_rxjs4.firstValueFrom)(guardrailsResult$);
             }
             responseStatus$.next(new SuccessResponseStatus());
             eventStreamSubscription == null ? void 0 : eventStreamSubscription.unsubscribe();
@@ -2772,29 +3509,29 @@ var CopilotResolver = class {
   }
 };
 __name(CopilotResolver, "CopilotResolver");
-_ts_decorate12([
-  (0, import_type_graphql13.Query)(() => String),
-  _ts_metadata12("design:type", Function),
-  _ts_metadata12("design:paramtypes", []),
-  _ts_metadata12("design:returntype", Promise)
+_ts_decorate14([
+  (0, import_type_graphql15.Query)(() => String),
+  _ts_metadata14("design:type", Function),
+  _ts_metadata14("design:paramtypes", []),
+  _ts_metadata14("design:returntype", Promise)
 ], CopilotResolver.prototype, "hello", null);
-_ts_decorate12([
-  (0, import_type_graphql13.Mutation)(() => CopilotResponse),
-  _ts_param(0, (0, import_type_graphql13.Ctx)()),
-  _ts_param(1, (0, import_type_graphql13.Arg)("data")),
-  _ts_param(2, (0, import_type_graphql13.Arg)("properties", () => import_graphql_scalars2.GraphQLJSONObject, {
+_ts_decorate14([
+  (0, import_type_graphql15.Mutation)(() => CopilotResponse),
+  _ts_param(0, (0, import_type_graphql15.Ctx)()),
+  _ts_param(1, (0, import_type_graphql15.Arg)("data")),
+  _ts_param(2, (0, import_type_graphql15.Arg)("properties", () => import_graphql_scalars2.GraphQLJSONObject, {
     nullable: true
   })),
-  _ts_metadata12("design:type", Function),
-  _ts_metadata12("design:paramtypes", [
+  _ts_metadata14("design:type", Function),
+  _ts_metadata14("design:paramtypes", [
     typeof GraphQLContext === "undefined" ? Object : GraphQLContext,
     typeof GenerateCopilotResponseInput === "undefined" ? Object : GenerateCopilotResponseInput,
     typeof CopilotRequestContextProperties === "undefined" ? Object : CopilotRequestContextProperties
   ]),
-  _ts_metadata12("design:returntype", Promise)
+  _ts_metadata14("design:returntype", Promise)
 ], CopilotResolver.prototype, "generateCopilotResponse", null);
-CopilotResolver = _ts_decorate12([
-  (0, import_type_graphql13.Resolver)(() => CopilotResponse)
+CopilotResolver = _ts_decorate14([
+  (0, import_type_graphql15.Resolver)(() => CopilotResponse)
 ], CopilotResolver);
 
 // src/lib/integrations/shared.ts
@@ -2849,7 +3586,7 @@ async function createContext(initialContext, copilotKitContext, contextLogger, p
 __name(createContext, "createContext");
 function buildSchema(options = {}) {
   logger.debug("Building GraphQL schema...");
-  const schema = (0, import_type_graphql14.buildSchemaSync)({
+  const schema = (0, import_type_graphql16.buildSchemaSync)({
     resolvers: [
       CopilotResolver
     ],