From 41f5d208a9b1204afb9d3bc580a81d633310f9b3 Mon Sep 17 00:00:00 2001 From: yangdx Date: Wed, 26 Feb 2025 13:32:15 +0800 Subject: [PATCH] fix: shared data intitialization failed for multi-worker --- lightrag/api/lightrag_server.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/lightrag/api/lightrag_server.py b/lightrag/api/lightrag_server.py index 56f55833..07108c52 100644 --- a/lightrag/api/lightrag_server.py +++ b/lightrag/api/lightrag_server.py @@ -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,