From 75a2b11a2b7843d6a8f059fce384066e265f8180 Mon Sep 17 00:00:00 2001 From: Lorenze Jay Date: Mon, 9 Dec 2024 08:26:58 -0800 Subject: [PATCH] cleanup --- src/crewai/knowledge/storage/knowledge_storage.py | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/src/crewai/knowledge/storage/knowledge_storage.py b/src/crewai/knowledge/storage/knowledge_storage.py index a8cc700cf..23fe688eb 100644 --- a/src/crewai/knowledge/storage/knowledge_storage.py +++ b/src/crewai/knowledge/storage/knowledge_storage.py @@ -16,6 +16,7 @@ from crewai.knowledge.storage.base_knowledge_storage import BaseKnowledgeStorage from crewai.utilities import EmbeddingConfigurator from crewai.utilities.logger import Logger from crewai.utilities.paths import db_storage_path +from crewai.utilities.constants import KNOWLEDGE_DIRECTORY @contextlib.contextmanager @@ -106,7 +107,7 @@ class KnowledgeStorage(BaseKnowledgeStorage): raise Exception("Failed to create or get collection") def reset(self): - base_path = os.path.join(db_storage_path(), "knowledge") + base_path = os.path.join(db_storage_path(), KNOWLEDGE_DIRECTORY) if not self.app: self.app = chromadb.PersistentClient( path=base_path, @@ -116,7 +117,7 @@ class KnowledgeStorage(BaseKnowledgeStorage): self.app.reset() self.app = None self.collection = None - self._remove_knowledge_storage_folders() + shutil.rmtree(base_path) def save( self, @@ -183,7 +184,3 @@ class KnowledgeStorage(BaseKnowledgeStorage): if embedder_config else self._create_default_embedding_function() ) - - def _remove_knowledge_storage_folders(self): - base_path = os.path.join(db_storage_path(), "knowledge") - shutil.rmtree(base_path)