@mastra/server 1.58.0-alpha.1 → 1.58.0-alpha.13

package/dist/server/handlers/workflows.cjs

@@ -1,11 +1,11 @@
 Object.defineProperty(exports, Symbol.toStringTag, { value: "Module" });
 const require_rolldown_runtime = require("../../rolldown-runtime-CVvi-lCc.cjs");
 const require_http_exception = require("../../http-exception-CKDHupKw.cjs");
-const require_common = require("../../common-spbwhiWS.cjs");
-const require_workflows = require("../../workflows-XfiENvkF.cjs");
-const require_agents = require("../../agents-Dk2qoJi_.cjs");
-const require_route_builder = require("../../route-builder-CgLVYL1c.cjs");
-const require_utils = require("../../utils-wESuBh27.cjs");
+const require_common = require("../../common-DQn8jX5T.cjs");
+const require_workflows = require("../../workflows--79xhOOb.cjs");
+const require_agents = require("../../agents-0Tw3kkD_.cjs");
+const require_route_builder = require("../../route-builder-BAAedfru.cjs");
+const require_utils = require("../../utils-Bna_JCkb.cjs");
 const require_server_handlers_error = require("./error.cjs");
 const require_utils$1 = require("../../utils-wTwyTjg3.cjs");
 let zod_v4 = require("zod/v4");
@@ -20,6 +20,7 @@ var workflows_exports = /* @__PURE__ */ require_rolldown_runtime.__exportAll({
 	GET_WORKFLOW_RUN_BY_ID_ROUTE: () => GET_WORKFLOW_RUN_BY_ID_ROUTE,
 	LIST_WORKFLOWS_ROUTE: () => LIST_WORKFLOWS_ROUTE,
 	LIST_WORKFLOW_RUNS_ROUTE: () => LIST_WORKFLOW_RUNS_ROUTE,
+	LIST_WORKFLOW_RUN_COUNTS_ROUTE: () => LIST_WORKFLOW_RUN_COUNTS_ROUTE,
 	OBSERVE_STREAM_LEGACY_WORKFLOW_ROUTE: () => OBSERVE_STREAM_LEGACY_WORKFLOW_ROUTE,
 	OBSERVE_STREAM_WORKFLOW_ROUTE: () => OBSERVE_STREAM_WORKFLOW_ROUTE,
 	RECEIVE_WORKFLOW_EVENT_ROUTE: () => RECEIVE_WORKFLOW_EVENT_ROUTE,
@@ -37,7 +38,8 @@ var workflows_exports = /* @__PURE__ */ require_rolldown_runtime.__exportAll({
 	STREAM_WORKFLOW_ROUTE: () => STREAM_WORKFLOW_ROUTE,
 	TIME_TRAVEL_ASYNC_WORKFLOW_ROUTE: () => TIME_TRAVEL_ASYNC_WORKFLOW_ROUTE,
 	TIME_TRAVEL_STREAM_WORKFLOW_ROUTE: () => TIME_TRAVEL_STREAM_WORKFLOW_ROUTE,
-	TIME_TRAVEL_WORKFLOW_ROUTE: () => TIME_TRAVEL_WORKFLOW_ROUTE
+	TIME_TRAVEL_WORKFLOW_ROUTE: () => TIME_TRAVEL_WORKFLOW_ROUTE,
+	__setWorkflowRunCountsNow: () => __setWorkflowRunCountsNow
 });
 /**
 * Statuses a run cannot come back from. Streaming one of these runIds again would
@@ -51,6 +53,43 @@ const TERMINAL_RUN_STATUSES = [
 	"canceled",
 	"tripwire"
 ];
+/**
+* Runs whose chunks are already being written to the cache in this process,
+* keyed by runId. A run has exactly one writer: without this, two concurrent
+* requests for the same runId each cache every chunk and the replayed history
+* comes back duplicated.
+*/
+const activeRunStreamCachers = /* @__PURE__ */ new Set();
+/**
+* Cache a run's chunks for later replay and return the stream to hand the client.
+*
+* The cached history belongs to the run, not to whoever happens to be watching,
+* so the caching side is driven by its own reader rather than by the client's
+* consumption. If the client disconnects mid-run its branch is cancelled while
+* this reader keeps draining, so `/observe` still replays a complete history —
+* the failure that makes reconnection necessary is exactly the one the cache has
+* to survive. Chunks buffer in the tee while a client lags, bounded by the run's
+* own length.
+*/
+function cacheRunStream({ cache, runId, source }) {
+	if (activeRunStreamCachers.has(runId)) return source;
+	const [toCache, toClient] = source.tee();
+	activeRunStreamCachers.add(runId);
+	(async () => {
+		const reader = toCache.getReader();
+		try {
+			while (true) {
+				const { done, value } = await reader.read();
+				if (done) break;
+				await cache.listPush(runId, value).catch(() => {});
+			}
+		} catch {} finally {
+			reader.releaseLock();
+			activeRunStreamCachers.delete(runId);
+		}
+	})();
+	return toClient;
+}
 async function listWorkflowsFromSystem({ mastra, workflowId }) {
 	const logger = mastra.getLogger();
 	if (!workflowId) throw new require_http_exception.HTTPException(400, { message: "Workflow ID is required" });
@@ -113,6 +152,81 @@ const LIST_WORKFLOWS_ROUTE = require_route_builder.createRoute({
 		}
 	})
 });
+const RUN_COUNTS_CACHE_TTL_MS = 5e3;
+const runCountsCache = /* @__PURE__ */ new WeakMap();
+let runCountsNow = () => Date.now();
+/** Test seam — lets TTL expiry be tested without wall-clock waits. */
+function __setWorkflowRunCountsNow(now) {
+	runCountsNow = now ?? (() => Date.now());
+}
+const LIST_WORKFLOW_RUN_COUNTS_ROUTE = require_route_builder.createRoute({
+	method: "GET",
+	path: "/workflows/run-counts",
+	responseType: "json",
+	responseSchema: require_workflows.workflowRunCountsResponseSchema,
+	summary: "List workflow run counts",
+	description: "Returns per-workflow counts of currently running and suspended (awaiting resume) runs, keyed by the workflow registry key used in the Mastra config",
+	tags: ["Workflows"],
+	requiresAuth: true,
+	handler: (async ({ mastra, requestContext }) => {
+		try {
+			const fgaProvider = mastra.getServer?.()?.fga;
+			const user = requestContext?.get("user");
+			if (fgaProvider && !user) return {};
+			const workflows = mastra.listWorkflows({ serialized: false });
+			const counts = {};
+			const workflowIdToRegistryKey = /* @__PURE__ */ new Map();
+			for (const [registryKey, workflow] of Object.entries(workflows)) {
+				counts[registryKey] = {
+					running: 0,
+					suspended: 0
+				};
+				workflowIdToRegistryKey.set(workflow.id, registryKey);
+			}
+			const effectiveResourceId = require_utils$1.getEffectiveResourceId(requestContext, void 0);
+			const cacheable = !fgaProvider && !effectiveResourceId;
+			if (cacheable) {
+				const cached = runCountsCache.get(mastra);
+				if (cached && runCountsNow() - cached.at < RUN_COUNTS_CACHE_TTL_MS) return cached.value;
+			}
+			const storage = mastra.getStorage();
+			const workflowsStore = storage ? await storage.getStore("workflows") : void 0;
+			if (workflowsStore) {
+				const [running, suspended] = await Promise.all([workflowsStore.listWorkflowRuns({
+					status: "running",
+					resourceId: effectiveResourceId
+				}), workflowsStore.listWorkflowRuns({
+					status: "suspended",
+					resourceId: effectiveResourceId
+				})]);
+				for (const run of running.runs) {
+					const registryKey = workflowIdToRegistryKey.get(run.workflowName);
+					const entry = registryKey ? counts[registryKey] : void 0;
+					if (entry) entry.running++;
+				}
+				for (const run of suspended.runs) {
+					const registryKey = workflowIdToRegistryKey.get(run.workflowName);
+					const entry = registryKey ? counts[registryKey] : void 0;
+					if (entry) entry.suspended++;
+				}
+			}
+			if (fgaProvider) {
+				const workflowList = Object.keys(counts).map((id) => ({ id }));
+				const accessible = await fgaProvider.filterAccessible(user, workflowList, "workflow", require_utils$1.MastraFGAPermissions.WORKFLOWS_READ);
+				const accessibleSet = new Set(accessible.map((w) => w.id));
+				for (const id of Object.keys(counts)) if (!accessibleSet.has(id)) delete counts[id];
+				return counts;
+			}
+			if (cacheable) runCountsCache.set(mastra, {
+				at: runCountsNow(),
+				value: counts
+			});
+			return counts;
+		} catch (error) {
+			return require_server_handlers_error.handleError(error, "Error getting workflow run counts");
+		}
+	})
+});
 const GET_WORKFLOW_BY_ID_ROUTE = require_route_builder.createRoute({
 	method: "GET",
 	path: "/workflows/:workflowId",
@@ -305,13 +419,11 @@ const STREAM_WORKFLOW_ROUTE = require_route_builder.createRoute({
 				...params,
 				requestContext
 			});
-			if (serverCache) {
-				const { transform } = (0, _mastra_core_stream.createCachingTransformStream)({
-					cache: serverCache,
-					cacheKey: runId
-				});
-				return result.fullStream.pipeThrough(transform);
-			}
+			if (serverCache) return cacheRunStream({
+				cache: serverCache,
+				runId,
+				source: result.fullStream
+			});
 			return result.fullStream;
 		} catch (error) {
 			return require_server_handlers_error.handleError(error, "Error streaming workflow");
@@ -352,13 +464,11 @@ const RESUME_STREAM_WORKFLOW_ROUTE = require_route_builder.createRoute({
 				...params,
 				requestContext
 			});
-			if (serverCache) {
-				const { transform } = (0, _mastra_core_stream.createCachingTransformStream)({
-					cache: serverCache,
-					cacheKey: runId
-				});
-				return resumeResult.fullStream.pipeThrough(transform);
-			}
+			if (serverCache) return cacheRunStream({
+				cache: serverCache,
+				runId,
+				source: resumeResult.fullStream
+			});
 			return resumeResult.fullStream;
 		} catch (error) {
 			return require_server_handlers_error.handleError(error, "Error resuming workflow");
@@ -597,6 +707,12 @@ const RESUME_WORKFLOW_ROUTE = require_route_builder.createRoute({
 			})).resume({
 				...params,
 				requestContext
+			}).catch((error) => {
+				mastra.getLogger().error("Failed to resume workflow run", {
+					error,
+					workflowId,
+					runId
+				});
 			});
 			return { message: "Workflow run resumed" };
 		} catch (error) {
@@ -808,6 +924,12 @@ const TIME_TRAVEL_WORKFLOW_ROUTE = require_route_builder.createRoute({
 			})).timeTravel({
 				...params,
 				requestContext
+			}).catch((error) => {
+				mastra.getLogger().error("Failed to time travel workflow run", {
+					error,
+					workflowId,
+					runId
+				});
 			});
 			return { message: "Workflow run time travel started" };
 		} catch (error) {
@@ -847,13 +969,11 @@ const TIME_TRAVEL_STREAM_WORKFLOW_ROUTE = require_route_builder.createRoute({
 				...params,
 				requestContext
 			});
-			if (serverCache) {
-				const { transform } = (0, _mastra_core_stream.createCachingTransformStream)({
-					cache: serverCache,
-					cacheKey: runId
-				});
-				return result.fullStream.pipeThrough(transform);
-			}
+			if (serverCache) return cacheRunStream({
+				cache: serverCache,
+				runId,
+				source: result.fullStream
+			});
 			return result.fullStream;
 		} catch (error) {
 			return require_server_handlers_error.handleError(error, "Error time traveling workflow stream");
@@ -1099,6 +1219,7 @@ exports.GET_WORKFLOW_BY_ID_ROUTE = GET_WORKFLOW_BY_ID_ROUTE;
 exports.GET_WORKFLOW_RUN_BY_ID_ROUTE = GET_WORKFLOW_RUN_BY_ID_ROUTE;
 exports.LIST_WORKFLOWS_ROUTE = LIST_WORKFLOWS_ROUTE;
 exports.LIST_WORKFLOW_RUNS_ROUTE = LIST_WORKFLOW_RUNS_ROUTE;
+exports.LIST_WORKFLOW_RUN_COUNTS_ROUTE = LIST_WORKFLOW_RUN_COUNTS_ROUTE;
 exports.OBSERVE_STREAM_LEGACY_WORKFLOW_ROUTE = OBSERVE_STREAM_LEGACY_WORKFLOW_ROUTE;
 exports.OBSERVE_STREAM_WORKFLOW_ROUTE = OBSERVE_STREAM_WORKFLOW_ROUTE;
 exports.RECEIVE_WORKFLOW_EVENT_ROUTE = RECEIVE_WORKFLOW_EVENT_ROUTE;
@@ -1117,6 +1238,7 @@ exports.STREAM_WORKFLOW_ROUTE = STREAM_WORKFLOW_ROUTE;
 exports.TIME_TRAVEL_ASYNC_WORKFLOW_ROUTE = TIME_TRAVEL_ASYNC_WORKFLOW_ROUTE;
 exports.TIME_TRAVEL_STREAM_WORKFLOW_ROUTE = TIME_TRAVEL_STREAM_WORKFLOW_ROUTE;
 exports.TIME_TRAVEL_WORKFLOW_ROUTE = TIME_TRAVEL_WORKFLOW_ROUTE;
+exports.__setWorkflowRunCountsNow = __setWorkflowRunCountsNow;
 Object.defineProperty(exports, "workflows_exports", {
 	enumerable: true,
 	get: function() {