Trait EmbeddingModel

pub trait EmbeddingModel:
    Clone
    + Sync
    + Send {
    const MAX_DOCUMENTS: usize;

    // Required methods
    fn ndims(&self) -> usize;
    fn embed_texts(
        &self,
        texts: impl IntoIterator<Item = String> + Send,
    ) -> impl Future<Output = Result<Vec<Embedding>, EmbeddingError>> + Send;

    // Provided method
    fn embed_text(
        &self,
        text: &str,
    ) -> impl Future<Output = Result<Embedding, EmbeddingError>> + Send { ... }
}

Trait for embedding models that can generate embeddings for documents.

Required Associated Constants

const MAX_DOCUMENTS: usize

The maximum number of documents that can be embedded in a single request.

Required Methods

fn ndims(&self) -> usize

The number of dimensions in the embedding vector.

fn embed_texts( &self, texts: impl IntoIterator<Item = String> + Send, ) -> impl Future<Output = Result<Vec<Embedding>, EmbeddingError>> + Send

Embed multiple text documents in a single request

Provided Methods

fn embed_text( &self, text: &str, ) -> impl Future<Output = Result<Embedding, EmbeddingError>> + Send

Embed a single text document.

Dyn Compatibility

This trait is not dyn compatible.

In older versions of Rust, dyn compatibility was called "object safety", so this trait is not object safe.

Implementors

impl EmbeddingModel for rig::providers::azure::EmbeddingModel

const MAX_DOCUMENTS: usize = 1_024usize

impl EmbeddingModel for rig::providers::cohere::embeddings::EmbeddingModel

const MAX_DOCUMENTS: usize = 96usize

impl EmbeddingModel for rig::providers::gemini::embedding::EmbeddingModel

const MAX_DOCUMENTS: usize = 1_024usize

impl EmbeddingModel for rig::providers::mistral::embedding::EmbeddingModel

const MAX_DOCUMENTS: usize = 1_024usize

impl EmbeddingModel for rig::providers::ollama::EmbeddingModel

const MAX_DOCUMENTS: usize = 1_024usize

impl EmbeddingModel for rig::providers::openai::embedding::EmbeddingModel

const MAX_DOCUMENTS: usize = 1_024usize

impl EmbeddingModel for rig::providers::together::embedding::EmbeddingModel

const MAX_DOCUMENTS: usize = 1_024usize

impl EmbeddingModel for rig::providers::xai::embedding::EmbeddingModel

const MAX_DOCUMENTS: usize = 1_024usize