qdrant_client/builders/
sparse_vector_builder.rs

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
use crate::qdrant::*;

#[derive(Default)]
pub struct SparseVectorBuilder {
    pub(crate) indices: Vec<u32>,
    pub(crate) values: Vec<f32>,
}

impl SparseVectorBuilder {
    pub fn new(indices: impl Into<Vec<u32>>, values: impl Into<Vec<f32>>) -> Self {
        Self {
            indices: indices.into(),
            values: values.into(),
        }
    }

    #[allow(unused_mut)]
    pub fn indices(mut self, indices: impl Into<Vec<u32>>) -> Self {
        self.indices = indices.into();
        self
    }

    #[allow(unused_mut)]
    pub fn values(mut self, values: impl Into<Vec<f32>>) -> Self {
        self.values = values.into();
        self
    }

    /// Builds the desired type. Can often be omitted.
    pub fn build(self) -> SparseVector {
        SparseVector {
            indices: self.indices,
            values: self.values,
        }
    }
}

impl From<(Vec<u32>, Vec<f32>)> for SparseVector {
    fn from((indices, values): (Vec<u32>, Vec<f32>)) -> Self {
        SparseVectorBuilder::new(indices, values).build()
    }
}

impl From<SparseVector> for Vector {
    fn from(dense_vector: SparseVector) -> Self {
        crate::qdrant::vector::Vector::from(dense_vector).into()
    }
}

impl From<SparseVectorBuilder> for Vector {
    fn from(dense_vector: SparseVectorBuilder) -> Self {
        crate::qdrant::vector::Vector::from(dense_vector.build()).into()
    }
}

impl From<SparseVector> for crate::qdrant::vector::Vector {
    fn from(dense_vector: SparseVector) -> Self {
        Self::Sparse(dense_vector)
    }
}