Avoid redundant llm cache updates

2025-03-10 01:45:58 +08:00
parent d2708b966d
commit 6b0acce644
3 changed files with 30 additions and 6 deletions
--- a/lightrag/kg/json_kv_impl.py
+++ b/lightrag/kg/json_kv_impl.py
@@ -18,6 +18,7 @@ from .shared_storage import (
    set_all_update_flags,
    clear_all_update_flags,
    try_initialize_namespace,
+    is_multiprocess,
 )


@@ -57,7 +58,7 @@ class JsonKVStorage(BaseKVStorage):

    async def index_done_callback(self) -> None:
        async with self._storage_lock:
-            if self.storage_updated:
+            if (is_multiprocess and self.storage_updated.value) or (not is_multiprocess and self.storage_updated):
                data_dict = (
                    dict(self._data) if hasattr(self._data, "_getvalue") else self._data
                )