Update MaxTokens for gemini model to 300 in test request

This commit is contained in:
CaIon 2025-04-08 17:37:25 +08:00
parent ba9cd9314b
commit c40404a8c1

View File

@ -193,7 +193,7 @@ func buildTestRequest(model string) *dto.GeneralOpenAIRequest {
testRequest.MaxTokens = 50
}
} else if strings.Contains(model, "gemini") {
testRequest.MaxTokens = 50
testRequest.MaxTokens = 300
} else {
testRequest.MaxTokens = 10
}