feat: add TokenLimiterProcessor to various agents and update README

- Integrated TokenLimiterProcessor into multiple agents to manage output token limits effectively.
- Updated README.md with new Wakatime badge link.
- Adjusted sampling rates for structure and creativity scorers in Content Strategist Agent.
- Enhanced Upstash configuration for improved vector search and memory management.

Author: ssdeanx

README.md

@@ -29,7 +29,7 @@
 <!-- Repository Links -->
 [![GitHub](https://img.shields.io/badge/GitHub-ssdeanx/AgentStack-181717?logo=github)](https://github.com/ssdeanx/AgentStack)
 [![GitMCP](https://img.shields.io/endpoint?url=https://gitmcp.io/badge/ssdeanx/AgentStack)](https://gitmcp.io/ssdeanx/AgentStack)
-[![wakatime](https://wakatime.com/badge/user/7a2fb9a0-188b-4568-887f-7645f9249e62/project/e44412f3-9bcc-4661-b79d-23160d90dfe0.svg)](https://wakatime.com/badge/user/7a2fb9a0-188b-4568-887f-7645f9249e62/project/e44412f3-9bcc-4661-b79d-23160d90dfe0)
+[![wakatime](https://wakatime.com/badge/user/7a2fb9a0-188b-4568-887f-7645f9249e62/project/e52d02a1-f64a-4f8d-bc13-caaa2dc37461.svg)](https://wakatime.com/badge/user/7a2fb9a0-188b-4568-887f-7645f9249e62/project/e52d02a1-f64a-4f8d-bc13-caaa2dc37461)
 
 **AgentStack** is a **production-grade multi-agent framework** built on Mastra, delivering **60+ enterprise tools**, **30+ specialized agents**, **14 workflows**, **4 agent networks**, **65 UI components** (30+ AI Elements + 35+ base), and **A2A/MCP orchestration** for scalable AI systems. Focuses on **financial intelligence**, **RAG pipelines**, **observability**, **secure governance**, and **AI chat interfaces**.
 
@@ -71,7 +71,7 @@
 ## ✨ **Core Capabilities**
 
 - **💰 Financial Intelligence**: 30+ tools (Polygon quotes/aggs/fundamentals, Finnhub analysis, AlphaVantage indicators)
-- **🔍 Semantic RAG**: PgVector (3072D embeddings) + MDocument chunking + rerank + graph traversal  
+- **🔍 Semantic RAG**: PgVector (3072D embeddings) + MDocument chunking + rerank + graph traversal
 - **🤖 30+ Agents**: Research → Learn → Report → Edit → Analyze (stock/crypto/copywriter/evaluator/data pipeline/business-legal/charting/image/coding/dane)
 - **📋 14 Workflows**: Weather, content, financial reports, document processing, research synthesis, learning extraction, governed RAG (index + answer), spec generation, repo ingestion, stock analysis
 - **🌐 4 Agent Networks**: Coding team, data pipeline, report generation, research pipeline

src/mastra/agents/acpAgent.ts

@@ -19,6 +19,7 @@ import { batchWebScraperTool, contentCleanerTool, htmlToMarkdownTool, linkExtrac
 import type { GoogleGenerativeAIProviderOptions } from '@ai-sdk/google';
 import type { RequestContext } from '@mastra/core/request-context';
 import { PGVECTOR_PROMPT } from '@mastra/pg';
+import { TokenLimiterProcessor } from '@mastra/core/processors';
 
 export interface ACPContext {
     userId?: string
@@ -129,7 +130,7 @@ export const acpAgent = new Agent({
     //		containerTags: ['acp-agent']
     //	}),
   },
-
+  outputProcessors: [new TokenLimiterProcessor(1048576)],
   workflows: {},
   scorers: {
     relevancy: {

src/mastra/agents/businessLegalAgents.ts

@@ -4,7 +4,7 @@ import { GoogleGenerativeAIProviderMetadata } from '@ai-sdk/google';
 import { googleTools } from '@ai-sdk/google/internal';
 import { Agent } from '@mastra/core/agent';
 
-import { BatchPartsProcessor, UnicodeNormalizer } from '@mastra/core/processors';
+import { BatchPartsProcessor, TokenLimiterProcessor, UnicodeNormalizer } from '@mastra/core/processors';
 import type { RequestContext } from '@mastra/core/request-context';
 import {
   createAnswerRelevancyScorer,
@@ -156,7 +156,8 @@ ${PGVECTOR_PROMPT}
       stripControlChars: true,
       collapseWhitespace: true,
     }),
-  ]
+  ],
+  outputProcessors: [new TokenLimiterProcessor(1048576)]
 })
 
 export const contractAnalysisAgent = new Agent({
@@ -239,7 +240,8 @@ You are a Senior Contract Analyst. Analyze legal documents for risks, obligation
       stripControlChars: true,
       collapseWhitespace: true,
     }),
-  ]
+  ],
+  outputProcessors: [new TokenLimiterProcessor(1048576)]
 })
 
 export const complianceMonitoringAgent = new Agent({
@@ -325,7 +327,8 @@ You are a Compliance Officer. Monitor regulatory compliance and identify risks a
       stripControlChars: true,
       collapseWhitespace: true,
     }),
-  ]
+  ],
+  outputProcessors: [new TokenLimiterProcessor(1048576)]
 })
 
 export const businessStrategyAgent = new Agent({
@@ -406,7 +409,8 @@ You are a Chief Strategy Officer with legal expertise. Align business strategy w
       stripControlChars: true,
       collapseWhitespace: true,
     }),
-  ]
+  ],
+  outputProcessors: [new TokenLimiterProcessor(1048576)]
 })
 
 

src/mastra/agents/calendarAgent.ts

@@ -3,6 +3,7 @@ import { listEvents, getTodayEvents, getUpcomingEvents, findFreeSlots } from '..
 import { googleAIFlashLite, pgMemory } from '../config';
 import type { GoogleGenerativeAIProviderOptions } from '@ai-sdk/google';
 import type { RequestContext } from '@mastra/core/request-context';
+import { TokenLimiterProcessor } from '@mastra/core/processors';
 
 export interface CalendarContext {
     userId?: string
@@ -54,5 +55,5 @@ Current user: ${userId ?? 'anonymous'}`,
     getUpcomingEvents,
     findFreeSlots,
   },
-
+  outputProcessors: [new TokenLimiterProcessor(1048576)]
 });

src/mastra/agents/codingAgents.ts

@@ -1,6 +1,6 @@
 import { Agent } from '@mastra/core/agent'
 
-import { BatchPartsProcessor, UnicodeNormalizer } from '@mastra/core/processors'
+import { BatchPartsProcessor, TokenLimiterProcessor, UnicodeNormalizer } from '@mastra/core/processors'
 import type { RequestContext } from '@mastra/core/request-context'
 import { createAnswerRelevancyScorer, createToxicityScorer } from '@mastra/evals/scorers/prebuilt'
 import type { GoogleGenerativeAIProviderOptions } from '@ai-sdk/google';
@@ -128,8 +128,7 @@ Always consider maintainability, scalability, and testability in your recommenda
   inputProcessors: [
     new UnicodeNormalizer({ stripControlChars: true, collapseWhitespace: true }),
   ],
-  outputProcessors: [
-  ],
+  outputProcessors: [new TokenLimiterProcessor(1048576)]
 })
 
 //log.info('Cached tokens:', providerMetadata.google?.usageMetadata);
@@ -239,6 +238,7 @@ Be constructive and educational in feedback.`,
   inputProcessors: [
     new UnicodeNormalizer({ stripControlChars: true, collapseWhitespace: true }),
   ],
+  outputProcessors: [new TokenLimiterProcessor(1048576)]
 })
 
 /**
@@ -340,7 +340,7 @@ Always use Vitest syntax: describe, it, expect, vi.mock, vi.fn.`,
   inputProcessors: [
     new UnicodeNormalizer({ stripControlChars: true, collapseWhitespace: true }),
   ],
-
+  outputProcessors: [new TokenLimiterProcessor(1048576)]
 })
 
 /**
@@ -448,7 +448,7 @@ For each refactoring:
   inputProcessors: [
     new UnicodeNormalizer({ stripControlChars: true, collapseWhitespace: true }),
   ],
-
+  outputProcessors: [new TokenLimiterProcessor(1048576)]
 })
 
 log.info('Coding Team Agents initialized: codeArchitectAgent, codeReviewerAgent, testEngineerAgent, refactoringAgent')

src/mastra/agents/contentStrategistAgent.ts

@@ -6,6 +6,7 @@ import { structureScorer, creativityScorer } from '../scorers';
 import { chartSupervisorTool } from '../tools/financial-chart-tools';
 import type { GoogleGenerativeAIProviderOptions } from '@ai-sdk/google';
 import type { RequestContext } from '@mastra/core/request-context'
+import { TokenLimiterProcessor } from '@mastra/core/processors';
 
 export type UserTier = 'free' | 'pro' | 'enterprise'
 export interface ContentAgentContext {
@@ -99,11 +100,12 @@ EXAMPLE FLOW:
   scorers: {
     structure: {
       scorer: structureScorer,
-      sampling: { type: 'ratio', rate: 1.0 },
+      sampling: { type: 'ratio', rate: 0.3 },
     },
     creativity: {
       scorer: creativityScorer,
-      sampling: { type: 'ratio', rate: 1.0 },
+      sampling: { type: 'ratio', rate: 0.3 },
     },
-  }
+  },
+  outputProcessors: [new TokenLimiterProcessor(1048576)]
 });

src/mastra/agents/copywriterAgent.ts

@@ -14,6 +14,7 @@ import { structureScorer } from '../scorers'
 import { chartSupervisorTool } from '../tools/financial-chart-tools'
 import type { GoogleGenerativeAIProviderOptions } from '@ai-sdk/google'
 import type { RequestContext } from '@mastra/core/request-context'
+import { TokenLimiterProcessor } from '@mastra/core/processors'
 
 // Define runtime context for this agent
 export interface CopywriterAgentContext {
@@ -142,6 +143,7 @@ Provide the final content in a clear, well-structured format appropriate for the
         },
     },
     workflows: {},
-    maxRetries: 5
+    maxRetries: 5,
+    outputProcessors: [new TokenLimiterProcessor(1048576)]
 })
 

src/mastra/agents/csv_to_excalidraw.ts

@@ -3,6 +3,7 @@ import { googleAI, pgMemory } from "../config";
 import { structureScorer } from "../scorers";
 import type { GoogleGenerativeAIProviderOptions } from "@ai-sdk/google";
 import type { RequestContext } from '@mastra/core/request-context'
+import { TokenLimiterProcessor } from "@mastra/core/processors";
 
 export interface CsvToExcalidrawRuntimeContext {
     userId?: string
@@ -182,5 +183,6 @@ Structure:
     },
   },
   workflows: {},
-  maxRetries: 5
+  maxRetries: 5,
+  outputProcessors: [new TokenLimiterProcessor(1048576)]
 });

src/mastra/agents/dane.ts

@@ -5,6 +5,7 @@ import { listEvents } from '../tools/calendar-tool'
 import { execaTool } from '../tools/execa-tool'
 import type { GoogleGenerativeAIProviderOptions } from '@ai-sdk/google'
 import type { RequestContext } from '@mastra/core/request-context'
+import { TokenLimiterProcessor } from '@mastra/core/processors'
 
 export interface DaneContext {
     userId?: string
@@ -39,6 +40,7 @@ export const daneCommitMessage = new Agent({
   },
   model: googleAIFlashLite,
   memory: pgMemory,
+  outputProcessors: [new TokenLimiterProcessor(1048576)]
 })
 
 export const daneIssueLabeler = new Agent({
@@ -67,6 +69,7 @@ export const daneIssueLabeler = new Agent({
   },
   model: googleAIFlashLite,
   memory: pgMemory,
+  outputProcessors: [new TokenLimiterProcessor(1048576)]
 })
 
 export const daneLinkChecker = new Agent({
@@ -100,6 +103,7 @@ export const daneLinkChecker = new Agent({
   },
   model: googleAIFlashLite,
   memory: pgMemory,
+  outputProcessors: [new TokenLimiterProcessor(1048576)]
 })
 
 export const daneChangeLog = new Agent({
@@ -135,6 +139,7 @@ export const daneChangeLog = new Agent({
   },
   model: googleAIFlashLite,
   memory: pgMemory,
+  outputProcessors: [new TokenLimiterProcessor(1048576)]
 })
 
 export const dane = new Agent({
@@ -201,4 +206,5 @@ export const dane = new Agent({
     googleSearch,
     listEvents,
   },
+  outputProcessors: [new TokenLimiterProcessor(1048576)]
 })

src/mastra/agents/dataExportAgent.ts

@@ -12,6 +12,7 @@ import {
     listDataDirTool,
 } from '../tools/data-file-manager'
 import type { RequestContext } from '@mastra/core/request-context'
+import { TokenLimiterProcessor } from '@mastra/core/processors'
 
 export interface DataExportContext {
     userId?: string
@@ -85,6 +86,7 @@ export const dataExportAgent = new Agent({
         backupDataTool,
         listDataDirTool,
     },
+    outputProcessors: [new TokenLimiterProcessor(1048576)]
 })
 
 log.info('Data Export Agent initialized')