MCP Memory Service (rust Implementation)

🚀 MCP 内存服务

MCP 内存服务是一个基于 Rust 编程语言打造的内存管理平台。它能提供高效的存储、检索、搜索和删除功能，支持多种存储后端和嵌入式模型，还可与不同工具集成。

🚀 快速开始

MCP 内存服务是一个基于 Rust 编程语言的内存管理平台，能高效地进行存储、检索、搜索和删除操作。它支持多种存储后端和嵌入式模型，还可与不同工具集成。

📦 安装指南

使用环境变量配置 MCP 服务

1. 设置环境变量

# 示例：通过命令行设置环境变量
export MCP_MEMORY_STORAGE_BACKEND="inmemory"
export MCP_MEMORY_EMBEDDING_MODEL="onnx"
export MCP_MEMORY_EMBEDDING_MODEL_PATH="/path/to/model.onnx"
export MCP_MEMORY_EMBEDDING_SIZE=768
export MCP_MEMORY_LOG_LEVEL="info"

2. 运行 MCP 服务

# 示例：通过命令行运行 MCP 服务
cargo run --release

✨ 主要特性

核心功能

• 存储：支持将内存数据存储到不同的后端（如内存或 ChromaDB）。
• 检索：能够根据 ID 检索特定的内存数据。
• 搜索：提供基于内容相似度的内存数据搜索功能。
• 删除：允许删除指定 ID 的内存数据。

工具集成

MCP 内存服务提供了多种工具接口，方便与其他系统和应用程序集成。这些工具包括：

• 命令行工具
• HTTP API
• WebSocket 接口

📚 详细文档

项目结构

# MCP 内存服务的 Rust 项目结构

├── src/
│   ├── main.rs         # 主程序入口
│   ├── config/        # 配置模块
│   │       └── config.rs     # 环境变量配置解析
│   ├── storage/       # 存储后端实现
│   │       ├── inmemory.rs    # 内存存储实现
│   │       └── chroma.rs      # ChromaDB 存储实现
│   ├── embedding/     # 嵌入式模型实现
│   │       ├── dummy.rs       # 简单嵌入式实现（用于测试）
│   │       └── onnx.rs        # ONNX 模型实现
│   └── tools/         # 工具模块
│           └── tool_logic.rs  # 工具逻辑实现

第三方依赖

MCP 内存服务使用了以下第三方 crates：

• ort: 提供对 ONNX 模型的支持。
• tokenizers: 提供 tokenizer 库，用于处理文本数据。
• reqwest: 提供 HTTP 客户端功能。

💻 使用示例

基础用法

# 示例：使用命令行工具进行基本操作
mcp store --id 1 --content "hello world"
mcp retrieve --id 1
mcp search --query "similar to hello world"
mcp delete --id 1

🔗 扩展阅读

• Rust 编程语言
• ONNX 模型规范
• ChromaDB 存储系统

📄 许可证

MCP 内存服务遵循 MIT 许可证，具体细节请参阅 LICENSE 文件。