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 e5c9c7c..5820eb8 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 @@ -127,7 +127,7 @@ public class ChatService { FileUtil.writeFromStream(file.getInputStream(), pdfPath); } else { FileUtil.writeFromStream(file.getInputStream(), srcPath); - if (StrUtil.equals(fileType, "doc") || StrUtil.equals(fileType, "docx")) { + if (Arrays.asList("doc", "docx").contains(fileType)) { OfficeFileUtil.doc2pdf(srcPath, pdfPath); } else { OfficeFileUtil.ppt2pdf(srcPath, pdfPath); @@ -148,9 +148,7 @@ public class ChatService { List newDocuments = new ArrayList<>(); StringBuilder stringBuilder = new StringBuilder(); for (Document document : documents) { - Map metadata = new HashMap<>(); - metadata.put("fileName", file.getOriginalFilename()); - metadata.put("fileMd5", fileMd5); + Map metadata = Map.of("fileName", file.getOriginalFilename(), "fileMd5", fileMd5); newDocuments.add(new Document(IdWorker.getSnowFlakeIdString(), Objects.requireNonNull(document.getText()), metadata)); stringBuilder.append(document.getText()); }