fix: close connection when response ended

This commit is contained in:
JustSong
2023-07-29 23:52:18 +08:00
parent 9e8c40ade4
commit bd80f77f2b

View File

@@ -205,6 +205,10 @@ func xunfeiStreamHandler(c *gin.Context, textRequest GeneralOpenAIRequest, appId
}
dataChan <- response
if response.Payload.Choices.Status == 2 {
err := conn.Close()
if err != nil {
common.SysError("error closing websocket connection: " + err.Error())
}
break
}
}