新增记录AI上传文件的大小记录。

This commit is contained in:
chenxudong 2025-04-03 12:07:02 +08:00
parent 87cfbb4d0e
commit e951153a9f
2 changed files with 4 additions and 0 deletions

View File

@ -21,4 +21,7 @@ public class AiFileUploadRecord extends BaseModel {
@TableField(value = "vector_id") @TableField(value = "vector_id")
private String vectorId; private String vectorId;
@TableField(value = "file_size")
private long fileSize;
} }

View File

@ -81,6 +81,7 @@ public class ChatService {
vectorStore.write(ListUtil.of(document)); vectorStore.write(ListUtil.of(document));
aiFileUploadRecordMapper.insert(new AiFileUploadRecord().setId(IdWorker.getSnowFlakeIdString()) aiFileUploadRecordMapper.insert(new AiFileUploadRecord().setId(IdWorker.getSnowFlakeIdString())
.setVectorId(document.getId()) .setVectorId(document.getId())
.setFileSize(file.getSize())
.setFileMd5(fileMd5) .setFileMd5(fileMd5)
.setFileName(file.getOriginalFilename())); .setFileName(file.getOriginalFilename()));