Просмотр исходного кода

fix: ignore data if not have proper prefix

JustSong 2 лет назад
Родитель
Сommit
8cbbeb784f
1 измененных файлов с 3 добавлено и 0 удалено
  1. 3 0
      controller/relay-openai.go

+ 3 - 0
controller/relay-openai.go

@@ -34,6 +34,9 @@ func openaiStreamHandler(c *gin.Context, resp *http.Response, relayMode int) (*O
 			if len(data) < 6 { // ignore blank line or wrong format
 			if len(data) < 6 { // ignore blank line or wrong format
 				continue
 				continue
 			}
 			}
+			if data[:6] != "data: " && data[:6] != "[DONE]" {
+				continue
+			}
 			dataChan <- data
 			dataChan <- data
 			data = data[6:]
 			data = data[6:]
 			if !strings.HasPrefix(data, "[DONE]") {
 			if !strings.HasPrefix(data, "[DONE]") {