From 8ede1bf12174ded20c492872c64218646f39d613 Mon Sep 17 00:00:00 2001
From: iszcz <74706321+iszcz@users.noreply.github.com>
Date: Fri, 3 Jan 2025 20:42:46 +0800
Subject: [PATCH] Update model-ratio.go

---
 common/model-ratio.go | 23 +++++++++++++++++------
 1 file changed, 17 insertions(+), 6 deletions(-)

diff --git a/common/model-ratio.go b/common/model-ratio.go
index 7d4b852b..e9e80e47 100644
--- a/common/model-ratio.go
+++ b/common/model-ratio.go
@@ -431,12 +431,23 @@ func GetCompletionRatio(name string) float64 {
 }
 
 func GetAudioRatio(name string) float64 {
-	if strings.HasPrefix(name, "gpt-4o-realtime") {
-		return 20
-	} else if strings.HasPrefix(name, "gpt-4o-audio") {
-		return 40
-	} else if strings.HasPrefix(name, "gpt-4o-mini-realtime") {
-		return 10 / 0.6
+	if strings.Contains(name, "-realtime") {
+		if strings.HasSuffix(name, "gpt-4o-realtime-preview-2024-12-17") {
+			return 8
+		} else if strings.Contains(name, "mini") {
+			return 10 / 0.6
+		} else {
+			return 20
+		}
+	}
+	if strings.Contains(name, "-audio") {
+		if strings.HasSuffix(name, "gpt-4o-audio-preview-2024-12-17") {
+			return 16
+		} else if strings.Contains(name, "mini") {
+			return 10 / 0.15
+		} else {
+			return 40
+		}
 	}
 	return 20
 }