Fine-tuning/backend/app/api/models.py

from fastapi import APIRouter

from app.schemas.model import ModelDownloadRequest, ModelDownloadResponse, ModelInfo

router = APIRouter()


@router.get("/", response_model=list[ModelInfo])
async def list_models():
    """列出所有本地缓存的模型。"""
    return []


@router.post("/download", response_model=ModelDownloadResponse)
async def download_model(req: ModelDownloadRequest):
    """从 HuggingFace 或 ModelScope 下载模型。"""
    return ModelDownloadResponse(
        model_id=req.model_id, status="downloading", path=None, error=None
    )


@router.get("/{model_id}", response_model=ModelInfo)
async def get_model_info(model_id: str):
    """获取已缓存模型的详细信息。"""
    return ModelInfo(
        id=model_id,
        name=model_id.split("/")[-1],
        model_type="text",
        is_downloaded=False,
    )