elid 0.4.24

Embedding Locality IDentifier - encode embeddings into sortable string IDs for vector search without vector stores, plus fast string similarity algorithms
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
33
34
35
36
37
38
39
40
41

//! Embedding model support for text and images
//!
//! This module provides ONNX-based embedding models that can run in WASM.
//! Models are loaded from files and run inference using tract-onnx.
//!
//! ## Features
//!
//! - `models`: Base ONNX model support (tract-onnx dependency)
//! - `models-text`: Text embedding using Model2Vec (potion-base-8M)
//! - `models-image`: Image embedding using MobileNetV3-Small
//!
//! ## Example
//!
//! ```rust,ignore
//! use elid::models::{embed_text, embed_image};
//!
//! // Text embedding
//! let text_embedding = embed_text("Hello, world!")?;
//! assert_eq!(text_embedding.len(), 256);
//!
//! // Image embedding
//! let image_bytes = std::fs::read("image.jpg")?;
//! let image_embedding = embed_image(&image_bytes)?;
//! assert_eq!(image_embedding.len(), 1024);
//! ```

mod error;

#[cfg(feature = "models-text")]
pub mod text;

#[cfg(feature = "models-image")]
pub mod image;

pub use error::ModelError;

#[cfg(feature = "models-text")]
pub use text::embed_text;

#[cfg(feature = "models-image")]
pub use image::embed_image;