Commit 86d93c46 by 1808837298@qq.com

fix: adjust max tokens configuration in test request builder

- Update max tokens default value to 10
parent 93511cfb
...@@ -158,10 +158,8 @@ func buildTestRequest(model string) *dto.GeneralOpenAIRequest { ...@@ -158,10 +158,8 @@ func buildTestRequest(model string) *dto.GeneralOpenAIRequest {
} }
if strings.HasPrefix(model, "o1") || strings.HasPrefix(model, "o3") { if strings.HasPrefix(model, "o1") || strings.HasPrefix(model, "o3") {
testRequest.MaxCompletionTokens = 10 testRequest.MaxCompletionTokens = 10
} else if strings.HasPrefix(model, "gemini-2.0-flash-thinking") {
testRequest.MaxTokens = 10
} else { } else {
testRequest.MaxTokens = 1 testRequest.MaxTokens = 10
} }
content, _ := json.Marshal("hi") content, _ := json.Marshal("hi")
testMessage := dto.Message{ testMessage := dto.Message{
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or sign in to comment