velesdb-cli 1.14.1

Interactive CLI and REPL for VelesDB with VelesQL support
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

//! CLI argument types for clap `ValueEnum` derivation.
//!
//! Contains `MetricArg`, `StorageModeArg`, and `IndexTypeArg` plus their
//! `From` conversions into the core domain types.

use clap::ValueEnum;
use velesdb_core::{DistanceMetric, StorageMode};

/// CLI metric option
#[derive(Debug, Clone, Copy, ValueEnum, Default)]
pub enum MetricArg {
    #[default]
    Cosine,
    Euclidean,
    Dot,
    Hamming,
    Jaccard,
}

impl From<MetricArg> for DistanceMetric {
    fn from(m: MetricArg) -> Self {
        match m {
            MetricArg::Cosine => DistanceMetric::Cosine,
            MetricArg::Euclidean => DistanceMetric::Euclidean,
            MetricArg::Dot => DistanceMetric::DotProduct,
            MetricArg::Hamming => DistanceMetric::Hamming,
            MetricArg::Jaccard => DistanceMetric::Jaccard,
        }
    }
}

/// CLI storage mode option
#[derive(Debug, Clone, Copy, ValueEnum, Default)]
pub enum StorageModeArg {
    #[default]
    Full,
    Sq8,
    Binary,
    Pq,
    Rabitq,
}

impl From<StorageModeArg> for StorageMode {
    fn from(m: StorageModeArg) -> Self {
        match m {
            StorageModeArg::Full => StorageMode::Full,
            StorageModeArg::Sq8 => StorageMode::SQ8,
            StorageModeArg::Binary => StorageMode::Binary,
            StorageModeArg::Pq => StorageMode::ProductQuantization,
            StorageModeArg::Rabitq => StorageMode::RaBitQ,
        }
    }
}

/// CLI index type option
#[derive(Debug, Clone, Copy, ValueEnum)]
pub enum IndexTypeArg {
    Secondary,
    Property,
    Range,
}