Fix imports

services/training/app/main.py

@@ -3,20 +3,20 @@ Training Service
 Handles ML model training for bakery demand forecasting
 """
 
-import logging
+import structlog
 from fastapi import FastAPI, BackgroundTasks
 from fastapi.middleware.cors import CORSMiddleware
 
 from app.core.config import settings
 from app.core.database import database_manager
 from app.api import training, models
-from app.services.messaging import message_publisher
+from app.services.messaging import setup_messaging, cleanup_messaging
 from shared.monitoring.logging import setup_logging
 from shared.monitoring.metrics import MetricsCollector
 
 # Setup logging
 setup_logging("training-service", settings.LOG_LEVEL)
-logger = logging.getLogger(__name__)
+logger = structlog.get_logger()
 
 # Create FastAPI app
 app = FastAPI(
@@ -50,7 +50,7 @@ async def startup_event():
     await database_manager.create_tables()
     
     # Initialize message publisher
-    await message_publisher.connect()
+    await setup_messaging()
     
     # Start metrics server
     metrics_collector.start_metrics_server(8080)
@@ -63,7 +63,7 @@ async def shutdown_event():
     logger.info("Shutting down Training Service")
     
     # Cleanup message publisher
-    await message_publisher.disconnect()
+    await cleanup_messaging()
     
     logger.info("Training Service shutdown complete")