from fastapi import APIRouter

from app.schemas.deployment import DeployConfig, DeployResponse
from app.services import deploy_service

router = APIRouter()


@router.post("/export", response_model=DeployResponse)
async def export_adapter(config: DeployConfig):
    """合并 adapter 与基础模型，可选导出为 GGUF。"""
    result = await deploy_service.export_adapter(
        config.job_id,
        {"merge_with_base": config.merge_with_base, "export_format": config.export_format},
    )
    return DeployResponse(**result)


@router.get("/{deploy_id}/status", response_model=DeployResponse)
async def get_deployment_status(deploy_id: str):
    """获取导出/部署任务状态。"""
    result = await deploy_service.get_deploy_status(deploy_id)
    return DeployResponse(**result)