embedd 0.2.1

Embedding interfaces + local backends (Candle/HF).
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23

//! Minimal example: embed two strings and compute cosine similarity.
//!
//! Run: `cargo run -p embedd --example hello_embed --features fastembed`

use embedd::fastembed::FastembedEmbedder;
use embedd::{EmbedMode, TextEmbedder};

fn main() -> anyhow::Result<()> {
    let embedder = FastembedEmbedder::new_default()?;
    println!(
        "model: {}  dim: {}",
        embedder.model_id().unwrap_or("?"),
        embedder.dimension().unwrap_or(0),
    );

    let a = embedder.embed_text("the cat sat on the mat", EmbedMode::Document)?;
    let b = embedder.embed_text("a dog lay on the rug", EmbedMode::Document)?;

    let sim = embedd::vector::cosine_f32(&a, &b);
    println!("cosine similarity: {sim:.4}");

    Ok(())
}