From 21804b7e2f82d0ff814d6971c926f54d10efbc65 Mon Sep 17 00:00:00 2001
From: JustSong <songquanpeng@foxmail.com>
Date: Tue, 1 Aug 2023 23:50:45 +0800
Subject: [PATCH] chore: update implementation

---
 controller/relay-zhipu.go | 17 +++++++++--------
 1 file changed, 9 insertions(+), 8 deletions(-)

diff --git a/controller/relay-zhipu.go b/controller/relay-zhipu.go
index 79f8106c..20a4fa42 100644
--- a/controller/relay-zhipu.go
+++ b/controller/relay-zhipu.go
@@ -210,15 +210,16 @@ func zhipuStreamHandler(c *gin.Context, resp *http.Response) (*OpenAIErrorWithSt
 			data := scanner.Text()
 			lines := strings.Split(data, "\n")
 			for i, line := range lines {
-				if len(line) >= 5 {
-					if line[:5] == "data:" {
-						dataChan <- line[5:]
-						if i != len(lines)-1 {
-							dataChan <- "\n"
-						}
-					} else if line[:5] == "meta:" {
-						metaChan <- line[5:]
+				if len(line) < 5 {
+					continue
+				}
+				if line[:5] == "data:" {
+					dataChan <- line[5:]
+					if i != len(lines)-1 {
+						dataChan <- "\n"
 					}
+				} else if line[:5] == "meta:" {
+					metaChan <- line[5:]
 				}
 			}
 		}