yujunjun/LightRAG
1
0
Fork 0
mirror of https://github.com/HKUDS/LightRAG.git synced 2025-11-15 09:33:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #831 from YanSte/drops

Browse Source 
fallback default drops
This commit is contained in:
Yannick Stephan 2025-02-18 08:46:42 +01:00 committed by GitHub
commit e4e678f80b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 16 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
lightrag/kg/json_kv_impl.py
View File

@ -50,3 +50,6 @@ 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,6 +117,10 @@ 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
@ -198,6 +202,10 @@ 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,3 +61,8 @@ 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)