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