mem0-rust 0.2.0

Rust implementation of mem0 - Universal memory layer for AI Agents
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21

mod cohere;

pub use cohere::CohereReranker;

use async_trait::async_trait;
use crate::models::ScoredMemory;
use crate::errors::MemoryError;
use crate::config::RerankerConfig;
use std::sync::Arc;

#[async_trait]
pub trait Reranker: Send + Sync {
    async fn rerank(&self, query: &str, results: Vec<ScoredMemory>) -> Result<Vec<ScoredMemory>, MemoryError>;
    fn model_name(&self) -> &str;
}

pub fn create_reranker(config: &RerankerConfig) -> Result<Arc<dyn Reranker>, MemoryError> {
    match config {
        RerankerConfig::Cohere(cfg) => Ok(Arc::new(CohereReranker::new(cfg.clone())?)),
    }
}