label_ai_service/app/routers/finetune.py

from fastapi import APIRouter, Depends

from app.clients.llm.base import LLMClient
from app.core.dependencies import get_llm_client
from app.models.finetune_models import (
    FinetuneStartRequest,
    FinetuneStartResponse,
    FinetuneStatusResponse,
)
from app.services import finetune_service

router = APIRouter(tags=["Finetune"])


@router.post("/finetune/start", response_model=FinetuneStartResponse)
async def start_finetune(
    req: FinetuneStartRequest,
    llm: LLMClient = Depends(get_llm_client),
) -> FinetuneStartResponse:
    return await finetune_service.submit_finetune(req, llm)


@router.get("/finetune/status/{job_id}", response_model=FinetuneStatusResponse)
async def get_status(
    job_id: str,
    llm: LLMClient = Depends(get_llm_client),
) -> FinetuneStatusResponse:
    return await finetune_service.get_finetune_status(job_id, llm)