From 4f459f31f63f92fbc85a242d26183dbb74d9b886 Mon Sep 17 00:00:00 2001 From: Urtzi Alfaro Date: Fri, 25 Jul 2025 19:03:20 +0200 Subject: [PATCH] Fix training start 2 --- services/training/app/api/training.py | 4 ++-- services/training/app/main.py | 6 +++--- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/services/training/app/api/training.py b/services/training/app/api/training.py index 6a92779d..c4e3040d 100644 --- a/services/training/app/api/training.py +++ b/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: diff --git a/services/training/app/main.py b/services/training/app/main.py index 04812f3f..827469ab 100644 --- a/services/training/app/main.py +++ b/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: