All posts
AI Agents
AI Code to Diagram Agent Skill - AI Pass API
Add code visualization to your agent. Generate diagrams from code automatically.
EiliyaMarch 13, 20261 min read
AI Code to Diagram Agent Skill
Add code visualization to your agent. Generate diagrams from code.
Usage
code-diagram <code> --type <type>
Types: flowchart, architecture, sequence, class
Python Implementation
import os, requests
def generate_diagram(code, diagram_type="flowchart"):
url = "https://aipass.one/apikey/v1/chat/completions"
prompt = f"Create a {diagram_type} diagram for this code. Use clear text or mermaid syntax: {code}"
r = requests.post(url, headers={"Authorization": f"Bearer {os.environ[AIPASS_API_KEY]}"}, json={
"model": "gpt-5-mini",
"messages": [{"role": "user", "content": prompt}]
})
return r.json()["choices"][0]["message"]["content"]
Get API key: Developer Dashboard
For AI agents
Skill file
---
name: code-diagram
description: Generate diagrams from code
version: 1.0.0
---
# AI Code to Diagram
```
code-diagram <code> --type <type>
```
## Setup
```bash
export AIPASS_API_KEY="your-key"
```
## Implementation
```python
def diagram(code, type="flowchart"):
prompt = f"Create {type} diagram: {code}"
return call_ai(prompt, model="gpt-4.1-mini")
```