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 -i --rm \
  -e PYATS_TESTBED_PATH=/app/testbed.yaml \
  -v /your/testbed/folder:/app \
  pyats-mcp-server

🧠 可用 MCP 工具

工具 描述

run_show_command 执行带安全性的 show 命令，可选解析

run_ping_command 执行 ping 测试并返回解析或原始结果

apply_configuration 应用受控的配置命令（支持多行）

learn_config 获取运行配置（show run brief）

learn_logging 获取系统日志（show logging last 250）

所有输入均通过 Pydantic 模式进行验证，以确保安全性和一致性。

🤖 LangGraph 集成

将 MCP 服务器作为工具节点添加到您的 LangGraph 管道中，如下所示：

("pyats-mcp", ["python3", "pyats_mcp_server.py", "--oneshot"],)

✍️ 作者

John Capobianco

Selector AI 的产品营销传教士

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

如果您有任何问题，请随时告知，我将乐于添加：

一个样本 LangGraph 图配置

配套客户端脚本

CI/CD 集成（例如 GitHub Actions）

我很乐意提供帮助！

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