feat: add queue to celery task (#688)

2023-07-31 13:13:08 +08:00
parent 396197e881
commit 0c18cab111
19 changed files with 20 additions and 19 deletions
--- a/api/docker/entrypoint.sh
+++ b/api/docker/entrypoint.sh
@@ -8,7 +8,8 @@ if [[ "${MIGRATION_ENABLED}" == "true" ]]; then
 fi

 if [[ "${MODE}" == "worker" ]]; then
-  celery -A app.celery worker -P ${CELERY_WORKER_CLASS:-gevent} -c ${CELERY_WORKER_AMOUNT:-1} --loglevel INFO
+  celery -A app.celery worker -P ${CELERY_WORKER_CLASS:-gevent} -c ${CELERY_WORKER_AMOUNT:-1} --loglevel INFO \
+    -Q ${CELERY_QUEUES:-dataset,generation,mail}
 else
  if [[ "${DEBUG}" == "true" ]]; then
    flask run --host=${DIFY_BIND_ADDRESS:-0.0.0.0} --port=${DIFY_PORT:-5001} --debug