Update MaxTokens for gemini model to 300 in test request
This commit is contained in:
@@ -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
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user