feat: add logs

keephq · Sep 8, 2024 · 6dfbbd9 · 6dfbbd9
1 parent 8fce52c
commit 6dfbbd9
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/keep/api/api.py b/keep/api/api.py
@@ -271,14 +271,17 @@ async def on_startup():
         if KEEP_ARQ_TASK_POOL != KEEP_ARQ_TASK_POOL_NONE:
             event_loop = asyncio.get_event_loop()
             if KEEP_ARQ_TASK_POOL == KEEP_ARQ_TASK_POOL_ALL:
+                logger.info("Starting all task pools")
                 basic_worker = get_arq_worker(KEEP_ARQ_QUEUE_BASIC)
                 ai_worker = get_arq_worker(KEEP_ARQ_QUEUE_AI)
                 event_loop.create_task(basic_worker.async_run())
                 event_loop.create_task(ai_worker.async_run())
             elif KEEP_ARQ_TASK_POOL == KEEP_ARQ_TASK_POOL_AI:
+                logger.info("Starting AI task pool")
                 arq_worker = get_arq_worker(KEEP_ARQ_QUEUE_AI)
                 event_loop.create_task(arq_worker.async_run())
             elif KEEP_ARQ_TASK_POOL == KEEP_ARQ_TASK_POOL_BASIC_PROCESSING:
+                logger.info("Starting Basic Processing task pool")
                 arq_worker = get_arq_worker(KEEP_ARQ_QUEUE_BASIC)
                 event_loop.create_task(arq_worker.async_run())
             else: