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

fix: gemini test MaxTokens

Browse Source
This commit is contained in:
zkp
2025-04-06 23:23:53 +08:00
parent f6e8887482
commit f24de65626
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
controller/channel-test.go
View File

@@ -192,6 +192,8 @@ func buildTestRequest(model string) *dto.GeneralOpenAIRequest {
if !strings.Contains(model, "claude") { if !strings.Contains(model, "claude") {
testRequest.MaxTokens = 50 testRequest.MaxTokens = 50
} }
} else if strings.Contains(model, "gemini") {
testRequest.MaxTokens = 500
} else { } else {
testRequest.MaxTokens = 10 testRequest.MaxTokens = 10
} }