Pyats Network Automation

🚀 pyATS MCP 服务器

本项目实现了一个基于模型的上下文协议（MCP）服务器，它封装了 Cisco pyATS 和 Genie 功能。通过 STDIO 使用 JSON-RPC 2.0 协议，该服务器实现了结构化、基于模型的网络设备交互，为网络设备管理提供了高效且安全的解决方案。

⚠️ 重要提示

此服务器不使用 HTTP 或 SSE，所有通信均通过 STDIN/STDOUT 进行，这使其非常适合安全嵌入式、容器化或 LangGraph 工具集成。

✨ 主要特性

• 可连接到 Cisco IOS/NX-OS 设备（设备需在 pyATS 测试台文件中定义）。
• 支持安全执行验证过的 CLI 命令，如 show、ping 等。
• 允许进行受控的配置更改。
• 可返回结构化（解析）或原始输出。
• 通过 tools/discover 和 tools/call 暴露一组预定义工具。
• 完全通过 STDIO 运行，以最小的表面面积实现最大可移植性。

📦 安装指南

设置测试台路径

export PYATS_TESTBED_PATH=/absolute/path/to/testbed.yaml

运行服务器

连续 STDIO 模式（默认）

python3 pyats_mcp_server.py

此模式会启动一个长期运行的进程，从标准输入读取 JSON-RPC 请求，并将响应写入标准输出。

一次性模式

echo '{"jsonrpc": "2.0", "id": 1, "method": "tools/discover"}' | python3 pyats_mcp_server.py --oneshot

该模式用于处理单个 JSON-RPC 请求并退出。

Docker 支持

构建容器

docker build -t pyats-mcp-server .

运行容器（STDIO 模式）

docker run -it --rm -v $(pwd):/app -e PYATS_TESTBED_PATH=/app/testbed.yaml pyats-mcp-server

⚠️ 重要提示

测试台文件 testbed.yaml 与 Cisco DevNet 的 Cisco Modeling Labs (CML) Sandbox 兼容！

💻 使用示例

快速开始

1. 确保已安装 Python 和 Docker。
2. 克隆此仓库：

git clone https://github.com/your-repository.git

3. 运行服务器：

python3 pyats_mcp_server.py

或者以 Docker 容器形式运行：

docker run -it --rm -v $(pwd):/app -e PYATS_TESTBED_PATH=/app/testbed.yaml pyats-mcp-server

MCP 工具

以下工具可供使用：

LangGraph 集成

可以在 LangGraph 中进行如下配置：

from pyats.topology import loader

# 加载测试台文件
testbed = loader.load("testbed.yaml")

# 创建 MCP 服务器实例
mcp_server = testbed.devices["iosxe"].connections.mcp

# 添加到 LangGraph 管道
graph.add_node(mcp_server, label="MCP Server")

完整的 LangGraph 图配置示例

from pyats.topology import loader
from langgraph.core.graph import Graph

# 加载测试台文件
testbed = loader.load("testbed.yaml")

# 创建 MCP 服务器实例
mcp_server = testbed.devices["iosxe"].connections.mcp

# 创建 LangGraph 管理器
graph = Graph()

# 添加节点和边
graph.add_node(mcp_server, label="MCP Server")

🔧 技术细节

项目结构

.
├── pyats_mcp_server.py   # MCP 服务器主程序
├── testbed.yaml         # 测试台配置文件，包含 Cisco 设备连接信息
└── Dockerfile          # Docker 镜像构建文件

MCP 服务器配置

可以在以下 JSON 格式中配置 MCP 服务器：

{
  "devices": {
    "iosxe": {
      "hostname": "192.168.1.1",
      "username": "admin",
      "password": "password"
    }
  }
}

安全特性

• 所有输入均经过验证和 sanitization。
• 阻止执行不安全命令，如 reload 或 shutdown。
• 支持 SSL/TLS 加密通信（即将推出）。

📄 许可证

文档未提及相关许可证信息。

✍️ 作者

John Capobianco，Selector AI 的产品营销布道者，《自动化你的网络》的作者。如有任何问题，可联系作者，作者乐意提供帮助，同时作者还计划添加一个 LangGraph 图配置示例、配套客户端脚本以及实现 CI/CD 集成（例如 GitHub Actions）。