@mastra/server 1.51.1-alpha.0 → 1.52.0-alpha.11

package/dist/{chunk-WEAXQLOP.js → chunk-JCN5OOXU.js}

@@ -13,11 +13,11 @@ function getAgentControllerOrThrow(mastra, controllerId) {
   }
   return controller;
 }
-async function getSession(controller, resourceId, options) {
+async function getSession(controller, resourceId, options, requestContext) {
   await controller.init();
-  const { tags, scope } = options ?? {};
-  const id = scope ? `${resourceId}::${scope}` : resourceId;
-  return controller.createSession({ resourceId, id, ownerId: controller.id, tags, scope });
+  const { tags, scope, threadId } = options ?? {};
+  const id = threadId ?? (scope ? `${resourceId}::${scope}` : resourceId);
+  return controller.createSession({ resourceId, id, ownerId: controller.id, tags, scope, threadId, requestContext });
 }
 var controllerIdPathParams = z.object({ controllerId: z.string() });
 var sessionPathParams = z.object({ controllerId: z.string(), resourceId: z.string() });
@@ -25,12 +25,15 @@ var sessionScopeQuerySchema = z.object({ sessionScope: z.string().optional() });
 var createSessionBodySchema = z.object({
   resourceId: z.string(),
   tags: z.record(z.string(), z.string()).optional(),
+  threadId: z.string().optional(),
   sessionScope: z.string().optional()
 });
 var MAX_FILE_DATA_LENGTH = 14 * 1024 * 1024;
 var MAX_TOTAL_FILE_DATA_LENGTH = 28 * 1024 * 1024;
+var bodyRequestContextSchema = z.record(z.string(), z.unknown()).optional();
 var sendMessageBodySchema = z.object({
   message: z.string(),
+  requestContext: bodyRequestContextSchema,
   // Optional attachments (e.g. pasted images). `data` is base64-encoded.
   files: z.array(
     z.object({
@@ -42,17 +45,19 @@ var sendMessageBodySchema = z.object({
     message: "Total attachment size exceeds limit"
   }).optional()
 });
-var steerBodySchema = z.object({ message: z.string() });
+var steerBodySchema = z.object({ message: z.string(), requestContext: bodyRequestContextSchema });
 var toolApprovalBodySchema = z.object({
   toolCallId: z.string(),
-  approved: z.boolean()
+  approved: z.boolean(),
+  requestContext: bodyRequestContextSchema
 });
 var toolSuspensionBodySchema = z.object({
   toolCallId: z.string(),
   // Free-form resume payload. For ask_user this is a string (or string[] for
   // multi-select); for submit_plan it's `{ action, feedback? }`; for
   // request_access it's "Yes"/"No".
-  resumeData: z.any()
+  resumeData: z.unknown(),
+  requestContext: bodyRequestContextSchema
 });
 var switchModeBodySchema = z.object({ modeId: z.string() });
 var switchModelBodySchema = z.object({
@@ -81,13 +86,13 @@ var listThreadsQuerySchema = z.object({
     }
   }, z.record(z.string(), z.string()).optional()).optional()
 });
-var followUpBodySchema = z.object({ message: z.string() });
+var followUpBodySchema = z.object({ message: z.string(), requestContext: bodyRequestContextSchema });
 var sendNotificationBodySchema = z.object({
   source: z.string(),
   kind: z.string(),
   summary: z.string(),
   priority: z.enum(["low", "medium", "high", "urgent"]).optional(),
-  payload: z.any().optional(),
+  payload: z.unknown().optional(),
   sourceId: z.string().optional(),
   dedupeKey: z.string().optional(),
   coalesceKey: z.string().optional(),
@@ -157,16 +162,23 @@ var threadResponseSchema = z.object({
   createdAt: z.string().optional(),
   updatedAt: z.string().optional()
 });
-var messageContentSchema = z.object({
+var messagePartSchema = z.object({
   type: z.string()
 }).passthrough();
+var messageContentV2Schema = z.object({
+  format: z.literal(2),
+  parts: z.array(messagePartSchema)
+}).passthrough();
 var listMessagesResponseSchema = z.object({
   messages: z.array(
     z.object({
       id: z.string(),
-      role: z.enum(["user", "assistant", "system"]),
-      content: z.array(messageContentSchema),
-      createdAt: z.string().optional()
+      role: z.enum(["user", "assistant", "system", "tool", "signal"]),
+      content: messageContentV2Schema,
+      createdAt: z.string().optional(),
+      threadId: z.string().optional(),
+      resourceId: z.string().optional(),
+      type: z.string().optional()
     })
   )
 });
@@ -186,7 +198,7 @@ var workspaceStatusResponseSchema = z.object({
   isReady: z.boolean()
 });
 var omRecordResponseSchema = z.object({
-  record: z.any().optional()
+  record: z.unknown().optional()
 });
 var permissionPolicyEnum = z.enum(["allow", "ask", "deny"]);
 var toolCategoryEnum = z.enum(["read", "edit", "execute", "mcp", "other"]);
@@ -236,10 +248,10 @@ var CREATE_AGENT_CONTROLLER_SESSION_ROUTE = createRoute({
   tags: ["AgentController"],
   requiresAuth: true,
   requiresPermission: "agent-controller:execute",
-  handler: async ({ mastra, controllerId, resourceId, sessionScope, tags }) => {
+  handler: async ({ mastra, controllerId, resourceId, sessionScope, tags, threadId, requestContext }) => {
     try {
       const controller = getAgentControllerOrThrow(mastra, controllerId);
-      const session = await getSession(controller, resourceId, { tags, scope: sessionScope });
+      const session = await getSession(controller, resourceId, { tags, scope: sessionScope, threadId }, requestContext);
       return {
         controllerId,
         resourceId,
@@ -270,10 +282,10 @@ var STREAM_AGENT_CONTROLLER_SESSION_ROUTE = createRoute({
   tags: ["AgentController", "Streaming"],
   requiresAuth: true,
   requiresPermission: "agent-controller:read",
-  handler: async ({ mastra, controllerId, resourceId, sessionScope, abortSignal }) => {
+  handler: async ({ mastra, controllerId, resourceId, sessionScope, abortSignal, requestContext }) => {
     try {
       const controller = getAgentControllerOrThrow(mastra, controllerId);
-      const session = await getSession(controller, resourceId, { scope: sessionScope });
+      const session = await getSession(controller, resourceId, { scope: sessionScope }, requestContext);
       let cleanedUp = false;
       let heartbeat;
       let unsubscribe;
@@ -350,7 +362,7 @@ var SEND_AGENT_CONTROLLER_MESSAGE_ROUTE = createRoute({
   handler: async ({ mastra, controllerId, resourceId, sessionScope, message, files, requestContext }) => {
     try {
       const controller = getAgentControllerOrThrow(mastra, controllerId);
-      const session = await getSession(controller, resourceId, { scope: sessionScope });
+      const session = await getSession(controller, resourceId, { scope: sessionScope }, requestContext);
       void session.sendMessage({ content: message, files, requestContext });
       return { ok: true };
     } catch (error) {
@@ -370,10 +382,10 @@ var ABORT_AGENT_CONTROLLER_SESSION_ROUTE = createRoute({
   tags: ["AgentController"],
   requiresAuth: true,
   requiresPermission: "agent-controller:execute",
-  handler: async ({ mastra, controllerId, resourceId, sessionScope }) => {
+  handler: async ({ mastra, controllerId, resourceId, sessionScope, requestContext }) => {
     try {
       const controller = getAgentControllerOrThrow(mastra, controllerId);
-      const session = await getSession(controller, resourceId, { scope: sessionScope });
+      const session = await getSession(controller, resourceId, { scope: sessionScope }, requestContext);
       session.abort();
       return { ok: true };
     } catch (error) {
@@ -397,7 +409,7 @@ var AGENT_CONTROLLER_TOOL_APPROVAL_ROUTE = createRoute({
   handler: async ({ mastra, controllerId, resourceId, sessionScope, toolCallId, approved, requestContext }) => {
     try {
       const controller = getAgentControllerOrThrow(mastra, controllerId);
-      const session = await getSession(controller, resourceId, { scope: sessionScope });
+      const session = await getSession(controller, resourceId, { scope: sessionScope }, requestContext);
       session.respondToToolApproval({ toolCallId, decision: approved ? "approve" : "decline", requestContext });
       return { ok: true };
     } catch (error) {
@@ -421,7 +433,7 @@ var AGENT_CONTROLLER_TOOL_SUSPENSION_ROUTE = createRoute({
   handler: async ({ mastra, controllerId, resourceId, sessionScope, toolCallId, resumeData, requestContext }) => {
     try {
       const controller = getAgentControllerOrThrow(mastra, controllerId);
-      const session = await getSession(controller, resourceId, { scope: sessionScope });
+      const session = await getSession(controller, resourceId, { scope: sessionScope }, requestContext);
       await session.respondToToolSuspension({ toolCallId, resumeData, requestContext });
       return { ok: true };
     } catch (error) {
@@ -445,7 +457,7 @@ var STEER_AGENT_CONTROLLER_SESSION_ROUTE = createRoute({
   handler: async ({ mastra, controllerId, resourceId, sessionScope, message, requestContext }) => {
     try {
       const controller = getAgentControllerOrThrow(mastra, controllerId);
-      const session = await getSession(controller, resourceId, { scope: sessionScope });
+      const session = await getSession(controller, resourceId, { scope: sessionScope }, requestContext);
       void session.steer({ content: message, requestContext });
       return { ok: true };
     } catch (error) {
@@ -466,10 +478,10 @@ var SWITCH_AGENT_CONTROLLER_MODE_ROUTE = createRoute({
   tags: ["AgentController"],
   requiresAuth: true,
   requiresPermission: "agent-controller:execute",
-  handler: async ({ mastra, controllerId, resourceId, sessionScope, modeId }) => {
+  handler: async ({ mastra, controllerId, resourceId, sessionScope, modeId, requestContext }) => {
     try {
       const controller = getAgentControllerOrThrow(mastra, controllerId);
-      const session = await getSession(controller, resourceId, { scope: sessionScope });
+      const session = await getSession(controller, resourceId, { scope: sessionScope }, requestContext);
       await session.mode.switch({ modeId });
       return { ok: true };
     } catch (error) {
@@ -490,10 +502,10 @@ var SWITCH_AGENT_CONTROLLER_MODEL_ROUTE = createRoute({
   tags: ["AgentController"],
   requiresAuth: true,
   requiresPermission: "agent-controller:execute",
-  handler: async ({ mastra, controllerId, resourceId, sessionScope, modelId, scope, modeId }) => {
+  handler: async ({ mastra, controllerId, resourceId, sessionScope, modelId, scope, modeId, requestContext }) => {
     try {
       const controller = getAgentControllerOrThrow(mastra, controllerId);
-      const session = await getSession(controller, resourceId, { scope: sessionScope });
+      const session = await getSession(controller, resourceId, { scope: sessionScope }, requestContext);
       await session.model.switch({ modelId, scope, modeId });
       return { ok: true };
     } catch (error) {
@@ -514,10 +526,10 @@ var SWITCH_AGENT_CONTROLLER_THREAD_ROUTE = createRoute({
   tags: ["AgentController"],
   requiresAuth: true,
   requiresPermission: "agent-controller:execute",
-  handler: async ({ mastra, controllerId, resourceId, sessionScope, threadId }) => {
+  handler: async ({ mastra, controllerId, resourceId, sessionScope, threadId, requestContext }) => {
     try {
       const controller = getAgentControllerOrThrow(mastra, controllerId);
-      const session = await getSession(controller, resourceId, { scope: sessionScope });
+      const session = await getSession(controller, resourceId, { scope: sessionScope }, requestContext);
       if (session.thread.getId() !== threadId) {
         await session.thread.switch({ threadId });
       }
@@ -539,10 +551,10 @@ var GET_AGENT_CONTROLLER_SESSION_STATE_ROUTE = createRoute({
   tags: ["AgentController"],
   requiresAuth: true,
   requiresPermission: "agent-controller:read",
-  handler: async ({ mastra, controllerId, resourceId, sessionScope }) => {
+  handler: async ({ mastra, controllerId, resourceId, sessionScope, requestContext }) => {
     try {
       const controller = getAgentControllerOrThrow(mastra, controllerId);
-      const session = await getSession(controller, resourceId, { scope: sessionScope });
+      const session = await getSession(controller, resourceId, { scope: sessionScope }, requestContext);
       const ds = session.displayState.get();
       const om = ds.omProgress;
       const reflectionSavings = om.buffered.reflection.inputObservationTokens - om.buffered.reflection.observationTokens;
@@ -613,10 +625,10 @@ var LIST_AGENT_CONTROLLER_THREADS_ROUTE = createRoute({
   tags: ["AgentController"],
   requiresAuth: true,
   requiresPermission: "agent-controller:read",
-  handler: async ({ mastra, controllerId, resourceId, sessionScope, limit, tags }) => {
+  handler: async ({ mastra, controllerId, resourceId, sessionScope, limit, tags, requestContext }) => {
     try {
       const controller = getAgentControllerOrThrow(mastra, controllerId);
-      const session = await getSession(controller, resourceId, { scope: sessionScope });
+      const session = await getSession(controller, resourceId, { scope: sessionScope }, requestContext);
       const threads = await session.thread.list();
       const getTags = (t) => {
         const metadata = t.metadata ?? {};
@@ -685,11 +697,12 @@ var SEND_AGENT_CONTROLLER_NOTIFICATION_ROUTE = createRoute({
     dedupeKey,
     coalesceKey,
     attributes,
-    metadata
+    metadata,
+    requestContext
   }) => {
     try {
       const controller = getAgentControllerOrThrow(mastra, controllerId);
-      const session = await getSession(controller, resourceId, { scope: sessionScope });
+      const session = await getSession(controller, resourceId, { scope: sessionScope }, requestContext);
       const result = await session.sendNotificationSignal({
         source,
         kind,
@@ -726,10 +739,10 @@ var CREATE_AGENT_CONTROLLER_THREAD_ROUTE = createRoute({
   tags: ["AgentController", "Threads"],
   requiresAuth: true,
   requiresPermission: "agent-controller:execute",
-  handler: async ({ mastra, controllerId, resourceId, sessionScope, title }) => {
+  handler: async ({ mastra, controllerId, resourceId, sessionScope, title, requestContext }) => {
     try {
       const controller = getAgentControllerOrThrow(mastra, controllerId);
-      const session = await getSession(controller, resourceId, { scope: sessionScope });
+      const session = await getSession(controller, resourceId, { scope: sessionScope }, requestContext);
       const thread = await session.thread.create({ title });
       return {
         id: thread.id,
@@ -755,10 +768,10 @@ var DELETE_AGENT_CONTROLLER_THREAD_ROUTE = createRoute({
   tags: ["AgentController", "Threads"],
   requiresAuth: true,
   requiresPermission: "agent-controller:execute",
-  handler: async ({ mastra, controllerId, resourceId, sessionScope, threadId }) => {
+  handler: async ({ mastra, controllerId, resourceId, sessionScope, threadId, requestContext }) => {
     try {
       const controller = getAgentControllerOrThrow(mastra, controllerId);
-      const session = await getSession(controller, resourceId, { scope: sessionScope });
+      const session = await getSession(controller, resourceId, { scope: sessionScope }, requestContext);
       await session.thread.delete({ threadId });
       return { ok: true };
     } catch (error) {
@@ -779,10 +792,10 @@ var RENAME_AGENT_CONTROLLER_THREAD_ROUTE = createRoute({
   tags: ["AgentController", "Threads"],
   requiresAuth: true,
   requiresPermission: "agent-controller:execute",
-  handler: async ({ mastra, controllerId, resourceId, sessionScope, threadId, title }) => {
+  handler: async ({ mastra, controllerId, resourceId, sessionScope, threadId, title, requestContext }) => {
     try {
       const controller = getAgentControllerOrThrow(mastra, controllerId);
-      const session = await getSession(controller, resourceId, { scope: sessionScope });
+      const session = await getSession(controller, resourceId, { scope: sessionScope }, requestContext);
       if (session.thread.getId() !== threadId) {
         await session.thread.switch({ threadId });
       }
@@ -806,10 +819,10 @@ var CLONE_AGENT_CONTROLLER_THREAD_ROUTE = createRoute({
   tags: ["AgentController", "Threads"],
   requiresAuth: true,
   requiresPermission: "agent-controller:execute",
-  handler: async ({ mastra, controllerId, resourceId, sessionScope, sourceThreadId, title }) => {
+  handler: async ({ mastra, controllerId, resourceId, sessionScope, sourceThreadId, title, requestContext }) => {
     try {
       const controller = getAgentControllerOrThrow(mastra, controllerId);
-      const session = await getSession(controller, resourceId, { scope: sessionScope });
+      const session = await getSession(controller, resourceId, { scope: sessionScope }, requestContext);
       const thread = await session.thread.clone({ sourceThreadId, title });
       return {
         id: thread.id,
@@ -835,17 +848,20 @@ var LIST_AGENT_CONTROLLER_THREAD_MESSAGES_ROUTE = createRoute({
   tags: ["AgentController", "Threads"],
   requiresAuth: true,
   requiresPermission: "agent-controller:read",
-  handler: async ({ mastra, controllerId, resourceId, sessionScope, threadId, limit }) => {
+  handler: async ({ mastra, controllerId, resourceId, sessionScope, threadId, limit, requestContext }) => {
     try {
       const controller = getAgentControllerOrThrow(mastra, controllerId);
-      const session = await getSession(controller, resourceId, { scope: sessionScope });
+      const session = await getSession(controller, resourceId, { scope: sessionScope }, requestContext);
       const messages = await session.thread.listMessages({ threadId, limit });
       return {
         messages: messages.map((m) => ({
           id: m.id,
           role: m.role,
           content: m.content,
-          createdAt: m.createdAt instanceof Date ? m.createdAt.toISOString() : void 0
+          createdAt: m.createdAt instanceof Date ? m.createdAt.toISOString() : void 0,
+          threadId: m.threadId,
+          resourceId: m.resourceId,
+          type: m.type
         }))
       };
     } catch (error) {
@@ -869,7 +885,7 @@ var FOLLOW_UP_AGENT_CONTROLLER_SESSION_ROUTE = createRoute({
   handler: async ({ mastra, controllerId, resourceId, sessionScope, message, requestContext }) => {
     try {
       const controller = getAgentControllerOrThrow(mastra, controllerId);
-      const session = await getSession(controller, resourceId, { scope: sessionScope });
+      const session = await getSession(controller, resourceId, { scope: sessionScope }, requestContext);
       void session.followUp({ content: message, requestContext });
       return { ok: true };
     } catch (error) {
@@ -943,10 +959,10 @@ var GET_AGENT_CONTROLLER_OM_RECORD_ROUTE = createRoute({
   tags: ["AgentController"],
   requiresAuth: true,
   requiresPermission: "agent-controller:read",
-  handler: async ({ mastra, controllerId, resourceId, sessionScope }) => {
+  handler: async ({ mastra, controllerId, resourceId, sessionScope, requestContext }) => {
     try {
       const controller = getAgentControllerOrThrow(mastra, controllerId);
-      const session = await getSession(controller, resourceId, { scope: sessionScope });
+      const session = await getSession(controller, resourceId, { scope: sessionScope }, requestContext);
       const record = await controller.getObservationalMemoryRecord(session);
       return { record: record ?? void 0 };
     } catch (error) {
@@ -967,10 +983,10 @@ var SET_AGENT_CONTROLLER_RESOURCE_ID_ROUTE = createRoute({
   tags: ["AgentController"],
   requiresAuth: true,
   requiresPermission: "agent-controller:execute",
-  handler: async ({ mastra, controllerId, resourceId, sessionScope, newResourceId }) => {
+  handler: async ({ mastra, controllerId, resourceId, sessionScope, newResourceId, requestContext }) => {
     try {
       const controller = getAgentControllerOrThrow(mastra, controllerId);
-      const session = await getSession(controller, resourceId, { scope: sessionScope });
+      const session = await getSession(controller, resourceId, { scope: sessionScope }, requestContext);
       await controller.setResourceId(session, { resourceId: newResourceId });
       return { ok: true };
     } catch (error) {
@@ -990,10 +1006,10 @@ var GET_AGENT_CONTROLLER_RESOURCE_IDS_ROUTE = createRoute({
   tags: ["AgentController"],
   requiresAuth: true,
   requiresPermission: "agent-controller:read",
-  handler: async ({ mastra, controllerId, resourceId, sessionScope }) => {
+  handler: async ({ mastra, controllerId, resourceId, sessionScope, requestContext }) => {
     try {
       const controller = getAgentControllerOrThrow(mastra, controllerId);
-      const session = await getSession(controller, resourceId, { scope: sessionScope });
+      const session = await getSession(controller, resourceId, { scope: sessionScope }, requestContext);
       const resourceIds = await controller.getKnownResourceIds(session);
       return { resourceIds };
     } catch (error) {
@@ -1038,10 +1054,10 @@ var GET_AGENT_CONTROLLER_GOAL_ROUTE = createRoute({
   tags: ["AgentController", "Goals"],
   requiresAuth: true,
   requiresPermission: "agent-controller:read",
-  handler: async ({ mastra, controllerId, resourceId, sessionScope }) => {
+  handler: async ({ mastra, controllerId, resourceId, sessionScope, requestContext }) => {
     try {
       const controller = getAgentControllerOrThrow(mastra, controllerId);
-      const session = await getSession(controller, resourceId, { scope: sessionScope });
+      const session = await getSession(controller, resourceId, { scope: sessionScope }, requestContext);
       const threadId = session.thread.getId();
       if (!threadId) return { goal: void 0 };
       const agent = getAgentForSession(controller, session);
@@ -1065,10 +1081,19 @@ var SET_AGENT_CONTROLLER_GOAL_ROUTE = createRoute({
   tags: ["AgentController", "Goals"],
   requiresAuth: true,
   requiresPermission: "agent-controller:execute",
-  handler: async ({ mastra, controllerId, resourceId, sessionScope, objective, judgeModelId, maxRuns }) => {
+  handler: async ({
+    mastra,
+    controllerId,
+    resourceId,
+    sessionScope,
+    objective,
+    judgeModelId,
+    maxRuns,
+    requestContext
+  }) => {
     try {
       const controller = getAgentControllerOrThrow(mastra, controllerId);
-      const session = await getSession(controller, resourceId, { scope: sessionScope });
+      const session = await getSession(controller, resourceId, { scope: sessionScope }, requestContext);
       const threadId = session.thread.getId();
       if (!threadId) throw new HTTPException(400, { message: "session has no active thread" });
       const agent = getAgentForSession(controller, session);
@@ -1097,10 +1122,19 @@ var UPDATE_AGENT_CONTROLLER_GOAL_ROUTE = createRoute({
   tags: ["AgentController", "Goals"],
   requiresAuth: true,
   requiresPermission: "agent-controller:execute",
-  handler: async ({ mastra, controllerId, resourceId, sessionScope, judgeModelId, maxRuns, status }) => {
+  handler: async ({
+    mastra,
+    controllerId,
+    resourceId,
+    sessionScope,
+    judgeModelId,
+    maxRuns,
+    status,
+    requestContext
+  }) => {
     try {
       const controller = getAgentControllerOrThrow(mastra, controllerId);
-      const session = await getSession(controller, resourceId, { scope: sessionScope });
+      const session = await getSession(controller, resourceId, { scope: sessionScope }, requestContext);
       const threadId = session.thread.getId();
       if (!threadId) throw new HTTPException(400, { message: "session has no active thread" });
       const agent = getAgentForSession(controller, session);
@@ -1128,10 +1162,10 @@ var CLEAR_AGENT_CONTROLLER_GOAL_ROUTE = createRoute({
   tags: ["AgentController", "Goals"],
   requiresAuth: true,
   requiresPermission: "agent-controller:execute",
-  handler: async ({ mastra, controllerId, resourceId, sessionScope }) => {
+  handler: async ({ mastra, controllerId, resourceId, sessionScope, requestContext }) => {
     try {
       const controller = getAgentControllerOrThrow(mastra, controllerId);
-      const session = await getSession(controller, resourceId, { scope: sessionScope });
+      const session = await getSession(controller, resourceId, { scope: sessionScope }, requestContext);
       const threadId = session.thread.getId();
       if (!threadId) throw new HTTPException(400, { message: "session has no active thread" });
       const agent = getAgentForSession(controller, session);
@@ -1154,10 +1188,10 @@ var GET_AGENT_CONTROLLER_PERMISSIONS_ROUTE = createRoute({
   tags: ["AgentController", "Permissions"],
   requiresAuth: true,
   requiresPermission: "agent-controller:read",
-  handler: async ({ mastra, controllerId, resourceId, sessionScope }) => {
+  handler: async ({ mastra, controllerId, resourceId, sessionScope, requestContext }) => {
     try {
       const controller = getAgentControllerOrThrow(mastra, controllerId);
-      const session = await getSession(controller, resourceId, { scope: sessionScope });
+      const session = await getSession(controller, resourceId, { scope: sessionScope }, requestContext);
       const rules = session.permissions.getRules();
       return {
         categories: rules.categories,
@@ -1181,10 +1215,10 @@ var SET_AGENT_CONTROLLER_CATEGORY_PERMISSION_ROUTE = createRoute({
   tags: ["AgentController", "Permissions"],
   requiresAuth: true,
   requiresPermission: "agent-controller:execute",
-  handler: async ({ mastra, controllerId, resourceId, sessionScope, category, policy }) => {
+  handler: async ({ mastra, controllerId, resourceId, sessionScope, category, policy, requestContext }) => {
     try {
       const controller = getAgentControllerOrThrow(mastra, controllerId);
-      const session = await getSession(controller, resourceId, { scope: sessionScope });
+      const session = await getSession(controller, resourceId, { scope: sessionScope }, requestContext);
       await session.permissions.setForCategory({ category, policy });
       return { ok: true };
     } catch (error) {
@@ -1205,10 +1239,10 @@ var SET_AGENT_CONTROLLER_TOOL_PERMISSION_ROUTE = createRoute({
   tags: ["AgentController", "Permissions"],
   requiresAuth: true,
   requiresPermission: "agent-controller:execute",
-  handler: async ({ mastra, controllerId, resourceId, sessionScope, toolName, policy }) => {
+  handler: async ({ mastra, controllerId, resourceId, sessionScope, toolName, policy, requestContext }) => {
     try {
       const controller = getAgentControllerOrThrow(mastra, controllerId);
-      const session = await getSession(controller, resourceId, { scope: sessionScope });
+      const session = await getSession(controller, resourceId, { scope: sessionScope }, requestContext);
       await session.permissions.setForTool({ toolName, policy });
       return { ok: true };
     } catch (error) {
@@ -1230,10 +1264,10 @@ var SET_AGENT_CONTROLLER_SESSION_STATE_ROUTE = createRoute({
   tags: ["AgentController"],
   requiresAuth: true,
   requiresPermission: "agent-controller:execute",
-  handler: async ({ mastra, controllerId, resourceId, sessionScope, state }) => {
+  handler: async ({ mastra, controllerId, resourceId, sessionScope, state, requestContext }) => {
     try {
       const controller = getAgentControllerOrThrow(mastra, controllerId);
-      const session = await getSession(controller, resourceId, { scope: sessionScope });
+      const session = await getSession(controller, resourceId, { scope: sessionScope }, requestContext);
       await session.state.set(state);
       return { ok: true };
     } catch (error) {
@@ -1243,5 +1277,5 @@ var SET_AGENT_CONTROLLER_SESSION_STATE_ROUTE = createRoute({
 });
 
 export { ABORT_AGENT_CONTROLLER_SESSION_ROUTE, AGENT_CONTROLLER_TOOL_APPROVAL_ROUTE, AGENT_CONTROLLER_TOOL_SUSPENSION_ROUTE, CLEAR_AGENT_CONTROLLER_GOAL_ROUTE, CLONE_AGENT_CONTROLLER_THREAD_ROUTE, CREATE_AGENT_CONTROLLER_SESSION_ROUTE, CREATE_AGENT_CONTROLLER_THREAD_ROUTE, DELETE_AGENT_CONTROLLER_THREAD_ROUTE, FOLLOW_UP_AGENT_CONTROLLER_SESSION_ROUTE, GET_AGENT_CONTROLLER_GOAL_ROUTE, GET_AGENT_CONTROLLER_OM_RECORD_ROUTE, GET_AGENT_CONTROLLER_PERMISSIONS_ROUTE, GET_AGENT_CONTROLLER_RESOURCE_IDS_ROUTE, GET_AGENT_CONTROLLER_SESSION_STATE_ROUTE, GET_AGENT_CONTROLLER_WORKSPACE_STATUS_ROUTE, LIST_AGENT_CONTROLLERS_ROUTE, LIST_AGENT_CONTROLLER_MODELS_ROUTE, LIST_AGENT_CONTROLLER_MODES_ROUTE, LIST_AGENT_CONTROLLER_THREADS_ROUTE, LIST_AGENT_CONTROLLER_THREAD_MESSAGES_ROUTE, RENAME_AGENT_CONTROLLER_THREAD_ROUTE, SEND_AGENT_CONTROLLER_MESSAGE_ROUTE, SEND_AGENT_CONTROLLER_NOTIFICATION_ROUTE, SET_AGENT_CONTROLLER_CATEGORY_PERMISSION_ROUTE, SET_AGENT_CONTROLLER_GOAL_ROUTE, SET_AGENT_CONTROLLER_RESOURCE_ID_ROUTE, SET_AGENT_CONTROLLER_SESSION_STATE_ROUTE, SET_AGENT_CONTROLLER_TOOL_PERMISSION_ROUTE, STEER_AGENT_CONTROLLER_SESSION_ROUTE, STREAM_AGENT_CONTROLLER_SESSION_ROUTE, SWITCH_AGENT_CONTROLLER_MODEL_ROUTE, SWITCH_AGENT_CONTROLLER_MODE_ROUTE, SWITCH_AGENT_CONTROLLER_THREAD_ROUTE, UPDATE_AGENT_CONTROLLER_GOAL_ROUTE };
-//# sourceMappingURL=chunk-WEAXQLOP.js.map
-//# sourceMappingURL=chunk-WEAXQLOP.js.map
+//# sourceMappingURL=chunk-JCN5OOXU.js.map
+//# sourceMappingURL=chunk-JCN5OOXU.js.map