fix: shared data intitialization failed for multi-worker

This commit is contained in:
yangdx
2025-02-26 13:32:15 +08:00
parent 2c019dbc7b
commit 41f5d208a9

View File

@@ -94,7 +94,6 @@ def create_app(args):
logger.setLevel(getattr(logging, args.log_level))
set_verbose_debug(args.verbose)
# Display splash screen
from lightrag.kg.shared_storage import is_multiprocess
logger.info(f"==== Multi-processor mode: {is_multiprocess} ====")
@@ -421,9 +420,9 @@ def get_application():
import types
args = types.SimpleNamespace(**json.loads(args_json))
# if args.workers > 1:
# from lightrag.kg.shared_storage import initialize_manager
# initialize_manager()
if args.workers > 1:
from lightrag.kg.shared_storage import initialize_share_data
initialize_share_data()
return create_app(args)
@@ -486,6 +485,7 @@ def main():
display_splash_screen(args)
uvicorn_config = {
"app": "lightrag.api.lightrag_server:get_application",
"factory": True,