Fix disk I/O error when resetting short-term memory.

Reset chromadb client and nullifies references before
removing directory.
This commit is contained in:
Carlos Souza
2024-12-07 06:20:04 -05:00
parent 6930b68484
commit 409fdb4b20

View File

@@ -150,9 +150,11 @@ class RAGStorage(BaseRAGStorage):
def reset(self) -> None:
try:
shutil.rmtree(f"{db_storage_path()}/{self.type}")
if self.app:
self.app.reset()
shutil.rmtree(f"{db_storage_path()}/{self.type}")
self.app = None
self.collection = None
except Exception as e:
if "attempt to write a readonly database" in str(e):
# Ignore this specific error