|
|
@@ -158,10 +158,8 @@ func buildTestRequest(model string) *dto.GeneralOpenAIRequest {
|
|
|
}
|
|
|
if strings.HasPrefix(model, "o1") || strings.HasPrefix(model, "o3") {
|
|
|
testRequest.MaxCompletionTokens = 10
|
|
|
- } else if strings.HasPrefix(model, "gemini-2.0-flash-thinking") {
|
|
|
- testRequest.MaxTokens = 10
|
|
|
} else {
|
|
|
- testRequest.MaxTokens = 1
|
|
|
+ testRequest.MaxTokens = 10
|
|
|
}
|
|
|
content, _ := json.Marshal("hi")
|
|
|
testMessage := dto.Message{
|