Documentation | 中文 | English

🚀 AI Red Teaming Platform by Tencent Zhuque Lab

A.I.G (AI-Infra-Guard) integrates capabilities such as AI infra vulnerability scan, MCP Server risk scan, and Jailbreak Evaluation, aiming to provide users with the most comprehensive, intelligent, and user-friendly solution for AI security risk self-examination.

Our project has been featured in 🔗Awesome DeepSeek Integrations.

Table of Contents

• ✨ Features
• 🖼️ Showcase
• 🚀 Quick Start
• 📖 User Guide
• 📝 Contribution Guide
• 🙏 Acknowledgements
• 💬 Join the Community
• 📄 License
• 📖 Citation
• 📚 Papers

✨ Features

🔍 Detect AI Infra Risk

• Precisely identifies 30+ AI framework components
• Covers nearly 400 known CVE vulnerabilities
• Including Ollama, ComfyUI, vLLM, etc.

🤖 Detect MCP Server Risk

• Powered by AI Agent
• Detects 9 major categories of MCP security risks
• Supports source code/remote URL scanning

⚡ Jailbreak Evaluation

• Rapidly assesses Prompt security risks
• Includes multiple curated jailbreak evaluation datasets
• Cross-model security performance comparison

📊 Detailed Report Analysis

• Comprehensive security assessment reports
• Shareable analysis with detailed vulnerability insights

💰 Free & Open Source with MIT license

• Completely free to use
• Open source with MIT license

🌍 Multi-Language Support

• 🇨🇳 Chinese and 🇺🇸 English interface
• Localized documentation and help

🖥️ Cross-Platform Compatibility

• 🐧 Linux, 🍎 macOS, and 🪟 Windows support
• Docker-based deployment

🖼️ Showcase（待增加GIF与演示视频外链）

A.I.G Main Interface

AI Infra Scan

MCP Scan

Jailbreak Evaluation

Plugin Management

🚀 Quick Start

Deployment with Docker

System Requirements

• Docker 20.10 or higher
• At least 4GB of available RAM
• At least 10GB of available disk space

1. One-Click Install Script （Recommended）

# This method will automatically install Docker and launch A.I.G with one command  
curl https://raw.githubusercontent.com/Tencent/AI-Infra-Guard/refs/heads/main/docker.sh | bash

2. Run with pre-built images (Recommended)

git clone https://github.com/Tencent/AI-Infra-Guard.git
cd AI-Infra-Guard
# This method pulls pre-built images from Docker Hub for a faster start
docker-compose -f docker-compose.images.yml up -d

3. Build from source and run

git clone https://github.com/Tencent/AI-Infra-Guard.git
cd AI-Infra-Guard
# This method builds a Docker image from local source code and starts the service
docker-compose up -d

Once the service is running, you can access the A.I.G web interface at: http://localhost:8088

📖 User Guide

Visit our online documentation for detailed documentation: https://tencent.github.io/AI-Infra-Guard/

For more detailed FAQs and troubleshooting guides, visit our documentation.（常见问题文档补充中）

📝 Contribution Guide

The extensible plugin framework​​ serves as A.I.G's architectural cornerstone, inviting community innovation through Plugin and Feature contributions.​

Plugin Contribution Rules

1. Fingerprint Rules: Add new YAML fingerprint files to the data/fingerprints/ directory.
2. Vulnerability Rules: Add new vulnerability scan rules to the data/vuln/ directory.
3. MCP Plugins: Add new MCP security scan rules to the data/mcp/ directory.
4. Jailbreak Evaluation Datasets: Add new Jailbreak evaluation datasets to the data/eval directory.

Please refer to the existing rule formats, create new files, and submit them via a Pull Request.

Other Ways to Contribute

• 🐛 Report a Bug
• 💡 Suggest a New Feature
• ⭐ Improve Documentation

🙏 Acknowledgements

Thanks to all the developers who have contributed to the A.I.G project:

We are deeply grateful to the following teams and organizations for their trust, and valuable feedback in using A.I.G. Your contributions have been instrumental in making A.I.G a more robust and reliable AI Red Team platform.
(待确定后改为logo)

• Tencent Zhuque Lab
• Tencent Keen Lab
• Tencent WeChat Security
• Tencent FIT Security
• DeepSeek

📊 User Registration

If you are using A.I.G, please let us know! Your usage is very important to us (new registrations will be prioritized in order, with dedicated support for priority Q&A and Pro Version): 💬 Submit Usage Registration

💬 Join the Community

🌐 Online Discussions

• GitHub Discussions: Join our community discussions
• Issues & Bug Reports: Report issues or suggest features

📱 WeChat Community

Scan the WeChat QR code to join the A.I.G community group

📧 Contact Us

For collaboration inquiries or feedback, please contact us at: zhuque@tencent.com

📄 License

This project is licensed under the MIT License. See the License.txt file for details.

🌟 Thank You to Our Stargazers!

We are deeply grateful to all the developers who have starred our repository!

Thank you to Peking University, Alibaba, Tsinghua University, ByteDance, Microsoft, Amazon, Huawei, Meituan, Douban, HFUT, cuit, and many more amazing users for your stars!

⭐ Every star encourages us to keep improving and innovating! ⭐

🚀 Help us reach more developers by starring this repository. 🚀

📖 Citation

If you use A.I.G in your research or product, please cite:

@misc{tencent2025aig,
  title={A.I.G},
  author={tencent},
  year={2025},
  url={https://github.com/Tencent/AI-Infra-Guard}
}

📚 Papers

1. Systematic Analysis of MCP Security - https://arxiv.org/abs/2508.12538
2. A Survey on AgentOps: Categorization, Challenges, and Future Directions - https://arxiv.org/pdf/2508.02121
3. MCPSecBench: A Systematic Security Benchmark and Playground for Testing Model Context Protocols - https://arxiv.org/pdf/2508.13220