diff --git a/lightrag/base.py b/lightrag/base.py index 2dc7b035..79cc5639 100644 --- a/lightrag/base.py +++ b/lightrag/base.py @@ -135,10 +135,6 @@ class BaseKVStorage(StorageNameSpace, ABC): async def upsert(self, data: dict[str, dict[str, Any]]) -> None: """Upsert data""" - @abstractmethod - async def drop(self) -> None: - """Drop the storage""" - @dataclass class BaseGraphStorage(StorageNameSpace, ABC): diff --git a/lightrag/kg/json_kv_impl.py b/lightrag/kg/json_kv_impl.py index 0566fb26..7e13dea7 100644 --- a/lightrag/kg/json_kv_impl.py +++ b/lightrag/kg/json_kv_impl.py @@ -50,6 +50,3 @@ class JsonKVStorage(BaseKVStorage): for doc_id in ids: self._data.pop(doc_id, None) await self.index_done_callback() - - async def drop(self) -> None: - self._data = {} diff --git a/lightrag/kg/mongo_impl.py b/lightrag/kg/mongo_impl.py index 87c9f9f1..8cfc84b9 100644 --- a/lightrag/kg/mongo_impl.py +++ b/lightrag/kg/mongo_impl.py @@ -117,10 +117,6 @@ class MongoKVStorage(BaseKVStorage): # Mongo handles persistence automatically pass - async def drop(self) -> None: - """Drop the collection""" - await self._data.drop() - @final @dataclass @@ -202,10 +198,6 @@ class MongoDocStatusStorage(DocStatusStorage): # Mongo handles persistence automatically pass - async def drop(self) -> None: - """Drop the collection""" - await self._data.drop() - @final @dataclass diff --git a/lightrag/kg/redis_impl.py b/lightrag/kg/redis_impl.py index 3b8ebee4..056fbc9e 100644 --- a/lightrag/kg/redis_impl.py +++ b/lightrag/kg/redis_impl.py @@ -61,8 +61,3 @@ class RedisKVStorage(BaseKVStorage): async def index_done_callback(self) -> None: # Redis handles persistence automatically pass - - async def drop(self) -> None: - keys = await self._redis.keys(f"{self.namespace}:*") - if keys: - await self._redis.delete(*keys)