mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-09-05 02:51:07 +00:00
refactor: The first refactored version for sdk release (#907)
Co-authored-by: chengfangyin2 <chengfangyin3@jd.com>
This commit is contained in:
36
dbgpt/app/_cli.py
Normal file
36
dbgpt/app/_cli.py
Normal file
@@ -0,0 +1,36 @@
|
||||
import click
|
||||
import os
|
||||
from dbgpt.app.base import WebServerParameters
|
||||
from dbgpt.configs.model_config import LOGDIR
|
||||
from dbgpt.util.parameter_utils import EnvArgumentParser
|
||||
from dbgpt.util.command_utils import _run_current_with_daemon, _stop_service
|
||||
|
||||
|
||||
@click.command(name="webserver")
|
||||
@EnvArgumentParser.create_click_option(WebServerParameters)
|
||||
def start_webserver(**kwargs):
|
||||
"""Start webserver(dbgpt_server.py)"""
|
||||
if kwargs["daemon"]:
|
||||
log_file = os.path.join(LOGDIR, "webserver_uvicorn.log")
|
||||
_run_current_with_daemon("WebServer", log_file)
|
||||
else:
|
||||
from dbgpt.app.dbgpt_server import run_webserver
|
||||
|
||||
run_webserver(WebServerParameters(**kwargs))
|
||||
|
||||
|
||||
@click.command(name="webserver")
|
||||
@click.option(
|
||||
"--port",
|
||||
type=int,
|
||||
default=None,
|
||||
required=False,
|
||||
help=("The port to stop"),
|
||||
)
|
||||
def stop_webserver(port: int):
|
||||
"""Stop webserver(dbgpt_server.py)"""
|
||||
_stop_service("webserver", "WebServer", port=port)
|
||||
|
||||
|
||||
def _stop_all_dbgpt_server():
|
||||
_stop_service("webserver", "WebServer")
|
Reference in New Issue
Block a user