From 2663fbc51a17539e8d0381b60a4efb956b18cdd0 Mon Sep 17 00:00:00 2001
From: JustSong <songquanpeng@foxmail.com>
Date: Sun, 30 Jul 2023 12:03:06 +0800
Subject: [PATCH] fix: ignore data if not have proper prefix

---
 controller/relay-openai.go | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/controller/relay-openai.go b/controller/relay-openai.go
index 2bcb99dc..b986822c 100644
--- a/controller/relay-openai.go
+++ b/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
 				continue
 			}
+			if data[:6] != "data: " && data[:6] != "[DONE]" {
+				continue
+			}
 			dataChan <- data
 			data = data[6:]
 			if !strings.HasPrefix(data, "[DONE]") {