diff --git a/electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/service/serviceimpl/ChatService.java b/electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/service/serviceimpl/ChatService.java index ea5f6de..2b389b8 100644 --- a/electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/service/serviceimpl/ChatService.java +++ b/electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/service/serviceimpl/ChatService.java @@ -114,12 +114,9 @@ public class ChatService { } case "xls", "xlsx" -> { String filePath = elePropertyConfig.getEleTmpPath() + File.separator + fileMd5 + "." + fileType; - List pageInfo; - if (StrUtil.equals("xlsx", filePath)) { - pageInfo = OfficeFileUtil.parseXlsxByPage(filePath); - } else { - pageInfo = OfficeFileUtil.parseXlsByPage(filePath); - } + List pageInfo = StrUtil.equals("xlsx", filePath) ? + OfficeFileUtil.parseXlsxByPage(filePath) : + OfficeFileUtil.parseXlsByPage(filePath); for (PageFile pageFile : pageInfo) { Map metaData = Map.of("page_number", pageFile.getPageNumber(), "file_name", file.getOriginalFilename()); Document document = new Document(pageFile.getContent(), metaData); @@ -157,7 +154,8 @@ public class ChatService { documents = documents.subList(0, Math.min(elePropertyConfig.getAiMaxDocs(), documents.size())); vectorStore.write(documents); for (Document document : documents) { - aiFileUploadRecordMapper.insert(new AiFileUploadRecord().setId(IdWorker.getSnowFlakeIdString()) + aiFileUploadRecordMapper.insert(new AiFileUploadRecord() + .setId(IdWorker.getSnowFlakeIdString()) .setVectorId(document.getId()) .setFileSize(file.getSize()) .setFileMd5(fileMd5)