Skip to content

phsl / new-api

Go to a project
Commit 93d9f141 by zkp
Options

fix: gemini test MaxTokens

parent 69d790b4
Showing with 2 additions and 0 deletions
controller/channel-test.go
...@@ -192,6 +192,8 @@ func buildTestRequest(model string) *dto.GeneralOpenAIRequest { ...@@ -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
} }
......
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