diff --git a/apps/common/management/commands/services/services/local_model.py b/apps/common/management/commands/services/services/local_model.py index 35c5ae92e..d1a248118 100644 --- a/apps/common/management/commands/services/services/local_model.py +++ b/apps/common/management/commands/services/services/local_model.py @@ -36,7 +36,8 @@ class GunicornLocalModelService(BaseService): '--max-requests-jitter', '2048', '--access-logformat', log_format, '--access-logfile', '/dev/null', - '--error-logfile', '-' + '--error-logfile', '-', + '-e', 'ENABLE_LOCAL_MODEL=1' ] if DEBUG: cmd.append('--reload') diff --git a/apps/maxkb/wsgi.py b/apps/maxkb/wsgi.py index fc271a7c3..6c44716c0 100644 --- a/apps/maxkb/wsgi.py +++ b/apps/maxkb/wsgi.py @@ -30,8 +30,10 @@ def post_scheduler_handler(): job.run() -# 启动后处理函数 -post_handler() +# 仅在非local_model模式下启动后处理函数,dev celery scheduler 不需要 +if os.environ.get("ENABLE_LOCAL_MODEL") != '1': + # 启动后处理函数 + post_handler() # 仅在scheduler中启动定时任务,dev local_model celery 不需要 if os.environ.get('ENABLE_SCHEDULER') == '1':