Fix training start 2

services/training/app/api/training.py

@@ -29,7 +29,7 @@ from app.services.messaging import (
 )
 
 from sqlalchemy.ext.asyncio import AsyncSession
-from app.core.database import get_db
+from app.core.database import get_db_session
 
 # Import unified authentication from shared library
 from shared.auth.decorators import (
@@ -48,7 +48,7 @@ async def start_training_job(
     tenant_id: str = Depends(get_current_tenant_id_dep),
     current_user: Dict[str, Any] = Depends(get_current_user_dep),
     training_service: TrainingService = Depends(),
-    db: AsyncSession = Depends(get_db) # Ensure db is available
+    db: AsyncSession = Depends(get_db_session) # Ensure db is available
 ):
     """Start a new training job for all products"""
     try: