shaliz-kong commited on
Commit Β·
6be95d4
1
Parent(s): b39a40c
changed worker manage import to get worker manager
Browse files- app/main.py +6 -1
app/main.py
CHANGED
|
@@ -21,7 +21,10 @@ from contextlib import asynccontextmanager
|
|
| 21 |
|
| 22 |
# βββ Internal Imports βββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
| 23 |
from app.core.event_hub import event_hub
|
| 24 |
-
|
|
|
|
|
|
|
|
|
|
| 25 |
from app.deps import rate_limit_org, verify_api_key, check_all_services
|
| 26 |
from app.tasks.analytics_worker import trigger_kpi_computation
|
| 27 |
from app.service.vector_service import cleanup_expired_vectors
|
|
@@ -106,6 +109,8 @@ async def lifespan(app: FastAPI):
|
|
| 106 |
if os.getenv("DISABLE_WORKER_MANAGER") != "1":
|
| 107 |
logger.info("π starting worker manager...")
|
| 108 |
try:
|
|
|
|
|
|
|
| 109 |
asyncio.create_task(worker_manager.start_listener(), name="worker-manager")
|
| 110 |
except Exception as e:
|
| 111 |
logger.error(f"β Failed to start worker manager: {e}")
|
|
|
|
| 21 |
|
| 22 |
# βββ Internal Imports βββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
| 23 |
from app.core.event_hub import event_hub
|
| 24 |
+
# NOTE: worker_manager is now created via async factory `get_worker_manager()`
|
| 25 |
+
# Old import kept as comment for reference:
|
| 26 |
+
# from app.core.worker_manager import worker_manager
|
| 27 |
+
from app.core.worker_manager import get_worker_manager,worker_manager
|
| 28 |
from app.deps import rate_limit_org, verify_api_key, check_all_services
|
| 29 |
from app.tasks.analytics_worker import trigger_kpi_computation
|
| 30 |
from app.service.vector_service import cleanup_expired_vectors
|
|
|
|
| 109 |
if os.getenv("DISABLE_WORKER_MANAGER") != "1":
|
| 110 |
logger.info("π starting worker manager...")
|
| 111 |
try:
|
| 112 |
+
# Use the async factory to get the singleton manager instance
|
| 113 |
+
worker_manager = await get_worker_manager()
|
| 114 |
asyncio.create_task(worker_manager.start_listener(), name="worker-manager")
|
| 115 |
except Exception as e:
|
| 116 |
logger.error(f"β Failed to start worker manager: {e}")
|