Skip to main content

Crate cerebro

Crate cerebro 

Source
Expand description

§Cerebro 🧠

Cerebro is a blazing-fast, storage-agnostic semantic memory engine for AI Agents and LLM applications, written in pure Rust.

§Quick Start

use cerebro::prelude::*;
use std::sync::Arc;

#[tokio::main]
async fn main() {
    let chunker = Arc::new(RecursiveCharacterChunker::new(512, 50));
    let embedder = Arc::new(MockEmbedder::new(1536));
    let store = Arc::new(MemoryVectorStore::new());

    let engine = MemoryEngine::new(chunker, embedder, store);

    let doc = Document::new("Rust ensures memory safety without a garbage collector.");
    engine.ingest_document(doc).await.unwrap();

    let results = engine.query("memory safety", 5).await.unwrap();
    for (node, score) in results {
        println!("Score {:.3}: {}", score, node.chunk.text);
    }
}

Re-exports§

pub use models::*;
pub use traits::*;

Modules§

chunker
compute
engine
ffi
ingest
models
prelude
Convenience prelude — use cerebro::prelude::* to import everything you need.
storage
swarm
Cerebro Swarm Module
traits