fix: baidu max_output_tokens (#353)

52b0c57a · CalciumIon · 19ba8525 · 52b0c57a
Commit 52b0c57a authored Jul 08, 2024 by CalciumIon
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 0 deletions

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

No files found.
--- a/relay/channel/baidu/relay-baidu.go
+++ b/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 {