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/chroma.py
+++ b/mem0/vector_stores/chroma.py
@@ -221,3 +221,9 @@ class ChromaDB(VectorStoreBase):
        """
        results = self.collection.get(where=filters, limit=limit)
        return [self._parse_output(results)]
+    
+    def reset(self):
+        """Reset the index by deleting and recreating it."""
+        logger.warning(f"Resetting index {self.collection_name}...")
+        self.delete_col()
+        self.collection = self.create_col(self.collection_name)