oasysdb 0.3.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
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
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80

#![warn(missing_docs)]
#![doc = include_str!("../readme.md")]
#![doc(html_favicon_url = "https://i.postimg.cc/W3T230zk/favicon.png")]
#![doc(html_logo_url = "https://i.postimg.cc/Vv0HPVwB/logo.png")]

#[cfg(test)]
mod tests;

mod db;
mod func;

/// Convenience re-exports for the public APIs.
pub mod prelude;

pub use db::database;
pub use func::collection;
pub use func::err;
pub use func::metadata;
pub use func::vector;

use pyo3::prelude::*;

type Module = fn(Python<'_>, &PyModule) -> PyResult<()>;

#[pymodule]
fn oasysdb(py: Python, m: &PyModule) -> PyResult<()> {
    let sys = py.import("sys")?;
    let modules = sys.getattr("modules")?;

    let mods: Vec<(&str, Module)> = vec![
        ("collection", collection_modules),
        ("vector", vector_modules),
        ("database", database_modules),
        ("prelude", prelude_modules),
    ];

    for (name, module) in mods {
        let full_name = format!("oasysdb.{}", name);
        let pymod = PyModule::new(py, &full_name)?;
        module(py, pymod)?;
        m.add(name, pymod)?;
        modules.set_item(full_name, pymod)?;
    }

    Ok(())
}

#[pymodule]
fn collection_modules(_py: Python, m: &PyModule) -> PyResult<()> {
    m.add_class::<collection::Config>()?;
    m.add_class::<collection::Record>()?;
    m.add_class::<collection::Collection>()?;
    m.add_class::<collection::SearchResult>()?;
    Ok(())
}

#[pymodule]
fn vector_modules(_py: Python, m: &PyModule) -> PyResult<()> {
    m.add_class::<vector::Vector>()?;
    m.add_class::<vector::VectorID>()?;
    Ok(())
}

#[pymodule]
fn database_modules(_py: Python, m: &PyModule) -> PyResult<()> {
    m.add_class::<database::Database>()?;
    Ok(())
}

#[pymodule]
fn prelude_modules(_py: Python, m: &PyModule) -> PyResult<()> {
    m.add_class::<collection::Config>()?;
    m.add_class::<collection::Record>()?;
    m.add_class::<collection::Collection>()?;
    m.add_class::<collection::SearchResult>()?;
    m.add_class::<vector::Vector>()?;
    m.add_class::<vector::VectorID>()?;
    m.add_class::<database::Database>()?;
    Ok(())
}