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,