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 を完全に使用して実行され、最小限のサーフェスエリアで最大の移植性を実現

📦 インストール

テストベッドパスの設定

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

⚠️ 重要提示

testbed.yaml ファイルは Cisco DevNet の Cisco Modeling Labs (CML) Sandbox と互換性があります！

💻 使用例

基本的な使用法

テストベッドパスの設定

# テストベッドパスの設定
export PYATS_TESTBED_PATH=/absolute/path/to/testbed.yaml

サーバーの起動（連続 STDIO モード）

# 連続 STDIO モードでサーバーを起動
python3 pyats_mcp_server.py

サーバーの起動（ワンショットモード）

# ワンショットモードでサーバーを起動
echo '{"jsonrpc": "2.0", "id": 1, "method": "tools/discover"}' | python3 pyats_mcp_server.py --oneshot

高度な使用法

Docker でのサーバーの起動

# Docker コンテナを構築
docker build -t pyats-mcp-server .
# Docker コンテナを 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 サーバーの設定

MCP サーバーは以下の JSON 形式で設定できます。

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

MCP ツール

以下のツールが利用可能です。

LangGraph 統合

LangGraph では以下のように構成できます。

from pyats.topology import loader

# テストベッドファイルを読み込む
testbed = loader.load("testbed.yaml")

# MCP サーバーインスタンスを作成
mcp_server = testbed.devices["iosxe"].connections.mcp

# LangGraph パイプラインに追加
graph.add_node(mcp_server, label="MCP Server")

セキュリティ機能

• すべての入力は検証とサニタイズが行われます。
• 不安全なコマンド（reload や shutdown）の実行はブロックされます。
• SSL/TLS 暗号化通信のサポート（近日実装予定）

✍️ 作者

John Capobianco

Selector AI の製品マーケティングブリーダー

『Automate Your Network』の著者

何か質問があれば、以下の追加事項を提供することができます。

• LangGraph グラフ構成の例を追加
• クライアントスクリプトの提供
• CI/CD 統合（GitHub Actions など）

お気軽にお問い合わせください！