@mastra/server 1.76.0-alpha.2 → 1.76.0-alpha.4

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

@@ -1,7 +1,7 @@
 import { t as HTTPException } from "../../http-exception-CyfxZB6W.js";
 import { n as createRoute } from "../../route-builder-lBil8Cvo.js";
 import { handleError } from "./error.js";
-import { t as enforceThreadAccess } from "../../utils-KcHk7C0D.js";
+import { t as enforceThreadAccess } from "../../utils-kaP2cyIO.js";
 import { E as includeSchema, F as messageOrderBySchema, h as filterSchema } from "../../memory-B58Q2LgU.js";
 import { z } from "zod/v4";
 //#region src/server/handlers/agent-controller.ts
@@ -13,8 +13,9 @@ import { z } from "zod/v4";
 * non-terminal clients — e.g. a browser-based MastraCode — can create sessions,
 * send messages, stream events, and drive run-control. Each route resolves its
 * target AgentController by id, then operates on a session bound to a
-* `resourceId` (get-or-create, so reconnects resume rather than fork the
-* conversation).
+* `resourceId`. Mutating routes use get-or-create resolution so reconnects
+* resume rather than fork the conversation; read routes only inspect existing
+* live sessions.
 */
 /**
 * Internal thread-metadata keys that `Session.loadMetadata()` reads back as
@@ -31,6 +32,8 @@ const RESERVED_THREAD_METADATA_KEYS = {
 	currentModeId: true,
 	observerModelId: true,
 	reflectorModelId: true,
+	observerModelSelection: true,
+	reflectorModelSelection: true,
 	observationThreshold: true,
 	reflectionThreshold: true,
 	tokenUsage: true,
@@ -63,6 +66,12 @@ async function getSession(controller, resourceId, options, requestContext) {
 		requestContext
 	});
 }
+async function getExistingSession(controller, resourceId, scope, requestContext) {
+	await controller.init();
+	const session = await controller.getSessionByResource(resourceId, scope, requestContext);
+	if (session) return session;
+	return getSession(controller, resourceId, { scope }, requestContext);
+}
 /**
 * Acknowledges a session operation that keeps running after the response is
 * sent.
@@ -503,7 +512,7 @@ const STREAM_AGENT_CONTROLLER_SESSION_ROUTE = createRoute({
 	requiresPermission: "agent-controller:read",
 	handler: async ({ mastra, controllerId, resourceId, sessionScope, abortSignal, requestContext }) => {
 		try {
-			const session = await getSession(getAgentControllerOrThrow(mastra, controllerId), resourceId, { scope: sessionScope }, requestContext);
+			const session = await getExistingSession(getAgentControllerOrThrow(mastra, controllerId), resourceId, sessionScope, requestContext);
 			let cleanedUp = false;
 			let heartbeat;
 			let unsubscribe;
@@ -819,7 +828,7 @@ const GET_AGENT_CONTROLLER_SESSION_STATE_ROUTE = createRoute({
 	requiresPermission: "agent-controller:read",
 	handler: async ({ mastra, controllerId, resourceId, sessionScope, threadId: requestedThreadId, requestContext }) => {
 		try {
-			const session = await getSession(getAgentControllerOrThrow(mastra, controllerId), resourceId, { scope: sessionScope }, requestContext);
+			const session = await getExistingSession(getAgentControllerOrThrow(mastra, controllerId), resourceId, sessionScope, requestContext);
 			const ds = session.displayState.get();
 			const threadId = requestedThreadId ?? session.thread.getId() ?? void 0;
 			const storage = mastra.getStorage();
@@ -1293,7 +1302,7 @@ const GET_AGENT_CONTROLLER_OM_RECORD_ROUTE = createRoute({
 	handler: async ({ mastra, controllerId, resourceId, sessionScope, requestContext }) => {
 		try {
 			const controller = getAgentControllerOrThrow(mastra, controllerId);
-			const session = await getSession(controller, resourceId, { scope: sessionScope }, requestContext);
+			const session = await getExistingSession(controller, resourceId, sessionScope, requestContext);
 			return { record: await controller.getObservationalMemoryRecord(session) ?? void 0 };
 		} catch (error) {
 			return handleError(error, "error reading controller OM record");
@@ -1339,7 +1348,7 @@ const GET_AGENT_CONTROLLER_RESOURCE_IDS_ROUTE = createRoute({
 	handler: async ({ mastra, controllerId, resourceId, sessionScope, requestContext }) => {
 		try {
 			const controller = getAgentControllerOrThrow(mastra, controllerId);
-			const session = await getSession(controller, resourceId, { scope: sessionScope }, requestContext);
+			const session = await getExistingSession(controller, resourceId, sessionScope, requestContext);
 			return { resourceIds: await controller.getKnownResourceIds(session) };
 		} catch (error) {
 			return handleError(error, "error listing controller resource IDs");
@@ -1394,7 +1403,7 @@ const GET_AGENT_CONTROLLER_GOAL_ROUTE = createRoute({
 	handler: async ({ mastra, controllerId, resourceId, sessionScope, requestContext }) => {
 		try {
 			const controller = getAgentControllerOrThrow(mastra, controllerId);
-			const session = await getSession(controller, resourceId, { scope: sessionScope }, requestContext);
+			const session = await getExistingSession(controller, resourceId, sessionScope, requestContext);
 			const threadId = session.thread.getId();
 			if (!threadId) return { goal: void 0 };
 			return { goal: await getAgentForSession(controller, session).getObjective({ threadId }) ?? void 0 };
@@ -1502,7 +1511,7 @@ const GET_AGENT_CONTROLLER_PERMISSIONS_ROUTE = createRoute({
 	requiresPermission: "agent-controller:read",
 	handler: async ({ mastra, controllerId, resourceId, sessionScope, requestContext }) => {
 		try {
-			const rules = (await getSession(getAgentControllerOrThrow(mastra, controllerId), resourceId, { scope: sessionScope }, requestContext)).permissions.getRules();
+			const rules = (await getExistingSession(getAgentControllerOrThrow(mastra, controllerId), resourceId, sessionScope, requestContext)).permissions.getRules();
 			return {
 				categories: rules.categories,
 				tools: rules.tools