From 708f80f43da3c1b521e0a92c24e5b6c605405f98 Mon Sep 17 00:00:00 2001
From: yangdx <gzdaniel@me.com>
Date: Mon, 17 Nov 2025 13:46:46 +0800
Subject: [PATCH] Add _default_workspace to shared storage finalization

- Add _default_workspace to global vars
- Set _default_workspace to None on cleanup
- Ensure complete resource cleanup
- Fix missing workspace finalization

(cherry picked from commit 6d6716e9f83508f04a4581f67506b3ea2b1869f4)
---
 lightrag/kg/shared_storage.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/lightrag/kg/shared_storage.py b/lightrag/kg/shared_storage.py
index 438fd65d..d9905f48 100644
--- a/lightrag/kg/shared_storage.py
+++ b/lightrag/kg/shared_storage.py
@@ -1456,7 +1456,8 @@ def finalize_share_data():
         _init_flags, \
         _initialized, \
         _update_flags, \
-        _async_locks
+        _async_locks, \
+        _default_workspace
 
     # Check if already initialized
     if not _initialized:
@@ -1522,5 +1523,6 @@ def finalize_share_data():
     _data_init_lock = None
     _update_flags = None
     _async_locks = None
+    _default_workspace = None
 
     direct_log(f"Process {os.getpid()} storage data finalization complete")