generate_diagram

Generate AWS, Kubernetes, and other architecture diagrams from Python code using the diagrams package to visualize cloud infrastructure and system designs.

Instructions

Generate a diagram from Python code using the diagrams package.

This tool accepts Python code as a string that uses the diagrams package DSL and generates a PNG diagram without displaying it. The code is executed with show=False to prevent automatic display.

USAGE INSTRUCTIONS: Never import. Start writing code immediately with with Diagram( and use the icons you found with list_icons.

First use get_diagram_examples to understand the syntax and capabilities
Then use list_icons to discover all available icons. These are the only icons you can work with.
You MUST use icon names exactly as they are in the list_icons response, case-sensitive.
Write your diagram code following python diagrams examples. Do not import any additional icons or packages, the runtime already imports everything needed.
Submit your code to this tool to generate the diagram
The tool returns the path to the generated PNG file
For complex diagrams, consider using Clusters to organize components
Diagrams should start with a user or end device on the left, with data flowing to the right.

CODE REQUIREMENTS:

Must include a Diagram() definition with appropriate parameters
Can use any of the supported diagram components (AWS, K8s, etc.)
Can include custom styling with Edge attributes (color, style)
Can use Cluster to group related components
Can use custom icons with the Custom class

COMMON PATTERNS:

Basic: provider.service("label")
Connections: service1 >> service2 >> service3
Grouping: with Cluster("name"): [components]
Styling: service1 >> Edge(color="red", style="dashed") >> service2

IMPORTANT FOR CLINE: Always send the current workspace directory when calling this tool! The workspace_dir parameter should be set to the directory where the user is currently working so that diagrams are saved to a location accessible to the user.

Supported diagram types:

AWS architecture diagrams
Sequence diagrams
Flow diagrams
Class diagrams
Kubernetes diagrams
On-premises diagrams
Custom diagrams with custom nodes

Returns: Dictionary with the path to the generated diagram and status information

Input Schema

Name	Required	Description
`code`	Yes	Python code using the diagrams package DSL. The runtime already imports everything needed so you can start immediately using `with Diagram(`
`filename`	No	The filename to save the diagram to. If not provided, a random name will be generated.
`timeout`	No	The timeout for diagram generation in seconds. Default is 90 seconds.
`workspace_dir`	No	The user's current workspace directory. CRITICAL: Client must always send the current workspace directory when calling this tool! If provided, diagrams will be saved to a 'generated-diagrams' subdirectory.

Input Schema (JSON Schema)

{
  "properties": {
    "code": {
      "description": "Python code using the diagrams package DSL. The runtime already imports everything needed so you can start immediately using `with Diagram(`",
      "title": "Code",
      "type": "string"
    },
    "filename": {
      "anyOf": [
        {
          "type": "string"
        },
        {
          "type": "null"
        }
      ],
      "default": null,
      "description": "The filename to save the diagram to. If not provided, a random name will be generated.",
      "title": "Filename"
    },
    "timeout": {
      "default": 90,
      "description": "The timeout for diagram generation in seconds. Default is 90 seconds.",
      "title": "Timeout",
      "type": "integer"
    },
    "workspace_dir": {
      "anyOf": [
        {
          "type": "string"
        },
        {
          "type": "null"
        }
      ],
      "default": null,
      "description": "The user's current workspace directory. CRITICAL: Client must always send the current workspace directory when calling this tool! If provided, diagrams will be saved to a 'generated-diagrams' subdirectory.",
      "title": "Workspace Dir"
    }
  },
  "required": [
    "code"
  ],
  "title": "mcp_generate_diagramArguments",
  "type": "object"
}

AWS Diagram MCP Server