Image MCP Servers

图像 MCP 服务

本项目实现了一个 SSE MCP 服务，提供与 HuggingFace 和 Replicate API 交互的工具，重点支持图像生成和模型管理。它基于 MCP Core 框架构建，为 AI 模型交互提供标准化接口。下面将介绍如何自行托管！

功能

HuggingFace 集成

工具	描述
获取模型信息	获取 HuggingFace 上模型的详细信息
获取模型示例图像	从模型 README 中提取示例图像
获取 README	获取模型的 README 内容
搜索模型	在 HuggingFace 上搜索模型，并提供筛选选项
WhoAmI	获取当前 HuggingFace API 令牌的信息

Replicate 集成

工具	描述
生成图像	使用 Replicate 的文本生成图像模型创建图像
获取模型信息	获取 Replicate 上模型的详细信息
获取预测结果	检查预测状态并获取其输出
列出模型	列出 Replicate 上可用的模型，并可选择筛选
WhoAmI	获取当前 Replicate API 令牌的信息

开始使用

前提条件

• Rust（2024 版）
• Docker（可选，用于容器化部署）

安装

1. 克隆仓库：

   git clone https://github.com/yourusername/image-mcp-servers.git
   cd image-mcp-servers
   

2. 根据 .env.example 创建 .env 文件：

   cp .env.example .env
   

3. 将 API 令牌添加到 .env 文件中：

   HF_API_TOKEN="your_huggingface_token"
   REPLICATE_API_TOKEN="your_replicate_token"
   

4. 构建项目：

   cargo build
   

运行服务

cargo run

服务将在 SERVER_PORT 环境变量指定的端口上启动（默认值：3000）。

使用 Docker

本项目支持 Docker，便于部署：

# Start the service
just docker-up service

# Stop the service
just docker-down service

配置

可以使用以下环境变量配置服务：

• SERVER_NAME：服务名称（默认值："image-mcp-servers"）
• SERVER_VERSION：服务版本（默认值："0.1.0"）
• SERVER_PORT：服务运行端口（默认值：3000）
• HF_API_TOKEN：HuggingFace API 令牌
• REPLICATE_API_TOKEN：Replicate API 令牌

功能

本项目支持两组主要功能，可启用或禁用：

• huggingface：启用 HuggingFace API 集成
• replicate：启用 Replicate API 集成

默认情况下，不启用任何功能。你可以构建并启用指定功能：

# Build with only HuggingFace
cargo build --no-default-features --features huggingface

# Build with only Replicate
cargo build --no-default-features --features replicate

许可证

本项目采用 MIT 许可证，详情请参阅 LICENSE 文件。

致谢

• MCP Core - 用于构建此服务的框架
• HuggingFace - 提供图像生成模型
• Replicate - 提供图像生成 API