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 03e8ab4126
commit 79b35e385f

View File

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