🚀 pyATS MCP 服務器

本項目實現了一個基於模型上下文協議（MCP）的服務器，它封裝了 Cisco pyATS 和 Genie 的功能。藉助該服務器，可通過 STDIO 運用 JSON - RPC 2.0 協議，實現對網絡設備的結構化、基於模型的交互。

🚀 快速開始

環境設置

設置您的測試床路徑：

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 請求，處理完成後即退出。

✨ 主要特性

• 設備連接：可連接到 Cisco IOS/NX - OS 設備，設備信息來自 pyATS 測試床。
• 命令執行：支持安全執行經過驗證的 CLI 命令，如 show、ping 等。
• 配置更改：允許進行受控的配置更改。
• 輸出返回：能返回結構化（解析）或原始輸出。
• 工具暴露：通過 tools/discover 和 tools/call 暴露一組定義良好的工具。
• 安全運行：完全通過 STDIO 運行，可實現最小的攻擊面和最大的可移植性。

📦 安裝指南

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

💻 使用示例

基礎用法

以下是一次性模式下執行 tools/discover 工具的示例：

echo '{"jsonrpc": "2.0", "id": 1, "method": "tools/discover"}' | python3 pyats_mcp_server.py --oneshot

高級用法

在 Docker 容器中運行服務器並指定測試床路徑：

docker run -i --rm \
  -e PYATS_TESTBED_PATH=/app/testbed.yaml \
  -v /your/testbed/folder:/app \
  pyats-mcp-server

📚 詳細文檔

可用 MCP 工具

所有輸入均通過 Pydantic 模式進行驗證，以確保安全性和一致性。

LangGraph 集成

將 MCP 服務器作為工具節點添加到您的 LangGraph 管道中：

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

🔧 技術細節

此服務器不使用 HTTP 或 SSE，所有通信均通過標準輸入輸出（STDIN/STDOUT）完成，這使其成為安全嵌入式、容器化或 LangGraph 工具集成的理想選擇。所有輸入均通過 Pydantic 模式進行驗證，保證了安全性和一致性。

📄 許可證

文檔中未提及許可證相關信息。

⚠️ 重要提示

此服務器不使用 HTTP 或 SSE，所有通信均通過標準輸入輸出（STDIN/STDOUT）完成。

✍️ 作者 John Capobianco，Selector AI 的產品營銷傳教士，《自動化你的網絡》的作者。

如果您有任何問題，請隨時告知，作者將樂於添加：

• 一個樣本 LangGraph 圖配置
• 配套客戶端腳本
• CI/CD 集成（例如 GitHub Actions）

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