入力プロセッサ

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', // Automatically redact detected PII
      detectionTypes: ['email', 'phone', 'credit-card', 'ssn', 'api-key', 'crypto-wallet', 'iban'],
      redactionMethod: 'mask', // Preserve format while masking
      preserveFormat: true, // Keep original structure in redacted values
      includeDetections: true, // Log details for compliance auditing
    }),
  ],
});

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 } from "@mastra/core/processors";
import type { MastraMessageV2 } from "@mastra/core/agent/message-list";
import { TripWire } from "@mastra/core/agent";
 
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(`メッセージが長すぎます: ${totalLength} 文字（上限: ${this.maxLength}）`); // TripWire エラーを投げる
      }
    } catch (error) {
      if (error instanceof TripWire) {
        throw error; // TripWire エラーは投げ直す
      }
      throw new Error(`長さの検証に失敗しました: ${error instanceof Error ? error.message : '不明なエラー'}`); // アプリケーションレベルでは標準のエラーを投げる
    }
 
    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);
}