@copilotkit/runtime 1.50.0-beta.9 → 1.50.0

package/dist/lib/integrations/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;
 };
@@ -79,14 +79,11 @@ var require_package = __commonJS({
         tsconfig: "workspace:*",
         tsup: "^6.7.0",
         typescript: "^5.2.3",
-        vitest: "^3.2.4",
         "zod-to-json-schema": "^3.23.5"
       },
       dependencies: {
         "@anthropic-ai/sdk": "^0.57.0",
         "@copilotkit/shared": "workspace:*",
-        "@copilotkitnext/agent": "^0.0.17",
-        "@copilotkitnext/runtime": "^0.0.17",
         "@graphql-yoga/plugin-defer-stream": "^3.3.1",
         "@langchain/aws": "^0.1.9",
         "@langchain/community": "^0.3.29",
@@ -156,7 +153,7 @@ var import_type_graphql27 = require("type-graphql");
 
 // src/graphql/resolvers/copilot.resolver.ts
 var import_type_graphql21 = require("type-graphql");
-var import_rxjs2 = require("rxjs");
+var import_rxjs3 = require("rxjs");
 
 // src/graphql/inputs/generate-copilot-response.input.ts
 var import_type_graphql19 = require("type-graphql");
@@ -168,11 +165,11 @@ var import_type_graphql3 = require("type-graphql");
 var import_type_graphql = require("type-graphql");
 var MessageRole;
 (function(MessageRole2) {
+  MessageRole2["user"] = "user";
   MessageRole2["assistant"] = "assistant";
-  MessageRole2["developer"] = "developer";
   MessageRole2["system"] = "system";
   MessageRole2["tool"] = "tool";
-  MessageRole2["user"] = "user";
+  MessageRole2["developer"] = "developer";
 })(MessageRole || (MessageRole = {}));
 var CopilotRequestType;
 (function(CopilotRequestType2) {
@@ -879,10 +876,9 @@ var MessageStatusCode;
 (0, import_type_graphql12.registerEnumType)(MessageStatusCode, {
   name: "MessageStatusCode"
 });
-var BaseMessageStatus = class {
+var BaseMessageStatus = /* @__PURE__ */ __name(class BaseMessageStatus2 {
   code;
-};
-__name(BaseMessageStatus, "BaseMessageStatus");
+}, "BaseMessageStatus");
 _ts_decorate11([
   (0, import_type_graphql12.Field)(() => MessageStatusCode),
   _ts_metadata11("design:type", String)
@@ -1689,24 +1685,13 @@ GenerateCopilotResponseInput = _ts_decorate18([
 var import_graphql_yoga = require("graphql-yoga");
 
 // src/service-adapters/events.ts
-var import_shared2 = require("@copilotkit/shared");
-var import_rxjs = require("rxjs");
+var import_shared5 = require("@copilotkit/shared");
+var import_class_transformer = require("class-transformer");
+var import_rxjs2 = require("rxjs");
 
 // src/graphql/types/converted/index.ts
-var import_shared = require("@copilotkit/shared");
-var Message = class {
+var Message = class extends BaseMessageInput {
   type;
-  id;
-  createdAt;
-  status;
-  constructor(props) {
-    props.id ?? (props.id = (0, import_shared.randomId)());
-    props.status ?? (props.status = {
-      code: MessageStatusCode.Success
-    });
-    props.createdAt ?? (props.createdAt = /* @__PURE__ */ new Date());
-    Object.assign(this, props);
-  }
   isTextMessage() {
     return this.type === "TextMessage";
   }
@@ -1725,14 +1710,10 @@ var Message = class {
 };
 __name(Message, "Message");
 var TextMessage = class extends Message {
+  type = "TextMessage";
   content;
-  parentMessageId;
   role;
-  type = "TextMessage";
-  constructor(props) {
-    super(props);
-    this.type = "TextMessage";
-  }
+  parentMessageId;
 };
 __name(TextMessage, "TextMessage");
 var ActionExecutionMessage = class extends Message {
@@ -1825,80 +1806,17 @@ var AgentStateMessage = class extends Message {
 };
 __name(AgentStateMessage, "AgentStateMessage");
 
-// src/service-adapters/events.ts
-var RuntimeEventTypes;
-(function(RuntimeEventTypes2) {
-  RuntimeEventTypes2["TextMessageStart"] = "TextMessageStart";
-  RuntimeEventTypes2["TextMessageContent"] = "TextMessageContent";
-  RuntimeEventTypes2["TextMessageEnd"] = "TextMessageEnd";
-  RuntimeEventTypes2["ActionExecutionStart"] = "ActionExecutionStart";
-  RuntimeEventTypes2["ActionExecutionArgs"] = "ActionExecutionArgs";
-  RuntimeEventTypes2["ActionExecutionEnd"] = "ActionExecutionEnd";
-  RuntimeEventTypes2["ActionExecutionResult"] = "ActionExecutionResult";
-  RuntimeEventTypes2["AgentStateMessage"] = "AgentStateMessage";
-  RuntimeEventTypes2["MetaEvent"] = "MetaEvent";
-  RuntimeEventTypes2["RunError"] = "RunError";
-})(RuntimeEventTypes || (RuntimeEventTypes = {}));
-var RuntimeMetaEventName;
-(function(RuntimeMetaEventName2) {
-  RuntimeMetaEventName2["LangGraphInterruptEvent"] = "LangGraphInterruptEvent";
-  RuntimeMetaEventName2["LangGraphInterruptResumeEvent"] = "LangGraphInterruptResumeEvent";
-  RuntimeMetaEventName2["CopilotKitLangGraphInterruptEvent"] = "CopilotKitLangGraphInterruptEvent";
-})(RuntimeMetaEventName || (RuntimeMetaEventName = {}));
-
-// src/graphql/resolvers/copilot.resolver.ts
-var import_graphql_scalars2 = require("graphql-scalars");
-var import_class_transformer = require("class-transformer");
-var import_graphql = require("graphql");
-
-// src/utils/failed-response-status-reasons.ts
-var GuardrailsValidationFailureResponse = class extends FailedResponseStatus {
-  reason = FailedResponseStatusReason.GUARDRAILS_VALIDATION_FAILED;
-  constructor({ guardrailsReason }) {
-    super();
-    this.details = {
-      guardrailsReason
-    };
-  }
-};
-__name(GuardrailsValidationFailureResponse, "GuardrailsValidationFailureResponse");
-var MessageStreamInterruptedResponse = class extends FailedResponseStatus {
-  reason = FailedResponseStatusReason.MESSAGE_STREAM_INTERRUPTED;
-  constructor({ messageId }) {
-    super();
-    this.details = {
-      messageId,
-      description: "Check the message for mode details"
-    };
-  }
-};
-__name(MessageStreamInterruptedResponse, "MessageStreamInterruptedResponse");
-var UnknownErrorResponse = class extends FailedResponseStatus {
-  reason = FailedResponseStatusReason.UNKNOWN_ERROR;
-  constructor({ description, originalError }) {
-    super();
-    this.details = {
-      description,
-      originalError
-    };
-  }
-};
-__name(UnknownErrorResponse, "UnknownErrorResponse");
+// src/lib/runtime/remote-actions.ts
+var import_shared3 = require("@copilotkit/shared");
 
 // src/lib/telemetry-client.ts
-var import_shared4 = require("@copilotkit/shared");
+var import_shared2 = require("@copilotkit/shared");
 var import_node_crypto = require("crypto");
 
 // src/lib/runtime/copilot-runtime.ts
-var import_shared3 = require("@copilotkit/shared");
-var import_runtime22 = require("@copilotkitnext/runtime");
-
-// src/lib/runtime/types.ts
-var EndpointType;
-(function(EndpointType2) {
-  EndpointType2["CopilotKit"] = "copilotKit";
-  EndpointType2["LangGraphPlatform"] = "langgraph-platform";
-})(EndpointType || (EndpointType = {}));
+var import_shared = require("@copilotkit/shared");
+var import_rxjs = require("rxjs");
+var import_langgraph_sdk = require("@langchain/langgraph-sdk");
 
 // src/agents/langgraph/events.ts
 var LangGraphEventTypes;
@@ -1933,7 +1851,6 @@ var CustomEventNames;
 })(CustomEventNames || (CustomEventNames = {}));
 
 // src/lib/runtime/copilot-runtime.ts
-var import_agent = require("@copilotkitnext/agent");
 function resolveEndpointType(endpoint) {
   if (!endpoint.type) {
     if ("deploymentUrl" in endpoint && "agents" in endpoint) {
@@ -1948,15 +1865,14 @@ __name(resolveEndpointType, "resolveEndpointType");
 
 // src/lib/telemetry-client.ts
 var packageJson = require_package();
-var telemetryClient = new import_shared4.TelemetryClient({
+var telemetryClient = new import_shared2.TelemetryClient({
   packageName: packageJson.name,
   packageVersion: packageJson.version
 });
 function getRuntimeInstanceTelemetryInfo(options) {
-  var _a, _b, _c, _d, _e;
+  var _a, _b;
   const runtime = options.runtime;
-  const remoteEndpoints = ((_a = runtime.params) == null ? void 0 : _a.remoteEndpoints) ?? [];
-  const endpointsInfo = remoteEndpoints.reduce((acc, endpoint) => {
+  const endpointsInfo = runtime.remoteEndpointDefinitions.reduce((acc, endpoint) => {
     let info = {
       ...acc
     };
@@ -1984,19 +1900,19 @@ function getRuntimeInstanceTelemetryInfo(options) {
     agentsAmount: null,
     hashedKey: null
   });
-  const publicApiKey = (_b = options.cloud) == null ? void 0 : _b.publicApiKey;
+  const publicApiKey = (_a = options.cloud) == null ? void 0 : _a.publicApiKey;
   const apiKeyProvided = !!publicApiKey && publicApiKey.trim().length > 0;
   return {
-    actionsAmount: ((_d = (_c = runtime.params) == null ? void 0 : _c.actions) == null ? void 0 : _d.length) ?? 0,
-    endpointsAmount: remoteEndpoints.length,
+    actionsAmount: runtime.actions.length,
+    endpointsAmount: runtime.remoteEndpointDefinitions.length,
     endpointTypes: endpointsInfo.endpointTypes,
-    agentsAmount: Object.keys(runtime.instance.agents).length,
+    agentsAmount: endpointsInfo.agentsAmount,
     hashedLgcKey: endpointsInfo.hashedKey,
     "cloud.api_key_provided": apiKeyProvided,
     ...apiKeyProvided ? {
       "cloud.public_api_key": publicApiKey
     } : {},
-    ...((_e = options.cloud) == null ? void 0 : _e.baseUrl) ? {
+    ...((_b = options.cloud) == null ? void 0 : _b.baseUrl) ? {
       "cloud.base_url": options.cloud.baseUrl
     } : {}
   };
@@ -2004,8 +1920,76 @@ function getRuntimeInstanceTelemetryInfo(options) {
 __name(getRuntimeInstanceTelemetryInfo, "getRuntimeInstanceTelemetryInfo");
 var telemetry_client_default = telemetryClient;
 
+// src/lib/runtime/remote-actions.ts
+var import_shared4 = require("@copilotkit/shared");
+var EndpointType;
+(function(EndpointType2) {
+  EndpointType2["CopilotKit"] = "copilotKit";
+  EndpointType2["LangGraphPlatform"] = "langgraph-platform";
+})(EndpointType || (EndpointType = {}));
+
+// src/service-adapters/events.ts
+var RuntimeEventTypes;
+(function(RuntimeEventTypes2) {
+  RuntimeEventTypes2["TextMessageStart"] = "TextMessageStart";
+  RuntimeEventTypes2["TextMessageContent"] = "TextMessageContent";
+  RuntimeEventTypes2["TextMessageEnd"] = "TextMessageEnd";
+  RuntimeEventTypes2["ActionExecutionStart"] = "ActionExecutionStart";
+  RuntimeEventTypes2["ActionExecutionArgs"] = "ActionExecutionArgs";
+  RuntimeEventTypes2["ActionExecutionEnd"] = "ActionExecutionEnd";
+  RuntimeEventTypes2["ActionExecutionResult"] = "ActionExecutionResult";
+  RuntimeEventTypes2["AgentStateMessage"] = "AgentStateMessage";
+  RuntimeEventTypes2["MetaEvent"] = "MetaEvent";
+  RuntimeEventTypes2["RunError"] = "RunError";
+})(RuntimeEventTypes || (RuntimeEventTypes = {}));
+var RuntimeMetaEventName;
+(function(RuntimeMetaEventName2) {
+  RuntimeMetaEventName2["LangGraphInterruptEvent"] = "LangGraphInterruptEvent";
+  RuntimeMetaEventName2["LangGraphInterruptResumeEvent"] = "LangGraphInterruptResumeEvent";
+  RuntimeMetaEventName2["CopilotKitLangGraphInterruptEvent"] = "CopilotKitLangGraphInterruptEvent";
+})(RuntimeMetaEventName || (RuntimeMetaEventName = {}));
+
 // src/graphql/resolvers/copilot.resolver.ts
-var import_shared5 = require("@copilotkit/shared");
+var import_graphql_scalars2 = require("graphql-scalars");
+var import_class_transformer2 = require("class-transformer");
+var import_graphql = require("graphql");
+
+// src/utils/failed-response-status-reasons.ts
+var GuardrailsValidationFailureResponse = class extends FailedResponseStatus {
+  reason = FailedResponseStatusReason.GUARDRAILS_VALIDATION_FAILED;
+  constructor({ guardrailsReason }) {
+    super();
+    this.details = {
+      guardrailsReason
+    };
+  }
+};
+__name(GuardrailsValidationFailureResponse, "GuardrailsValidationFailureResponse");
+var MessageStreamInterruptedResponse = class extends FailedResponseStatus {
+  reason = FailedResponseStatusReason.MESSAGE_STREAM_INTERRUPTED;
+  constructor({ messageId }) {
+    super();
+    this.details = {
+      messageId,
+      description: "Check the message for mode details"
+    };
+  }
+};
+__name(MessageStreamInterruptedResponse, "MessageStreamInterruptedResponse");
+var UnknownErrorResponse = class extends FailedResponseStatus {
+  reason = FailedResponseStatusReason.UNKNOWN_ERROR;
+  constructor({ description, originalError }) {
+    super();
+    this.details = {
+      description,
+      originalError
+    };
+  }
+};
+__name(UnknownErrorResponse, "UnknownErrorResponse");
+
+// src/graphql/resolvers/copilot.resolver.ts
+var import_shared6 = require("@copilotkit/shared");
 
 // src/graphql/types/agents-response.type.ts
 var import_type_graphql20 = require("type-graphql");
@@ -2061,7 +2045,7 @@ AgentsResponse = _ts_decorate19([
 ], AgentsResponse);
 
 // src/graphql/resolvers/copilot.resolver.ts
-var import_shared6 = require("@copilotkit/shared");
+var import_shared7 = require("@copilotkit/shared");
 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")
@@ -2124,14 +2108,14 @@ var CopilotResolver = class {
       component: "CopilotResolver.availableAgents"
     });
     logger2.debug("Processing");
-    const agentsWithEndpoints = [];
+    const agentsWithEndpoints = await ctx._copilotkit.runtime.discoverAgentsFromEndpoints(ctx);
     logger2.debug("Event source created, creating response");
     return {
       agents: agentsWithEndpoints.map(({ endpoint, ...agentWithoutEndpoint }) => agentWithoutEndpoint)
     };
   }
   async generateCopilotResponse(ctx, data, properties) {
-    var _a, _b, _c, _d;
+    var _a, _b, _c, _d, _e, _f, _g;
     telemetry_client_default.capture("oss.runtime.copilot_request_created", {
       "cloud.guardrails.enabled": ((_a = data.cloud) == null ? void 0 : _a.guardrails) !== void 0,
       requestType: data.metadata.requestType,
@@ -2173,12 +2157,21 @@ var CopilotResolver = class {
       logger2.debug("Cloud configuration provided, checking for public API key in headers");
       if (!copilotCloudPublicApiKey) {
         logger2.error("Public API key not found in headers");
+        await copilotRuntime.errorGraphQLError({
+          message: "X-CopilotCloud-Public-API-Key header is required",
+          code: "MISSING_PUBLIC_API_KEY",
+          type: "GraphQLError"
+        }, {
+          operation: "generateCopilotResponse",
+          cloudConfigPresent: Boolean(data.cloud),
+          guardrailsEnabled: Boolean((_b = data.cloud) == null ? void 0 : _b.guardrails)
+        });
         throw new import_graphql.GraphQLError("X-CopilotCloud-Public-API-Key header is required");
       }
       if (process.env.COPILOT_CLOUD_BASE_URL) {
         copilotCloudBaseUrl = process.env.COPILOT_CLOUD_BASE_URL;
-      } else if ((_b = ctx._copilotkit.cloud) == null ? void 0 : _b.baseUrl) {
-        copilotCloudBaseUrl = (_c = ctx._copilotkit.cloud) == null ? void 0 : _c.baseUrl;
+      } else if ((_c = ctx._copilotkit.cloud) == null ? void 0 : _c.baseUrl) {
+        copilotCloudBaseUrl = (_d = ctx._copilotkit.cloud) == null ? void 0 : _d.baseUrl;
       } else {
         copilotCloudBaseUrl = "https://api.cloud.copilotkit.ai";
       }
@@ -2187,9 +2180,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_rxjs3.ReplaySubject();
+    const interruptStreaming$ = new import_rxjs3.ReplaySubject();
+    const guardrailsResult$ = new import_rxjs3.ReplaySubject();
     let outputMessages = [];
     let resolveOutputMessagesPromise;
     let rejectOutputMessagesPromise;
@@ -2202,11 +2195,41 @@ var CopilotResolver = class {
     }
     logger2.debug("Processing");
     let runtimeResponse;
-    const { eventSource, threadId = (0, import_shared5.randomId)(), runId, serverSideActions, actionInputsWithoutAgents, extensions } = runtimeResponse;
+    try {
+      runtimeResponse = await copilotRuntime.processRuntimeRequest({
+        serviceAdapter,
+        messages: data.messages,
+        actions: data.frontend.actions.filter((action) => action.available !== ActionInputAvailability.disabled),
+        threadId: data.threadId,
+        runId: data.runId,
+        publicApiKey: copilotCloudPublicApiKey,
+        outputMessagesPromise,
+        graphqlContext: ctx,
+        forwardedParameters: data.forwardedParameters,
+        agentSession: data.agentSession,
+        agentStates: data.agentStates,
+        url: data.frontend.url,
+        extensions: data.extensions,
+        metaEvents: data.metaEvents,
+        context: data.context
+      });
+    } catch (error) {
+      if ((0, import_shared7.isStructuredCopilotKitError)(error) || ((_e = error == null ? void 0 : error.extensions) == null ? void 0 : _e.visibility)) {
+        throw new import_graphql.GraphQLError(error.message || "Agent error occurred", {
+          extensions: {
+            ...error.extensions,
+            code: error.code || ((_f = error.extensions) == null ? void 0 : _f.code) || "AGENT_ERROR",
+            originalError: error
+          }
+        });
+      }
+      throw error;
+    }
+    const { eventSource, threadId = (0, import_shared6.randomId)(), runId, serverSideActions, actionInputsWithoutAgents, extensions } = runtimeResponse;
     logger2.debug("Event source created, creating response");
     const eventStream = eventSource.processRuntimeEvents({
       serverSideActions,
-      guardrailsResult$: ((_d = data.cloud) == null ? void 0 : _d.guardrails) ? guardrailsResult$ : null,
+      guardrailsResult$: ((_g = data.cloud) == null ? void 0 : _g.guardrails) ? guardrailsResult$ : null,
       actionInputsWithoutAgents: actionInputsWithoutAgents.filter(
         // TODO-AGENTS: do not exclude ALL server side actions
         (action) => !serverSideActions.find((serverSideAction) => serverSideAction.name == action.name)
@@ -2215,15 +2238,15 @@ var CopilotResolver = class {
     }).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_rxjs3.shareReplay)(),
+      (0, import_rxjs3.finalize)(() => {
         logger2.debug("Event stream finalized");
       })
     );
     const response = {
       threadId,
       runId,
-      status: (0, import_rxjs2.firstValueFrom)(responseStatus$),
+      status: (0, import_rxjs3.firstValueFrom)(responseStatus$),
       extensions,
       metaEvents: new import_graphql_yoga.Repeater(async (push, stop) => {
         let eventStreamSubscription;
@@ -2234,7 +2257,7 @@ var CopilotResolver = class {
             }
             switch (event.name) {
               case LangGraphEventTypes.OnInterrupt:
-                push((0, import_class_transformer.plainToInstance)(LangGraphInterruptEvent, {
+                push((0, import_class_transformer2.plainToInstance)(LangGraphInterruptEvent, {
                   // @ts-ignore
                   type: event.type,
                   // @ts-ignore
@@ -2244,21 +2267,21 @@ var CopilotResolver = class {
                 }));
                 break;
               case RuntimeMetaEventName.LangGraphInterruptEvent:
-                push((0, import_class_transformer.plainToInstance)(LangGraphInterruptEvent, {
+                push((0, import_class_transformer2.plainToInstance)(LangGraphInterruptEvent, {
                   type: event.type,
                   name: event.name,
                   value: event.value
                 }));
                 break;
               case RuntimeMetaEventName.CopilotKitLangGraphInterruptEvent:
-                push((0, import_class_transformer.plainToInstance)(CopilotKitLangGraphInterruptEvent, {
+                push((0, import_class_transformer2.plainToInstance)(CopilotKitLangGraphInterruptEvent, {
                   type: event.type,
                   name: event.name,
                   data: {
                     value: event.data.value,
                     messages: event.data.messages.map((message) => {
                       if (message.type === "TextMessage" || "content" in message && "role" in message) {
-                        return (0, import_class_transformer.plainToInstance)(TextMessage, {
+                        return (0, import_class_transformer2.plainToInstance)(TextMessage, {
                           id: message.id,
                           createdAt: /* @__PURE__ */ new Date(),
                           content: [
@@ -2269,7 +2292,7 @@ var CopilotResolver = class {
                         });
                       }
                       if ("arguments" in message) {
-                        return (0, import_class_transformer.plainToInstance)(ActionExecutionMessage, {
+                        return (0, import_class_transformer2.plainToInstance)(ActionExecutionMessage, {
                           name: message.name,
                           id: message.id,
                           arguments: [
@@ -2333,8 +2356,8 @@ var CopilotResolver = class {
                   reason: `Interrupted due to Guardrails validation failure. Reason: ${result.reason}`
                 });
                 outputMessages = [
-                  (0, import_class_transformer.plainToInstance)(TextMessage, {
-                    id: (0, import_shared5.randomId)(),
+                  (0, import_class_transformer2.plainToInstance)(TextMessage, {
+                    id: (0, import_shared6.randomId)(),
                     createdAt: /* @__PURE__ */ new Date(),
                     content: result.reason,
                     role: MessageRole.assistant
@@ -2367,30 +2390,30 @@ 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_rxjs3.skipWhile)((e) => e !== event),
                   // take until the message end event
-                  (0, import_rxjs2.takeWhile)((e) => !(e.type === RuntimeEventTypes.TextMessageEnd && e.messageId == event.messageId)),
+                  (0, import_rxjs3.takeWhile)((e) => !(e.type === RuntimeEventTypes.TextMessageEnd && e.messageId == event.messageId)),
                   // filter out any other message events or message ids
-                  (0, import_rxjs2.filter)((e) => e.type == RuntimeEventTypes.TextMessageContent && e.messageId == event.messageId)
+                  (0, import_rxjs3.filter)((e) => e.type == RuntimeEventTypes.TextMessageContent && e.messageId == event.messageId)
                 );
-                const streamingTextStatus = new import_rxjs2.Subject();
+                const streamingTextStatus = new import_rxjs3.Subject();
                 const messageId = event.messageId;
                 pushMessage({
                   id: messageId,
                   parentMessageId: event.parentMessageId,
-                  status: (0, import_rxjs2.firstValueFrom)(streamingTextStatus),
+                  status: (0, import_rxjs3.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_rxjs3.shareReplay)(), (0, import_rxjs3.take)(1), (0, import_rxjs3.tap)(({ reason, messageId: messageId2 }) => {
                       logger2.debug({
                         reason,
                         messageId: messageId2
                       }, "Text streaming interrupted");
-                      streamingTextStatus.next((0, import_class_transformer.plainToInstance)(FailedMessageStatus, {
+                      streamingTextStatus.next((0, import_class_transformer2.plainToInstance)(FailedMessageStatus, {
                         reason
                       }));
                       responseStatus$.next(new MessageStreamInterruptedResponse({
@@ -2423,7 +2446,7 @@ var CopilotResolver = class {
                         streamingTextStatus.next(new SuccessMessageStatus());
                         stopStreamingText();
                         textSubscription == null ? void 0 : textSubscription.unsubscribe();
-                        outputMessages.push((0, import_class_transformer.plainToInstance)(TextMessage, {
+                        outputMessages.push((0, import_class_transformer2.plainToInstance)(TextMessage, {
                           id: messageId,
                           createdAt: /* @__PURE__ */ new Date(),
                           content: textChunks.join(""),
@@ -2437,17 +2460,17 @@ var CopilotResolver = class {
               case RuntimeEventTypes.ActionExecutionStart:
                 logger2.debug("Action execution start event received");
                 const actionExecutionArgumentStream = eventStream.pipe(
-                  (0, import_rxjs2.skipWhile)((e) => e !== event),
+                  (0, import_rxjs3.skipWhile)((e) => e !== event),
                   // take until the action execution end event
-                  (0, import_rxjs2.takeWhile)((e) => !(e.type === RuntimeEventTypes.ActionExecutionEnd && e.actionExecutionId == event.actionExecutionId)),
+                  (0, import_rxjs3.takeWhile)((e) => !(e.type === RuntimeEventTypes.ActionExecutionEnd && e.actionExecutionId == event.actionExecutionId)),
                   // filter out any other action execution events or action execution ids
-                  (0, import_rxjs2.filter)((e) => e.type == RuntimeEventTypes.ActionExecutionArgs && e.actionExecutionId == event.actionExecutionId)
+                  (0, import_rxjs3.filter)((e) => e.type == RuntimeEventTypes.ActionExecutionArgs && e.actionExecutionId == event.actionExecutionId)
                 );
-                const streamingArgumentsStatus = new import_rxjs2.Subject();
+                const streamingArgumentsStatus = new import_rxjs3.Subject();
                 pushMessage({
                   id: event.actionExecutionId,
                   parentMessageId: event.parentMessageId,
-                  status: (0, import_rxjs2.firstValueFrom)(streamingArgumentsStatus),
+                  status: (0, import_rxjs3.firstValueFrom)(streamingArgumentsStatus),
                   createdAt: /* @__PURE__ */ new Date(),
                   name: event.actionName,
                   arguments: new import_graphql_yoga.Repeater(async (pushArgumentsChunk, stopStreamingArguments) => {
@@ -2465,7 +2488,7 @@ var CopilotResolver = class {
                         logger2.error({
                           err
                         }, "Error in action execution argument stream");
-                        streamingArgumentsStatus.next((0, import_class_transformer.plainToInstance)(FailedMessageStatus, {
+                        streamingArgumentsStatus.next((0, import_class_transformer2.plainToInstance)(FailedMessageStatus, {
                           reason: "An unknown error has occurred in the action execution argument stream"
                         }));
                         stopStreamingArguments();
@@ -2476,7 +2499,7 @@ var CopilotResolver = class {
                         streamingArgumentsStatus.next(new SuccessMessageStatus());
                         stopStreamingArguments();
                         actionExecutionArgumentSubscription == null ? void 0 : actionExecutionArgumentSubscription.unsubscribe();
-                        outputMessages.push((0, import_class_transformer.plainToInstance)(ActionExecutionMessage, {
+                        outputMessages.push((0, import_class_transformer2.plainToInstance)(ActionExecutionMessage, {
                           id: event.actionExecutionId,
                           createdAt: /* @__PURE__ */ new Date(),
                           name: event.actionName,
@@ -2499,7 +2522,7 @@ var CopilotResolver = class {
                   actionName: event.actionName,
                   result: event.result
                 });
-                outputMessages.push((0, import_class_transformer.plainToInstance)(ResultMessage, {
+                outputMessages.push((0, import_class_transformer2.plainToInstance)(ResultMessage, {
                   id: "result-" + event.actionExecutionId,
                   createdAt: /* @__PURE__ */ new Date(),
                   actionExecutionId: event.actionExecutionId,
@@ -2512,7 +2535,7 @@ var CopilotResolver = class {
                   event
                 }, "Agent message event received");
                 pushMessage({
-                  id: (0, import_shared5.randomId)(),
+                  id: (0, import_shared6.randomId)(),
                   status: new SuccessMessageStatus(),
                   threadId: event.threadId,
                   agentName: event.agentName,
@@ -2524,8 +2547,8 @@ var CopilotResolver = class {
                   role: MessageRole.assistant,
                   createdAt: /* @__PURE__ */ new Date()
                 });
-                outputMessages.push((0, import_class_transformer.plainToInstance)(AgentStateMessage, {
-                  id: (0, import_shared5.randomId)(),
+                outputMessages.push((0, import_class_transformer2.plainToInstance)(AgentStateMessage, {
+                  id: (0, import_shared6.randomId)(),
                   threadId: event.threadId,
                   agentName: event.agentName,
                   nodeName: event.nodeName,
@@ -2540,8 +2563,8 @@ var CopilotResolver = class {
             }
           },
           error: (err) => {
-            var _a3, _b2, _c2, _d2, _e, _f;
-            if (err instanceof import_shared6.CopilotKitError || err instanceof import_shared6.CopilotKitLowLevelError || err instanceof Error && err.name && err.name.includes("CopilotKit") || ((_a3 = err == null ? void 0 : err.extensions) == null ? void 0 : _a3.visibility)) {
+            var _a3, _b2, _c2, _d2, _e2, _f2;
+            if (err instanceof import_shared7.CopilotKitError || err instanceof import_shared7.CopilotKitLowLevelError || err instanceof Error && err.name && err.name.includes("CopilotKit") || ((_a3 = err == null ? void 0 : err.extensions) == null ? void 0 : _a3.visibility)) {
               responseStatus$.next(new UnknownErrorResponse({
                 description: err.message || "Agent error occurred",
                 // Include original error information for frontend to extract
@@ -2549,8 +2572,8 @@ var CopilotResolver = class {
                   code: err.code || ((_b2 = err.extensions) == null ? void 0 : _b2.code),
                   statusCode: err.statusCode || ((_c2 = err.extensions) == null ? void 0 : _c2.statusCode),
                   severity: err.severity || ((_d2 = err.extensions) == null ? void 0 : _d2.severity),
-                  visibility: err.visibility || ((_e = err.extensions) == null ? void 0 : _e.visibility),
-                  originalErrorType: err.originalErrorType || ((_f = err.extensions) == null ? void 0 : _f.originalErrorType),
+                  visibility: err.visibility || ((_e2 = err.extensions) == null ? void 0 : _e2.visibility),
+                  originalErrorType: err.originalErrorType || ((_f2 = err.extensions) == null ? void 0 : _f2.originalErrorType),
                   extensions: err.extensions
                 }
               }));
@@ -2571,7 +2594,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_rxjs3.firstValueFrom)(guardrailsResult$);
             }
             responseStatus$.next(new SuccessResponseStatus());
             eventStreamSubscription == null ? void 0 : eventStreamSubscription.unsubscribe();
@@ -2736,7 +2759,7 @@ LoadAgentStateInput = _ts_decorate22([
 ], LoadAgentStateInput);
 
 // src/graphql/resolvers/state.resolver.ts
-var import_shared7 = require("@copilotkit/shared");
+var import_shared8 = require("@copilotkit/shared");
 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")
@@ -2761,10 +2784,10 @@ function _ts_param2(paramIndex, decorator) {
 __name(_ts_param2, "_ts_param");
 var StateResolver = class {
   async loadAgentState(ctx, data) {
-    const agents = [];
+    const agents = await ctx._copilotkit.runtime.getAllAgents(ctx);
     const hasAgent = agents.some((agent) => agent.name === data.agentName);
     if (!hasAgent) {
-      throw new import_shared7.CopilotKitAgentDiscoveryError({
+      throw new import_shared8.CopilotKitAgentDiscoveryError({
         agentName: data.agentName,
         availableAgents: agents.map((a) => ({
           name: a.name,
@@ -2772,7 +2795,7 @@ var StateResolver = class {
         }))
       });
     }
-    const state = {};
+    const state = await ctx._copilotkit.runtime.loadAgentState(ctx, data.threadId, data.agentName);
     return state;
   }
 };
@@ -2794,7 +2817,7 @@ StateResolver = _ts_decorate23([
 
 // src/lib/integrations/shared.ts
 var packageJson2 = __toESM(require_package());
-var import_shared8 = require("@copilotkit/shared");
+var import_shared9 = require("@copilotkit/shared");
 var logger = createLogger();
 var addCustomHeaderPlugin = {
   onResponse({ response }) {
@@ -2860,11 +2883,11 @@ function getCommonConfig(options) {
     }
   });
   const userErrorCodes = [
-    import_shared8.CopilotKitErrorCode.AGENT_NOT_FOUND,
-    import_shared8.CopilotKitErrorCode.API_NOT_FOUND,
-    import_shared8.CopilotKitErrorCode.REMOTE_ENDPOINT_NOT_FOUND,
-    import_shared8.CopilotKitErrorCode.CONFIGURATION_ERROR,
-    import_shared8.CopilotKitErrorCode.MISSING_PUBLIC_API_KEY_ERROR
+    import_shared9.CopilotKitErrorCode.AGENT_NOT_FOUND,
+    import_shared9.CopilotKitErrorCode.API_NOT_FOUND,
+    import_shared9.CopilotKitErrorCode.REMOTE_ENDPOINT_NOT_FOUND,
+    import_shared9.CopilotKitErrorCode.CONFIGURATION_ERROR,
+    import_shared9.CopilotKitErrorCode.MISSING_PUBLIC_API_KEY_ERROR
   ];
   return {
     logging: createLogger({
@@ -2887,7 +2910,7 @@ function getCommonConfig(options) {
           console.debug("User configuration error:", error.message);
           return error;
         }
-        if (originalError instanceof import_shared8.CopilotKitError && userErrorCodes.includes(originalError.code)) {
+        if (originalError instanceof import_shared9.CopilotKitError && userErrorCodes.includes(originalError.code)) {
           console.debug("User configuration error:", error.message);
           return error;
         }
@@ -2900,7 +2923,7 @@ function getCommonConfig(options) {
 __name(getCommonConfig, "getCommonConfig");
 
 // src/lib/integrations/nextjs/app-router.ts
-var import_runtime32 = require("@copilotkitnext/runtime");
+var import_graphql_yoga2 = require("graphql-yoga");
 function copilotRuntimeNextJSAppRouterEndpoint(options) {
   var _a;
   const commonConfig = getCommonConfig(options);
@@ -2917,17 +2940,24 @@ function copilotRuntimeNextJSAppRouterEndpoint(options) {
   telemetry_client_default.capture("oss.runtime.instance_created", getRuntimeInstanceTelemetryInfo(options));
   const logger2 = commonConfig.logging;
   logger2.debug("Creating NextJS App Router endpoint");
-  const serviceAdapter = options.serviceAdapter;
-  options.runtime.handleServiceAdapter(serviceAdapter);
-  return (0, import_runtime32.createCopilotEndpoint)({
-    runtime: options.runtime.instance,
-    basePath: options.endpoint
+  const yoga = (0, import_graphql_yoga2.createYoga)({
+    ...commonConfig,
+    graphqlEndpoint: options.endpoint,
+    fetchAPI: {
+      Response: globalThis.Response
+    }
   });
+  return {
+    handleRequest: yoga,
+    GET: yoga,
+    POST: yoga,
+    OPTIONS: yoga
+  };
 }
 __name(copilotRuntimeNextJSAppRouterEndpoint, "copilotRuntimeNextJSAppRouterEndpoint");
 
 // src/lib/integrations/nextjs/pages-router.ts
-var import_runtime34 = require("@copilotkitnext/runtime");
+var import_graphql_yoga3 = require("graphql-yoga");
 var config = {
   api: {
     bodyParser: false
@@ -2949,17 +2979,16 @@ function copilotRuntimeNextJSPagesRouterEndpoint(options) {
   telemetry_client_default.capture("oss.runtime.instance_created", getRuntimeInstanceTelemetryInfo(options));
   const logger2 = commonConfig.logging;
   logger2.debug("Creating NextJS Pages Router endpoint");
-  const serviceAdapter = options.serviceAdapter;
-  options.runtime.handleServiceAdapter(serviceAdapter);
-  return (0, import_runtime34.createCopilotEndpoint)({
-    runtime: options.runtime.instance,
-    basePath: options.baseUrl
+  const yoga = (0, import_graphql_yoga3.createYoga)({
+    ...commonConfig,
+    graphqlEndpoint: options.endpoint
   });
+  return yoga;
 }
 __name(copilotRuntimeNextJSPagesRouterEndpoint, "copilotRuntimeNextJSPagesRouterEndpoint");
 
 // src/lib/integrations/node-http/index.ts
-var import_runtime36 = require("@copilotkitnext/runtime");
+var import_graphql_yoga4 = require("graphql-yoga");
 function copilotRuntimeNodeHttpEndpoint(options) {
   var _a;
   const commonConfig = getCommonConfig(options);
@@ -2976,12 +3005,11 @@ function copilotRuntimeNodeHttpEndpoint(options) {
   telemetry_client_default.capture("oss.runtime.instance_created", getRuntimeInstanceTelemetryInfo(options));
   const logger2 = commonConfig.logging;
   logger2.debug("Creating Node HTTP endpoint");
-  const serviceAdapter = options.serviceAdapter;
-  options.runtime.handleServiceAdapter(serviceAdapter);
-  return (0, import_runtime36.createCopilotEndpoint)({
-    runtime: options.runtime.instance,
-    basePath: options.baseUrl
+  const yoga = (0, import_graphql_yoga4.createYoga)({
+    ...commonConfig,
+    graphqlEndpoint: options.endpoint
   });
+  return yoga;
 }
 __name(copilotRuntimeNodeHttpEndpoint, "copilotRuntimeNodeHttpEndpoint");