MCP Memory Service (rust Implementation)

🚀 MCP Memory Service

MCP Memory Service is a memory management platform based on the Rust programming language. It aims to provide efficient storage, retrieval, search, and deletion functions. The platform supports multiple storage backends and embedded models and can be integrated with various tools.

🚀 Quick Start

Configure the MCP Service Using Environment Variables

1. Set Environment Variables

# Example: Set environment variables via the command line
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. Run the MCP Service

# Example: Run the MCP service via the command line
cargo run --release

✨ Features

Core Features

• Storage: Supports storing memory data in different backends (e.g., in-memory or ChromaDB).
• Retrieval: Can retrieve specific memory data based on the ID.
• Search: Provides a memory data search function based on content similarity.
• Deletion: Allows deleting memory data with a specified ID.

Tool Integration

The MCP Memory Service provides multiple tool interfaces for easy integration with other systems and applications. These tools include:

• Command-line Tool
• HTTP API
• WebSocket Interface

📚 Documentation

Project Structure

# Rust project structure of the MCP Memory Service

├── src/
│   ├── main.rs         # Main program entry
│   ├── config/        # Configuration module
│   │       └── config.rs     # Environment variable configuration parsing
│   ├── storage/       # Storage backend implementation
│   │       ├── inmemory.rs    # In-memory storage implementation
│   │       └── chroma.rs      # ChromaDB storage implementation
│   ├── embedding/     # Embedded model implementation
│   │       ├── dummy.rs       # Simple embedded implementation (for testing)
│   │       └── onnx.rs        # ONNX model implementation
│   └── tools/         # Tool module
│           └── tool_logic.rs  # Tool logic implementation

Third-party Dependencies

The MCP Memory Service uses the following third-party crates:

• ort: Provides support for ONNX models.
• tokenizers: Provides a tokenizer library for processing text data.
• reqwest: Provides HTTP client functionality.

💻 Usage Examples

Basic Usage

# Example: Perform basic operations using the command-line tool
mcp store --id 1 --content "hello world"
mcp retrieve --id 1
mcp search --query "similar to hello world"
mcp delete --id 1

📄 License

The MCP Memory Service is licensed under the MIT License. For specific details, please refer to the LICENSE file.

📚 Extended Reading

• Rust Programming Language
• ONNX Model Specification
• ChromaDB Storage System