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

📦 安装指南

环境准备

本项目依赖 Python 和 Docker，请确保已安装。

克隆仓库

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

💻 使用示例

基础用法

设置测试台路径

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

📚 详细文档

测试台文件结构

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

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

MCP 工具

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

安全特性

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

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 的产品营销布道者
• 作品：《自动化你的网络》的作者

如有任何问题，请联系我，我很乐意提供帮助，同时也欢迎提出以下方面的需求：

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