Merge pull request #21 from linto-ai/refactoring/all

use env var QUEUE_NAME and rollback queue name for healthcheck
linto-ai · Feb 7, 2024 · 8dfc7d4 · 8dfc7d4
2 parents 487a544 + e51eb6d
commit 8dfc7d4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/celery_app/celeryapp.py b/celery_app/celeryapp.py
@@ -22,7 +22,7 @@
 celery.conf.update(
     {
         "task_routes": {
-            "diarization_task": {"queue": service_name},
+            "diarization_task": {"queue": "diarization"},
         }
     }
 )

diff --git a/celery_app/register.py b/celery_app/register.py
@@ -70,7 +70,7 @@ def unregister() -> None:
 
 
 def queue() -> str:
-    return service_name
+    return os.environ.get("QUEUE_NAME", service_name)
 
 
 def service_info() -> dict: