16 lines
422 B
Markdown
16 lines
422 B
Markdown
|
```rust title="Rust"
|
||
|
|
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);
|
||
|
|
```
|