diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/modelassesstaskauto/ModelAssessTaskAutoController.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/modelassesstaskauto/ModelAssessTaskAutoController.java index 94a1e8c0f..7a5fb310b 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/modelassesstaskauto/ModelAssessTaskAutoController.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/modelassesstaskauto/ModelAssessTaskAutoController.java @@ -110,14 +110,18 @@ public class ModelAssessTaskAutoController { @PostMapping("/backedUp") @Operation(summary = "备份") - public void backedUp(ModelAssessTaskManualPageReqVO pageReqVO) { + public CommonResult backedUp(@RequestParam("id") Long id) { //TODO 自动评估-备份待补充 + modelAssessTaskAutoService.dataBackup(id); + return success(true); } @PostMapping("/cancelTheBackUp") @Operation(summary = "取消备份") - public void cancelTheBackUp(ModelAssessTaskManualPageReqVO pageReqVO) { + public CommonResult cancelTheBackUp(@RequestParam("id") Long id) { //TODO 自动评估-取消备份待补充 + modelAssessTaskAutoService.cancelTheBackUp(id); + return success(true); } @PostMapping("/reEvaluation") diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/modelassesstaskauto/vo/ModelAssessTaskAutoRespVO.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/modelassesstaskauto/vo/ModelAssessTaskAutoRespVO.java index ca7c04d2f..0f670b9f0 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/modelassesstaskauto/vo/ModelAssessTaskAutoRespVO.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/modelassesstaskauto/vo/ModelAssessTaskAutoRespVO.java @@ -59,4 +59,7 @@ public class ModelAssessTaskAutoRespVO { @Schema(description = "数据集名称") private String datasetName; + @Schema(description = "备份ID") + private Long backupId; + } 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 new file mode 100644 index 000000000..9f039155e --- /dev/null +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/dataobject/modelassesstaskauto/ModelAssessTaskAutoBackupDO.java @@ -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; + +} diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/dataobject/modelassesstaskauto/ModelAssessTaskAutoDO.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/dataobject/modelassesstaskauto/ModelAssessTaskAutoDO.java index ea03d71c2..cf8b961c3 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/dataobject/modelassesstaskauto/ModelAssessTaskAutoDO.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/dataobject/modelassesstaskauto/ModelAssessTaskAutoDO.java @@ -28,6 +28,10 @@ public class ModelAssessTaskAutoDO extends BaseDO { */ @TableId private Long id; + /** + * 备份ID + */ + private Long backupId; /** * 任务名称 */ diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/dataobject/modelassesstaskauto/ModelAssessTaskStoplistBackupDO.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/dataobject/modelassesstaskauto/ModelAssessTaskStoplistBackupDO.java new file mode 100644 index 000000000..64474c9a1 --- /dev/null +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/dataobject/modelassesstaskauto/ModelAssessTaskStoplistBackupDO.java @@ -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; + +} 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 new file mode 100644 index 000000000..835668d71 --- /dev/null +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/mysql/modelassesstaskauto/ModelAssessTaskAutoBackupMapper.java @@ -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 { + + + +} diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/mysql/modelassesstaskauto/ModelAssessTaskStoplistBackupMapper.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/mysql/modelassesstaskauto/ModelAssessTaskStoplistBackupMapper.java new file mode 100644 index 000000000..6c994bbb2 --- /dev/null +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/mysql/modelassesstaskauto/ModelAssessTaskStoplistBackupMapper.java @@ -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 { + + +} diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelassesstaskauto/ModelAssessTaskAutoBackupService.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelassesstaskauto/ModelAssessTaskAutoBackupService.java new file mode 100644 index 000000000..898d2f52a --- /dev/null +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelassesstaskauto/ModelAssessTaskAutoBackupService.java @@ -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 { + +} diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelassesstaskauto/ModelAssessTaskAutoBackupServiceImpl.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelassesstaskauto/ModelAssessTaskAutoBackupServiceImpl.java new file mode 100644 index 000000000..078c87580 --- /dev/null +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelassesstaskauto/ModelAssessTaskAutoBackupServiceImpl.java @@ -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 { + + +} diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelassesstaskauto/ModelAssessTaskAutoService.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelassesstaskauto/ModelAssessTaskAutoService.java index 93d4a87a1..2e68ebef2 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelassesstaskauto/ModelAssessTaskAutoService.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelassesstaskauto/ModelAssessTaskAutoService.java @@ -53,4 +53,8 @@ public interface ModelAssessTaskAutoService { PageResult getModelAssessTaskAutoPage(ModelAssessTaskAutoPageReqVO pageReqVO); PageResult getModelAssessTaskAutoPage1(ModelAssessTaskAutoPageReqVO pageReqVO); -} \ No newline at end of file + + void dataBackup(Long id); + + void cancelTheBackUp(Long id); +} diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelassesstaskauto/ModelAssessTaskAutoServiceImpl.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelassesstaskauto/ModelAssessTaskAutoServiceImpl.java index 4c23a6b44..1ea02d1e9 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelassesstaskauto/ModelAssessTaskAutoServiceImpl.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelassesstaskauto/ModelAssessTaskAutoServiceImpl.java @@ -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 modelAssessTaskStoplistDOS = modelingTaskStoplistMapper.selectList(new LambdaQueryWrapper().eq(ModelAssessTaskStoplistDO::getTaskId, id)); + if (CollectionUtils.isEmpty(modelAssessTaskStoplistDOS)){ + throw exception(MODEL_ASSESS_TASK_STOPLIST_NOT_EXISTS); + } + List bean1 = BeanUtils.toBean(modelAssessTaskStoplistDOS, ModelAssessTaskStoplistBackupDO.class); + modelAssessTaskStoplistBackupMapper.insertBatch(bean1); + + } + + @Override + public void cancelTheBackUp(Long id) { + modelAssessTaskAutoBackupMapper.delete(new LambdaQueryWrapper().eq(ModelAssessTaskAutoBackupDO::getId, id)); + ModelAssessTaskAutoDO modelAssessTaskAutoDO = new ModelAssessTaskAutoDO(); + modelAssessTaskAutoDO.setId(id); + modelAssessTaskAutoDO.setBackupId(null); + modelAssessTaskAutoMapper.updateById(modelAssessTaskAutoDO); + modelAssessTaskStoplistBackupMapper.delete(new LambdaQueryWrapper().eq(ModelAssessTaskStoplistBackupDO::getBackupId, id)); + } + } diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelassesstaskauto/ModelAssessTaskStoplistBackupService.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelassesstaskauto/ModelAssessTaskStoplistBackupService.java new file mode 100644 index 000000000..427f22c81 --- /dev/null +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelassesstaskauto/ModelAssessTaskStoplistBackupService.java @@ -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 { + +} diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelassesstaskauto/ModelAssessTaskStoplistBackupServiceImpl.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelassesstaskauto/ModelAssessTaskStoplistBackupServiceImpl.java new file mode 100644 index 000000000..093b26865 --- /dev/null +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelassesstaskauto/ModelAssessTaskStoplistBackupServiceImpl.java @@ -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 { +} diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/resources/mapper/modelassesstaskauto/ModelAssessTaskAutoBackupMapper.xml b/yudao-module-llm/yudao-module-llm-biz/src/main/resources/mapper/modelassesstaskauto/ModelAssessTaskAutoBackupMapper.xml new file mode 100644 index 000000000..b75309f3a --- /dev/null +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/resources/mapper/modelassesstaskauto/ModelAssessTaskAutoBackupMapper.xml @@ -0,0 +1,12 @@ + + + + + + + diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/resources/mapper/modelassesstaskauto/ModelAssessTaskStoplistBackupMapper.xml b/yudao-module-llm/yudao-module-llm-biz/src/main/resources/mapper/modelassesstaskauto/ModelAssessTaskStoplistBackupMapper.xml new file mode 100644 index 000000000..8064d0862 --- /dev/null +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/resources/mapper/modelassesstaskauto/ModelAssessTaskStoplistBackupMapper.xml @@ -0,0 +1,12 @@ + + + + + + +