Skip to content

Commit 1ba6faa

Browse files
committed
fix: fix max_tokens
1 parent 7fa53af commit 1ba6faa

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

adapter/sparkdesk/chat.go

+4-4
Original file line numberDiff line numberDiff line change
@@ -20,13 +20,13 @@ func GetToken(props *adaptercommon.ChatProps) *int {
2020

2121
switch props.Model {
2222
case globals.SparkDeskLite, globals.SparkDeskPro128K:
23-
if *props.MaxTokens > 8192 {
24-
return utils.ToPtr(8192)
25-
}
26-
case globals.SparkDeskPro, globals.SparkDeskMax, globals.SparkDeskMax32K, globals.SparkDeskV4Ultra:
2723
if *props.MaxTokens > 4096 {
2824
return utils.ToPtr(4096)
2925
}
26+
case globals.SparkDeskPro, globals.SparkDeskMax, globals.SparkDeskMax32K, globals.SparkDeskV4Ultra:
27+
if *props.MaxTokens > 8192 {
28+
return utils.ToPtr(8192)
29+
}
3030
}
3131

3232
return props.MaxTokens

0 commit comments

Comments
 (0)