Always watching over your data

🗄️ zoey-storage-mongo

MongoDB storage adapter for ZoeyAI

A MongoDB implementation of the IDatabaseAdapter trait from zoey-core. Provides document-based storage with local vector search capabilities.

Installation

Add to your Cargo.toml:

[dependencies]
zoey-core = "0.1"
zoey-storage-mongo = "0.1"

Features

• 🍃 MongoDB Native - Full MongoDB driver integration
• 🔍 Vector Search - Local cosine similarity search (no Atlas required)
• 📊 Schema Management - Automatic collection and index creation
• 🔐 HIPAA Ready - Optional compliance features

Quick Start

use zoey_core::{AgentRuntime, RuntimeOpts, IDatabaseAdapter};
use zoey_storage_mongo::MongoAdapter;
use std::sync::Arc;

#[tokio::main]
async fn main() -> zoey_core::Result<()> {
    // Create MongoDB adapter
    let mut adapter = MongoAdapter::new(
        "mongodb://localhost:27017",
        "zoey_db"
    );
    
    // Initialize (creates collections and indexes)
    adapter.init().await?;
    
    // Use with runtime
    let opts = RuntimeOpts::default()
        .with_adapter(Arc::new(adapter));
    
    let runtime = AgentRuntime::new(opts).await?;
    
    Ok(())
}

Vector Search

This adapter implements local vector search using MongoDB aggregation pipelines with cosine similarity calculation—no MongoDB Atlas required.

use zoey_storage_mongo::vector_search::MongoVectorSearch;

let searcher = MongoVectorSearch::new(collection, 1536);

let results = searcher.search(SearchMemoriesParams {
    embedding: query_vector,
    count: 10,
    ..Default::default()
}).await?;

Configuration

Environment Variables

MONGODB_URI=mongodb://localhost:27017
MONGODB_DATABASE=zoey_db

Connection Options

// Local development
let adapter = MongoAdapter::new("mongodb://localhost:27017", "zoey_db");

// With authentication
let adapter = MongoAdapter::new(
    "mongodb://user:password@localhost:27017",
    "zoey_db"
);

// Replica set
let adapter = MongoAdapter::new(
    "mongodb://host1:27017,host2:27017,host3:27017/?replicaSet=rs0",
    "zoey_db"
);

Related Crates

License

MIT License - See LICENSE for details.

🔐 Your secrets are safe with Zoey