Agent-Wiz

Stale
GitHub Python Apache-2.0

Description

A CLI threat-modeling tool from Repello AI that visualizes attack surfaces of LangGraph, AutoGen, and CrewAI agent builds.

Key Features

  • Framework-native parsers - Built-in support for LangGraph, AutoGen, CrewAI, and more
  • Attack-surface graph - Generates dependency and data-flow graphs for every agent node
  • Risk scoring - Maps each node to OWASP LLM Top 10 risk categories
  • Offline CLI - Runs locally with no cloud dependency, CI-friendly
  • Exportable reports - Outputs SVG and JSON artifacts for security reviews

Use Cases

💡 Review attack surface of agent code in pull requests
💡 Generate threat-modeling diagrams during security reviews of LangGraph/AutoGen projects
💡 Produce OWASP LLM Top 10 risk inventories for compliance audits
💡 Demonstrate common agent risk patterns in training and workshops

Strengths & Limitations

Strengths

  • Permissive open-source license (Apache-2.0)

⚠️ Limitations

  • No updates in over 9 months
  • High issue backlog (23 open issues)

Quick Start

# Install Agent-Wiz
pip install agent-wiz

# Scan a LangGraph project in the current directory
agent-wiz scan ./my_agent --framework langgraph

# Produce a visual report
agent-wiz scan ./my_agent --output report.svg

Related Projects