xieyinlu/ruoyi-ai
1
0
Fork 0
mirror of https://gitcode.com/ageerle/ruoyi-ai.git synced 2026-04-14 20:33:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(小程序): 1. 修复小程序一直停留在加载页面 2. 修复小程序无法正常对话

Browse Source
This commit is contained in:
ageer
2025-03-28 22:25:32 +08:00
parent 065ff3bed0
commit bd6f8a9558
6 changed files with 30 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
ruoyi-common/ruoyi-common-chat/src/main/java/org/ruoyi/common/chat/handler/PlusWebSocketHandler.java
View File

@@ -61,7 +61,7 @@ public class PlusWebSocketHandler extends AbstractWebSocketHandler {
}
ChatCompletion chatCompletion = ChatCompletion
.builder()
.model(ChatCompletion.Model.GPT_3_5_TURBO.getName())
.model("gpt-4o-mini")
.messages(messages)
.temperature(0.2)
.stream(true)

19
ruoyi-common/ruoyi-common-chat/src/main/java/org/ruoyi/common/chat/listener/WebSocketEventListener.java
View File

@@ -9,6 +9,7 @@ import okhttp3.sse.EventSource;
import okhttp3.sse.EventSourceListener;
import org.ruoyi.common.chat.constant.OpenAIConst;
import org.ruoyi.common.chat.entity.chat.ChatCompletionResponse;
import org.ruoyi.common.chat.entity.chat.Message;
import org.springframework.web.socket.TextMessage;
import org.springframework.web.socket.WebSocketSession;
@@ -57,7 +58,12 @@ public class WebSocketEventListener extends EventSourceListener {
ObjectMapper mapper = new ObjectMapper();
// 读取Json
ChatCompletionResponse completionResponse = mapper.readValue(data, ChatCompletionResponse.class);
String delta = mapper.writeValueAsString(completionResponse.getChoices().get(0).getDelta());
String delta = "";
try {
delta = mapper.writeValueAsString(completionResponse.getChoices().get(0).getDelta());
}catch (Exception e){
log.error("转换失败{}",e.getMessage());
}
session.sendMessage(new TextMessage(delta));
}
@@ -75,18 +81,9 @@ public class WebSocketEventListener extends EventSourceListener {
return;
}
ResponseBody body = response.body();
if (Objects.nonNull(body)) {
// 返回非流式回复内容
if(response.code() == OpenAIConst.SUCCEED_CODE){
ObjectMapper mapper = new ObjectMapper();
ChatCompletionResponse completionResponse = mapper.readValue(body.string(), ChatCompletionResponse.class);
String delta = mapper.writeValueAsString(completionResponse.getChoices().get(0).getMessage().getContent());
session.sendMessage(new TextMessage(delta));
}else {
log.error("Socket连接异常data：{}，异常：{}", body.string(), t);
}
log.error("Socket连接异常data：{}，异常：{}", body.string(), t);
} else {
log.error("Socket连接异常data：{}，异常：{}", response, t);
}