Persistence cache data to disk before exiting

2025-04-28 23:16:50 +08:00
parent 0ecae90002
commit 40a2357c14
1 changed files with 7 additions and 0 deletions
--- a/lightrag/kg/json_kv_impl.py
+++ b/lightrag/kg/json_kv_impl.py
@@ -197,3 +197,10 @@ class JsonKVStorage(BaseKVStorage):
        except Exception as e:
            logger.error(f"Error dropping {self.namespace}: {e}")
            return {"status": "error", "message": str(e)}
+
+    async def finalize(self):
+        """Finalize storage resources
+        Persistence cache data to disk before exiting
+        """
+        if self.namespace.endswith("cache"):
+            await self.index_done_callback()