Add all the code for training service

services/training/app/api/models.py

@@ -8,10 +8,11 @@ from typing import List
 import structlog
 
 from app.core.database import get_db
-from app.core.auth import verify_token
+from app.core.auth import get_current_tenant_id
 from app.schemas.training import TrainedModelResponse
 from app.services.training_service import TrainingService
 
+
 logger = structlog.get_logger()
 router = APIRouter()
 
@@ -19,12 +20,12 @@ training_service = TrainingService()
 
 @router.get("/", response_model=List[TrainedModelResponse])
 async def get_trained_models(
-    user_data: dict = Depends(verify_token),
+    tenant_id: str = Depends(get_current_tenant_id),
     db: AsyncSession = Depends(get_db)
 ):
     """Get trained models"""
     try:
-        return await training_service.get_trained_models(user_data, db)
+        return await training_service.get_trained_models(tenant_id, db)
     except Exception as e:
         logger.error(f"Get trained models error: {e}")
         raise HTTPException(