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 (close #353)

Browse Source
This commit is contained in:
CalciumIon 2024-07-08 19:46:45 +08:00
parent 49cec50908
commit 02545e4856
2 changed files with 12 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
relay/channel/baidu/dto.go
View File

@ -19,7 +19,7 @@ type BaiduChatRequest struct {
System string `json:"system,omitempty"` System string `json:"system,omitempty"`
DisableSearch bool `json:"disable_search,omitempty"` DisableSearch bool `json:"disable_search,omitempty"`
EnableCitation bool `json:"enable_citation,omitempty"` EnableCitation bool `json:"enable_citation,omitempty"`
MaxOutputTokens int `json:"max_output_tokens,omitempty"` MaxOutputTokens *int `json:"max_output_tokens,omitempty"`
UserId string `json:"user_id,omitempty"` UserId string `json:"user_id,omitempty"`
} }

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

@ -29,9 +29,12 @@ func requestOpenAI2Baidu(request dto.GeneralOpenAIRequest) *BaiduChatRequest {
Stream: request.Stream, Stream: request.Stream,
DisableSearch: false, DisableSearch: false,
EnableCitation: false, EnableCitation: false,
MaxOutputTokens: int(request.MaxTokens),
UserId: request.User, UserId: request.User,
} }
if request.MaxTokens != 0 {
maxTokens := int(request.MaxTokens)
baiduRequest.MaxOutputTokens = &maxTokens
}
for _, message := range request.Messages { for _, message := range request.Messages {
if message.Role == "system" { if message.Role == "system" {
baiduRequest.System = message.StringContent() baiduRequest.System = message.StringContent()