Scott
66ee6b8f98
* fix: preserve qwen thinking budget * test: address qwen thinking budget review comments * chore: remove unreachable adaptor code * test: cover zero Qwen thinking budgets
29 lines
693 B
Go
29 lines
693 B
Go
package ali
|
|
|
|
import (
|
|
"github.com/QuantumNous/new-api/relaykit/dto"
|
|
"github.com/samber/lo"
|
|
)
|
|
|
|
// https://help.aliyun.com/document_detail/613695.html?spm=a2c4g.2399480.0.0.1adb778fAdzP9w#341800c0f8w0r
|
|
|
|
const EnableSearchModelSuffix = "-internet"
|
|
|
|
func requestOpenAI2Ali(request dto.GeneralOpenAIRequest, upstreamModelName string) *dto.GeneralOpenAIRequest {
|
|
modelName := upstreamModelName
|
|
if modelName == "" {
|
|
modelName = request.Model
|
|
}
|
|
if !dto.IsQwenThinkingBudgetModel(modelName) {
|
|
request.ThinkingBudget = nil
|
|
}
|
|
|
|
topP := lo.FromPtrOr(request.TopP, 0)
|
|
if topP >= 1 {
|
|
request.TopP = lo.ToPtr(0.999)
|
|
} else if topP <= 0 {
|
|
request.TopP = lo.ToPtr(0.001)
|
|
}
|
|
return &request
|
|
}
|