q792602257/DataMate
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: Add original file ID to document metadata in RagEtlService

Browse Source
This commit is contained in:
Dallas98
2025-12-01 17:04:52 +08:00
parent bb3345268e
commit 9fc35f066f
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
backend/services/rag-indexer-service/src/main/java/com/datamate/rag/indexer/infrastructure/event/RagEtlService.java
View File

@@ -106,6 +106,7 @@ public class RagEtlService {
document = new HtmlToTextDocumentTransformer().transform(document); document = new HtmlToTextDocumentTransformer().transform(document);
} }
document.metadata().put("rag_file_id", ragFile.getId()); document.metadata().put("rag_file_id", ragFile.getId());
document.metadata().put("original_file_id", ragFile.getFileId());
// 使用文档分块器对文档进行分块 // 使用文档分块器对文档进行分块
DocumentSplitter splitter = documentSplitter(event.addFilesReq()); DocumentSplitter splitter = documentSplitter(event.addFilesReq());
List<TextSegment> split = splitter.split(document); List<TextSegment> split = splitter.split(document);