Update entrypoint.sh

entrypoint.sh

@@ -1,11 +1,16 @@
+#!/bin/bash
+
 set -e
 
+cd /app/api
+
 if [[ "${MIGRATION_ENABLED}" == "true" ]]; then
   echo "Running migrations"
-  cd /app/api && poetry run flask db upgrade
+  poetry run flask db upgrade
 fi
 
 if [[ "${MODE}" == "worker" ]]; then
+  # Get the number of available CPU cores
   if [ "${CELERY_AUTO_SCALE,,}" = "true" ]; then
     AVAILABLE_CORES=$(nproc)
     MAX_WORKERS=${CELERY_MAX_WORKERS:-$AVAILABLE_CORES}
@@ -15,11 +20,20 @@ if [[ "${MODE}" == "worker" ]]; then
     CONCURRENCY_OPTION="-c ${CELERY_WORKER_AMOUNT:-1}"
   fi
 
-  exec poetry run celery -A app.celery worker -P ${CELERY_WORKER_CLASS:-gevent} $CONCURRENCY_OPTION --loglevel ${LOG_LEVEL} \
+  exec poetry run celery -A app.celery worker \
+    -P ${CELERY_WORKER_CLASS:-gevent} \
+    $CONCURRENCY_OPTION \
+    --loglevel ${LOG_LEVEL} \
     -Q ${CELERY_QUEUES:-dataset,mail,ops_trace,app_deletion}
+
+elif [[ "${MODE}" == "beat" ]]; then
+  exec poetry run celery -A app.celery beat --loglevel ${LOG_LEVEL}
 else
   if [[ "${DEBUG}" == "true" ]]; then
-    exec poetry run flask run --host=${DIFY_BIND_ADDRESS:-0.0.0.0} --port=7860 --debug
+    exec poetry run flask run \
+      --host=${DIFY_BIND_ADDRESS:-0.0.0.0} \
+      --port=7860 \
+      --debug
   else
     exec poetry run gunicorn \
       --bind "0.0.0.0:7860" \