CaIon
bae799ccb1
Thread int32 saturation clamps from tiered settlement and video task recompute into the consume/task logs under admin_info, so oversized or malformed billing inputs stay auditable. Clamp negative audio duration before token conversion and gate the saturation UI markers on admin.
87 lines
2.4 KiB
Go
87 lines
2.4 KiB
Go
package service
|
|
|
|
import (
|
|
"github.com/QuantumNous/new-api/common"
|
|
"github.com/QuantumNous/new-api/setting/operation_setting"
|
|
)
|
|
|
|
// ToolCallUsage captures all tool call counts from a single request.
|
|
type ToolCallUsage struct {
|
|
ModelName string
|
|
WebSearchCalls int
|
|
WebSearchToolName string // "web_search_preview", "web_search", etc.
|
|
FileSearchCalls int
|
|
ImageGenerationCall bool
|
|
ImageGenerationQuality string
|
|
ImageGenerationSize string
|
|
}
|
|
|
|
// ToolCallItem represents a single billed tool usage line.
|
|
type ToolCallItem struct {
|
|
Name string `json:"name"`
|
|
CallCount int `json:"call_count"`
|
|
PricePer1K float64 `json:"price_per_1k"`
|
|
TotalPrice float64 `json:"total_price"`
|
|
Quota int `json:"quota"`
|
|
}
|
|
|
|
// ToolCallResult holds the aggregated tool call billing for a request.
|
|
type ToolCallResult struct {
|
|
TotalQuota int `json:"total_quota"`
|
|
Items []ToolCallItem `json:"items,omitempty"`
|
|
}
|
|
|
|
// ComputeToolCallQuota calculates the total quota for all tool calls in a
|
|
// request. Tool prices are resolved via GetToolPriceForModel which supports
|
|
// model-prefix overrides. groupRatio is applied.
|
|
func ComputeToolCallQuota(usage ToolCallUsage, groupRatio float64) ToolCallResult {
|
|
var items []ToolCallItem
|
|
totalQuota := 0
|
|
|
|
addItem := func(toolName string, count int) {
|
|
if count <= 0 {
|
|
return
|
|
}
|
|
pricePer1K := operation_setting.GetToolPriceForModel(toolName, usage.ModelName)
|
|
if pricePer1K <= 0 {
|
|
return
|
|
}
|
|
totalPrice := pricePer1K * float64(count) / 1000
|
|
quota := common.QuotaRound(totalPrice * common.QuotaPerUnit * groupRatio)
|
|
items = append(items, ToolCallItem{
|
|
Name: toolName,
|
|
CallCount: count,
|
|
PricePer1K: pricePer1K,
|
|
TotalPrice: totalPrice,
|
|
Quota: quota,
|
|
})
|
|
totalQuota += quota
|
|
}
|
|
|
|
if usage.WebSearchCalls > 0 && usage.WebSearchToolName != "" {
|
|
addItem(usage.WebSearchToolName, usage.WebSearchCalls)
|
|
}
|
|
|
|
if usage.FileSearchCalls > 0 {
|
|
addItem("file_search", usage.FileSearchCalls)
|
|
}
|
|
|
|
if usage.ImageGenerationCall {
|
|
price := operation_setting.GetGPTImage1PriceOnceCall(usage.ImageGenerationQuality, usage.ImageGenerationSize)
|
|
quota := common.QuotaRound(price * common.QuotaPerUnit * groupRatio)
|
|
items = append(items, ToolCallItem{
|
|
Name: "image_generation",
|
|
CallCount: 1,
|
|
PricePer1K: price,
|
|
TotalPrice: price,
|
|
Quota: quota,
|
|
})
|
|
totalQuota += quota
|
|
}
|
|
|
|
return ToolCallResult{
|
|
TotalQuota: totalQuota,
|
|
Items: items,
|
|
}
|
|
}
|