From 9d4b2715e8119963432787c4edb6ab0e19d38e12 Mon Sep 17 00:00:00 2001 From: GareArc Date: Sun, 1 Mar 2026 22:27:44 -0800 Subject: [PATCH] fix(celery): register enterprise_telemetry_task in worker imports Fixes Celery worker error where process_enterprise_telemetry task was unregistered despite being dispatched from the app. Added conditional import when ENTERPRISE_TELEMETRY_ENABLED=true to ensure the task is available in the worker process. Resolves: KeyError 'tasks.enterprise_telemetry_task.process_enterprise_telemetry' --- api/extensions/ext_celery.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/api/extensions/ext_celery.py b/api/extensions/ext_celery.py index af983f6d87..9944e768b9 100644 --- a/api/extensions/ext_celery.py +++ b/api/extensions/ext_celery.py @@ -184,6 +184,8 @@ def init_app(app: DifyApp) -> Celery: "task": "schedule.trigger_provider_refresh_task.trigger_provider_refresh", "schedule": timedelta(minutes=dify_config.TRIGGER_PROVIDER_REFRESH_INTERVAL), } + if dify_config.ENTERPRISE_TELEMETRY_ENABLED: + imports.append("tasks.enterprise_telemetry_task") celery_app.conf.update(beat_schedule=beat_schedule, imports=imports) return celery_app