ruvector-onnx-embeddings 0.1.0

ONNX-based embedding generation for RuVector - Reimagined embedding pipeline in pure Rust
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32

//! Basic embedding example demonstrating single text embedding

use anyhow::Result;
use ruvector_onnx_embeddings::{Embedder, EmbedderConfig, PretrainedModel};

#[tokio::main]
async fn main() -> Result<()> {
    // Create embedder with a specific model
    let config = EmbedderConfig::pretrained(PretrainedModel::AllMiniLmL6V2);
    let mut embedder = Embedder::new(config).await?;

    // Embed text
    let text = "Hello, RuVector!";
    let embedding = embedder.embed_one(text)?;

    println!("Text: {}", text);
    println!("Embedding dimension: {}", embedding.len());
    println!("First 10 values: {:?}", &embedding[..10]);

    // Compute similarity
    let similar_text = "Greetings, RuVector!";
    let different_text = "The weather is sunny.";

    let sim1 = embedder.similarity(text, similar_text)?;
    let sim2 = embedder.similarity(text, different_text)?;

    println!("\nSimilarity scores:");
    println!("  '{}' <-> '{}': {:.4}", text, similar_text, sim1);
    println!("  '{}' <-> '{}': {:.4}", text, different_text, sim2);

    Ok(())
}