SuperAPI AI 工具集

Agent-Wiz

Agent-Wiz 是一个 Python 命令行工具,用于从 LangGraph、AutoGen、CrewAI 等主流 AI 框架中提取智能体工作流,进行可视化展示,并基于 MAESTRO 等威胁建模方法自动执行安全威胁评估。它面向开发者、研究人员和安全团队,帮助理清复杂 LLM 编排中的智能体、工具与数据流关系,识别潜在漏洞并生成安全报告。Agent-Wiz 需要配置 OpenAI API key 才能运行分析命令。相关框架解析器、标准化图表示、安全报告生成、STRIDE/PASTA/LINDDUN 等更多威胁模型分析以及智能体仿真威胁探索等特性仍在规划或开发中。

访问官网

工具介绍

用途:Agent-Wiz 是一个基于 Python 的命令行工具(CLI),用于从主流 AI 智能体编排框架中提取智能体工作流,并执行自动化威胁评估。它面向开发者、研究人员和安全团队,目的是为基于 LLM 的复杂编排提供可见性,帮助可视化流程、映射工具与智能体交互,并生成可操作的安全报告。

主要能力:

  1. 工作流提取:为支持的框架提供各自的 AST 静态解析器,用于提取基于类或函数的智能体、智能体之间的转换、工具调用链以及组智能体(如 selector、round-robin)。
  2. 可视化:生成基于 HTML d3 的智能体工作流可视化;可选的 open 标志会自动在默认浏览器中打开该可视化。
  3. 安全分析:当前以 MAESTRO 作为主要威胁建模框架,按照 Mission(系统目的与安全目标)、Assets(智能体、工具、数据流等关键组件清单)、Entrypoints(攻击面与访问向量映射)、Security Controls(现有防御措施评估)、Threats(潜在漏洞与攻击场景识别)、Risks(安全事件影响与可能性计算)、Operations(运行时安全评估)的结构评估智能体工作流。
  4. 报告生成:基于提供的图和威胁建模框架生成报告,例如 autogen_report.md。

使用方式:

  • 运行任何分析命令前,必须先配置 OpenAI API key,可通过环境变量或 .env 文件(推荐)两种方式设置。
  • 提取智能体工作流:该命令会生成具有特定结构的图 JSON。
  • 可视化智能体工作流:该命令会生成基于 d3 的 HTML 可视化。
  • 威胁建模分析:该命令会生成类似 autogen_report.md 的报告。
  • 可运行 agent-wiz --help 查看更多信息。
  • 如需持久使用,可将相关配置行加入 .bashrc、.zshrc 或环境设置脚本。

适用场景:适用于使用 LangGraph、AutoGen、CrewAI 等框架构建智能体系统的开发者、研究人员与安全团队,用于梳理复杂 LLM 编排中的智能体与工具关系、发现潜在安全漏洞,并产出安全评估报告。

限制与规划:

  • 当前主要支持 MAESTRO 威胁建模框架。
  • STRIDE、PASTA、LINDDUN 等更多威胁模型分析仍在开发中。
  • 计划中的功能(未按特定顺序)包括:为主要智能体框架构建解析器、生成智能体流程的标准化 JSON 图表示、安全报告生成、扩展到 STRIDE/PASTA/LINDDUN 等,以及基于智能体仿真的威胁探索。