Skip to main content

embed_passages_controlled

Function embed_passages_controlled 

Source
pub fn embed_passages_controlled(
    embedder: &Mutex<TextEmbedding>,
    texts: &[&str],
    token_counts: &[usize],
) -> Result<Vec<Vec<f32>>, AppError>
Expand description

Embeds passages grouped into token-budget-aware batches to avoid OOM on variable-length inputs.

texts and token_counts must have the same length. Batches are planned using an internal budget algorithm and single-item batches fall back to embed_passage.

ยงErrors

Returns Err when lengths differ, the mutex is poisoned, or inference fails.