From 61912f5e2c454b2f34e442c359daa7587043d283 Mon Sep 17 00:00:00 2001
From: ckt1031 <65409152+ckt1031@users.noreply.github.com>
Date: Thu, 13 Jul 2023 19:40:36 +0800
Subject: [PATCH] fix: patch testing

---
 controller/channel-test.go | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/controller/channel-test.go b/controller/channel-test.go
index 867cacd3..7f3a3025 100644
--- a/controller/channel-test.go
+++ b/controller/channel-test.go
@@ -58,7 +58,8 @@ func testChannel(channel *model.Channel, request ChatRequest) error {
 		return errors.New("invalid status code: " + strconv.Itoa(resp.StatusCode))
 	}
 
-	var streamResponseText string
+	var done = false
+	var streamResponseText = ""
 
 	scanner := bufio.NewScanner(resp.Body)
 	scanner.Split(func(data []byte, atEOF bool) (advance int, token []byte, err error) {
@@ -93,13 +94,16 @@ func testChannel(channel *model.Channel, request ChatRequest) error {
 			for _, choice := range streamResponse.Choices {
 				streamResponseText += choice.Delta.Content
 			}
+		} else {
+			done = true
+			break
 		}
 	}
 
 	defer resp.Body.Close()
 
 	// Check if streaming is complete and streamResponseText is populated
-	if streamResponseText == "" {
+	if streamResponseText == "" || !done {
 		return errors.New("Streaming not complete")
 	}