c5405b2a12
- Introduced a new rule for the Billing Expression System, emphasizing the importance of reading `pkg/billingexpr/expr.md` for dynamic billing. - Updated the billing expression logic to support new variables and improved handling of image and audio tokens. - Enhanced the tiered billing functionality with versioning support for expressions and refined quota calculations. - Added tests to validate the new billing expression features and ensure correctness in pricing calculations.
125 lines
3.7 KiB
Go
125 lines
3.7 KiB
Go
package operation_setting
|
|
|
|
import (
|
|
"strings"
|
|
|
|
"github.com/QuantumNous/new-api/setting/config"
|
|
)
|
|
|
|
// ---------------------------------------------------------------------------
|
|
// Tool call prices ($/1K calls, admin-configurable)
|
|
// DB keys: tool_price_setting.prices
|
|
// ---------------------------------------------------------------------------
|
|
|
|
var defaultToolPrices = map[string]float64{
|
|
"web_search": 10.0,
|
|
"web_search_high": 25.0,
|
|
"claude_web_search": 10.0,
|
|
"file_search": 2.5,
|
|
}
|
|
|
|
// ToolPriceSetting is managed by config.GlobalConfig.Register.
|
|
type ToolPriceSetting struct {
|
|
Prices map[string]float64 `json:"prices"`
|
|
}
|
|
|
|
var toolPriceSetting = ToolPriceSetting{
|
|
Prices: func() map[string]float64 {
|
|
m := make(map[string]float64, len(defaultToolPrices))
|
|
for k, v := range defaultToolPrices {
|
|
m[k] = v
|
|
}
|
|
return m
|
|
}(),
|
|
}
|
|
|
|
func init() {
|
|
config.GlobalConfig.Register("tool_price_setting", &toolPriceSetting)
|
|
}
|
|
|
|
// GetToolPrice returns the configured price for a tool key ($/1K calls),
|
|
// falling back to hardcoded default if not overridden.
|
|
func GetToolPrice(key string) float64 {
|
|
if v, ok := toolPriceSetting.Prices[key]; ok {
|
|
return v
|
|
}
|
|
if v, ok := defaultToolPrices[key]; ok {
|
|
return v
|
|
}
|
|
return 0
|
|
}
|
|
|
|
// ---------------------------------------------------------------------------
|
|
// GPT Image 1 per-call pricing (special: depends on quality + size)
|
|
// ---------------------------------------------------------------------------
|
|
|
|
const (
|
|
GPTImage1Low1024x1024 = 0.011
|
|
GPTImage1Low1024x1536 = 0.016
|
|
GPTImage1Low1536x1024 = 0.016
|
|
GPTImage1Medium1024x1024 = 0.042
|
|
GPTImage1Medium1024x1536 = 0.063
|
|
GPTImage1Medium1536x1024 = 0.063
|
|
GPTImage1High1024x1024 = 0.167
|
|
GPTImage1High1024x1536 = 0.25
|
|
GPTImage1High1536x1024 = 0.25
|
|
)
|
|
|
|
func GetGPTImage1PriceOnceCall(quality string, size string) float64 {
|
|
prices := map[string]map[string]float64{
|
|
"low": {
|
|
"1024x1024": GPTImage1Low1024x1024,
|
|
"1024x1536": GPTImage1Low1024x1536,
|
|
"1536x1024": GPTImage1Low1536x1024,
|
|
},
|
|
"medium": {
|
|
"1024x1024": GPTImage1Medium1024x1024,
|
|
"1024x1536": GPTImage1Medium1024x1536,
|
|
"1536x1024": GPTImage1Medium1536x1024,
|
|
},
|
|
"high": {
|
|
"1024x1024": GPTImage1High1024x1024,
|
|
"1024x1536": GPTImage1High1024x1536,
|
|
"1536x1024": GPTImage1High1536x1024,
|
|
},
|
|
}
|
|
|
|
if qualityMap, exists := prices[quality]; exists {
|
|
if price, exists := qualityMap[size]; exists {
|
|
return price
|
|
}
|
|
}
|
|
|
|
return GPTImage1High1024x1024
|
|
}
|
|
|
|
// ---------------------------------------------------------------------------
|
|
// Gemini audio input pricing (per-million tokens, model-specific)
|
|
// ---------------------------------------------------------------------------
|
|
|
|
const (
|
|
Gemini25FlashPreviewInputAudioPrice = 1.00
|
|
Gemini25FlashProductionInputAudioPrice = 1.00
|
|
Gemini25FlashLitePreviewInputAudioPrice = 0.50
|
|
Gemini25FlashNativeAudioInputAudioPrice = 3.00
|
|
Gemini20FlashInputAudioPrice = 0.70
|
|
GeminiRoboticsER15InputAudioPrice = 1.00
|
|
)
|
|
|
|
func GetGeminiInputAudioPricePerMillionTokens(modelName string) float64 {
|
|
if strings.HasPrefix(modelName, "gemini-2.5-flash-preview-native-audio") {
|
|
return Gemini25FlashNativeAudioInputAudioPrice
|
|
} else if strings.HasPrefix(modelName, "gemini-2.5-flash-preview-lite") {
|
|
return Gemini25FlashLitePreviewInputAudioPrice
|
|
} else if strings.HasPrefix(modelName, "gemini-2.5-flash-preview") {
|
|
return Gemini25FlashPreviewInputAudioPrice
|
|
} else if strings.HasPrefix(modelName, "gemini-2.5-flash") {
|
|
return Gemini25FlashProductionInputAudioPrice
|
|
} else if strings.HasPrefix(modelName, "gemini-2.0-flash") {
|
|
return Gemini20FlashInputAudioPrice
|
|
} else if strings.HasPrefix(modelName, "gemini-robotics-er-1.5") {
|
|
return GeminiRoboticsER15InputAudioPrice
|
|
}
|
|
return 0
|
|
}
|