🚀 图像 MCP 服务器
本项目实现了一个基于 SSE 的 MCP 服务器，提供了与 HuggingFace 和 Replicate API 交互的工具。它专注于图像生成和模型管理，使用 MCP 核心框架构建，并提供标准化接口用于 AI 模型交互。以下将为你介绍如何自行部署该项目。
🚀 快速开始
先决条件
- Rust（2024 版本）
- Docker（可选，用于容器化部署）
安装
- 克隆仓库：
git clone https://github.com/yourusername/image-mcp-servers.git cd image-mcp-servers - 基于
.env.example创建一个.env文件：cp .env.example .env - 将你的 API 令牌添加到
.env文件中：HF_API_TOKEN="your_huggingface_token" REPLICATE_API_TOKEN="your_replicate_token" - 构建项目：
cargo build
运行服务器
cargo run
服务器将在 SERVER_PORT 环境变量指定的端口上运行（默认：3000）。
使用 Docker
该项目包含 Docker 支持，以实现轻松部署：
# 启动服务
just docker-up service
# 停止服务
just docker-down service
✨ 主要特性
HuggingFace 集成
| 工具 | 描述 |
|---|---|
| 获取模型信息 | 检索 HuggingFace 上模型的详细信息 |
| 获取模型示例图像 | 从模型 README 中提取样本图像 |
| 获取 README | 获取模型的 README 内容 |
| 搜索模型 | 使用过滤选项在 HuggingFace 上搜索模型 |
| 我是谁（WhoAmI） | 检索当前 HuggingFace API 令牌的信息 |
Replicate 集成
| 工具 | 描述 |
|---|---|
| 生成图像 | 使用 Replicate 的文本到图像模型创建图像 |
| 获取模型信息 | 检索 Replicate 上模型的详细信息 |
| 获取预测 | 检查预测状态并检索输出 |
| 列出模型 | 列出可用的 Replicate 模型，带可选过滤 |
| 我是谁（WhoAmI） | 检索当前 Replicate API 令牌的信息 |
📦 安装指南
环境依赖
- Rust（2024 版本）
- Docker（可选，用于容器化部署）
安装步骤
- 克隆仓库：
git clone https://github.com/yourusername/image-mcp-servers.git cd image-mcp-servers - 基于
.env.example创建一个.env文件：cp .env.example .env - 将你的 API 令牌添加到
.env文件中：HF_API_TOKEN="your_huggingface_token" REPLICATE_API_TOKEN="your_replicate_token" - 构建项目：
cargo build
📚 详细文档
配置
服务器可以通过环境变量进行配置：
| 属性 | 详情 |
|---|---|
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 集成
默认情况下没有启用任何功能。你可以根据需要构建带有特定功能的版本：
# 仅使用 HuggingFace 构建
cargo build --no-default-features --features huggingface
# 仅使用 Replicate 构建
cargo build --no-default-features --features replicate
📄 许可证
该项目在 MIT License 下许可 - 有关详细信息，请参阅 LICENSE 文件。
👏 致谢
- MCP Core - 用于构建此服务器的框架
- HuggingFace - 提供其图像生成模型
- Replicate - 提供其图像生成 API