Enum EmbeddingModel 

pub enum EmbeddingModel {
    MiniLM,
    BgeSmall,
    E5Small,
}

Supported embedding models.

Variants

MiniLM

all-MiniLM-L6-v2 - Fast and efficient, good for general use

• Dimensions: 384
• Max tokens: 256
• Speed: Fastest

BgeSmall

BAAI/bge-small-en-v1.5 - Balanced quality and speed

• Dimensions: 384
• Max tokens: 512
• Speed: Medium

E5Small

intfloat/e5-small-v2 - Higher quality embeddings

• Dimensions: 384
• Max tokens: 512
• Speed: Medium

Implementations

impl EmbeddingModel

pub fn model_id(&self) -> &'static str

Get the HuggingFace model ID.

pub fn dimension(&self) -> usize

Get the embedding dimension for this model.

pub fn max_seq_length(&self) -> usize

Get the maximum sequence length (in tokens).

pub fn query_prefix(&self) -> Option<&'static str>

Get the query prefix for models that require it. Some models like E5 require a prefix for queries vs documents.

pub fn document_prefix(&self) -> Option<&'static str>

Get the document/passage prefix for models that require it.

pub fn use_mean_pooling(&self) -> bool

Whether this model uses mean pooling (vs CLS token).

pub fn normalize_embeddings(&self) -> bool

Whether embeddings should be normalized.

pub fn tokens_per_second_cpu(&self) -> usize

Get approximate tokens per second on CPU (for estimation).

pub fn all() -> &'static [EmbeddingModel]

List all available models.

pub fn parse(s: &str) -> Option<Self>

Parse model from string (case-insensitive).

Trait Implementations

impl Clone for EmbeddingModel

fn clone(&self) -> EmbeddingModel

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for EmbeddingModel

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

Formats the value using the given formatter.

impl Default for EmbeddingModel

fn default() -> EmbeddingModel

Returns the “default value” for a type.

impl<'de> Deserialize<'de> for EmbeddingModel

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Display for EmbeddingModel

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

Formats the value using the given formatter.

impl Hash for EmbeddingModel

fn hash<__H: Hasher>(&self, state: &mut __H)

Feeds this value into the given Hasher.

fn hash_slice<H>(data: &[Self], state: &mut H)

where H: Hasher, Self: Sized,

Feeds a slice of this type into the given Hasher.

impl PartialEq for EmbeddingModel

fn eq(&self, other: &EmbeddingModel) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl Serialize for EmbeddingModel

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.

impl Copy for EmbeddingModel

impl Eq for EmbeddingModel

impl StructuralPartialEq for EmbeddingModel