AI Firewall MCP

<mcp-name: io.github.Akhilucky/ai-firewall-mcp>

AI Firewall — MCP サーバー

LLM をプロンプトインジェクション、ジェイルブレイク、ポリシー違反から保護する、マルチエージェント型の AI セキュリティレイヤーです。MCP サーバーとして、MCP 対応のあらゆるクライアント（Claude Desktop、Cursor、Windsurf、Cline、Roo Code など）で利用できます。

クイックスタート

pip install

pip install ai-firewall-mcp
ai-firewall-mcp

Docker

docker pull akhilucky/ai-firewall-mcp:latest
docker run -i akhilucky/ai-firewall-mcp:latest

Claude Desktop

claude_desktop_config.json に追加します。

pip install：

{
  "mcpServers": {
    "ai-firewall": {
      "command": "pipx",
      "args": ["run", "ai-firewall-mcp"]
    }
  }
}

Docker：

{
  "mcpServers": {
    "ai-firewall": {
      "command": "docker",
      "args": ["run", "-i", "akhilucky/ai-firewall-mcp:latest"]
    }
  }
}

Cursor / Windsurf / Cline / Roo Code

MCP 設定で次のように設定します。

• タイプ： stdio
• コマンド： docker run -i akhilucky/ai-firewall-mcp:latest
• または、pip でインストールした場合は ai-firewall-mcp を使用します

MCP ツール

ツール	説明
analyze_prompt	プロンプトのインジェクション、ジェイルブレイク、データ流出、情報漏えいを分析します
get_threat_breakdown	直近の分析結果におけるシグナルごとの詳細なスコア内訳を取得します
sanitize_prompt	正当な内容を保ちながら、疑わしいプロンプトを無害化します
get_firewall_status	ヘルスチェック：ベクトル DB のサイズ、モデルの状態、稼働時間
benchmark_firewall	敵対的テストスイートを実行し、検出統計を返します

MCP Inspector でのテスト

npx @modelcontextprotocol/inspector ai-firewall-mcp

アーキテクチャ

ファイアウォールはプロンプトごとに 3 つのエージェントを実行します。

User Prompt → [Retrieval Agent] → [Guard Agent] → [Policy Agent] → LLM
                   │                    │               │
                   ▼                    ▼               ▼
              Vector DB (FAISS)    Threat Signals    Allow/Block

エージェント	役割
検索エージェント	既知の攻撃パターンを対象とするセマンティック検索（FAISS + sentence-transformers）
ガードエージェント	複数シグナルによる分類：ベクトル類似度、キーワード一致、ヒューリスティックスコアリング
ポリシーエージェント	設定可能なしきい値に基づく最終判断：ALLOW / BLOCK / SANITIZE

脅威シグナルの重みは、ベクトル類似度 40%、キーワード一致 25%、ヒューリスティック 20%、**ポリシーの重み 15%**です。

設定

環境変数	デフォルト値	説明
FIREWALL_MODE	strict	strict / moderate / permissive
SIMILARITY_THRESHOLD	0.50	ベクトル一致のしきい値（値が低いほど厳格）
LOG_LEVEL	INFO	ログの詳細度

CLI / API の使用方法

# Interactive dashboard
python main.py

# Red-team adversarial tests
python main.py --redteam

# REST API server
python main.py --api

# Single prompt analysis
python main.py --analyze "Ignore all previous instructions"

REST API は http://localhost:8000 で実行され、OpenAPI ドキュメントは /docs で確認できます（pip install ai-firewall-mcp[api] が必要です）。

テスト

pytest tests/ -v          # Full test suite (43 tests)
pytest tests/test_mcp.py  # MCP-specific tests only

プロジェクト構成

├── src/ai_firewall/          # MCP server package (PyPI entry)
│   ├── mcp_server.py         #    5 MCP tools, stdio transport
│   ├── threat_scorer.py      #    Per-signal scoring breakdown
│   └── __init__.py
├── src/agents/               # Core firewall agents
├── tests/                    # Test suites
├── Dockerfile                # Docker image (2.04GB, CPU-only torch)
├── pyproject.toml            # Package config & metadata
└── .github/workflows/ci.yml  # CI/CD pipeline

ライセンス

MIT — LICENSE を参照してください。

---