Trait Embedder

Source

pub trait Embedder: Send + Sync {
    // Required methods
    fn dimensions(&self) -> usize;
    fn model_name(&self) -> &'static str;
    fn embed(&self, text: &str) -> Result<Vec<f32>>;

    // Provided method
    fn embed_batch(&self, texts: &[&str]) -> Result<Vec<Vec<f32>>> { ... }
}

Expand description

Trait for embedding generators.

Implementations must be thread-safe (Send + Sync) to support parallel embedding generation during chunk loading.

§Examples

use rlm_rs::embedding::{Embedder, FallbackEmbedder, DEFAULT_DIMENSIONS};

let embedder = FallbackEmbedder::new(DEFAULT_DIMENSIONS);
let embedding = embedder.embed("Hello, world!").unwrap();
assert_eq!(embedding.len(), DEFAULT_DIMENSIONS);