xieyinlu/ruoyi-ai
1
0
Fork 0
mirror of https://gitcode.com/ageerle/ruoyi-ai.git synced 2026-03-14 05:03:43 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #254 from StevenJack666/v3.0.0

Browse Source 
修复异常场景，请求链接无法关闭问题
This commit is contained in:
ageerle
2026-02-11 17:03:44 +08:00
committed by GitHub
parent c78b456dbf 91a44e1ba8
commit afa3c78180
1 changed files with 17 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/service/chat/impl/AbstractStreamingChatService.java
View File

@@ -98,6 +98,8 @@ public abstract class AbstractStreamingChatService implements IChatService {
}
} catch (Exception e) {
SseMessageUtils.sendMessage(userId, "对话出错：" + e.getMessage());
SseMessageUtils.completeConnection(userId, tokenValue);
log.error("{}请求失败：{}", getProviderName(), e.getMessage(), e);
}
return emitter;
@@ -223,6 +225,21 @@ public abstract class AbstractStreamingChatService implements IChatService {
@Override
public void onError(Throwable error) {
log.error("{}流式响应错误: {}", getProviderName(), error.getMessage(), error);
// 发送错误消息到前端
try {
String errorMessage = String.format("模型调用失败: %s", error.getMessage());
SseMessageUtils.sendMessage(userId, errorMessage);
} catch (Exception e) {
log.error("发送错误消息失败: {}", e.getMessage(), e);
}
// 关闭SSE连接，避免前端一直等待
try {
SseMessageUtils.completeConnection(userId, tokenValue);
} catch (Exception e) {
log.error("关闭SSE连接失败: {}", e.getMessage(), e);
}
}
};
}