From e14b54a7f85d1f11d51d9526d171fb9d55992e05 Mon Sep 17 00:00:00 2001 From: Liuyang <2746366019@qq.com> Date: Fri, 14 Mar 2025 09:14:08 +0800 Subject: [PATCH] =?UTF-8?q?fix(llm):=20=E4=BF=AE=E5=A4=8D=E7=9F=A5?= =?UTF-8?q?=E8=AF=86=E5=BA=93=E6=96=87=E6=A1=A3=E9=87=8D=E5=A4=8D=E7=94=9F?= =?UTF-8?q?=E6=88=90=E5=90=91=E9=87=8F=E9=97=AE=E9=A2=98?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - 将 knowledge.getId() 替换为 knowledge.getFileId()，确保正确使用文件 ID - 修改 knowledgeEmbed 方法调用，使用 fileID替代 knowledgeBaseId - 更新 KnowledgeRagEmbedReqVO 中的 fileId 设置 --- .../yudao/module/llm/service/async/AsyncKnowledgeBase.java | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/async/AsyncKnowledgeBase.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/async/AsyncKnowledgeBase.java index 3c10295c4..63c8113d2 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/async/AsyncKnowledgeBase.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/async/AsyncKnowledgeBase.java @@ -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"))