Mastra•3mo ago

@mastra/ai-sdk chatRoute with middleware context

Hey all, I'm trying to use the chatRoute from the mastra ai-sdk and I have middleware configured that is grabbing some additional metadata my backend sends and adding it to the runtime context. When the agent is called up with a users chat my middleware sees and sets the runtimeContext successfully but the agent runtimeContext is empty.

Is this a known issue or perhaps something I'm missing with this setup? This is from the docs here https://mastra.ai/en/docs/frameworks/agentic-uis/ai-sdk

export const mastra = new Mastra({
  workflows: { fileUploadProcessorWorkflow },
  agents: { fileProcessorAgent, investmentChatAgent },
  storage: getStorage(),

  // Server configuration to bind to all interfaces
  server: {
    port: 4111,
    host: "0.0.0.0",

    middleware: [
      async (c, next) => {
        const runtimeContext = c.get("runtimeContext");
        if (c.req.method === "POST" && c.req.url.includes("/chat")) {
          try {
            const clonedReq = c.req.raw.clone();
            const body = await clonedReq.json();
            if (body?.data) {
              for (const [key, value] of Object.entries(body.data)) {
                runtimeContext.set(key, value);
              }
            }
          } catch {}
        }
        await next();
      },
    ],

    apiRoutes: [
      chatRoute({
        path: "/chat",
        agent: "investmentChatAgent",
      }),
    ],
  }
})

export const mastra = new Mastra({
  workflows: { fileUploadProcessorWorkflow },
  agents: { fileProcessorAgent, investmentChatAgent },
  storage: getStorage(),

  // Server configuration to bind to all interfaces
  server: {
    port: 4111,
    host: "0.0.0.0",

    middleware: [
      async (c, next) => {
        const runtimeContext = c.get("runtimeContext");
        if (c.req.method === "POST" && c.req.url.includes("/chat")) {
          try {
            const clonedReq = c.req.raw.clone();
            const body = await clonedReq.json();
            if (body?.data) {
              for (const [key, value] of Object.entries(body.data)) {
                runtimeContext.set(key, value);
              }
            }
          } catch {}
        }
        await next();
      },
    ],

    apiRoutes: [
      chatRoute({
        path: "/chat",
        agent: "investmentChatAgent",
      }),
    ],
  }
})

export const mastra = new Mastra({
  workflows: { fileUploadProcessorWorkflow },
  agents: { fileProcessorAgent, investmentChatAgent },
  storage: getStorage(),

  // Server configuration to bind to all interfaces
  server: {
    port: 4111,
    host: "0.0.0.0",

    middleware: [
      async (c, next) => {
        const runtimeContext = c.get("runtimeContext");
        if (c.req.method === "POST" && c.req.url.includes("/chat")) {
          try {
            const clonedReq = c.req.raw.clone();
            const body = await clonedReq.json();
            if (body?.data) {
              for (const [key, value] of Object.entries(body.data)) {
                runtimeContext.set(key, value);
              }
            }
          } catch {}
        }
        await next();
      },
    ],

    apiRoutes: [
      chatRoute({
        path: "/chat",
        agent: "investmentChatAgent",
      }),
    ],
  }
})

export const mastra = new Mastra({
  workflows: { fileUploadProcessorWorkflow },
  agents: { fileProcessorAgent, investmentChatAgent },
  storage: getStorage(),

  // Server configuration to bind to all interfaces
  server: {
    port: 4111,
    host: "0.0.0.0",

    middleware: [
      async (c, next) => {
        const runtimeContext = c.get("runtimeContext");
        if (c.req.method === "POST" && c.req.url.includes("/chat")) {
          try {
            const clonedReq = c.req.raw.clone();
            const body = await clonedReq.json();
            if (body?.data) {
              for (const [key, value] of Object.entries(body.data)) {
                runtimeContext.set(key, value);
              }
            }
          } catch {}
        }
        await next();
      },
    ],

    apiRoutes: [
      chatRoute({
        path: "/chat",
        agent: "investmentChatAgent",
      }),
    ],
  }
})

Incisiv3OP•10/8/25, 1:22 AM

Or is there another way when ineracting with this new /chat/chat endpoint to send along data i'd like to be put in runtimeContext automatically? my setup is basically react frontend -> NestJS backend server -> mastra API

Mastra Triager•10/8/25, 12:05 PM

Created GitHub issue: https://github.com/mastra-ai/mastra/issues/8609

GitHub

[DISCORD:1425287716114792499] @mastra/ai-sdk chatRoute with middlew...

This issue was created from Discord post: https://discord.com/channels/1309558646228779139/1425287716114792499 Hey all, I'm trying to use the chatRoute from the mastra ai-sdk and I have middlew...

_roamin_•10/8/25, 2:28 PM

Hi @Incisiv3 ! I think that's because you're not setting the runtimeContextruntimeContext back into the hono context, can you try adding c.set("runtimeContext", runtimeContext);c.set("runtimeContext", runtimeContext); after you're do setting up your runtimeContext?

Incisiv3OP•10/8/25, 5:54 PM

Oh fair point yes let me give it a try shortly and I’ll get back to you

Incisiv3OP•10/8/25, 6:50 PM

@Romain I adjusted it to re-set the runtimeContext and my agent still is getting an empty object when it's invoked

    middleware: [
      async (c, next) => {
        if (c.req.method === "POST" && c.req.url.includes("/chat")) {
          try {
            const clonedReq = c.req.raw.clone();
            const body = await clonedReq.json();
            if (body?.data) {
              let runtimeContext = c.get("runtimeContext");
              // Initialize if it doesn't exist
              if (!runtimeContext) {
                runtimeContext = new Map();
              }
              for (const [key, value] of Object.entries(body.data)) {
                runtimeContext.set(key, value);
              }
              c.set("runtimeContext", runtimeContext);
            }
          } catch (error) {
            console.error("Error setting runtime context:", error);
          }
        }
        await next();
      },
    ],

    middleware: [
      async (c, next) => {
        if (c.req.method === "POST" && c.req.url.includes("/chat")) {
          try {
            const clonedReq = c.req.raw.clone();
            const body = await clonedReq.json();
            if (body?.data) {
              let runtimeContext = c.get("runtimeContext");
              // Initialize if it doesn't exist
              if (!runtimeContext) {
                runtimeContext = new Map();
              }
              for (const [key, value] of Object.entries(body.data)) {
                runtimeContext.set(key, value);
              }
              c.set("runtimeContext", runtimeContext);
            }
          } catch (error) {
            console.error("Error setting runtime context:", error);
          }
        }
        await next();
      },
    ],

IIncisiv3 @Romain I adjusted it to re-set the runtimeContext and my agent still is getting...

_roamin_•10/8/25, 9:11 PM

Hmm, I think that's a bug in chatRoute... I think that the first code you shared should have been working. Let me check something.

Incisiv3OP•10/8/25, 9:16 PM

Ok thanks

IIncisiv3 Ok thanks

_roamin_•10/8/25, 10:09 PM

Could you try using the tag fix-runtimeContext-passing-chatRoute as your Mastra packages version and let me know if it's correctly passing the runtime context?

Incisiv3OP•10/8/25, 11:38 PM

@Romain yep that seems to have fixed it! seems to work without me explicitly setting the runtime context back too with c.set("runtimeContext", runtimeContext);c.set("runtimeContext", runtimeContext);

Omicrxn•10/9/25, 1:39 PM

hey @Romain i was just about to post the same issue, is the fix going to be merged with main branch or should I use the fix-runtimeContext-passing-chatRoutefix-runtimeContext-passing-chatRoute?

_roamin_•10/9/25, 2:03 PM

Thanks @Omicrxn & @Incisiv3 for testing out the fix, I'll have the PR reviewed and merged asap.

Omicrxn•10/9/25, 2:04 PM

Thanks @Romain !

_roamin_•10/9/25, 2:07 PM

You can use the snapshort version above for now btw, I'll try to have an alpha released today, but not sure we'll have another stable release before next tuesday

Omicrxn•10/9/25, 2:10 PM

Okay I'll wait till stable release then thank you!

Incisiv3OP•10/9/25, 3:07 PM

Thanks all!

Ata Sanchez•12/13/25, 2:46 AM

hey guys, could this problem have spread to the beta version V1? I'm using the beta versions of Mastra, and I'm experiencing the exact same issue with chatRoute and networkRoute. If I use Mastra Studio directly, the requestContext is set correctly and I can use it, but when using the routes for the AI SDK, it isn't set and is always empty in the agent, tools, workflows, etc.

Ata Sanchez•12/13/25, 2:46 AM

import { chatRoute } from '@mastra/ai-sdk';
import { quoteAgentMiddleware } from '../middleware';
import { gmailRoutes } from '../routes/integrations/gmail.routes';
import { platformRoutes } from '../routes/platform.routes';
import { corsConfig } from './cors';

export const serverConfig = {
    cors: corsConfig,
    middleware: [
        {
            handler: quoteAgentMiddleware,
            path: '/agent/quote-agent/*',
        },
    ],
    apiRoutes: [
        chatRoute({
            path: '/platform/chat',
            agent: 'retripAgent',
        }),
        chatRoute({
            path: '/agent/quote-agent/chat',
            agent: 'quoteAgent',
        }),
        ...platformRoutes, // Mantienen su lógica separada
        ...gmailRoutes, // Mantienen su lógica separada
    ],
};

import { chatRoute } from '@mastra/ai-sdk';
import { quoteAgentMiddleware } from '../middleware';
import { gmailRoutes } from '../routes/integrations/gmail.routes';
import { platformRoutes } from '../routes/platform.routes';
import { corsConfig } from './cors';

export const serverConfig = {
    cors: corsConfig,
    middleware: [
        {
            handler: quoteAgentMiddleware,
            path: '/agent/quote-agent/*',
        },
    ],
    apiRoutes: [
        chatRoute({
            path: '/platform/chat',
            agent: 'retripAgent',
        }),
        chatRoute({
            path: '/agent/quote-agent/chat',
            agent: 'quoteAgent',
        }),
        ...platformRoutes, // Mantienen su lógica separada
        ...gmailRoutes, // Mantienen su lógica separada
    ],
};

Before the update (with runtimeContext) this code worked.

Ata Sanchez•12/13/25, 2:46 AM

export const quoteAgentMiddleware: MiddlewareHandler = async (c, next) => {
    const requestContext = c.get('requestContext') as RequestContext;

    // Extract data from body
    let body: { agentId?: string; organizationId?: string };
    try {
        const clonedRequest = c.req.raw.clone();
        body = await clonedRequest.json();
    } catch {
        return c.json({ error: 'Invalid request body. Expected JSON.' }, 400);
    }

    // Validate required fields
    if (!body.agentId) {
        return c.json({ error: 'agentId is required in request body' }, 400);
    }

    if (!body.organizationId) {
        return c.json({ error: 'organizationId is required in request body' }, 400);
    }

    // Store in requestContext
    requestContext.set('agentId', body.agentId);
    requestContext.set('organizationId', body.organizationId);

    await next();
};

export const quoteAgentMiddleware: MiddlewareHandler = async (c, next) => {
    const requestContext = c.get('requestContext') as RequestContext;

    // Extract data from body
    let body: { agentId?: string; organizationId?: string };
    try {
        const clonedRequest = c.req.raw.clone();
        body = await clonedRequest.json();
    } catch {
        return c.json({ error: 'Invalid request body. Expected JSON.' }, 400);
    }

    // Validate required fields
    if (!body.agentId) {
        return c.json({ error: 'agentId is required in request body' }, 400);
    }

    if (!body.organizationId) {
        return c.json({ error: 'organizationId is required in request body' }, 400);
    }

    // Store in requestContext
    requestContext.set('agentId', body.agentId);
    requestContext.set('organizationId', body.organizationId);

    await next();
};

Ata Sanchez•12/13/25, 2:46 AM

CC @_roamin_