From 1ba6faaa79ecedc1f897a65b14fb38d192ec9bd6 Mon Sep 17 00:00:00 2001 From: George Date: Mon, 14 Oct 2024 17:05:02 +0800 Subject: [PATCH] fix: fix max_tokens --- adapter/sparkdesk/chat.go | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/adapter/sparkdesk/chat.go b/adapter/sparkdesk/chat.go index b19d4da..5316ea5 100644 --- 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