From 0a75c9c77394ea4d9e39da86c6b9b1a6a7bef05b Mon Sep 17 00:00:00 2001
From: chenxudong <chenxudong@cqida.cn>
Date: Wed, 9 Apr 2025 18:00:45 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96=E4=BB=A3=E7=A0=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../manage/service/serviceimpl/ChatService.java      | 12 +++++-------
 1 file changed, 5 insertions(+), 7 deletions(-)

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<PageFile> pageInfo;
-                if (StrUtil.equals("xlsx", filePath)) {
-                    pageInfo = OfficeFileUtil.parseXlsxByPage(filePath);
-                } else {
-                    pageInfo = OfficeFileUtil.parseXlsByPage(filePath);
-                }
+                List<PageFile> pageInfo = StrUtil.equals("xlsx", filePath) ?
+                        OfficeFileUtil.parseXlsxByPage(filePath) :
+                        OfficeFileUtil.parseXlsByPage(filePath);
                 for (PageFile pageFile : pageInfo) {
                     Map<String, Object> 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)