GithubProxy/coai
1
0
Fork 0
mirror of https://github.com/coaidev/coai.git synced 2025-05-28 17:30:15 +09:00
Code Issues Actions Packages Projects Releases Wiki Activity

update claude-2-100k prize

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

2
auth/payment.go
View File

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

8
utils/tokenizer.go
View File

@ -106,8 +106,10 @@ func CountInputToken(model string, v []globals.Message) float32 {
return float32(CountTokenPrice(v, model)) / 1000 * 4.2
case globals.SparkDesk:
return float32(CountTokenPrice(v, model)) / 1000 * 0.36
case globals.Claude2, globals.Claude2100k:
case globals.Claude2:
return 0
case globals.Claude2100k:
return float32(CountTokenPrice(v, model)) / 1000 * 0.008
default:
return 0
}
@ -125,8 +127,10 @@ func CountOutputToken(model string, t int) float32 {
return float32(t*GetWeightByModel(model)) / 1000 * 8.6
case globals.SparkDesk:
return float32(t*GetWeightByModel(model)) / 1000 * 0.36
case globals.Claude2, globals.Claude2100k:
case globals.Claude2:
return 0
case globals.Claude2100k:
return float32(t*GetWeightByModel(model)) / 1000 * 0.008
default:
return 0
}