laurus 0.4.0

Unified search library for lexical, vector, and semantic retrieval
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

//! Factory for creating vector index readers.

use std::sync::Arc;

use crate::error::Result;
use crate::vector::index::flat::reader::FlatVectorIndexReader;
use crate::vector::index::hnsw::reader::HnswIndexReader;
use crate::vector::index::ivf::reader::IvfIndexReader;
use crate::vector::reader::VectorIndexReader;

/// Factory for creating vector index readers.
pub struct VectorIndexReaderFactory;

impl VectorIndexReaderFactory {
    /// Create a reader for a specific index type.
    pub fn create_reader(
        index_type: &str,
        index_data: &[u8],
    ) -> Result<Arc<dyn VectorIndexReader>> {
        match index_type.to_lowercase().as_str() {
            "flat" => {
                let reader = FlatVectorIndexReader::from_bytes(index_data)?;
                Ok(Arc::new(reader))
            }
            "hnsw" => {
                let reader = HnswIndexReader::from_bytes(index_data)?;
                Ok(Arc::new(reader))
            }
            "ivf" => {
                let reader = IvfIndexReader::from_bytes(index_data)?;
                Ok(Arc::new(reader))
            }
            _ => Err(crate::error::LaurusError::InvalidOperation(format!(
                "Unknown index type: {index_type}"
            ))),
        }
    }
}