@copilotkit/runtime 1.10.5-next.0 → 1.10.5-next.10

package/dist/index.js

@@ -44,7 +44,7 @@ var require_package = __commonJS({
       publishConfig: {
         access: "public"
       },
-      version: "1.10.5-next.0",
+      version: "1.10.5-next.10",
       sideEffects: false,
       main: "./dist/index.js",
       module: "./dist/index.mjs",
@@ -82,11 +82,11 @@ var require_package = __commonJS({
         "zod-to-json-schema": "^3.23.5"
       },
       dependencies: {
-        "@ag-ui/client": "^0.0.37",
-        "@ag-ui/core": "^0.0.37",
-        "@ag-ui/encoder": "^0.0.37",
-        "@ag-ui/langgraph": "^0.0.12",
-        "@ag-ui/proto": "^0.0.37",
+        "@ag-ui/client": "^0.0.39",
+        "@ag-ui/core": "^0.0.39",
+        "@ag-ui/encoder": "^0.0.39",
+        "@ag-ui/langgraph": "^0.0.17",
+        "@ag-ui/proto": "^0.0.39",
         "@anthropic-ai/sdk": "^0.57.0",
         "@copilotkit/shared": "workspace:*",
         "@graphql-yoga/plugin-defer-stream": "^3.3.1",
@@ -115,10 +115,10 @@ var require_package = __commonJS({
         zod: "^3.23.3"
       },
       peerDependencies: {
-        "@ag-ui/client": ">=0.0.37",
-        "@ag-ui/core": ">=0.0.37",
-        "@ag-ui/encoder": ">=0.0.37",
-        "@ag-ui/proto": ">=0.0.37"
+        "@ag-ui/client": ">=0.0.39",
+        "@ag-ui/core": ">=0.0.39",
+        "@ag-ui/encoder": ">=0.0.39",
+        "@ag-ui/proto": ">=0.0.39"
       },
       keywords: [
         "copilotkit",
@@ -1309,14 +1309,14 @@ var GroqAdapter = class {
 __name(GroqAdapter, "GroqAdapter");
 
 // src/lib/integrations/shared.ts
-var import_type_graphql26 = require("type-graphql");
+var import_type_graphql27 = require("type-graphql");
 
 // src/graphql/resolvers/copilot.resolver.ts
-var import_type_graphql20 = require("type-graphql");
+var import_type_graphql21 = require("type-graphql");
 var import_rxjs5 = require("rxjs");
 
 // src/graphql/inputs/generate-copilot-response.input.ts
-var import_type_graphql18 = require("type-graphql");
+var import_type_graphql19 = require("type-graphql");
 
 // src/graphql/inputs/message.input.ts
 var import_type_graphql3 = require("type-graphql");
@@ -2674,7 +2674,8 @@ MetaEventInput = _ts_decorate16([
   (0, import_type_graphql17.InputType)()
 ], MetaEventInput);
 
-// src/graphql/inputs/generate-copilot-response.input.ts
+// src/graphql/inputs/copilot-context.input.ts
+var import_type_graphql18 = require("type-graphql");
 function _ts_decorate17(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")
@@ -2691,18 +2692,52 @@ function _ts_metadata17(k, v) {
     return Reflect.metadata(k, v);
 }
 __name(_ts_metadata17, "_ts_metadata");
+var CopilotContextInput = class {
+  description;
+  value;
+};
+__name(CopilotContextInput, "CopilotContextInput");
+_ts_decorate17([
+  (0, import_type_graphql18.Field)(() => String),
+  _ts_metadata17("design:type", String)
+], CopilotContextInput.prototype, "description", void 0);
+_ts_decorate17([
+  (0, import_type_graphql18.Field)(() => String),
+  _ts_metadata17("design:type", String)
+], CopilotContextInput.prototype, "value", void 0);
+CopilotContextInput = _ts_decorate17([
+  (0, import_type_graphql18.InputType)()
+], CopilotContextInput);
+
+// src/graphql/inputs/generate-copilot-response.input.ts
+function _ts_decorate18(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_decorate18, "_ts_decorate");
+function _ts_metadata18(k, v) {
+  if (typeof Reflect === "object" && typeof Reflect.metadata === "function")
+    return Reflect.metadata(k, v);
+}
+__name(_ts_metadata18, "_ts_metadata");
 var GenerateCopilotResponseMetadataInput = class {
   requestType;
 };
 __name(GenerateCopilotResponseMetadataInput, "GenerateCopilotResponseMetadataInput");
-_ts_decorate17([
-  (0, import_type_graphql18.Field)(() => CopilotRequestType, {
+_ts_decorate18([
+  (0, import_type_graphql19.Field)(() => CopilotRequestType, {
     nullable: true
   }),
-  _ts_metadata17("design:type", typeof CopilotRequestType === "undefined" ? Object : CopilotRequestType)
+  _ts_metadata18("design:type", typeof CopilotRequestType === "undefined" ? Object : CopilotRequestType)
 ], GenerateCopilotResponseMetadataInput.prototype, "requestType", void 0);
-GenerateCopilotResponseMetadataInput = _ts_decorate17([
-  (0, import_type_graphql18.InputType)()
+GenerateCopilotResponseMetadataInput = _ts_decorate18([
+  (0, import_type_graphql19.InputType)()
 ], GenerateCopilotResponseMetadataInput);
 var GenerateCopilotResponseInput = class {
   metadata;
@@ -2717,84 +2752,93 @@ var GenerateCopilotResponseInput = class {
   agentStates;
   extensions;
   metaEvents;
+  context;
 };
 __name(GenerateCopilotResponseInput, "GenerateCopilotResponseInput");
-_ts_decorate17([
-  (0, import_type_graphql18.Field)(() => GenerateCopilotResponseMetadataInput, {
+_ts_decorate18([
+  (0, import_type_graphql19.Field)(() => GenerateCopilotResponseMetadataInput, {
     nullable: false
   }),
-  _ts_metadata17("design:type", typeof GenerateCopilotResponseMetadataInput === "undefined" ? Object : GenerateCopilotResponseMetadataInput)
+  _ts_metadata18("design:type", typeof GenerateCopilotResponseMetadataInput === "undefined" ? Object : GenerateCopilotResponseMetadataInput)
 ], GenerateCopilotResponseInput.prototype, "metadata", void 0);
-_ts_decorate17([
-  (0, import_type_graphql18.Field)(() => String, {
+_ts_decorate18([
+  (0, import_type_graphql19.Field)(() => String, {
     nullable: true
   }),
-  _ts_metadata17("design:type", String)
+  _ts_metadata18("design:type", String)
 ], GenerateCopilotResponseInput.prototype, "threadId", void 0);
-_ts_decorate17([
-  (0, import_type_graphql18.Field)(() => String, {
+_ts_decorate18([
+  (0, import_type_graphql19.Field)(() => String, {
     nullable: true
   }),
-  _ts_metadata17("design:type", String)
+  _ts_metadata18("design:type", String)
 ], GenerateCopilotResponseInput.prototype, "runId", void 0);
-_ts_decorate17([
-  (0, import_type_graphql18.Field)(() => [
+_ts_decorate18([
+  (0, import_type_graphql19.Field)(() => [
     MessageInput
   ]),
-  _ts_metadata17("design:type", Array)
+  _ts_metadata18("design:type", Array)
 ], GenerateCopilotResponseInput.prototype, "messages", void 0);
-_ts_decorate17([
-  (0, import_type_graphql18.Field)(() => FrontendInput),
-  _ts_metadata17("design:type", typeof FrontendInput === "undefined" ? Object : FrontendInput)
+_ts_decorate18([
+  (0, import_type_graphql19.Field)(() => FrontendInput),
+  _ts_metadata18("design:type", typeof FrontendInput === "undefined" ? Object : FrontendInput)
 ], GenerateCopilotResponseInput.prototype, "frontend", void 0);
-_ts_decorate17([
-  (0, import_type_graphql18.Field)(() => CloudInput, {
+_ts_decorate18([
+  (0, import_type_graphql19.Field)(() => CloudInput, {
     nullable: true
   }),
-  _ts_metadata17("design:type", typeof CloudInput === "undefined" ? Object : CloudInput)
+  _ts_metadata18("design:type", typeof CloudInput === "undefined" ? Object : CloudInput)
 ], GenerateCopilotResponseInput.prototype, "cloud", void 0);
-_ts_decorate17([
-  (0, import_type_graphql18.Field)(() => ForwardedParametersInput, {
+_ts_decorate18([
+  (0, import_type_graphql19.Field)(() => ForwardedParametersInput, {
     nullable: true
   }),
-  _ts_metadata17("design:type", typeof ForwardedParametersInput === "undefined" ? Object : ForwardedParametersInput)
+  _ts_metadata18("design:type", typeof ForwardedParametersInput === "undefined" ? Object : ForwardedParametersInput)
 ], GenerateCopilotResponseInput.prototype, "forwardedParameters", void 0);
-_ts_decorate17([
-  (0, import_type_graphql18.Field)(() => AgentSessionInput, {
+_ts_decorate18([
+  (0, import_type_graphql19.Field)(() => AgentSessionInput, {
     nullable: true
   }),
-  _ts_metadata17("design:type", typeof AgentSessionInput === "undefined" ? Object : AgentSessionInput)
+  _ts_metadata18("design:type", typeof AgentSessionInput === "undefined" ? Object : AgentSessionInput)
 ], GenerateCopilotResponseInput.prototype, "agentSession", void 0);
-_ts_decorate17([
-  (0, import_type_graphql18.Field)(() => AgentStateInput, {
+_ts_decorate18([
+  (0, import_type_graphql19.Field)(() => AgentStateInput, {
     nullable: true
   }),
-  _ts_metadata17("design:type", typeof AgentStateInput === "undefined" ? Object : AgentStateInput)
+  _ts_metadata18("design:type", typeof AgentStateInput === "undefined" ? Object : AgentStateInput)
 ], GenerateCopilotResponseInput.prototype, "agentState", void 0);
-_ts_decorate17([
-  (0, import_type_graphql18.Field)(() => [
+_ts_decorate18([
+  (0, import_type_graphql19.Field)(() => [
     AgentStateInput
   ], {
     nullable: true
   }),
-  _ts_metadata17("design:type", Array)
+  _ts_metadata18("design:type", Array)
 ], GenerateCopilotResponseInput.prototype, "agentStates", void 0);
-_ts_decorate17([
-  (0, import_type_graphql18.Field)(() => ExtensionsInput, {
+_ts_decorate18([
+  (0, import_type_graphql19.Field)(() => ExtensionsInput, {
     nullable: true
   }),
-  _ts_metadata17("design:type", typeof ExtensionsInput === "undefined" ? Object : ExtensionsInput)
+  _ts_metadata18("design:type", typeof ExtensionsInput === "undefined" ? Object : ExtensionsInput)
 ], GenerateCopilotResponseInput.prototype, "extensions", void 0);
-_ts_decorate17([
-  (0, import_type_graphql18.Field)(() => [
+_ts_decorate18([
+  (0, import_type_graphql19.Field)(() => [
     MetaEventInput
   ], {
     nullable: true
   }),
-  _ts_metadata17("design:type", Array)
+  _ts_metadata18("design:type", Array)
 ], GenerateCopilotResponseInput.prototype, "metaEvents", void 0);
-GenerateCopilotResponseInput = _ts_decorate17([
-  (0, import_type_graphql18.InputType)()
+_ts_decorate18([
+  (0, import_type_graphql19.Field)(() => [
+    CopilotContextInput
+  ], {
+    nullable: true
+  }),
+  _ts_metadata18("design:type", Array)
+], GenerateCopilotResponseInput.prototype, "context", void 0);
+GenerateCopilotResponseInput = _ts_decorate18([
+  (0, import_type_graphql19.InputType)()
 ], GenerateCopilotResponseInput);
 
 // src/graphql/resolvers/copilot.resolver.ts
@@ -4809,6 +4853,7 @@ var CopilotRuntime = class {
   langserve = [];
   onBeforeRequest;
   onAfterRequest;
+  onStopGeneration;
   delegateAgentProcessingToServiceAdapter;
   observability;
   availableAgents;
@@ -4836,6 +4881,7 @@ var CopilotRuntime = class {
     this.remoteEndpointDefinitions = (params == null ? void 0 : params.remoteEndpoints) ?? (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;
+    this.onStopGeneration = params == null ? void 0 : params.onStopGeneration;
     this.delegateAgentProcessingToServiceAdapter = (params == null ? void 0 : params.delegateAgentProcessingToServiceAdapter) || false;
     this.observability = params == null ? void 0 : params.observability_c;
     this.agents = (params == null ? void 0 : params.agents) ?? {};
@@ -4914,10 +4960,22 @@ var CopilotRuntime = class {
   }
   async processRuntimeRequest(request) {
     var _a, _b, _c, _d, _e;
-    const { serviceAdapter, messages: rawMessages, actions: clientSideActionsInput, threadId, runId, outputMessagesPromise, graphqlContext, forwardedParameters, url, extensions, agentSession, agentStates, publicApiKey } = request;
+    const { serviceAdapter, messages: rawMessages, actions: clientSideActionsInput, threadId, runId, outputMessagesPromise, graphqlContext, forwardedParameters, url, extensions, agentSession, agentStates, publicApiKey, context } = request;
+    graphqlContext.request.signal.addEventListener("abort", () => {
+      var _a2;
+      return (_a2 = this.onStopGeneration) == null ? void 0 : _a2.call(this, {
+        threadId,
+        runId,
+        url,
+        agentName: agentSession == null ? void 0 : agentSession.agentName,
+        lastMessage: rawMessages[rawMessages.length - 1]
+      });
+    }, {
+      once: true
+    });
     const eventSource = new RuntimeEventSource({
-      errorHandler: async (error, context) => {
-        await this.error("error", context, error, publicApiKey);
+      errorHandler: async (error, context2) => {
+        await this.error("error", context2, error, publicApiKey);
       },
       errorContext: {
         threadId,
@@ -5363,7 +5421,7 @@ please use an LLM adapter instead.`
   }
   async processAgentRequest(request) {
     var _a, _b, _c, _d, _e;
-    const { messages: rawMessages, outputMessagesPromise, graphqlContext, agentSession, threadId: threadIdFromRequest, metaEvents, publicApiKey, forwardedParameters } = request;
+    const { messages: rawMessages, outputMessagesPromise, graphqlContext, agentSession, threadId: threadIdFromRequest, metaEvents, publicApiKey, forwardedParameters, context } = request;
     const { agentName, nodeName } = agentSession;
     const requestStartTime = Date.now();
     const streamedChunks = [];
@@ -5437,8 +5495,8 @@ please use an LLM adapter instead.`
     }));
     try {
       const eventSource = new RuntimeEventSource({
-        errorHandler: async (error, context) => {
-          await this.error("error", context, error, publicApiKey);
+        errorHandler: async (error, context2) => {
+          await this.error("error", context2, error, publicApiKey);
         },
         errorContext: {
           threadId,
@@ -5663,7 +5721,8 @@ please use an LLM adapter instead.`
       frontendUrl: url,
       agents: this.agents,
       metaEvents: request.metaEvents,
-      nodeName: (_a = request.agentSession) == null ? void 0 : _a.nodeName
+      nodeName: (_a = request.agentSession) == null ? void 0 : _a.nodeName,
+      context: request.context
     });
     const configuredActions = typeof this.actions === "function" ? this.actions({
       properties: graphqlContext.properties,
@@ -6407,7 +6466,7 @@ var import_shared26 = require("@copilotkit/shared");
 // src/lib/runtime/agui-action.ts
 var import_rxjs3 = require("rxjs");
 var import_shared24 = require("@copilotkit/shared");
-function constructAGUIRemoteAction({ logger: logger2, messages, agentStates, agent, metaEvents, threadMetadata, nodeName, graphqlContext }) {
+function constructAGUIRemoteAction({ logger: logger2, messages, agentStates, agent, metaEvents, threadMetadata, nodeName, context, graphqlContext }) {
   const action = {
     name: agent.agentId,
     description: agent.description,
@@ -6416,6 +6475,11 @@ function constructAGUIRemoteAction({ logger: logger2, messages, agentStates, age
     },
     remoteAgentHandler: async ({ actionInputsWithoutAgents, threadId }) => {
       var _a;
+      graphqlContext.request.signal.addEventListener("abort", () => {
+        agent.abortRun();
+      }, {
+        once: true
+      });
       logger2.debug({
         actionName: agent.agentId
       }, "Executing remote agent");
@@ -6466,7 +6530,8 @@ function constructAGUIRemoteAction({ logger: logger2, messages, agentStates, age
       };
       return agent.legacy_to_be_removed_runAgentBridged({
         tools,
-        forwardedProps
+        forwardedProps,
+        context
       }).pipe((0, import_rxjs3.mergeMap)((event) => {
         if (event.type === RuntimeEventTypes.RunError) {
           const { message } = event;
@@ -6591,7 +6656,7 @@ async function fetchRemoteInfo({ url, onBeforeRequest, graphqlContext, logger: l
   }
 }
 __name(fetchRemoteInfo, "fetchRemoteInfo");
-async function setupRemoteActions({ remoteEndpointDefinitions, graphqlContext, messages, agentStates, frontendUrl, agents, metaEvents, nodeName }) {
+async function setupRemoteActions({ remoteEndpointDefinitions, graphqlContext, messages, agentStates, frontendUrl, agents, metaEvents, nodeName, context }) {
   var _a;
   const logger2 = graphqlContext.logger.child({
     component: "remote-actions.fetchRemoteActions"
@@ -6659,7 +6724,8 @@ async function setupRemoteActions({ remoteEndpointDefinitions, graphqlContext, m
       metaEvents,
       threadMetadata,
       nodeName,
-      graphqlContext
+      graphqlContext,
+      context
     }));
   }
   return result.flat();
@@ -7042,8 +7108,8 @@ __name(UnknownErrorResponse, "UnknownErrorResponse");
 var import_shared28 = require("@copilotkit/shared");
 
 // src/graphql/types/agents-response.type.ts
-var import_type_graphql19 = require("type-graphql");
-function _ts_decorate18(decorators, target, key, desc) {
+var import_type_graphql20 = require("type-graphql");
+function _ts_decorate19(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);
@@ -7053,50 +7119,50 @@ function _ts_decorate18(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_decorate18, "_ts_decorate");
-function _ts_metadata18(k, v) {
+__name(_ts_decorate19, "_ts_decorate");
+function _ts_metadata19(k, v) {
   if (typeof Reflect === "object" && typeof Reflect.metadata === "function")
     return Reflect.metadata(k, v);
 }
-__name(_ts_metadata18, "_ts_metadata");
+__name(_ts_metadata19, "_ts_metadata");
 var Agent = class {
   id;
   name;
   description;
 };
 __name(Agent, "Agent");
-_ts_decorate18([
-  (0, import_type_graphql19.Field)(() => String),
-  _ts_metadata18("design:type", String)
+_ts_decorate19([
+  (0, import_type_graphql20.Field)(() => String),
+  _ts_metadata19("design:type", String)
 ], Agent.prototype, "id", void 0);
-_ts_decorate18([
-  (0, import_type_graphql19.Field)(() => String),
-  _ts_metadata18("design:type", String)
+_ts_decorate19([
+  (0, import_type_graphql20.Field)(() => String),
+  _ts_metadata19("design:type", String)
 ], Agent.prototype, "name", void 0);
-_ts_decorate18([
-  (0, import_type_graphql19.Field)(() => String),
-  _ts_metadata18("design:type", String)
+_ts_decorate19([
+  (0, import_type_graphql20.Field)(() => String),
+  _ts_metadata19("design:type", String)
 ], Agent.prototype, "description", void 0);
-Agent = _ts_decorate18([
-  (0, import_type_graphql19.ObjectType)()
+Agent = _ts_decorate19([
+  (0, import_type_graphql20.ObjectType)()
 ], Agent);
 var AgentsResponse = class {
   agents;
 };
 __name(AgentsResponse, "AgentsResponse");
-_ts_decorate18([
-  (0, import_type_graphql19.Field)(() => [
+_ts_decorate19([
+  (0, import_type_graphql20.Field)(() => [
     Agent
   ]),
-  _ts_metadata18("design:type", Array)
+  _ts_metadata19("design:type", Array)
 ], AgentsResponse.prototype, "agents", void 0);
-AgentsResponse = _ts_decorate18([
-  (0, import_type_graphql19.ObjectType)()
+AgentsResponse = _ts_decorate19([
+  (0, import_type_graphql20.ObjectType)()
 ], AgentsResponse);
 
 // src/graphql/resolvers/copilot.resolver.ts
 var import_shared29 = require("@copilotkit/shared");
-function _ts_decorate19(decorators, target, key, desc) {
+function _ts_decorate20(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);
@@ -7106,12 +7172,12 @@ function _ts_decorate19(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_decorate19, "_ts_decorate");
-function _ts_metadata19(k, v) {
+__name(_ts_decorate20, "_ts_decorate");
+function _ts_metadata20(k, v) {
   if (typeof Reflect === "object" && typeof Reflect.metadata === "function")
     return Reflect.metadata(k, v);
 }
-__name(_ts_metadata19, "_ts_metadata");
+__name(_ts_metadata20, "_ts_metadata");
 function _ts_param(paramIndex, decorator) {
   return function(target, key) {
     decorator(target, key, paramIndex);
@@ -7260,7 +7326,8 @@ var CopilotResolver = class {
         agentStates: data.agentStates,
         url: data.frontend.url,
         extensions: data.extensions,
-        metaEvents: data.metaEvents
+        metaEvents: data.metaEvents,
+        context: data.context
       });
     } catch (error) {
       if ((0, import_shared29.isStructuredCopilotKitError)(error) || ((_e = error == null ? void 0 : error.extensions) == null ? void 0 : _e.visibility)) {
@@ -7657,38 +7724,38 @@ var CopilotResolver = class {
   }
 };
 __name(CopilotResolver, "CopilotResolver");
-_ts_decorate19([
-  (0, import_type_graphql20.Query)(() => String),
-  _ts_metadata19("design:type", Function),
-  _ts_metadata19("design:paramtypes", []),
-  _ts_metadata19("design:returntype", Promise)
+_ts_decorate20([
+  (0, import_type_graphql21.Query)(() => String),
+  _ts_metadata20("design:type", Function),
+  _ts_metadata20("design:paramtypes", []),
+  _ts_metadata20("design:returntype", Promise)
 ], CopilotResolver.prototype, "hello", null);
-_ts_decorate19([
-  (0, import_type_graphql20.Query)(() => AgentsResponse),
-  _ts_param(0, (0, import_type_graphql20.Ctx)()),
-  _ts_metadata19("design:type", Function),
-  _ts_metadata19("design:paramtypes", [
+_ts_decorate20([
+  (0, import_type_graphql21.Query)(() => AgentsResponse),
+  _ts_param(0, (0, import_type_graphql21.Ctx)()),
+  _ts_metadata20("design:type", Function),
+  _ts_metadata20("design:paramtypes", [
     typeof GraphQLContext === "undefined" ? Object : GraphQLContext
   ]),
-  _ts_metadata19("design:returntype", Promise)
+  _ts_metadata20("design:returntype", Promise)
 ], CopilotResolver.prototype, "availableAgents", null);
-_ts_decorate19([
-  (0, import_type_graphql20.Mutation)(() => CopilotResponse),
-  _ts_param(0, (0, import_type_graphql20.Ctx)()),
-  _ts_param(1, (0, import_type_graphql20.Arg)("data")),
-  _ts_param(2, (0, import_type_graphql20.Arg)("properties", () => import_graphql_scalars2.GraphQLJSONObject, {
+_ts_decorate20([
+  (0, import_type_graphql21.Mutation)(() => CopilotResponse),
+  _ts_param(0, (0, import_type_graphql21.Ctx)()),
+  _ts_param(1, (0, import_type_graphql21.Arg)("data")),
+  _ts_param(2, (0, import_type_graphql21.Arg)("properties", () => import_graphql_scalars2.GraphQLJSONObject, {
     nullable: true
   })),
-  _ts_metadata19("design:type", Function),
-  _ts_metadata19("design:paramtypes", [
+  _ts_metadata20("design:type", Function),
+  _ts_metadata20("design:paramtypes", [
     typeof GraphQLContext === "undefined" ? Object : GraphQLContext,
     typeof GenerateCopilotResponseInput === "undefined" ? Object : GenerateCopilotResponseInput,
     typeof CopilotRequestContextProperties === "undefined" ? Object : CopilotRequestContextProperties
   ]),
-  _ts_metadata19("design:returntype", Promise)
+  _ts_metadata20("design:returntype", Promise)
 ], CopilotResolver.prototype, "generateCopilotResponse", null);
-CopilotResolver = _ts_decorate19([
-  (0, import_type_graphql20.Resolver)(() => CopilotResponse)
+CopilotResolver = _ts_decorate20([
+  (0, import_type_graphql21.Resolver)(() => CopilotResponse)
 ], CopilotResolver);
 
 // src/lib/integrations/shared.ts
@@ -7723,13 +7790,13 @@ function createLogger(options) {
 __name(createLogger, "createLogger");
 
 // src/graphql/resolvers/state.resolver.ts
-var import_type_graphql23 = require("type-graphql");
 var import_type_graphql24 = require("type-graphql");
 var import_type_graphql25 = require("type-graphql");
+var import_type_graphql26 = require("type-graphql");
 
 // src/graphql/types/load-agent-state-response.type.ts
-var import_type_graphql21 = require("type-graphql");
-function _ts_decorate20(decorators, target, key, desc) {
+var import_type_graphql22 = require("type-graphql");
+function _ts_decorate21(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);
@@ -7739,12 +7806,12 @@ function _ts_decorate20(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_decorate20, "_ts_decorate");
-function _ts_metadata20(k, v) {
+__name(_ts_decorate21, "_ts_decorate");
+function _ts_metadata21(k, v) {
   if (typeof Reflect === "object" && typeof Reflect.metadata === "function")
     return Reflect.metadata(k, v);
 }
-__name(_ts_metadata20, "_ts_metadata");
+__name(_ts_metadata21, "_ts_metadata");
 var LoadAgentStateResponse = class {
   threadId;
   threadExists;
@@ -7752,29 +7819,29 @@ var LoadAgentStateResponse = class {
   messages;
 };
 __name(LoadAgentStateResponse, "LoadAgentStateResponse");
-_ts_decorate20([
-  (0, import_type_graphql21.Field)(() => String),
-  _ts_metadata20("design:type", String)
+_ts_decorate21([
+  (0, import_type_graphql22.Field)(() => String),
+  _ts_metadata21("design:type", String)
 ], LoadAgentStateResponse.prototype, "threadId", void 0);
-_ts_decorate20([
-  (0, import_type_graphql21.Field)(() => Boolean),
-  _ts_metadata20("design:type", Boolean)
+_ts_decorate21([
+  (0, import_type_graphql22.Field)(() => Boolean),
+  _ts_metadata21("design:type", Boolean)
 ], LoadAgentStateResponse.prototype, "threadExists", void 0);
-_ts_decorate20([
-  (0, import_type_graphql21.Field)(() => String),
-  _ts_metadata20("design:type", String)
+_ts_decorate21([
+  (0, import_type_graphql22.Field)(() => String),
+  _ts_metadata21("design:type", String)
 ], LoadAgentStateResponse.prototype, "state", void 0);
-_ts_decorate20([
-  (0, import_type_graphql21.Field)(() => String),
-  _ts_metadata20("design:type", String)
+_ts_decorate21([
+  (0, import_type_graphql22.Field)(() => String),
+  _ts_metadata21("design:type", String)
 ], LoadAgentStateResponse.prototype, "messages", void 0);
-LoadAgentStateResponse = _ts_decorate20([
-  (0, import_type_graphql21.ObjectType)()
+LoadAgentStateResponse = _ts_decorate21([
+  (0, import_type_graphql22.ObjectType)()
 ], LoadAgentStateResponse);
 
 // src/graphql/inputs/load-agent-state.input.ts
-var import_type_graphql22 = require("type-graphql");
-function _ts_decorate21(decorators, target, key, desc) {
+var import_type_graphql23 = require("type-graphql");
+function _ts_decorate22(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);
@@ -7784,32 +7851,32 @@ function _ts_decorate21(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_decorate21, "_ts_decorate");
-function _ts_metadata21(k, v) {
+__name(_ts_decorate22, "_ts_decorate");
+function _ts_metadata22(k, v) {
   if (typeof Reflect === "object" && typeof Reflect.metadata === "function")
     return Reflect.metadata(k, v);
 }
-__name(_ts_metadata21, "_ts_metadata");
+__name(_ts_metadata22, "_ts_metadata");
 var LoadAgentStateInput = class {
   threadId;
   agentName;
 };
 __name(LoadAgentStateInput, "LoadAgentStateInput");
-_ts_decorate21([
-  (0, import_type_graphql22.Field)(() => String),
-  _ts_metadata21("design:type", String)
+_ts_decorate22([
+  (0, import_type_graphql23.Field)(() => String),
+  _ts_metadata22("design:type", String)
 ], LoadAgentStateInput.prototype, "threadId", void 0);
-_ts_decorate21([
-  (0, import_type_graphql22.Field)(() => String),
-  _ts_metadata21("design:type", String)
+_ts_decorate22([
+  (0, import_type_graphql23.Field)(() => String),
+  _ts_metadata22("design:type", String)
 ], LoadAgentStateInput.prototype, "agentName", void 0);
-LoadAgentStateInput = _ts_decorate21([
-  (0, import_type_graphql22.InputType)()
+LoadAgentStateInput = _ts_decorate22([
+  (0, import_type_graphql23.InputType)()
 ], LoadAgentStateInput);
 
 // src/graphql/resolvers/state.resolver.ts
 var import_shared30 = require("@copilotkit/shared");
-function _ts_decorate22(decorators, target, key, desc) {
+function _ts_decorate23(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);
@@ -7819,12 +7886,12 @@ function _ts_decorate22(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_decorate22, "_ts_decorate");
-function _ts_metadata22(k, v) {
+__name(_ts_decorate23, "_ts_decorate");
+function _ts_metadata23(k, v) {
   if (typeof Reflect === "object" && typeof Reflect.metadata === "function")
     return Reflect.metadata(k, v);
 }
-__name(_ts_metadata22, "_ts_metadata");
+__name(_ts_metadata23, "_ts_metadata");
 function _ts_param2(paramIndex, decorator) {
   return function(target, key) {
     decorator(target, key, paramIndex);
@@ -7849,19 +7916,19 @@ var StateResolver = class {
   }
 };
 __name(StateResolver, "StateResolver");
-_ts_decorate22([
-  (0, import_type_graphql25.Query)(() => LoadAgentStateResponse),
-  _ts_param2(0, (0, import_type_graphql24.Ctx)()),
-  _ts_param2(1, (0, import_type_graphql23.Arg)("data")),
-  _ts_metadata22("design:type", Function),
-  _ts_metadata22("design:paramtypes", [
+_ts_decorate23([
+  (0, import_type_graphql26.Query)(() => LoadAgentStateResponse),
+  _ts_param2(0, (0, import_type_graphql25.Ctx)()),
+  _ts_param2(1, (0, import_type_graphql24.Arg)("data")),
+  _ts_metadata23("design:type", Function),
+  _ts_metadata23("design:paramtypes", [
     typeof GraphQLContext === "undefined" ? Object : GraphQLContext,
     typeof LoadAgentStateInput === "undefined" ? Object : LoadAgentStateInput
   ]),
-  _ts_metadata22("design:returntype", Promise)
+  _ts_metadata23("design:returntype", Promise)
 ], StateResolver.prototype, "loadAgentState", null);
-StateResolver = _ts_decorate22([
-  (0, import_type_graphql23.Resolver)(() => LoadAgentStateResponse)
+StateResolver = _ts_decorate23([
+  (0, import_type_graphql24.Resolver)(() => LoadAgentStateResponse)
 ], StateResolver);
 
 // src/lib/integrations/shared.ts
@@ -7892,7 +7959,7 @@ async function createContext(initialContext, copilotKitContext, contextLogger, p
 __name(createContext, "createContext");
 function buildSchema(options = {}) {
   logger.debug("Building GraphQL schema...");
-  const schema = (0, import_type_graphql26.buildSchemaSync)({
+  const schema = (0, import_type_graphql27.buildSchemaSync)({
     resolvers: [
       CopilotResolver,
       StateResolver
@@ -8205,12 +8272,22 @@ var LangGraphAgent = class extends import_langgraph.LangGraphAgent {
       return processedEvent;
     }));
   }
-  langGraphDefaultMergeState(state, messages, tools) {
-    const { tools: returnedTools, ...rest } = super.langGraphDefaultMergeState(state, messages, tools);
+  langGraphDefaultMergeState(state, messages, input) {
+    const aguiMergedState = super.langGraphDefaultMergeState(state, messages, input);
+    const { tools: returnedTools, "ag-ui": agui } = aguiMergedState;
+    const rawCombinedTools = [
+      ...returnedTools ?? [],
+      ...(agui == null ? void 0 : agui.tools) ?? []
+    ];
+    const combinedTools = Array.from(new Map(rawCombinedTools.map((t) => [
+      (t == null ? void 0 : t.id) ?? (t == null ? void 0 : t.name) ?? (t == null ? void 0 : t.key) ?? JSON.stringify(t),
+      t
+    ])).values());
     return {
-      ...rest,
+      ...aguiMergedState,
       copilotkit: {
-        actions: returnedTools
+        actions: combinedTools,
+        context: (agui == null ? void 0 : agui.context) ?? []
       }
     };
   }
@@ -8228,6 +8305,10 @@ var LangGraphAgent = class extends import_langgraph.LangGraphAgent {
       output: schemaKeys.output ? [
         ...schemaKeys.output,
         ...CONSTANT_KEYS
+      ] : null,
+      context: schemaKeys.context ? [
+        ...schemaKeys.context,
+        ...CONSTANT_KEYS
       ] : null
     };
   }