fix: fix xunfei crash (#451)

songquanpeng · Aug 20, 2023 · efeb9a1 · efeb9a1
1 parent 05e4f2b
commit efeb9a1
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/controller/relay-text.go b/controller/relay-text.go
@@ -315,11 +315,11 @@ func relayTextHelper(c *gin.Context, relayMode int) *OpenAIErrorWithStatusCode {
 			return errorWrapper(err, "close_request_body_failed", http.StatusInternalServerError)
 		}
 		isStream = isStream || strings.HasPrefix(resp.Header.Get("Content-Type"), "text/event-stream")
-	}
 
-	if resp.StatusCode != http.StatusOK {
-		return errorWrapper(
-			fmt.Errorf("bad status code: %d", resp.StatusCode), "bad_status_code", resp.StatusCode)
+		if resp.StatusCode != http.StatusOK {
+			return errorWrapper(
+				fmt.Errorf("bad status code: %d", resp.StatusCode), "bad_status_code", resp.StatusCode)
+		}
 	}
 
 	var textResponse TextResponse