diff --git a/controller/channel-test.go b/controller/channel-test.go
index 84197dfd..4ec00e1e 100644
--- a/controller/channel-test.go
+++ b/controller/channel-test.go
@@ -99,10 +99,6 @@ func testChannel(channel *model.Channel, request ChatRequest) error {
 				}
 			}
 		}
-
-		if streamResponseText == "" {
-			return errors.New("empty stream response")
-		}
 	} else {
 		body, err := ioutil.ReadAll(resp.Body)
 		if err != nil {
@@ -121,6 +117,11 @@ func testChannel(channel *model.Channel, request ChatRequest) error {
 
 	defer resp.Body.Close()
 
+	// Check if streaming is complete and streamResponseText is populated
+	if isStream && streamResponseText == "" {
+		return errors.New("Streaming not complete")
+	}
+
 	return nil
 }
 
@@ -128,6 +129,7 @@ func buildTestRequest() *ChatRequest {
 	testRequest := &ChatRequest{
 		Model:     "", // this will be set later
 		MaxTokens: 1,
+		Stream:    false,
 	}
 	testMessage := Message{
 		Role:    "user",
diff --git a/controller/relay.go b/controller/relay.go
index 81850925..3b37910e 100644
--- a/controller/relay.go
+++ b/controller/relay.go
@@ -45,6 +45,7 @@ type ChatRequest struct {
 	Model     string    `json:"model"`
 	Messages  []Message `json:"messages"`
 	MaxTokens int       `json:"max_tokens"`
+	Stream    bool      `json:"stream"`
 }
 
 type TextRequest struct {