Skip to content

phsl / new-api

Go to a project
Commit ba9cd931 by Calcium-Ion Committed by GitHub
Options

Merge pull request #936 from lamcodes/main 

fix: gemini test MaxTokens
parents d85e25fc 950572b0
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 = 50
} 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