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
115
116
117
118
119
120
121
122
123
124
125
// SPDX-License-Identifier: Apache-2.0
//! Vector search and hybrid search scalar function registrations.
use nodedb_types::columnar::ColumnType;
use crate::functions::arg_types;
use crate::functions::registry::{FunctionCategory::Scalar, FunctionMeta, SearchTrigger};
use super::super::helpers::{m, no_trigger};
pub(super) fn vector_functions() -> Vec<FunctionMeta> {
vec![
m(
"vector_distance",
Scalar,
2,
3,
SearchTrigger::VectorSearch,
Some(ColumnType::Float64),
arg_types::VECTOR_DISTANCE_ARGS,
),
m(
"vector_cosine_distance",
Scalar,
2,
3,
SearchTrigger::VectorSearch,
Some(ColumnType::Float64),
arg_types::VECTOR_DISTANCE_ARGS,
),
m(
"vector_neg_inner_product",
Scalar,
2,
3,
SearchTrigger::VectorSearch,
Some(ColumnType::Float64),
arg_types::VECTOR_DISTANCE_ARGS,
),
m(
"multi_vector_search",
Scalar,
1,
2,
SearchTrigger::MultiVectorSearch,
Some(ColumnType::Float64),
arg_types::MULTI_VECTOR_SEARCH_ARGS,
),
m(
"multi_vector_score",
Scalar,
3,
3,
no_trigger(),
Some(ColumnType::Float64),
arg_types::MULTI_VECTOR_SCORE_ARGS,
),
m(
"sparse_score",
Scalar,
3,
3,
no_trigger(),
Some(ColumnType::Float64),
arg_types::SPARSE_SCORE_ARGS,
),
m(
"bm25_score",
Scalar,
2,
2,
SearchTrigger::TextSearch,
Some(ColumnType::Float64),
arg_types::BM25_SCORE_ARGS,
),
m(
"search_score",
Scalar,
2,
2,
SearchTrigger::TextSearch,
Some(ColumnType::Float64),
arg_types::SEARCH_SCORE_ARGS,
),
m(
"text_match",
Scalar,
2,
3,
SearchTrigger::TextMatch,
None,
arg_types::TEXT_MATCH_ARGS,
),
// `SEARCH(field, 'query text')` — alias for WHERE-clause FTS search.
// Routes the same as `text_match` through the TextMatch WHERE-search path.
m(
"search",
Scalar,
2,
3,
SearchTrigger::TextMatch,
None,
arg_types::TEXT_MATCH_ARGS,
),
m(
"rrf_score",
Scalar,
2,
6,
SearchTrigger::HybridSearch,
Some(ColumnType::Float64),
arg_types::RRF_SCORE_ARGS,
),
m(
"graph_score",
Scalar,
2,
usize::MAX,
SearchTrigger::GraphSearch,
Some(ColumnType::Float64),
arg_types::GRAPH_SCORE_ARGS,
),
]
}