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

//! Backend contracts for storage implementations.

use anyhow::Result;
use serde::{Deserialize, Serialize};

use crate::store::{Drawer, DrawerFilter, SearchResult};

#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
pub struct PalaceRef {
    pub backend: String,
    pub location: String,
}

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct QueryResult {
    pub results: Vec<SearchResult>,
}

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct GetResult {
    pub drawer: Option<Drawer>,
}

pub trait Backend {
    fn palace_ref(&self) -> PalaceRef;
    fn count_drawers(&self) -> Result<i64>;
    fn get_drawer(&self, id: &str) -> Result<GetResult>;
    fn query(&self, query_vec: &[f32], filter: &DrawerFilter, limit: usize) -> Result<QueryResult>;
}

pub trait SourceAdapter {
    fn name(&self) -> &str;
    fn ingest(&self) -> Result<usize>;
}