yujunjun/LightRAG
1
0
Fork 0
mirror of https://github.com/HKUDS/LightRAG.git synced 2025-09-26 16:47:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #836 from YanSte/drop

Browse Source 
fixed drop
This commit is contained in:
Yannick Stephan 2025-02-18 10:22:37 +01:00 committed by GitHub
commit 00704aa76a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
4 changed files with 0 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
lightrag/base.py
View File

@ -135,10 +135,6 @@ class BaseKVStorage(StorageNameSpace, ABC):
async def upsert(self, data: dict[str, dict[str, Any]]) -> None: async def upsert(self, data: dict[str, dict[str, Any]]) -> None:
"""Upsert data""" """Upsert data"""
@abstractmethod
async def drop(self) -> None:
"""Drop the storage"""
@dataclass @dataclass
class BaseGraphStorage(StorageNameSpace, ABC): class BaseGraphStorage(StorageNameSpace, ABC):

3
lightrag/kg/json_kv_impl.py
View File

@ -50,6 +50,3 @@ class JsonKVStorage(BaseKVStorage):
for doc_id in ids: for doc_id in ids:
self._data.pop(doc_id, None) self._data.pop(doc_id, None)
await self.index_done_callback() await self.index_done_callback()
async def drop(self) -> None:
self._data = {}

8
lightrag/kg/mongo_impl.py
View File

@ -117,10 +117,6 @@ class MongoKVStorage(BaseKVStorage):
# Mongo handles persistence automatically # Mongo handles persistence automatically
pass pass
async def drop(self) -> None:
"""Drop the collection"""
await self._data.drop()
@final @final
@dataclass @dataclass
@ -202,10 +198,6 @@ class MongoDocStatusStorage(DocStatusStorage):
# Mongo handles persistence automatically # Mongo handles persistence automatically
pass pass
async def drop(self) -> None:
"""Drop the collection"""
await self._data.drop()
@final @final
@dataclass @dataclass

5
lightrag/kg/redis_impl.py
View File

@ -61,8 +61,3 @@ class RedisKVStorage(BaseKVStorage):
async def index_done_callback(self) -> None: async def index_done_callback(self) -> None:
# Redis handles persistence automatically # Redis handles persistence automatically
pass pass
async def drop(self) -> None:
keys = await self._redis.keys(f"{self.namespace}:*")
if keys:
await self._redis.delete(*keys)