快適なMCPサーバー

FastMCP フレームワークを使用して、リモート Comfy サーバー経由のプロンプトに基づいて画像を生成するサーバー。

概要

このスクリプトは、FastMCPフレームワークを使用してサーバーを設定し、指定されたワークフローを使用してプロンプトに基づいて画像を生成します。リモートのComfyサーバーと連携してプロンプトを送信し、生成された画像を取得します。

前提条件

• Python 用のuvパッケージとプロジェクト マネージャー。
• Comfy UIからエクスポートされたワークフローファイル。このコードには、ここでは参照用としてのみ使用されるサンプルのFlux-Dev-ComfyUI-Workflow.jsonが含まれています。ワークフローからエクスポートし、環境変数を適切に設定する必要があります。

ローカル開発に必要なパッケージをインストールできます。

構成

次の環境変数を設定します。

• COMFY_URL Comfy サーバーの URL を指します。
• COMFY_WORKFLOW_JSON_FILE 、comfyui ワークフローの API エクスポート json ファイルの絶対パスを指します。
• PROMPT_NODE_IDテキストプロンプトノードの ID に設定します。
• OUTPUT_NODE_ID最終画像を含む出力ノードの ID に設定します。
• OUTPUT_MODE``urlまたはfileに設定して、必要な出力を選択します。

オプションとして、 Ollamaサーバーが実行中の場合は、プロンプトを生成するためにサーバーに接続できます。

• OLLAMA_API_BASE ollama が実行されている URL に設定します。
• PROMPT_LLMプロンプト生成用の ollama でホストされているモデルの名前に設定します。

例：

使用法

Comfy MCP Server は次のコマンドで起動できます。

Claude デスクトップ設定の例

機能性

generate_image(prompt: str, ctx: Context) -> Image | str

この関数は、指定されたプロンプトを使用して画像を生成します。以下の手順に従います。

1. すべての環境変数が設定されているかどうかを確認します。
2. JSON ファイルからプロンプト テンプレートを読み込みます。
3. プロンプトを Comfy サーバーに送信します。
4. プロンプト処理のステータスをサーバーにポーリングします。
5. 生成された画像が準備できたら取得して返します。

generate_prompt(topic: str, ctx: Context) -> str

この関数は、指定されたトピックから包括的な画像生成プロンプトを生成します。

依存関係

• mcp : FastMCP サーバーをセットアップします。
• json : JSON データを処理します。
• urllib : HTTP リクエストを行うため。
• time : ポーリングに遅延を追加します。
• os : 環境変数にアクセスします。
• langchain : トピックから画像生成プロンプトを生成するためのシンプルな LLM プロンプト チェーンを作成します。
• langchain-ollama : LangChain の ollama 固有のモジュール用。

ライセンス

このプロジェクトは MIT ライセンスに基づいてライセンスされています - 詳細についてはLICENSEファイルを参照してください。