fix: fix claude 3 request missing the 'max_token' field

This commit is contained in:
CaIon 2024-03-08 21:16:12 +08:00
parent 37113c0e96
commit f43b268520

View File

@ -65,6 +65,9 @@ func requestOpenAI2ClaudeMessage(textRequest dto.GeneralOpenAIRequest) (*ClaudeR
TopP: textRequest.TopP, TopP: textRequest.TopP,
Stream: textRequest.Stream, Stream: textRequest.Stream,
} }
if claudeRequest.MaxTokens == 0 {
claudeRequest.MaxTokens = 4096
}
claudeMessages := make([]ClaudeMessage, 0) claudeMessages := make([]ClaudeMessage, 0)
for _, message := range textRequest.Messages { for _, message := range textRequest.Messages {
if message.Role == "system" { if message.Role == "system" {