fix: fix max_tokens

vdsmitnov52 · Oct 14, 2024 · 1ba6faa · 1ba6faa
1 parent 7fa53af
commit 1ba6faa
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/adapter/sparkdesk/chat.go b/adapter/sparkdesk/chat.go
@@ -20,13 +20,13 @@ func GetToken(props *adaptercommon.ChatProps) *int {
 
 	switch props.Model {
 	case globals.SparkDeskLite, globals.SparkDeskPro128K:
-		if *props.MaxTokens > 8192 {
-			return utils.ToPtr(8192)
-		}
-	case globals.SparkDeskPro, globals.SparkDeskMax, globals.SparkDeskMax32K, globals.SparkDeskV4Ultra:
 		if *props.MaxTokens > 4096 {
 			return utils.ToPtr(4096)
 		}
+	case globals.SparkDeskPro, globals.SparkDeskMax, globals.SparkDeskMax32K, globals.SparkDeskV4Ultra:
+		if *props.MaxTokens > 8192 {
+			return utils.ToPtr(8192)
+		}
 	}
 
 	return props.MaxTokens