Skip to main content

UML diagram generation tool based on MCP (Model Context Protocol)

Project description

UML-MCP-Server

English | 中文

smithery badge

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

  1. Clone repository:
git clone https://github.com/yourusername/UML-MCP-Server.git
cd UML-MCP-Server
  1. 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
  1. 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:

  1. Ensure that the latest version of Cursor IDE (which supports MCP functionality) is installed.

  2. 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
  1. 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
  1. Save the configuration file and restart the Cursor.

  2. 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:

1. Understand the certification process of the project
2. Generate UML code for the authentication process and generate a flowchart through UML-MCP Server
3. Attention: "output_dir": "/Users/my-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:

  1. PlantUML Code - You can copy this code to use in other PlantUML tools
  2. PlantUML URL - You can open this URL in a browser to view the generated UML diagram
  3. 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:

  1. Check log files: View the log files in the 'logs' directory for error details
  2. Verify Dependency Installation: Ensure that all dependencies are installed correctly
  3. Check network connection: Ensure that PlantUML server (www.plantuml.com) can be accessed
  4. 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.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

iflow_mcp_uml_mcp_server-0.1.0.tar.gz (11.4 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

iflow_mcp_uml_mcp_server-0.1.0-py3-none-any.whl (21.0 kB view details)

Uploaded Python 3

File details

Details for the file iflow_mcp_uml_mcp_server-0.1.0.tar.gz.

File metadata

File hashes

Hashes for iflow_mcp_uml_mcp_server-0.1.0.tar.gz
Algorithm Hash digest
SHA256 ae7f9597998d700497672f65b1265c3805913c546b20c54359b1b2b42a248a49
MD5 5d3cc3d8453a4aa5279196b40a9a1265
BLAKE2b-256 2450b357d78199f97ea04c98605ea5e3874baa9537498d866bd2c6569590b818

See more details on using hashes here.

File details

Details for the file iflow_mcp_uml_mcp_server-0.1.0-py3-none-any.whl.

File metadata

File hashes

Hashes for iflow_mcp_uml_mcp_server-0.1.0-py3-none-any.whl
Algorithm Hash digest
SHA256 00fcf1ecc607614f7223ba3668a138639bb6cfb4ddf2f66284084af103f3540a
MD5 1350e6a84462f51237e327ff70289927
BLAKE2b-256 0bbf2b95a330eb14ee97e4258ae4f38c9461070b3ce6bb907eb708c51eda854b

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page