@mastra/server 1.58.0-alpha.0 → 1.58.0-alpha.12

package/dist/server/handlers/agent-controller.cjs

@@ -1,6 +1,6 @@
 Object.defineProperty(exports, Symbol.toStringTag, { value: "Module" });
 const require_http_exception = require("../../http-exception-CKDHupKw.cjs");
-const require_route_builder = require("../../route-builder-CgLVYL1c.cjs");
+const require_route_builder = require("../../route-builder-BBovb2TE.cjs");
 const require_server_handlers_error = require("./error.cjs");
 let zod_v4 = require("zod/v4");
 //#region src/server/handlers/agent-controller.ts
@@ -50,6 +50,38 @@ async function getSession(controller, resourceId, options, requestContext) {
 		requestContext
 	});
 }
+/**
+* Acknowledges a session operation that keeps running after the response is
+* sent.
+*
+* The messages/steer/follow-up routes answer `{ ok: true }` as soon as the
+* message is handed to the session: the reply itself arrives on the session's
+* SSE stream, so the request must not block for the whole turn. Those session
+* methods can still reject — `sendMessage` deliberately rejects when signal
+* submission fails before a stream starts — and a bare `void promise` turns
+* that into an unhandled rejection, which terminates the process on Node's
+* default `--unhandled-rejections=throw`.
+*
+* Observing the promise keeps the immediate acknowledgement while logging the
+* failure and emitting an `error` event on the session, so a client streaming
+* the session is told the turn died instead of waiting for an `agent_end` that
+* will never come.
+*/
+function ackBackgroundSessionWork({ work, session, mastra, operation }) {
+	work.catch((error) => {
+		const failure = error instanceof Error ? error : new Error(String(error));
+		mastra.getLogger?.()?.error?.(`AgentController ${operation} failed after the request was acknowledged`, {
+			operation,
+			error: failure
+		});
+		try {
+			session.emit({
+				type: "error",
+				error: failure
+			});
+		} catch {}
+	});
+}
 const controllerIdPathParams = zod_v4.z.object({ controllerId: zod_v4.z.string() });
 const sessionPathParams = zod_v4.z.object({
 	controllerId: zod_v4.z.string(),
@@ -189,13 +221,15 @@ const omProgressSummarySchema = zod_v4.z.object({
 });
 const sessionSettingsSchema = zod_v4.z.object({
 	yolo: zod_v4.z.boolean(),
+	/** Session override only — absent when the session inherits a configured default. */
 	thinkingLevel: zod_v4.z.enum([
 		"off",
 		"low",
 		"medium",
 		"high",
-		"xhigh"
-	]),
+		"xhigh",
+		"max"
+	]).optional(),
 	notifications: zod_v4.z.enum([
 		"off",
 		"bell",
@@ -345,9 +379,14 @@ const CREATE_AGENT_CONTROLLER_SESSION_ROUTE = require_route_builder.createRoute(
 * non-enumerable, so JSON serialization in the SSE adapter would send
 * `"error": {}` and clients could only render a generic "Error". Flatten the
 * Error into a plain object so the actual failure reaches the client.
+*
+* `display_state_changed` Maps JSON-serialize to `{}`; convert them to plain
+* records so wire clients get the tool state the in-process TUI sees.
 */
 function toWireEvent(event) {
-	if (typeof event === "object" && event !== null && event.type === "error" && event.error instanceof Error) {
+	if (typeof event !== "object" || event === null) return event;
+	const { type } = event;
+	if (type === "error" && event.error instanceof Error) {
 		const error = event.error;
 		return {
 			...event,
@@ -357,6 +396,16 @@ function toWireEvent(event) {
 			}
 		};
 	}
+	if (type === "display_state_changed") {
+		const { displayState } = event;
+		if (typeof displayState !== "object" || displayState === null) return event;
+		const wireDisplayState = { ...displayState };
+		for (const [key, value] of Object.entries(wireDisplayState)) if (value instanceof Map) wireDisplayState[key] = Object.fromEntries(value);
+		return {
+			...event,
+			displayState: wireDisplayState
+		};
+	}
 	return event;
 }
 const STREAM_AGENT_CONTROLLER_SESSION_ROUTE = require_route_builder.createRoute({
@@ -447,10 +496,16 @@ const SEND_AGENT_CONTROLLER_MESSAGE_ROUTE = require_route_builder.createRoute({
 	requiresPermission: "agent-controller:execute",
 	handler: async ({ mastra, controllerId, resourceId, sessionScope, message, files, requestContext }) => {
 		try {
-			(await getSession(getAgentControllerOrThrow(mastra, controllerId), resourceId, { scope: sessionScope }, requestContext)).sendMessage({
-				content: message,
-				files,
-				requestContext
+			const session = await getSession(getAgentControllerOrThrow(mastra, controllerId), resourceId, { scope: sessionScope }, requestContext);
+			ackBackgroundSessionWork({
+				work: session.sendMessage({
+					content: message,
+					files,
+					requestContext
+				}),
+				session,
+				mastra,
+				operation: "sendMessage"
 			});
 			return { ok: true };
 		} catch (error) {
@@ -546,9 +601,15 @@ const STEER_AGENT_CONTROLLER_SESSION_ROUTE = require_route_builder.createRoute({
 	requiresPermission: "agent-controller:execute",
 	handler: async ({ mastra, controllerId, resourceId, sessionScope, message, requestContext }) => {
 		try {
-			(await getSession(getAgentControllerOrThrow(mastra, controllerId), resourceId, { scope: sessionScope }, requestContext)).steer({
-				content: message,
-				requestContext
+			const session = await getSession(getAgentControllerOrThrow(mastra, controllerId), resourceId, { scope: sessionScope }, requestContext);
+			ackBackgroundSessionWork({
+				work: session.steer({
+					content: message,
+					requestContext
+				}),
+				session,
+				mastra,
+				operation: "steer"
 			});
 			return { ok: true };
 		} catch (error) {
@@ -647,6 +708,7 @@ const GET_AGENT_CONTROLLER_SESSION_STATE_ROUTE = require_route_builder.createRou
 			const reflectionSavings = om.buffered.reflection.inputObservationTokens - om.buffered.reflection.observationTokens;
 			const st = session.state.get();
 			const oneOf = (value, allowed, fallback) => allowed.includes(value) ? value : fallback;
+			const oneOfOptional = (value, allowed) => allowed.includes(value) ? value : void 0;
 			return {
 				controllerId,
 				resourceId,
@@ -668,13 +730,14 @@ const GET_AGENT_CONTROLLER_SESSION_STATE_ROUTE = require_route_builder.createRou
 				tokenUsage: ds.tokenUsage,
 				settings: {
 					yolo: st.yolo === true,
-					thinkingLevel: oneOf(st.thinkingLevel, [
+					thinkingLevel: oneOfOptional(st.thinkingLevel, [
 						"off",
 						"low",
 						"medium",
 						"high",
-						"xhigh"
-					], "off"),
+						"xhigh",
+						"max"
+					]),
 					notifications: oneOf(st.notifications, [
 						"off",
 						"bell",
@@ -955,9 +1018,15 @@ const FOLLOW_UP_AGENT_CONTROLLER_SESSION_ROUTE = require_route_builder.createRou
 	requiresPermission: "agent-controller:execute",
 	handler: async ({ mastra, controllerId, resourceId, sessionScope, message, requestContext }) => {
 		try {
-			(await getSession(getAgentControllerOrThrow(mastra, controllerId), resourceId, { scope: sessionScope }, requestContext)).followUp({
-				content: message,
-				requestContext
+			const session = await getSession(getAgentControllerOrThrow(mastra, controllerId), resourceId, { scope: sessionScope }, requestContext);
+			ackBackgroundSessionWork({
+				work: session.followUp({
+					content: message,
+					requestContext
+				}),
+				session,
+				mastra,
+				operation: "followUp"
 			});
 			return { ok: true };
 		} catch (error) {