人工备份 导出

This commit is contained in:
limin 2025-01-04 16:27:20 +08:00
parent af09e51c69
commit 88e56f6cb0
4 changed files with 64 additions and 10 deletions

View File

@ -2,8 +2,10 @@ package cn.iocoder.yudao.module.llm.controller.admin.modelassesstaskmanualbackup
import cn.iocoder.yudao.module.llm.controller.admin.modelassesstaskauto.vo.ModelAssessTaskAutoPageReqVO;
import cn.iocoder.yudao.module.llm.controller.admin.modelassesstaskauto.vo.ModelAssessTaskAutoRespVO;
import cn.iocoder.yudao.module.llm.controller.admin.modelassesstaskmanual.vo.ManualAssessmentPageRespVO;
import cn.iocoder.yudao.module.llm.controller.admin.modelassesstaskmanual.vo.ManualEvaluationReportRespVO;
import cn.iocoder.yudao.module.llm.controller.admin.modelassesstaskmanual.vo.ModelAssessTaskManualRespVO;
import cn.iocoder.yudao.module.llm.utils.ManalExcelExportUtils;
import org.springframework.web.bind.annotation.*;
import javax.annotation.Resource;
import org.springframework.validation.annotation.Validated;
@ -104,13 +106,13 @@ public class ModelAssessTaskManualBackupController {
@Operation(summary = "导出人工评估备份 Excel")
@PreAuthorize("@ss.hasPermission('llm:model-assess-task-manual-backup:export')")
@ApiAccessLog(operateType = EXPORT)
public void exportModelAssessTaskManualBackupExcel(@Valid ModelAssessTaskManualBackupPageReqVO pageReqVO,
public void exportModelAssessTaskManualBackupExcel(@Valid ManualModelAnswerBackupPageReqVO pageReqVO,
HttpServletResponse response) throws IOException {
pageReqVO.setPageSize(PageParam.PAGE_SIZE_NONE);
List<ModelAssessTaskManualBackupDO> list = modelAssessTaskManualBackupService.getModelAssessTaskManualBackupPage(pageReqVO).getList();
List<ManualAssessmentPageRespVO> list = modelAssessTaskManualBackupService.getManualAssessment(pageReqVO).getList();
// 导出 Excel
ExcelUtils.write(response, "人工评估备份.xls", "数据", ModelAssessTaskManualBackupRespVO.class,
BeanUtils.toBean(list, ModelAssessTaskManualBackupRespVO.class));
ManalExcelExportUtils.write(response, "人工评估备份.xls", "人工评估导出备份", ModelAssessTaskManualBackupRespVO.class,
list);
}
@GetMapping("/gitReport")
@Operation(summary = "查看报告详情")

View File

@ -14,7 +14,8 @@ import static cn.iocoder.yudao.framework.common.util.date.DateUtils.FORMAT_YEAR_
@EqualsAndHashCode(callSuper = true)
@ToString(callSuper = true)
public class ModelAssessTaskManualBackupPageReqVO extends PageParam {
@Schema(description = "导出id 主表id")
private Long id;
@Schema(description = "原评估信息Id", example = "8848")
private Long srcId;

View File

@ -5,6 +5,8 @@ import javax.validation.*;
import cn.iocoder.yudao.module.llm.controller.admin.modelassesstaskauto.vo.ModelAssessTaskAutoPageReqVO;
import cn.iocoder.yudao.module.llm.controller.admin.modelassesstaskauto.vo.ModelAssessTaskAutoRespVO;
import cn.iocoder.yudao.module.llm.controller.admin.modelassesstaskmanual.manualAnnoVo.ManualModelAnswerPageReqVO;
import cn.iocoder.yudao.module.llm.controller.admin.modelassesstaskmanual.vo.ManualAssessmentPageRespVO;
import cn.iocoder.yudao.module.llm.controller.admin.modelassesstaskmanual.vo.ManualEvaluationReportRespVO;
import cn.iocoder.yudao.module.llm.controller.admin.modelassesstaskmanualbackup.vo.*;
import cn.iocoder.yudao.module.llm.dal.dataobject.modelassesstaskmanualbackup.ModelAssessTaskManualBackupDO;
@ -69,5 +71,9 @@ public interface ModelAssessTaskManualBackupService {
ModelAssessTaskAutoRespVO getModelAssessTaskAuto(Long id);
PageResult<ModelAssessTaskManualBackupRespVO> getModelAssessTaskManualPage1(ModelAssessTaskManualBackupPageReqVO pageReqVO);
// 备份导出列表查询
PageResult<ManualAssessmentPageRespVO> getManualAssessment(ManualModelAnswerBackupPageReqVO pageReqVO);
}

View File

@ -1,14 +1,15 @@
package cn.iocoder.yudao.module.llm.service.modelassesstaskmanualbackup;
import cn.iocoder.yudao.module.llm.controller.admin.dataset.vo.DatasetAnswerRespVO;
import cn.iocoder.yudao.module.llm.controller.admin.modelassesstaskauto.vo.ModelAssessTaskAutoPageReqVO;
import cn.iocoder.yudao.module.llm.controller.admin.modelassesstaskauto.vo.ModelAssessTaskAutoRespVO;
import cn.iocoder.yudao.module.llm.controller.admin.modelassesstaskauto.vo.ModelAssessTaskStoplistRespVO;
import cn.iocoder.yudao.module.llm.controller.admin.modelassesstaskmanual.vo.LabelInformationVO;
import cn.iocoder.yudao.module.llm.controller.admin.modelassesstaskmanual.vo.ManualEvaluationReportRespVO;
import cn.iocoder.yudao.module.llm.controller.admin.modelassesstaskmanual.vo.ModelAssessTaskDimensionRespVO;
import cn.iocoder.yudao.module.llm.controller.admin.modelassesstaskmanual.vo.ModelAssessTaskManualRespVO;
import cn.iocoder.yudao.module.llm.controller.admin.modelassesstaskmanual.manualAnnoVo.ManualModelAnswerPageReqVO;
import cn.iocoder.yudao.module.llm.controller.admin.modelassesstaskmanual.vo.*;
import cn.iocoder.yudao.module.llm.dal.dataobject.basemodel.BaseModelDO;
import cn.iocoder.yudao.module.llm.dal.dataobject.dataset.DatasetAnswerDO;
import cn.iocoder.yudao.module.llm.dal.dataobject.dataset.DatasetDO;
import cn.iocoder.yudao.module.llm.dal.dataobject.dataset.DatasetQuestionDO;
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.ModelAssessTaskAutoDO;
@ -22,7 +23,9 @@ import cn.iocoder.yudao.module.llm.dal.dataobject.modelassesstaskmanualbackup.Ma
import cn.iocoder.yudao.module.llm.dal.dataobject.modelassesstaskmanualbackup.ModelAssessTaskDimensionBackupDO;
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.DatasetAnswerMapper;
import cn.iocoder.yudao.module.llm.dal.mysql.dataset.DatasetMapper;
import cn.iocoder.yudao.module.llm.dal.mysql.dataset.DatasetQuestionMapper;
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;
@ -84,6 +87,10 @@ public class ModelAssessTaskManualBackupServiceImpl implements ModelAssessTaskMa
private ModelAssessTaskStoplistBackupMapper modelAssessTaskStoplistBackupMapper;
@Resource
private ModelAssessStoplistMapper modelAssessStoplistMapper;
@Resource
private DatasetQuestionMapper datasetQuestionMapper;
@Resource
private DatasetAnswerMapper datasetAnswerMapper;
@Override
public Long createModelAssessTaskManualBackup(ModelAssessTaskManualBackupSaveReqVO createReqVO) {
@ -367,6 +374,44 @@ public class ModelAssessTaskManualBackupServiceImpl implements ModelAssessTaskMa
}
return result;
}
@Override
public PageResult<ManualAssessmentPageRespVO> getManualAssessment(ManualModelAnswerBackupPageReqVO pageReqVO) {
ModelAssessTaskManualBackupRespVO modelAssessTaskManual = getModelAssessTaskManualBackup1(pageReqVO.getManalTaskId());
PageResult<ManualAssessmentPageRespVO> result = new PageResult<>();
PageResult<ManualModelAnswerBackupDO> modelAnswerDOPageResult = manualModelAnswerBackupMapper.selectPage(pageReqVO);
List<ManualAssessmentPageRespVO> res = new ArrayList<>();
if(!cn.iocoder.yudao.framework.common.util.collection.CollectionUtils.isAnyEmpty(modelAnswerDOPageResult.getList())){
modelAnswerDOPageResult.getList().forEach(modelAnswerDO -> {
ManualAssessmentPageRespVO resp = BeanUtils.toBean(modelAssessTaskManual, ManualAssessmentPageRespVO.class);
DatasetQuestionDO datasetQuestionDO = datasetQuestionMapper.selectById(modelAnswerDO.getQuestionId());
List<DatasetAnswerDO> datasetAnswerDOS = datasetAnswerMapper.selectList(new LambdaQueryWrapper<>(DatasetAnswerDO.class).eq(DatasetAnswerDO::getQuestionId, modelAnswerDO.getQuestionId()));
resp.setQuestion(datasetQuestionDO.getQuestion());
resp.setSystem(datasetQuestionDO.getSystem());
resp.setAnswers(BeanUtils.toBean(datasetAnswerDOS, DatasetAnswerRespVO.class));
resp.setModelAnswer(modelAnswerDO.getModelAnswer());
resp.setId(modelAnswerDO.getId());
resp.setStatus(modelAnswerDO.getStatus());
resp.setManalTaskId(modelAnswerDO.getManalTaskId());
if (!cn.iocoder.yudao.framework.common.util.collection.CollectionUtils.isAnyEmpty(modelAnswerDO.getReqRespVos())){
resp.setReqRespVos(modelAnswerDO.getReqRespVos());
}else {
if (!cn.iocoder.yudao.framework.common.util.collection.CollectionUtils.isAnyEmpty(modelAssessTaskManual.getDimensions())){
List<HashMap<String, Object>> map = new ArrayList<>();
modelAssessTaskManual.getDimensions().forEach(dimension -> {
HashMap<String, Object> map1 = new HashMap<>();
map1.put("label", dimension.getDimensionName());
map1.put("score", 0);
map.add(map1);
});
resp.setReqRespVos(map);
}
}
res.add(resp);
});
}
result.setList(res);
result.setTotal(modelAnswerDOPageResult.getTotal());
return result;
}
}