@mastra/koa 1.5.7-alpha.4 → 1.5.7-alpha.8

package/CHANGELOG.md

@@ -1,5 +1,56 @@
 # @mastra/koa
 
+## 1.5.7-alpha.8
+
+### Patch Changes
+
+- Improved agent thread subscription resilience by keeping server streams active during idle periods and allowing the JavaScript client to reconnect when subscription streams close or resubscribe requests fail. ([#17045](https://github.com/mastra-ai/mastra/pull/17045))
+
+  Enable automatic reconnection with `subscription.processDataStream({ onChunk: chunk => console.log(chunk), reconnect: true })`.
+
+- Updated dependencies [[`c35b962`](https://github.com/mastra-ai/mastra/commit/c35b9625c7e854fcfdeee226a3338a750d0ff211), [`c35b962`](https://github.com/mastra-ai/mastra/commit/c35b9625c7e854fcfdeee226a3338a750d0ff211), [`4084113`](https://github.com/mastra-ai/mastra/commit/408411370fc48a822e8b616b3b63f9409774e0e9), [`bc01b1b`](https://github.com/mastra-ai/mastra/commit/bc01b1bfafe381d90af909f8bce7eeb4eee779f2)]:
+  - @mastra/core@1.37.0-alpha.8
+  - @mastra/server@1.37.0-alpha.8
+
+## 1.5.7-alpha.7
+
+### Patch Changes
+
+- Updated dependencies [[`168fa09`](https://github.com/mastra-ai/mastra/commit/168fa09d6b39114cb8c13bd06f1dccb9bc81c6cd)]:
+  - @mastra/core@1.37.0-alpha.7
+  - @mastra/server@1.37.0-alpha.7
+
+## 1.5.7-alpha.6
+
+### Patch Changes
+
+- Developers can now cancel long-running custom API route work when clients disconnect. Node-based adapters pass abort signals into custom route handlers, clean up response streams correctly, and still surface upstream response body errors. ([#16335](https://github.com/mastra-ai/mastra/pull/16335))
+
+  ```ts
+  registerApiRoute('/stream', {
+    method: 'GET',
+    handler: async c => {
+      const stream = await agent.stream(prompt, {
+        abortSignal: c.req.raw.signal,
+      });
+
+      return stream.toTextStreamResponse();
+    },
+  });
+  ```
+
+- Updated dependencies [[`fafed7a`](https://github.com/mastra-ai/mastra/commit/fafed7a24dc320f7c92ee872c347f4be087fd689), [`0cbece9`](https://github.com/mastra-ai/mastra/commit/0cbece9d832cb134a74cdbf3682d390a058215a4), [`7dfe1bc`](https://github.com/mastra-ai/mastra/commit/7dfe1bcfe71d261a6fd6bbf29b1dec49d78fb98f), [`70cb714`](https://github.com/mastra-ai/mastra/commit/70cb7149c8f16f478e15b58498254a53181750a4), [`7f9da22`](https://github.com/mastra-ai/mastra/commit/7f9da22efd5aa595e138a31de55a5f0f2f28b33d)]:
+  - @mastra/server@1.37.0-alpha.6
+  - @mastra/core@1.37.0-alpha.6
+
+## 1.5.7-alpha.5
+
+### Patch Changes
+
+- Updated dependencies [[`6096445`](https://github.com/mastra-ai/mastra/commit/60964459733f0ab384584d95e19c36607ffdf7b0), [`91cf0e0`](https://github.com/mastra-ai/mastra/commit/91cf0e027e511b871481a8576b56b7af83b15afd)]:
+  - @mastra/core@1.37.0-alpha.5
+  - @mastra/server@1.37.0-alpha.5
+
 ## 1.5.7-alpha.4
 
 ### Patch Changes

package/dist/index.cjs

@@ -6,6 +6,9 @@ var serverAdapter = require('@mastra/server/server-adapter');
 var requestContext = require('@mastra/core/request-context');
 
 // src/index.ts
+
+// ../../node_modules/.pnpm/zod@4.4.3/node_modules/zod/v4/core/core.js
+var _a;
 // @__NO_SIDE_EFFECTS__
 function $constructor(name, initializer3, params) {
   function init(inst, def) {
@@ -58,15 +61,16 @@ function $constructor(name, initializer3, params) {
   Object.defineProperty(_, "name", { value: name });
   return _;
 }
+(_a = globalThis).__zod_globalConfig ?? (_a.__zod_globalConfig = {});
 
-// ../../node_modules/.pnpm/zod@4.3.6/node_modules/zod/v4/core/util.js
+// ../../node_modules/.pnpm/zod@4.4.3/node_modules/zod/v4/core/util.js
 function jsonStringifyReplacer(_, value) {
   if (typeof value === "bigint")
     return value.toString();
   return value;
 }
 
-// ../../node_modules/.pnpm/zod@4.3.6/node_modules/zod/v4/core/errors.js
+// ../../node_modules/.pnpm/zod@4.4.3/node_modules/zod/v4/core/errors.js
 var initializer = (inst, def) => {
   inst.name = "$ZodError";
   Object.defineProperty(inst, "_zod", {
@@ -84,7 +88,7 @@ var initializer = (inst, def) => {
   });
 };
 var $ZodError = $constructor("$ZodError", initializer);
-function flattenError(error, mapper = (issue2) => issue2.message) {
+function flattenError(error, mapper = (issue) => issue.message) {
   const fieldErrors = {};
   const formErrors = [];
   for (const sub of error.issues) {
@@ -97,32 +101,35 @@ function flattenError(error, mapper = (issue2) => issue2.message) {
   }
   return { formErrors, fieldErrors };
 }
-function formatError(error, mapper = (issue2) => issue2.message) {
+function formatError(error, mapper = (issue) => issue.message) {
   const fieldErrors = { _errors: [] };
-  const processError = (error2) => {
-    for (const issue2 of error2.issues) {
-      if (issue2.code === "invalid_union" && issue2.errors.length) {
-        issue2.errors.map((issues) => processError({ issues }));
-      } else if (issue2.code === "invalid_key") {
-        processError({ issues: issue2.issues });
-      } else if (issue2.code === "invalid_element") {
-        processError({ issues: issue2.issues });
-      } else if (issue2.path.length === 0) {
-        fieldErrors._errors.push(mapper(issue2));
+  const processError = (error2, path = []) => {
+    for (const issue of error2.issues) {
+      if (issue.code === "invalid_union" && issue.errors.length) {
+        issue.errors.map((issues) => processError({ issues }, [...path, ...issue.path]));
+      } else if (issue.code === "invalid_key") {
+        processError({ issues: issue.issues }, [...path, ...issue.path]);
+      } else if (issue.code === "invalid_element") {
+        processError({ issues: issue.issues }, [...path, ...issue.path]);
       } else {
-        let curr = fieldErrors;
-        let i = 0;
-        while (i < issue2.path.length) {
-          const el = issue2.path[i];
-          const terminal = i === issue2.path.length - 1;
-          if (!terminal) {
-            curr[el] = curr[el] || { _errors: [] };
-          } else {
-            curr[el] = curr[el] || { _errors: [] };
-            curr[el]._errors.push(mapper(issue2));
+        const fullpath = [...path, ...issue.path];
+        if (fullpath.length === 0) {
+          fieldErrors._errors.push(mapper(issue));
+        } else {
+          let curr = fieldErrors;
+          let i = 0;
+          while (i < fullpath.length) {
+            const el = fullpath[i];
+            const terminal = i === fullpath.length - 1;
+            if (!terminal) {
+              curr[el] = curr[el] || { _errors: [] };
+            } else {
+              curr[el] = curr[el] || { _errors: [] };
+              curr[el]._errors.push(mapper(issue));
+            }
+            curr = curr[el];
+            i++;
           }
-          curr = curr[el];
-          i++;
         }
       }
     }
@@ -131,54 +138,7 @@ function formatError(error, mapper = (issue2) => issue2.message) {
   return fieldErrors;
 }
 
-// ../../node_modules/.pnpm/zod@4.3.6/node_modules/zod/v4/core/registries.js
-var _a;
-var $ZodRegistry = class {
-  constructor() {
-    this._map = /* @__PURE__ */ new WeakMap();
-    this._idmap = /* @__PURE__ */ new Map();
-  }
-  add(schema, ..._meta) {
-    const meta = _meta[0];
-    this._map.set(schema, meta);
-    if (meta && typeof meta === "object" && "id" in meta) {
-      this._idmap.set(meta.id, schema);
-    }
-    return this;
-  }
-  clear() {
-    this._map = /* @__PURE__ */ new WeakMap();
-    this._idmap = /* @__PURE__ */ new Map();
-    return this;
-  }
-  remove(schema) {
-    const meta = this._map.get(schema);
-    if (meta && typeof meta === "object" && "id" in meta) {
-      this._idmap.delete(meta.id);
-    }
-    this._map.delete(schema);
-    return this;
-  }
-  get(schema) {
-    const p = schema._zod.parent;
-    if (p) {
-      const pm = { ...this.get(p) ?? {} };
-      delete pm.id;
-      const f = { ...pm, ...this._map.get(schema) };
-      return Object.keys(f).length ? f : void 0;
-    }
-    return this._map.get(schema);
-  }
-  has(schema) {
-    return this._map.has(schema);
-  }
-};
-function registry() {
-  return new $ZodRegistry();
-}
-(_a = globalThis).__zod_globalRegistry ?? (_a.__zod_globalRegistry = registry());
-
-// ../../node_modules/.pnpm/zod@4.3.6/node_modules/zod/v4/classic/errors.js
+// ../../node_modules/.pnpm/zod@4.4.3/node_modules/zod/v4/classic/errors.js
 var initializer2 = (inst, issues) => {
   $ZodError.init(inst, issues);
   inst.name = "ZodError";
@@ -192,8 +152,8 @@ var initializer2 = (inst, issues) => {
       // enumerable: false,
     },
     addIssue: {
-      value: (issue2) => {
-        inst.issues.push(issue2);
+      value: (issue) => {
+        inst.issues.push(issue);
         inst.message = JSON.stringify(inst.issues, jsonStringifyReplacer, 2);
       }
       // enumerable: false,
@@ -213,7 +173,7 @@ var initializer2 = (inst, issues) => {
     }
   });
 };
-var ZodError = $constructor("ZodError", initializer2);
+var ZodError = /* @__PURE__ */ $constructor("ZodError", initializer2);
 function toWebRequest(ctx) {
   const protocol = ctx.protocol || "http";
   const host = ctx.host || "localhost";
@@ -676,6 +636,10 @@ var MastraServer = class extends serverAdapter.MastraServer {
         const { done, value } = await reader.read();
         if (done) break;
         if (value) {
+          if (streamFormat === "sse" && typeof value === "string" && value.startsWith(":")) {
+            ctx.res.write(value);
+            continue;
+          }
           const shouldRedact = this.streamOptions?.redact ?? true;
           const outputValue = shouldRedact ? serverAdapter.redactStreamChunk(value) : value;
           if (streamFormat === "sse") {
@@ -915,77 +879,94 @@ var MastraServer = class extends serverAdapter.MastraServer {
       );
       const shouldRunCustomRouteAuth = auth.isProtectedCustomRoute(path, method, server.customRouteAuthConfig);
       const shouldRunCustomRouteFGA = !!matchedRoute?.route.fga;
-      if (shouldRunCustomRouteAuth || shouldRunCustomRouteFGA) {
-        const serverRoute = {
-          method: matchedRoute?.route.method ?? method,
-          path: matchedRoute?.route.path ?? path,
-          responseType: "json",
-          handler: async () => {
-          },
-          requiresAuth: matchedRoute?.route.requiresAuth,
-          requiresPermission: matchedRoute?.route.requiresPermission,
-          fga: matchedRoute?.route.fga
-        };
-        if (shouldRunCustomRouteAuth) {
-          const authError = await server.checkRouteAuth(serverRoute, {
-            path,
-            method,
-            getHeader: (name) => ctx.headers[name.toLowerCase()],
-            getQuery: (name) => ctx.query[name],
-            requestContext: ctx.state.requestContext,
-            request: toWebRequest2(ctx),
-            buildAuthorizeContext: () => toWebRequest2(ctx)
-          });
-          if (authError) {
-            if (authError.headers) {
-              for (const [key, value] of Object.entries(authError.headers)) {
-                ctx.set(key, value);
+      const customRouteAbortController = new AbortController();
+      const abortCustomRoute = () => {
+        customRouteAbortController.abort();
+      };
+      const abortCustomRouteIfOpen = () => {
+        if (!ctx.res.writableEnded) {
+          abortCustomRoute();
+        }
+      };
+      ctx.res.once("close", abortCustomRouteIfOpen);
+      ctx.res.once("error", abortCustomRouteIfOpen);
+      try {
+        if (shouldRunCustomRouteAuth || shouldRunCustomRouteFGA) {
+          const serverRoute = {
+            method: matchedRoute?.route.method ?? method,
+            path: matchedRoute?.route.path ?? path,
+            responseType: "json",
+            handler: async () => {
+            },
+            requiresAuth: matchedRoute?.route.requiresAuth,
+            requiresPermission: matchedRoute?.route.requiresPermission,
+            fga: matchedRoute?.route.fga
+          };
+          if (shouldRunCustomRouteAuth) {
+            const authError = await server.checkRouteAuth(serverRoute, {
+              path,
+              method,
+              getHeader: (name) => ctx.headers[name.toLowerCase()],
+              getQuery: (name) => ctx.query[name],
+              requestContext: ctx.state.requestContext,
+              request: toWebRequest2(ctx),
+              buildAuthorizeContext: () => toWebRequest2(ctx)
+            });
+            if (authError) {
+              if (authError.headers) {
+                for (const [key, value] of Object.entries(authError.headers)) {
+                  ctx.set(key, value);
+                }
               }
-            }
-            if (authError.error) {
-              ctx.status = authError.status;
-              ctx.body = { error: authError.error };
-              return;
-            }
-          }
-          const authConfig = server.mastra.getServer()?.auth;
-          if (authConfig) {
-            const hasPermission = await loadHasPermission();
-            if (hasPermission) {
-              const userPermissions = ctx.state.requestContext.get("mastra__userPermissions");
-              const permissionError = server.checkRoutePermission(serverRoute, userPermissions, hasPermission);
-              if (permissionError) {
-                ctx.status = permissionError.status;
-                ctx.body = {
-                  error: permissionError.error,
-                  message: permissionError.message
-                };
+              if (authError.error) {
+                ctx.status = authError.status;
+                ctx.body = { error: authError.error };
                 return;
               }
             }
+            const authConfig = server.mastra.getServer()?.auth;
+            if (authConfig) {
+              const hasPermission = await loadHasPermission();
+              if (hasPermission) {
+                const userPermissions = ctx.state.requestContext.get("mastra__userPermissions");
+                const permissionError = server.checkRoutePermission(serverRoute, userPermissions, hasPermission);
+                if (permissionError) {
+                  ctx.status = permissionError.status;
+                  ctx.body = {
+                    error: permissionError.error,
+                    message: permissionError.message
+                  };
+                  return;
+                }
+              }
+            }
+          }
+          const fgaError = await serverAdapter.checkRouteFGA(server.mastra, serverRoute, ctx.state.requestContext, {
+            ...matchedRoute?.params ?? {},
+            ...ctx.query,
+            ...typeof ctx.request.body === "object" && ctx.request.body !== null ? ctx.request.body : {}
+          });
+          if (fgaError) {
+            ctx.status = fgaError.status;
+            ctx.body = { error: fgaError.error, message: fgaError.message };
+            return;
           }
         }
-        const fgaError = await serverAdapter.checkRouteFGA(server.mastra, serverRoute, ctx.state.requestContext, {
-          ...matchedRoute?.params ?? {},
-          ...ctx.query,
-          ...typeof ctx.request.body === "object" && ctx.request.body !== null ? ctx.request.body : {}
-        });
-        if (fgaError) {
-          ctx.status = fgaError.status;
-          ctx.body = { error: fgaError.error, message: fgaError.message };
-          return;
-        }
+        const response = await server.handleCustomRouteRequest(
+          `${ctx.protocol}://${ctx.host}${ctx.originalUrl || ctx.url}`,
+          ctx.method,
+          ctx.headers,
+          ctx.request.body,
+          ctx.state.requestContext,
+          customRouteAbortController.signal
+        );
+        if (!response) return next();
+        ctx.respond = false;
+        await server.writeCustomRouteResponse(response, ctx.res, customRouteAbortController.signal);
+      } finally {
+        ctx.res.off("close", abortCustomRouteIfOpen);
+        ctx.res.off("error", abortCustomRouteIfOpen);
       }
-      const response = await server.handleCustomRouteRequest(
-        `${ctx.protocol}://${ctx.host}${ctx.originalUrl || ctx.url}`,
-        ctx.method,
-        ctx.headers,
-        ctx.request.body,
-        ctx.state.requestContext
-      );
-      if (!response) return next();
-      ctx.respond = false;
-      await server.writeCustomRouteResponse(response, ctx.res);
     });
   }
   registerContextMiddleware() {
@@ -1006,14 +987,14 @@ var MastraServer = class extends serverAdapter.MastraServer {
       const method = ctx.method;
       const path = ctx.path;
       await next();
-      const duration2 = Date.now() - start;
+      const duration = Date.now() - start;
       const status = ctx.status;
       const level = server.httpLoggingConfig?.level || "info";
       const logData = {
         method,
         path,
         status,
-        duration: `${duration2}ms`
+        duration: `${duration}ms`
       };
       if (server.httpLoggingConfig?.includeQueryParams) {
         logData.query = ctx.query;
@@ -1029,7 +1010,7 @@ var MastraServer = class extends serverAdapter.MastraServer {
         });
         logData.headers = headers;
       }
-      server.logger[level](`${method} ${path} ${status} ${duration2}ms`, logData);
+      server.logger[level](`${method} ${path} ${status} ${duration}ms`, logData);
     });
   }
 };