from fastapi import APIRouter, HTTPException

from engine.loader import get_inference_lock, reload_engine
from schema import ReloadRequest, ReloadResponse

router = APIRouter()


@router.post("/model/reload", response_model=ReloadResponse)
async def model_reload(req: ReloadRequest):
    async with get_inference_lock():
        try:
            engine = reload_engine(req.model_path, req.runtime)
        except Exception as e:
            raise HTTPException(status_code=500, detail=f"模型重载失败: {e}")
    return ReloadResponse(status="ok", model_version=engine.model_version)