mcp-server

mcp-server

作成者：Zack Chaffee A20478873

HDF5 ファイル操作と Slurm ジョブ管理のための Model Coupling Protocol（MCP）機能を実装するサーバーです。

機能

HDF5 ファイル操作：

• データセットの読み取り
• ファイルの内容の一覧表示

Slurm ジョブ管理：

• ジョブの送信
• ジョブのステータスの確認

ノードのハードウェア操作

• CPU 情報の取得
• メモリ情報の取得
• ディスク情報の取得
• システム情報の包括的な取得

圧縮操作

• gzip または zlib による文字列データの圧縮
• gzip または zlib によるファイルの圧縮
• データの解凍

初期化

このリポジトリをクローンしたら、ディレクトリに移動します。

続いて仮想環境を作成し、すべての依存関係をインストールします。

uv venv .venv
source .venv/bin/activate  # On Windows: .venv\Scripts\activate
uv pip install -e .
uv pip install -e ".[test]"

実行

サーバーを起動するには、次を実行します。

python -m src.server

サーバーは自動的に http://localhost:8000 で起動します。

エンドポイント

• POST /mcp：MCP リクエストのメインエンドポイント
• GET /health：ヘルスチェックエンドポイント

例：

import httpx

async with httpx.AsyncClient() as client:
    # Read a dataset
    response = await client.post("http://localhost:8000/mcp", json={
        "capability": "hdf5",
        "action": "read_dataset",
        "parameters": {
            "file_path": "/path/to/data.h5",
            "dataset_path": "/path/to/dataset"
        }
    })
    
    # List contents
    response = await client.post("http://localhost:8000/mcp", json={
        "capability": "hdf5",
        "action": "list_contents",
        "parameters": {
            "file_path": "/path/to/data.h5",
            "group_path": "/"
        }
    })

curl -X POST http://localhost:8000/mcp \
  -H "Content-Type: application/json" \
  -d '{
    "jsonrpc": "2.0",
    "method": "mcp/listTools", 
    "params": {}, 
    "id": "1"
  }'

テスト

テストを実行するには、次を実行します。

pytest

カバレッジ付きでテストするには、次を実行します。

pytest --cov=src