Content
<p align="center">
<h1 align="center"><img vertical-align=“middle” width="400px" src="img/logo-full-new.png" alt="A.I.G"/></h1>
</p>
<h4 align="center">
<p>
<a href="https://tencent.github.io/AI-Infra-Guard/">Documentation</a> |
<a href="./README_ZH.md">中文</a> |
<a href="#">English</a>
<p>
</h4>
<p align="center">
<a href="https://github.com/Tencent/AI-Infra-Guard">
<img alt="GitHub Stars" src="https://img.shields.io/github/stars/Tencent/AI-Infra-Guard?color=gold">
</a>
<a href="https://github.com/Tencent/AI-Infra-Guard">
<img alt="GitHub downloads" src="https://img.shields.io/github/downloads/Tencent/AI-Infra-Guard/total">
</a>
<a href="https://github.com/Tencent/AI-Infra-Guard">
<img alt="docker pulls" src="https://img.shields.io/docker/pulls/zhuquelab/aig-server.svg?color=gold">
</a>
<a href="https://github.com/Tencent/AI-Infra-Guard">
<img alt="Release" src="https://img.shields.io/github/v/release/Tencent/AI-Infra-Guard?color=green">
</a>
<a href="https://deepwiki.com/Tencent/AI-Infra-Guard">
<img src="https://deepwiki.com/badge.svg" alt="Ask DeepWiki">
</a>
</p>
<p align="center">
<a href="https://trendshift.io/repositories/13637" target="_blank"><img src="https://trendshift.io/api/badge/repositories/13637" alt="Tencent%2FAI-Infra-Guard | Trendshift" style="width: 250px; height: 55px;" width="250" height="55"/></a>
<a href="https://www.blackhat.com/eu-25/arsenal/schedule/index.html#aigai-infra-guard-48381" target="_blank"><img src="img/blackhat.png" alt="Tencent%2FAI-Infra-Guard | blackhat" style="width: 175px; height: 55px;" width="175" height="55"/></a>
<a href="https://github.com/deepseek-ai/awesome-deepseek-integration" target="_blank"><img src="img/awesome-deepseek.png" alt="Tencent%2FAI-Infra-Guard | awesome-deepseek-integration" style="width: 273px; height: 55px;" width="273" height="55"/></a>
</p>
<br>
<p align="center">
<h2 align="center">🚀 AI Red Teaming Platform by Tencent Zhuque Lab</h2>
</p>
**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.
<p>
We are committed to making A.I.G(AI-Infra-Guard) the industry-leading AI red teaming platform. More stars help this project reach a wider audience, attracting more developers to contribute, which accelerates iteration and improvement. Your star is crucial to us!
</p>
<p align="center">
<a href="https://github.com/Tencent/AI-Infra-Guard">
<img src="https://img.shields.io/badge/⭐-Give%20us%20a%20Star-yellow?style=for-the-badge&logo=github" alt="Give us a Star">
</a>
</p>
## Table of Contents
- [✨ Features](#-features)
- [🖼️ Showcase](#-showcase)
- [🚀 Quick Start](#-quick-start)
- [📖 User Guide](#-user-guide)
- [📝 Contribution Guide](#-contribution-guide)
- [🙏 Acknowledgements](#-acknowledgements)
- [💬 Join the Community](#-join-the-community)
- [📖 Citation](#-citation)
- [📚 Related Papers](#-related-papers)
- [📄 License](#-license)
## ✨ Features
| Feature | More Info |
|:--------|:------------|
| **AI Infra Scan** | Precisely identifies over 30 AI framework components and covers nearly 400 known CVE vulnerabilities, including Ollama, ComfyUI, vLLM, etc. |
| **MCP Server Scan** | 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. |
| **Easy-to-use Web Interface** | Modern, user-friendly web UI for seamless operation, One-click scanning with real-time progress tracking, Comprehensive Analysis Reports. |
| **Multi-Language Support** | Chinese and English interface, Localized documentation and help. |
| **Cross-Platform Compatibility** | Linux, macOS, and Windows support, Docker-based deployment. |
| **Free & Open Source** | Offered completely free of charge under the MIT license. |
<br />
## 🖼️ Showcase
### A.I.G Main Interface
### AI Infra Scan
### MCP Server Scan
### Jailbreak Evaluation
### Plugin Management
<br />
## 🚀 Quick Start
### Deployment with Docker
**System Requirements**
| Docker | RAM | Disk Space |
|:-------|:----|:----------|
| 20.10 or higher | 4GB+ | 10GB+ |
**1. One-Click Install Script （Recommended）**
```bash
# 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)**
```bash
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**
```bash
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`
<br>
<br>
## 📖 User Guide
Visit our online documentation: [https://tencent.github.io/AI-Infra-Guard/](https://tencent.github.io/AI-Infra-Guard/)
For more detailed FAQs and troubleshooting guides, visit our [documentation](https://tencent.github.io/AI-Infra-Guard/?menu=faq).
<br />
<br>
## 📝 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](https://github.com/Tencent/AI-Infra-Guard/issues)
- 💡 [Suggest a New Feature](https://github.com/Tencent/AI-Infra-Guard/issues)
- ⭐ [Improve Documentation](https://github.com/Tencent/AI-Infra-Guard/pulls)
<br />
<br />
## 🙏 Acknowledgements
### 👥 Gratitude to Contributing Developers
Thanks to all the developers who have contributed to the A.I.G project, Your contributions have been instrumental in making A.I.G a more robust and reliable AI Red Team platform.
<br />
<table style="border: none; border-collapse: inherit;">
<tr>
<td width="33%" style="border: none;"><img src="img/keen_lab_logo.svg" alt="Keen Lab" height="85%"></td>
<td width="33%" style="border: none;"><img src="img/wechat_security.png" alt="WeChat Security" height="85%"></td>
<td width="33%" style="border: none;"><img src="img/fit_sec_logo.png" alt="Fit Security" height="85%"></td>
</tr>
</table>
<a href="https://github.com/Tencent/AI-Infra-Guard/graphs/contributors">
<img src="https://contrib.rocks/image?repo=Tencent/AI-Infra-Guard" />
</a>
<br>
<br>
### 🤝 Appreciation for Our Users
We are deeply grateful to the following teams and organizations for their trust, and valuable feedback in using A.I.G.
<br>
<br>
<div align="center">
<img src="img/tencent.png" alt="Tencent" height="30px">
<img src="img/deepseek.png" alt="DeepSeek" height="38px">
</div>
<br>
<br>
### 🌟 Thanks to Our Stargazers!
We are deeply grateful to all the developers who have starred our repository!
<div align="center">
<img src="img/stargazers.png" alt="Stargazers" width="100%">
</div>
Thank you to users from <strong>Google, Microsoft, Amazon, ByteDance, Alibaba, Huawei, Meituan, Douban, Peking University, Tsinghua University, HFUT, cuit</strong>, and many more amazing stargazers!
<br>
<br>
<p align="center">
⭐ Every star encourages us to keep improving and innovating! ⭐
</p>
<p align="center">
🚀 Help us reach more developers by starring this repository. 🚀
</p>
<p align="center">
<a href="https://github.com/Tencent/AI-Infra-Guard">
<img src="https://img.shields.io/badge/⭐-Give%20us%20a%20Star-yellow?style=for-the-badge&logo=github" alt="Give us a Star">
</a>
</p>
<br>
## 💬 Join the Community
### 🌐 Online Discussions
- **GitHub Discussions**: [Join our community discussions](https://github.com/Tencent/AI-Infra-Guard/discussions)
- **Issues & Bug Reports**: [Report issues or suggest features](https://github.com/Tencent/AI-Infra-Guard/issues)
### 📱 Discussion Community
<table>
<thead>
<tr>
<th>WeChat Group</th>
<th>Discord <a href="https://discord.gg/U9dnPnyadZ">[link]</a></th>
</tr>
</thead>
<tbody>
<tr>
<td><img src="img/wechatgroup.png" alt="WeChat Group" width="200"></td>
<td><img src="img/discord.png" alt="discord" width="200"></td>
</tr>
</tbody>
</table>
### 📧 Contact Us
For collaboration inquiries or feedback, please contact us at: zhuque@tencent.com
<br>
<br>
## 📖 Citation
If you use A.I.G in your research or product, please cite:
```bibtex
@misc{Tencent_AI-Infra-Guard_2025,
author={{Tencent Zhuque Lab}},
title={{AI-Infra-Guard: A Comprehensive, Intelligent, and Easy-to-Use AI Red Teaming Platform}},
year={2025},
howpublished={GitHub repository},
url={https://github.com/Tencent/AI-Infra-Guard}
}
```
<br>
## 📚 Related Papers
We are deeply grateful to the research teams who have used A.I.G in their academic work and contributed to advancing AI security research:
[1] Yongjian Guo, Puzhuo Liu, et al. **"Systematic Analysis of MCP Security."** arXiv preprint arXiv:2508.12538 (2025). [[pdf]](https://arxiv.org/abs/2508.12538)
[2] Zexin Wang, Jingjing Li, et al. **"A Survey on AgentOps: Categorization, Challenges, and Future Directions."** arXiv preprint arXiv:2508.02121 (2025). [[pdf]](https://arxiv.org/abs/2508.02121)
[3] Yixuan Yang, Daoyuan Wu, Yufan Chen. **"MCPSecBench: A Systematic Security Benchmark and Playground for Testing Model Context Protocols."** arXiv preprint arXiv:2508.13220 (2025). [[pdf]](https://arxiv.org/abs/2508.13220)
[4] Ping He, Changjiang Li, et al. **"Automatic Red Teaming LLM-based Agents with Model Context Protocol Tools."** arXiv preprint arXiv:2509.21011 (2025). [[pdf]](https://arxiv.org/abs/2509.21011)
📧 If you have used A.I.G in your research, we would love to hear from you! [Contact us here](#-join-the-community).
<br>
<br>
## 📄 License
This project is licensed under the **MIT License**. See the [License.txt](./License.txt) file for details.
<div>
[](https://star-history.com/#Tencent/AI-Infra-Guard&Date)
