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 运行，以最小的表面面积实现最大可移植性

🚀 使用说明

1. 设置测试台路径

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

2. 运行服务器

连续 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

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

✍️ 作者

John Capobianco

Selector AI 的产品营销布道者

《自动化你的网络》的作者

如有任何问题，请联系我：

• 添加一个 LangGraph 图配置示例
• 配套客户端脚本
• CI/CD 集成（例如 GitHub Actions）

我很乐意提供帮助！

测试台文件结构

项目包含以下文件和目录：

.
├── README.md            # 项目说明文档
├── pyats_mcp_server.py   # MCP 服务器主程序
└── testbed.yaml         # 测试台配置文件，与 Cisco Modeling Labs 兼容

使用方法

快速开始

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")

安全特性

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

项目结构

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

MCP 服务器配置

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

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

示例

以下是一个完整的 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")

作者信息

John Capobianco

• 职位：Selector AI 的产品营销布道者
• 作品：《自动化你的网络》的作者

如需帮助，请随时联系我！