|
|
@@ -192,6 +192,8 @@ func buildTestRequest(model string) *dto.GeneralOpenAIRequest {
|
|
|
if !strings.Contains(model, "claude") {
|
|
|
testRequest.MaxTokens = 50
|
|
|
}
|
|
|
+ } else if strings.Contains(model, "gemini") {
|
|
|
+ testRequest.MaxTokens = 500
|
|
|
} else {
|
|
|
testRequest.MaxTokens = 10
|
|
|
}
|