Reset function for VectorDBs (#2584)

2025-04-25 00:01:53 +05:30
parent ff6ae478f1
commit 64c3d34deb
18 changed files with 224 additions and 20 deletions
--- a/mem0/vector_stores/supabase.py
+++ b/mem0/vector_stores/supabase.py
@@ -229,3 +229,9 @@ class Supabase(VectorStoreBase):
        records = self.collection.fetch(ids=ids)

        return [[OutputData(id=str(record[0]), score=None, payload=record[2]) for record in records]]
+    
+    def reset(self):
+        """Reset the index by deleting and recreating it."""
+        logger.warning(f"Resetting index {self.collection_name}...")
+        self.delete_col()
+        self.create_col(self.embedding_model_dims)