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-03 19:06:41 +08:00
parent 4a19a8f743
commit 89792d05fc
15 changed files with 303 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/modelassesstaskauto/ModelAssessTaskAutoController.java
View File

@ -110,14 +110,18 @@ public class ModelAssessTaskAutoController {
@PostMapping("/backedUp")
@Operation(summary = "备份")
public void backedUp(ModelAssessTaskManualPageReqVO pageReqVO) {
public CommonResult<Boolean> backedUp(@RequestParam("id") Long id) {
//TODO 自动评估-备份待补充
modelAssessTaskAutoService.dataBackup(id);
return success(true);
}
@PostMapping("/cancelTheBackUp")
@Operation(summary = "取消备份")
public void cancelTheBackUp(ModelAssessTaskManualPageReqVO pageReqVO) {
public CommonResult<Boolean> cancelTheBackUp(@RequestParam("id") Long id) {
//TODO 自动评估-取消备份待补充
modelAssessTaskAutoService.cancelTheBackUp(id);
return success(true);
}
@PostMapping("/reEvaluation")

3
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/modelassesstaskauto/vo/ModelAssessTaskAutoRespVO.java
View File

@ -59,4 +59,7 @@ public class ModelAssessTaskAutoRespVO {
@Schema(description = "数据集名称")
private String datasetName;
@Schema(description = "备份ID")
private Long backupId;
}

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

@ -0,0 +1,67 @@
package cn.iocoder.yudao.module.llm.dal.dataobject.modelassesstaskauto;
import lombok.*;
import java.util.*;
import java.time.LocalDateTime;
import java.time.LocalDateTime;
import com.baomidou.mybatisplus.annotation.*;
import cn.iocoder.yudao.framework.mybatis.core.dataobject.BaseDO;
/**
* 模型评估任务 —— 自动评估备份 DO
*
* @author 华大大模型
*/
@TableName("llm_model_assess_task_auto_backup")
@KeySequence("llm_model_assess_task_auto_backup_seq") // 用于 Oracle、PostgreSQL、Kingbase、DB2、H2 数据库的主键自增。如果是 MySQL 等数据库，可不写。
@Data
@EqualsAndHashCode(callSuper = true)
@ToString(callSuper = true)
@Builder
@NoArgsConstructor
@AllArgsConstructor
public class ModelAssessTaskAutoBackupDO extends BaseDO {
/**
* 评估备份ID
*/
@TableId
private Long backupId;
/**
* 评估任务ID
*/
private Long id;
/**
* 任务名称
*/
private String taskName;
/**
* 任务描述
*/
private String taskInfro;
/**
* CPU类型，使用字典（llm_cpu_type）
*/
private Integer cpuType;
/**
* 模型服务
*/
private Long modelService;
/**
* 数据集
*/
private Long dataset;
/**
* 数据集类型 0 普通 1 官方
*/
private Long modelType;
/**
* 评估方向，使用字典（llm_dimension_type）
*/
private Integer dimension;
/**
* 模型评估任务状态，使用字典（llm_model_assess_task_status）
*/
private Integer status;
}

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

@ -28,6 +28,10 @@ public class ModelAssessTaskAutoDO extends BaseDO {
*/
@TableId
private Long id;
/**
* 备份ID
*/
private Long backupId;
/**
* 任务名称
*/

43
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/dataobject/modelassesstaskauto/ModelAssessTaskStoplistBackupDO.java Normal file
View File

@ -0,0 +1,43 @@
package cn.iocoder.yudao.module.llm.dal.dataobject.modelassesstaskauto;
import lombok.*;
import java.util.*;
import java.time.LocalDateTime;
import java.time.LocalDateTime;
import com.baomidou.mybatisplus.annotation.*;
import cn.iocoder.yudao.framework.mybatis.core.dataobject.BaseDO;
/**
* 模型评估任务 —— 自动评估 ——停用词备份 DO
*
* @author 华大大模型
*/
@TableName("llm_model_assess_task_stoplist_backup")
@KeySequence("llm_model_assess_task_stoplist_backup_seq") // 用于 Oracle、PostgreSQL、Kingbase、DB2、H2 数据库的主键自增。如果是 MySQL 等数据库，可不写。
@Data
@EqualsAndHashCode(callSuper = true)
@ToString(callSuper = true)
@Builder
@NoArgsConstructor
@AllArgsConstructor
public class ModelAssessTaskStoplistBackupDO extends BaseDO {
/**
* 评估备份ID
*/
@TableId
private Long backupId;
/**
* 评估任务ID
*/
private Long id;
/**
* 自动任务ID
*/
private Long taskId;
/**
* 停用词ID
*/
private Long stoplistId;
}

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

@ -0,0 +1,21 @@
package cn.iocoder.yudao.module.llm.dal.mysql.modelassesstaskauto;
import cn.iocoder.yudao.framework.mybatis.core.mapper.BaseMapperX;
import cn.iocoder.yudao.framework.mybatis.core.query.LambdaQueryWrapperX;
import cn.iocoder.yudao.module.llm.dal.dataobject.modelassesstaskauto.ModelAssessTaskAutoBackupDO;
import org.apache.ibatis.annotations.Mapper;
import java.util.*;
/**
* 模型评估任务 —— 自动评估备份 Mapper
*
* @author 华大大模型
*/
@Mapper
public interface ModelAssessTaskAutoBackupMapper extends BaseMapperX<ModelAssessTaskAutoBackupDO> {
}

22
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/mysql/modelassesstaskauto/ModelAssessTaskStoplistBackupMapper.java Normal file
View File

@ -0,0 +1,22 @@
package cn.iocoder.yudao.module.llm.dal.mysql.modelassesstaskauto;
import java.util.*;
import cn.iocoder.yudao.framework.common.pojo.PageResult;
import cn.iocoder.yudao.framework.mybatis.core.query.LambdaQueryWrapperX;
import cn.iocoder.yudao.framework.mybatis.core.mapper.BaseMapperX;
import cn.iocoder.yudao.module.llm.dal.dataobject.modelassesstaskauto.ModelAssessTaskStoplistBackupDO;
import org.apache.ibatis.annotations.Mapper;
/**
* 模型评估任务 —— 自动评估 ——停用词备份 Mapper
*
* @author 华大大模型
*/
@Mapper
public interface ModelAssessTaskStoplistBackupMapper extends BaseMapperX<ModelAssessTaskStoplistBackupDO> {
}

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

@ -0,0 +1,16 @@
package cn.iocoder.yudao.module.llm.service.modelassesstaskauto;
import cn.iocoder.yudao.module.llm.dal.dataobject.modelassesstaskauto.ModelAssessTaskAutoBackupDO;
import java.util.*;
import javax.validation.*;
/**
* 模型评估任务 —— 自动评估备份 Service 接口
*
* @author 华大大模型
*/
public interface ModelAssessTaskAutoBackupService {
}

19
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelassesstaskauto/ModelAssessTaskAutoBackupServiceImpl.java Normal file
View File

@ -0,0 +1,19 @@
package cn.iocoder.yudao.module.llm.service.modelassesstaskauto;
import org.springframework.stereotype.Service;
import javax.annotation.Resource;
import org.springframework.validation.annotation.Validated;
import org.springframework.transaction.annotation.Transactional;
/**
* 模型评估任务 —— 自动评估备份 Service 实现类
*
* @author 华大大模型
*/
@Service
@Validated
public class ModelAssessTaskAutoBackupServiceImpl implements ModelAssessTaskAutoBackupService {
}

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

@ -53,4 +53,8 @@ public interface ModelAssessTaskAutoService {
PageResult<ModelAssessTaskAutoDO> getModelAssessTaskAutoPage(ModelAssessTaskAutoPageReqVO pageReqVO);
PageResult<ModelAssessTaskAutoRespVO> getModelAssessTaskAutoPage1(ModelAssessTaskAutoPageReqVO pageReqVO);
}
void dataBackup(Long id);
void cancelTheBackUp(Long id);
}

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

@ -1,15 +1,20 @@
package cn.iocoder.yudao.module.llm.service.modelassesstaskauto;
import cn.iocoder.yudao.framework.mybatis.core.query.LambdaQueryWrapperX;
import cn.iocoder.yudao.module.llm.controller.admin.modelassesstaskmanual.vo.ModelAssessTaskManualRespVO;
import cn.iocoder.yudao.module.llm.dal.dataobject.basemodel.BaseModelDO;
import cn.iocoder.yudao.module.llm.dal.dataobject.dataset.DatasetDO;
import cn.iocoder.yudao.module.llm.dal.dataobject.modelassessstoplist.ModelAssessStoplistDO;
import cn.iocoder.yudao.module.llm.dal.dataobject.modelassesstaskauto.ModelAssessTaskAutoBackupDO;
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.modelassesstaskmanual.ModelAssessTaskManualDO;
import cn.iocoder.yudao.module.llm.dal.dataobject.modelservice.ModelServiceDO;
import cn.iocoder.yudao.module.llm.dal.mysql.basemodel.BaseModelMapper;
import cn.iocoder.yudao.module.llm.dal.mysql.dataset.DatasetMapper;
import cn.iocoder.yudao.module.llm.dal.mysql.modelassessstoplist.ModelAssessStoplistMapper;
import cn.iocoder.yudao.module.llm.dal.mysql.modelassesstaskauto.ModelAssessTaskAutoBackupMapper;
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.modelservice.ModelServiceMapper;
import com.alibaba.druid.util.StringUtils;
@ -56,6 +61,10 @@ public class ModelAssessTaskAutoServiceImpl implements ModelAssessTaskAutoServic
private ModelAssessStoplistMapper modelAssessStoplistMapper;
@Resource
private BaseModelMapper baseModelMapper;
@Resource
private ModelAssessTaskAutoBackupMapper modelAssessTaskAutoBackupMapper;
@Resource
private ModelAssessTaskStoplistBackupMapper modelAssessTaskStoplistBackupMapper;
@Override
@ -211,4 +220,33 @@ public class ModelAssessTaskAutoServiceImpl implements ModelAssessTaskAutoServic
return result;
}
@Override
public void dataBackup(Long id) {
ModelAssessTaskAutoDO modelAssessTaskAutoDO = modelAssessTaskAutoMapper.selectById(id);
if (modelAssessTaskAutoDO == null){
throw exception(MODEL_ASSESS_TASK_AUTO_NOT_EXISTS);
}
ModelAssessTaskAutoBackupDO bean = BeanUtils.toBean(modelAssessTaskAutoDO, ModelAssessTaskAutoBackupDO.class);
modelAssessTaskAutoBackupMapper.insert(bean);
modelAssessTaskAutoDO.setBackupId(bean.getId());
modelAssessTaskAutoMapper.updateById(modelAssessTaskAutoDO);
List<ModelAssessTaskStoplistDO> modelAssessTaskStoplistDOS = modelingTaskStoplistMapper.selectList(new LambdaQueryWrapper<ModelAssessTaskStoplistDO>().eq(ModelAssessTaskStoplistDO::getTaskId, id));
if (CollectionUtils.isEmpty(modelAssessTaskStoplistDOS)){
throw exception(MODEL_ASSESS_TASK_STOPLIST_NOT_EXISTS);
}
List<ModelAssessTaskStoplistBackupDO> bean1 = BeanUtils.toBean(modelAssessTaskStoplistDOS, ModelAssessTaskStoplistBackupDO.class);
modelAssessTaskStoplistBackupMapper.insertBatch(bean1);
}
@Override
public void cancelTheBackUp(Long id) {
modelAssessTaskAutoBackupMapper.delete(new LambdaQueryWrapper<ModelAssessTaskAutoBackupDO>().eq(ModelAssessTaskAutoBackupDO::getId, id));
ModelAssessTaskAutoDO modelAssessTaskAutoDO = new ModelAssessTaskAutoDO();
modelAssessTaskAutoDO.setId(id);
modelAssessTaskAutoDO.setBackupId(null);
modelAssessTaskAutoMapper.updateById(modelAssessTaskAutoDO);
modelAssessTaskStoplistBackupMapper.delete(new LambdaQueryWrapper<ModelAssessTaskStoplistBackupDO>().eq(ModelAssessTaskStoplistBackupDO::getBackupId, id));
}
}

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

@ -0,0 +1,16 @@
package cn.iocoder.yudao.module.llm.service.modelassesstaskauto;
import cn.iocoder.yudao.module.llm.dal.dataobject.modelassesstaskauto.ModelAssessTaskStoplistBackupDO;
import java.util.*;
import javax.validation.*;
/**
* 模型评估任务 —— 自动评估 ——停用词备份 Service 接口
*
* @author 华大大模型
*/
public interface ModelAssessTaskStoplistBackupService {
}

19
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelassesstaskauto/ModelAssessTaskStoplistBackupServiceImpl.java Normal file
View File

@ -0,0 +1,19 @@
package cn.iocoder.yudao.module.llm.service.modelassesstaskauto;
import cn.iocoder.yudao.framework.common.util.object.BeanUtils;
import cn.iocoder.yudao.module.llm.dal.dataobject.modelassesstaskauto.ModelAssessTaskStoplistBackupDO;
import cn.iocoder.yudao.module.llm.dal.mysql.modelassesstaskauto.ModelAssessTaskStoplistBackupMapper;
import org.springframework.stereotype.Service;
import javax.annotation.Resource;
import org.springframework.validation.annotation.Validated;
import org.springframework.transaction.annotation.Transactional;
/**
* 模型评估任务 —— 自动评估 ——停用词备份 Service 实现类
*
* @author 华大大模型
*/
@Service
@Validated
public class ModelAssessTaskStoplistBackupServiceImpl implements ModelAssessTaskStoplistBackupService {
}

12
yudao-module-llm/yudao-module-llm-biz/src/main/resources/mapper/modelassesstaskauto/ModelAssessTaskAutoBackupMapper.xml Normal file
View File

@ -0,0 +1,12 @@
<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
<mapper namespace="cn.iocoder.yudao.module.llm.dal.mysql.modelassesstaskauto.ModelAssessTaskAutoBackupMapper">
<!--
一般情况下，尽可能使用 Mapper 进行 CRUD 增删改查即可。
无法满足的场景，例如说多表关联查询，才使用 XML 编写 SQL。
代码生成器暂时只生成 Mapper XML 文件本身，更多推荐 MybatisX 快速开发插件来生成查询。
文档可见：https://www.iocoder.cn/MyBatis/x-plugins/
-->
</mapper>

12
yudao-module-llm/yudao-module-llm-biz/src/main/resources/mapper/modelassesstaskauto/ModelAssessTaskStoplistBackupMapper.xml Normal file
View File

@ -0,0 +1,12 @@
<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
<mapper namespace="cn.iocoder.yudao.module.llm.dal.mysql.modelassesstaskauto.ModelAssessTaskStoplistBackupMapper">
<!--
一般情况下，尽可能使用 Mapper 进行 CRUD 增删改查即可。
无法满足的场景，例如说多表关联查询，才使用 XML 编写 SQL。
代码生成器暂时只生成 Mapper XML 文件本身，更多推荐 MybatisX 快速开发插件来生成查询。
文档可见：https://www.iocoder.cn/MyBatis/x-plugins/
-->
</mapper>