REFACTOR external service and improve websocket training

services/training/app/api/models.py

@@ -10,14 +10,12 @@ from sqlalchemy import text
 
 from app.core.database import get_db
 from app.schemas.training import TrainedModelResponse, ModelMetricsResponse
-from app.services.training_service import TrainingService
+from app.services.training_service import EnhancedTrainingService
 from datetime import datetime
 from sqlalchemy import select, delete, func
 import uuid
 import shutil
 
-from app.services.messaging import publish_models_deleted_event
-
 from shared.auth.decorators import (
     get_current_user_dep,
     require_admin_role
@@ -38,7 +36,7 @@ route_builder = RouteBuilder('training')
 logger = structlog.get_logger()
 router = APIRouter()
 
-training_service = TrainingService()
+training_service = EnhancedTrainingService()
 
 @router.get(
     route_builder.build_base_route("models") + "/{inventory_product_id}/active"
@@ -472,12 +470,7 @@ async def delete_tenant_models_complete(
             deletion_stats["errors"].append(error_msg)
             logger.warning(error_msg)
         
-        # Step 5: Publish deletion event
-        try:
-            await publish_models_deleted_event(tenant_id, deletion_stats)
-        except Exception as e:
-            logger.warning("Failed to publish models deletion event", error=str(e))
-        
+        # Models deleted successfully
         return {
             "success": True,
             "message": f"All training data for tenant {tenant_id} deleted successfully",