dreadnode/sdk

GitHub: dreadnode/sdk

Strikes 是一个用于构建、实验和评估 AI 安全 agent 的 Python SDK,集任务编排、数据追踪、指标衡量与可扩展运行于一体。

Stars: 26 | Forks: 2

Logo

Dreadnode Strikes SDK

PyPI - Python Version PyPI - Version GitHub License Tests Pre-Commit Renovate


Strikes 是一个用于构建、实验和评估 AI 安全 agent 代码的平台。 - **实验 + 任务分配 + 可观测性** 集于一处,轻量且易于扩展。 - **跟踪你的数据**,将参数、输入和输出全部与你的任务关联。 - **记录你的 artifact** —— 数据、模型、文件和文件夹 —— 以跟踪你的 Dreadnode 运行数据,从而实现轻松的复用和重现。 - **衡量一切**,在你的代码中以及任何你需要的地方使用指标。 - **扩展你的代码**,从单次运行扩展到数千次。 ``` import dreadnode as dn import rigging as rg from .tools import reversing_tools dn.configure() @dataclass class Finding: name: str severity: str description: str exploit_code: str @dn.scorer(name="Score Finding") async def score_finding(finding: Finding) -> float: if finding.severity == "critical": return 1.0 elif finding.severity == "high": return 0.8 else: return 0.2 @dn.task(scorers=[score_finding]) @rg.prompt(tools=[reversing_tools]) async def analyze_binary(binary: str) -> list[Finding]: """ Analyze the binary for vulnerabilities. """ ... with dn.run(tags=["reverse-engineering"]): binary = "c2/downloads/service.exe" dn.log_params( model="gpt-4", temperature=0.5, binary=binary ) findings = await analyze_binary(binary) dn.log_metric("findings", len(findings)) ``` ## 安装 我们将每个版本发布到 PyPi: ``` pip install -U dreadnode ``` 如果你想从源码构建: ``` uv sync # 安装 multimodal extras uv sync --extras multimodal # 安装 training extras uv sync --extras training # 安装所有 extras uv sync --all-extras ``` ## 从 PyPI 安装可选功能 如需高级媒体处理功能(音频、视频、图像),请安装 multimodal 扩展: ``` # Multimodal 支持(音频、视频处理) pip install -U "dreadnode[multimodal]" # Training 支持(ML model 集成) pip install -U "dreadnode[training]" # 所有可选功能 pip install -U "dreadnode[all]" ``` 有关更多选项,请参阅我们的**[安装指南](https://docs.dreadnode.io/strikes/install)**。 ## 快速开始 请阅读我们文档中的**[介绍指南](https://docs.dreadnode.io/strikes/intro)**。 ## 示例 查看 **[dreadnode/example-agents](https://github.com/dreadnode/example-agents)**,寻找你最喜欢的用例。
标签:AI安全, Apex, Chat Copilot, Python, 实验追踪, 无后门, 机器学习, 自动化评估, 逆向工具