Skip to content

Commit 06563a1

Browse files
authored
Add log statement for gateway sending async task (#459)
* log * fix
1 parent 34552b8 commit 06563a1

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

model-engine/model_engine_server/infra/gateways/celery_task_queue_gateway.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,8 +8,10 @@
88
)
99
from model_engine_server.core.celery import TaskVisibility, celery_app
1010
from model_engine_server.core.config import infra_config
11+
from model_engine_server.core.loggers import logger_name, make_logger
1112
from model_engine_server.domain.gateways.task_queue_gateway import TaskQueueGateway
1213

14+
logger = make_logger(logger_name())
1315
backend_protocol = "abs" if infra_config().cloud_provider == "azure" else "s3"
1416

1517
celery_redis = celery_app(
@@ -72,6 +74,7 @@ def send_task(
7274
kwargs=kwargs,
7375
queue=queue_name,
7476
)
77+
logger.info(f"Task {res.id} sent to queue {queue_name} from gateway") # pragma: no cover
7578
return CreateAsyncTaskV1Response(task_id=res.id)
7679

7780
def get_task(self, task_id: str) -> GetAsyncTaskV1Response:

0 commit comments

Comments
 (0)