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 bfa30e921..3408e9964 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 @@ -66,7 +66,7 @@ public class AsyncKnowledgeBase { * @param knowledgeList 文件列表 * @param id */ - @Async +// @Async public void knowledgeEmbed (List knowledgeList, Long id) { if (!CollectionUtils.isAnyEmpty(knowledgeList)) { knowledgeList.forEach(knowledge -> { diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/RagHttpService.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/RagHttpService.java index c71514f8c..f6c5dafdd 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/RagHttpService.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/RagHttpService.java @@ -225,15 +225,20 @@ public class RagHttpService { // 获取知识库向量嵌入的url String ragEmbed = llmBackendProperties.getEmbed(); log.info("url : {}", ragEmbed); - + aDo.setKnowledgeBaseIntro("8-- url: "+ragEmbed); + knowledgeBaseMapper.updateById(aDo); // 构建请求参数 HttpResponse response = Unirest.post(ragEmbed) .field("file_id", reqVO.getFileId()) .field("file", reqVO.getFileInputStream(), reqVO.getFileName()) .asString(); + + aDo.setKnowledgeBaseIntro("9-- url: "+ragEmbed); + knowledgeBaseMapper.updateById(aDo); String body = response.getBody(); log.info("body : {}", body); - + aDo.setKnowledgeBaseIntro("10-- url: "+ragEmbed); + knowledgeBaseMapper.updateById(aDo); JSONObject jsonObject = JSON.parseObject(body); diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/knowledgebase/KnowledgeBaseServiceImpl.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/knowledgebase/KnowledgeBaseServiceImpl.java index ce7e887d4..b83eccd87 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/knowledgebase/KnowledgeBaseServiceImpl.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/knowledgebase/KnowledgeBaseServiceImpl.java @@ -95,12 +95,14 @@ public class KnowledgeBaseServiceImpl implements KnowledgeBaseService { knowledgeDocumentsMapper.insertOrUpdate(knowledgeDocumentsDO); } ); + // 4.5 异步处理知识库外挂 + asyncKnowledgeBase.knowledgeEmbed(knowledgeDocumentsList, updateReqVO.getId()); List deleteIds = knowledgeDocumentsMapper.selectDeleteIds(updateReqVO.getId()); asyncKnowledgeBase.createKnowledgeBase(knowledgeDocumentsList, deleteIds); updateObj.setKnowledgeBaseIntro("4"); knowledgeBaseMapper.updateById(updateObj); // 4.5 异步处理知识库外挂 - asyncKnowledgeBase.knowledgeEmbed(knowledgeDocumentsList, updateReqVO.getId()); +// asyncKnowledgeBase.knowledgeEmbed(knowledgeDocumentsList, updateReqVO.getId()); } else { knowledgeDocumentsMapper.delete(new LambdaQueryWrapperX() .eq(KnowledgeDocumentsDO::getKnowledgeBaseId, updateReqVO.getId()));