solidb 1.0.2

A lightweight, high-performance structured database server written in Rust.
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

//! Driver handlers for Graph RAG operations.

use super::DriverHandler;
use crate::driver::protocol::{DriverError, Response};
use crate::sdbql::QueryExecutor;
use serde_json::{json, Value};

fn graph_executor<'a>(handler: &'a DriverHandler, database: &str) -> QueryExecutor<'a> {
    QueryExecutor::with_database(&handler.storage, database.to_string())
}

pub fn handle_graph_neighbors(
    handler: &DriverHandler,
    database: String,
    edge_collection: String,
    seeds: Value,
    options: Option<Value>,
) -> Response {
    match graph_executor(handler, &database).neighbors(&edge_collection, seeds, options) {
        Ok(results) => {
            let count = results.as_array().map(|a| a.len()).unwrap_or(0);
            Response::ok(json!({ "results": results, "count": count }))
        }
        Err(e) => Response::error(DriverError::DatabaseError(e.to_string())),
    }
}

pub fn handle_graph_rag(
    handler: &DriverHandler,
    database: String,
    seed_collection: String,
    vector_index: String,
    edge_collection: String,
    query_vector: Vec<f32>,
    options: Option<Value>,
) -> Response {
    match graph_executor(handler, &database).graph_rag(
        &seed_collection,
        &vector_index,
        &edge_collection,
        json!(query_vector),
        options,
    ) {
        Ok(results) => {
            let count = results.as_array().map(|a| a.len()).unwrap_or(0);
            Response::ok(json!({ "results": results, "count": count }))
        }
        Err(e) => Response::error(DriverError::DatabaseError(e.to_string())),
    }
}

pub fn handle_community_search(
    handler: &DriverHandler,
    database: String,
    query_text: String,
    options: Option<Value>,
) -> Response {
    match graph_executor(handler, &database).community_search(&query_text, options) {
        Ok(results) => {
            let count = results.as_array().map(|a| a.len()).unwrap_or(0);
            Response::ok(json!({ "results": results, "count": count }))
        }
        Err(e) => Response::error(DriverError::DatabaseError(e.to_string())),
    }
}