github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-08-12 21:46:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix:port define

Browse Source 
can re-define port
This commit is contained in:
aries_ckt 2023-06-29 16:07:39 +08:00
parent 295b759ba4
commit 5951215c0b
1 changed files with 10 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
pilot/server/dbgpt_server.py
View File

@ -1,3 +1,4 @@
import signal
import traceback import traceback
import os import os
import shutil import shutil
@ -9,10 +10,7 @@ sys.path.append(ROOT_PATH)
from pilot.configs.config import Config from pilot.configs.config import Config
from pilot.configs.model_config import ( from pilot.configs.model_config import (
DATASETS_DIR, LOGDIR
KNOWLEDGE_UPLOAD_ROOT_PATH,
LLM_MODEL_CONFIG,
LOGDIR,
) )
from pilot.utils import build_logger from pilot.utils import build_logger
@ -35,6 +33,11 @@ CFG = Config()
logger = build_logger("webserver", LOGDIR + "webserver.log") logger = build_logger("webserver", LOGDIR + "webserver.log")
def signal_handler(sig, frame):
print("in order to avoid chroma db atexit problem")
os._exit(0)
def swagger_monkey_patch(*args, **kwargs): def swagger_monkey_patch(*args, **kwargs):
return get_swagger_ui_html( return get_swagger_ui_html(
*args, *args,
@ -72,9 +75,10 @@ if __name__ == "__main__":
# old version server config # old version server config
parser.add_argument("--host", type=str, default="0.0.0.0") parser.add_argument("--host", type=str, default="0.0.0.0")
parser.add_argument("--port", type=int, default=CFG.WEB_SERVER_PORT) parser.add_argument("--port", type=int, default=5000)
parser.add_argument("--concurrency-count", type=int, default=10) parser.add_argument("--concurrency-count", type=int, default=10)
parser.add_argument("--share", default=False, action="store_true") parser.add_argument("--share", default=False, action="store_true")
signal.signal(signal.SIGINT, signal_handler)
# init server config # init server config
args = parser.parse_args() args = parser.parse_args()
@ -82,4 +86,4 @@ if __name__ == "__main__":
CFG.NEW_SERVER_MODE = True CFG.NEW_SERVER_MODE = True
import uvicorn import uvicorn
uvicorn.run(app, host="0.0.0.0", port=5000) uvicorn.run(app, host="0.0.0.0", port=args.port)