From b990888316cc62e806380a702499f02faee72e70 Mon Sep 17 00:00:00 2001 From: Arnav Chopra Date: Fri, 3 Apr 2026 19:16:21 -0400 Subject: [PATCH 1/2] fix(gcp): use redis backend for celery autoscaler when broker is redis --- .../model_engine_server/core/celery/celery_autoscaler.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/model-engine/model_engine_server/core/celery/celery_autoscaler.py b/model-engine/model_engine_server/core/celery/celery_autoscaler.py index 66a9430d..fe2c39bf 100644 --- a/model-engine/model_engine_server/core/celery/celery_autoscaler.py +++ b/model-engine/model_engine_server/core/celery/celery_autoscaler.py @@ -604,7 +604,8 @@ async def main(): inspect = { db_index: inspect_app( app=celery_app( - None, broker_type=broker_type, task_visibility=db_index, aws_role=aws_profile + None, broker_type=broker_type, task_visibility=db_index, + aws_role=aws_profile, backend_protocol="redis", ) ) for db_index in get_all_db_indexes() From 4ab03ae5df19c9aed442b31741819b54c1c949e5 Mon Sep 17 00:00:00 2001 From: Arnav Chopra Date: Fri, 3 Apr 2026 19:21:53 -0400 Subject: [PATCH 2/2] fix --- .../model_engine_server/core/celery/celery_autoscaler.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/model-engine/model_engine_server/core/celery/celery_autoscaler.py b/model-engine/model_engine_server/core/celery/celery_autoscaler.py index fe2c39bf..73aada54 100644 --- a/model-engine/model_engine_server/core/celery/celery_autoscaler.py +++ b/model-engine/model_engine_server/core/celery/celery_autoscaler.py @@ -604,8 +604,11 @@ async def main(): inspect = { db_index: inspect_app( app=celery_app( - None, broker_type=broker_type, task_visibility=db_index, - aws_role=aws_profile, backend_protocol="redis", + None, + broker_type=broker_type, + task_visibility=db_index, + aws_role=aws_profile, + backend_protocol="redis", ) ) for db_index in get_all_db_indexes()