From 20be3393d248506c2d3b3b4f2b3123f4d7c532f7 Mon Sep 17 00:00:00 2001
From: Per Stark <perstark.se@gmail.com>
Date: Mon, 2 Dec 2024 08:53:15 +0100
Subject: [PATCH] chore: renaming indexes

---
 src/storage/db.rs | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/src/storage/db.rs b/src/storage/db.rs
index 005c702..ecd782e 100644
--- a/src/storage/db.rs
+++ b/src/storage/db.rs
@@ -34,12 +34,19 @@ impl SurrealDbClient {
         Ok(SurrealDbClient { client: db })
     }
 
+    pub async fn build_indexes(&self) -> Result<(), Error> {
+        self.client.query("DEFINE INDEX idx_embedding_chunks ON text_chunk FIELDS embedding HNSW DIMENSION 1536").await?;
+        self.client.query("DEFINE INDEX idx_embedding_entities ON knowledge_entity FIELDS embedding HNSW DIMENSION 1536").await?;
+
+        Ok(())
+    }
+
     pub async fn rebuild_indexes(&self) -> Result<(), Error> {
         self.client
-            .query("REBUILD INDEX IF EXISTS idx_embedding ON text_chunk")
+            .query("REBUILD INDEX IF EXISTS idx_embedding_chunks ON text_chunk")
             .await?;
         self.client
-            .query("REBUILD INDEX IF EXISTS embeddings ON knowledge_entity")
+            .query("REBUILD INDEX IF EXISTS idx_embeddings_entities ON knowledge_entity")
             .await?;
         Ok(())
     }