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

fix: update MaxCompletionTokens for model prefix handling in buildTestRequest function

Browse Source
This commit is contained in:
CalciumIon
2024-12-25 17:55:20 +08:00
parent ba56e2e8ca
commit ed4d26fc9e
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
controller/channel-test.go
View File

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