AI Code to Diagram Agent Skill - AI Pass API
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": "openai/gpt-4.1-mini",
"messages": [{"role": "user", "content": prompt}]
})
return r.json()["choices"][0]["message"]["content"]
Get API key: Developer Dashboard
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")
```