andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-11 00:58:30 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

fix: Clean up both ChromaDB files and temp directories in reset

Browse Source 
Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
Devin AI
2025-02-09 23:28:47 +00:00
parent 0dfa5b05d3
commit 08b541cc75
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/crewai/memory/storage/rag_storage.py
View File

@@ -153,9 +153,13 @@ class RAGStorage(BaseRAGStorage):
try: try:
if self.app: if self.app:
self.app.reset() self.app.reset()
# Clean up ChromaDB files
storage_path = os.path.join(db_storage_path(), self.type) storage_path = os.path.join(db_storage_path(), self.type)
if os.path.exists(storage_path): if os.path.exists(storage_path):
shutil.rmtree(storage_path) shutil.rmtree(storage_path)
# Clean up temporary directory
if os.path.exists(self.path):
shutil.rmtree(self.path)
self.app = None self.app = None
self.collection = None self.collection = None
except Exception as e: except Exception as e: