Skip to main content

Embedder

punch_memory::embeddings

Trait Embedder 

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

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

Trait for computing vector embeddings from text.

Required Methods§

Source

fn embed(&self, text: &str) -> PunchResult<Vec<f32>>

Compute an embedding vector for a single piece of text.

Source

fn dimensions(&self) -> usize

The dimensionality of vectors produced by this embedder.

Provided Methods§

Source

fn embed_batch(&self, texts: &[&str]) -> PunchResult<Vec<Vec<f32>>>

Compute embedding vectors for a batch of texts.

Implementors§