生成式AI漏洞检测工具,评 估语言模型安全性
该工具检查LLM的潜在故障,包括幻觉、数据泄露、提示注入、错误信息、生成有害内容和越狱等。作为免费的命令行工具,支持多种生成模型,如Hugging Face、Replicate、OpenAI等。用户可通过pip或GitHub安装最新版本,并提供详尽的日志记录和报告功能,帮助识别和修复语言模型的漏洞。
garak
是一个专为大规模语言模型(LLM)设计的漏洞扫描工具,旨在检测诸如幻觉、数据泄露、提示注入、错误信息传播、生成毒性内容以及其他潜在弱点。可以将其视作 LLMs 的 nmap
,着重于探索使对话系统或语言模型失败的方法,通过静态、动态和自适应的探测技术来实现这些目标。
garak
是一个免费工具,开发团队对它的开发充满热情,并且希望不断拓展其功能,以支持更多的应用场景。
目前,garak
支持以下生成模型和接口:
pip
安装标准安装通过 PyPI 完成,只需执行以下命令即可:
python -m pip install -U garak
如果希望获得更新的版本,可以通过以下命令从 GitHub 安装:
python -m pip install -U git+https://github.com/leondz/garak.git@main
garak
有其自身的依赖,可以在独立的 Conda 环境中安装:
conda create --name garak "python>=3.10,<=3.12"
conda activate garak
gh repo clone leondz/garak
cd garak
python -m pip install -e .
garak
一般语法是:
garak <options>
garak
需要知道要扫描哪个模型,默认情况下,它会对该模型应用所有已知的探针,并使用每个探针推荐的脆弱性检测器。可以使用 garak --list_probes
查看探针列表。通过 --model_type
和(可选的)--model_name
来指定模型类型和名称。
例如,以下命令用于检查 ChatGPT 的基于编码的提示注入(需要设置 OPENAI_API_KEY):
export OPENAI_API_KEY="sk-123XXXXXXXXXXXX"
python3 -m garak --model_type openai --model_name gpt-3.5-turbo --probes encoding
以下是 garak
支持的一些探针:
探针 | 描述 |
---|---|
blank | 总是发送空提示的简单探针。 |
atkgen | 自动攻击生成,红队 LLM 探针模拟目标并与其交互,尝试生成毒性输出。 |
encoding | 通过文本编码进行的提示注入。 |
dan | 各种DAN 和类似攻击。 |
garak
生成多种日志:
garak.log
:包括调试信息,持续记录所有运行。garak
的代码结构模块化,主要包括以下几个部分:
garak/probes/
- 生成与 LLM 交互的类garak/detectors/
- 检测 LLM 是否出现给定失败模式的类garak/evaluators/
- 评估报告机制garak/generators/
- 要探测的 LLM 插件garak/harnesses/
- 结构化测试的类想要添加自己的插件?可以参考其他插件的实现,继承基类并尽量少进行覆盖。测试新代码可以通过交互式 Python 环境或运行扫描测试插件。
详细的常见问题解答可在 FAQ 查看。
如果在使用 garak
,请参考我们的 前置论文 进行引用。
AI数字人视频创作平台
Keevx 一款开箱即用的AI数字人视频创作平台,广泛适用于电商广告、企业培训与社媒宣传,让全球企业与个人创作者无需拍摄剪辑,就能快速生成多语言、高质量的专业视频。
一站式AI创作平台
提供 AI 驱动的图片、视频生成及数字人等功能,助力创意创作
AI办公助手,复杂任务高效处理
AI办公助手,复杂任务高效处理。办公效率低?扣子空间AI助手支持播客生成、PPT制作、网页开发及报告写作,覆盖科研、商业、舆情等领域的专家Agent 7x24小时响应,生活工作无缝切换,提升50%效率!
AI辅助编程,代码自 动修复
Trae是一种自适应的集成开发环境(IDE),通过自动化和多元协作改变开发流程。利用Trae,团队能够更快速、精确地编写和部署代码,从而提高编程效率和项目交付速度。Trae具备上下文感知和代码自动完成功能,是提升开发效率的理想工具。
AI小说写作助手,一站式润色、改写、扩写
蛙蛙写作—国内先进的AI写作平台,涵盖小说、学术、社交媒体等多场景。提供续写、改写、润色等功能,助力创作者高效优化写作流程。界面简洁,功能全面,适合各类写作者提升内容品质和工作效率。