REFACTOR external service and improve websocket training

services/training/app/api/__init__.py

@@ -1,14 +1,16 @@
 """
 Training API Layer
-HTTP endpoints for ML training operations
+HTTP endpoints for ML training operations and WebSocket connections
 """
 
 from .training_jobs import router as training_jobs_router
 from .training_operations import router as training_operations_router
 from .models import router as models_router
+from .websocket_operations import router as websocket_operations_router
 
 __all__ = [
     "training_jobs_router",
     "training_operations_router",
-    "models_router"
+    "models_router",
+    "websocket_operations_router"
 ]