From 671766700e9e603268fa9afec07323a2fbe58012 Mon Sep 17 00:00:00 2001 From: Liuyang <2746366019@qq.com> Date: Wed, 12 Feb 2025 17:16:00 +0800 Subject: [PATCH] =?UTF-8?q?refactor(module-llm):=E4=BC=98=E5=8C=96?= =?UTF-8?q?=E6=97=A5=E5=BF=97=E6=89=93=E5=8D=B0=E5=92=8C=E5=BC=82=E5=B8=B8?= =?UTF-8?q?=E5=A4=84=E7=90=86-=20=E4=BF=AE=E6=94=B9=E4=BA=86=20AsyncKnowle?= =?UTF-8?q?dgeBase=20=E7=B1=BB=E4=B8=AD=E7=9A=84=E6=97=A5=E5=BF=97?= =?UTF-8?q?=E6=89=93=E5=8D=B0=E6=A0=BC=E5=BC=8F=20-=20=E7=A7=BB=E9=99=A4?= =?UTF-8?q?=E4=BA=86=20RagHttpService=20=E7=B1=BB=E4=B8=AD=E4=B8=8D?= =?UTF-8?q?=E5=BF=85=E8=A6=81=E7=9A=84=20Unirest.shutDown()=E8=B0=83?= =?UTF-8?q?=E7=94=A8=20-=20=E5=A2=9E=E5=8A=A0=E4=BA=86=20JSON=20=E8=A7=A3?= =?UTF-8?q?=E6=9E=90=E5=89=8D=E7=9A=84=E5=AD=97=E7=AC=A6=E4=B8=B2=E5=93=8D?= =?UTF-8?q?=E5=BA=94=E4=BD=93=E6=97=A5=E5=BF=97=E6=89=93=E5=8D=B0=20-=20?= =?UTF-8?q?=E4=BC=98=E5=8C=96=E4=BA=86=E5=BC=82=E5=B8=B8=E5=A4=84=E7=90=86?= =?UTF-8?q?=EF=BC=8C=E7=A7=BB=E9=99=A4=E4=BA=86=E5=A4=9A=E4=BD=99=E7=9A=84?= =?UTF-8?q?=20finally=20=E5=9D=97?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../yudao/module/llm/service/async/AsyncKnowledgeBase.java | 2 +- .../yudao/module/llm/service/http/RagHttpService.java | 7 ++----- .../service/knowledgebase/KnowledgeBaseServiceImpl.java | 1 - 3 files changed, 3 insertions(+), 7 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 776a22eb5..1a88a940f 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 @@ -119,7 +119,7 @@ public class AsyncKnowledgeBase { ragHttpService.knowledgeEmbed(ragEmbedReqVo, id); } catch (Exception e) { - log.error("the creation of the knowledge base error {}", e.getMessage(), e); + log.error("the creation of the knowledge base error : {}", e.getMessage(), e); throw e; } } 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 23ccb08fc..8c2db4cae 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 @@ -157,8 +157,6 @@ public class RagHttpService { } else { throw new RuntimeException("文件上传失败: " + e.getMessage(), e); } - } finally { - Unirest.shutDown(); } } @@ -311,8 +309,9 @@ public class RagHttpService { // 检查响应状态 String responseBody = response.getBody(); + log.info(" ========= String Response Body Result: {}", responseBody); JSONObject parseObject = JSON.parseObject(responseBody); - log.info(" ========= Response Body Result: {}", responseBody); + log.info(" ========= JSON Response Body Result: {}", responseBody); if (parseObject.containsKey("status") && parseObject.getBoolean("status")) { // 修改状态为 上传成功 @@ -330,8 +329,6 @@ public class RagHttpService { } else { throw new RuntimeException("文件上传失败: " + e.getMessage(), e); } - } finally { - Unirest.shutDown(); } } 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 8c88bc40f..397fcfc76 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 @@ -114,7 +114,6 @@ public class KnowledgeBaseServiceImpl implements KnowledgeBaseService { // 4.4 异步处理新增文档和删除的文档 List deleteIds = knowledgeDocumentsMapper.selectDeleteIds(updateReqVO.getId()); - log.info("注释.."); asyncKnowledgeBase.createKnowledgeBase(newDocuments, deleteIds); } else { // 5. 如果传入的文档列表为空,则删除所有关联文档