use crate::segment::types::{
Distance, HnswConfig, MultiVectorConfig, QuantizationConfig, VectorStorageDatatype,
};
use crate::edge::config::vectors::EdgeVectorParams;
#[derive(Debug, Clone)]
pub struct EdgeVectorParamsBuilder {
size: usize,
distance: Distance,
on_disk: Option<bool>,
multivector_config: Option<MultiVectorConfig>,
datatype: Option<VectorStorageDatatype>,
quantization_config: Option<QuantizationConfig>,
hnsw_config: Option<HnswConfig>,
}
impl EdgeVectorParamsBuilder {
pub fn new(size: usize, distance: Distance) -> Self {
Self {
size,
distance,
on_disk: None,
multivector_config: None,
datatype: None,
quantization_config: None,
hnsw_config: None,
}
}
pub fn on_disk(mut self, on_disk: bool) -> Self {
self.on_disk = Some(on_disk);
self
}
pub fn multivector_config(mut self, multivector_config: MultiVectorConfig) -> Self {
self.multivector_config = Some(multivector_config);
self
}
pub fn datatype(mut self, datatype: VectorStorageDatatype) -> Self {
self.datatype = Some(datatype);
self
}
pub fn quantization_config(mut self, quantization_config: QuantizationConfig) -> Self {
self.quantization_config = Some(quantization_config);
self
}
pub fn hnsw_config(mut self, hnsw_config: HnswConfig) -> Self {
self.hnsw_config = Some(hnsw_config);
self
}
pub fn build(self) -> EdgeVectorParams {
let Self {
size,
distance,
on_disk,
multivector_config,
datatype,
quantization_config,
hnsw_config,
} = self;
EdgeVectorParams {
size,
distance,
on_disk,
multivector_config,
datatype,
quantization_config,
hnsw_config,
}
}
}