فهرست منبع

fix: fix sse not ending properly in some case

JustSong 2 سال پیش
والد
کامیت
c9d2e42a9e
1فایلهای تغییر یافته به همراه2 افزوده شده و 2 حذف شده
  1. 2 2
      controller/relay-openai.go

+ 2 - 2
controller/relay-openai.go

@@ -46,7 +46,7 @@ func openaiStreamHandler(c *gin.Context, resp *http.Response, relayMode int) (*O
 					err := json.Unmarshal([]byte(data), &streamResponse)
 					err := json.Unmarshal([]byte(data), &streamResponse)
 					if err != nil {
 					if err != nil {
 						common.SysError("error unmarshalling stream response: " + err.Error())
 						common.SysError("error unmarshalling stream response: " + err.Error())
-						return
+						continue // just ignore the error
 					}
 					}
 					for _, choice := range streamResponse.Choices {
 					for _, choice := range streamResponse.Choices {
 						responseText += choice.Delta.Content
 						responseText += choice.Delta.Content
@@ -56,7 +56,7 @@ func openaiStreamHandler(c *gin.Context, resp *http.Response, relayMode int) (*O
 					err := json.Unmarshal([]byte(data), &streamResponse)
 					err := json.Unmarshal([]byte(data), &streamResponse)
 					if err != nil {
 					if err != nil {
 						common.SysError("error unmarshalling stream response: " + err.Error())
 						common.SysError("error unmarshalling stream response: " + err.Error())
-						return
+						continue
 					}
 					}
 					for _, choice := range streamResponse.Choices {
 					for _, choice := range streamResponse.Choices {
 						responseText += choice.Text
 						responseText += choice.Text