Fix training start 2

services/training/app/main.py

@@ -16,7 +16,7 @@ from fastapi.responses import JSONResponse
 import uvicorn
 
 from app.core.config import settings
-from app.core.database import database_manager, get_db_health
+from app.core.database import initialize_training_database, cleanup_training_database
 from app.api import training, models
 from app.services.messaging import setup_messaging, cleanup_messaging
 from shared.monitoring.logging import setup_logging
@@ -41,7 +41,7 @@ async def lifespan(app: FastAPI):
     try:
         # Initialize database
         logger.info("Initializing database connection")
-        await database_manager.create_tables()
+        initialize_training_database()
         logger.info("Database initialized successfully")
         
         # Initialize messaging
@@ -81,7 +81,7 @@ async def lifespan(app: FastAPI):
         logger.info("Messaging cleanup completed")
         
         # Close database connections
-        await database_manager.close_connections()
+        cleanup_training_database()
         logger.info("Database connections closed")
         
     except Exception as e: