REFACTOR - Database logic

services/training/app/main.py

@@ -16,8 +16,9 @@ from fastapi.responses import JSONResponse
 import uvicorn
 
 from app.core.config import settings
-from app.core.database import initialize_training_database, cleanup_training_database
+from app.core.database import initialize_training_database, cleanup_training_database, get_db_health
 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
@@ -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"])