README
🚀 图像 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
微信扫一扫