From 4fae477af8079f85a0269aa874ae0df695545ba6 Mon Sep 17 00:00:00 2001
From: Sakurasan <26715255+Sakurasan@users.noreply.github.com>
Date: Tue, 14 May 2024 12:24:18 +0800
Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0gpt-4o=E8=AE=A1=E8=B4=B9?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 pkg/tokenizer/tokenizer.go | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/pkg/tokenizer/tokenizer.go b/pkg/tokenizer/tokenizer.go
index e68389f..aabda19 100644
--- 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