From 90abe7f27d6e35ae28befae5a382ec81caa5c789 Mon Sep 17 00:00:00 2001
From: CalciumIon <1808837298@qq.com>
Date: Mon, 8 Jul 2024 19:50:12 +0800
Subject: [PATCH] fix: baidu max_output_tokens (#353)

---
 relay/channel/baidu/relay-baidu.go | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/relay/channel/baidu/relay-baidu.go b/relay/channel/baidu/relay-baidu.go
index e313316..918108c 100644
--- 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 {