llm-draw 0.1.2

LLM diagram generation library for Mermaid, PlantUML, and draw.io style workflows.

llm-draw

Generate diagram source from natural-language prompts using LLMs.

llm-draw infers the diagram type and output format from your prompt, then returns Mermaid, PlantUML, draw.io XML, or Markdown diagram text.

Install

Install from PyPI:

python -m pip install "llm-draw[openai]"

Other extras are available for specific integrations:

python -m pip install "llm-draw[langchain]"

Example

import os

from llm_draw import LlmDraw

client = LlmDraw(provider="openai")
result = client.generate(
    prompt="Show a web app with browser, API, PostgreSQL, Redis, and Stripe.",
    apikey=os.environ["OPENAI_API_KEY"],
    model_id="gpt-4.1-mini",
)

print(result.content)
print(result.diagram_type, result.output_format)

CLI

export OPENAI_API_KEY="..."

llm-draw \
  "Draw a payment workflow with fraud checks and refund handling" \
  --model-id gpt-4.1-mini \
  --output workflow.mmd

llm-draw writes inferred metadata to stderr and writes diagram source to stdout, unless --output is provided.

Supported Diagrams

• Architecture, system architecture, network, AWS, Azure, GCP, Kubernetes, Terraform
• Flowchart, process flow, workflow, process map, BPMN, swimlane, state, sequence
• ERD, UML, use case, block, data flow, user flow, user journey map, org chart
• Mermaid editor and PlantUML editor modes