lora-executor 0.11.2

Query-plan executor for LoraDB's Cypher implementation.
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
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114

//! `vector.*` — vector similarity, distance, coordinate extraction, and norms.
//!
//! VECTOR construction is handled by `cast.to(list, VECTOR<T>(N))` /
//! `CAST(list AS VECTOR<T>(N))`.

use crate::value::LoraValue;

use super::super::errors::set_eval_error;
use super::super::vector::{
    eval_vector_distance_fn, eval_vector_norm_fn, eval_vector_sim_cosine, eval_vector_sim_euclidean,
};

pub(super) fn dispatch(op: &str, args: &[LoraValue]) -> Option<LoraValue> {
    Some(match op {
        "dimension" | "dim" => dimension(args),
        "distance" => distance(args),
        "similarity" => similarity(args),
        "norm" => norm(args),
        "coordinates" => coordinates(args),
        _ => return None,
    })
}

fn dimension(args: &[LoraValue]) -> LoraValue {
    match args.first() {
        Some(LoraValue::Null) | None => LoraValue::Null,
        Some(LoraValue::Vector(v)) => LoraValue::Int(v.dimension as i64),
        Some(other) => {
            set_eval_error(format!(
                "vector.dimension expected VECTOR, got {}",
                crate::errors::value_kind(other)
            ));
            LoraValue::Null
        }
    }
}

fn distance(args: &[LoraValue]) -> LoraValue {
    eval_vector_distance_fn(args)
}

fn similarity(args: &[LoraValue]) -> LoraValue {
    // Two valid call shapes:
    //   vector.similarity(a, b)                 — defaults to cosine
    //   vector.similarity(a, b, 'cosine'|'euclidean')
    let metric = args
        .get(2)
        .and_then(|v| match v {
            LoraValue::String(s) => Some(s.to_ascii_lowercase()),
            _ => None,
        })
        .unwrap_or_else(|| "cosine".to_string());
    match metric.as_str() {
        "cosine" => eval_vector_sim_cosine(args),
        "euclidean" => eval_vector_sim_euclidean(args),
        _ => LoraValue::Null,
    }
}

fn norm(args: &[LoraValue]) -> LoraValue {
    eval_vector_norm_fn(args)
}

fn coordinates(args: &[LoraValue]) -> LoraValue {
    let Some(LoraValue::String(target)) = args.get(1) else {
        return LoraValue::Null;
    };

    match target.trim().to_ascii_uppercase().as_str() {
        "INTEGER" | "INT" => integer_coordinates(args),
        "FLOAT" | "REAL" => float_coordinates(args),
        _ => LoraValue::Null,
    }
}

fn integer_coordinates(args: &[LoraValue]) -> LoraValue {
    match args.first() {
        Some(LoraValue::Null) | None => LoraValue::Null,
        Some(LoraValue::Vector(v)) => LoraValue::List(
            v.values
                .to_i64_vec()
                .into_iter()
                .map(LoraValue::Int)
                .collect(),
        ),
        Some(other) => {
            set_eval_error(format!(
                "vector.coordinates expected VECTOR, got {}",
                crate::errors::value_kind(other)
            ));
            LoraValue::Null
        }
    }
}

fn float_coordinates(args: &[LoraValue]) -> LoraValue {
    match args.first() {
        Some(LoraValue::Null) | None => LoraValue::Null,
        Some(LoraValue::Vector(v)) => LoraValue::List(
            v.values
                .as_f64_vec()
                .into_iter()
                .map(LoraValue::Float)
                .collect(),
        ),
        Some(other) => {
            set_eval_error(format!(
                "vector.coordinates expected VECTOR, got {}",
                crate::errors::value_kind(other)
            ));
            LoraValue::Null
        }
    }
}