xinyin025/one-api
1
0
Fork 0
mirror of https://github.com/songquanpeng/one-api.git synced 2025-11-12 11:23:42 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: change GetAudioTokens to return float64 and update related functions

Browse Source
This commit is contained in:
Laisky.Cai
2025-01-26 12:17:31 +00:00
parent bcba9bf3a1
commit b83e400297
5 changed files with 14 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
relay/billing/ratio/model.go
View File

@@ -3,7 +3,6 @@ package ratio
import (
"encoding/json"
"fmt"
"math"
"strings"
"github.com/songquanpeng/one-api/common/logger"
@@ -391,7 +390,7 @@ var AudioPromptTokensPerSecond = map[string]float64{
// GetAudioPromptTokensPerSecond returns the number of audio tokens per second
// for the given model.
func GetAudioPromptTokensPerSecond(actualModelName string) int {
func GetAudioPromptTokensPerSecond(actualModelName string) float64 {
var v float64
if tokensPerSecond, ok := AudioPromptTokensPerSecond[actualModelName]; ok {
v = tokensPerSecond
@@ -399,7 +398,7 @@ func GetAudioPromptTokensPerSecond(actualModelName string) int {
v = 10
}
return int(math.Ceil(v))
return v
}
var CompletionRatio = map[string]float64{