DATA_MIDDLE_PLATFORM/Data_Middle_Platform_Server
3
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

fix(llm): 修复向量库返回结果解析异常

Browse Source 
- 在解析向量库返回结果时增加异常捕获处理
- 当解析 JSON 格式出错时，记录错误日志并抛出 RuntimeException
- 优化了错误处理逻辑
This commit is contained in:
Liuyang 2025-02-12 17:21:10 +08:00
parent 671766700e
commit 764c9455f2
1 changed files with 15 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/RagHttpService.java
View File

@ -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 连接已关闭，尝试重新上传...");