ollama-MCP-server

Ollamaと通信するModel Context Protocol (MCP) サーバー

概要

このMCPサーバーは、ローカルのOllama LLMインスタンスとMCP互換アプリケーションの間でシームレスな統合を可能にし、高度なタスク分解、評価、ワークフロー管理を提供します。

主な機能:

複雑な問題のタスク分解
結果の評価と検証
Ollamaモデルの管理と実行
MCPプロトコルによる標準化された通信
高度なエラー処理と詳細なエラーメッセージ
パフォーマンス最適化（コネクションプーリング、LRUキャッシュ）

コンポーネント

リソース

サーバーは以下のリソースを実装しています:

task:// - 個別のタスクにアクセスするためのURIスキーム
result:// - 評価結果にアクセスするためのURIスキーム
model:// - 利用可能なOllamaモデルにアクセスするためのURIスキーム

各リソースには、最適なLLMとの対話のための適切なメタデータとMIMEタイプが設定されています。

プロンプトとツールの関係

MCPサーバーでは、プロンプトとツールは密接に関連していますが、異なる役割を持っています。

プロンプト：LLMに特定の思考方法や構造を提供するスキーマ（Schema）のような役割
ツール：実際にアクションを実行するハンドラー（Handler）のような役割

各ツールには対応するスキーマ（プロンプト）が必要であり、これによりLLMの思考能力と実際のシステム機能を効果的に連携させることができます。

プロンプト

サーバーはいくつかの特殊なプロンプトを提供します:

decompose-task - 複雑なタスクを管理しやすいサブタスクに分解
- タスクの説明と粒度レベルのオプションパラメータを取得
- 依存関係と推定複雑性を含む構造化された内訳を返す
evaluate-result - 指定された基準に対してタスク結果を分析
- 結果の内容と評価パラメータを取得
- スコアと改善提案を含む詳細な評価を返す

ツール

サーバーはいくつかの強力なツールを実装しています:

add-task
- 必須パラメータ: name (文字列), description (文字列)
- オプションパラメータ: priority (数値), deadline (文字列), tags (配列)
- システムに新しいタスクを作成し、その識別子を返す
- 対応するスキーマ: タスク作成のためのデータ検証スキーマ
decompose-task
- 必須パラメータ: task_id (文字列), granularity (文字列: "high"|"medium"|"low")
- オプションパラメータ: max_subtasks (数値)
- Ollamaを使用して複雑なタスクを管理可能なサブタスクに分解
- 対応するスキーマ: 上記のdecompose-taskプロンプト
evaluate-result
- 必須パラメータ: result_id (文字列), criteria (オブジェクト)
- オプションパラメータ: detailed (ブール値)
- 指定された基準に対して結果を評価し、フィードバックを提供
- 対応するスキーマ: 上記のevaluate-resultプロンプト
run-model
- 必須パラメータ: model (文字列), prompt (文字列)
- オプションパラメータ: temperature (数値), max_tokens (数値)
- 指定されたパラメータでOllamaモデルを実行
- 対応するスキーマ: Ollamaモデル実行パラメータの検証スキーマ

新機能と改善点

拡張エラー処理

サーバーは、より詳細で構造化されたエラーメッセージを提供します。これにより、クライアントアプリケーションはエラーをより効果的に処理できます。エラーレスポンスの例:

{
  "error": {
    "message": "Task not found: task-123",
    "status_code": 404,
    "details": {
      "provided_id": "task-123"
    }
  }
}

パフォーマンス最適化

コネクションプーリング: 共有HTTP接続プールを使用することで、リクエストのパフォーマンスが向上し、リソース使用率が低減されます。
LRUキャッシュ: 同一または類似のリクエストに対する応答をキャッシュすることで、レスポンス時間が短縮され、Ollamaサーバーの負荷が軽減されます。

これらの設定は config.py で調整できます:

# パフォーマンス関連設定
cache_size: int = 100                 # キャッシュに保存する最大エントリ数
max_connections: int = 10             # 同時接続の最大数
max_connections_per_host: int = 10    # ホストごとの最大接続数
request_timeout: int = 60             # リクエストタイムアウト（秒）

モデル指定機能

概要

Ollama-MCP-Serverは、複数の方法でOllamaモデルを指定できる柔軟な機能を提供します。

モデル指定の優先順位

モデルは以下の優先順位で指定されます：

ツール呼び出し時のパラメータ (model パラメータ)
MCP設定ファイルの env セクション
環境変数 (OLLAMA_DEFAULT_MODEL)
デフォルト値 (llama3)

MCP設定ファイルを使ったモデル指定

Claude Desktopなどのクライアントで使用する場合、MCP設定ファイルを使用してモデルを指定できます：

{
  "mcpServers": {
    "ollama-MCP-server": {
      "command": "python",
      "args": [
        "-m",
        "ollama_mcp_server"
      ],
      "env": [
        {"model": "llama3:latest"}
      ]
    }
  }
}

利用可能なモデルの確認

サーバー起動時に、設定されたモデルが存在するかチェックされます。モデルが見つからない場合は警告ログが出力されます。また、run-modelツールは利用可能なモデル一覧を返すため、ユーザーは有効なモデルを選択できます。

エラーハンドリングの改善

指定したモデルが存在しない場合や通信エラーが発生した場合、詳細なエラーメッセージが提供されます。エラーメッセージには利用可能なモデル一覧が含まれるため、ユーザーは素早く問題を解決できます。

テスト

プロジェクトには包括的なテストスイートが含まれています:

ユニットテスト: 個々のコンポーネントの機能をテスト
統合テスト: エンドツーエンドのワークフローをテスト

テストを実行するには:

# すべてのテストを実行
python -m unittest discover

# 特定のテストを実行
python -m unittest tests.test_integration

設定

環境変数

OLLAMA_HOST=http://localhost:11434
DEFAULT_MODEL=llama3
LOG_LEVEL=info

Ollamaのセットアップ

Ollamaがインストールされ、適切なモデルで実行されていることを確認してください:

# Ollamaをインストール（まだインストールされていない場合）
curl -fsSL https://ollama.com/install.sh | sh

# 推奨モデルをダウンロード
ollama pull llama3
ollama pull mistral
ollama pull qwen2

クイックスタート

インストール

pip install ollama-mcp-server

Claude Desktop設定

MacOS

パス: ~/Library/Application\ Support/Claude/claude_desktop_config.json

Windows

パス: %APPDATA%/Claude/claude_desktop_config.json

開発/未公開サーバーの設定

"mcpServers": {
  "ollama-MCP-server": {
    "command": "uv",
    "args": [
      "--directory",
      "/path/to/ollama-MCP-server",
      "run",
      "ollama-MCP-server"
    ],
    "ENV":["model":"deepseek:r14B"]
  }
}

公開サーバーの設定

"mcpServers": {
  "ollama-MCP-server": {
    "command": "uvx",
    "args": [
      "ollama-MCP-server"
    ]
  }
}

使用例

タスク分解

複雑なタスクを管理可能なサブタスクに分解するには:

result = await mcp.use_mcp_tool({
    "server_name": "ollama-MCP-server",
    "tool_name": "decompose-task",
    "arguments": {
        "task_id": "task://123",
        "granularity": "medium",
        "max_subtasks": 5
    }
})

結果評価

特定の基準に対して結果を評価するには:

evaluation = await mcp.use_mcp_tool({
    "server_name": "ollama-MCP-server",
    "tool_name": "evaluate-result",
    "arguments": {
        "result_id": "result://456",
        "criteria": {
            "accuracy": 0.4,
            "completeness": 0.3,
            "clarity": 0.3
        },
        "detailed": true
    }
})

Ollamaモデルの実行

Ollamaモデルに対して直接クエリを実行するには:

response = await mcp.use_mcp_tool({
    "server_name": "ollama-MCP-server",
    "tool_name": "run-model",
    "arguments": {
        "model": "llama3",
        "prompt": "量子コンピューティングを簡単な言葉で説明してください",
        "temperature": 0.7
    }
})

開発

プロジェクトのセットアップ

リポジトリをクローン:

git clone https://github.com/yourusername/ollama-MCP-server.git
cd ollama-MCP-server

仮想環境を作成してアクティベート:

python -m venv venv
source venv/bin/activate  # Windowsの場合: venv\Scripts\activate

開発依存関係をインストール:

uv sync --dev --all-extras

ローカル開発

プロジェクトには便利な開発用スクリプトが含まれています：

サーバーの実行

./run_server.sh

オプション:

--debug: デバッグモードで実行（ログレベル: DEBUG）
--log=LEVEL: ログレベルを指定（DEBUG, INFO, WARNING, ERROR, CRITICAL）

テストの実行

./run_tests.sh