Просмотр исходного кода

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

CaIon 2 лет назад
Родитель
Сommit
f43b268520
1 измененных файлов с 3 добавлено и 0 удалено
  1. 3 0
      relay/channel/claude/relay-claude.go

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

@@ -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" {