[bug]: Memory.reset() deletes collection and table without re-creating it (#1952)

This commit is contained in:
Mayank
2024-10-15 16:46:50 +05:30
committed by GitHub
parent 9341d9f597
commit 3f2d5bee34
3 changed files with 16 additions and 5 deletions

View File

@@ -542,7 +542,7 @@ class Memory(MemoryBase):
try:
existing_memory = self.vector_store.get(vector_id=memory_id)
except Exception as e:
except Exception:
raise ValueError(f"Error getting memory with ID {memory_id}. Please provide a valid 'memory_id'")
prev_value = existing_memory.payload.get("data")
@@ -591,6 +591,9 @@ class Memory(MemoryBase):
"""
logger.warning("Resetting all memories")
self.vector_store.delete_col()
self.vector_store = VectorStoreFactory.create(
self.config.vector_store.provider, self.config.vector_store.config
)
self.db.reset()
capture_event("mem0.reset", self)