UML-MCP-Server

English | 中文

UML-MCP Server is a UML diagram generation tool based on MCP (Model Context Protocol), which can help users generate various types of UML diagrams through natural language description or directly writing PlantUML code.

Functional Features

Supports multiple UML diagram types: class diagram, sequence diagram, activity diagram, use case diagram, state diagram, component diagram, deployment diagram, object diagram
UML diagrams can be generated through natural language description
You can directly use PlantUML code to generate UML diagrams
Return PlantUML code and accessible URL links for easy sharing and viewing
Simultaneously save the generated UML diagram locally and provide the local file path
Support custom save path and specify the output directory for UML images
As an MCP server, it can integrate with clients that support MCP, such as Claude
A comprehensive logging system that records server operating status and operation logs

Installation

Installing via Smithery

To install UML Diagram Generation Tool for Claude Desktop automatically via Smithery:

npx -y @smithery/cli install @Swayingleaves/uml-mcp-server --client claude

Manual Installation

Clone repository:

git clone https://github.com/yourusername/UML-MCP-Server.git
cd UML-MCP-Server

Create and activate a virtual environment:

python -m venv uml-mcp-venv
source uml-mcp-venv/bin/activate  # Linux/Mac
# Or
uml-mcp-venv\Scripts\activate  # Windows

Install dependencies:

pip install -r requirements.txt

Usage Method

Used as a Python Library

from fix_plantuml import generate_uml

# Create UML code
uml_code = """
@startuml
Title: Simple Class Diagram

class User {
-String name
-String email
+login()
+logout()
}

class Order {
-int id
-Date date
+process()
}

User "1" -- "many" Order: places
@enduml
"""

# Generate URLs, code, and local paths for UML diagrams
result = generate_uml(uml_code)

# Output result
print("PlantUML code: ")
print(result["code"])
print("\nPlantUML URL:")
print(result["url"])
print("\nLocal file path: ")
print(result["local_path"])

Configure MCP in Cursor

The Cursor supports MCP (Model Context Protocol) servers, allowing you to directly generate UML diagrams using UML-MCP Server in the Cursor. The configuration steps are as follows:

Ensure that the latest version of Cursor IDE (which supports MCP functionality) is installed.
Open the configuration file of Cursor:

On macOS: ~/Library/Application Support/Cursor/config.json
On Windows: %APPDATA%\Cursor\config.json
On Linux: ~/.config/Cursor/config.json

Add or modify the 'mcpServer' section in the configuration file:

{
    "mcpServers": {
        "UML-MCP-Server": {
            "command": "uv",
            "args": [
                "--directory",
                "/Users/yourpath/UML-MCP-Server",
                "run",
                "uml_mcp_server.py"
            ],
            "output_dir": "/Users/yourpath/uml-output"
        }
    }
}

Configuration Description:

UML-MCP-Server: The name of the MCP server can be modified as needed
Command: Use UV as the running command
args:
- directory: Specify the absolute path of the project directory
- run: Run the command
- uml_mcp_server.py: Main program file
output_dir: Specify the output directory for UML images

Please modify the following path according to your actual situation:

Replace '/Users/Yourpath/UML-MCP Server' with the actual path of your UML-MCP Server project
Replace '/Users/Yourpath/uml-output' with the directory path where you want to save the UML image

Save the configuration file and restart the Cursor.
Use UML-MCP Server in Cursor:

Open a new chat window
In the toolbar at the bottom of the chat interface, you will see the "UML-MCP Server" tool icon
Clicking on this icon will bring up the options menu for UML tools
Select the UML diagram type you need (such as "Generate Class Diagram", "Generate Sequence Diagram", etc.)

Using UML Tools in Cursor

In the Cursor, you can input:

For example:

Understand the certification process of the project
Generate UML code for the authentication process and generate a flowchart through UML-MCP Server
Attention: "output_dir": "/Users/edy/vs-code/bjwa-task-project/uml-output"

The Results Returned by UML Tools

Regardless of how UML tools are used, Cursor will call UML-MCP-Server and return the following:

PlantUML Code - You can copy this code to use in other PlantUML tools
PlantUML URL - You can open this URL in a browser to view the generated UML diagram
Local file path - The generated UML image is saved in this local path

For example:

Class diagram generated:

PlantUML code:
@startuml
Title: User and Order System
class User {
-String name
-String email
+login()
+logout()
}
class Order {
-int id
-Date date
+process()
}
User "1" -- "many" Order: places
@enduml

PlantUML URL：
http://www.plantuml.com/plantuml/png/~1UDgCqB5Bn0G1k1zYWM_EfPYQYY0Qd9oQc9oQaPcKYYcKc9gMYaiKc9gK...

Local file path:
/Users/username/projects/UML-MCP-Server/output/class_diagram_12345.png

You can view the generated UML diagram in the following ways:

Click on the returned URL link to view in the browser
Open the local file path in the file browser to view the saved image
In Cursor, you can use Markdown syntax to directly display images in the chat window

Troubleshooting

If you encounter problems while using UML-MCP Server, you can try the following steps:

Check log files: View the log files in the 'logs' directory for error details
Verify Dependency Installation: Ensure that all dependencies are installed correctly
Check network connection: Ensure that PlantUML server (www.plantuml.com) can be accessed
Check output directory permissions: Ensure that the program has permission to write to the 'output' directory

Common problems and solutions:

Unable to generate UML diagram: Check for error messages in the log, which may be due to network issues or temporary unavailability of PlantUML server
Image not saved locally: Check if the 'output' directory exists and has write permission
MCP server cannot start: Check the log file to ensure there are no port conflicts or other program errors

Contribution

Welcome to contribute code, report issues, or provide improvement suggestions! Please participate in project development through GitHub Issues or Pull Requests.

License

This project adopts the MIT license. Please refer to the LICENSE document for details.

Integrations