From 8acc2dc8613d7acbacbc345deb1929f20ed36cab Mon Sep 17 00:00:00 2001 From: aries_ckt <916701291@qq.com> Date: Thu, 19 Oct 2023 21:10:33 +0800 Subject: [PATCH] style:fmt --- pilot/server/base.py | 1 + pilot/server/dbgpt_server.py | 3 +++ pilot/server/llmserver.py | 3 +++ 3 files changed, 7 insertions(+) diff --git a/pilot/server/base.py b/pilot/server/base.py index 75e16ab85..e66efcf43 100644 --- a/pilot/server/base.py +++ b/pilot/server/base.py @@ -21,6 +21,7 @@ def signal_handler(sig, frame): def async_db_summary(system_app: SystemApp): + """async db schema into vector db""" from pilot.summary.db_summary_client import DBSummaryClient client = DBSummaryClient(system_app=system_app) diff --git a/pilot/server/dbgpt_server.py b/pilot/server/dbgpt_server.py index 5b57d18bb..c6e084a93 100644 --- a/pilot/server/dbgpt_server.py +++ b/pilot/server/dbgpt_server.py @@ -115,6 +115,9 @@ def _get_webserver_params(args: List[str] = None): def initialize_app(param: WebWerverParameters = None, args: List[str] = None): """Initialize app If you use gunicorn as a process manager, initialize_app can be invoke in `on_starting` hook. + Args: + param:WebWerverParameters + args:List[str] """ if not param: param = _get_webserver_params(args) diff --git a/pilot/server/llmserver.py b/pilot/server/llmserver.py index 1a2dd49ce..9526f38b5 100644 --- a/pilot/server/llmserver.py +++ b/pilot/server/llmserver.py @@ -16,6 +16,9 @@ CFG = Config() model_path = LLM_MODEL_CONFIG.get(CFG.LLM_MODEL) if __name__ == "__main__": + """run llm server including controller, manager worker + If you use gunicorn as a process manager, initialize_app can be invoke in `on_starting` hook. + """ run_worker_manager( model_name=CFG.LLM_MODEL, model_path=model_path,