sqlite-vector-rs 0.2.2

SQLite extension providing PGVector-like native vector types with HNSW indexing
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
42
43
44
45

use rusqlite::Connection;
use std::path::Path;

/// Create an in-memory SQLite connection with the vector extension loaded.
pub fn open_with_extension() -> Connection {
    let conn = Connection::open_in_memory().unwrap();
    unsafe {
        conn.load_extension_enable().unwrap();
        let ext_path = find_extension_path();
        conn.load_extension(ext_path, None::<&str>).unwrap();
        conn.load_extension_disable().unwrap();
    }
    conn
}

/// Create a file-backed SQLite connection (for persistence tests).
#[allow(dead_code)]
pub fn open_file_with_extension(path: &Path) -> Connection {
    let conn = Connection::open(path).unwrap();
    unsafe {
        conn.load_extension_enable().unwrap();
        let ext_path = find_extension_path();
        conn.load_extension(ext_path, None::<&str>).unwrap();
        conn.load_extension_disable().unwrap();
    }
    conn
}

fn find_extension_path() -> &'static str {
    if cfg!(target_os = "macos") {
        "target/debug/libsqlite_vector_rs"
    } else if cfg!(target_os = "windows") {
        "target/debug/sqlite_vector_rs"
    } else {
        "target/debug/libsqlite_vector_rs"
    }
}

/// Generate a random f32 vector of given dimension.
#[allow(dead_code)]
pub fn random_f32_vector(dim: usize) -> Vec<f32> {
    use rand::Rng;
    let mut rng = rand::rng();
    (0..dim).map(|_| rng.random_range(-1.0..1.0)).collect()
}