@mastra/server 1.70.0-alpha.0 → 1.70.0-alpha.1

package/dist/server/handlers/agents.cjs

@@ -1,7 +1,7 @@
 Object.defineProperty(exports, Symbol.toStringTag, { value: "Module" });
 const require_rolldown_runtime = require("../../rolldown-runtime-Bd6kNlEP.cjs");
 const require_http_exception = require("../../http-exception-CKDHupKw.cjs");
-const require_agents = require("../../agents-CB7mDJp7.cjs");
+const require_agents = require("../../agents-DBtumkdm.cjs");
 const require_schema_e0VHpMt0 = require("../../schema-e0VHpMt0-BS9KHnuT.cjs");
 const require_route_builder = require("../../route-builder-2yrD15TL.cjs");
 require("../../schema-BVcoNfni.cjs");
@@ -25,6 +25,7 @@ var agents_exports = /* @__PURE__ */ require_rolldown_runtime.__exportAll({
 	APPROVE_NETWORK_TOOL_CALL_ROUTE: () => APPROVE_NETWORK_TOOL_CALL_ROUTE,
 	APPROVE_TOOL_CALL_GENERATE_ROUTE: () => APPROVE_TOOL_CALL_GENERATE_ROUTE,
 	APPROVE_TOOL_CALL_ROUTE: () => APPROVE_TOOL_CALL_ROUTE,
+	CANCEL_AGENT_PENDING_SIGNALS_ROUTE: () => CANCEL_AGENT_PENDING_SIGNALS_ROUTE,
 	CLONE_AGENT_ROUTE: () => CLONE_AGENT_ROUTE,
 	DECLINE_NETWORK_TOOL_CALL_ROUTE: () => DECLINE_NETWORK_TOOL_CALL_ROUTE,
 	DECLINE_TOOL_CALL_GENERATE_ROUTE: () => DECLINE_TOOL_CALL_GENERATE_ROUTE,
@@ -1450,7 +1451,7 @@ const ABORT_AGENT_THREAD_ROUTE = require_route_builder.createRoute({
 	tags: ["Agents", "Streaming"],
 	requiresAuth: true,
 	requiresPermission: "agents:execute",
-	handler: async ({ mastra, agentId, resourceId, threadId, expectedRunId, requestContext: serverRequestContext }) => {
+	handler: async ({ mastra, agentId, resourceId, threadId, clearPendingSignals, expectedRunId, requestContext: serverRequestContext }) => {
 		try {
 			const agent = await getAgentFromSystem({
 				mastra,
@@ -1458,16 +1459,22 @@ const ABORT_AGENT_THREAD_ROUTE = require_route_builder.createRoute({
 				requestContext: serverRequestContext
 			});
 			if (typeof agent.abortThreadStream !== "function") throw new require_http_exception.HTTPException(501, { message: "agent thread aborts are not supported by this Mastra core version" });
+			if (clearPendingSignals && agent.__supportsThreadSignalCancellation !== true) throw new require_http_exception.HTTPException(501, { message: "clear-on-abort requires a newer @mastra/core version. Upgrade @mastra/core alongside @mastra/server." });
 			const effectiveResourceId = require_utils$1.getEffectiveResourceId(serverRequestContext, resourceId);
 			const effectiveThreadId = require_utils$1.getEffectiveThreadId(serverRequestContext, threadId);
 			if (!effectiveThreadId) throw new require_http_exception.HTTPException(400, { message: "threadId is required" });
-			if (effectiveResourceId) {
-				const memory = await agent.getMemory({ requestContext: serverRequestContext });
-				if (memory) await require_utils$1.validateThreadOwnership(await memory.getThreadById({ threadId: effectiveThreadId }), effectiveResourceId);
-			}
+			await require_utils$1.enforceThreadAccess({
+				mastra,
+				requestContext: serverRequestContext,
+				threadId: effectiveThreadId,
+				thread: await (await agent.getMemory({ requestContext: serverRequestContext }))?.getThreadById({ threadId: effectiveThreadId }),
+				effectiveResourceId,
+				permission: require_utils$1.MastraFGAPermissions.MEMORY_WRITE
+			});
 			return { aborted: await agent.abortThreadStream({
 				resourceId: effectiveResourceId,
 				threadId: effectiveThreadId,
+				...clearPendingSignals === void 0 ? {} : { clearPendingSignals },
 				expectedRunId
 			}) };
 		} catch (error) {
@@ -1475,6 +1482,47 @@ const ABORT_AGENT_THREAD_ROUTE = require_route_builder.createRoute({
 		}
 	}
 });
+const CANCEL_AGENT_PENDING_SIGNALS_ROUTE = require_route_builder.createRoute({
+	method: "POST",
+	path: "/agents/:agentId/threads/signals/cancel",
+	responseType: "json",
+	pathParamSchema: require_agents.agentIdPathParams,
+	bodySchema: require_agents.cancelPendingAgentSignalsBodySchema,
+	responseSchema: require_agents.cancelPendingAgentSignalsResponseSchema,
+	summary: "Cancel pending thread signals",
+	description: "Cancels selected pending thread signals and propagates requested IDs through PubSub",
+	tags: ["Agents", "Streaming"],
+	requiresAuth: true,
+	requiresPermission: "agents:execute",
+	handler: async ({ mastra, agentId, resourceId, threadId, signalIds, requestContext }) => {
+		try {
+			const agent = await getAgentFromSystem({
+				mastra,
+				agentId,
+				requestContext
+			});
+			if (typeof agent.cancelQueuedMessages !== "function" || agent.__supportsThreadSignalCancellation !== true) throw new require_http_exception.HTTPException(501, { message: "thread-wide cancellation requires a newer @mastra/core version. Upgrade @mastra/core alongside @mastra/server." });
+			const effectiveResourceId = require_utils$1.getEffectiveResourceId(requestContext, resourceId);
+			const effectiveThreadId = require_utils$1.getEffectiveThreadId(requestContext, threadId);
+			if (!effectiveThreadId) throw new require_http_exception.HTTPException(400, { message: "threadId is required" });
+			await require_utils$1.enforceThreadAccess({
+				mastra,
+				requestContext,
+				threadId: effectiveThreadId,
+				thread: await (await agent.getMemory({ requestContext }))?.getThreadById({ threadId: effectiveThreadId }),
+				effectiveResourceId,
+				permission: require_utils$1.MastraFGAPermissions.MEMORY_WRITE
+			});
+			return agent.cancelQueuedMessages({
+				resourceId: effectiveResourceId,
+				threadId: effectiveThreadId,
+				signalIds
+			});
+		} catch (error) {
+			return require_server_handlers_error.handleError(error, "error cancelling pending thread signals");
+		}
+	}
+});
 const SUBSCRIBE_AGENT_THREAD_ROUTE = require_route_builder.createRoute({
 	method: "POST",
 	path: "/agents/:agentId/threads/subscribe",
@@ -2554,6 +2602,7 @@ exports.ABORT_AGENT_THREAD_ROUTE = ABORT_AGENT_THREAD_ROUTE;
 exports.APPROVE_NETWORK_TOOL_CALL_ROUTE = APPROVE_NETWORK_TOOL_CALL_ROUTE;
 exports.APPROVE_TOOL_CALL_GENERATE_ROUTE = APPROVE_TOOL_CALL_GENERATE_ROUTE;
 exports.APPROVE_TOOL_CALL_ROUTE = APPROVE_TOOL_CALL_ROUTE;
+exports.CANCEL_AGENT_PENDING_SIGNALS_ROUTE = CANCEL_AGENT_PENDING_SIGNALS_ROUTE;
 exports.CLONE_AGENT_ROUTE = CLONE_AGENT_ROUTE;
 exports.DECLINE_NETWORK_TOOL_CALL_ROUTE = DECLINE_NETWORK_TOOL_CALL_ROUTE;
 exports.DECLINE_TOOL_CALL_GENERATE_ROUTE = DECLINE_TOOL_CALL_GENERATE_ROUTE;