You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
71 lines
2.1 KiB
Python
71 lines
2.1 KiB
Python
import logging
|
|
import os
|
|
import signal
|
|
import sys
|
|
import threading
|
|
import time
|
|
import traceback
|
|
|
|
from config import settings, show_configs
|
|
from utils import file_utils
|
|
|
|
stop_event = threading.Event()
|
|
|
|
RAGFLOW_DEBUGPY_LISTEN = int(os.environ.get('RAGFLOW_DEBUGPY_LISTEN', "0"))
|
|
|
|
|
|
def signal_handler(sig, frame):
|
|
logging.info("Received interrupt signal, shutting down...")
|
|
stop_event.set()
|
|
time.sleep(1)
|
|
sys.exit(0)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
logging.info(r"""
|
|
_____ _ ____ _ _ _
|
|
| __ \ | | | _ \ (_) | | |
|
|
| | | | __ _| |_ __ _| |_) |_ _ _| | __| | ___ _ __
|
|
| | | |/ _` | __/ _` | _ <| | | | | |/ _` |/ _ \ '__|
|
|
| |__| | (_| | || (_| | |_) | |_| | | | (_| | __/ |
|
|
|_____/ \__,_|\__\__,_|____/ \__,_|_|_|\__,_|\___|_|
|
|
""")
|
|
|
|
logging.info(
|
|
f'project base: {file_utils.get_project_base_directory()}'
|
|
)
|
|
show_configs()
|
|
# import argparse
|
|
# parser = argparse.ArgumentParser()
|
|
#
|
|
# args = parser.parse_args()
|
|
signal.signal(signal.SIGINT, signal_handler)
|
|
signal.signal(signal.SIGTERM, signal_handler)
|
|
try:
|
|
logging.info("服务启动ing...")
|
|
import uvicorn
|
|
# 配置Uvicorn参数
|
|
uvicorn_config = {
|
|
# "app": app, # FastAPI应用实例
|
|
"app": "config.fastapi_config:app", # FastAPI应用实例
|
|
"host": settings.host_ip,
|
|
"port": settings.host_port,
|
|
"reload": settings.debug, # 开发模式启用热重载
|
|
"log_level": "debug" if settings.debug else "info",
|
|
"access_log": True,
|
|
}
|
|
# 如果是调试模式，添加额外配置
|
|
if settings.debug:
|
|
uvicorn_config.update({
|
|
"reload_dirs": ["."], # 监视当前目录变化
|
|
"reload_delay": 0.5, # 重载延迟
|
|
})
|
|
|
|
# 启动Uvicorn服务器
|
|
uvicorn.run(**uvicorn_config)
|
|
except Exception:
|
|
traceback.print_exc()
|
|
stop_event.set()
|
|
time.sleep(1)
|
|
os.kill(os.getpid(), signal.SIGKILL)
|