chenxudong
/
electromagnetic-data-new
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

优化代码

This commit is contained in:
chenxudong 2025-04-09 18:00:45 +08:00
parent 815bf60b65
commit 0a75c9c773
1 changed files with 5 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/service/serviceimpl/ChatService.java
View File

@ -114,12 +114,9 @@ public class ChatService {
} }
case "xls", "xlsx" -> { case "xls", "xlsx" -> {
String filePath = elePropertyConfig.getEleTmpPath() + File.separator + fileMd5 + "." + fileType; String filePath = elePropertyConfig.getEleTmpPath() + File.separator + fileMd5 + "." + fileType;
List<PageFile> pageInfo; List<PageFile> pageInfo = StrUtil.equals("xlsx", filePath) ?
if (StrUtil.equals("xlsx", filePath)) { OfficeFileUtil.parseXlsxByPage(filePath) :
pageInfo = OfficeFileUtil.parseXlsxByPage(filePath); OfficeFileUtil.parseXlsByPage(filePath);
} else {
pageInfo = OfficeFileUtil.parseXlsByPage(filePath);
}
for (PageFile pageFile : pageInfo) { for (PageFile pageFile : pageInfo) {
Map<String, Object> metaData = Map.of("page_number", pageFile.getPageNumber(), "file_name", file.getOriginalFilename()); Map<String, Object> metaData = Map.of("page_number", pageFile.getPageNumber(), "file_name", file.getOriginalFilename());
Document document = new Document(pageFile.getContent(), metaData); Document document = new Document(pageFile.getContent(), metaData);
@ -157,7 +154,8 @@ public class ChatService {
documents = documents.subList(0, Math.min(elePropertyConfig.getAiMaxDocs(), documents.size())); documents = documents.subList(0, Math.min(elePropertyConfig.getAiMaxDocs(), documents.size()));
vectorStore.write(documents); vectorStore.write(documents);
for (Document document : documents) { for (Document document : documents) {
aiFileUploadRecordMapper.insert(new AiFileUploadRecord().setId(IdWorker.getSnowFlakeIdString()) aiFileUploadRecordMapper.insert(new AiFileUploadRecord()
.setId(IdWorker.getSnowFlakeIdString())
.setVectorId(document.getId()) .setVectorId(document.getId())
.setFileSize(file.getSize()) .setFileSize(file.getSize())
.setFileMd5(fileMd5) .setFileMd5(fileMd5)