minne/eval/src/db_helpers.rs

use anyhow::{Context, Result};
use common::storage::{db::SurrealDbClient, indexes::ensure_runtime_indexes};
use tracing::info;

// Remove and recreate HNSW indexes for changing embedding lengths, used at beginning if embedding length differs from default system settings.
pub async fn change_embedding_length_in_hnsw_indexes(
    db: &SurrealDbClient,
    dimension: usize,
) -> Result<()> {
    // No-op for now; runtime indexes are created after ingestion with the correct dimension.
    let _ = (db, dimension);
    Ok(())
}

// Helper functions for index management during namespace reseed
pub async fn remove_all_indexes(db: &SurrealDbClient) -> Result<()> {
    let _ = db;
    info!("Removing ALL indexes before namespace reseed (no-op placeholder)");
    Ok(())
}

pub async fn recreate_indexes(db: &SurrealDbClient, dimension: usize) -> Result<()> {
    info!("Recreating ALL indexes after namespace reseed via shared runtime helper");
    ensure_runtime_indexes(db, dimension)
        .await
        .context("creating runtime indexes")
}

pub async fn reset_namespace(db: &SurrealDbClient, namespace: &str, database: &str) -> Result<()> {
    let query = format!(
        "REMOVE NAMESPACE {ns};
         DEFINE NAMESPACE {ns};
         DEFINE DATABASE {db};",
        ns = namespace,
        db = database
    );
    db.client
        .query(query)
        .await
        .context("resetting SurrealDB namespace")?;
    db.client
        .use_ns(namespace)
        .use_db(database)
        .await
        .context("selecting namespace/database after reset")?;
    Ok(())
}

#[cfg(test)]
mod tests {
    use super::*;
    use serde::Deserialize;
    use uuid::Uuid;

    #[derive(Debug, Deserialize)]
    struct FooRow {
        label: String,
    }

    #[tokio::test]
    async fn reset_namespace_drops_existing_rows() {
        let namespace = format!("reset_ns_{}", Uuid::new_v4().simple());
        let database = format!("reset_db_{}", Uuid::new_v4().simple());
        let db = SurrealDbClient::memory(&namespace, &database)
            .await
            .expect("in-memory db");

        db.client
            .query(
                "DEFINE TABLE foo SCHEMALESS;
                 CREATE foo:foo SET label = 'before';",
            )
            .await
            .expect("seed namespace")
            .check()
            .expect("seed response");

        let mut before = db
            .client
            .query("SELECT * FROM foo")
            .await
            .expect("select before reset");
        let existing: Vec<FooRow> = before.take(0).expect("rows before reset");
        assert_eq!(existing.len(), 1);
        assert_eq!(existing[0].label, "before");

        reset_namespace(&db, &namespace, &database)
            .await
            .expect("namespace reset");

        match db.client.query("SELECT * FROM foo").await {
            Ok(mut response) => {
                let rows: Vec<FooRow> = response.take(0).unwrap_or_default();
                assert!(
                    rows.is_empty(),
                    "reset namespace should drop rows, found {:?}",
                    rows
                );
            }
            Err(error) => {
                let message = error.to_string();
                assert!(
                    message.to_ascii_lowercase().contains("table")
                        || message.to_ascii_lowercase().contains("namespace")
                        || message.to_ascii_lowercase().contains("foo"),
                    "unexpected error after namespace reset: {message}"
                );
            }
        }
    }
}