From e951153a9f281e9765cbff1392307165e599121e Mon Sep 17 00:00:00 2001 From: chenxudong Date: Thu, 3 Apr 2025 12:07:02 +0800 Subject: [PATCH] =?UTF-8?q?=E6=96=B0=E5=A2=9E=E8=AE=B0=E5=BD=95AI=E4=B8=8A?= =?UTF-8?q?=E4=BC=A0=E6=96=87=E4=BB=B6=E7=9A=84=E5=A4=A7=E5=B0=8F=E8=AE=B0?= =?UTF-8?q?=E5=BD=95=E3=80=82?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../software/manage/pojo/models/AiFileUploadRecord.java | 3 +++ .../software/manage/service/serviceimpl/ChatService.java | 1 + 2 files changed, 4 insertions(+) diff --git a/electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/pojo/models/AiFileUploadRecord.java b/electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/pojo/models/AiFileUploadRecord.java index ce9c304..abbe98e 100644 --- a/electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/pojo/models/AiFileUploadRecord.java +++ b/electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/pojo/models/AiFileUploadRecord.java @@ -21,4 +21,7 @@ public class AiFileUploadRecord extends BaseModel { @TableField(value = "vector_id") private String vectorId; + + @TableField(value = "file_size") + private long fileSize; } 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 6feb601..095877d 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 @@ -81,6 +81,7 @@ public class ChatService { vectorStore.write(ListUtil.of(document)); aiFileUploadRecordMapper.insert(new AiFileUploadRecord().setId(IdWorker.getSnowFlakeIdString()) .setVectorId(document.getId()) + .setFileSize(file.getSize()) .setFileMd5(fileMd5) .setFileName(file.getOriginalFilename()));