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/milvus.py
+++ b/mem0/vector_stores/milvus.py
@@ -237,3 +237,9 @@ class MilvusDB(VectorStoreBase):
            obj = OutputData(id=data.get("id"), score=None, payload=data.get("metadata"))
            memories.append(obj)
        return [memories]
+    
+    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.collection_name, self.embedding_model_dims, self.metric_type)