Pywrdrb MCP

pywrdrb-mcp

MCP服务器，使LLM能够高效访问 Pywr DRB 水资源模型代码库。使用静态分析（Python ast)从源文件中提取结构和数据，而无需导入pywrdrb或其严重依赖关系。

快速开始

先决条件

• 紫外线 安装
• Pywr DRB源代码在本地克隆（ src/pywrdrb/ 启动时读取目录）

运行服务器

uv run python -m pywrdrb_mcp.server

运行测试

uv run pytest tests/ -v

配置

服务器通过静态分析（Python ast)在启动时。它从不导入或执行pywrdrb代码，因此pywrdrb的重依赖项（numpy、pandas、pywr等）是 不 必修的。

集 PYWRDRB_ROOT 如果您的Pywr DRB签出不在默认的同级位置：

# Linux/macOS
export PYWRDRB_ROOT=/path/to/Pywr-DRB/src/pywrdrb

# Windows (PowerShell)
$env:PYWRDRB_ROOT = "C:\path\to\Pywr-DRB\src\pywrdrb"

客户端设置

克劳德代码（CLI）

claude mcp add --transport stdio pywrdrb-mcp -- \
  uv run --directory /path/to/pywrdrb-mcp python -m pywrdrb_mcp.server

克劳德代码（VS代码扩展）

添加 .claude/settings.json （项目或用户级别）：

{
  "mcpServers": {
    "pywrdrb-mcp": {
      "type": "stdio",
      "command": "uv",
      "args": [
        "run",
        "--directory", "/path/to/pywrdrb-mcp",
        "python", "-m", "pywrdrb_mcp.server"
      ]
    }
  }
}

克劳德桌面版

编辑Claude Desktop配置文件：

• 窗户： %APPDATA%\Claude\claude_desktop_config.json
• macOS： ~/Library/Application Support/Claude/claude_desktop_config.json

（通过克劳德桌面打开： 设置>开发人员>编辑配置)

{
  "mcpServers": {
    "pywrdrb-mcp": {
      "command": "uv",
      "args": [
        "run",
        "--directory", "/path/to/pywrdrb-mcp",
        "python", "-m", "pywrdrb_mcp.server"
      ],
      "env": {
        "PYWRDRB_ROOT": "/path/to/Pywr-DRB/src/pywrdrb"
      }
    }
  }
}

Windows路径： 使用正斜杠(C:/Users/...)或逃过一劫(C:\\Users\\...)在JSON中。

保存后， 完全退出并重新启动 Claude Desktop（系统托盘>退出，而不仅仅是关闭窗口）。

Claude桌面兼容性说明

在实践中，这意味着 工具是主要界面 在克劳德桌面上。领域知识资源（FFMP规则、STARFIT规则、洪水操作）和 how_to_* 提示有效，但需要手动用户操作才能调用。在Claude Code中，资源和提示更加无缝地集成在一起。

包含什么

工具（14）

资源（22）

拓扑和数据

API参考资料

领域知识指南

项目

参数化模板

提示（6）

所有提示都使用 how_to_* 命名约定，表明它们是教学指南。

建筑

pywrdrb-mcp/
├── src/pywrdrb_mcp/
│   ├── server.py          # FastMCP entry point
│   ├── config.py          # PYWRDRB_ROOT path
│   ├── index/             # AST-based static analysis engine
│   │   ├── builder.py     # PywrDRBIndex — topology, lists, capacity, rating curves
│   │   ├── ast_utils.py   # Safe AST extraction utilities
│   │   └── file_utils.py  # File system utilities
│   ├── tools/             # 14 MCP tools (8 modules)
│   ├── resources/         # 22 MCP resources
│   ├── prompts/           # 6 MCP prompt templates
│   └── content/           # Hand-written domain knowledge (7 markdown files)
└── tests/

服务器在启动时通过以下方式从Pywr DRB源代码树读取 ast.parse() 和 ast.literal_eval() --它从不导入或执行pywrdrb代码。