33 lines
986 B
Python
33 lines
986 B
Python
|
"""
|
||
|
|
Models API endpoints
|
||
|
|
"""
|
||
|
|
|
||
|
|
from fastapi import APIRouter, Depends, HTTPException, status
|
||
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
||
|
|
from typing import List
|
||
|
|
import logging
|
||
|
|
|
||
|
|
from app.core.database import get_db
|
||
|
|
from app.core.auth import verify_token
|
||
|
|
from app.schemas.training import TrainedModelResponse
|
||
|
|
from app.services.training_service import TrainingService
|
||
|
|
|
||
|
|
logger = logging.getLogger(__name__)
|
||
|
|
router = APIRouter()
|
||
|
|
|
||
|
|
training_service = TrainingService()
|
||
|
|
|
||
|
|
@router.get("/", response_model=List[TrainedModelResponse])
|
||
|
|
async def get_trained_models(
|
||
|
|
user_data: dict = Depends(verify_token),
|
||
|
|
db: AsyncSession = Depends(get_db)
|
||
|
|
):
|
||
|
|
"""Get trained models"""
|
||
|
|
try:
|
||
|
|
return await training_service.get_trained_models(user_data, db)
|
||
|
|
except Exception as e:
|
||
|
|
logger.error(f"Get trained models error: {e}")
|
||
|
|
raise HTTPException(
|
||
|
|
status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
|
||
|
|
detail="Failed to get trained models"
|
||
|
|
)
|