llamator
LLM/GenAI红队测试框架,28种攻击策略,OWASP LLM Top 10全覆盖
加载项目详情…
本应用为开源项目,仅供学习研究,请遵守其开源协议。
LLM/GenAI红队测试框架,28种攻击策略,OWASP LLM Top 10全覆盖
加载项目详情…
本应用为开源项目,仅供学习研究,请遵守其开源协议。

图1:LLAMATOR GitHub仓库概览
你有没有想过,当你和AI聊天机器人对话时,它到底有多容易被诱导说出不该说的话?研究人员早就发现,大语言模型(LLM)在面对精心设计的对抗性提问时,可能会绕过安全限制,输出有害内容、泄露系统提示词,甚至传播虚假信息。这些风险在金融、医疗、法律等高敏感场景中尤为致命。LLAMATOR就是为解决这一痛点而生的——它是一个专业的LLM/GenAI红队测试Python框架,让安全研究人员和AI开发者能够系统化地评估AI系统的安全边界。
LLAMATOR由LLAMATOR-Core团队开发,定位为AI安全评估的全链路工具链。与单一的Prompt注入测试工具不同,LLAMATOR覆盖了Prompt注入、越狱(Jailbreak)、系统提示词泄露、虚假信息传播、RAG检索投毒、多模态攻击等多个攻击维度,支持对聊天机器人、RAG系统、AI Agent、多模态模型(VLM)进行批量自动化安全评估。项目遵循OWASP LLM Top 10标准,与行业安全基准保持同步,2025年还获得了OWASP Gen AI Red Teaming Guide的实践推荐。
如果把AI系统的安全评估比作一场模拟黑客演习,LLAMATOR就是演习指挥中心。它预置了20+攻击策略(如Suffix、AutoDAN、DAN角色扮演、Base64编码注入等),内置攻击数据集(Parquet格式),支持通过YAML配置自定义攻击流程,通过Excel/CSV记录每次攻击的请求与响应,最终生成DOCX格式的测试报告,全程无需编写代码。
LLAMATOR支持三大类客户端接入:一是所有LangChain兼容的聊天客户端;二是OpenAI-like API(任意支持OpenAI SDK的模型);三是自定义类客户端(通过继承基类实现Telegram、WhatsApp、Selenium浏览器自动化等)。框架预置了Excel报告生成器和Word报告生成器,支持将攻击测试结果以结构化文档输出,便于安全审计和合规汇报。
框架本身采用标准Python项目结构,代码位于src/llamator/下,核心模块包括:attacks/(20+攻击策略实现)、attack_data/(攻击数据集和资源文件)、attack_provider/(攻击加载与分发)、client/(聊天客户端抽象层)、report_generators/(Excel/Word报告生成)。开发规范严格,配置了mypy静态类型检查、pylint代码风格检查、pre-commit自动格式化,测试覆盖率通过.coveragerc管理。
上手门槛非常低。安装仅需pip install llamator==3.5.0,运行测试只需几行代码。Docker部署也有官方支持,基于python:3.11-slim镜像,预装了Jupyter Notebook环境,适合快速验证。对于非技术背景的安全审计人员,项目提供了Sphinx文档、多格式指南,以及YouTube视频教程。唯一的限制是:作为纯CLI工具,没有Web界面,可视化程度有限。
LLAMATOR所在的AI安全测试赛道竞争激烈,竞争对手包括Microsoft's Counterfit、Google's Garak、IBM's Adversarial Robustness Toolbox等。但LLAMATOR的差异化在于:覆盖的攻击类型最全面(28种),支持多模态VLM攻击,内置OWASP LLM01-10全部分类,且提供了开箱即用的DOCX报告生成,这在企业合规场景中非常实用。随着OWASP Gen AI Red Teaming Guide的推广,AI红队测试正在从可选变为必选,LLAMATOR作为这一趋势中的成熟工具,有望持续获得关注。