From 0994d478f08917a15ad01da93e87dcd7607ba995 Mon Sep 17 00:00:00 2001 From: Yannick Stephan Date: Tue, 18 Feb 2025 10:21:54 +0100 Subject: [PATCH] cleaned code --- lightrag/kg/json_kv_impl.py | 2 +- lightrag/kg/mongo_impl.py | 2 ++ lightrag/kg/redis_impl.py | 2 +- 3 files changed, 4 insertions(+), 2 deletions(-) diff --git a/lightrag/kg/json_kv_impl.py b/lightrag/kg/json_kv_impl.py index 54748a27..7e13dea7 100644 --- a/lightrag/kg/json_kv_impl.py +++ b/lightrag/kg/json_kv_impl.py @@ -49,4 +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() \ No newline at end of file + await self.index_done_callback() diff --git a/lightrag/kg/mongo_impl.py b/lightrag/kg/mongo_impl.py index 0e58e538..8cfc84b9 100644 --- a/lightrag/kg/mongo_impl.py +++ b/lightrag/kg/mongo_impl.py @@ -117,6 +117,7 @@ class MongoKVStorage(BaseKVStorage): # Mongo handles persistence automatically pass + @final @dataclass class MongoDocStatusStorage(DocStatusStorage): @@ -197,6 +198,7 @@ class MongoDocStatusStorage(DocStatusStorage): # Mongo handles persistence automatically pass + @final @dataclass class MongoGraphStorage(BaseGraphStorage): diff --git a/lightrag/kg/redis_impl.py b/lightrag/kg/redis_impl.py index 3aa5c892..056fbc9e 100644 --- a/lightrag/kg/redis_impl.py +++ b/lightrag/kg/redis_impl.py @@ -60,4 +60,4 @@ class RedisKVStorage(BaseKVStorage): async def index_done_callback(self) -> None: # Redis handles persistence automatically - pass \ No newline at end of file + pass