Trait EmbeddingModel 

pub trait EmbeddingModel: WasmCompatSend + WasmCompatSync {
    type Client;

    const MAX_DOCUMENTS: usize;

    // Required methods
    fn make(
        client: &Self::Client,
        model: impl Into<String>,
        dims: Option<usize>,
    ) -> Self;
    fn ndims(&self) -> usize;
    fn embed_texts(
        &self,
        texts: impl IntoIterator<Item = String> + WasmCompatSend,
    ) -> impl Future<Output = Result<Vec<Embedding>, EmbeddingError>> + WasmCompatSend;

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

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 Associated Types

type Client

Required Methods

fn make( client: &Self::Client, model: impl Into<String>, dims: Option<usize>, ) -> Self

fn ndims(&self) -> usize

The number of dimensions in the embedding vector.

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

Embed multiple text documents in a single request

Provided Methods

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

Embed a single text document.

Dyn Compatibility

This trait is not dyn compatible.

In older versions of Rust, dyn compatibility was called "object safety".

Implementors

impl<T> EmbeddingModel for rig_extra::providers::gemini::EmbeddingModel<T>

where T: Clone + HttpClientExt + 'static,

const MAX_DOCUMENTS: usize = 1024

type Client = Client<GeminiExt, T>

impl<T> EmbeddingModel for rig_extra::providers::mistral::EmbeddingModel<T>

where T: HttpClientExt + Clone + 'static,

const MAX_DOCUMENTS: usize = MAX_DOCUMENTS

type Client = Client<MistralExt, T>

impl<T> EmbeddingModel for rig_extra::providers::ollama::EmbeddingModel<T>

where T: HttpClientExt + Clone + 'static,

const MAX_DOCUMENTS: usize = 1024

type Client = Client<OllamaExt, T>

impl<T> EmbeddingModel for rig_extra::providers::voyageai::EmbeddingModel<T>

where T: HttpClientExt + Clone + Debug + Default + 'static,

const MAX_DOCUMENTS: usize = 1024

type Client = Client<VoyageExt, T>

impl<T> EmbeddingModel for rig_extra::providers::openai::EmbeddingModel<T>

where T: HttpClientExt + Clone + Debug + Default + Send + 'static,

const MAX_DOCUMENTS: usize = 1024

type Client = Client<OpenAIResponsesExt, T>

impl<T> EmbeddingModel for rig_extra::providers::openrouter::EmbeddingModel<T>

where T: HttpClientExt + Clone + Debug + Default + WasmCompatSend + 'static,

const MAX_DOCUMENTS: usize = 1024

type Client = Client<OpenRouterExt, T>

impl<T> EmbeddingModel for rig_extra::providers::cohere::EmbeddingModel<T>

where T: HttpClientExt + Clone + WasmCompatSend + WasmCompatSync + 'static,

const MAX_DOCUMENTS: usize = 96

type Client = Client<CohereExt, T>

impl<T> EmbeddingModel for rig_extra::providers::azure::EmbeddingModel<T>

where T: HttpClientExt + Default + Clone + 'static,

const MAX_DOCUMENTS: usize = 1024

type Client = Client<AzureExt, T>

impl<T> EmbeddingModel for rig_extra::providers::together::EmbeddingModel<T>

where T: HttpClientExt + Default + Clone + Send + 'static,

const MAX_DOCUMENTS: usize = 1024

type Client = Client<TogetherExt, T>