Cluster Execution MCP Server

Cluster Execution MCP Server

Enables cluster-aware command execution and automatic task routing across distributed nodes based on system load, architecture, and OS requirements. It supports parallel execution, remote node management via SSH, and dynamic load balancing for agentic workflows.

Category
访问服务器

README

Cluster Execution MCP Server

Cluster-aware command execution for distributed task routing across the AGI agentic cluster.

Version: 0.2.0

Features

  • Automatic task routing: Commands routed to optimal nodes based on load, capabilities, and requirements
  • Multi-node support: macpro51 (Linux x86_64), mac-studio (macOS ARM64), macbook-air (macOS ARM64), inference node
  • Dynamic IP resolution: mDNS, DNS, and fallback methods with caching
  • Security hardened: No shell injection, environment-based configuration, command validation
  • SSH connectivity verification: Retry logic with configurable timeouts
  • Parallel execution: Distribute commands across cluster for maximum throughput

Installation

cd /mnt/agentic-system/mcp-servers/cluster-execution-mcp
pip install -e .

# For development:
pip install -e ".[dev]"

Configuration

Claude Code Configuration

Add to ~/.claude.json:

{
  "mcpServers": {
    "cluster-execution": {
      "command": "/mnt/agentic-system/.venv/bin/python3",
      "args": ["-m", "cluster_execution_mcp.server"]
    }
  }
}

Environment Variables

All configuration is externalized via environment variables:

Variable Default Description
CLUSTER_SSH_USER marc SSH username for remote execution
CLUSTER_SSH_TIMEOUT 5 SSH connection timeout (seconds)
CLUSTER_SSH_CONNECT_TIMEOUT 2 Initial SSH connect timeout (seconds)
CLUSTER_SSH_RETRIES 2 Number of SSH retry attempts
CLUSTER_CPU_THRESHOLD 40 CPU usage % threshold for offloading
CLUSTER_LOAD_THRESHOLD 4 Load average threshold for offloading
CLUSTER_MEMORY_THRESHOLD 80 Memory usage % threshold for offloading
CLUSTER_CMD_TIMEOUT 300 Command execution timeout (seconds)
CLUSTER_STATUS_TIMEOUT 5 Status check timeout (seconds)
CLUSTER_IP_CACHE_TTL 300 IP resolution cache TTL (seconds)
CLUSTER_GATEWAY 192.168.1.1 Gateway IP for route detection
CLUSTER_DNS 8.8.8.8 DNS server for IP detection
AGENTIC_SYSTEM_PATH /mnt/agentic-system Base path for databases

Node Configuration

Node hostnames and IPs can be customized:

Variable Default Description
CLUSTER_MACPRO51_HOST macpro51.local Mac Pro hostname
CLUSTER_MACPRO51_IP 192.168.1.183 Mac Pro fallback IP
CLUSTER_MACSTUDIO_HOST Marcs-Mac-Studio.local Mac Studio hostname
CLUSTER_MACSTUDIO_IP 192.168.1.16 Mac Studio fallback IP
CLUSTER_MACBOOKAIR_HOST Marcs-MacBook-Air.local MacBook Air hostname
CLUSTER_MACBOOKAIR_IP 192.168.1.172 MacBook Air fallback IP
CLUSTER_INFERENCE_HOST completeu-server.local Inference node hostname
CLUSTER_INFERENCE_IP 192.168.1.186 Inference node fallback IP

MCP Tools

Tool Description
cluster_bash Execute bash commands with automatic cluster routing
cluster_status Get current cluster state and load distribution
offload_to Explicitly route command to specific node
parallel_execute Run multiple commands in parallel across nodes

Usage Examples

Automatic Routing

# Heavy commands auto-route to least loaded node
result = await cluster_bash("make -j8 all")

# Simple commands run locally
result = await cluster_bash("ls -la")

Force Specific Requirements

# Force Linux execution
result = await cluster_bash("docker build .", requires_os="linux")

# Force x86_64 architecture
result = await cluster_bash("cargo build", requires_arch="x86_64")

Explicit Node Routing

# Run on Linux builder
result = await offload_to("podman run -it ubuntu:22.04", node_id="macpro51")

# Run on Mac Studio
result = await offload_to("swift build", node_id="mac-studio")

Parallel Execution

# Run tests across cluster
results = await parallel_execute([
    "pytest tests/unit/",
    "pytest tests/integration/",
    "pytest tests/e2e/"
])

Cluster Status

# Get cluster health before heavy operations
status = await cluster_status()
# Returns:
# {
#   "local_node": "macpro51",
#   "nodes": {
#     "macpro51": {"cpu_percent": 15.2, "memory_percent": 45.3, ...},
#     "mac-studio": {"cpu_percent": 8.1, "memory_percent": 32.1, ...},
#     ...
#   }
# }

Cluster Nodes

Node OS Arch Capabilities Specialties
macpro51 Linux x86_64 docker, podman, raid, nvme, compilation, testing, tpu compilation, testing, containerization, benchmarking
mac-studio macOS ARM64 orchestration, coordination, temporal, mlx-gpu, arduino orchestration, coordination, monitoring
macbook-air macOS ARM64 research, documentation, analysis research, documentation, mobile
inference macOS ARM64 ollama, inference, model-serving, llm-api ollama-inference, model-serving

Offload Patterns

Commands matching these patterns are automatically offloaded:

  • Build: make, cargo, npm, yarn, pnpm
  • Test: pytest, jest, mocha, test
  • Compile: gcc, g++, clang
  • Container: docker, podman, kubectl
  • File ops: rsync, scp, tar, zip, find, grep -r

Commands that stay local:

  • Simple: ls, pwd, cd, echo, cat, head, tail, which, type

Security

Shell Injection Prevention

All commands use subprocess.run() with list arguments where possible:

# SAFE: List arguments
subprocess.run(["ssh", "-o", "ConnectTimeout=5", f"{user}@{ip}", command])

# Complex shell commands are validated before execution

Command Validation

Commands are validated for dangerous patterns:

  • rm -rf /
  • rm -rf /*
  • > /dev/sda
  • Fork bombs
  • And more...

SSH Configuration

  • StrictHostKeyChecking=accept-new - Accept new hosts but verify returning hosts
  • BatchMode=yes - Non-interactive mode for scripting
  • Configurable timeouts and retries

Development

Running Tests

# Install dev dependencies
pip install -e ".[dev]"

# Run tests
pytest tests/ -v

# With coverage
pytest tests/ --cov=cluster_execution_mcp --cov-report=html

Project Structure

cluster-execution-mcp/
├── src/cluster_execution_mcp/
│   ├── __init__.py      # Package exports
│   ├── config.py        # Configuration, validation, node definitions
│   ├── router.py        # Task routing and IP resolution
│   └── server.py        # FastMCP server and tools
├── tests/
│   ├── conftest.py      # Pytest fixtures
│   ├── test_config.py   # Config module tests (29 tests)
│   ├── test_router.py   # Router module tests (21 tests)
│   └── test_server.py   # Server and tool tests (21 tests)
└── pyproject.toml       # Package configuration

CLI Interface

# Submit a command
cluster-router submit "make -j8 all"

# Check task status
cluster-router status <task_id>

# Show cluster status
cluster-router cluster-status

Monitoring

Check cluster health before operations:

User: "Show me cluster status"

Claude Code: cluster_status tool

Output:
  macpro51:
    CPU: 45.2%
    Memory: 18.3%
    Load: 3.21
    Status: healthy

  mac-studio:
    CPU: 22.1%
    Memory: 54.7%
    Load: 2.15
    Status: healthy

  macbook-air:
    CPU: 12.8%
    Memory: 38.2%
    Load: 1.03
    Status: healthy

Troubleshooting

MCP server not loading:

# Check config
cat ~/.claude.json | jq '.mcpServers["cluster-execution"]'

# Test server import
python3 -c "from cluster_execution_mcp.server import main; print('OK')"

Node unreachable:

# Test SSH connectivity
ssh marc@macpro51.local hostname
ssh marc@Marcs-Mac-Studio.local hostname

# Check with fallback IP
ssh marc@192.168.1.183 hostname

Commands timing out:

# Increase timeout via environment
export CLUSTER_CMD_TIMEOUT=600  # 10 minutes
export CLUSTER_SSH_TIMEOUT=10   # 10 seconds

Changelog

v0.2.0

  • New Features:

    • Proper package structure with pyproject.toml
    • Environment-based configuration (no hardcoded credentials)
    • Shared config module with validation functions
    • Retry logic for SSH connectivity
    • IP resolution caching with TTL
    • Inference node support
  • Security Improvements:

    • Eliminated shell injection vulnerabilities
    • Command validation for dangerous patterns
    • IP validation rejecting loopback/Docker/link-local
    • SSH host key handling (accept-new)
  • Code Quality:

    • Full type hints throughout codebase
    • Replaced bare except clauses with specific exceptions
    • Added comprehensive logging
    • 71 unit tests with mocking
  • Bug Fixes:

    • Fixed darwin/macos OS alias handling
    • Proper timeout handling in SSH operations
    • Better error messages for failed operations

v0.1.0

  • Initial release with basic cluster execution

License

MIT


Part of the AGI Agentic System

See also:

  • Node Chat MCP - Inter-node communication
  • Enhanced Memory MCP - Persistent memory with RAG
  • Agent Runtime MCP - Goals and task queue

推荐服务器

Baidu Map

Baidu Map

百度地图核心API现已全面兼容MCP协议,是国内首家兼容MCP协议的地图服务商。

官方
精选
JavaScript
Playwright MCP Server

Playwright MCP Server

一个模型上下文协议服务器,它使大型语言模型能够通过结构化的可访问性快照与网页进行交互,而无需视觉模型或屏幕截图。

官方
精选
TypeScript
Magic Component Platform (MCP)

Magic Component Platform (MCP)

一个由人工智能驱动的工具,可以从自然语言描述生成现代化的用户界面组件,并与流行的集成开发环境(IDE)集成,从而简化用户界面开发流程。

官方
精选
本地
TypeScript
Audiense Insights MCP Server

Audiense Insights MCP Server

通过模型上下文协议启用与 Audiense Insights 账户的交互,从而促进营销洞察和受众数据的提取和分析,包括人口统计信息、行为和影响者互动。

官方
精选
本地
TypeScript
VeyraX

VeyraX

一个单一的 MCP 工具,连接你所有喜爱的工具:Gmail、日历以及其他 40 多个工具。

官方
精选
本地
graphlit-mcp-server

graphlit-mcp-server

模型上下文协议 (MCP) 服务器实现了 MCP 客户端与 Graphlit 服务之间的集成。 除了网络爬取之外,还可以将任何内容(从 Slack 到 Gmail 再到播客订阅源)导入到 Graphlit 项目中,然后从 MCP 客户端检索相关内容。

官方
精选
TypeScript
Kagi MCP Server

Kagi MCP Server

一个 MCP 服务器,集成了 Kagi 搜索功能和 Claude AI,使 Claude 能够在回答需要最新信息的问题时执行实时网络搜索。

官方
精选
Python
e2b-mcp-server

e2b-mcp-server

使用 MCP 通过 e2b 运行代码。

官方
精选
Neon MCP Server

Neon MCP Server

用于与 Neon 管理 API 和数据库交互的 MCP 服务器

官方
精选
Exa MCP Server

Exa MCP Server

模型上下文协议(MCP)服务器允许像 Claude 这样的 AI 助手使用 Exa AI 搜索 API 进行网络搜索。这种设置允许 AI 模型以安全和受控的方式获取实时的网络信息。

官方
精选