@copilotkit/runtime 1.9.1 → 1.9.2-next.1

package/dist/lib/index.js

@@ -44,7 +44,7 @@ var require_package = __commonJS({
       publishConfig: {
         access: "public"
       },
-      version: "1.9.1",
+      version: "1.9.2-next.1",
       sideEffects: false,
       main: "./dist/index.js",
       module: "./dist/index.mjs",
@@ -2734,14 +2734,14 @@ GenerateCopilotResponseInput = _ts_decorate17([
 var import_graphql_yoga = require("graphql-yoga");
 
 // src/service-adapters/events.ts
-var import_shared20 = require("@copilotkit/shared");
+var import_shared21 = require("@copilotkit/shared");
 var import_rxjs3 = require("rxjs");
 
 // src/lib/telemetry-client.ts
-var import_shared19 = require("@copilotkit/shared");
+var import_shared20 = require("@copilotkit/shared");
 
 // src/lib/runtime/remote-actions.ts
-var import_shared14 = require("@copilotkit/shared");
+var import_shared15 = require("@copilotkit/shared");
 
 // src/lib/runtime/remote-action-constructors.ts
 var import_node_crypto2 = require("crypto");
@@ -2982,6 +2982,9 @@ var RemoteLangGraphEventSource = class {
       return events;
     }), (0, import_rxjs.catchError)((error) => {
       console.error(error);
+      if (error instanceof import_shared7.CopilotKitError || (error == null ? void 0 : error.name) && error.name.includes("CopilotKit")) {
+        throw error;
+      }
       const events = [];
       if ((lastEventWithState == null ? void 0 : lastEventWithState.lastMessageId) && !lastEventWithState.isToolCall) {
         events.push({
@@ -3022,32 +3025,119 @@ var import_shared8 = require("@copilotkit/shared");
 var import_partial_json = require("partial-json");
 var import_shared9 = require("@copilotkit/shared");
 var import_messages3 = require("@langchain/core/messages");
+
+// src/lib/runtime/retry-utils.ts
+var RETRY_CONFIG = {
+  maxRetries: 3,
+  baseDelayMs: 1e3,
+  maxDelayMs: 5e3,
+  // HTTP status codes that should be retried
+  retryableStatusCodes: [
+    502,
+    503,
+    504,
+    408,
+    429
+  ],
+  // Network error patterns that should be retried
+  retryableErrorMessages: [
+    "fetch failed",
+    "network error",
+    "connection timeout",
+    "ECONNREFUSED",
+    "ETIMEDOUT",
+    "ENOTFOUND",
+    "ECONNRESET"
+  ]
+};
+function isRetryableError(error, response) {
+  var _a, _b;
+  if (response && RETRY_CONFIG.retryableStatusCodes.includes(response.status)) {
+    return true;
+  }
+  const errorCode = ((_a = error == null ? void 0 : error.cause) == null ? void 0 : _a.code) || (error == null ? void 0 : error.code);
+  if (errorCode && RETRY_CONFIG.retryableErrorMessages.includes(errorCode)) {
+    return true;
+  }
+  const errorMessage = ((_b = error == null ? void 0 : error.message) == null ? void 0 : _b.toLowerCase()) || "";
+  return RETRY_CONFIG.retryableErrorMessages.some((msg) => errorMessage.includes(msg));
+}
+__name(isRetryableError, "isRetryableError");
+function sleep(ms) {
+  return new Promise((resolve) => setTimeout(resolve, ms));
+}
+__name(sleep, "sleep");
+function calculateDelay(attempt) {
+  const delay = RETRY_CONFIG.baseDelayMs * Math.pow(2, attempt);
+  return Math.min(delay, RETRY_CONFIG.maxDelayMs);
+}
+__name(calculateDelay, "calculateDelay");
+async function fetchWithRetry(url, options, logger2) {
+  let lastError;
+  for (let attempt = 0; attempt <= RETRY_CONFIG.maxRetries; attempt++) {
+    try {
+      const response = await fetch(url, options);
+      if (isRetryableError(null, response) && attempt < RETRY_CONFIG.maxRetries) {
+        const delay = calculateDelay(attempt);
+        logger2 == null ? void 0 : logger2.warn(`Request to ${url} failed with status ${response.status}. Retrying attempt ${attempt + 1}/${RETRY_CONFIG.maxRetries + 1} in ${delay}ms.`);
+        await sleep(delay);
+        continue;
+      }
+      return response;
+    } catch (error) {
+      lastError = error;
+      if (isRetryableError(error) && attempt < RETRY_CONFIG.maxRetries) {
+        const delay = calculateDelay(attempt);
+        logger2 == null ? void 0 : logger2.warn(`Request to ${url} failed with network error. Retrying attempt ${attempt + 1}/${RETRY_CONFIG.maxRetries + 1} in ${delay}ms. Error: ${(error == null ? void 0 : error.message) || String(error)}`);
+        await sleep(delay);
+        continue;
+      }
+      break;
+    }
+  }
+  throw lastError;
+}
+__name(fetchWithRetry, "fetchWithRetry");
+
+// src/lib/runtime/remote-lg-action.ts
 var activeInterruptEvent = false;
 async function execute(args) {
   return new ReadableStream({
     async start(controller) {
-      try {
-        await streamEvents(controller, args);
-        controller.close();
-      } catch (err) {
-        const cause = err == null ? void 0 : err.cause;
-        const errorCode = (cause == null ? void 0 : cause.code) || (err == null ? void 0 : err.code);
-        if (errorCode === "ECONNREFUSED") {
-          throw new import_shared9.CopilotKitMisuseError({
-            message: `
-              The LangGraph client could not connect to the graph. Please further check previous logs, which includes further details.
-              
-              See more: https://docs.copilotkit.ai/troubleshooting/common-issues`
-          });
-        } else {
-          throw new import_shared9.CopilotKitMisuseError({
-            message: `
-              The LangGraph client threw unhandled error ${err}.
-              
-              See more: https://docs.copilotkit.ai/troubleshooting/common-issues`
-          });
+      let lastError;
+      for (let attempt = 0; attempt <= RETRY_CONFIG.maxRetries; attempt++) {
+        try {
+          await streamEvents(controller, args);
+          controller.close();
+          return;
+        } catch (err) {
+          lastError = err;
+          if (isRetryableError(err) && attempt < RETRY_CONFIG.maxRetries) {
+            const delay = calculateDelay(attempt);
+            console.warn(`LangGraph connection attempt ${attempt + 1}/${RETRY_CONFIG.maxRetries + 1} failed. Retrying in ${delay}ms. Error: ${(err == null ? void 0 : err.message) || String(err)}`);
+            await sleep(delay);
+            continue;
+          }
+          break;
         }
       }
+      const cause = lastError == null ? void 0 : lastError.cause;
+      const errorCode = (cause == null ? void 0 : cause.code) || (lastError == null ? void 0 : lastError.code);
+      if (errorCode === "ECONNREFUSED") {
+        throw new import_shared9.CopilotKitMisuseError({
+          message: `
+            The LangGraph client could not connect to the graph after ${RETRY_CONFIG.maxRetries + 1} attempts. Please further check previous logs, which includes further details.
+            
+            See more: https://docs.copilotkit.ai/troubleshooting/common-issues`
+        });
+      } else {
+        throw new import_shared9.CopilotKitMisuseError({
+          message: `
+            The LangGraph client threw unhandled error ${lastError}.
+            
+            See more: https://docs.copilotkit.ai/troubleshooting/common-issues`
+        });
+      }
     }
   });
 }
@@ -3647,9 +3737,10 @@ function filterObjectBySchemaKeys(obj, schemaKeys) {
 __name(filterObjectBySchemaKeys, "filterObjectBySchemaKeys");
 
 // src/lib/runtime/remote-action-constructors.ts
-var import_shared10 = require("@copilotkit/shared");
+var import_shared11 = require("@copilotkit/shared");
 
 // src/lib/streaming.ts
+var import_shared10 = require("@copilotkit/shared");
 async function writeJsonLineResponseToEventStream(response, eventStream$) {
   const reader = response.getReader();
   const decoder = new TextDecoder();
@@ -3688,16 +3779,44 @@ async function writeJsonLineResponseToEventStream(response, eventStream$) {
     }
   } catch (error) {
     console.error("Error in stream", error);
-    eventStream$.error(error);
+    const structuredError = convertStreamingErrorToStructured(error);
+    eventStream$.error(structuredError);
     return;
   }
   eventStream$.complete();
 }
 __name(writeJsonLineResponseToEventStream, "writeJsonLineResponseToEventStream");
+function convertStreamingErrorToStructured(error) {
+  var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j;
+  if (((_a = error == null ? void 0 : error.message) == null ? void 0 : _a.includes("terminated")) || ((_b = error == null ? void 0 : error.cause) == null ? void 0 : _b.code) === "UND_ERR_SOCKET" || ((_c = error == null ? void 0 : error.message) == null ? void 0 : _c.includes("other side closed")) || (error == null ? void 0 : error.code) === "UND_ERR_SOCKET") {
+    return new import_shared10.CopilotKitError({
+      message: "Connection to agent was unexpectedly terminated. This is likely due to the agent service being down or experiencing issues. Please check your agent logs and try again.",
+      code: import_shared10.CopilotKitErrorCode.NETWORK_ERROR
+    });
+  }
+  if (((_d = error == null ? void 0 : error.message) == null ? void 0 : _d.includes("fetch failed")) || ((_e = error == null ? void 0 : error.message) == null ? void 0 : _e.includes("ECONNREFUSED")) || ((_f = error == null ? void 0 : error.message) == null ? void 0 : _f.includes("ENOTFOUND")) || ((_g = error == null ? void 0 : error.message) == null ? void 0 : _g.includes("ETIMEDOUT"))) {
+    return new import_shared10.CopilotKitLowLevelError({
+      error: error instanceof Error ? error : new Error(String(error)),
+      url: "streaming connection",
+      message: "Network error occurred during streaming. Please check your connection and try again."
+    });
+  }
+  if (((_h = error == null ? void 0 : error.message) == null ? void 0 : _h.includes("aborted")) || ((_i = error == null ? void 0 : error.message) == null ? void 0 : _i.includes("canceled")) || ((_j = error == null ? void 0 : error.message) == null ? void 0 : _j.includes("signal is aborted"))) {
+    return new import_shared10.CopilotKitError({
+      message: "Request was cancelled",
+      code: import_shared10.CopilotKitErrorCode.UNKNOWN
+    });
+  }
+  return new import_shared10.CopilotKitError({
+    message: `Streaming error: ${(error == null ? void 0 : error.message) || String(error)}`,
+    code: import_shared10.CopilotKitErrorCode.UNKNOWN
+  });
+}
+__name(convertStreamingErrorToStructured, "convertStreamingErrorToStructured");
 
 // src/lib/runtime/remote-action-constructors.ts
-var import_shared11 = require("@copilotkit/shared");
 var import_shared12 = require("@copilotkit/shared");
+var import_shared13 = require("@copilotkit/shared");
 function constructLGCRemoteAction({ endpoint, graphqlContext, logger: logger2, messages, agentStates }) {
   const agents = endpoint.agents.map((agent) => ({
     name: agent.name,
@@ -3720,8 +3839,8 @@ function constructLGCRemoteAction({ endpoint, graphqlContext, logger: logger2, m
       if (agentStates) {
         const jsonState = agentStates.find((state2) => state2.agentName === name);
         if (jsonState) {
-          state = (0, import_shared12.parseJson)(jsonState.state, {});
-          config2 = (0, import_shared12.parseJson)(jsonState.config, {});
+          state = (0, import_shared13.parseJson)(jsonState.state, {});
+          config2 = (0, import_shared13.parseJson)(jsonState.config, {});
         }
       }
       try {
@@ -3741,7 +3860,7 @@ function constructLGCRemoteAction({ endpoint, graphqlContext, logger: logger2, m
           state,
           config: config2,
           properties: graphqlContext.properties,
-          actions: (0, import_shared12.tryMap)(actionInputsWithoutAgents, (action) => ({
+          actions: (0, import_shared13.tryMap)(actionInputsWithoutAgents, (action) => ({
             name: action.name,
             description: action.description,
             parameters: JSON.parse(action.jsonSchema)
@@ -3790,7 +3909,7 @@ function constructRemoteActions({ json, url, onBeforeRequest, graphqlContext, lo
       });
       const fetchUrl = `${url}/actions/execute`;
       try {
-        const response = await fetch(fetchUrl, {
+        const response = await fetchWithRetry(fetchUrl, {
           method: "POST",
           headers,
           body: JSON.stringify({
@@ -3798,7 +3917,7 @@ function constructRemoteActions({ json, url, onBeforeRequest, graphqlContext, lo
             arguments: args,
             properties: graphqlContext.properties
           })
-        });
+        }, logger2);
         if (!response.ok) {
           logger2.error({
             url,
@@ -3806,11 +3925,11 @@ function constructRemoteActions({ json, url, onBeforeRequest, graphqlContext, lo
             body: await response.text()
           }, "Failed to execute remote action");
           if (response.status === 404) {
-            throw new import_shared11.CopilotKitApiDiscoveryError({
+            throw new import_shared12.CopilotKitApiDiscoveryError({
               url: fetchUrl
             });
           }
-          throw new import_shared11.ResolvedCopilotKitError({
+          throw new import_shared12.ResolvedCopilotKitError({
             status: response.status,
             url: fetchUrl,
             isRemoteEndpoint: true
@@ -3824,10 +3943,10 @@ function constructRemoteActions({ json, url, onBeforeRequest, graphqlContext, lo
         }, "Executed remote action");
         return result;
       } catch (error) {
-        if (error instanceof import_shared10.CopilotKitError) {
+        if (error instanceof import_shared11.CopilotKitError) {
           throw error;
         }
-        throw new import_shared10.CopilotKitLowLevelError({
+        throw new import_shared11.CopilotKitLowLevelError({
           error,
           url: fetchUrl
         });
@@ -3855,13 +3974,13 @@ function constructRemoteActions({ json, url, onBeforeRequest, graphqlContext, lo
       if (agentStates) {
         const jsonState = agentStates.find((state2) => state2.agentName === name);
         if (jsonState) {
-          state = (0, import_shared12.parseJson)(jsonState.state, {});
-          config2 = (0, import_shared12.parseJson)(jsonState.config, {});
+          state = (0, import_shared13.parseJson)(jsonState.state, {});
+          config2 = (0, import_shared13.parseJson)(jsonState.config, {});
         }
       }
       const fetchUrl = `${url}/agents/execute`;
       try {
-        const response = await fetch(fetchUrl, {
+        const response = await fetchWithRetry(fetchUrl, {
           method: "POST",
           headers,
           body: JSON.stringify({
@@ -3875,14 +3994,14 @@ function constructRemoteActions({ json, url, onBeforeRequest, graphqlContext, lo
             state,
             config: config2,
             properties: graphqlContext.properties,
-            actions: (0, import_shared12.tryMap)(actionInputsWithoutAgents, (action) => ({
+            actions: (0, import_shared13.tryMap)(actionInputsWithoutAgents, (action) => ({
               name: action.name,
               description: action.description,
               parameters: JSON.parse(action.jsonSchema)
             })),
             metaEvents
           })
-        });
+        }, logger2);
         if (!response.ok) {
           logger2.error({
             url,
@@ -3890,11 +4009,11 @@ function constructRemoteActions({ json, url, onBeforeRequest, graphqlContext, lo
             body: await response.text()
           }, "Failed to execute remote agent");
           if (response.status === 404) {
-            throw new import_shared11.CopilotKitApiDiscoveryError({
+            throw new import_shared12.CopilotKitApiDiscoveryError({
               url: fetchUrl
             });
           }
-          throw new import_shared11.ResolvedCopilotKitError({
+          throw new import_shared12.ResolvedCopilotKitError({
             status: response.status,
             url: fetchUrl,
             isRemoteEndpoint: true
@@ -3912,10 +4031,10 @@ function constructRemoteActions({ json, url, onBeforeRequest, graphqlContext, lo
           throw new Error("Unsupported agent type");
         }
       } catch (error) {
-        if (error instanceof import_shared10.CopilotKitError) {
+        if (error instanceof import_shared11.CopilotKitError) {
           throw error;
         }
-        throw new import_shared10.CopilotKitLowLevelError({
+        throw new import_shared11.CopilotKitLowLevelError({
           error,
           url: fetchUrl
         });
@@ -3945,10 +4064,10 @@ function createHeaders(onBeforeRequest, graphqlContext) {
 __name(createHeaders, "createHeaders");
 
 // src/lib/runtime/remote-actions.ts
-var import_shared15 = require("@copilotkit/shared");
+var import_shared16 = require("@copilotkit/shared");
 
 // src/lib/runtime/agui-action.ts
-var import_shared13 = require("@copilotkit/shared");
+var import_shared14 = require("@copilotkit/shared");
 function constructAGUIRemoteAction({ logger: logger2, messages, agentStates, agent, metaEvents }) {
   const action = {
     name: agent.agentId,
@@ -3973,7 +4092,7 @@ function constructAGUIRemoteAction({ logger: logger2, messages, agentStates, age
       if (agentStates) {
         const jsonState = agentStates.find((state2) => state2.agentName === agent.agentId);
         if (jsonState) {
-          state = (0, import_shared13.parseJson)(jsonState.state, {});
+          state = (0, import_shared14.parseJson)(jsonState.state, {});
         }
       }
       agent.state = state;
@@ -4080,7 +4199,7 @@ async function fetchRemoteInfo({ url, onBeforeRequest, graphqlContext, logger: l
         status: response.status,
         body: await response.text()
       }, "Failed to fetch actions from url");
-      throw new import_shared15.ResolvedCopilotKitError({
+      throw new import_shared16.ResolvedCopilotKitError({
         status: response.status,
         url: fetchUrl,
         isRemoteEndpoint: true
@@ -4092,10 +4211,10 @@ async function fetchRemoteInfo({ url, onBeforeRequest, graphqlContext, logger: l
     }, "Fetched actions from url");
     return json;
   } catch (error) {
-    if (error instanceof import_shared15.CopilotKitError) {
+    if (error instanceof import_shared16.CopilotKitError) {
       throw error;
     }
-    throw new import_shared15.CopilotKitLowLevelError({
+    throw new import_shared16.CopilotKitLowLevelError({
       error,
       url: fetchUrl
     });
@@ -4153,9 +4272,9 @@ async function setupRemoteActions({ remoteEndpointDefinitions, graphqlContext, m
   }));
   for (const [key, agent] of Object.entries(agents)) {
     if (agent.agentId !== void 0 && agent.agentId !== key) {
-      throw new import_shared15.CopilotKitError({
+      throw new import_shared16.CopilotKitError({
         message: `Agent ${key} has agentId ${agent.agentId} which does not match the key ${key}`,
-        code: import_shared14.CopilotKitErrorCode.UNKNOWN
+        code: import_shared15.CopilotKitErrorCode.UNKNOWN
       });
     } else if (agent.agentId === void 0) {
       agent.agentId = key;
@@ -4176,7 +4295,7 @@ __name(setupRemoteActions, "setupRemoteActions");
 var import_node_crypto3 = require("crypto");
 
 // src/lib/runtime/copilot-runtime.ts
-var import_shared18 = require("@copilotkit/shared");
+var import_shared19 = require("@copilotkit/shared");
 
 // src/service-adapters/langchain/langserve.ts
 var import_remote = require("langchain/runnables/remote");
@@ -4257,11 +4376,11 @@ var RemoteChain = class {
 __name(RemoteChain, "RemoteChain");
 
 // src/service-adapters/empty/empty-adapter.ts
-var import_shared16 = require("@copilotkit/shared");
+var import_shared17 = require("@copilotkit/shared");
 var EmptyAdapter = class {
   async process(request) {
     return {
-      threadId: request.threadId || (0, import_shared16.randomUUID)()
+      threadId: request.threadId || (0, import_shared17.randomUUID)()
     };
   }
 };
@@ -4394,9 +4513,9 @@ __name(ImageMessage, "ImageMessage");
 
 // src/service-adapters/conversion.ts
 var import_class_transformer = require("class-transformer");
-var import_shared17 = require("@copilotkit/shared");
+var import_shared18 = require("@copilotkit/shared");
 function convertGqlInputToMessages(inputMessages) {
-  const messages = (0, import_shared17.tryMap)(inputMessages, (message) => {
+  const messages = (0, import_shared18.tryMap)(inputMessages, (message) => {
     if (message.textMessage) {
       return (0, import_class_transformer.plainToInstance)(TextMessage, {
         id: message.id,
@@ -4590,7 +4709,7 @@ var CopilotRuntime = class {
     this.mcpServersConfig = params == null ? void 0 : params.mcpServers;
     this.createMCPClientImpl = params == null ? void 0 : params.createMCPClient;
     if (this.mcpServersConfig && this.mcpServersConfig.length > 0 && !this.createMCPClientImpl) {
-      throw new import_shared18.CopilotKitMisuseError({
+      throw new import_shared19.CopilotKitMisuseError({
         message: "MCP Integration Error: `mcpServers` were provided, but the `createMCPClient` function was not passed to the CopilotRuntime constructor. Please provide an implementation for `createMCPClient`."
       });
     }
@@ -4646,7 +4765,7 @@ var CopilotRuntime = class {
       }
     } else {
       newMessages.unshift({
-        id: (0, import_shared18.randomId)(),
+        id: (0, import_shared19.randomId)(),
         createdAt: /* @__PURE__ */ new Date(),
         textMessage: {
           role: MessageRole.system,
@@ -4666,11 +4785,16 @@ var CopilotRuntime = class {
     const requestStartTime = Date.now();
     const streamedChunks = [];
     try {
+      if (Object.keys(this.agents).length && (agentSession == null ? void 0 : agentSession.agentName) && !this.delegateAgentProcessingToServiceAdapter) {
+        this.agents = {
+          [agentSession.agentName]: this.agents[agentSession.agentName]
+        };
+      }
       if (agentSession && !this.delegateAgentProcessingToServiceAdapter) {
         return await this.processAgentRequest(request);
       }
       if (serviceAdapter instanceof EmptyAdapter) {
-        throw new import_shared18.CopilotKitMisuseError({
+        throw new import_shared19.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.`
@@ -4700,7 +4824,7 @@ please use an LLM adapter instead.`
       const serverSideActionsInput = serverSideActions.map((action) => ({
         name: action.name,
         description: action.description,
-        jsonSchema: JSON.stringify((0, import_shared18.actionParametersToJsonSchema)(action.parameters))
+        jsonSchema: JSON.stringify((0, import_shared19.actionParametersToJsonSchema)(action.parameters))
       }));
       const actionInputs = flattenToolCallsNoDuplicates([
         ...serverSideActionsInput,
@@ -4829,12 +4953,12 @@ please use an LLM adapter instead.`
           console.error("Error logging LLM error:", logError);
         }
       }
-      if (error instanceof import_shared18.CopilotKitError) {
+      if (error instanceof import_shared19.CopilotKitError) {
         throw error;
       }
       console.error("Error getting response:", error);
-      eventSource.sendErrorMessageToChat();
-      throw error;
+      const structuredError = this.convertStreamingErrorToStructured(error);
+      throw structuredError;
     }
   }
   async discoverAgentsFromEndpoints(graphqlContext) {
@@ -4855,12 +4979,12 @@ please use an LLM adapter instead.`
         try {
           data = await client.assistants.search();
           if (data && "detail" in data && data.detail.toLowerCase() === "not found") {
-            throw new import_shared18.CopilotKitAgentDiscoveryError({
+            throw new import_shared19.CopilotKitAgentDiscoveryError({
               availableAgents: this.availableAgents
             });
           }
         } catch (e) {
-          throw new import_shared18.CopilotKitMisuseError({
+          throw new import_shared19.CopilotKitMisuseError({
             message: `
               Failed to find or contact remote endpoint at url ${endpoint.deploymentUrl}.
               Make sure the API is running and that it's indeed a LangGraph platform url.
@@ -4882,7 +5006,7 @@ please use an LLM adapter instead.`
       const cpkEndpoint = endpoint;
       const fetchUrl = `${endpoint.url}/info`;
       try {
-        const response = await fetch(fetchUrl, {
+        const response = await fetchWithRetry(fetchUrl, {
           method: "POST",
           headers: createHeaders(cpkEndpoint.onBeforeRequest, graphqlContext),
           body: JSON.stringify({
@@ -4891,11 +5015,11 @@ please use an LLM adapter instead.`
         });
         if (!response.ok) {
           if (response.status === 404) {
-            throw new import_shared18.CopilotKitApiDiscoveryError({
+            throw new import_shared19.CopilotKitApiDiscoveryError({
               url: fetchUrl
             });
           }
-          throw new import_shared18.ResolvedCopilotKitError({
+          throw new import_shared19.ResolvedCopilotKitError({
             status: response.status,
             url: fetchUrl,
             isRemoteEndpoint: true
@@ -4905,7 +5029,7 @@ please use an LLM adapter instead.`
         const endpointAgents = ((data == null ? void 0 : data.agents) ?? []).map((agent) => ({
           name: agent.name,
           description: agent.description ?? "" ?? "",
-          id: (0, import_shared18.randomId)(),
+          id: (0, import_shared19.randomId)(),
           endpoint
         }));
         return [
@@ -4913,10 +5037,10 @@ please use an LLM adapter instead.`
           ...endpointAgents
         ];
       } catch (error) {
-        if (error instanceof import_shared18.CopilotKitError) {
+        if (error instanceof import_shared19.CopilotKitError) {
           throw error;
         }
-        throw new import_shared18.CopilotKitLowLevelError({
+        throw new import_shared19.CopilotKitLowLevelError({
           error,
           url: fetchUrl
         });
@@ -4971,7 +5095,7 @@ please use an LLM adapter instead.`
       const cpkEndpoint = agentWithEndpoint.endpoint;
       const fetchUrl = `${cpkEndpoint.url}/agents/state`;
       try {
-        const response = await fetch(fetchUrl, {
+        const response = await fetchWithRetry(fetchUrl, {
           method: "POST",
           headers: createHeaders(cpkEndpoint.onBeforeRequest, graphqlContext),
           body: JSON.stringify({
@@ -4982,11 +5106,11 @@ please use an LLM adapter instead.`
         });
         if (!response.ok) {
           if (response.status === 404) {
-            throw new import_shared18.CopilotKitApiDiscoveryError({
+            throw new import_shared19.CopilotKitApiDiscoveryError({
               url: fetchUrl
             });
           }
-          throw new import_shared18.ResolvedCopilotKitError({
+          throw new import_shared19.ResolvedCopilotKitError({
             status: response.status,
             url: fetchUrl,
             isRemoteEndpoint: true
@@ -4999,10 +5123,10 @@ please use an LLM adapter instead.`
           messages: JSON.stringify(data.messages)
         };
       } catch (error) {
-        if (error instanceof import_shared18.CopilotKitError) {
+        if (error instanceof import_shared19.CopilotKitError) {
           throw error;
         }
-        throw new import_shared18.CopilotKitLowLevelError({
+        throw new import_shared19.CopilotKitLowLevelError({
           error,
           url: fetchUrl
         });
@@ -5022,7 +5146,7 @@ please use an LLM adapter instead.`
     const messages = convertGqlInputToMessages(rawMessages);
     const currentAgent = serverSideActions.find((action) => action.name === agentName && isRemoteAgentAction(action));
     if (!currentAgent) {
-      throw new import_shared18.CopilotKitAgentDiscoveryError({
+      throw new import_shared19.CopilotKitAgentDiscoveryError({
         agentName,
         availableAgents: this.availableAgents
       });
@@ -5034,7 +5158,7 @@ please use an LLM adapter instead.`
     )).map((action) => ({
       name: action.name,
       description: action.description,
-      jsonSchema: JSON.stringify((0, import_shared18.actionParametersToJsonSchema)(action.parameters))
+      jsonSchema: JSON.stringify((0, import_shared19.actionParametersToJsonSchema)(action.parameters))
     }));
     const allAvailableActions = flattenToolCallsNoDuplicates([
       ...availableActionsForCurrentAgent,
@@ -5134,7 +5258,8 @@ please use an LLM adapter instead.`
                 console.error("Error logging agent error:", logError);
               }
             }
-            eventStream$.error(err);
+            const structuredError = this.convertStreamingErrorToStructured(err);
+            eventStream$.error(structuredError);
             eventStream$.complete();
           },
           complete: () => eventStream$.complete()
@@ -5296,6 +5421,32 @@ please use an LLM adapter instead.`
       return "langchain";
     return void 0;
   }
+  convertStreamingErrorToStructured(error) {
+    var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j;
+    if (((_a = error == null ? void 0 : error.message) == null ? void 0 : _a.includes("terminated")) || ((_b = error == null ? void 0 : error.cause) == null ? void 0 : _b.code) === "UND_ERR_SOCKET" || ((_c = error == null ? void 0 : error.message) == null ? void 0 : _c.includes("other side closed")) || (error == null ? void 0 : error.code) === "UND_ERR_SOCKET") {
+      return new import_shared19.CopilotKitError({
+        message: "Connection to agent was unexpectedly terminated. This may be due to the agent service being restarted or network issues. Please try again.",
+        code: import_shared19.CopilotKitErrorCode.NETWORK_ERROR
+      });
+    }
+    if (((_d = error == null ? void 0 : error.message) == null ? void 0 : _d.includes("fetch failed")) || ((_e = error == null ? void 0 : error.message) == null ? void 0 : _e.includes("ECONNREFUSED")) || ((_f = error == null ? void 0 : error.message) == null ? void 0 : _f.includes("ENOTFOUND")) || ((_g = error == null ? void 0 : error.message) == null ? void 0 : _g.includes("ETIMEDOUT"))) {
+      return new import_shared19.CopilotKitLowLevelError({
+        error: error instanceof Error ? error : new Error(String(error)),
+        url: "agent streaming connection",
+        message: "Network error occurred during agent streaming. Please check your connection and try again."
+      });
+    }
+    if (((_h = error == null ? void 0 : error.message) == null ? void 0 : _h.includes("aborted")) || ((_i = error == null ? void 0 : error.message) == null ? void 0 : _i.includes("canceled")) || ((_j = error == null ? void 0 : error.message) == null ? void 0 : _j.includes("signal is aborted"))) {
+      return new import_shared19.CopilotKitError({
+        message: "Agent request was cancelled",
+        code: import_shared19.CopilotKitErrorCode.UNKNOWN
+      });
+    }
+    return new import_shared19.CopilotKitError({
+      message: `Agent streaming error: ${(error == null ? void 0 : error.message) || String(error)}`,
+      code: import_shared19.CopilotKitErrorCode.UNKNOWN
+    });
+  }
 };
 __name(CopilotRuntime, "CopilotRuntime");
 function flattenToolCallsNoDuplicates(toolsByPriority) {
@@ -5338,7 +5489,7 @@ __name(resolveEndpointType, "resolveEndpointType");
 
 // src/lib/telemetry-client.ts
 var packageJson = require_package();
-var telemetryClient = new import_shared19.TelemetryClient({
+var telemetryClient = new import_shared20.TelemetryClient({
   packageName: packageJson.name,
   packageVersion: packageJson.version
 });
@@ -5517,16 +5668,17 @@ var RuntimeEventSource = class {
     const errorMessage = `\u274C ${message}`;
     if (!this.callback) {
       this.stream(async (eventStream$) => {
-        eventStream$.sendTextMessage((0, import_shared20.randomId)(), errorMessage);
+        eventStream$.sendTextMessage((0, import_shared21.randomId)(), errorMessage);
       });
     } else {
-      this.eventStream$.sendTextMessage((0, import_shared20.randomId)(), errorMessage);
+      this.eventStream$.sendTextMessage((0, import_shared21.randomId)(), errorMessage);
     }
   }
   processRuntimeEvents({ serverSideActions, guardrailsResult$, actionInputsWithoutAgents, threadId }) {
     this.callback(this.eventStream$).catch((error) => {
       console.error("Error in event source callback", error);
-      this.sendErrorMessageToChat();
+      const structuredError = convertStreamingErrorToStructured2(error);
+      this.eventStream$.error(structuredError);
       this.eventStream$.complete();
     });
     return this.eventStream$.pipe(
@@ -5565,7 +5717,15 @@ var RuntimeEventSource = class {
           telemetry_client_default.capture("oss.runtime.server_action_executed", {});
           return (0, import_rxjs3.concat)((0, import_rxjs3.of)(eventWithState.event), toolCallEventStream$).pipe((0, import_rxjs3.catchError)((error) => {
             console.error("Error in tool call stream", error);
-            this.sendErrorMessageToChat();
+            const structuredError = convertStreamingErrorToStructured2(error);
+            toolCallEventStream$.sendActionExecutionResult({
+              actionExecutionId: eventWithState.actionExecutionId,
+              actionName: eventWithState.action.name,
+              error: {
+                code: structuredError.code,
+                message: structuredError.message
+              }
+            });
             return import_rxjs3.EMPTY;
           }));
         } else {
@@ -5638,12 +5798,13 @@ async function executeAction(eventStream$, guardrailsResult$, action, actionArgu
       next: (event) => eventStream$.next(event),
       error: (err) => {
         console.error("Error in stream", err);
+        const structuredError = convertStreamingErrorToStructured2(err);
         eventStream$.sendActionExecutionResult({
           actionExecutionId,
           actionName: action.name,
           error: {
-            code: "STREAM_ERROR",
-            message: err.message
+            code: structuredError.code,
+            message: structuredError.message
           }
         });
         eventStream$.complete();
@@ -5676,6 +5837,40 @@ async function executeAction(eventStream$, guardrailsResult$, action, actionArgu
   }
 }
 __name(executeAction, "executeAction");
+function convertStreamingErrorToStructured2(error) {
+  var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j;
+  if (((_a = error == null ? void 0 : error.message) == null ? void 0 : _a.includes("terminated")) || ((_b = error == null ? void 0 : error.cause) == null ? void 0 : _b.code) === "UND_ERR_SOCKET" || ((_c = error == null ? void 0 : error.message) == null ? void 0 : _c.includes("other side closed")) || (error == null ? void 0 : error.code) === "UND_ERR_SOCKET") {
+    return new import_shared21.CopilotKitError({
+      message: "Connection to agent was unexpectedly terminated. This may be due to the agent service being restarted or network issues. Please try again.",
+      code: import_shared21.CopilotKitErrorCode.NETWORK_ERROR
+    });
+  }
+  if (((_d = error == null ? void 0 : error.message) == null ? void 0 : _d.includes("fetch failed")) || ((_e = error == null ? void 0 : error.message) == null ? void 0 : _e.includes("ECONNREFUSED")) || ((_f = error == null ? void 0 : error.message) == null ? void 0 : _f.includes("ENOTFOUND")) || ((_g = error == null ? void 0 : error.message) == null ? void 0 : _g.includes("ETIMEDOUT"))) {
+    return new import_shared21.CopilotKitLowLevelError({
+      error: error instanceof Error ? error : new Error(String(error)),
+      url: "event streaming connection",
+      message: "Network error occurred during event streaming. Please check your connection and try again."
+    });
+  }
+  if (((_h = error == null ? void 0 : error.message) == null ? void 0 : _h.includes("aborted")) || ((_i = error == null ? void 0 : error.message) == null ? void 0 : _i.includes("canceled")) || ((_j = error == null ? void 0 : error.message) == null ? void 0 : _j.includes("signal is aborted"))) {
+    return new import_shared21.CopilotKitError({
+      message: "Request was cancelled",
+      code: import_shared21.CopilotKitErrorCode.UNKNOWN
+    });
+  }
+  const errorMessage = (error == null ? void 0 : error.message) || String(error);
+  if (errorMessage.includes("401") || errorMessage.toLowerCase().includes("api key") || errorMessage.toLowerCase().includes("unauthorized") || errorMessage.toLowerCase().includes("authentication") || errorMessage.toLowerCase().includes("incorrect api key")) {
+    return new import_shared21.CopilotKitError({
+      message: `Event streaming error: ${errorMessage}`,
+      code: import_shared21.CopilotKitErrorCode.MISSING_PUBLIC_API_KEY_ERROR
+    });
+  }
+  return new import_shared21.CopilotKitError({
+    message: `Event streaming error: ${errorMessage}`,
+    code: import_shared21.CopilotKitErrorCode.UNKNOWN
+  });
+}
+__name(convertStreamingErrorToStructured2, "convertStreamingErrorToStructured");
 
 // src/graphql/resolvers/copilot.resolver.ts
 var import_graphql_scalars2 = require("graphql-scalars");
@@ -5716,7 +5911,7 @@ var UnknownErrorResponse = class extends FailedResponseStatus {
 __name(UnknownErrorResponse, "UnknownErrorResponse");
 
 // src/graphql/resolvers/copilot.resolver.ts
-var import_shared21 = require("@copilotkit/shared");
+var import_shared22 = require("@copilotkit/shared");
 
 // src/graphql/types/agents-response.type.ts
 var import_type_graphql19 = require("type-graphql");
@@ -5772,6 +5967,7 @@ AgentsResponse = _ts_decorate18([
 ], AgentsResponse);
 
 // src/graphql/resolvers/copilot.resolver.ts
+var import_shared23 = require("@copilotkit/shared");
 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")
@@ -5911,7 +6107,7 @@ var CopilotResolver = class {
       ctx.properties["copilotCloudPublicApiKey"] = copilotCloudPublicApiKey;
     }
     logger2.debug("Processing");
-    const { eventSource, threadId = (0, import_shared21.randomId)(), runId, serverSideActions, actionInputsWithoutAgents, extensions } = await copilotRuntime.processRuntimeRequest({
+    const { eventSource, threadId = (0, import_shared22.randomId)(), runId, serverSideActions, actionInputsWithoutAgents, extensions } = await copilotRuntime.processRuntimeRequest({
       serviceAdapter,
       messages: data.messages,
       actions: data.frontend.actions.filter((action) => action.available !== ActionInputAvailability.disabled),
@@ -6054,7 +6250,7 @@ var CopilotResolver = class {
                 });
                 outputMessages = [
                   (0, import_class_transformer3.plainToInstance)(TextMessage, {
-                    id: (0, import_shared21.randomId)(),
+                    id: (0, import_shared22.randomId)(),
                     createdAt: /* @__PURE__ */ new Date(),
                     content: result.reason,
                     role: MessageRole.assistant
@@ -6232,7 +6428,7 @@ var CopilotResolver = class {
                   event
                 }, "Agent message event received");
                 pushMessage({
-                  id: (0, import_shared21.randomId)(),
+                  id: (0, import_shared22.randomId)(),
                   status: new SuccessMessageStatus(),
                   threadId: event.threadId,
                   agentName: event.agentName,
@@ -6245,7 +6441,7 @@ var CopilotResolver = class {
                   createdAt: /* @__PURE__ */ new Date()
                 });
                 outputMessages.push((0, import_class_transformer3.plainToInstance)(AgentStateMessage, {
-                  id: (0, import_shared21.randomId)(),
+                  id: (0, import_shared22.randomId)(),
                   threadId: event.threadId,
                   agentName: event.agentName,
                   nodeName: event.nodeName,
@@ -6263,6 +6459,12 @@ var CopilotResolver = class {
             logger2.error({
               err
             }, "Error in event stream");
+            if (err instanceof import_shared23.CopilotKitError || err instanceof Error && err.name && err.name.includes("CopilotKit")) {
+              eventStreamSubscription == null ? void 0 : eventStreamSubscription.unsubscribe();
+              rejectOutputMessagesPromise(err);
+              stopStreamingMessages(err);
+              return;
+            }
             responseStatus$.next(new UnknownErrorResponse({
               description: `An unknown error has occurred in the event stream`
             }));
@@ -6440,6 +6642,7 @@ LoadAgentStateInput = _ts_decorate21([
 ], LoadAgentStateInput);
 
 // src/graphql/resolvers/state.resolver.ts
+var import_shared24 = require("@copilotkit/shared");
 function _ts_decorate22(decorators, target, key, desc) {
   var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
   if (typeof Reflect === "object" && typeof Reflect.decorate === "function")
@@ -6467,12 +6670,13 @@ var StateResolver = class {
     const agents = await ctx._copilotkit.runtime.discoverAgentsFromEndpoints(ctx);
     const agent = agents.find((agent2) => agent2.name === data.agentName);
     if (!agent) {
-      return {
-        threadId: data.threadId || "",
-        threadExists: false,
-        state: JSON.stringify({}),
-        messages: JSON.stringify([])
-      };
+      throw new import_shared24.CopilotKitAgentDiscoveryError({
+        agentName: data.agentName,
+        availableAgents: agents.map((a) => ({
+          name: a.name,
+          id: a.name
+        }))
+      });
     }
     const state = await ctx._copilotkit.runtime.loadAgentState(ctx, data.threadId, data.agentName);
     return state;