feat: enhance agent functionality and add new tools

- Updated access control enforcement in tools and workflows to use `RequestContext` and agent-specific runtime context types.
- Added optional fields `userId` and `researchPhase` to `ResearchRuntimeContext` for improved context handling.
- Imported new agents: `calendarAgent`, `bgColorAgent`, and `danePackagePublisher` into the main index.
- Implemented a new tool `colorChangeTool` to change the background color with a defined input schema.

Author: ssdeanx

src/mastra/AGENTS.md

@@ -45,7 +45,7 @@ This folder contains the middleware and high-level components that define Mastra
 
 - Use explicit Zod schemas for every tool input/output.
 - Keep tools small and side-effect-free when possible; agents orchestrate tools and handle context.
-- Use `RuntimeContext` to enforce access control in tools and workflows.
+- Use `RequestContext` (per-request) or agent-specific runtime context types (e.g. `EditorRuntimeContext`) to enforce access control in tools and workflows.
 
 ---
 Last updated: 2025-11-26

src/mastra/agents/researchAgent.ts

@@ -34,6 +34,9 @@ export type UserTier = 'free' | 'pro' | 'enterprise'
 export interface ResearchRuntimeContext {
   'user-tier': UserTier
   language: 'en' | 'es' | 'ja' | 'fr'
+  // Optional runtime fields the server middleware may populate
+  userId?: string
+  researchPhase?: 'initial' | 'followup' | 'validation' | string
 }
 log.info('Initializing Research Agent...')
 
@@ -46,13 +49,18 @@ export const researchAgent = new Agent({
     // runtimeContext is read at invocation time
     const userTier = requestContext.get('user-tier') ?? 'free'
     const language = requestContext.get('language') ?? 'en'
+    const userId = requestContext.get('userId') ?? 'anonymous'
+    const researchPhase = requestContext.get('researchPhase') ?? 'initial'
+
     return {
       role: 'system',
       content: `
         <role>
 
         Tier: ${userTier}
         Language: ${language}
+        UserId: ${userId}
+        Research Phase: ${researchPhase}
 
         You are a Senior Research Analyst. Your goal is to research topics thoroughly by following a precise, multi-phase process.
         </role>
@@ -160,8 +168,8 @@ export const researchAgent = new Agent({
       // higher quality (chat style) for enterprise
       return google.chat('gemini-3-pro-preview')
     } else if (userTier === 'pro') {
-      // Chat bison for pro as well
-      return googleAI
+      // cheaper/faster model for pro tier
+      return 'google/gemini-2.5-flash-preview-09-2025'
     }
     // cheaper/faster model for free tier
     return google.chat('gemini-2.5-flash-preview-09-2025')

src/mastra/index.ts

@@ -51,6 +51,9 @@ import { researchPaperAgent } from './agents/researchPaperAgent';
 
 // Utility Agents
 import { daneNewContributor } from './workflows/new-contributor';
+import { calendarAgent } from './agents/calendarAgent';
+import { bgColorAgent } from './agents/bgColorAgent';
+import { danePackagePublisher } from './agents/package-publisher';
 
 // Financial Chart Agents
 import {
@@ -72,6 +75,7 @@ import { codingA2ACoordinator } from './a2a/codingA2ACoordinator';
 
 // Workflows
 import { trace } from "@opentelemetry/api";
+import type { RequestContext } from '@mastra/core/request-context';
 import { acpAgent } from './agents/acpAgent';
 import { businessStrategyAgent, complianceMonitoringAgent, contractAnalysisAgent, legalResearchAgent } from "./agents/businessLegalAgents";
 import { dane, daneChangeLog, daneCommitMessage, daneIssueLabeler, daneLinkChecker } from './agents/dane';
@@ -140,6 +144,11 @@ export const mastra = new Mastra({
     daneLinkChecker,
     daneChangeLog,
     dane,
+    // Calendar and misc
+    calendarAgent,
+    bgColorAgent,
+    // Package publisher
+    danePackagePublisher,
     // Financial Chart Agents
     chartTypeAdvisorAgent,
     chartDataProcessorAgent,
@@ -453,37 +462,63 @@ export const mastra = new Mastra({
         sendSources: true,
       }),
     ],
-//    cors: {
-//      origin: ["*"], // Allow specific origins or '*' for all
-//      allowMethods: ["GET", "POST", "PUT", "DELETE", "OPTIONS"],
-//      allowHeaders: ["Content-Type", "Authorization"],
-//      exposeHeaders: ["Content-Length", "X-Requested-With"],
-//      credentials: false,
-//    },
-//    middleware: [
-      // Middleware to extract data from the request body
-//      async (c, next) => {
-//        const runtimeContext = c.get("runtimeContext");
-//
-//        if (c.req.method === "POST") {
-//          try {
-//            const clonedReq = c.req.raw.clone();
-///            const body = await clonedReq.json();
-//
-            // Ensure body and body.data are objects before using them
-//            if (body !== null && typeof body === 'object' && body.data !== null && typeof body.data === 'object') {
-//              const data = body.data as Record<string, unknown>;
-//              for (const [key, value] of Object.entries(data)) {
-//                runtimeContext.set(key, value);
-//              }
-//            }
-//          } catch {
-//            log.error("Failed to parse request body for middleware");
-//          }
-//        }
-//        await next();
-//      },
-//0          ],
+    middleware: [
+      // Populate RequestContext with real runtime values derived from headers (used by agents/tools)
+      async (c, next) => {
+        const country = c.req.header('CF-IPCountry') ?? ''
+        const authHeader = c.req.header('Authorization') ?? ''
+        const headerUserId = c.req.header('x-user-id')
+        const headerUserTier = c.req.header('x-user-tier')
+        const acceptLanguage = c.req.header('accept-language') ?? ''
+        const researchPhaseHeader = c.req.header('x-research-phase')
+
+        const requestContext = c.get('requestContext') as RequestContext | undefined
+        if (requestContext?.set) {
+          // Temperature unit (from Cloudflare geo header)
+          const unit = country === 'US' ? 'fahrenheit' : 'celsius'
+          requestContext.set('temperature-unit', unit)
+
+          // userId: prefer explicit header, otherwise try to parse from a bearer token (format: "Bearer user:<id>")
+          let userId = headerUserId
+          if (!userId && authHeader !== null && authHeader !== '' && authHeader.startsWith('Bearer ')) {
+            const token = authHeader.slice('Bearer '.length)
+            const exec = /user:([^;\s]+)/.exec(token)
+            if (exec) { userId = exec[1] }
+          }
+          if (userId !== null && userId !== '') { requestContext.set('userId', userId) }
+
+          // user-tier: prefer explicit header, otherwise derive from token hints
+          let userTier = headerUserTier
+          if (!userTier && authHeader !== null && authHeader !== '' && authHeader.startsWith('Bearer ')) {
+            const token = authHeader.slice('Bearer '.length)
+            if (token.includes('enterprise')) { userTier = 'enterprise' }
+            else if (token.includes('pro')) { userTier = 'pro' }
+            else { userTier = 'free' }
+          }
+          if (userTier !== null && userTier !== '') { requestContext.set('user-tier', userTier) }
+
+          // language: prefer Accept-Language header (primary language subtag), fallback to 'en'
+          const language = acceptLanguage.split(',')[0]?.split('-')[0] ?? 'en'
+          const supported = ['en', 'es', 'ja', 'fr']
+          requestContext.set('language', supported.includes(language) ? language : 'en')
+
+          // research phase
+          requestContext.set('researchPhase', researchPhaseHeader ?? 'initial')
+
+          // runtime API key (for tools that may accept runtimeContext.apiKey)
+        //  if (apiKeyHeader !== null && apiKeyHeader !== '') { requestContext.set('apiKey', apiKeyHeader) }
+        }
+
+        await next()
+      },
+      // Request timing logger
+      async (c, next) => {
+        const start = Date.now()
+        await next()
+        const duration = Date.now() - start
+        log.info(`${c.req.method} ${c.req.url} - ${duration}ms`)
+      },
+    ]
     }
 });
 

src/mastra/tools/color-change-tool.ts

@@ -0,0 +1,16 @@
+import { createTool } from "@mastra/client-js";
+import z from "zod";
+
+export function changeBgColor(color: string) {
+  if (typeof window !== "undefined") {
+    document.body.style.setProperty("--background", color);
+  }
+}
+
+export const colorChangeTool = createTool({
+  id: "changeColor",
+  description: "Changes the background color",
+  inputSchema: z.object({
+    color: z.string(),
+  }),
+});