EmbeddingGenerator

Trait EmbeddingGenerator 

Source
pub trait EmbeddingGenerator:
    Send
    + Sync
    + AsAny {
    // Required methods
    fn generate(&self, content: &EmbeddableContent) -> Result<Vector>;
    fn dimensions(&self) -> usize;
    fn config(&self) -> &EmbeddingConfig;

    // Provided method
    fn generate_batch(
        &self,
        contents: &[EmbeddableContent],
    ) -> Result<Vec<Vector>> { ... }
}
Expand description

Embedding generator trait

Required Methods§

Source

fn generate(&self, content: &EmbeddableContent) -> Result<Vector>

Generate embedding for content

Source

fn dimensions(&self) -> usize

Get the embedding dimensions

Source

fn config(&self) -> &EmbeddingConfig

Get the model configuration

Provided Methods§

Source

fn generate_batch(&self, contents: &[EmbeddableContent]) -> Result<Vec<Vector>>

Generate embeddings for multiple contents in batch

Implementors§