diff --git a/api/extensions/ext_celery.py b/api/extensions/ext_celery.py index 5cf4984709..b1a970230a 100644 --- a/api/extensions/ext_celery.py +++ b/api/extensions/ext_celery.py @@ -47,7 +47,11 @@ def _get_celery_ssl_options() -> dict[str, Any] | None: def init_app(app: DifyApp) -> Celery: class FlaskTask(Task): def __call__(self, *args: object, **kwargs: object) -> object: + from core.logging.context import init_request_context + with app.app_context(): + # Initialize logging context for this task (similar to before_request in Flask) + init_request_context() return self.run(*args, **kwargs) broker_transport_options = {}