embedcache 0.1.1

High-performance text embedding service with caching capabilities
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29

//! Embedding generation module
//!
//! This module provides traits and implementations for generating text embeddings.

mod fastembed;
mod registry;

use anyhow::Result;
use async_trait::async_trait;

pub use self::fastembed::FastEmbedder;
pub use registry::{get_embedding_model, initialize_models, SUPPORTED_MODELS};

/// Trait for embedding generation
///
/// Implementors of this trait can provide custom embedding logic.
#[async_trait]
pub trait Embedder: Send + Sync {
    /// Generate embeddings for the given text chunks
    ///
    /// # Arguments
    ///
    /// * `chunks` - The text chunks to embed
    ///
    /// # Returns
    ///
    /// A vector of embeddings, one for each chunk
    async fn embed(&self, chunks: &[String]) -> Result<Vec<Vec<f32>>>;
}