From f43b268520d40bfb9b20de90e5e6319e81b970b0 Mon Sep 17 00:00:00 2001
From: CaIon <1808837298@qq.com>
Date: Fri, 8 Mar 2024 21:16:12 +0800
Subject: [PATCH] fix: fix claude 3 request missing the 'max_token' field

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

diff --git a/relay/channel/claude/relay-claude.go b/relay/channel/claude/relay-claude.go
index f43dfca..db29fe8 100644
--- a/relay/channel/claude/relay-claude.go
+++ b/relay/channel/claude/relay-claude.go
@@ -65,6 +65,9 @@ func requestOpenAI2ClaudeMessage(textRequest dto.GeneralOpenAIRequest) (*ClaudeR
 		TopP:          textRequest.TopP,
 		Stream:        textRequest.Stream,
 	}
+	if claudeRequest.MaxTokens == 0 {
+		claudeRequest.MaxTokens = 4096
+	}
 	claudeMessages := make([]ClaudeMessage, 0)
 	for _, message := range textRequest.Messages {
 		if message.Role == "system" {