diff --git a/pilot/scene/chat_db/auto_execute/out_parser.py b/pilot/scene/chat_db/auto_execute/out_parser.py index fe490fb8d..9b02d1ba1 100644 --- a/pilot/scene/chat_db/auto_execute/out_parser.py +++ b/pilot/scene/chat_db/auto_execute/out_parser.py @@ -35,15 +35,15 @@ class DbChatOutputParser(BaseOutputParser): if len(data) <= 1: data.insert(0, ["result"]) df = pd.DataFrame(data[1:], columns=data[0]) - if CFG.NEW_SERVER_MODE: - html = df.to_html(index=False, escape=False, sparsify=False) - html = "".join(html.split()) - else: + if not CFG.NEW_SERVER_MODE: table_style = """""" html_table = df.to_html(index=False, escape=False) html = f"{table_style}{html_table}" + else: + html = df.to_html(index=False, escape=False, sparsify=False) + html = "".join(html.split()) view_text = f"##### {str(speak)}" + "\n" + html.replace("\n", " ") return view_text diff --git a/pilot/server/dbgpt_server.py b/pilot/server/dbgpt_server.py index 84b667876..0279a2b86 100644 --- a/pilot/server/dbgpt_server.py +++ b/pilot/server/dbgpt_server.py @@ -91,15 +91,19 @@ if __name__ == "__main__": parser.add_argument("--port", type=int, default=5000) parser.add_argument("--concurrency-count", type=int, default=10) parser.add_argument("--share", default=False, action="store_true") + parser.add_argument("-light", "--light", default=False,action="store_true", help="enable light mode") signal.signal(signal.SIGINT, signal_handler) # init server config args = parser.parse_args() - - from pilot.server.llmserver import worker - worker.start_check() server_init(args) - CFG.NEW_SERVER_MODE = True - import uvicorn + if not args.light: + print("Model Unified Deployment Mode!") + from pilot.server.llmserver import worker + worker.start_check() + CFG.NEW_SERVER_MODE = True + + + import uvicorn uvicorn.run(app, host="0.0.0.0", port=args.port)