diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/dataobject/modelassesstaskauto/ModelAssessTaskAutoBackupDO.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/dataobject/modelassesstaskauto/ModelAssessTaskAutoBackupDO.java index 9f039155e..d0b5f9a48 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/dataobject/modelassesstaskauto/ModelAssessTaskAutoBackupDO.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/dataobject/modelassesstaskauto/ModelAssessTaskAutoBackupDO.java @@ -26,11 +26,11 @@ public class ModelAssessTaskAutoBackupDO extends BaseDO { * 评估备份ID */ @TableId - private Long backupId; + private Long id; /** * 评估任务ID */ - private Long id; + private Long taskId; /** * 任务名称 */ @@ -47,20 +47,20 @@ public class ModelAssessTaskAutoBackupDO extends BaseDO { * 模型服务 */ private Long modelService; + /** + * 是否预知模型(0普通 1官方) + */ + private Integer modelType; /** * 数据集 */ private Long dataset; - /** - * 数据集类型 0 普通 1 官方 - */ - private Long modelType; /** * 评估方向,使用字典(llm_dimension_type) */ - private Integer dimension; + private String dimension; /** - * 模型评估任务状态,使用字典(llm_model_assess_task_status) + * 状态 */ private Integer status; diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/mysql/modelassesstaskauto/ModelAssessTaskAutoBackupMapper.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/mysql/modelassesstaskauto/ModelAssessTaskAutoBackupMapper.java index 8f83b2cee..ac98307e0 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/mysql/modelassesstaskauto/ModelAssessTaskAutoBackupMapper.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/mysql/modelassesstaskauto/ModelAssessTaskAutoBackupMapper.java @@ -29,7 +29,7 @@ public interface ModelAssessTaskAutoBackupMapper extends BaseMapperX modelAssessTaskStoplistDOS = modelingTaskStoplistMapper.selectList(new LambdaQueryWrapper().eq(ModelAssessTaskStoplistDO::getTaskId, id)); @@ -269,19 +275,21 @@ public class ModelAssessTaskAutoServiceImpl implements ModelAssessTaskAutoServic List listByTaskId = modelAssessTaskAutoInfoService.getListByTaskId(modelAssessTaskAutoInfoPageReqVO); for (ModelAssessTaskAutoInfoDO infoDO : listByTaskId) { ModelAssessTaskAutoInfoSaveReqVO modelAssessTaskAutoInfoSaveReqVO = BeanUtils.toBean(infoDO, ModelAssessTaskAutoInfoSaveReqVO.class); - modelAssessTaskAutoInfoSaveReqVO.setTaskId(bean.getId()); + modelAssessTaskAutoInfoSaveReqVO.setTaskId(id); modelAssessTaskAutoInfoBackupService.createModelAssessTaskAutoInfo(modelAssessTaskAutoInfoSaveReqVO); + } } @Override public void cancelTheBackUp (Long id) { - modelAssessTaskAutoBackupMapper.delete(new LambdaQueryWrapper().eq(ModelAssessTaskAutoBackupDO::getId, id)); + modelAssessTaskAutoBackupMapper.delete(new LambdaQueryWrapper().eq(ModelAssessTaskAutoBackupDO::getTaskId, id)); ModelAssessTaskAutoDO modelAssessTaskAutoDO = modelAssessTaskAutoMapper.selectById(id); modelAssessTaskAutoDO.setBackupId(0L); modelAssessTaskAutoMapper.updateById(modelAssessTaskAutoDO); - modelAssessTaskStoplistBackupMapper.delete(new LambdaQueryWrapper().eq(ModelAssessTaskStoplistBackupDO::getBackupId, id)); + modelAssessTaskStoplistBackupMapper.delete(new LambdaQueryWrapper().eq(ModelAssessTaskStoplistBackupDO::getTaskId, id)); + modelAssessTaskAutoInfoBackupMapper.delete(new LambdaQueryWrapper().eq(ModelAssessTaskAutoInfoBackupDO::getTaskId, id)); } /**