palace-rs 0.20.0

Local memory palace for AI assistants — Rust edition
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

//! SQLite backend implementation.

use anyhow::Result;
use rusqlite::Connection;
use std::path::{Path, PathBuf};

use crate::backend::{Backend, GetResult, PalaceRef, QueryResult};
use crate::store::{self, DrawerFilter};

pub struct SqliteBackend {
    conn: Connection,
    location: PathBuf,
}

impl SqliteBackend {
    pub fn open(path: &Path) -> Result<Self> {
        Ok(Self {
            conn: crate::db::open(path)?,
            location: path.to_path_buf(),
        })
    }

    pub fn open_in_memory() -> Result<Self> {
        Ok(Self {
            conn: crate::db::open_in_memory()?,
            location: PathBuf::from(":memory:"),
        })
    }

    pub fn conn(&self) -> &Connection {
        &self.conn
    }
}

impl Backend for SqliteBackend {
    fn palace_ref(&self) -> PalaceRef {
        PalaceRef {
            backend: "sqlite".to_string(),
            location: self.location.to_string_lossy().into_owned(),
        }
    }

    fn count_drawers(&self) -> Result<i64> {
        store::count_drawers(&self.conn)
    }

    fn get_drawer(&self, id: &str) -> Result<GetResult> {
        Ok(GetResult {
            drawer: store::get_drawer(&self.conn, id)?,
        })
    }

    fn query(&self, query_vec: &[f32], filter: &DrawerFilter, limit: usize) -> Result<QueryResult> {
        Ok(QueryResult {
            results: store::vector_search(&self.conn, query_vec, filter, limit)?,
        })
    }
}