Преглед на файлове

fix: baidu max_output_tokens (#353)

CalciumIon преди 1 година
родител
ревизия
90abe7f27d
променени са 1 файла, в които са добавени 3 реда и са изтрити 0 реда
  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 {