Skip to content

phsl / new-api

Go to a project
Commit a48679fb by CalciumIon
Options

fix: update MaxCompletionTokens for model prefix handling in buildTestRequest function

parent 314ad072
Showing with 2 additions and 2 deletions
controller/channel-test.go
...@@ -152,8 +152,8 @@ func buildTestRequest(model string) *dto.GeneralOpenAIRequest { ...@@ -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 {
......
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