UML-MCP - MCP Server | MCP Directory

UML-MCP: A Diagram Generation Server with MCP Interface

License: MIT

Python 3.10+

smithery badge

UML-MCP is a powerful diagram generation server that implements the Model Context Protocol (MCP), enabling seamless diagram creation directly from AI assistants and other applications.

🌟 Features

Multiple Diagram Types: Support for UML diagrams (Class, Sequence, Activity, etc.), Mermaid, D2, and more
MCP Integration: Seamless integration with LLM assistants supporting the Model Context Protocol
Playground Links: Direct links to online editors for each diagram type
Multiple Output Formats: SVG, PNG, PDF, and other format options
Easy Configuration: Works with local and remote diagram rendering services

📋 Supported Diagram Types

UML-MCP supports a wide variety of diagram types:

| Category | Diagram Types | |----------|---------------| | UML | Class, Sequence, Activity, Use Case, State, Component, Deployment, Object | | Other | Mermaid, D2, Graphviz, ERD, BlockDiag, BPMN, C4 with PlantUML |

🚀 Getting Started

Prerequisites

Python 3.10 or higher
pip (Python package installer)

Installation

Clone the repository:

git clone https://github.com/yourusername/uml-mcp.git
cd uml-mcp

Install the dependencies:

pip install -r requirements.txt

For development environments:

pip install -r requirements-dev.txt

Running the Server

Start the MCP server:

python mcp_server.py

This will start the server using stdio for communication with MCP clients.

🔧 Configuration

Editor Integration

Cursor

To integrate with Cursor:

python mcp/install_to_cursor.py

Or manually configure in Cursor settings:

"mcpServers": {
  "UML-MCP-Server": {
    "command": "python",
    "args": ["/path/to/uml-mcp/mcp_server.py"],
    "output_dir": "/path/to/output"
  }
}

Environment Variables

MCP_OUTPUT_DIR - Directory to save generated diagrams (default: ./output)
KROKI_SERVER - URL of the Kroki server (default: https://kroki.io)
PLANTUML_SERVER - URL of the PlantUML server (default: http://plantuml-server:8080)
USE_LOCAL_KROKI - Use local Kroki server (true/false)
USE_LOCAL_PLANTUML - Use local PlantUML server (true/false)

📚 Documentation

For detailed documentation, visit the docs directory or our documentation site.

🧩 Architecture

UML-MCP is built with a modular architecture:

MCP Server Core: Handles MCP protocol communication
Diagram Generators: Supporting different diagram types
Tools: Expose diagram generation functionality through MCP
Resources: Provide templates and examples for various diagram types

🛠️ Local Development

For local development:

Set up local PlantUML and/or Kroki servers:

# PlantUML
docker run -d -p 8080:8080 plantuml/plantuml-server

# Kroki
docker run -d -p 8000:8000 yuzutech/kroki

Configure environment variables:

export USE_LOCAL_PLANTUML=true
export PLANTUML_SERVER=http://localhost:8080
export USE_LOCAL_KROKI=true  
export KROKI_SERVER=http://localhost:8000

🤝 Contributing

Contributions are welcome! Please feel free to submit a Pull Request.

📄 License

This project is licensed under the MIT License - see the LICENSE file for details.

👏 Acknowledgements

PlantUML - UML diagram generation
Kroki - Unified diagram generation service
Mermaid - Generation of diagrams from text
D2 - Modern diagram scripting language

UML-MCP Server

An MCP Server that provides UML diagram generation capabilities through various diagram rendering engines.

Components

Resources

The server provides several resources via the uml:// URI scheme:

uml://types: List of available UML diagram types
uml://templates: Templates for creating UML diagrams
uml://examples: Example UML diagrams for reference
uml://formats: Supported output formats for diagrams
uml://server-info: Information about the UML-MCP server

Tools

The server implements multiple diagram generation tools:

Universal UML Generator

generate_uml: Generate any UML diagram
- Parameters: diagram_type, code, output_dir

Specific UML Diagram Tools

generate_class_diagram: Generate UML class diagrams
- Parameters: code, output_dir
generate_sequence_diagram: Generate UML sequence diagrams
- Parameters: code, output_dir
generate_activity_diagram: Generate UML activity diagrams
- Parameters: code, output_dir
generate_usecase_diagram: Generate UML use case diagrams
- Parameters: code, output_dir
generate_state_diagram: Generate UML state diagrams
- Parameters: code, output_dir
generate_component_diagram: Generate UML component diagrams
- Parameters: code, output_dir
generate_deployment_diagram: Generate UML deployment diagrams
- Parameters: code, output_dir
generate_object_diagram: Generate UML object diagrams
- Parameters: code, output_dir

Other Diagram Formats

generate_mermaid_diagram: Generate diagrams using Mermaid syntax
- Parameters: code, output_dir
generate_d2_diagram: Generate diagrams using D2 syntax
- Parameters: code, output_dir
generate_graphviz_diagram: Generate diagrams using Graphviz DOT syntax
- Parameters: code, output_dir
generate_erd_diagram: Generate Entity-Relationship diagrams
- Parameters: code, output_dir

Prompts

The server provides prompts to help create UML diagrams:

class_diagram: Create a UML class diagram showing classes, attributes, methods, and relationships
sequence_diagram: Create a UML sequence diagram showing interactions between objects over time
activity_diagram: Create a UML activity diagram showing workflows and business processes

Configuration

Install

Claude Desktop

On MacOS: ~/Library/Application\ Support/Claude/claude_desktop_config.json
On Windows: %APPDATA%/Claude/claude_desktop_config.json

<details> <summary>Claude Desktop Configuration</summary>

{
  "mcpServers": {
    "uml_diagram_generator": {
      "command": "python",
      "args": [
        "/path/to/uml-mcp/mcp_server.py"
      ]
    }
  }
}

</details>

Cursor Integration

The UML-MCP Server can also be integrated with Cursor:

<details> <summary>Cursor Configuration</summary>

{
  "mcpServers": {
    "uml_diagram_generator": {
      "command": "python",
      "args": [
        "/path/to/uml-mcp/mcp_server.py"
      ]
    }
  }
}

</details>

Usage

Command Line Arguments

usage: mcp_server.py [-h] [--debug] [--host HOST] [--port PORT] [--transport {stdio,http}] [--list-tools]

UML-MCP Diagram Generation Server

options:
  -h, --help            show this help message and exit
  --debug               Enable debug logging
  --host HOST           Server host (default: 127.0.0.1)
  --port PORT           Server port (default: 8000)
  --transport {stdio,http}
                        Transport protocol (default: stdio)
  --list-tools          List available tools and exit

Environment Variables

LOG_LEVEL: Set logging level (DEBUG, INFO, WARNING, ERROR, CRITICAL)
UML_MCP_OUTPUT_DIR: Directory to store generated diagram files
KROKI_SERVER: Kroki server URL for diagram rendering
PLANTUML_SERVER: PlantUML server URL for diagram rendering
LIST_TOOLS: Set to "true" to display tools and exit

Example: Generating a Class Diagram

result = tool.call("generate_class_diagram", {
    "code": """
        @startuml
        class User {
          -id: int
          -name: string
          +login(): boolean
        }
        class Order {
          -id: int
          +addItem(item: string): void
        }
        User "1" -- "many" Order
        @enduml
    """,
    "output_dir": "/path/to/output"
})

Development

Building and Running

# Clone the repository
git clone https://github.com/your-username/uml-mcp.git
cd uml-mcp

# Install dependencies
pip install -r requirements.txt

# Run the server
python mcp_server.py

Debugging

For debugging, you can run the server with:

python mcp_server.py --debug

Debug logs will be stored in the logs/ directory.

Running Tests

# Run all tests
pytest

# Run specific tests
pytest tests/test_diagram_tools.py