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 c5479e11e..76b6c289c 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 @@ -11,6 +11,7 @@ import cn.iocoder.yudao.module.llm.service.http.vo.KnowledgeRagEmbedReqVO; import cn.iocoder.yudao.module.llm.service.http.vo.RegUploadReqVO; import org.slf4j.Logger; import org.slf4j.LoggerFactory; +import org.springframework.scheduling.annotation.Async; import org.springframework.stereotype.Service; import javax.annotation.Resource; @@ -37,6 +38,7 @@ public class AsyncKnowledgeBase { // 向向量知识库创建文件 + @Async public void createKnowledgeBase (List knowledgeList, List ids) { if (!CollectionUtils.isAnyEmpty(ids)) { String mes = ragHttpService.ragDocumentsDel(llmBackendProperties.getRagDocumentsDel(), ids); @@ -60,13 +62,13 @@ public class AsyncKnowledgeBase { int lastIndex = knowledge.getDocumentName().lastIndexOf("."); if (lastIndex != -1) { - - String extension = knowledge.getDocumentName().substring(lastIndex + 1).toLowerCase(); - if ("txt".equals(extension)) { - ragHttpService.embedUploadFile(regUploadReqVO); - } else { - knowledgeEmbed(knowledge, knowledge.getKnowledgeBaseId()); - } + ragHttpService.embedUploadFile(regUploadReqVO); +// 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());