From 1e9665c11d7b6257852b0ebafbc389a1986ada35 Mon Sep 17 00:00:00 2001 From: Ramtin Yazdanian Date: Tue, 4 Nov 2025 14:56:11 +0100 Subject: [PATCH 1/2] add celery config to avoid accidentally missing a heartbeat --- graphai/celery/common/celery_config.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/graphai/celery/common/celery_config.py b/graphai/celery/common/celery_config.py index e23e2ea6..cbb0ba47 100644 --- a/graphai/celery/common/celery_config.py +++ b/graphai/celery/common/celery_config.py @@ -113,6 +113,21 @@ def create_celery(): # Every 24 hours } }) + # Configuring the broker to avoid accidentally-missed heartbeats + celery_app.conf.update(task_acks_late=True) + celery_app.conf.update(worker_send_task_events=True) + celery_app.conf.update(send_events=True) + celery_app.conf.update(send_sent_event=True) + celery_app.conf.update(task_track_started=True) + celery_app.conf.update(redis_socket_keepalive=True) + celery_app.conf.update(broker_pool_limit=None) + celery_app.conf.update(broker_connection_timeout=300) + celery_app.conf.update(broker_max_retries=None) + celery_app.conf.update(worker_lost_wait=300) + celery_app.conf.update(worker_cancel_long_running_tasks_on_connection_loss=True) + celery_app.conf.update(broker_connection_retry_on_startup=True) + celery_app.conf.update(broker_channel_error_retry=True) + return celery_app From 86f05d1707a76633d46f59b5ba095d755764bf02 Mon Sep 17 00:00:00 2001 From: Ramtin Yazdanian Date: Wed, 5 Nov 2025 12:55:48 +0100 Subject: [PATCH 2/2] linter fix --- graphai/celery/common/celery_config.py | 1 - 1 file changed, 1 deletion(-) diff --git a/graphai/celery/common/celery_config.py b/graphai/celery/common/celery_config.py index cbb0ba47..d2890b9b 100644 --- a/graphai/celery/common/celery_config.py +++ b/graphai/celery/common/celery_config.py @@ -128,7 +128,6 @@ def create_celery(): celery_app.conf.update(broker_connection_retry_on_startup=True) celery_app.conf.update(broker_channel_error_retry=True) - return celery_app