Improve the event messaging for training service 2

services/training/app/main.py

@@ -18,6 +18,7 @@ import uvicorn
 from app.core.config import settings
 from app.core.database import initialize_training_database, cleanup_training_database
 from app.api import training, models
+from app.api.websocket import websocket_router
 from app.services.messaging import setup_messaging, cleanup_messaging
 from shared.monitoring.logging import setup_logging
 from shared.monitoring.metrics import MetricsCollector
@@ -176,6 +177,7 @@ async def global_exception_handler(request: Request, exc: Exception):
 # Include API routers
 app.include_router(training.router, prefix="/api/v1", tags=["training"])
 app.include_router(models.router, prefix="/api/v1", tags=["models"])
+app.include_router(websocket_router, prefix="/api/v1/ws", tags=["websocket"])
 
 
 # Health check endpoints