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 文件。