Pārlūkot izejas kodu

fix: baidu max_output_tokens (#353)

CalciumIon 1 gadu atpakaļ
vecāks
revīzija
90abe7f27d
1 mainītis faili ar 3 papildinājumiem un 0 dzēšanām
  1. 3 0
      relay/channel/baidu/relay-baidu.go

+ 3 - 0
relay/channel/baidu/relay-baidu.go

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