xinyin025/new-api
1
0
Fork 0
mirror of https://github.com/linux-do/new-api.git synced 2025-09-17 16:06:38 +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"`
} }

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

@ -23,14 +23,17 @@ var baiduTokenStore sync.Map
func requestOpenAI2Baidu(request dto.GeneralOpenAIRequest) *BaiduChatRequest { func requestOpenAI2Baidu(request dto.GeneralOpenAIRequest) *BaiduChatRequest {
baiduRequest := BaiduChatRequest{ baiduRequest := BaiduChatRequest{
Temperature: request.Temperature, Temperature: request.Temperature,
TopP: request.TopP, TopP: request.TopP,
PenaltyScore: request.FrequencyPenalty, PenaltyScore: request.FrequencyPenalty,
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" {