ruvector-postgres 2.0.5

High-performance PostgreSQL vector database extension v2 - pgvector drop-in replacement with 230+ SQL functions, SIMD acceleration, Flash Attention, GNN layers, hybrid search, multi-tenancy, self-healing, and self-learning capabilities
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

//! Extension initialization and SQL functions

use pgrx::prelude::*;
use super::guc;
use super::state::DAG_STATE;

/// Initialize the DAG extension
pub fn init() {
    // Initialize GUC variables
    guc::init_guc();

    // Register background worker
    super::worker::register_worker();
}

// SQL Functions

#[pg_extern]
fn ruvector_dag_set_enabled(enabled: bool) {
    if enabled {
        DAG_STATE.enable();
    } else {
        DAG_STATE.disable();
    }
}

#[pg_extern]
fn ruvector_dag_is_enabled() -> bool {
    DAG_STATE.is_enabled()
}

#[pg_extern]
fn ruvector_dag_status() -> pgrx::JsonB {
    let status = serde_json::json!({
        "enabled": DAG_STATE.is_enabled(),
        "pattern_count": DAG_STATE.get_pattern_count(),
        "trajectory_count": DAG_STATE.get_trajectory_count(),
        "learning_rate": guc::get_learning_rate(),
        "attention_mechanism": guc::get_attention_mechanism(),
    });

    pgrx::JsonB(status)
}

#[pg_extern]
fn ruvector_dag_set_learning_rate(rate: f64) {
    // Would update GUC variable
    // For now, just validate
    if rate < 0.0 || rate > 1.0 {
        pgrx::error!("Learning rate must be between 0 and 1");
    }
}

#[pg_extern]
fn ruvector_dag_set_attention(mechanism: &str) {
    let valid = ["topological", "causal_cone", "critical_path",
                 "mincut_gated", "hierarchical_lorentz",
                 "parallel_branch", "temporal_btsp", "auto"];

    if !valid.contains(&mechanism) {
        pgrx::error!("Invalid attention mechanism: {}. Valid: {:?}", mechanism, valid);
    }
}