2023-05-15 08:51:32 +08:00
|
|
|
#!/bin/bash
|
|
|
|
|
|
|
|
|
|
set -e
|
|
|
|
|
|
|
|
|
|
if [[ "${MIGRATION_ENABLED}" == "true" ]]; then
|
|
|
|
|
echo "Running migrations"
|
|
|
|
|
flask db upgrade
|
|
|
|
|
fi
|
|
|
|
|
|
|
|
|
|
if [[ "${MODE}" == "worker" ]]; then
|
2023-07-31 13:13:08 +08:00
|
|
|
celery -A app.celery worker -P ${CELERY_WORKER_CLASS:-gevent} -c ${CELERY_WORKER_AMOUNT:-1} --loglevel INFO \
|
|
|
|
|
-Q ${CELERY_QUEUES:-dataset,generation,mail}
|
2024-01-02 15:29:18 +08:00
|
|
|
elif [[ "${MODE}" == "beat" ]]; then
|
|
|
|
|
celery -A app.celery beat --loglevel INFO
|
2023-05-15 08:51:32 +08:00
|
|
|
else
|
|
|
|
|
if [[ "${DEBUG}" == "true" ]]; then
|
|
|
|
|
flask run --host=${DIFY_BIND_ADDRESS:-0.0.0.0} --port=${DIFY_PORT:-5001} --debug
|
|
|
|
|
else
|
|
|
|
|
gunicorn \
|
|
|
|
|
--bind "${DIFY_BIND_ADDRESS:-0.0.0.0}:${DIFY_PORT:-5001}" \
|
|
|
|
|
--workers ${SERVER_WORKER_AMOUNT:-1} \
|
|
|
|
|
--worker-class ${SERVER_WORKER_CLASS:-gevent} \
|
|
|
|
|
--timeout ${GUNICORN_TIMEOUT:-200} \
|
|
|
|
|
--preload \
|
|
|
|
|
app:app
|
|
|
|
|
fi
|
|
|
|
|
fi
|
