mirrors2/opencatd-open
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update:计费

Browse Source
This commit is contained in:
Sakurasan
2023-07-02 21:00:58 +08:00
parent d99407127f
commit c3000c401c
1 changed files with 15 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
router/router.go
View File

@@ -638,12 +638,24 @@ func Cost(model string, promptCount, completionCount int) float64 {
completion = float64(completionCount) completion = float64(completionCount)
switch model { switch model {
case "gpt-3.5-turbo", "gpt-3.5-turbo-0301": case "gpt-3.5-turbo-0301":
cost = 0.002 * float64((prompt+completion)/1000) cost = 0.002 * float64((prompt+completion)/1000)
case "gpt-4", "gpt-4-0314": case "gpt-3.5-turbo", "gpt-3.5-turbo-0613":
cost = 0.0015*float64((prompt)/1000) + 0.002*float64(completion/1000)
case "gpt-3.5-turbo-16k", "gpt-3.5-turbo-16k-0613":
cost = 0.003*float64((prompt)/1000) + 0.004*float64(completion/1000)
case "gpt-4", "gpt-4-0613", "gpt-4-0314":
cost = 0.03*float64(prompt/1000) + 0.06*float64(completion/1000) cost = 0.03*float64(prompt/1000) + 0.06*float64(completion/1000)
case "gpt-4-32k", "gpt-4-32k-0314": case "gpt-4-32k", "gpt-4-32k-0613":
cost = 0.06*float64(prompt/1000) + 0.12*float64(completion/1000) cost = 0.06*float64(prompt/1000) + 0.12*float64(completion/1000)
default:
if strings.Contains(model, "gpt-3.5-turbo") {
cost = 0.003 * float64((prompt+completion)/1000)
} else if strings.Contains(model, "gpt-4") {
cost = 0.06 * float64((prompt+completion)/1000)
} else {
cost = 0.002 * float64((prompt+completion)/1000)
}
} }
return cost return cost
} }