AI Infra Guard MCP Server

<p align="center"> <h1 align="center"><img vertical-align="middle" width="400px" src="img/logo-full-new.png" alt="A.I.G"/></h1> </p> <p align="center"> <a href="https://tencent.github.io/AI-Infra-Guard/">📖 Documentation</a> &nbsp;|&nbsp; 🌐 <b>🇬🇧 English</b> · <a href="./readme/README_ZH.md">🇨🇳 中文</a> </p> <p align="center"> <a href="https://github.com/tencent/AI-Infra-Guard/stargazers"> <img src="https://img.shields.io/github/stars/tencent/AI-Infra-Guard.svg" alt="GitHub stars"> </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> </p> <p align="center"> <a href="https://clawhub.ai/aigsec/edgeone-clawscan" target="_blank"> <img src="https://img.shields.io/badge/ClawHub-EdgeOne%20ClawScan-a870dc" alt="EdgeOne ClawScan"> </a> <a href="https://clawhub.ai/aigsec/edgeone-skill-scanner" target="_blank"> <img src="https://img.shields.io/badge/ClawHub-EdgeOne%20Skill%20Scanner-2ea44f" alt="EdgeOne Skill Scanner"> </a> <a href="https://clawhub.ai/aigsec/aig-scanner" target="_blank"> <img src="https://img.shields.io/badge/ClawHub-AIG%20Scanner-e6a817" alt="AIG Scanner"> </a> </p> <p align="center"> <a href="https://github.com/openclaw/clawscan" target="_blank"> <img src="https://img.shields.io/badge/OpenClaw-Recommended-ff6b6b" alt="OpenClaw Recommended"> </a> </p> <p align="center"> <a href="https://trendshift.io/repositories/13637" target="_blank"><picture><source media="(prefers-color-scheme: dark)" srcset="https://trendshift.io/api/badge/repositories/13637"><source media="(prefers-color-scheme: light)" srcset="https://trendshift.io/api/badge/repositories/13637"><img src="https://trendshift.io/api/badge/repositories/13637" alt="Tencent%2FAI-Infra-Guard | Trendshift" width="250" height="55"/></picture></a>&nbsp; <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" width="175" height="55"/></a>&nbsp; <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" width="273" height="55"/></a> </p> <br> <p align="center"> <h3 align="center">🚀 One-Stop AI Red Teaming Platform by Tencent Zhuque Lab</h3> </p>

<b>A.I.G (AI-Infra-Guard)</b> brings together OpenClaw security checks, Agent security scanning, AI tool and skill scanning, AI infrastructure vulnerability scanning, and LLM security assessment (jailbreak evaluation), aiming to provide the most comprehensive, intelligent, and easy-to-use way to check AI security risks.

<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> <br>

Table of Contents

• 🚀 Quick Start
• 🧰 Standalone Use
• 🖼️ Showcase
• 📖 User Guide
• 🏗️ Architecture Evolution
• 📝 Contribution Guide
• 🛡️ About the Team
• 🙏 Acknowledgements
• 💬 Join the Community
• 📖 Citation
• 📄 License <br><br>

🚀 Quick Start

🐳 Deploy A.I.G with Docker

Docker	RAM	Disk Space
20.10 or higher	4GB+	10GB+

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

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

Other Installation Methods

Method 2: 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

Method 3: Build and run from source

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
# (For Docker Compose V2+, replace 'docker-compose' with 'docker compose')
docker-compose up -d

Note: The AI-Infra-Guard project is positioned as an AI red teaming platform for internal use by enterprises or individuals. It currently lacks an authentication mechanism and should not be deployed on public networks.

For more information, see: https://tencent.github.io/AI-Infra-Guard/?menu=getting-started

🧰 Standalone Use

The tools below can be used from the command line without starting the web service. Only the shortest usage is shown. See each README for parameters and examples.

AI Infrastructure Scan

Scan a domain, IP, or URL and match fingerprints and vulnerabilities.

pip install aig-infra-scan
aig-infra-scan -t https://example.com

View README

AI Safety Compliance Check

Open a local page to check whether a model refuses unsafe prompts, and optionally scan infrastructure. Evaluation sets live in data/eval. Run from the repository root, or set AIG_EVAL_DIR.

pip install aig-compliance
aig-compliance

View README

LLM Security Assessment / Jailbreak

Evaluate a model that exposes an OpenAI-compatible API.

pip install aig-model-scan
aig-model-scan \
  --model "gpt-4o-mini" \
  --base_url "https://api.openai.com/v1" \
  --api_key "your-api-key" \
  --max_concurrent 10 \
  --scenarios Bias \
  --techniques Raw

View README

MCP Scan

Scan an MCP server source directory, or connect to a service that is already running.

pip install aig-mcp-scan
export LLM_API_KEY="your-api-key"
aig-mcp-scan --repo /path/to/mcp-server -o result.sarif.json

View README

Skill Scan

Audit a local Agent Skill directory.

pip install aig-skill-scan
export LLM_API_KEY="your-api-key"
aig-skill-scan --repo /path/to/your/skill -o result.sarif.json

View README

Agent Scan

Send test prompts to a running Agent. target.yaml describes the Agent under test. The API key is for the model used by the scanner.

pip install aig-agent-scan
export LLM_API_KEY="your-api-key"
aig-agent-scan --agent_provider target.yaml -o result.json

View README

🖼️ Showcase

A.I.G Main Interface

<br />

📖 User Guide

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

For more detailed FAQs and troubleshooting guides, visit our documentation. <br /> <br>

📝 Contribution Guide

One of A.I.G's core strengths is its rich plugin system, which can be configured quickly. We welcome high-quality plugins and features from the community.

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. Model Evaluation Datasets: Add new model 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 <br />

<br />

🛡️ About the Team

This project is led and developed by Tencent Zhuque Lab, part of the Tencent Security Platform Department. Founded in 2019, Tencent Zhuque Lab is a top-tier security research lab focused on real-world offensive and defensive research and frontier technology in the AI security space, covering large model security, AI agent security, AI-empowered security, and AI-generated content detection.

The team has helped major vendors such as NVIDIA, Google, and Microsoft, as well as open-source communities like OpenClaw, Linux, and Hugging Face, fix a large number of high-risk vulnerabilities, and has been publicly acknowledged by them.

We have released open-source AI security products including the AI Red Team Security Testing Platform <b>A.I.G (AI-Infra-Guard)</b> and the Zhuque AI Detection Assistant. Our research has been widely published at top international security and AI conferences such as Black Hat, DEF CON, ICLR, CVPR, NeurIPS, and ACL, and we have authored the book "AI Security: Technology and Practice".

👥 Core Members & Contributions

Role	Member	Contribution
Head of Tencent Security Platform Department	Yong Yang	Initiated A.I.G and proposed automated assessment of AI agent loss-of-control risks, guiding the platform's expansion from AI infrastructure vulnerability scanning to agent execution risk, tool misuse, and permission-boundary evaluation.
Head of Tencent Zhuque Lab	Xing Zheng	Proposed the automated vulnerability-update and benchmark-alignment mechanism, helping AI Infra fingerprints, CVE/GHSA rules, and benchmarks iterate continuously.
Project Lead	Nicky	Frontier security research, product planning, technical-route decisions, internal and external collaboration, and communications.
Technical Lead	Python	Overall architecture design, core module development, and version iteration.
Core Contributor	Zona	Frontend interaction, product experience, community operations, and user-feedback loop.
Core Contributor	Fyoung	AI Infra vulnerability component fingerprint updates and Benchmark system construction.
Core Contributor	Xiangfan	Security capability development for Skill risks and agent loss-of-control scenarios.
Core Contributor	Elwood	Enhancing Agent security scanning capabilities and updating technical reports.
Core Contributor	Robert	LLM safety assessment and jailbreak-evaluation strategy operations.
Core Contributor	Zoe	LLM safety assessment, jailbreak evaluation, and model-integration module development.
Contributor	Ronin	Participated in AI agent security scanning development.
Contributor	Rsin	Participated in community operations and campaign communications.

<br />

🙏 Acknowledgements

🎓 Academic Collaborations

We thank our academic partners for their research contributions and technical support.

<img src="img/北大未来网络重点实验室2.png" height="30" align="middle"/>

<table> <tr> <td align="center" width="90"> <a href="#"> <img src="https://avatars.githubusercontent.com/u/0?v=4" width="70px;" style="border-radius: 50%;" alt=""/> </a> <br /> <a href="#"> _{<b>Prof.&nbsp;hui&nbsp;Li</b>} </a> </td> <td align="center" width="90"> <a href="https://github.com/TheBinKing"> <img src="https://avatars.githubusercontent.com/TheBinKing" width="70px;" style="border-radius: 50%;" alt=""/> </a> <br /> <a href="mailto:1546697086@qq.com"> _{<b>Bin&nbsp;Wang</b>} </a> </td> <td align="center" width="90"> <a href="https://github.com/KPGhat"> <img src="https://avatars.githubusercontent.com/KPGhat" width="70px;" style="border-radius: 50%;" alt=""/> </a> <br /> <a href="mailto:kpghat@gmail.com"> _{<b>Zexin&nbsp;Liu</b>} </a> </td> <td align="center" width="90"> <a href="https://github.com/GioldDiorld"> <img src="https://avatars.githubusercontent.com/GioldDiorld" width="70px;" style="border-radius: 50%;" alt=""/> </a> <br /> <a href="mailto:g.diorld@gmail.com"> _{<b>Hao&nbsp;Yu</b>} </a> </td> <td align="center" width="90"> <a href="https://github.com/Jarvisni"> <img src="https://avatars.githubusercontent.com/Jarvisni" width="70px;" style="border-radius: 50%;" alt=""/> </a> <br /> <a href="mailto:719001405@qq.com"> _{<b>Ao&nbsp;Yang</b>} </a> </td> <td align="center" width="90"> <a href="https://github.com/Zhengxi7"> <img src="https://avatars.githubusercontent.com/Zhengxi7" width="70px;" style="border-radius: 50%;" alt=""/> </a> <br /> <a href="mailto:linzhengxi7@126.com"> _{<b>Zhengxi&nbsp;Lin</b>} </a> </td> </tr> </table>

<img src="img/复旦大学2.png" height="30" align="middle" style="vertical-align: middle;"/>

<table> <tr> <td align="center" width="120"> <a href="https://yangzhemin.github.io/"> <img src="https://avatars.githubusercontent.com/yangzhemin" width="70px;" style="border-radius: 50%;" alt=""/> </a> <br /> <a href="mailto:yangzhemin@fudan.edu.cn"> _{<b>Prof.&nbsp;Zhemin&nbsp;Yang</b>} </a> </td> <td align="center" width="100"> <a href="https://github.com/kangwei-zhong"> <img src="https://avatars.githubusercontent.com/kangwei-zhong" width="70px;" style="border-radius: 50%;" alt=""/> </a> <br /> <a href="mailto:kwzhong23@m.fudan.edu.cn"> _{<b>Kangwei&nbsp;Zhong</b>} </a> </td> <td align="center" width="90"> <a href="https://github.com/MoonBirdLin"> <img src="https://avatars.githubusercontent.com/MoonBirdLin" width="70px;" style="border-radius: 50%;" alt=""/> </a> <br /> <a href="mailto:linjp23@m.fudan.edu.cn"> _{<b>Jiapeng&nbsp;Lin</b>} </a> </td> <td align="center" width="90"> <a href="https://vanilla-tiramisu.github.io/"> <img src="https://avatars.githubusercontent.com/vanilla-tiramisu" width="70px;" style="border-radius: 50%;" alt=""/> </a> <br /> <a href="mailto:csheng25@m.fudan.edu.cn"> _{<b>Cheng&nbsp;Sheng</b>} </a> </td> </tr> </table> <br>

👥 Thanks to the following teams and developers

<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>

🤝 Thanks to the following organizations and teams for using and supporting A.I.G

<br> <div align="center"> <img src="img/tencent.png" alt="Tencent" height="28px"> <img src="img/deepseek.png" alt="DeepSeek" height="38px"> <img src="img/antintl.svg" alt="Antintl" height="45px"> <img src="img/lenovo.png" alt="Lenovo" height="35px"> <img src="img/ICBC.jpg" alt="ICBC" height="40px"> <img src="img/vivo.png" alt="Vivo" height="30px"> <img src="img/oppo.png" alt="Oppo" height="30px"> <img src="img/haier.png" alt="Haier" height="30px"> <img src="img/abc.png" alt="Abc" height="40px"> <img src="img/JkOvmDOXpr.png" alt="招商银行" height="40px"> <img src="img/中国电信.png" alt="中国电信" height="40px"> <img src="img/bilibili.jpg" alt="Bilibili" height="38px"> <img src="img/qunar.png" alt="Qunar" height="35px"> <img src="img/蜜雪冰城.png" alt="蜜雪冰城" height="40px"> <img src="img/IDG.webp" alt="IDG" height="55px"> <img src="img/kingdee.png" alt="kingdee" height="40px"> <img src="img/unicom.png" alt="联通数科" height="40px"> <img src="img/changan.png" alt="长安汽车" height="40px"> <img src="img/tiane.png" alt="天鹅到家" height="35px"> </div> <div align="center"> <img src="img/清华大学.jpg" alt="清华大学" height="40px"> <img src="img/北京大学.png" alt="北京大学" height="40px"> <img src="img/nanyang.png" alt="南洋理工大学" height="40px"> <img src="img/fudan.png" alt="复旦大学" height="40px"> <img src="img/浙江大学.png" alt="浙江大学" height="40px"> <img src="img/南京大学.png" alt="南京大学" height="40px"> <img src="img/wuhan.png" alt="武汉大学" height="40px"> <img src="img/An-NajahNationalUniversity.png" alt="An-Najah National University" height="40px"> <img src="img/西安交通大学.png" alt="西安交通大学" height="40px"> <img src="img/huazhong.png" alt="华中科技大学" height="45px"> <img src="img/南开大学.jpg" alt="南开大学" height="45px"> <img src="img/四川大学.png" alt="四川大学" height="40px"> <img src="img/Binus_University.svg" alt="Binus University" height="45px"> </div> <br> <br>

💬 Join the Community

🌐 Online Discussions

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

📱 Discussion Community

<table> <thead> <tr> <th>WeChat Group</th> </tr> </thead> <tbody> <tr> <td><img src="img/wechatgroup.png" alt="WeChat Group" width="200"></td> </tr> </tbody> </table>

📧 Contact Us

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

🔗 More Security Tools

If you are interested in code security, we recommend A.S.E (AICGSecEval), the industry's first project-level AI-generated code security evaluation framework open-sourced by the Tencent Wukong Code Security Team.

<br> <br>

📖 Citation

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

@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>

📄 License

This project is open-sourced under the Apache License 2.0. See the LICENSE file for details.

⚖️ License & Attribution

This project is open-sourced under the Apache License 2.0. We warmly welcome and encourage community contributions, integrations, and derivative works, subject to the following attribution requirements:

1. Retain notices: You must retain the LICENSE and NOTICE files from the original project in any distribution.
2. Product attribution: If you integrate AI-Infra-Guard's core code, components, or scanning engine into your open-source project, commercial product, or internal platform, you must clearly state the following in your product documentation, usage guide, or UI "About" page:

   "This project integrates AI-Infra-Guard, open-sourced by Tencent Zhuque Lab."

3. Academic & article citation: If you use this tool in vulnerability analysis reports, security research articles, or academic papers, please explicitly mention "Tencent Zhuque Lab AI-Infra-Guard" and include a link to the repository.

Repackaging this project as an original product without disclosing its origin is strictly prohibited.

<div> <a href="https://www.star-history.com/?type=date&repos=Tencent%2FAI-Infra-Guard"> <picture> <source media="(prefers-color-scheme: dark)" srcset="https://api.star-history.com/chart?repos=Tencent/AI-Infra-Guard&type=date&theme=dark&legend=top-left&sealed_token=3aux6V5PKs5SBSuoVgw6MReXo5IzleLoV22UkaLZtWuN6kK4PltQDiq-hrtHpF4smNRGO9dbhrjk9Q4m7FWsPPUQqIsQUUrZkwev7vTDanFVCAHfU1qusQ" /> <source media="(prefers-color-scheme: light)" srcset="https://api.star-history.com/chart?repos=Tencent/AI-Infra-Guard&type=date&legend=top-left&sealed_token=3aux6V5PKs5SBSuoVgw6MReXo5IzleLoV22UkaLZtWuN6kK4PltQDiq-hrtHpF4smNRGO9dbhrjk9Q4m7FWsPPUQqIsQUUrZkwev7vTDanFVCAHfU1qusQ" /> <img alt="Star History Chart" src="https://api.star-history.com/chart?repos=Tencent/AI-Infra-Guard&type=date&legend=top-left&sealed_token=3aux6V5PKs5SBSuoVgw6MReXo5IzleLoV22UkaLZtWuN6kK4PltQDiq-hrtHpF4smNRGO9dbhrjk9Q4m7FWsPPUQqIsQUUrZkwev7vTDanFVCAHfU1qusQ" /> </picture> </a>