fix:maxTokens

2025-05-04 02:52:04 +08:00
parent 9c604460b1
commit 6b2d78fe56
1 changed files with 3 additions and 1 deletions
--- a/llm/claude/v2/chat.go
+++ b/llm/claude/v2/chat.go
@@ -94,7 +94,9 @@ func (c *Claude) Chat(ctx context.Context, chatReq llm.ChatRequest) (*llm.ChatRe
 	if chatReq.MaxTokens > 0 {
 		maxTokens = chatReq.MaxTokens
 	} else {
-		if strings.Contains(chatReq.Model, "sonnet") || strings.Contains(chatReq.Model, "haiku") {
+		if strings.Contains(chatReq.Model, "3-7") {
+			maxTokens = 64000
+		} else if strings.Contains(chatReq.Model, "3-5") {
 			maxTokens = 8192
 		} else {
 			maxTokens = 4096