oadmin/new-api
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: Adjust MaxTokens logic for non-Claude models in test request

Browse Source
This commit is contained in:
1808837298@qq.com
2025-03-17 23:44:32 +08:00
parent 7143b0f160
commit 73263e02d6
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
controller/channel-test.go
View File

@@ -187,7 +187,9 @@ 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.Contains(model, "thinking") { } else if strings.Contains(model, "thinking") {
testRequest.MaxTokens = 50 if !strings.Contains(model, "claude") {
testRequest.MaxTokens = 50
}
} else { } else {
testRequest.MaxTokens = 10 testRequest.MaxTokens = 10
} }