GithubProxy/coai
1
0
Fork 0
mirror of https://github.com/coaidev/coai.git synced 2025-06-03 20:30:19 +09:00
Code Issues Actions Packages Projects Releases Wiki Activity

update sparkdesk prize

Browse Source
This commit is contained in:
Zhang Minghan 2023-10-01 11:41:46 +08:00
parent 91d650a383
commit f8cb3d99d8
2 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
auth/payment.go
View File

@ -78,6 +78,8 @@ func CanEnableModel(db *sql.DB, user *User, model string) bool {
return user != nil && user.GetQuota(db) >= 5
case globals.GPT432k, globals.GPT432k0613, globals.GPT432k0314:
return user != nil && user.GetQuota(db) >= 50
case globals.SparkDesk:
return user != nil && user.GetQuota(db) >= 1
default:
return true
}

7
utils/tokenizer.go
View File

@ -49,7 +49,8 @@ func GetWeightByModel(model string) int {
globals.GPT4,
globals.GPT40314,
globals.GPT40613:
globals.GPT40613,
globals.SparkDesk:
return 3
case globals.GPT3Turbo0301, globals.GPT3Turbo16k0301:
return 4 // every message follows <|start|>{role/name}\n{content}<|end|>\n
@ -103,6 +104,8 @@ func CountInputToken(model string, v []globals.Message) float32 {
return float32(CountTokenPrice(v, model)) / 1000 * 2.1
case globals.GPT432k:
return float32(CountTokenPrice(v, model)) / 1000 * 4.2
case globals.SparkDesk:
return float32(CountTokenPrice(v, model)) / 1000 * 0.36
case globals.Claude2, globals.Claude2100k:
return 0
default:
@ -120,6 +123,8 @@ func CountOutputToken(model string, t int) float32 {
return float32(t*GetWeightByModel(model)) / 1000 * 4.3
case globals.GPT432k:
return float32(t*GetWeightByModel(model)) / 1000 * 8.6
case globals.SparkDesk:
return float32(t*GetWeightByModel(model)) / 1000 * 0.36
case globals.Claude2, globals.Claude2100k:
return 0
default: