Просмотр исходного кода

fix: fix stream mode determine related logic (close #360)

JustSong 2 лет назад
Родитель
Сommit
d96cf2e84d
1 измененных файлов с 1 добавлено и 1 удалено
  1. 1 1
      controller/relay-text.go

+ 1 - 1
controller/relay-text.go

@@ -302,7 +302,7 @@ func relayTextHelper(c *gin.Context, relayMode int) *OpenAIErrorWithStatusCode {
 		if err != nil {
 		if err != nil {
 			return errorWrapper(err, "close_request_body_failed", http.StatusInternalServerError)
 			return errorWrapper(err, "close_request_body_failed", http.StatusInternalServerError)
 		}
 		}
-		isStream = strings.HasPrefix(resp.Header.Get("Content-Type"), "text/event-stream")
+		isStream = isStream || strings.HasPrefix(resp.Header.Get("Content-Type"), "text/event-stream")
 	}
 	}
 
 
 	var textResponse TextResponse
 	var textResponse TextResponse