From fc0cf2934e4b60a5fca94dfaf99d55c1a27ae4ad Mon Sep 17 00:00:00 2001
From: Yannick Stephan <stephan.yannick@me.com>
Date: Tue, 18 Feb 2025 10:21:14 +0100
Subject: [PATCH] fixed drop

---
 lightrag/base.py            |  4 ----
 lightrag/kg/json_kv_impl.py |  5 +----
 lightrag/kg/mongo_impl.py   | 10 ----------
 lightrag/kg/redis_impl.py   |  7 +------
 4 files changed, 2 insertions(+), 24 deletions(-)

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..54748a27 100644
--- a/lightrag/kg/json_kv_impl.py
+++ b/lightrag/kg/json_kv_impl.py
@@ -49,7 +49,4 @@ class JsonKVStorage(BaseKVStorage):
     async def delete(self, ids: list[str]) -> None:
         for doc_id in ids:
             self._data.pop(doc_id, None)
-        await self.index_done_callback()
-
-    async def drop(self) -> None:
-        self._data = {}
+        await self.index_done_callback()
\ No newline at end of file
diff --git a/lightrag/kg/mongo_impl.py b/lightrag/kg/mongo_impl.py
index 87c9f9f1..0e58e538 100644
--- a/lightrag/kg/mongo_impl.py
+++ b/lightrag/kg/mongo_impl.py
@@ -117,11 +117,6 @@ class MongoKVStorage(BaseKVStorage):
         # Mongo handles persistence automatically
         pass
 
-    async def drop(self) -> None:
-        """Drop the collection"""
-        await self._data.drop()
-
-
 @final
 @dataclass
 class MongoDocStatusStorage(DocStatusStorage):
@@ -202,11 +197,6 @@ class MongoDocStatusStorage(DocStatusStorage):
         # Mongo handles persistence automatically
         pass
 
-    async def drop(self) -> None:
-        """Drop the collection"""
-        await self._data.drop()
-
-
 @final
 @dataclass
 class MongoGraphStorage(BaseGraphStorage):
diff --git a/lightrag/kg/redis_impl.py b/lightrag/kg/redis_impl.py
index 3b8ebee4..3aa5c892 100644
--- a/lightrag/kg/redis_impl.py
+++ b/lightrag/kg/redis_impl.py
@@ -60,9 +60,4 @@ 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)
+        pass
\ No newline at end of file