fix: fix max_tokens check

songquanpeng · songquanpeng · commit 67c64e71c800 · 2023-12-20T21:45:33.000+08:00
diff --git a/controller/relay-text.go b/controller/relay-text.go
@@ -58,6 +58,9 @@ func relayTextHelper(c *gin.Context, relayMode int) *OpenAIErrorWithStatusCode {
 	if err != nil {
 		return errorWrapper(err, "bind_request_body_failed", http.StatusBadRequest)
 	}
+	if textRequest.MaxTokens < 0 || textRequest.MaxTokens > math.MaxInt32/2 {
+		return errorWrapper(errors.New("max_tokens is invalid"), "invalid_max_tokens", http.StatusBadRequest)
+	}
 	if relayMode == RelayModeModerations && textRequest.Model == "" {
 		textRequest.Model = "text-moderation-latest"
 	}

Original file line number	Diff line number	Diff line change
`@@ -58,6 +58,9 @@ func relayTextHelper(c gin.Context, relayMode int) OpenAIErrorWithStatusCode {`
`58`	`58`	`if err != nil {`
`59`	`59`	`return errorWrapper(err, "bind_request_body_failed", http.StatusBadRequest)`
`60`	`60`	`}`
	`61`	`+ if textRequest.MaxTokens < 0 \|\| textRequest.MaxTokens > math.MaxInt32/2 {`
	`62`	`+ return errorWrapper(errors.New("max_tokens is invalid"), "invalid_max_tokens", http.StatusBadRequest)`
	`63`	`+ }`
`61`	`64`	`if relayMode == RelayModeModerations && textRequest.Model == "" {`
`62`	`65`	`textRequest.Model = "text-moderation-latest"`
`63`	`66`	`}`