GithubProxy/coai
1
0
Fork 0
mirror of https://github.com/coaidev/coai.git synced 2025-05-19 04:50:14 +09:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: fix max_tokens

Browse Source
This commit is contained in:
George 2024-10-14 17:05:02 +08:00
parent 7fa53af2af
commit 1ba6faaa79
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
adapter/sparkdesk/chat.go
View File

@ -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