422 B
422 B
use kreuzberg::{EmbeddingConfig, EmbeddingModelType, embed_texts};
let config = EmbeddingConfig {
model: EmbeddingModelType::Preset { name: "balanced".to_string() },
normalize: true,
..Default::default()
};
let texts = vec!["Hello, world!", "Kreuzberg is fast"];
let embeddings = embed_texts(&texts, &config)?;
assert_eq!(embeddings.len(), 2);
assert_eq!(embeddings[0].len(), 768);