Function index_embedding

src/vector_store.rs:782–940

(
    config: &Config,
    hnsw_index: &HNSWIndex,
    parent_lazy_item_latest_ptr: SharedLatestNode,
    prop_value: Arc<NodePropValue>,
    prop_metadata: Option<Arc<NodePropMetadata>>,
    current_

#[allow(clippy::too_many_arguments)]
pub fn index_embedding(
    config: &Config,
    hnsw_index: &HNSWIndex,
    parent_lazy_item_latest_ptr: SharedLatestNode,
    prop_value: Arc<NodePropValue>,
    prop_metadata: Option<Arc<NodePropMetadata>>,
    current_lazy_item_latest_ptr: SharedLatestNode,
    cur_level: HNSWLevel,
    version: VersionNumber,
    file_id: IndexFileId,
    hnsw_params: &HNSWHyperParams,
    max_level: u8,
    offset_counter: &HNSWIndexFileOffsetCounter,
    distance_metric: DistanceMetric,
) -> Result<(), WaCustomError> {
    let fvec = &prop_value.vec;
    let mut skipm = PerformantFixedSet::new(if cur_level.0 == 0 {
        hnsw_params.level_0_neighbors_count
    } else {
        hnsw_params.neighbors_count
    });
    let new_node_id = match &prop_metadata {
        Some(metadata) => *metadata.replica_id,
        None => *prop_value.id,
    };
    skipm.insert(new_node_id);

    let current_lazy_item = unsafe { &*current_lazy_item_latest_ptr }.latest;
    let current_node = unsafe { &*current_lazy_item }.try_get_data(&hnsw_index.cache)?;

    let cur_node_id = &current_node.get_id();

    let mdims = prop_metadata.clone().map(|pm| pm.vec.clone());

    let z = traverse_find_nearest(
        config,
        hnsw_index,
        current_lazy_item_latest_ptr,
        fvec,
        Some(&prop_value.id),
        mdims.as_deref(),
        &mut 0,
        &mut skipm,
        &distance_metric,
        true,
        hnsw_params.ef_construction,
    )?;

    let z = if z.is_empty() {
        let fvec_data = VectorData {
            id: None,
            quantized_vec: fvec,
            metadata: mdims.as_deref(),
        };
        let cur_node_metadata = current_node.prop_metadata.clone().map(|pm| pm.vec.clone());
        let cur_node_data = VectorData {
            id: Some(cur_node_id),
            quantized_vec: &current_node.prop_value.vec,