DATA_MIDDLE_PLATFORM/Data_Middle_Platform_Server
3
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

fix(llm): 修复知识库文档重复生成向量问题

Browse Source 
- 将 knowledge.getId() 替换为 knowledge.getFileId()，确保正确使用文件 ID
- 修改 knowledgeEmbed 方法调用，使用 fileID替代 knowledgeBaseId
- 更新 KnowledgeRagEmbedReqVO 中的 fileId 设置
This commit is contained in:
Liuyang 2025-03-14 09:14:08 +08:00
parent e303b6f78c
commit e14b54a7f8
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/async/AsyncKnowledgeBase.java
View File

@ -78,7 +78,7 @@ public class AsyncKnowledgeBase {
log.info("为文档准备 RegUploadReqVO，文档 ID: {}", knowledge.getId());
RegUploadReqVO regUploadReqVO = new RegUploadReqVO()
.setUrl(llmBackendProperties.getRagEmbed())
.setFileId(String.valueOf(knowledge.getId()))
.setFileId(String.valueOf(knowledge.getFileId()))
.setFileName(knowledge.getDocumentName())
.setFileUrl(knowledge.getFileUrl());
@ -87,7 +87,7 @@ public class AsyncKnowledgeBase {
if (lastIndex != -1) {
String extension = knowledge.getDocumentName().substring(lastIndex + 1).toLowerCase();
log.info("文档扩展名: {}", extension);
knowledgeEmbed(knowledge, knowledge.getKnowledgeBaseId(), knowledgeParameters);
knowledgeEmbed(knowledge, knowledge.getFileId(), knowledgeParameters);
} else {
log.warn("文档无扩展名，跳过处理，文档 ID: {}", knowledge.getId());
}
@ -130,7 +130,7 @@ public class AsyncKnowledgeBase {
// 创建知识向量
KnowledgeRagEmbedReqVO ragEmbedReqVo = new KnowledgeRagEmbedReqVO()
.setFileId(String.valueOf(knowledge.getId()))
.setFileId(String.valueOf(knowledge.getFileId()))
.setFileName(knowledge.getDocumentName())
.setFileUrl(knowledge.getFileUrl())
.setChunkSize(knowledgeParameters.get("chunkSize"))