xinyin025/new-api
1
0
Fork 0
mirror of https://github.com/linux-do/new-api.git synced 2025-09-18 00:16:37 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: baidu max_output_tokens (#353)

Browse Source
This commit is contained in:
CalciumIon 2024-07-08 19:50:12 +08:00
parent bb313eb26f
commit 90abe7f27d
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
relay/channel/baidu/relay-baidu.go
View File

@ -33,6 +33,9 @@ func requestOpenAI2Baidu(request dto.GeneralOpenAIRequest) *BaiduChatRequest {
} }
if request.MaxTokens != 0 { if request.MaxTokens != 0 {
maxTokens := int(request.MaxTokens) maxTokens := int(request.MaxTokens)
if request.MaxTokens == 1 {
maxTokens = 2
}
baiduRequest.MaxOutputTokens = &maxTokens baiduRequest.MaxOutputTokens = &maxTokens
} }
for _, message := range request.Messages { for _, message := range request.Messages {