Function build_collection_codec

nodedb-vector/src/collection/codec_dispatch.rs:69–100

Build a `CollectionCodec` from a quantization tag and training vectors. Returns `None` for unsupported or unrecognised tags (e.g. "sq8", "pq", "none") — those variants use separate per-segment code paths.

(
    quantization: &str,
    vectors: &[Vec<f32>],
    dim: usize,
    m: usize,
    ef_construction: usize,
    seed: u64,
)

/// Returns `None` for unsupported or unrecognised tags (e.g. "sq8", "pq",
/// "none") — those variants use separate per-segment code paths.
pub fn build_collection_codec(
    quantization: &str,
    vectors: &[Vec<f32>],
    dim: usize,
    m: usize,
    ef_construction: usize,
    seed: u64,
) -> Option<CollectionCodec> {
    if vectors.is_empty() {
        return None;
    }
    let refs: Vec<&[f32]> = vectors.iter().map(|v| v.as_slice()).collect();
    match quantization {
        "rabitq" => {
            let codec = RaBitQCodec::calibrate(&refs, dim, seed);
            let mut idx = HnswCodecIndex::new(dim, m, ef_construction, codec, seed);
            for (i, v) in vectors.iter().enumerate() {
                idx.insert(i as u32, v);
            }
            Some(CollectionCodec::RaBitQ(idx))
        }
        "bbq" => {
            let codec = BbqCodec::calibrate(&refs, dim, 3);
            let mut idx = HnswCodecIndex::new(dim, m, ef_construction, codec, seed);
            for (i, v) in vectors.iter().enumerate() {
                idx.insert(i as u32, v);
            }
            Some(CollectionCodec::Bbq(idx))
        }
        _ => None,
    }
}

#[cfg(test)]
mod tests {