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/weaviate.py
+++ b/mem0/vector_stores/weaviate.py
@@ -308,3 +308,9 @@ class Weaviate(VectorStoreBase):
            payload["id"] = str(obj.uuid).split("'")[0]
            results.append(OutputData(id=str(obj.uuid).split("'")[0], score=1.0, payload=payload))
        return [results]
+    
+    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()