Parcourir la source

fix: enhance data trimming and validation in stream scanner

CaIon il y a 1 semaine
Parent
commit
0835e15091
1 fichiers modifiés avec 4 ajouts et 2 suppressions
  1. 4 2
      relay/helper/stream_scanner.go

+ 4 - 2
relay/helper/stream_scanner.go

@@ -215,8 +215,10 @@ func StreamScannerHandler(c *gin.Context, resp *http.Response, info *relaycommon
 				continue
 			}
 			data = data[5:]
-			data = strings.TrimLeft(data, " ")
-			data = strings.TrimSuffix(data, "\r")
+			data = strings.TrimSpace(data)
+			if data == "" {
+				continue
+			}
 			if !strings.HasPrefix(data, "[DONE]") {
 				info.SetFirstResponseTime()
 				info.ReceivedResponseCount++