Add migration services

services/training/app/main.py

@@ -8,6 +8,7 @@ ML training service for bakery demand forecasting
 
 import asyncio
 from fastapi import FastAPI, Request
+from sqlalchemy import text
 from app.core.config import settings
 from app.core.database import initialize_training_database, cleanup_training_database, database_manager
 from app.api import training, models
@@ -19,6 +20,27 @@ from shared.service_base import StandardFastAPIService
 class TrainingService(StandardFastAPIService):
     """Training Service with standardized setup"""
 
+    expected_migration_version = "001_initial_training"
+
+    async def on_startup(self, app):
+        """Custom startup logic including migration verification"""
+        await self.verify_migrations()
+        await super().on_startup(app)
+
+    async def verify_migrations(self):
+        """Verify database schema matches the latest migrations."""
+        try:
+            async with self.database_manager.get_session() as session:
+                result = await session.execute(text("SELECT version_num FROM alembic_version"))
+                version = result.scalar()
+                if version != self.expected_migration_version:
+                    self.logger.error(f"Migration version mismatch: expected {self.expected_migration_version}, got {version}")
+                    raise RuntimeError(f"Migration version mismatch: expected {self.expected_migration_version}, got {version}")
+                self.logger.info(f"Migration verification successful: {version}")
+        except Exception as e:
+            self.logger.error(f"Migration verification failed: {e}")
+            raise
+
     def __init__(self):
         # Define expected database tables for health checks
         training_expected_tables = [