DATA_MIDDLE_PLATFORM/Data_Middle_Platform_Server
3
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

修改自动评估备份按钮

Browse Source
This commit is contained in:
sunxiqing 2025-01-24 16:51:55 +08:00
parent 392d4ce473
commit 2e255c059f
3 changed files with 21 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/dataobject/modelassesstaskauto/ModelAssessTaskAutoBackupDO.java
View File

@ -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;

2
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/mysql/modelassesstaskauto/ModelAssessTaskAutoBackupMapper.java
View File

@ -29,7 +29,7 @@ public interface ModelAssessTaskAutoBackupMapper extends BaseMapperX<ModelAssess
.eqIfPresent(ModelAssessTaskAutoBackupDO::getDataset, reqVO.getDataset())
.betweenIfPresent(ModelAssessTaskAutoBackupDO::getCreateTime, reqVO.getCreateTime())
.eqIfPresent(ModelAssessTaskAutoBackupDO::getDimension, reqVO.getDimension())
.orderByDesc(ModelAssessTaskAutoBackupDO::getBackupId));
.orderByDesc(ModelAssessTaskAutoBackupDO::getId));
}

16
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelassesstaskauto/ModelAssessTaskAutoServiceImpl.java
View File

@ -11,6 +11,7 @@ import cn.iocoder.yudao.module.llm.dal.dataobject.modelassesstaskauto.ModelAsses
import cn.iocoder.yudao.module.llm.dal.dataobject.modelassesstaskauto.ModelAssessTaskAutoDO;
import cn.iocoder.yudao.module.llm.dal.dataobject.modelassesstaskauto.ModelAssessTaskStoplistBackupDO;
import cn.iocoder.yudao.module.llm.dal.dataobject.modelassesstaskauto.ModelAssessTaskStoplistDO;
import cn.iocoder.yudao.module.llm.dal.dataobject.modelassesstaskautoinfo.ModelAssessTaskAutoInfoBackupDO;
import cn.iocoder.yudao.module.llm.dal.dataobject.modelassesstaskautoinfo.ModelAssessTaskAutoInfoDO;
import cn.iocoder.yudao.module.llm.dal.dataobject.modelservice.ModelServiceDO;
import cn.iocoder.yudao.module.llm.dal.mysql.basemodel.BaseModelMapper;
@ -20,6 +21,7 @@ import cn.iocoder.yudao.module.llm.dal.mysql.modelassesstaskauto.ModelAssessTask
import cn.iocoder.yudao.module.llm.dal.mysql.modelassesstaskauto.ModelAssessTaskAutoMapper;
import cn.iocoder.yudao.module.llm.dal.mysql.modelassesstaskauto.ModelAssessTaskStoplistBackupMapper;
import cn.iocoder.yudao.module.llm.dal.mysql.modelassesstaskauto.ModelAssessTaskStoplistMapper;
import cn.iocoder.yudao.module.llm.dal.mysql.modelassesstaskautoinfo.ModelAssessTaskAutoInfoBackupMapper;
import cn.iocoder.yudao.module.llm.dal.mysql.modelservice.ModelServiceMapper;
import cn.iocoder.yudao.module.llm.service.async.AsyncModelAccessManualService;
import cn.iocoder.yudao.module.llm.service.modelassesstaskautoinfo.ModelAssessTaskAutoInfoBackupService;
@ -74,6 +76,8 @@ public class ModelAssessTaskAutoServiceImpl implements ModelAssessTaskAutoServic
private ModelAssessTaskAutoInfoBackupService modelAssessTaskAutoInfoBackupService;
@Resource
private ModelAssessStoplistMapper modelAssessStoplistMapper;
@Resource
private ModelAssessTaskAutoInfoBackupMapper modelAssessTaskAutoInfoBackupMapper;
@Override
@ -253,8 +257,10 @@ public class ModelAssessTaskAutoServiceImpl implements ModelAssessTaskAutoServic
throw exception(MODEL_ASSESS_TASK_AUTO_NOT_EXISTS);
}
ModelAssessTaskAutoBackupDO bean = BeanUtils.toBean(modelAssessTaskAutoDO, ModelAssessTaskAutoBackupDO.class);
bean.setId(null);
bean.setTaskId(id);
modelAssessTaskAutoBackupMapper.insert(bean);
modelAssessTaskAutoDO.setBackupId(bean.getBackupId());
modelAssessTaskAutoDO.setBackupId(bean.getId());
modelAssessTaskAutoMapper.updateById(modelAssessTaskAutoDO);
List<ModelAssessTaskStoplistDO> modelAssessTaskStoplistDOS = modelingTaskStoplistMapper.selectList(new LambdaQueryWrapper<ModelAssessTaskStoplistDO>().eq(ModelAssessTaskStoplistDO::getTaskId, id));
@ -269,19 +275,21 @@ public class ModelAssessTaskAutoServiceImpl implements ModelAssessTaskAutoServic
List<ModelAssessTaskAutoInfoDO> 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<ModelAssessTaskAutoBackupDO>().eq(ModelAssessTaskAutoBackupDO::getId, id));
modelAssessTaskAutoBackupMapper.delete(new LambdaQueryWrapper<ModelAssessTaskAutoBackupDO>().eq(ModelAssessTaskAutoBackupDO::getTaskId, id));
ModelAssessTaskAutoDO modelAssessTaskAutoDO = modelAssessTaskAutoMapper.selectById(id);
modelAssessTaskAutoDO.setBackupId(0L);
modelAssessTaskAutoMapper.updateById(modelAssessTaskAutoDO);
modelAssessTaskStoplistBackupMapper.delete(new LambdaQueryWrapper<ModelAssessTaskStoplistBackupDO>().eq(ModelAssessTaskStoplistBackupDO::getBackupId, id));
modelAssessTaskStoplistBackupMapper.delete(new LambdaQueryWrapper<ModelAssessTaskStoplistBackupDO>().eq(ModelAssessTaskStoplistBackupDO::getTaskId, id));
modelAssessTaskAutoInfoBackupMapper.delete(new LambdaQueryWrapper<ModelAssessTaskAutoInfoBackupDO>().eq(ModelAssessTaskAutoInfoBackupDO::getTaskId, id));
}
/**