新增记录AI上传文件的大小记录。
This commit is contained in:
parent
87cfbb4d0e
commit
e951153a9f
|
|
@ -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;
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -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()));
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue