Enum EmbeddingModel 

pub enum EmbeddingModel {
    OpenAISmall,
    OpenAILarge,
    OpenAIAda,
    MiniLM,
    MPNet,
    Cohere,
    GoogleGecko,
    Voyage,
    Ollama {
        model: String,
    },
    Custom {
        name: String,
        dim: usize,
    },
}

Embedding model options

Variants

OpenAISmall

OpenAI text-embedding-3-small (1536 dims)

OpenAILarge

OpenAI text-embedding-3-large (3072 dims)

OpenAIAda

OpenAI text-embedding-ada-002 (1536 dims)

MiniLM

Sentence Transformers all-MiniLM-L6-v2 (384 dims)

MPNet

Sentence Transformers all-mpnet-base-v2 (768 dims)

Cohere

Cohere embed-english-v3.0 (1024 dims)

GoogleGecko

Google text-embedding-004 (768 dims)

Voyage

Voyage AI voyage-2 (1024 dims)

Ollama

Local model via Ollama

Fields

model: String

Custom

Custom model

Fields

name: String

dim: usize

Implementations

impl EmbeddingModel

pub fn dimension(&self) -> usize

Get the dimension for this model

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 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.