Agent-Wiz
Agent Wiz is a Python CLI that extracts agentic workflows from popular AI frameworks using static analysis and performs automated threat assessments with the MAESTRO framework. It helps developers, researchers, and security teams visualize agent graphs, map interactions, and generate security reports.
✨ Key features
- Extracts agent workflows via AST-based static parsing.
- Visualizes agent-to-agent and tool connections interactively.
- Generates MAESTRO threat assessment reports automatically.
- Supports multiple frameworks including Autogen, CrewAI, LangGraph.
- Provides simple CLI and JSON exports for integration.
- Framework agnostic with extensible SDK.
🎯 Use cases
- Audit complex LLM agent systems for security vulnerabilities.
- Document agent interactions and data flows for compliance.
- Visualize agent graphs to understand orchestration logic.
- Integrate threat modeling into CI/CD pipelines for agent code.
📦 Installation
🧰 Requirements: Python environment with pip; OpenAI API key required for analysis commands.
pip install repello-agent-wiz
Before running analysis, set your OpenAI API key:
export OPENAI_API_KEY=sk-...
or copy .env.sample to .env and fill in the key.
🚀 Usage
agent-wiz extract --framework agent_chat --directory ./examples/code/agent_chat --output agentchat_graph.jsonagent-wiz visualize --input agentchat_graph.json --openagent-wiz analyze --input agentchat_graph.json⚠️ Good to know
Currently supports only MAESTRO threat modeling; other frameworks (STRIDE, PASTA, LINDDUN) are under development.
❓ FAQ
Which frameworks are supported?
Autogen (core), AgentChat, CrewAI, LangGraph, LlamaIndex, n8n, OpenAI Agents, Pydantic-AI, Swarm, and Google-ADK.
How do I configure the OpenAI API key?
Set the OPENAI_API_KEY environment variable or copy .env.sample to .env and add your key there.
What output does the extract command produce?
It generates a JSON file with nodes, edges, and metadata containing the framework name.
Can I use Agent Wiz without an OpenAI API key?
Extraction and visualization do not require it, but the analyze command does.
📊 Repository
🤖 Overview, features, install steps and FAQ were generated from the project's README on Sep 4, 2026. Always check the original source before running commands.