Struct HuggingFaceProvider 

pub struct HuggingFaceProvider { /* private fields */ }

HuggingFace Inference API embedding provider

Implementations

impl HuggingFaceProvider

pub fn new(api_key: String, model: String) -> Self

Create a new HuggingFace provider

pub async fn detect_dimension(&mut self) -> Result<usize>

Detect dimension by making a test API call

pub fn get_model_info(model: &str) -> (Option<usize>, usize)

Get common HuggingFace model information (dimension, max_length)

Trait Implementations

impl EmbeddingProvider for HuggingFaceProvider

fn embed_text<'life0, 'life1, 'async_trait>( &'life0 self, text: &'life1 str, ) -> Pin<Box<dyn Future<Output = Result<Vec<f32>>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

Generate embeddings for a single text

fn get_dimension<'life0, 'async_trait>( &'life0 self, ) -> Pin<Box<dyn Future<Output = Result<usize>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait,

Get the dimension of embeddings produced by this provider

fn health_check<'life0, 'async_trait>( &'life0 self, ) -> Pin<Box<dyn Future<Output = Result<()>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait,

Test if the provider is available and working

fn get_info(&self) -> ProviderInfo

Get provider-specific information