mirror of
https://github.com/Wei-Shaw/sub2api.git
synced 2026-10-16 07:33:06 +08:00
IanShaw027
f3d9491071
运营需要按分组覆盖渠道/内置价，且部分套餐不应自动吃 200k 倍率。 原先只能改渠道价卡，分组侧只剩 Voice 三列。 - groups 新增 model_pricing / long_context_pricing_enabled，解析链改为 Group → Channel → 内置 - 关闭长上下文时 token 模型只取最低档；video 按秒计费可写进同一价卡 - 回退价对齐官方卡：4.5 缓存 $0.30、4.3/imagine/audio/search 默认值一并校正
43 lines
1.7 KiB
Go
43 lines
1.7 KiB
Go
package service
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestCalculateSearchCost(t *testing.T) {
|
|
t.Parallel()
|
|
s := &BillingService{}
|
|
require.Equal(t, 0.0, s.CalculateSearchCost(0, floatPtr(10), 1).ActualCost)
|
|
// nil price -> official xAI default $5/1k: 5 calls = 0.025
|
|
require.InDelta(t, 0.025, s.CalculateSearchCost(5, nil, 1).ActualCost, 1e-9)
|
|
// explicit 0 → free
|
|
require.Equal(t, 0.0, s.CalculateSearchCost(5, floatPtr(0), 1).ActualCost)
|
|
price := 10.0
|
|
cost := s.CalculateSearchCost(100, &price, 1.5)
|
|
// 10 / 1000 * 100 * 1.5 = 1.5
|
|
require.InDelta(t, 1.0, cost.TotalCost, 1e-9)
|
|
require.InDelta(t, 1.5, cost.ActualCost, 1e-9)
|
|
}
|
|
|
|
func TestCalculateAudioCost(t *testing.T) {
|
|
t.Parallel()
|
|
s := &BillingService{}
|
|
rt, tts, stt := 0.10, 15.0, 0.50
|
|
cfg := &audioPriceConfig{RealtimePerMin: &rt, TTSPerMChars: &tts, STTPerHour: &stt}
|
|
require.InDelta(t, 0.20, s.CalculateAudioCost("realtime", 2, cfg, 1).ActualCost, 1e-9)
|
|
require.InDelta(t, 1.5, s.CalculateAudioCost("tts", 0.1, cfg, 1).ActualCost, 1e-9)
|
|
require.InDelta(t, 0.25, s.CalculateAudioCost("stt", 0.5, cfg, 1).ActualCost, 1e-9)
|
|
require.Equal(t, 0.0, s.CalculateAudioCost("unknown", 1, cfg, 1).ActualCost)
|
|
// nil config -> official defaults (think-fast-1 $0.05/min, TTS $15/M, REST STT $0.10/hr)
|
|
require.InDelta(t, 0.05, s.CalculateAudioCost("realtime", 1, nil, 1).ActualCost, 1e-9)
|
|
require.InDelta(t, 15.0, s.CalculateAudioCost("tts", 1, nil, 1).ActualCost, 1e-9)
|
|
require.InDelta(t, 0.10, s.CalculateAudioCost("stt", 1, nil, 1).ActualCost, 1e-9)
|
|
// explicit 0 → free
|
|
zero := 0.0
|
|
require.Equal(t, 0.0, s.CalculateAudioCost("realtime", 1, &audioPriceConfig{RealtimePerMin: &zero}, 1).ActualCost)
|
|
}
|
|
|
|
func floatPtr(v float64) *float64 { return &v }
|