pyATS MCP 服务器

本项目实现了一个模型上下文协议 (MCP) 服务器，它封装了 Cisco pyATS 和 Genie 的功能。它支持使用 JSON-RPC 2.0 协议通过 STDIO 与网络设备进行结构化的、模型驱动的交互。

🚨 此服务器不使用 HTTP 或 SSE。所有通信都通过 STDIN/STDOUT（标准输入/输出）完成，使其非常适合安全、嵌入式、容器化或基于 LangGraph 的工具集成。

🔧 功能

连接到 pyATS 测试平台中定义的 Cisco IOS/NX-OS 设备

支持安全执行经过验证的 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

启动一个长时间运行的进程，该进程从 stdin 读取 JSON-RPC 请求并将响应写入 stdout。

单次模式

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"], "tools/discover", "tools/call")

名称: pyats-mcp

命令: python3 pyats_mcp_server.py --oneshot

发现方法: tools/discover

调用方法: tools/call

基于 STDIO 的通信确保与 LangGraph 的工具调用模型紧密集成，而无需打开 HTTP 端口或公开 REST 端点。

📜 示例请求

发现工具

{
  "jsonrpc": "2.0",
  "id": 1,
  "method": "tools/discover"
}

运行 Show 命令

{
  "jsonrpc": "2.0",
  "id": 2,
  "method": "tools/call",
  "params": {
    "name": "run_show_command",
    "arguments": {
      "device_name": "router1",
      "command": "show ip interface brief"
    }
  }
}

🔒 安全特性

使用 Pydantic 进行输入验证

阻止不安全的命令，如 erase、reload、write

防止管道/重定向滥用（例如，| include、>、copy 等）

优雅地处理解析回退和错误

📁 项目结构

.
├── pyats_mcp_server.py     # 具有 JSON-RPC 和 pyATS 集成的 MCP 服务器
├── Dockerfile              # Docker 容器定义
├── testbed.yaml            # pyATS 测试平台（用户提供）
└── README.md               # 此文件

✍️ 作者

John Capobianco

Selector AI 产品营销布道者

Automate Your Network 作者

如果您想添加以下内容，请告诉我：

示例 LangGraph 图配置

配套客户端脚本

CI/CD 集成（例如，GitHub Actions）

很乐意帮忙！

testbed.yaml 文件适用于 Cisco DevNet Cisco Modeling Labs (CML) 沙箱！