Draw Architecture MCP Server

一个专业的架构图绘制 MCP 服务器，集成智谱AI大模型，能够根据文本描述自动生成 draw.io 格式的系统架构图。

功能特性

🤖 AI 驱动生成：集成智谱AI免费大模型，智能理解架构需求
🎨 智能架构图生成：根据文本描述自动生成专业的系统架构图
📊 Draw.io 兼容：生成标准的 .drawio 格式文件，可直接导入 draw.io 编辑
🏗️ 多种架构模式：支持微服务、分层架构、事件驱动等多种架构模式
🎯 专业提示词：内置专业的架构设计提示词模板
🔧 MCP 协议：基于 Model Context Protocol，可与支持 MCP 的 AI 助手集成
💰 免费使用：使用智谱AI的免费额度，无需付费即可体验

安装配置

1. 环境要求

Python 3.8+
支持 MCP 的 AI 客户端（如 Claude Desktop、Cline 等）

2. 安装依赖

# 克隆或下载项目
cd draw_architecture_mcp

# 安装依赖
pip install -r requirements.txt

3. 配置智谱AI API Key

获取API Key

访问智谱AI开放平台
注册账号并登录
在控制台获取API Key（新用户有免费额度）

配置环境变量

# 复制环境变量模板
cp .env.example .env

# 编辑 .env 文件，填入你的API Key
echo "ZHIPUAI_API_KEY=your_actual_api_key_here" > .env

4. 配置 MCP 客户端

Claude Desktop 配置

编辑 Claude Desktop 的配置文件：

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

添加以下配置：

{
  "mcpServers": {
    "draw-architecture": {
      "command": "python3",
      "args": ["/path/to/draw_architecture_mcp/mcp_server.py"],
      "env": {
        "PYTHONPATH": "/path/to/draw_architecture_mcp",
        "ZHIPUAI_API_KEY": "your_api_key_here"
      }
    }
  }
}

Cline 配置

在 Cline 的设置中添加 MCP 服务器：

{
  "mcpServers": {
    "draw-architecture": {
      "command": "python3",
      "args": ["/path/to/draw_architecture_mcp/mcp_server.py"],
      "env": {
        "ZHIPUAI_API_KEY": "your_api_key_here"
      }
    }
  }
}

4. 验证安装

重启 AI 客户端，在对话中询问是否可以使用架构图生成功能。如果配置正确，AI 助手应该能够访问以下工具：

generate_architecture_diagram: 生成架构图
get_architecture_prompt: 获取提示词模板
validate_drawio_file: 验证文件格式

使用方法

基本用法

在支持 MCP 的 AI 客户端中，你可以这样使用：

请帮我绘制一个电商系统的架构图，包括：
- 前端：Web应用、移动App
- 后端：用户服务、商品服务、订单服务
- 数据库：MySQL、Redis
- 消息队列：Kafka

高级用法

绘制一个微服务架构的在线教育平台：

前端层：
- React Web应用
- Flutter移动应用
- 管理后台

网关层：
- API网关（Kong）
- 负载均衡（Nginx）

业务服务层：
- 用户认证服务
- 课程管理服务
- 视频播放服务
- 支付服务
- 消息通知服务

数据层：
- MySQL（用户数据、课程数据）
- MongoDB（视频元数据）
- Redis（缓存、会话）
- Elasticsearch（搜索）

基础设施：
- Docker容器化
- Kubernetes编排
- 监控（Prometheus + Grafana）
- 日志（ELK Stack）

技术架构

AI 模型集成

智谱AI GLM-4-Flash：免费的大语言模型，专门优化架构图生成
专业提示词模板：内置完整的架构设计指导模板
智能回退机制：AI调用失败时自动回退到规则引擎

生成流程

需求分析：解析用户的架构描述
模板整合：将用户需求与专业提示词模板结合
AI生成：调用智谱AI生成完整的draw.io XML代码
格式验证：确保生成的XML符合draw.io标准
文件保存：保存为.drawio格式文件

支持的架构模式

分层架构（Layered Architecture）
微服务架构（Microservices）
事件驱动架构（Event-Driven）
六边形架构（Hexagonal）
CQRS架构
服务网格（Service Mesh）

AI 助手会使用 generate_architecture_diagram 工具生成对应的 draw.io 文件。

API 参考

工具列表

`generate_architecture_diagram`

生成 draw.io 格式的架构图。

参数:

description (string, 必需): 架构描述
diagram_name (string, 可选): 图表名称，默认为"系统架构图"
output_file (string, 可选): 输出文件路径

示例:

{
  "description": "微服务架构，包含用户服务、订单服务、支付服务",
  "diagram_name": "电商系统架构",
  "output_file": "./ecommerce_architecture.drawio"
}

资源列表

`prompt://draw-architecture`

提供完整的 draw.io 架构图绘制提示词模板。

项目结构

draw_architecture_mcp/
├── mcp_server.py              # MCP 服务器主文件
├── draw_architecture_prompt.md # 提示词模板
├── requirements.txt           # Python 依赖
├── mcp_config.json           # MCP 配置示例
├── README.md                 # 项目文档
└── examples/                 # 示例文件
    ├── sample_architecture.drawio
    └── usage_examples.md

开发指南

本地开发

# 安装开发依赖
pip install -r requirements.txt

# 运行测试
pytest tests/

# 代码格式化
black mcp_server.py

# 类型检查
mypy mcp_server.py

扩展功能

添加新的图表类型: 在 generate_drawio_xml 函数中添加新的模板
增强验证功能: 在 validate_drawio_file 工具中添加更多检查项
支持更多格式: 添加导出为 PNG、SVG 等格式的功能

常见问题

Q: 生成的文件无法在 draw.io 中打开？

A: 使用 validate_drawio_file 工具检查文件格式，常见问题包括：

缺少 XML 声明
HTML 标签未正确转义
XML 结构不完整

Q: 如何自定义架构图样式？

A: 修改 mcp_server.py 中的 generate_drawio_xml 函数，调整颜色、字体、布局等样式参数。

Q: 支持哪些 AI 客户端？

A: 支持所有实现 MCP 协议的客户端，包括：

Claude Desktop
Cline (VS Code 扩展)
其他支持 MCP 的 AI 工具

贡献指南

Fork 项目
创建功能分支 (git checkout -b feature/amazing-feature)
提交更改 (git commit -m 'Add amazing feature')
推送到分支 (git push origin feature/amazing-feature)
创建 Pull Request

许可证

MIT License - 详见 LICENSE 文件

更新日志

v1.0.0 (2025-06-14)

初始版本发布
支持基础架构图生成
提供文件验证功能
完整的 MCP 协议实现

联系方式

项目主页: GitHub Repository
问题反馈: Issues
功能建议: Discussions

Draw Architecture