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 76b6c289c..464339724 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 @@ -62,13 +62,13 @@ public class AsyncKnowledgeBase { int lastIndex = knowledge.getDocumentName().lastIndexOf("."); if (lastIndex != -1) { - ragHttpService.embedUploadFile(regUploadReqVO); -// String extension = knowledge.getDocumentName().substring(lastIndex + 1).toLowerCase(); -// if ("txt".equals(extension)) { -// ragHttpService.embedUploadFile(regUploadReqVO); -// } else { -// knowledgeEmbed(knowledge, knowledge.getKnowledgeBaseId()); -// } + + String extension = knowledge.getDocumentName().substring(lastIndex + 1).toLowerCase(); + if ("txt".equals(extension)) { + ragHttpService.embedUploadFile(regUploadReqVO); + } else { + knowledgeEmbed(knowledge, knowledge.getKnowledgeBaseId()); + } } } catch (Exception e) { log.error("the creation of the knowledge base error {}", e.getMessage());