package cn.iocoder.yudao.module.llm.service.finetuningtask;

import cn.iocoder.yudao.framework.common.pojo.PageResult;
import cn.iocoder.yudao.module.llm.controller.admin.finetuningtask.vo.FineTuningTaskPageReqVO;
import cn.iocoder.yudao.module.llm.controller.admin.finetuningtask.vo.FineTuningTaskRespVO;
import cn.iocoder.yudao.module.llm.controller.admin.finetuningtask.vo.FineTuningTaskSaveReqVO;
import cn.iocoder.yudao.module.llm.dal.dataobject.finetuningtask.FineTuningTaskDO;

import javax.validation.Valid;
import java.util.List;
import java.util.Map;

/**
 * 模型调优 —— 微调任务 Service 接口
 *
 * @author 华大大模型
 */
public interface FineTuningTaskService {

    /**
     * 创建模型调优 —— 微调任务
     *
     * @param createReqVO 创建信息
     * @return 编号
     */
    Long createFineTuningTask (@Valid FineTuningTaskSaveReqVO createReqVO);

    /**
     * 更新模型调优 —— 微调任务
     *
     * @param updateReqVO 更新信息
     */
    void updateFineTuningTask (@Valid FineTuningTaskSaveReqVO updateReqVO);

    /**
     * 删除模型调优 —— 微调任务
     *
     * @param id 编号
     */
    void deleteFineTuningTask (Long id);

    /**
     * 获得模型调优 —— 微调任务
     *
     * @param id 编号
     * @return 模型调优 —— 微调任务
     */
    FineTuningTaskRespVO getFineTuningTask (Long id);

    /**
     * 获得模型调优 —— 微调任务分页
     *
     * @param pageReqVO 分页查询
     * @return 模型调优 —— 微调任务分页
     */
    PageResult<FineTuningTaskDO> getFineTuningTaskPage (FineTuningTaskPageReqVO pageReqVO);

    /**
     * 获得模型调优 —— 微调任务分页
     *
     * @param pageReqVO 分页查询
     * @return 模型调优 —— 微调任务分页
     */
    PageResult<FineTuningTaskRespVO> getFineTuningTaskPage1 (FineTuningTaskPageReqVO pageReqVO);

    /**
     * 停止微调任务
     *
     * @param id
     */
    void stopFineTuningTask (Long id);

    /**
     * 部署微调任务
     *
     * @param id
     */
    void startFineTuningTask (Long id);

    List<FineTuningTaskRespVO> selectAll ();

    /**
     * 获得模型调优 —— 获取已部署完成的调优任务
     *
     * @return 已部署完成的调优任务
     */
    List<FineTuningTaskRespVO> selectEnable ();

    /**
     * 根据 基础模型id 获取 模型调优
     *
     * @param baseModelId 基础模型id
     * @return 模型调优
     */
    Map<Long, String> getModelTuningByModelId (Long baseModelId);

    /**
     * 根据 数据集id 获取 模型调优
     *
     * @param datasetId 数据集id
     * @return 模型调优
     */
    Map<Long, String> getModelTuningByDatasetId (Long datasetId);

    String getFineTuningLog(Long id);
}
