fix: update MaxCompletionTokens for model prefix handling in buildTestRequest function

This commit is contained in:
CalciumIon
2024-12-25 17:55:20 +08:00
parent ba56e2e8ca
commit ed4d26fc9e

View File

@@ -152,8 +152,8 @@ func buildTestRequest(model string) *dto.GeneralOpenAIRequest {
Model: "", // this will be set later
Stream: false,
}
if strings.HasPrefix(model, "o1-") {
testRequest.MaxCompletionTokens = 1
if strings.HasPrefix(model, "o1") {
testRequest.MaxCompletionTokens = 10
} else if strings.HasPrefix(model, "gemini-2.0-flash-thinking") {
testRequest.MaxTokens = 2
} else {