Quellcode durchsuchen

feat: update model ratio

CalciumIon vor 1 Jahr
Ursprung
Commit
cbf0688b80
2 geänderte Dateien mit 4 neuen und 3 gelöschten Zeilen
  1. 1 1
      common/model-ratio.go
  2. 3 2
      service/quota.go

+ 1 - 1
common/model-ratio.go

@@ -430,7 +430,7 @@ func GetAudioRatio(name string) float64 {
 
 
 func GetAudioCompletionRatio(name string) float64 {
 func GetAudioCompletionRatio(name string) float64 {
 	if strings.HasPrefix(name, "gpt-4o-realtime") {
 	if strings.HasPrefix(name, "gpt-4o-realtime") {
-		return 10
+		return 2
 	}
 	}
 	return 2
 	return 2
 }
 }

+ 3 - 2
service/quota.go

@@ -78,8 +78,9 @@ func PostWssConsumeQuota(ctx *gin.Context, relayInfo *relaycommon.RelayInfo, mod
 
 
 	quota := 0
 	quota := 0
 	if !usePrice {
 	if !usePrice {
-		quota = int(math.Round(float64(textInputTokens)*ratio + float64(textOutTokens)*ratio*completionRatio))
-		quota += int(math.Round(float64(audioInputTokens)*ratio*audioRatio + float64(audioOutTokens)*ratio*audioRatio*audioCompletionRatio))
+		quota = int(math.Round(float64(textInputTokens) + float64(textOutTokens)*completionRatio))
+		quota += int(math.Round(float64(audioInputTokens)*audioRatio + float64(audioOutTokens)*audioRatio*audioCompletionRatio))
+		quota = int(math.Round(float64(quota) * ratio))
 		if ratio != 0 && quota <= 0 {
 		if ratio != 0 && quota <= 0 {
 			quota = 1
 			quota = 1
 		}
 		}