Browse Source

yinruoxi

修改:
    1.重构websocketClient
tags/B.2.6.7_20240112_base
kira 1 year ago
parent
commit
206bde413f
1 changed files with 13 additions and 1 deletions
  1. +13
    -1
      xueyi-modules/xueyi-nlt/src/main/java/com/xueyi/nlt/netty/client/listener/LlmWebSocketListener.java

+ 13
- 1
xueyi-modules/xueyi-nlt/src/main/java/com/xueyi/nlt/netty/client/listener/LlmWebSocketListener.java View File

@@ -193,7 +193,19 @@ public class LlmWebSocketListener extends WebSocketListener {
//关闭释放资源
this.wsCloseFlag = true;
// 如果是非流式调用,释放锁
if (!stream) {
if (stream) {
// 讯飞返回status为2时,可能不包含。结尾,所以需要判断answerBuf是否为空,如果不为空,则直接发送给前端
Channel ch = ServerConfig.sessionMap.get(llmContext.getDevId());
logger.info("当前ch:{}",ch.id().asLongText());
if (ch != null) {
if (!StringUtils.isEmpty(answerBuf)) {
JSONObject jo = formatToChannel(answerBuf, responseData.header.status, responseData.header.code);
logger.info("发送到client:{},id:{},内容:{}", llmContext.getDevId(), ch.id().asLongText(), jo.toJSONString());
ch.writeAndFlush(new TextWebSocketFrame(jo.toJSONString()));
}
}
}
else {
this.notifyAll();
}



Loading…
Cancel
Save