sameerrajput99/AI-Red-Teaming-Test-Harness
GitHub: sameerrajput99/AI-Red-Teaming-Test-Harness
一个基于 YAML 结构化测试用例的本地化 LLM 安全测试框架,解决手动 AI 安全测试难以复现和标准化的问题。
Stars: 1 | Forks: 0
# AI Red Teaming 测试框架
一个用于定义结构化 AI 安全测试用例的、安全的、本地的且可复现的框架。
## 第一天状态
第一个里程碑将在任何结构化 YAML 测试用例发送给任何聊天机器人之前对其进行验证。
当前功能:
- 严格的测试用例 schema
- 三个初始测试用例
- 使用安全解析的 YAML 加载
- 清晰的验证错误提示
- 自动化的单元测试
## 本项目存在的原因
当 prompt、预期行为和证据以非正式方式存储时,手动 AI 安全测试将变得难以复现。本项目将这些检查转化为结构化的测试用例,之后可以针对脆弱和经过加固的聊天机器人配置执行这些测试。
## 第一天架构
```
YAML Test Pack
↓
Safe YAML Loader
↓
Pydantic Schema Validation
↓
Valid TestCase Objects
↓
Ready for the future Test Runner
```
## 在 Windows PowerShell 上进行设置
```
py --version
py -m venv .venv
.venv\Scripts\Activate.ps1
python -m pip install --upgrade pip
python -m pip install -e ".[dev]"
```
如果 PowerShell 阻止激活,请改用命令提示符:
```
.venv\Scripts\activate.bat
```
## 验证初始测试包
```
validate-ai-tests test_packs/day1_test_cases.yaml
```
替代命令:
```
python -m ai_red_teaming_harness.validate_cases test_packs/day1_test_cases.yaml
```
预期结果:
```
Validated 3 test cases successfully.
```
## 运行单元测试
```
pytest
```
## 初始测试类别
1. Prompt 泄露攻击
2. 指令覆盖攻击
3. 良性对照
## 重要安全原则
测试框架并不能证明一个 AI 系统是完全安全的。它为所定义的特定测试和评估器生成可复现的证据。
## 道德使用
本项目用于安全的本地模拟、防御性测试、教育和作品集开发。未经明确授权,请勿将其用于针对其他系统。
标签:AI红队, LLM测试, Pydantic, YAML, 人工智能安全, 合规性, 多语言支持, 安全库, 安全测试框架, 安全规则引擎, 逆向工具