更新gpt-4o计费

2024-05-14 12:24:18 +08:00
parent dd2d583ffc
commit 4fae477af8
1 changed files with 2 additions and 0 deletions
--- a/pkg/tokenizer/tokenizer.go
+++ b/pkg/tokenizer/tokenizer.go
@@ -95,6 +95,8 @@ func Cost(model string, promptCount, completionCount int) float64 {
 		cost = 0.01*float64(prompt/1000) + 0.03*float64(completion/1000)
 	case "gpt-4-turbo", "gpt-4-turbo-2024-04-09":
 		cost = 0.01*float64(prompt/1000) + 0.03*float64(completion/1000)
+	case "gpt-4o", "gpt-4o-2024-05-13":
+		cost = 0.005*float64(prompt/1000) + 0.015*float64(completion/1000)
 	case "whisper-1":
 		// 0.006$/min
 		cost = 0.006 * float64(prompt+completion) / 60