diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/dataset/DatasetQuestionController.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/dataset/DatasetQuestionController.java index 9f4409a6a..fa585ce3b 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/dataset/DatasetQuestionController.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/dataset/DatasetQuestionController.java @@ -80,33 +80,40 @@ public class DatasetQuestionController { int count = 0; List id = new ArrayList<>(); HSSFRow row = sheet.createRow(count); - row.createCell(0).setCellValue("问题内容"); + row.createCell(0).setCellValue("system"); row.getCell(0).setCellStyle(cellStyle); - row.createCell(1).setCellValue("标注内容"); + row.createCell(1).setCellValue("question"); row.getCell(1).setCellStyle(cellStyle); + row.createCell(2).setCellValue("answer"); + row.getCell(2).setCellStyle(cellStyle); id.add(count); count++; pageReqVO.setPageSize(PageParam.PAGE_SIZE_NONE); List list = datasetQuestionService.getDatasetQuestionPage(pageReqVO).getList(); for (DatasetQuestionRespVO item : list){ + String system = item.getSystem(); String question = item.getQuestion(); List datasetAnswerRespVO = item.getDatasetAnswerRespVO(); List collect = datasetAnswerRespVO.stream().map(DatasetAnswerRespVO::getAnswer).collect(Collectors.toList()); if (collect.size() == 0){ row = sheet.createRow(count); - row.createCell(0).setCellValue(question); + row.createCell(0).setCellValue(system); row.getCell(0).setCellStyle(cellStyle); - row.createCell(1).setCellValue(""); + row.createCell(1).setCellValue(question); row.getCell(1).setCellStyle(cellStyle); + row.createCell(2).setCellValue(""); + row.getCell(2).setCellStyle(cellStyle); id.add(count); count++; }else { for (String s : collect) { row = sheet.createRow(count); - row.createCell(0).setCellValue(question); + row.createCell(0).setCellValue(system); row.getCell(0).setCellStyle(cellStyle); - row.createCell(1).setCellValue(s); + row.createCell(1).setCellValue(question); row.getCell(1).setCellStyle(cellStyle); + row.createCell(2).setCellValue(s); + row.getCell(2).setCellStyle(cellStyle); count++; } id.add(count-1); @@ -115,7 +122,7 @@ public class DatasetQuestionController { //合并相同内容的单元格 for (int i = 0; i < id.size() - 1; i++){ if (id.get(i+1)-id.get(i)>1) { - sheet.addMergedRegion(new CellRangeAddress(id.get(i)+1, id.get(i + 1), 0, 0)); + sheet.addMergedRegion(new CellRangeAddress(id.get(i)+1, id.get(i + 1), 0, 1)); } } // 导出 Excel