@copilotkit/runtime 1.5.12-next.5 → 1.5.12-next.7

package/dist/index.js

@@ -44,7 +44,7 @@ var require_package = __commonJS({
       publishConfig: {
         access: "public"
       },
-      version: "1.5.12-next.5",
+      version: "1.5.12-next.7",
       sideEffects: false,
       main: "./dist/index.js",
       module: "./dist/index.mjs",
@@ -157,7 +157,7 @@ module.exports = __toCommonJS(src_exports);
 var import_reflect_metadata = require("reflect-metadata");
 
 // src/lib/runtime/copilot-runtime.ts
-var import_shared12 = require("@copilotkit/shared");
+var import_shared15 = require("@copilotkit/shared");
 
 // src/service-adapters/langchain/langserve.ts
 var import_remote = require("langchain/runnables/remote");
@@ -397,8 +397,9 @@ var OpenAIAdapter = class {
     this.disableParallelToolCalls = (params == null ? void 0 : params.disableParallelToolCalls) || false;
   }
   async process(request) {
-    const { threadId, model = this.model, messages, actions, eventSource, forwardedParameters } = request;
+    const { threadId: threadIdFromRequest, model = this.model, messages, actions, eventSource, forwardedParameters } = request;
     const tools = actions.map(convertActionInputToOpenAITool);
+    const threadId = threadIdFromRequest ?? (0, import_shared.randomUUID)();
     let openaiMessages = messages.map(convertMessageToOpenAIMessage);
     openaiMessages = limitMessagesToTokenCount(openaiMessages, tools, model);
     let toolChoice = forwardedParameters == null ? void 0 : forwardedParameters.toolChoice;
@@ -496,7 +497,7 @@ var OpenAIAdapter = class {
       eventStream$.complete();
     });
     return {
-      threadId: threadId || (0, import_shared.randomId)()
+      threadId
     };
   }
 };
@@ -742,8 +743,8 @@ var LangChainAdapter = class {
   }
   async process(request) {
     try {
-      const { eventSource, model, actions, messages, runId } = request;
-      const threadId = request.threadId ?? (0, import_shared3.randomId)();
+      const { eventSource, model, actions, messages, runId, threadId: threadIdFromRequest } = request;
+      const threadId = threadIdFromRequest ?? (0, import_shared3.randomUUID)();
       const result = await this.options.chainFn({
         messages: messages.map(convertMessageToLangChainMessage),
         tools: actions.map(convertActionInputToLangChainTool),
@@ -804,8 +805,12 @@ var OpenAIAssistantAdapter = class {
     this.disableParallelToolCalls = (params == null ? void 0 : params.disableParallelToolCalls) || false;
   }
   async process(request) {
+    var _a, _b;
     const { messages, actions, eventSource, runId, forwardedParameters } = request;
-    let threadId = request.threadId || (await this.openai.beta.threads.create()).id;
+    let threadId = (_b = (_a = request.extensions) == null ? void 0 : _a.openaiAssistantAPI) == null ? void 0 : _b.threadId;
+    if (!threadId) {
+      threadId = (await this.openai.beta.threads.create()).id;
+    }
     const lastMessage = messages.at(-1);
     let nextRunId = void 0;
     if (lastMessage.isResultMessage() && runId) {
@@ -816,8 +821,15 @@ var OpenAIAssistantAdapter = class {
       throw new Error("No actionable message found in the messages");
     }
     return {
+      runId: nextRunId,
       threadId,
-      runId: nextRunId
+      extensions: {
+        ...request.extensions,
+        openaiAssistantAPI: {
+          threadId,
+          runId: nextRunId
+        }
+      }
     };
   }
   async submitToolOutputs(threadId, runId, messages, eventSource) {
@@ -1085,7 +1097,7 @@ var UnifyAdapter = class {
       eventStream$.complete();
     });
     return {
-      threadId: request.threadId || (0, import_shared4.randomId)()
+      threadId: request.threadId || (0, import_shared4.randomUUID)()
     };
   }
 };
@@ -1206,7 +1218,7 @@ var GroqAdapter = class {
       eventStream$.complete();
     });
     return {
-      threadId: threadId || (0, import_shared5.randomId)()
+      threadId: request.threadId || (0, import_shared5.randomUUID)()
     };
   }
 };
@@ -1457,7 +1469,7 @@ var AnthropicAdapter = class {
       eventStream$.complete();
     });
     return {
-      threadId: threadId || (0, import_shared6.randomId)()
+      threadId: threadId || (0, import_shared6.randomUUID)()
     };
   }
 };
@@ -1532,7 +1544,7 @@ var ExperimentalOllamaAdapter = class {
       eventStream$.complete();
     });
     return {
-      threadId: request.threadId || (0, import_shared7.randomId)()
+      threadId: request.threadId || (0, import_shared7.randomUUID)()
     };
   }
 };
@@ -1543,7 +1555,7 @@ var import_shared8 = require("@copilotkit/shared");
 var EmptyAdapter = class {
   async process(request) {
     return {
-      threadId: request.threadId || (0, import_shared8.randomId)()
+      threadId: request.threadId || (0, import_shared8.randomUUID)()
     };
   }
 };
@@ -1551,11 +1563,11 @@ __name(EmptyAdapter, "EmptyAdapter");
 var ExperimentalEmptyAdapter = EmptyAdapter;
 
 // src/service-adapters/events.ts
-var import_shared11 = require("@copilotkit/shared");
+var import_shared14 = require("@copilotkit/shared");
 var import_rxjs2 = require("rxjs");
 
 // src/lib/telemetry-client.ts
-var import_shared10 = require("@copilotkit/shared");
+var import_shared13 = require("@copilotkit/shared");
 
 // src/lib/runtime/remote-action-constructors.ts
 var import_node_crypto2 = require("crypto");
@@ -1803,6 +1815,7 @@ __name(RemoteLangGraphEventSource, "RemoteLangGraphEventSource");
 // src/lib/runtime/remote-lg-action.ts
 var import_langgraph_sdk = require("@langchain/langgraph-sdk");
 var import_node_crypto = require("crypto");
+var import_shared10 = require("@copilotkit/shared");
 var import_partial_json = require("partial-json");
 
 // src/graphql/types/enums.ts
@@ -1856,7 +1869,7 @@ async function execute(args) {
 __name(execute, "execute");
 async function streamEvents(controller, args) {
   var _a, _b, _c, _d, _e, _f;
-  const { deploymentUrl, langsmithApiKey, threadId: agrsInitialThreadId, agent, nodeName: initialNodeName, state: initialState, messages, actions, logger: logger2 } = args;
+  const { deploymentUrl, langsmithApiKey, threadId: argsInitialThreadId, agent, nodeName: initialNodeName, state: initialState, messages, actions, logger: logger2 } = args;
   let nodeName = initialNodeName;
   let state = initialState;
   const { name, assistantId: initialAssistantId } = agent;
@@ -1864,15 +1877,18 @@ async function streamEvents(controller, args) {
     apiUrl: deploymentUrl,
     apiKey: langsmithApiKey
   });
-  let initialThreadId = agrsInitialThreadId;
-  const wasInitiatedWithExistingThread = !!initialThreadId;
-  if (initialThreadId && initialThreadId.startsWith("ck-")) {
-    initialThreadId = initialThreadId.substring(3);
+  let threadId = argsInitialThreadId ?? (0, import_shared10.randomUUID)();
+  if (argsInitialThreadId && argsInitialThreadId.startsWith("ck-")) {
+    threadId = argsInitialThreadId.substring(3);
+  }
+  if (!(0, import_shared10.isValidUUID)(threadId)) {
+    console.warn(`Cannot use the threadId ${threadId} with LangGraph Platform. Must be a valid UUID.`);
   }
-  const threadId = initialThreadId ?? (0, import_node_crypto.randomUUID)();
-  if (initialThreadId === threadId) {
+  let wasInitiatedWithExistingThread = true;
+  try {
     await client.threads.get(threadId);
-  } else {
+  } catch (error) {
+    wasInitiatedWithExistingThread = false;
     await client.threads.create({
       threadId
     });
@@ -1885,7 +1901,7 @@ async function streamEvents(controller, args) {
   }
   const agentStateValues = agentState.values;
   state.messages = agentStateValues.messages;
-  const mode = wasInitiatedWithExistingThread && nodeName != "__end__" ? "continue" : "start";
+  const mode = threadId && nodeName != "__end__" && nodeName != void 0 && nodeName != null ? "continue" : "start";
   let formattedMessages = [];
   try {
     formattedMessages = copilotkitMessagesToLangChain(messages);
@@ -2305,6 +2321,7 @@ function copilotkitMessagesToLangChain(messages) {
 __name(copilotkitMessagesToLangChain, "copilotkitMessagesToLangChain");
 
 // src/lib/runtime/remote-action-constructors.ts
+var import_shared11 = require("@copilotkit/shared");
 function constructLGCRemoteAction({ endpoint, graphqlContext, logger: logger2, messages, agentStates }) {
   const agents = endpoint.agents.map((agent) => ({
     name: agent.name,
@@ -2385,8 +2402,9 @@ function constructRemoteActions({ json, url, onBeforeRequest, graphqlContext, lo
         type: "self-hosted",
         agentsAmount: totalAgents
       });
+      const fetchUrl = `${url}/actions/execute`;
       try {
-        const response = await fetch(`${url}/actions/execute`, {
+        const response = await fetch(fetchUrl, {
           method: "POST",
           headers,
           body: JSON.stringify({
@@ -2411,10 +2429,13 @@ function constructRemoteActions({ json, url, onBeforeRequest, graphqlContext, lo
         }, "Executed remote action");
         return result;
       } catch (error) {
-        logger2.error({
-          error: error.message ? error.message : error + ""
-        }, "Failed to execute remote action");
-        return "Failed to execute remote action";
+        if (error instanceof import_shared11.CopilotKitError) {
+          throw error;
+        }
+        throw new import_shared11.CopilotKitLowLevelError({
+          error,
+          url: fetchUrl
+        });
       }
     }
   }));
@@ -2442,37 +2463,48 @@ function constructRemoteActions({ json, url, onBeforeRequest, graphqlContext, lo
           state = JSON.parse(jsonState);
         }
       }
-      const response = await fetch(`${url}/agents/execute`, {
-        method: "POST",
-        headers,
-        body: JSON.stringify({
-          name,
-          threadId,
-          nodeName,
-          messages: [
-            ...messages,
-            ...additionalMessages
-          ],
-          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 fetchUrl = `${url}/agents/execute`;
+      try {
+        const response = await fetch(fetchUrl, {
+          method: "POST",
+          headers,
+          body: JSON.stringify({
+            name,
+            threadId,
+            nodeName,
+            messages: [
+              ...messages,
+              ...additionalMessages
+            ],
+            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();
+        streamResponse(response.body, eventSource.eventStream$);
+        return eventSource.processLangGraphEvents();
+      } catch (error) {
+        if (error instanceof import_shared11.CopilotKitError) {
+          throw error;
+        }
+        throw new import_shared11.CopilotKitLowLevelError({
+          error,
+          url: fetchUrl
+        });
       }
-      const eventSource = new RemoteLangGraphEventSource();
-      streamResponse(response.body, eventSource.eventStream$);
-      return eventSource.processLangGraphEvents();
     }
   })) : [];
   return [
@@ -2542,6 +2574,7 @@ function createHeaders(onBeforeRequest, graphqlContext) {
 __name(createHeaders, "createHeaders");
 
 // src/lib/runtime/remote-actions.ts
+var import_shared12 = require("@copilotkit/shared");
 var EndpointType;
 (function(EndpointType2) {
   EndpointType2["CopilotKit"] = "copilotKit";
@@ -2559,8 +2592,9 @@ async function fetchRemoteInfo({ url, onBeforeRequest, graphqlContext, logger: l
     url
   }, "Fetching actions from url");
   const headers = createHeaders(onBeforeRequest, graphqlContext);
+  const fetchUrl = `${url}/info`;
   try {
-    const response = await fetch(`${url}/info`, {
+    const response = await fetch(fetchUrl, {
       method: "POST",
       headers,
       body: JSON.stringify({
@@ -2574,10 +2608,10 @@ async function fetchRemoteInfo({ url, onBeforeRequest, graphqlContext, logger: l
         status: response.status,
         body: await response.text()
       }, "Failed to fetch actions from url");
-      return {
-        actions: [],
-        agents: []
-      };
+      throw new import_shared12.ResolvedCopilotKitError({
+        status: response.status,
+        isRemoteEndpoint: true
+      });
     }
     const json = await response.json();
     logger2.debug({
@@ -2585,13 +2619,13 @@ async function fetchRemoteInfo({ url, onBeforeRequest, graphqlContext, logger: l
     }, "Fetched actions from url");
     return json;
   } catch (error) {
-    logger2.error({
-      error: error.message ? error.message : error + ""
-    }, "Failed to fetch actions from url");
-    return {
-      actions: [],
-      agents: []
-    };
+    if (error instanceof import_shared12.CopilotKitError) {
+      throw error;
+    }
+    throw new import_shared12.CopilotKitLowLevelError({
+      error,
+      url: fetchUrl
+    });
   }
 }
 __name(fetchRemoteInfo, "fetchRemoteInfo");
@@ -2651,7 +2685,7 @@ __name(setupRemoteActions, "setupRemoteActions");
 // src/lib/telemetry-client.ts
 var import_node_crypto3 = require("crypto");
 var packageJson = require_package();
-var telemetryClient = new import_shared10.TelemetryClient({
+var telemetryClient = new import_shared13.TelemetryClient({
   packageName: packageJson.name,
   packageVersion: packageJson.version
 });
@@ -2961,13 +2995,13 @@ var RuntimeEventSource = class {
     const errorMessage = `\u274C ${message}`;
     if (!this.callback) {
       this.stream(async (eventStream$) => {
-        eventStream$.sendTextMessage((0, import_shared11.randomId)(), errorMessage);
+        eventStream$.sendTextMessage((0, import_shared14.randomId)(), errorMessage);
       });
     } else {
-      this.eventStream$.sendTextMessage((0, import_shared11.randomId)(), errorMessage);
+      this.eventStream$.sendTextMessage((0, import_shared14.randomId)(), errorMessage);
     }
   }
-  processRuntimeEvents({ serverSideActions, guardrailsResult$, actionInputsWithoutAgents }) {
+  processRuntimeEvents({ serverSideActions, guardrailsResult$, actionInputsWithoutAgents, threadId }) {
     this.callback(this.eventStream$).catch((error) => {
       console.error("Error in event source callback", error);
       this.sendErrorMessageToChat();
@@ -3003,7 +3037,7 @@ var RuntimeEventSource = class {
       (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.actionExecutionParentMessageId, eventWithState.actionExecutionId, actionInputsWithoutAgents).catch((error) => {
+          executeAction(toolCallEventStream$, guardrailsResult$ ? guardrailsResult$ : null, eventWithState.action, eventWithState.args, eventWithState.actionExecutionParentMessageId, eventWithState.actionExecutionId, actionInputsWithoutAgents, threadId).catch((error) => {
             console.error(error);
           });
           telemetry_client_default.capture("oss.runtime.server_action_executed", {});
@@ -3020,7 +3054,7 @@ var RuntimeEventSource = class {
   }
 };
 __name(RuntimeEventSource, "RuntimeEventSource");
-async function executeAction(eventStream$, guardrailsResult$, action, actionArguments, actionExecutionParentMessageId, actionExecutionId, actionInputsWithoutAgents) {
+async function executeAction(eventStream$, guardrailsResult$, action, actionArguments, actionExecutionParentMessageId, actionExecutionId, actionInputsWithoutAgents, threadId) {
   var _a;
   if (guardrailsResult$) {
     const { status } = await (0, import_rxjs2.firstValueFrom)(guardrailsResult$);
@@ -3071,6 +3105,7 @@ async function executeAction(eventStream$, guardrailsResult$, action, actionArgu
     });
     const stream = await action.langGraphAgentHandler({
       name: action.name,
+      threadId,
       actionInputsWithoutAgents,
       additionalMessages: [
         agentExecution,
@@ -3170,6 +3205,7 @@ __name(convertGqlInputToMessages, "convertGqlInputToMessages");
 
 // src/lib/runtime/copilot-runtime.ts
 var import_rxjs3 = require("rxjs");
+var import_langgraph_sdk2 = require("@langchain/langgraph-sdk");
 var CopilotRuntime = class {
   actions;
   remoteEndpointDefinitions;
@@ -3189,16 +3225,18 @@ var CopilotRuntime = class {
   }
   async processRuntimeRequest(request) {
     var _a;
-    const { serviceAdapter, messages: rawMessages, actions: clientSideActionsInput, threadId, runId, outputMessagesPromise, graphqlContext, forwardedParameters, agentSession, url } = request;
+    const { serviceAdapter, messages: rawMessages, actions: clientSideActionsInput, threadId, runId, outputMessagesPromise, graphqlContext, forwardedParameters, agentSession, url, extensions } = request;
     const eventSource = new RuntimeEventSource();
     try {
       if (agentSession) {
         return await this.processAgentRequest(request);
       }
       if (serviceAdapter instanceof EmptyAdapter) {
-        throw new Error(`Invalid adapter configuration: EmptyAdapter is only meant to be used with agent lock mode. 
+        throw new import_shared15.CopilotKitMisuseError({
+          message: `Invalid adapter configuration: EmptyAdapter is only meant to be used with agent lock mode. 
 For non-agent components like useCopilotChatSuggestions, CopilotTextarea, or CopilotTask, 
-please use an LLM adapter instead.`);
+please use an LLM adapter instead.`
+        });
       }
       const messages = rawMessages.filter((message) => !message.agentStateMessage);
       const inputMessages = convertGqlInputToMessages(messages);
@@ -3206,7 +3244,7 @@ please use an LLM adapter instead.`);
       const serverSideActionsInput = serverSideActions.map((action) => ({
         name: action.name,
         description: action.description,
-        jsonSchema: JSON.stringify((0, import_shared12.actionParametersToJsonSchema)(action.parameters))
+        jsonSchema: JSON.stringify((0, import_shared15.actionParametersToJsonSchema)(action.parameters))
       }));
       const actionInputs = flattenToolCallsNoDuplicates([
         ...serverSideActionsInput,
@@ -3228,12 +3266,14 @@ please use an LLM adapter instead.`);
         threadId,
         runId,
         eventSource,
-        forwardedParameters
+        forwardedParameters,
+        extensions
       });
+      const nonEmptyThreadId = threadId ?? result.threadId;
       outputMessagesPromise.then((outputMessages) => {
         var _a2;
         (_a2 = this.onAfterRequest) == null ? void 0 : _a2.call(this, {
-          threadId: result.threadId,
+          threadId: nonEmptyThreadId,
           runId: result.runId,
           inputMessages,
           outputMessages,
@@ -3243,16 +3283,20 @@ please use an LLM adapter instead.`);
       }).catch((_error) => {
       });
       return {
-        threadId: result.threadId,
+        threadId: nonEmptyThreadId,
         runId: result.runId,
         eventSource,
         serverSideActions,
         actionInputsWithoutAgents: actionInputs.filter((action) => (
           // TODO-AGENTS: do not exclude ALL server side actions
           !serverSideActions.find((serverSideAction) => serverSideAction.name == action.name)
-        ))
+        )),
+        extensions: result.extensions
       };
     } catch (error) {
+      if (error instanceof import_shared15.CopilotKitError) {
+        throw error;
+      }
       console.error("Error getting response:", error);
       eventSource.sendErrorMessageToChat();
       throw error;
@@ -3263,54 +3307,131 @@ please use an LLM adapter instead.`);
     const agents = this.remoteEndpointDefinitions.reduce(async (acc, endpoint) => {
       const agents2 = await acc;
       if (endpoint.type === EndpointType.LangGraphPlatform) {
-        const response2 = await fetch(`${endpoint.deploymentUrl}/assistants/search`, {
-          method: "POST",
-          headers
+        const client = new import_langgraph_sdk2.Client({
+          apiUrl: endpoint.deploymentUrl,
+          apiKey: endpoint.langsmithApiKey
         });
-        const data2 = await response2.json();
-        const endpointAgents2 = (data2 ?? []).map((entry) => ({
+        const data = await client.assistants.search();
+        const endpointAgents = (data ?? []).map((entry) => ({
           name: entry.graph_id,
-          id: entry.assistant_id
+          id: entry.assistant_id,
+          description: "",
+          endpoint
+        }));
+        return [
+          ...agents2,
+          ...endpointAgents
+        ];
+      }
+      const fetchUrl = `${endpoint.url}/info`;
+      try {
+        const response = await fetch(fetchUrl, {
+          method: "POST",
+          headers,
+          body: JSON.stringify({
+            properties: graphqlContext.properties
+          })
+        });
+        if (!response.ok) {
+          if (response.status === 404) {
+            throw new import_shared15.CopilotKitApiDiscoveryError();
+          }
+          throw new import_shared15.ResolvedCopilotKitError({
+            status: response.status,
+            isRemoteEndpoint: true
+          });
+        }
+        const data = await response.json();
+        const endpointAgents = ((data == null ? void 0 : data.agents) ?? []).map((agent) => ({
+          name: agent.name,
+          description: agent.description ?? "" ?? "",
+          id: (0, import_shared15.randomId)(),
+          endpoint
         }));
         return [
           ...agents2,
-          ...endpointAgents2
+          ...endpointAgents
         ];
+      } catch (error) {
+        if (error instanceof import_shared15.CopilotKitError) {
+          throw error;
+        }
+        throw new import_shared15.CopilotKitLowLevelError({
+          error,
+          url: fetchUrl
+        });
+      }
+    }, Promise.resolve([]));
+    return agents;
+  }
+  async loadAgentState(graphqlContext, threadId, agentName) {
+    const agentsWithEndpoints = await this.discoverAgentsFromEndpoints(graphqlContext);
+    const agentWithEndpoint = agentsWithEndpoints.find((agent) => agent.name === agentName);
+    if (!agentWithEndpoint) {
+      throw new Error("Agent not found");
+    }
+    const headers = createHeaders(null, graphqlContext);
+    if (agentWithEndpoint.endpoint.type === EndpointType.LangGraphPlatform) {
+      const client = new import_langgraph_sdk2.Client({
+        apiUrl: agentWithEndpoint.endpoint.deploymentUrl,
+        apiKey: agentWithEndpoint.endpoint.langsmithApiKey
+      });
+      const state = (await client.threads.getState(threadId)).values;
+      if (Object.keys(state).length === 0) {
+        return {
+          threadId,
+          threadExists: false,
+          state: JSON.stringify({}),
+          messages: JSON.stringify([])
+        };
+      } else {
+        console.log(state);
+        const { messages, ...stateWithoutMessages } = state;
+        const copilotkitMessages = langchainMessagesToCopilotKit(messages);
+        return {
+          threadId,
+          threadExists: true,
+          state: JSON.stringify(stateWithoutMessages),
+          messages: JSON.stringify(copilotkitMessages)
+        };
       }
-      const response = await fetch(`${endpoint.url}/info`, {
+    } else if (agentWithEndpoint.endpoint.type === EndpointType.CopilotKit || !("type" in agentWithEndpoint.endpoint)) {
+      const response = await fetch(`${agentWithEndpoint.endpoint.url}/agents/state`, {
         method: "POST",
         headers,
         body: JSON.stringify({
-          properties: graphqlContext.properties
+          properties: graphqlContext.properties,
+          threadId,
+          name: agentName
         })
       });
       const data = await response.json();
-      const endpointAgents = ((data == null ? void 0 : data.agents) ?? []).map((agent) => ({
-        name: agent.name,
-        description: agent.description,
-        id: (0, import_shared12.randomId)()
-      }));
-      return [
-        ...agents2,
-        ...endpointAgents
-      ];
-    }, Promise.resolve([]));
-    return agents;
+      return {
+        ...data,
+        state: JSON.stringify(data.state),
+        messages: JSON.stringify(data.messages)
+      };
+    } else {
+      throw new Error(`Unknown endpoint type: ${agentWithEndpoint.endpoint.type}`);
+    }
   }
   async processAgentRequest(request) {
     var _a;
-    const { messages: rawMessages, outputMessagesPromise, graphqlContext, agentSession } = request;
-    const { threadId, agentName, nodeName } = agentSession;
+    const { messages: rawMessages, outputMessagesPromise, graphqlContext, agentSession, threadId: threadIdFromRequest } = request;
+    const { agentName, nodeName } = agentSession;
+    const threadId = threadIdFromRequest ?? agentSession.threadId;
     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`);
+      throw new import_shared15.CopilotKitAgentDiscoveryError({
+        agentName
+      });
     }
     const serverSideActionsInput = serverSideActions.filter((action) => !isLangGraphAgentAction(action)).map((action) => ({
       name: action.name,
       description: action.description,
-      jsonSchema: JSON.stringify((0, import_shared12.actionParametersToJsonSchema)(action.parameters))
+      jsonSchema: JSON.stringify((0, import_shared15.actionParametersToJsonSchema)(action.parameters))
     }));
     const actionInputsWithoutAgents = flattenToolCallsNoDuplicates([
       ...serverSideActionsInput,
@@ -3438,14 +3559,14 @@ function resolveEndpointType(endpoint) {
 __name(resolveEndpointType, "resolveEndpointType");
 
 // src/lib/integrations/shared.ts
-var import_type_graphql17 = require("type-graphql");
+var import_type_graphql24 = require("type-graphql");
 
 // src/graphql/resolvers/copilot.resolver.ts
-var import_type_graphql16 = require("type-graphql");
+var import_type_graphql18 = require("type-graphql");
 var import_rxjs4 = require("rxjs");
 
 // src/graphql/inputs/generate-copilot-response.input.ts
-var import_type_graphql11 = require("type-graphql");
+var import_type_graphql12 = require("type-graphql");
 
 // src/graphql/inputs/message.input.ts
 var import_type_graphql3 = require("type-graphql");
@@ -3961,7 +4082,8 @@ AgentStateInput = _ts_decorate9([
   (0, import_type_graphql10.InputType)()
 ], AgentStateInput);
 
-// src/graphql/inputs/generate-copilot-response.input.ts
+// src/graphql/inputs/extensions.input.ts
+var import_type_graphql11 = require("type-graphql");
 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")
@@ -3978,18 +4100,69 @@ function _ts_metadata10(k, v) {
     return Reflect.metadata(k, v);
 }
 __name(_ts_metadata10, "_ts_metadata");
+var ExtensionsInput = class {
+  openaiAssistantAPI;
+};
+__name(ExtensionsInput, "ExtensionsInput");
+_ts_decorate10([
+  (0, import_type_graphql11.Field)(() => OpenAIApiAssistantAPIInput, {
+    nullable: true
+  }),
+  _ts_metadata10("design:type", typeof OpenAIApiAssistantAPIInput === "undefined" ? Object : OpenAIApiAssistantAPIInput)
+], ExtensionsInput.prototype, "openaiAssistantAPI", void 0);
+ExtensionsInput = _ts_decorate10([
+  (0, import_type_graphql11.InputType)()
+], ExtensionsInput);
+var OpenAIApiAssistantAPIInput = class {
+  runId;
+  threadId;
+};
+__name(OpenAIApiAssistantAPIInput, "OpenAIApiAssistantAPIInput");
+_ts_decorate10([
+  (0, import_type_graphql11.Field)(() => String, {
+    nullable: true
+  }),
+  _ts_metadata10("design:type", String)
+], OpenAIApiAssistantAPIInput.prototype, "runId", void 0);
+_ts_decorate10([
+  (0, import_type_graphql11.Field)(() => String, {
+    nullable: true
+  }),
+  _ts_metadata10("design:type", String)
+], OpenAIApiAssistantAPIInput.prototype, "threadId", void 0);
+OpenAIApiAssistantAPIInput = _ts_decorate10([
+  (0, import_type_graphql11.InputType)()
+], OpenAIApiAssistantAPIInput);
+
+// src/graphql/inputs/generate-copilot-response.input.ts
+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);
+  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_decorate11, "_ts_decorate");
+function _ts_metadata11(k, v) {
+  if (typeof Reflect === "object" && typeof Reflect.metadata === "function")
+    return Reflect.metadata(k, v);
+}
+__name(_ts_metadata11, "_ts_metadata");
 var GenerateCopilotResponseMetadataInput = class {
   requestType;
 };
 __name(GenerateCopilotResponseMetadataInput, "GenerateCopilotResponseMetadataInput");
-_ts_decorate10([
-  (0, import_type_graphql11.Field)(() => CopilotRequestType, {
+_ts_decorate11([
+  (0, import_type_graphql12.Field)(() => CopilotRequestType, {
     nullable: true
   }),
-  _ts_metadata10("design:type", typeof CopilotRequestType === "undefined" ? Object : CopilotRequestType)
+  _ts_metadata11("design:type", typeof CopilotRequestType === "undefined" ? Object : CopilotRequestType)
 ], GenerateCopilotResponseMetadataInput.prototype, "requestType", void 0);
-GenerateCopilotResponseMetadataInput = _ts_decorate10([
-  (0, import_type_graphql11.InputType)()
+GenerateCopilotResponseMetadataInput = _ts_decorate11([
+  (0, import_type_graphql12.InputType)()
 ], GenerateCopilotResponseMetadataInput);
 var GenerateCopilotResponseInput = class {
   metadata;
@@ -4002,78 +4175,85 @@ var GenerateCopilotResponseInput = class {
   agentSession;
   agentState;
   agentStates;
+  extensions;
 };
 __name(GenerateCopilotResponseInput, "GenerateCopilotResponseInput");
-_ts_decorate10([
-  (0, import_type_graphql11.Field)(() => GenerateCopilotResponseMetadataInput, {
+_ts_decorate11([
+  (0, import_type_graphql12.Field)(() => GenerateCopilotResponseMetadataInput, {
     nullable: false
   }),
-  _ts_metadata10("design:type", typeof GenerateCopilotResponseMetadataInput === "undefined" ? Object : GenerateCopilotResponseMetadataInput)
+  _ts_metadata11("design:type", typeof GenerateCopilotResponseMetadataInput === "undefined" ? Object : GenerateCopilotResponseMetadataInput)
 ], GenerateCopilotResponseInput.prototype, "metadata", void 0);
-_ts_decorate10([
-  (0, import_type_graphql11.Field)(() => String, {
+_ts_decorate11([
+  (0, import_type_graphql12.Field)(() => String, {
     nullable: true
   }),
-  _ts_metadata10("design:type", String)
+  _ts_metadata11("design:type", String)
 ], GenerateCopilotResponseInput.prototype, "threadId", void 0);
-_ts_decorate10([
-  (0, import_type_graphql11.Field)(() => String, {
+_ts_decorate11([
+  (0, import_type_graphql12.Field)(() => String, {
     nullable: true
   }),
-  _ts_metadata10("design:type", String)
+  _ts_metadata11("design:type", String)
 ], GenerateCopilotResponseInput.prototype, "runId", void 0);
-_ts_decorate10([
-  (0, import_type_graphql11.Field)(() => [
+_ts_decorate11([
+  (0, import_type_graphql12.Field)(() => [
     MessageInput
   ]),
-  _ts_metadata10("design:type", Array)
+  _ts_metadata11("design:type", Array)
 ], GenerateCopilotResponseInput.prototype, "messages", void 0);
-_ts_decorate10([
-  (0, import_type_graphql11.Field)(() => FrontendInput),
-  _ts_metadata10("design:type", typeof FrontendInput === "undefined" ? Object : FrontendInput)
+_ts_decorate11([
+  (0, import_type_graphql12.Field)(() => FrontendInput),
+  _ts_metadata11("design:type", typeof FrontendInput === "undefined" ? Object : FrontendInput)
 ], GenerateCopilotResponseInput.prototype, "frontend", void 0);
-_ts_decorate10([
-  (0, import_type_graphql11.Field)(() => CloudInput, {
+_ts_decorate11([
+  (0, import_type_graphql12.Field)(() => CloudInput, {
     nullable: true
   }),
-  _ts_metadata10("design:type", typeof CloudInput === "undefined" ? Object : CloudInput)
+  _ts_metadata11("design:type", typeof CloudInput === "undefined" ? Object : CloudInput)
 ], GenerateCopilotResponseInput.prototype, "cloud", void 0);
-_ts_decorate10([
-  (0, import_type_graphql11.Field)(() => ForwardedParametersInput, {
+_ts_decorate11([
+  (0, import_type_graphql12.Field)(() => ForwardedParametersInput, {
     nullable: true
   }),
-  _ts_metadata10("design:type", typeof ForwardedParametersInput === "undefined" ? Object : ForwardedParametersInput)
+  _ts_metadata11("design:type", typeof ForwardedParametersInput === "undefined" ? Object : ForwardedParametersInput)
 ], GenerateCopilotResponseInput.prototype, "forwardedParameters", void 0);
-_ts_decorate10([
-  (0, import_type_graphql11.Field)(() => AgentSessionInput, {
+_ts_decorate11([
+  (0, import_type_graphql12.Field)(() => AgentSessionInput, {
     nullable: true
   }),
-  _ts_metadata10("design:type", typeof AgentSessionInput === "undefined" ? Object : AgentSessionInput)
+  _ts_metadata11("design:type", typeof AgentSessionInput === "undefined" ? Object : AgentSessionInput)
 ], GenerateCopilotResponseInput.prototype, "agentSession", void 0);
-_ts_decorate10([
-  (0, import_type_graphql11.Field)(() => AgentStateInput, {
+_ts_decorate11([
+  (0, import_type_graphql12.Field)(() => AgentStateInput, {
     nullable: true
   }),
-  _ts_metadata10("design:type", typeof AgentStateInput === "undefined" ? Object : AgentStateInput)
+  _ts_metadata11("design:type", typeof AgentStateInput === "undefined" ? Object : AgentStateInput)
 ], GenerateCopilotResponseInput.prototype, "agentState", void 0);
-_ts_decorate10([
-  (0, import_type_graphql11.Field)(() => [
+_ts_decorate11([
+  (0, import_type_graphql12.Field)(() => [
     AgentStateInput
   ], {
     nullable: true
   }),
-  _ts_metadata10("design:type", Array)
+  _ts_metadata11("design:type", Array)
 ], GenerateCopilotResponseInput.prototype, "agentStates", void 0);
-GenerateCopilotResponseInput = _ts_decorate10([
-  (0, import_type_graphql11.InputType)()
+_ts_decorate11([
+  (0, import_type_graphql12.Field)(() => ExtensionsInput, {
+    nullable: true
+  }),
+  _ts_metadata11("design:type", typeof ExtensionsInput === "undefined" ? Object : ExtensionsInput)
+], GenerateCopilotResponseInput.prototype, "extensions", void 0);
+GenerateCopilotResponseInput = _ts_decorate11([
+  (0, import_type_graphql12.InputType)()
 ], GenerateCopilotResponseInput);
 
 // src/graphql/types/copilot-response.type.ts
-var import_type_graphql14 = require("type-graphql");
+var import_type_graphql16 = require("type-graphql");
 
 // src/graphql/types/message-status.type.ts
-var import_type_graphql12 = require("type-graphql");
-function _ts_decorate11(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);
@@ -4083,58 +4263,58 @@ 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_decorate12, "_ts_decorate");
+function _ts_metadata12(k, v) {
   if (typeof Reflect === "object" && typeof Reflect.metadata === "function")
     return Reflect.metadata(k, v);
 }
-__name(_ts_metadata11, "_ts_metadata");
+__name(_ts_metadata12, "_ts_metadata");
 var MessageStatusCode;
 (function(MessageStatusCode2) {
   MessageStatusCode2["Pending"] = "pending";
   MessageStatusCode2["Success"] = "success";
   MessageStatusCode2["Failed"] = "failed";
 })(MessageStatusCode || (MessageStatusCode = {}));
-(0, import_type_graphql12.registerEnumType)(MessageStatusCode, {
+(0, import_type_graphql13.registerEnumType)(MessageStatusCode, {
   name: "MessageStatusCode"
 });
 var BaseMessageStatus = /* @__PURE__ */ __name(class BaseMessageStatus2 {
   code;
 }, "BaseMessageStatus");
-_ts_decorate11([
-  (0, import_type_graphql12.Field)(() => MessageStatusCode),
-  _ts_metadata11("design:type", String)
+_ts_decorate12([
+  (0, import_type_graphql13.Field)(() => MessageStatusCode),
+  _ts_metadata12("design:type", String)
 ], BaseMessageStatus.prototype, "code", void 0);
-BaseMessageStatus = _ts_decorate11([
-  (0, import_type_graphql12.ObjectType)()
+BaseMessageStatus = _ts_decorate12([
+  (0, import_type_graphql13.ObjectType)()
 ], BaseMessageStatus);
 var PendingMessageStatus = class extends BaseMessageStatus {
   code = "pending";
 };
 __name(PendingMessageStatus, "PendingMessageStatus");
-PendingMessageStatus = _ts_decorate11([
-  (0, import_type_graphql12.ObjectType)()
+PendingMessageStatus = _ts_decorate12([
+  (0, import_type_graphql13.ObjectType)()
 ], PendingMessageStatus);
 var SuccessMessageStatus = class extends BaseMessageStatus {
   code = "success";
 };
 __name(SuccessMessageStatus, "SuccessMessageStatus");
-SuccessMessageStatus = _ts_decorate11([
-  (0, import_type_graphql12.ObjectType)()
+SuccessMessageStatus = _ts_decorate12([
+  (0, import_type_graphql13.ObjectType)()
 ], SuccessMessageStatus);
 var FailedMessageStatus = class extends BaseMessageStatus {
   code = "failed";
   reason;
 };
 __name(FailedMessageStatus, "FailedMessageStatus");
-_ts_decorate11([
-  (0, import_type_graphql12.Field)(() => String),
-  _ts_metadata11("design:type", String)
+_ts_decorate12([
+  (0, import_type_graphql13.Field)(() => String),
+  _ts_metadata12("design:type", String)
 ], FailedMessageStatus.prototype, "reason", void 0);
-FailedMessageStatus = _ts_decorate11([
-  (0, import_type_graphql12.ObjectType)()
+FailedMessageStatus = _ts_decorate12([
+  (0, import_type_graphql13.ObjectType)()
 ], FailedMessageStatus);
-var MessageStatusUnion = (0, import_type_graphql12.createUnionType)({
+var MessageStatusUnion = (0, import_type_graphql13.createUnionType)({
   name: "MessageStatus",
   types: () => [
     PendingMessageStatus,
@@ -4145,8 +4325,8 @@ var MessageStatusUnion = (0, import_type_graphql12.createUnionType)({
 
 // src/graphql/types/response-status.type.ts
 var import_graphql_scalars = require("graphql-scalars");
-var import_type_graphql13 = require("type-graphql");
-function _ts_decorate12(decorators, target, key, desc) {
+var import_type_graphql14 = require("type-graphql");
+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);
@@ -4156,30 +4336,30 @@ 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_decorate13, "_ts_decorate");
+function _ts_metadata13(k, v) {
   if (typeof Reflect === "object" && typeof Reflect.metadata === "function")
     return Reflect.metadata(k, v);
 }
-__name(_ts_metadata12, "_ts_metadata");
+__name(_ts_metadata13, "_ts_metadata");
 var ResponseStatusCode;
 (function(ResponseStatusCode2) {
   ResponseStatusCode2["Pending"] = "pending";
   ResponseStatusCode2["Success"] = "success";
   ResponseStatusCode2["Failed"] = "failed";
 })(ResponseStatusCode || (ResponseStatusCode = {}));
-(0, import_type_graphql13.registerEnumType)(ResponseStatusCode, {
+(0, import_type_graphql14.registerEnumType)(ResponseStatusCode, {
   name: "ResponseStatusCode"
 });
 var BaseResponseStatus = /* @__PURE__ */ __name(class BaseResponseStatus2 {
   code;
 }, "BaseResponseStatus");
-_ts_decorate12([
-  (0, import_type_graphql13.Field)(() => ResponseStatusCode),
-  _ts_metadata12("design:type", String)
+_ts_decorate13([
+  (0, import_type_graphql14.Field)(() => ResponseStatusCode),
+  _ts_metadata13("design:type", String)
 ], BaseResponseStatus.prototype, "code", void 0);
-BaseResponseStatus = _ts_decorate12([
-  (0, import_type_graphql13.InterfaceType)({
+BaseResponseStatus = _ts_decorate13([
+  (0, import_type_graphql14.InterfaceType)({
     resolveType(value) {
       if (value.code === "success") {
         return SuccessResponseStatus;
@@ -4191,14 +4371,14 @@ BaseResponseStatus = _ts_decorate12([
       return void 0;
     }
   }),
-  (0, import_type_graphql13.ObjectType)()
+  (0, import_type_graphql14.ObjectType)()
 ], BaseResponseStatus);
 var PendingResponseStatus = class extends BaseResponseStatus {
   code = "pending";
 };
 __name(PendingResponseStatus, "PendingResponseStatus");
-PendingResponseStatus = _ts_decorate12([
-  (0, import_type_graphql13.ObjectType)({
+PendingResponseStatus = _ts_decorate13([
+  (0, import_type_graphql14.ObjectType)({
     implements: BaseResponseStatus
   })
 ], PendingResponseStatus);
@@ -4206,8 +4386,8 @@ var SuccessResponseStatus = class extends BaseResponseStatus {
   code = "success";
 };
 __name(SuccessResponseStatus, "SuccessResponseStatus");
-SuccessResponseStatus = _ts_decorate12([
-  (0, import_type_graphql13.ObjectType)({
+SuccessResponseStatus = _ts_decorate13([
+  (0, import_type_graphql14.ObjectType)({
     implements: BaseResponseStatus
   })
 ], SuccessResponseStatus);
@@ -4217,7 +4397,7 @@ var FailedResponseStatusReason;
   FailedResponseStatusReason2["MESSAGE_STREAM_INTERRUPTED"] = "MESSAGE_STREAM_INTERRUPTED";
   FailedResponseStatusReason2["UNKNOWN_ERROR"] = "UNKNOWN_ERROR";
 })(FailedResponseStatusReason || (FailedResponseStatusReason = {}));
-(0, import_type_graphql13.registerEnumType)(FailedResponseStatusReason, {
+(0, import_type_graphql14.registerEnumType)(FailedResponseStatusReason, {
   name: "FailedResponseStatusReason"
 });
 var FailedResponseStatus = class extends BaseResponseStatus {
@@ -4226,22 +4406,22 @@ var FailedResponseStatus = class extends BaseResponseStatus {
   details = null;
 };
 __name(FailedResponseStatus, "FailedResponseStatus");
-_ts_decorate12([
-  (0, import_type_graphql13.Field)(() => FailedResponseStatusReason),
-  _ts_metadata12("design:type", String)
+_ts_decorate13([
+  (0, import_type_graphql14.Field)(() => FailedResponseStatusReason),
+  _ts_metadata13("design:type", String)
 ], FailedResponseStatus.prototype, "reason", void 0);
-_ts_decorate12([
-  (0, import_type_graphql13.Field)(() => import_graphql_scalars.GraphQLJSON, {
+_ts_decorate13([
+  (0, import_type_graphql14.Field)(() => import_graphql_scalars.GraphQLJSON, {
     nullable: true
   }),
-  _ts_metadata12("design:type", typeof Record === "undefined" ? Object : Record)
+  _ts_metadata13("design:type", typeof Record === "undefined" ? Object : Record)
 ], FailedResponseStatus.prototype, "details", void 0);
-FailedResponseStatus = _ts_decorate12([
-  (0, import_type_graphql13.ObjectType)({
+FailedResponseStatus = _ts_decorate13([
+  (0, import_type_graphql14.ObjectType)({
     implements: BaseResponseStatus
   })
 ], FailedResponseStatus);
-var ResponseStatusUnion = (0, import_type_graphql13.createUnionType)({
+var ResponseStatusUnion = (0, import_type_graphql14.createUnionType)({
   name: "ResponseStatus",
   types: () => [
     PendingResponseStatus,
@@ -4250,8 +4430,60 @@ var ResponseStatusUnion = (0, import_type_graphql13.createUnionType)({
   ]
 });
 
+// src/graphql/types/extensions-response.type.ts
+var import_type_graphql15 = require("type-graphql");
+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);
+  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_decorate14, "_ts_decorate");
+function _ts_metadata14(k, v) {
+  if (typeof Reflect === "object" && typeof Reflect.metadata === "function")
+    return Reflect.metadata(k, v);
+}
+__name(_ts_metadata14, "_ts_metadata");
+var ExtensionsResponse = class {
+  openaiAssistantAPI;
+};
+__name(ExtensionsResponse, "ExtensionsResponse");
+_ts_decorate14([
+  (0, import_type_graphql15.Field)(() => OpenAIApiAssistantAPIResponse, {
+    nullable: true
+  }),
+  _ts_metadata14("design:type", typeof OpenAIApiAssistantAPIResponse === "undefined" ? Object : OpenAIApiAssistantAPIResponse)
+], ExtensionsResponse.prototype, "openaiAssistantAPI", void 0);
+ExtensionsResponse = _ts_decorate14([
+  (0, import_type_graphql15.ObjectType)()
+], ExtensionsResponse);
+var OpenAIApiAssistantAPIResponse = class {
+  runId;
+  threadId;
+};
+__name(OpenAIApiAssistantAPIResponse, "OpenAIApiAssistantAPIResponse");
+_ts_decorate14([
+  (0, import_type_graphql15.Field)(() => String, {
+    nullable: true
+  }),
+  _ts_metadata14("design:type", String)
+], OpenAIApiAssistantAPIResponse.prototype, "runId", void 0);
+_ts_decorate14([
+  (0, import_type_graphql15.Field)(() => String, {
+    nullable: true
+  }),
+  _ts_metadata14("design:type", String)
+], OpenAIApiAssistantAPIResponse.prototype, "threadId", void 0);
+OpenAIApiAssistantAPIResponse = _ts_decorate14([
+  (0, import_type_graphql15.ObjectType)()
+], OpenAIApiAssistantAPIResponse);
+
 // src/graphql/types/copilot-response.type.ts
-function _ts_decorate13(decorators, target, key, desc) {
+function _ts_decorate15(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);
@@ -4261,31 +4493,32 @@ function _ts_decorate13(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_decorate13, "_ts_decorate");
-function _ts_metadata13(k, v) {
+__name(_ts_decorate15, "_ts_decorate");
+function _ts_metadata15(k, v) {
   if (typeof Reflect === "object" && typeof Reflect.metadata === "function")
     return Reflect.metadata(k, v);
 }
-__name(_ts_metadata13, "_ts_metadata");
-var BaseMessageOutput = /* @__PURE__ */ __name(class BaseMessageOutput2 {
+__name(_ts_metadata15, "_ts_metadata");
+var BaseMessageOutput = class {
   id;
   createdAt;
   status;
-}, "BaseMessageOutput");
-_ts_decorate13([
-  (0, import_type_graphql14.Field)(() => String),
-  _ts_metadata13("design:type", String)
+};
+__name(BaseMessageOutput, "BaseMessageOutput");
+_ts_decorate15([
+  (0, import_type_graphql16.Field)(() => String),
+  _ts_metadata15("design:type", String)
 ], BaseMessageOutput.prototype, "id", void 0);
-_ts_decorate13([
-  (0, import_type_graphql14.Field)(() => Date),
-  _ts_metadata13("design:type", typeof Date === "undefined" ? Object : Date)
+_ts_decorate15([
+  (0, import_type_graphql16.Field)(() => Date),
+  _ts_metadata15("design:type", typeof Date === "undefined" ? Object : Date)
 ], BaseMessageOutput.prototype, "createdAt", void 0);
-_ts_decorate13([
-  (0, import_type_graphql14.Field)(() => MessageStatusUnion),
-  _ts_metadata13("design:type", Object)
+_ts_decorate15([
+  (0, import_type_graphql16.Field)(() => MessageStatusUnion),
+  _ts_metadata15("design:type", Object)
 ], BaseMessageOutput.prototype, "status", void 0);
-BaseMessageOutput = _ts_decorate13([
-  (0, import_type_graphql14.InterfaceType)({
+BaseMessageOutput = _ts_decorate15([
+  (0, import_type_graphql16.InterfaceType)({
     resolveType(value) {
       if (value.hasOwnProperty("content")) {
         return TextMessageOutput;
@@ -4306,24 +4539,24 @@ var TextMessageOutput = class {
   parentMessageId;
 };
 __name(TextMessageOutput, "TextMessageOutput");
-_ts_decorate13([
-  (0, import_type_graphql14.Field)(() => MessageRole),
-  _ts_metadata13("design:type", typeof MessageRole === "undefined" ? Object : MessageRole)
+_ts_decorate15([
+  (0, import_type_graphql16.Field)(() => MessageRole),
+  _ts_metadata15("design:type", typeof MessageRole === "undefined" ? Object : MessageRole)
 ], TextMessageOutput.prototype, "role", void 0);
-_ts_decorate13([
-  (0, import_type_graphql14.Field)(() => [
+_ts_decorate15([
+  (0, import_type_graphql16.Field)(() => [
     String
   ]),
-  _ts_metadata13("design:type", Array)
+  _ts_metadata15("design:type", Array)
 ], TextMessageOutput.prototype, "content", void 0);
-_ts_decorate13([
-  (0, import_type_graphql14.Field)(() => String, {
+_ts_decorate15([
+  (0, import_type_graphql16.Field)(() => String, {
     nullable: true
   }),
-  _ts_metadata13("design:type", String)
+  _ts_metadata15("design:type", String)
 ], TextMessageOutput.prototype, "parentMessageId", void 0);
-TextMessageOutput = _ts_decorate13([
-  (0, import_type_graphql14.ObjectType)({
+TextMessageOutput = _ts_decorate15([
+  (0, import_type_graphql16.ObjectType)({
     implements: BaseMessageOutput
   })
 ], TextMessageOutput);
@@ -4334,31 +4567,31 @@ var ActionExecutionMessageOutput = class {
   parentMessageId;
 };
 __name(ActionExecutionMessageOutput, "ActionExecutionMessageOutput");
-_ts_decorate13([
-  (0, import_type_graphql14.Field)(() => String),
-  _ts_metadata13("design:type", String)
+_ts_decorate15([
+  (0, import_type_graphql16.Field)(() => String),
+  _ts_metadata15("design:type", String)
 ], ActionExecutionMessageOutput.prototype, "name", void 0);
-_ts_decorate13([
-  (0, import_type_graphql14.Field)(() => String, {
+_ts_decorate15([
+  (0, import_type_graphql16.Field)(() => String, {
     nullable: true,
     deprecationReason: "This field will be removed in a future version"
   }),
-  _ts_metadata13("design:type", String)
+  _ts_metadata15("design:type", String)
 ], ActionExecutionMessageOutput.prototype, "scope", void 0);
-_ts_decorate13([
-  (0, import_type_graphql14.Field)(() => [
+_ts_decorate15([
+  (0, import_type_graphql16.Field)(() => [
     String
   ]),
-  _ts_metadata13("design:type", Array)
+  _ts_metadata15("design:type", Array)
 ], ActionExecutionMessageOutput.prototype, "arguments", void 0);
-_ts_decorate13([
-  (0, import_type_graphql14.Field)(() => String, {
+_ts_decorate15([
+  (0, import_type_graphql16.Field)(() => String, {
     nullable: true
   }),
-  _ts_metadata13("design:type", String)
+  _ts_metadata15("design:type", String)
 ], ActionExecutionMessageOutput.prototype, "parentMessageId", void 0);
-ActionExecutionMessageOutput = _ts_decorate13([
-  (0, import_type_graphql14.ObjectType)({
+ActionExecutionMessageOutput = _ts_decorate15([
+  (0, import_type_graphql16.ObjectType)({
     implements: BaseMessageOutput
   })
 ], ActionExecutionMessageOutput);
@@ -4368,20 +4601,20 @@ var ResultMessageOutput = class {
   result;
 };
 __name(ResultMessageOutput, "ResultMessageOutput");
-_ts_decorate13([
-  (0, import_type_graphql14.Field)(() => String),
-  _ts_metadata13("design:type", String)
+_ts_decorate15([
+  (0, import_type_graphql16.Field)(() => String),
+  _ts_metadata15("design:type", String)
 ], ResultMessageOutput.prototype, "actionExecutionId", void 0);
-_ts_decorate13([
-  (0, import_type_graphql14.Field)(() => String),
-  _ts_metadata13("design:type", String)
+_ts_decorate15([
+  (0, import_type_graphql16.Field)(() => String),
+  _ts_metadata15("design:type", String)
 ], ResultMessageOutput.prototype, "actionName", void 0);
-_ts_decorate13([
-  (0, import_type_graphql14.Field)(() => String),
-  _ts_metadata13("design:type", String)
+_ts_decorate15([
+  (0, import_type_graphql16.Field)(() => String),
+  _ts_metadata15("design:type", String)
 ], ResultMessageOutput.prototype, "result", void 0);
-ResultMessageOutput = _ts_decorate13([
-  (0, import_type_graphql14.ObjectType)({
+ResultMessageOutput = _ts_decorate15([
+  (0, import_type_graphql16.ObjectType)({
     implements: BaseMessageOutput
   })
 ], ResultMessageOutput);
@@ -4396,40 +4629,40 @@ var AgentStateMessageOutput = class {
   running;
 };
 __name(AgentStateMessageOutput, "AgentStateMessageOutput");
-_ts_decorate13([
-  (0, import_type_graphql14.Field)(() => String),
-  _ts_metadata13("design:type", String)
+_ts_decorate15([
+  (0, import_type_graphql16.Field)(() => String),
+  _ts_metadata15("design:type", String)
 ], AgentStateMessageOutput.prototype, "threadId", void 0);
-_ts_decorate13([
-  (0, import_type_graphql14.Field)(() => String),
-  _ts_metadata13("design:type", String)
+_ts_decorate15([
+  (0, import_type_graphql16.Field)(() => String),
+  _ts_metadata15("design:type", String)
 ], AgentStateMessageOutput.prototype, "agentName", void 0);
-_ts_decorate13([
-  (0, import_type_graphql14.Field)(() => String),
-  _ts_metadata13("design:type", String)
+_ts_decorate15([
+  (0, import_type_graphql16.Field)(() => String),
+  _ts_metadata15("design:type", String)
 ], AgentStateMessageOutput.prototype, "nodeName", void 0);
-_ts_decorate13([
-  (0, import_type_graphql14.Field)(() => String),
-  _ts_metadata13("design:type", String)
+_ts_decorate15([
+  (0, import_type_graphql16.Field)(() => String),
+  _ts_metadata15("design:type", String)
 ], AgentStateMessageOutput.prototype, "runId", void 0);
-_ts_decorate13([
-  (0, import_type_graphql14.Field)(() => Boolean),
-  _ts_metadata13("design:type", Boolean)
+_ts_decorate15([
+  (0, import_type_graphql16.Field)(() => Boolean),
+  _ts_metadata15("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)
+_ts_decorate15([
+  (0, import_type_graphql16.Field)(() => MessageRole),
+  _ts_metadata15("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)
+_ts_decorate15([
+  (0, import_type_graphql16.Field)(() => String),
+  _ts_metadata15("design:type", String)
 ], AgentStateMessageOutput.prototype, "state", void 0);
-_ts_decorate13([
-  (0, import_type_graphql14.Field)(() => Boolean),
-  _ts_metadata13("design:type", Boolean)
+_ts_decorate15([
+  (0, import_type_graphql16.Field)(() => Boolean),
+  _ts_metadata15("design:type", Boolean)
 ], AgentStateMessageOutput.prototype, "running", void 0);
-AgentStateMessageOutput = _ts_decorate13([
-  (0, import_type_graphql14.ObjectType)({
+AgentStateMessageOutput = _ts_decorate15([
+  (0, import_type_graphql16.ObjectType)({
     implements: BaseMessageOutput
   })
 ], AgentStateMessageOutput);
@@ -4438,30 +4671,37 @@ var CopilotResponse = class {
   status;
   runId;
   messages;
+  extensions;
 };
 __name(CopilotResponse, "CopilotResponse");
-_ts_decorate13([
-  (0, import_type_graphql14.Field)(() => String),
-  _ts_metadata13("design:type", String)
+_ts_decorate15([
+  (0, import_type_graphql16.Field)(() => String),
+  _ts_metadata15("design:type", String)
 ], CopilotResponse.prototype, "threadId", void 0);
-_ts_decorate13([
-  (0, import_type_graphql14.Field)(() => ResponseStatusUnion),
-  _ts_metadata13("design:type", Object)
+_ts_decorate15([
+  (0, import_type_graphql16.Field)(() => ResponseStatusUnion),
+  _ts_metadata15("design:type", Object)
 ], CopilotResponse.prototype, "status", void 0);
-_ts_decorate13([
-  (0, import_type_graphql14.Field)({
+_ts_decorate15([
+  (0, import_type_graphql16.Field)({
     nullable: true
   }),
-  _ts_metadata13("design:type", String)
+  _ts_metadata15("design:type", String)
 ], CopilotResponse.prototype, "runId", void 0);
-_ts_decorate13([
-  (0, import_type_graphql14.Field)(() => [
+_ts_decorate15([
+  (0, import_type_graphql16.Field)(() => [
     BaseMessageOutput
   ]),
-  _ts_metadata13("design:type", Array)
+  _ts_metadata15("design:type", Array)
 ], CopilotResponse.prototype, "messages", void 0);
-CopilotResponse = _ts_decorate13([
-  (0, import_type_graphql14.ObjectType)()
+_ts_decorate15([
+  (0, import_type_graphql16.Field)(() => ExtensionsResponse, {
+    nullable: true
+  }),
+  _ts_metadata15("design:type", typeof ExtensionsResponse === "undefined" ? Object : ExtensionsResponse)
+], CopilotResponse.prototype, "extensions", void 0);
+CopilotResponse = _ts_decorate15([
+  (0, import_type_graphql16.ObjectType)()
 ], CopilotResponse);
 
 // src/graphql/resolvers/copilot.resolver.ts
@@ -4504,11 +4744,11 @@ var UnknownErrorResponse = class extends FailedResponseStatus {
 __name(UnknownErrorResponse, "UnknownErrorResponse");
 
 // src/graphql/resolvers/copilot.resolver.ts
-var import_shared13 = require("@copilotkit/shared");
+var import_shared16 = require("@copilotkit/shared");
 
 // src/graphql/types/agents-response.type.ts
-var import_type_graphql15 = require("type-graphql");
-function _ts_decorate14(decorators, target, key, desc) {
+var import_type_graphql17 = require("type-graphql");
+function _ts_decorate16(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);
@@ -4518,49 +4758,49 @@ function _ts_decorate14(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_decorate14, "_ts_decorate");
-function _ts_metadata14(k, v) {
+__name(_ts_decorate16, "_ts_decorate");
+function _ts_metadata16(k, v) {
   if (typeof Reflect === "object" && typeof Reflect.metadata === "function")
     return Reflect.metadata(k, v);
 }
-__name(_ts_metadata14, "_ts_metadata");
+__name(_ts_metadata16, "_ts_metadata");
 var Agent = class {
   id;
   name;
   description;
 };
 __name(Agent, "Agent");
-_ts_decorate14([
-  (0, import_type_graphql15.Field)(() => String),
-  _ts_metadata14("design:type", String)
+_ts_decorate16([
+  (0, import_type_graphql17.Field)(() => String),
+  _ts_metadata16("design:type", String)
 ], Agent.prototype, "id", void 0);
-_ts_decorate14([
-  (0, import_type_graphql15.Field)(() => String),
-  _ts_metadata14("design:type", String)
+_ts_decorate16([
+  (0, import_type_graphql17.Field)(() => String),
+  _ts_metadata16("design:type", String)
 ], Agent.prototype, "name", void 0);
-_ts_decorate14([
-  (0, import_type_graphql15.Field)(() => String),
-  _ts_metadata14("design:type", String)
+_ts_decorate16([
+  (0, import_type_graphql17.Field)(() => String),
+  _ts_metadata16("design:type", String)
 ], Agent.prototype, "description", void 0);
-Agent = _ts_decorate14([
-  (0, import_type_graphql15.ObjectType)()
+Agent = _ts_decorate16([
+  (0, import_type_graphql17.ObjectType)()
 ], Agent);
 var AgentsResponse = class {
   agents;
 };
 __name(AgentsResponse, "AgentsResponse");
-_ts_decorate14([
-  (0, import_type_graphql15.Field)(() => [
+_ts_decorate16([
+  (0, import_type_graphql17.Field)(() => [
     Agent
   ]),
-  _ts_metadata14("design:type", Array)
+  _ts_metadata16("design:type", Array)
 ], AgentsResponse.prototype, "agents", void 0);
-AgentsResponse = _ts_decorate14([
-  (0, import_type_graphql15.ObjectType)()
+AgentsResponse = _ts_decorate16([
+  (0, import_type_graphql17.ObjectType)()
 ], AgentsResponse);
 
 // src/graphql/resolvers/copilot.resolver.ts
-function _ts_decorate15(decorators, target, key, desc) {
+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")
     r = Reflect.decorate(decorators, target, key, desc);
@@ -4570,12 +4810,12 @@ function _ts_decorate15(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_decorate15, "_ts_decorate");
-function _ts_metadata15(k, v) {
+__name(_ts_decorate17, "_ts_decorate");
+function _ts_metadata17(k, v) {
   if (typeof Reflect === "object" && typeof Reflect.metadata === "function")
     return Reflect.metadata(k, v);
 }
-__name(_ts_metadata15, "_ts_metadata");
+__name(_ts_metadata17, "_ts_metadata");
 function _ts_param(paramIndex, decorator) {
   return function(target, key) {
     decorator(target, key, paramIndex);
@@ -4622,10 +4862,10 @@ var CopilotResolver = class {
       component: "CopilotResolver.availableAgents"
     });
     logger2.debug("Processing");
-    const agents = await ctx._copilotkit.runtime.discoverAgentsFromEndpoints(ctx);
+    const agentsWithEndpoints = await ctx._copilotkit.runtime.discoverAgentsFromEndpoints(ctx);
     logger2.debug("Event source created, creating response");
     return {
-      agents
+      agents: agentsWithEndpoints.map(({ endpoint, ...agentWithoutEndpoint }) => agentWithoutEndpoint)
     };
   }
   async generateCopilotResponse(ctx, data, properties) {
@@ -4687,7 +4927,7 @@ var CopilotResolver = class {
       rejectOutputMessagesPromise = reject;
     });
     logger2.debug("Processing");
-    const { eventSource, threadId = (0, import_shared13.randomId)(), runId, serverSideActions, actionInputsWithoutAgents } = await copilotRuntime.processRuntimeRequest({
+    const { eventSource, threadId = (0, import_shared16.randomId)(), runId, serverSideActions, actionInputsWithoutAgents, extensions } = await copilotRuntime.processRuntimeRequest({
       serviceAdapter,
       messages: data.messages,
       actions: data.frontend.actions.filter((action) => action.available !== ActionInputAvailability.disabled),
@@ -4699,13 +4939,15 @@ var CopilotResolver = class {
       forwardedParameters: data.forwardedParameters,
       agentSession: data.agentSession,
       agentStates: data.agentStates,
-      url: data.frontend.url
+      url: data.frontend.url,
+      extensions: data.extensions
     });
     logger2.debug("Event source created, creating response");
     const response = {
       threadId,
       runId,
       status: (0, import_rxjs4.firstValueFrom)(responseStatus$),
+      extensions,
       messages: new import_graphql_yoga.Repeater(async (pushMessage, stopStreamingMessages) => {
         var _a2, _b2;
         logger2.debug("Messages repeater created");
@@ -4732,7 +4974,7 @@ var CopilotResolver = class {
                 });
                 outputMessages = [
                   (0, import_class_transformer3.plainToInstance)(TextMessage, {
-                    id: (0, import_shared13.randomId)(),
+                    id: (0, import_shared16.randomId)(),
                     createdAt: /* @__PURE__ */ new Date(),
                     content: result.reason,
                     role: MessageRole.assistant
@@ -4762,7 +5004,8 @@ var CopilotResolver = class {
           actionInputsWithoutAgents: actionInputsWithoutAgents.filter(
             // TODO-AGENTS: do not exclude ALL server side actions
             (action) => !serverSideActions.find((serverSideAction) => serverSideAction.name == action.name)
-          )
+          ),
+          threadId
         }).pipe(
           // shareReplay() ensures that later subscribers will see the whole stream instead of
           // just the events that were emitted after the subscriber was added.
@@ -4923,7 +5166,7 @@ var CopilotResolver = class {
                   event
                 }, "Agent message event received");
                 pushMessage({
-                  id: (0, import_shared13.randomId)(),
+                  id: (0, import_shared16.randomId)(),
                   status: new SuccessMessageStatus(),
                   threadId: event.threadId,
                   agentName: event.agentName,
@@ -4936,7 +5179,7 @@ var CopilotResolver = class {
                   createdAt: /* @__PURE__ */ new Date()
                 });
                 outputMessages.push((0, import_class_transformer3.plainToInstance)(AgentStateMessage, {
-                  id: (0, import_shared13.randomId)(),
+                  id: (0, import_shared16.randomId)(),
                   threadId: event.threadId,
                   agentName: event.agentName,
                   nodeName: event.nodeName,
@@ -4980,38 +5223,38 @@ var CopilotResolver = class {
   }
 };
 __name(CopilotResolver, "CopilotResolver");
-_ts_decorate15([
-  (0, import_type_graphql16.Query)(() => String),
-  _ts_metadata15("design:type", Function),
-  _ts_metadata15("design:paramtypes", []),
-  _ts_metadata15("design:returntype", Promise)
+_ts_decorate17([
+  (0, import_type_graphql18.Query)(() => String),
+  _ts_metadata17("design:type", Function),
+  _ts_metadata17("design:paramtypes", []),
+  _ts_metadata17("design:returntype", Promise)
 ], CopilotResolver.prototype, "hello", null);
-_ts_decorate15([
-  (0, import_type_graphql16.Query)(() => AgentsResponse),
-  _ts_param(0, (0, import_type_graphql16.Ctx)()),
-  _ts_metadata15("design:type", Function),
-  _ts_metadata15("design:paramtypes", [
+_ts_decorate17([
+  (0, import_type_graphql18.Query)(() => AgentsResponse),
+  _ts_param(0, (0, import_type_graphql18.Ctx)()),
+  _ts_metadata17("design:type", Function),
+  _ts_metadata17("design:paramtypes", [
     typeof GraphQLContext === "undefined" ? Object : GraphQLContext
   ]),
-  _ts_metadata15("design:returntype", Promise)
+  _ts_metadata17("design:returntype", Promise)
 ], CopilotResolver.prototype, "availableAgents", null);
-_ts_decorate15([
-  (0, import_type_graphql16.Mutation)(() => CopilotResponse),
-  _ts_param(0, (0, import_type_graphql16.Ctx)()),
-  _ts_param(1, (0, import_type_graphql16.Arg)("data")),
-  _ts_param(2, (0, import_type_graphql16.Arg)("properties", () => import_graphql_scalars2.GraphQLJSONObject, {
+_ts_decorate17([
+  (0, import_type_graphql18.Mutation)(() => CopilotResponse),
+  _ts_param(0, (0, import_type_graphql18.Ctx)()),
+  _ts_param(1, (0, import_type_graphql18.Arg)("data")),
+  _ts_param(2, (0, import_type_graphql18.Arg)("properties", () => import_graphql_scalars2.GraphQLJSONObject, {
     nullable: true
   })),
-  _ts_metadata15("design:type", Function),
-  _ts_metadata15("design:paramtypes", [
+  _ts_metadata17("design:type", Function),
+  _ts_metadata17("design:paramtypes", [
     typeof GraphQLContext === "undefined" ? Object : GraphQLContext,
     typeof GenerateCopilotResponseInput === "undefined" ? Object : GenerateCopilotResponseInput,
     typeof CopilotRequestContextProperties === "undefined" ? Object : CopilotRequestContextProperties
   ]),
-  _ts_metadata15("design:returntype", Promise)
+  _ts_metadata17("design:returntype", Promise)
 ], CopilotResolver.prototype, "generateCopilotResponse", null);
-CopilotResolver = _ts_decorate15([
-  (0, import_type_graphql16.Resolver)(() => CopilotResponse)
+CopilotResolver = _ts_decorate17([
+  (0, import_type_graphql18.Resolver)(() => CopilotResponse)
 ], CopilotResolver);
 
 // src/lib/integrations/shared.ts
@@ -5045,6 +5288,141 @@ function createLogger(options) {
 }
 __name(createLogger, "createLogger");
 
+// src/graphql/resolvers/state.resolver.ts
+var import_type_graphql21 = require("type-graphql");
+var import_type_graphql22 = require("type-graphql");
+var import_type_graphql23 = require("type-graphql");
+
+// src/graphql/types/load-agent-state-response.type.ts
+var import_type_graphql19 = require("type-graphql");
+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 LoadAgentStateResponse = class {
+  threadId;
+  threadExists;
+  state;
+  messages;
+};
+__name(LoadAgentStateResponse, "LoadAgentStateResponse");
+_ts_decorate18([
+  (0, import_type_graphql19.Field)(() => String),
+  _ts_metadata18("design:type", String)
+], LoadAgentStateResponse.prototype, "threadId", void 0);
+_ts_decorate18([
+  (0, import_type_graphql19.Field)(() => Boolean),
+  _ts_metadata18("design:type", Boolean)
+], LoadAgentStateResponse.prototype, "threadExists", void 0);
+_ts_decorate18([
+  (0, import_type_graphql19.Field)(() => String),
+  _ts_metadata18("design:type", String)
+], LoadAgentStateResponse.prototype, "state", void 0);
+_ts_decorate18([
+  (0, import_type_graphql19.Field)(() => String),
+  _ts_metadata18("design:type", String)
+], LoadAgentStateResponse.prototype, "messages", void 0);
+LoadAgentStateResponse = _ts_decorate18([
+  (0, import_type_graphql19.ObjectType)()
+], LoadAgentStateResponse);
+
+// src/graphql/inputs/load-agent-state.input.ts
+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);
+  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_decorate19, "_ts_decorate");
+function _ts_metadata19(k, v) {
+  if (typeof Reflect === "object" && typeof Reflect.metadata === "function")
+    return Reflect.metadata(k, v);
+}
+__name(_ts_metadata19, "_ts_metadata");
+var LoadAgentStateInput = class {
+  threadId;
+  agentName;
+};
+__name(LoadAgentStateInput, "LoadAgentStateInput");
+_ts_decorate19([
+  (0, import_type_graphql20.Field)(() => String),
+  _ts_metadata19("design:type", String)
+], LoadAgentStateInput.prototype, "threadId", void 0);
+_ts_decorate19([
+  (0, import_type_graphql20.Field)(() => String),
+  _ts_metadata19("design:type", String)
+], LoadAgentStateInput.prototype, "agentName", void 0);
+LoadAgentStateInput = _ts_decorate19([
+  (0, import_type_graphql20.InputType)()
+], LoadAgentStateInput);
+
+// src/graphql/resolvers/state.resolver.ts
+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);
+  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_decorate20, "_ts_decorate");
+function _ts_metadata20(k, v) {
+  if (typeof Reflect === "object" && typeof Reflect.metadata === "function")
+    return Reflect.metadata(k, v);
+}
+__name(_ts_metadata20, "_ts_metadata");
+function _ts_param2(paramIndex, decorator) {
+  return function(target, key) {
+    decorator(target, key, paramIndex);
+  };
+}
+__name(_ts_param2, "_ts_param");
+var StateResolver = class {
+  async loadAgentState(ctx, data) {
+    const agents = await ctx._copilotkit.runtime.discoverAgentsFromEndpoints(ctx);
+    const agent = agents.find((agent2) => agent2.name === data.agentName);
+    if (!agent) {
+      throw new Error("Agent not found");
+    }
+    const state = await ctx._copilotkit.runtime.loadAgentState(ctx, data.threadId, data.agentName);
+    return state;
+  }
+};
+__name(StateResolver, "StateResolver");
+_ts_decorate20([
+  (0, import_type_graphql23.Query)(() => LoadAgentStateResponse),
+  _ts_param2(0, (0, import_type_graphql22.Ctx)()),
+  _ts_param2(1, (0, import_type_graphql21.Arg)("data")),
+  _ts_metadata20("design:type", Function),
+  _ts_metadata20("design:paramtypes", [
+    typeof GraphQLContext === "undefined" ? Object : GraphQLContext,
+    typeof LoadAgentStateInput === "undefined" ? Object : LoadAgentStateInput
+  ]),
+  _ts_metadata20("design:returntype", Promise)
+], StateResolver.prototype, "loadAgentState", null);
+StateResolver = _ts_decorate20([
+  (0, import_type_graphql21.Resolver)(() => LoadAgentStateResponse)
+], StateResolver);
+
 // src/lib/integrations/shared.ts
 var logger = createLogger();
 async function createContext(initialContext, copilotKitContext, contextLogger, properties = {}) {
@@ -5066,9 +5444,10 @@ async function createContext(initialContext, copilotKitContext, contextLogger, p
 __name(createContext, "createContext");
 function buildSchema(options = {}) {
   logger.debug("Building GraphQL schema...");
-  const schema = (0, import_type_graphql17.buildSchemaSync)({
+  const schema = (0, import_type_graphql24.buildSchemaSync)({
     resolvers: [
-      CopilotResolver
+      CopilotResolver,
+      StateResolver
     ],
     emitSchemaFile: options.emitSchemaFile
   });