瀏覽代碼

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 {