From ed583f7e4b2c9defeb0d1d3a92114880714a30a2 Mon Sep 17 00:00:00 2001
From: Byron Wang <byron@dify.ai>
Date: Thu, 25 Dec 2025 17:18:37 +0800
Subject: [PATCH] add request context to celery

---
 api/extensions/ext_celery.py | 4 ++++
 1 file changed, 4 insertions(+)

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 = {}