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

refactor: update variable names and improve code readability; add Makefile for linting and installation

Browse Source
This commit is contained in:
Laisky.Cai
2025-01-23 03:32:41 +00:00
parent 81153f764b
commit 531f832d1e
12 changed files with 659 additions and 60 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"
@@ -396,7 +395,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
@@ -404,7 +403,7 @@ func GetAudioPromptTokensPerSecond(actualModelName string) int {
v = 10
}
return int(math.Ceil(v))
return v
}
var CompletionRatio = map[string]float64{