Input Processors

import { Agent } from "@mastra/core/agent";
import { UnicodeNormalizer } from "@mastra/core/processors";
import { openai } from "@ai-sdk/openai";
 
const agent = new Agent({
  name: 'normalized-agent',
  instructions: 'You are a helpful assistant',
  model: openai("gpt-4o"),
  inputProcessors: [
    new UnicodeNormalizer({
      stripControlChars: true,
      collapseWhitespace: true,
    }),
  ],
});

import { ModerationProcessor } from "@mastra/core/processors";
 
const agent = new Agent({
  inputProcessors: [
    new ModerationProcessor({
      model: openai("gpt-4.1-nano"), // 高速でコスト効率の高いモデルを使用
      threshold: 0.7, // フラグ付けの信頼度しきい値
      strategy: 'block', // フラグ付けされたコンテンツをブロック
      categories: ['hate', 'harassment', 'violence'], // カスタムカテゴリー
    }),
  ],
});

import { PromptInjectionDetector } from "@mastra/core/processors";
 
const agent = new Agent({
  inputProcessors: [
    new PromptInjectionDetector({
      model: openai("gpt-4.1-nano"),
      threshold: 0.8, // 偽陽性を減らすため高めのしきい値
      strategy: 'rewrite', // 意図を保ちながら無害化を試みる
      detectionTypes: ['injection', 'jailbreak', 'system-override'],
    }),
  ],
});

import { PIIDetector } from "@mastra/core/processors";
 
const agent = new Agent({
  inputProcessors: [
    new PIIDetector({
      model: openai("gpt-4.1-nano"),
      threshold: 0.6,
      strategy: 'redact', // 検出された PII を自動的に編集
      detectionTypes: ['email', 'phone', 'credit-card', 'ssn', 'api-key', 'crypto-wallet', 'iban'],
      redactionMethod: 'mask', // 形式を維持しつつマスク
      preserveFormat: true, // マスク後も元の構造を保持
      includeDetections: true, // コンプライアンス監査用に詳細を記録
    }),
  ],
});

import { LanguageDetector } from "@mastra/core/processors";
 
const agent = new Agent({
  inputProcessors: [
    new LanguageDetector({
      model: openai("gpt-4o-mini"),
      targetLanguages: ['English', 'en'], // 英語コンテンツを許可
      strategy: 'translate', // 非英語コンテンツを自動翻訳
      threshold: 0.8, // 高い信頼度しきい値
    }),
  ],
});

import { Agent } from "@mastra/core/agent";
import { 
  UnicodeNormalizer, 
  ModerationProcessor, 
  PromptInjectionDetector,
  PIIDetector 
} from "@mastra/core/processors";
 
const secureAgent = new Agent({
  inputProcessors: [
    // 1. まずテキストを正規化
    new UnicodeNormalizer({ stripControlChars: true }),
    // 2. セキュリティ上の脅威をチェック
    new PromptInjectionDetector({ model: openai("gpt-4.1-nano") }),
    // 3. コンテンツをモデレーション
    new ModerationProcessor({ model: openai("gpt-4.1-nano") }),
    // 4. 最後に個人情報（PII）を処理
    new PIIDetector({ model: openai("gpt-4.1-nano"), strategy: 'redact' }),
  ],
});

import type { Processor, MastraMessageV2, TripWire } from "@mastra/core/processors";
 
class MessageLengthLimiter implements Processor {
  readonly name = 'message-length-limiter';
  
  constructor(private maxLength: number = 1000) {}
 
  processInput({ messages, abort }: { 
    messages: MastraMessageV2[]; 
    abort: (reason?: string) => never 
  }): MastraMessageV2[] {
    // メッセージの合計文字数をチェック
    try {
      const totalLength = messages.reduce((sum, msg) => {
        return sum + msg.content.parts
          .filter(part => part.type === 'text')
          .reduce((partSum, part) => partSum + (part as any).text.length, 0);
      }, 0);
      
      if (totalLength > this.maxLength) {
        abort(`Message too long: ${totalLength} characters (max: ${this.maxLength})`); // TripWire エラーを送出
      }
    } catch (error) {
      if (error instanceof TripWire) {
        throw error; // TripWire エラーは再送出
      }
      throw new Error(`Length validation failed: ${error instanceof Error ? error.message : 'Unknown error'}`); // アプリケーションレベルでは標準的なエラーを送出
    }
    
    return messages;
  }
}
 
// カスタムプロセッサを使用する
const agent = new Agent({
  inputProcessors: [
    new MessageLengthLimiter(2000), // 2000 文字に制限
  ],
});

// Processors run before generate()
const result = await agent.generate('Hello');
 
// Processors also run before streamVNext()
const stream = await agent.streamVNext('Hello');
for await (const chunk of stream) {
  console.log(chunk);
}