Struct HuggingFaceEmbedder 

pub struct HuggingFaceEmbedder { /* private fields */ }

HuggingFace transformer model for embedding generation

Implementations

impl HuggingFaceEmbedder

pub fn new(config: HuggingFaceConfig) -> Result<Self>

Create a new HuggingFace embedder

pub fn with_default_config() -> Result<Self>

Create embedder with default configuration

pub async fn load_model(&mut self, model_name: &str) -> Result<()>

Load a model and prepare it for inference

pub async fn embed_batch( &mut self, contents: &[EmbeddableContent], ) -> Result<Vec<Vector>>

Generate embeddings for a batch of content

pub async fn embed(&mut self, content: &EmbeddableContent) -> Result<Vector>

Generate a single embedding

pub fn get_cached_models(&self) -> Vec<String>

Get available models from cache

pub fn clear_cache(&mut self)

Clear model cache

pub fn get_model_dimensions(&self, model_name: &str) -> Option<usize>

Get model dimensions

Trait Implementations

impl Debug for HuggingFaceEmbedder

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.