From 764c9455f2d77c7fbdafc43279def03f3855cc7b Mon Sep 17 00:00:00 2001 From: Liuyang <2746366019@qq.com> Date: Wed, 12 Feb 2025 17:21:10 +0800 Subject: [PATCH] =?UTF-8?q?fix(llm):=20=E4=BF=AE=E5=A4=8D=E5=90=91?= =?UTF-8?q?=E9=87=8F=E5=BA=93=E8=BF=94=E5=9B=9E=E7=BB=93=E6=9E=9C=E8=A7=A3?= =?UTF-8?q?=E6=9E=90=E5=BC=82=E5=B8=B8?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - 在解析向量库返回结果时增加异常捕获处理 - 当解析 JSON 格式出错时，记录错误日志并抛出 RuntimeException - 优化了错误处理逻辑 --- .../llm/service/http/RagHttpService.java | 24 ++++++++++++------- 1 file changed, 15 insertions(+), 9 deletions(-) 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 8c2db4cae..4f5e00c1e 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 @@ -310,18 +310,24 @@ public class RagHttpService { // 检查响应状态 String responseBody = response.getBody(); log.info(" ========= String Response Body Result: {}", responseBody); - JSONObject parseObject = JSON.parseObject(responseBody); - log.info(" ========= JSON Response Body Result: {}", responseBody); + try { + JSONObject parseObject = JSON.parseObject(responseBody); + log.info(" ========= JSON Response Body Result: {}", responseBody); - if (parseObject.containsKey("status") && parseObject.getBoolean("status")) { - // 修改状态为 上传成功 - updateFileState(documents, KnowledgeStatusEnum.UPLOAD_SUCCESS); - } else { - // 修改状态为 上传失败 - updateFileState(documents, KnowledgeStatusEnum.UPLOAD_FAILED); - throw exception(new ErrorCode(10047, responseBody)); + if (parseObject.containsKey("status") && parseObject.getBoolean("status")) { + // 修改状态为 上传成功 + updateFileState(documents, KnowledgeStatusEnum.UPLOAD_SUCCESS); + } else { + // 修改状态为 上传失败 + updateFileState(documents, KnowledgeStatusEnum.UPLOAD_FAILED); + throw exception(new ErrorCode(10047, responseBody)); + } + } catch (Exception e) { + log.error("返回结果 解析为 JSON格式错误: {}", e.getMessage()); + throw new RuntimeException("返回结果 解析为 JSON格式错误: " + e.getMessage(), e); } + } catch (UnirestException e) { if (isSocketClosedException(e)) { log.warn("knowledgeEmbed Socket 连接已关闭，尝试重新上传...");