oasysdb 0.6.0

Fast embedded vector database with incremental HNSW indexing.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20

use oasysdb::prelude::*;

/// Creates a collection with random vector records.
/// * `dimension`: Dimensionality of the vector embeddings
/// * `len`: Number of records in the database
pub fn build_test_collection(dimension: usize, len: usize) -> Collection {
    let records = Record::many_random(dimension, len);
    let config = Config::default();
    Collection::build(&config, &records).unwrap()
}

/// Creates a pre-populated database with a collection for testing.
/// * `dimension`: Dimensionality of the vector embeddings
/// * `size`: Number of records in the collection
pub fn create_test_database(dimension: usize, size: usize) -> Database {
    let collection = build_test_collection(dimension, size);
    let mut db = Database::new("data/bench").unwrap();
    db.save_collection("bench", &collection).unwrap();
    db
}