qdrant-client 1.17.0

Rust client for Qdrant Vector Search Engine
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

use std::collections::HashMap;

use crate::qdrant::vectors_config::Config;
use crate::qdrant::{
    MultiVectorComparator, MultiVectorConfigBuilder, VectorParams, VectorParamsMap, VectorsConfig,
};

const DEFAULT_VECTOR_NAME: &str = "";

#[derive(Debug, Clone, Default)]
pub struct VectorsConfigBuilder {
    params: HashMap<String, VectorParams>,
}

impl VectorsConfigBuilder {
    /// Add a named vector with the given parameters
    pub fn add_named_vector_params(
        &mut self,
        name: impl Into<String>,
        params: impl Into<VectorParams>,
    ) -> &mut Self {
        self.params.insert(name.into(), params.into());
        self
    }

    /// Add the default vector with the given parameters
    pub fn add_vector_params(&mut self, params: impl Into<VectorParams>) -> &mut Self {
        self.params
            .insert(DEFAULT_VECTOR_NAME.to_string(), params.into());
        self
    }
}

impl From<VectorsConfigBuilder> for VectorsConfig {
    fn from(mut builder: VectorsConfigBuilder) -> Self {
        if builder.params.is_empty() {
            return VectorsConfig::default();
        }

        if builder.params.len() == 1 && builder.params.contains_key(DEFAULT_VECTOR_NAME) {
            return VectorsConfig {
                config: Some(Config::from(
                    builder.params.remove(DEFAULT_VECTOR_NAME).unwrap(),
                )),
            };
        }

        VectorsConfig {
            config: Some(Config::from(VectorParamsMap::from(builder.params))),
        }
    }
}

impl MultiVectorConfigBuilder {
    pub fn new(comparator: impl Into<MultiVectorComparator>) -> Self {
        let builder = Self::empty();
        builder.comparator(comparator.into() as i32)
    }
}