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

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

@@ -1,5 +1,5 @@
 import { t as HTTPException } from "../../http-exception-CyfxZB6W.js";
-import { n as createRoute } from "../../route-builder-CIk5k35M.js";
+import { n as createRoute } from "../../route-builder-CQkb7tQY.js";
 import { handleError } from "./error.js";
 import { z } from "zod/v4";
 //#region src/server/handlers/agent-controller.ts
@@ -49,6 +49,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 = z.object({ controllerId: z.string() });
 const sessionPathParams = z.object({
 	controllerId: z.string(),
@@ -188,13 +220,15 @@ const omProgressSummarySchema = z.object({
 });
 const sessionSettingsSchema = z.object({
 	yolo: z.boolean(),
+	/** Session override only — absent when the session inherits a configured default. */
 	thinkingLevel: z.enum([
 		"off",
 		"low",
 		"medium",
 		"high",
-		"xhigh"
-	]),
+		"xhigh",
+		"max"
+	]).optional(),
 	notifications: z.enum([
 		"off",
 		"bell",
@@ -344,9 +378,14 @@ const CREATE_AGENT_CONTROLLER_SESSION_ROUTE = 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,
@@ -356,6 +395,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 = createRoute({
@@ -446,10 +495,16 @@ const SEND_AGENT_CONTROLLER_MESSAGE_ROUTE = 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) {
@@ -545,9 +600,15 @@ const STEER_AGENT_CONTROLLER_SESSION_ROUTE = 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) {
@@ -646,6 +707,7 @@ const GET_AGENT_CONTROLLER_SESSION_STATE_ROUTE = createRoute({
 			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,
@@ -667,13 +729,14 @@ const GET_AGENT_CONTROLLER_SESSION_STATE_ROUTE = createRoute({
 				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",
@@ -954,9 +1017,15 @@ const FOLLOW_UP_AGENT_CONTROLLER_SESSION_ROUTE = createRoute({
 	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) {