Method search_embeddings

atomic-core/src/pristine/txn/read.rs:1173–1224

(
        &self,
        query_vector: &[f32],
        top_k: usize,
    )

    }

    fn search_embeddings(
        &self,
        query_vector: &[f32],
        top_k: usize,
    ) -> PristineResult<Vec<SearchResult>> {
        let table = match self.txn.open_table(EMBEDDINGS) {
            Ok(table) => table,
            Err(redb::TableError::TableDoesNotExist(_)) => return Ok(Vec::new()),
            Err(e) => return Err(PristineError::from(e)),
        };

        let mut scored: Vec<SearchResult> = Vec::new();

        for item in table.iter()? {
            let (key_guard, value_guard) = item?;
            let key_str = key_guard.value();

            let (path, chunk_idx) = match decode_embedding_key(key_str) {
                Some(decoded) => decoded,
                None => continue,
            };

            let bytes = value_guard.value();
            let record: EmbeddingRecord =
                postcard::from_bytes(bytes).map_err(|e| PristineError::Serialization {
                    message: format!(
                        "failed to deserialize EmbeddingRecord at '{}': {}",
                        key_str, e
                    ),
                })?;

            let score = cosine_similarity(query_vector, &record.vector);

            scored.push(SearchResult {
                path: path.to_string(),
                chunk_idx,
                score,
                preview: record.preview,
            });
        }

        // Sort by descending score
        scored.sort_by(|a, b| {
            b.score
                .partial_cmp(&a.score)
                .unwrap_or(std::cmp::Ordering::Equal)
        });

        // Return top-k
        scored.truncate(top_k);
        Ok(scored)
    }
}

// CrdtTxnT (read accessors) implementation for ReadTxn.