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）