celery worker log format following LOG_FORMAT env#9404

Signed-off-by: root <[email protected]>
langgenius · Oct 29, 2024 · b53bce5 · b53bce5
1 parent 5580bcf
commit b53bce5
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/api/extensions/ext_celery.py b/api/extensions/ext_celery.py
@@ -5,6 +5,7 @@
 from flask import Flask
 
 from configs import dify_config
+import pytz
 
 
 def init_app(app: Flask) -> Celery:
@@ -43,6 +44,11 @@ def __call__(self, *args: object, **kwargs: object) -> object:
         result_backend=dify_config.CELERY_RESULT_BACKEND,
         broker_transport_options=broker_transport_options,
         broker_connection_retry_on_startup=True,
+        worker_log_format=dify_config.LOG_FORMAT,
+        worker_task_log_format=dify_config.LOG_FORMAT,
+        worker_logfile=dify_config.LOG_FILE,
+        worker_hijack_root_logger=False,
+        timezone=pytz.timezone(dify_config.LOG_TZ),
     )
 
     if dify_config.BROKER_USE_SSL: