Sub2API v1.0 - AI API 网关(二开初始版本,基于上游 Wei-Shaw/sub2api)
Release / update-version (push) Has been cancelled
Release / build-frontend (push) Has been cancelled
Release / release (push) Has been cancelled
Release / sync-version-file (push) Has been cancelled
CI / shell (push) Canceled after 0s
CI / test (push) Canceled after 0s
CI / frontend (push) Canceled after 0s
CI / golangci-lint (push) Canceled after 0s
Security Scan / backend-security (push) Canceled after 0s
Security Scan / frontend-security (push) Canceled after 0s
Release / update-version (push) Has been cancelled
Release / build-frontend (push) Has been cancelled
Release / release (push) Has been cancelled
Release / sync-version-file (push) Has been cancelled
CI / shell (push) Canceled after 0s
CI / test (push) Canceled after 0s
CI / frontend (push) Canceled after 0s
CI / golangci-lint (push) Canceled after 0s
Security Scan / backend-security (push) Canceled after 0s
Security Scan / frontend-security (push) Canceled after 0s
This commit is contained in:
@@ -0,0 +1,643 @@
|
||||
package service
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"sort"
|
||||
"strings"
|
||||
"time"
|
||||
)
|
||||
|
||||
// BillingMode 计费模式
|
||||
type BillingMode string
|
||||
|
||||
const (
|
||||
BillingModeToken BillingMode = "token" // 按 token 区间计费
|
||||
BillingModePerRequest BillingMode = "per_request" // 按次计费(支持上下文窗口分层)
|
||||
BillingModeImage BillingMode = "image" // 图片计费(当前按次,预留 token 计费)
|
||||
BillingModeVideo BillingMode = "video" // 视频生成计费(按视频生成次数)
|
||||
)
|
||||
|
||||
// IsValid 检查 BillingMode 是否为合法值
|
||||
func (m BillingMode) IsValid() bool {
|
||||
switch m {
|
||||
case BillingModeToken, BillingModePerRequest, BillingModeImage, BillingModeVideo, "":
|
||||
return true
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
// IsValidUsageFilter 检查 BillingMode 是否可用于使用记录筛选。
|
||||
func (m BillingMode) IsValidUsageFilter() bool {
|
||||
switch m {
|
||||
case BillingModeToken, BillingModePerRequest, BillingModeImage, BillingModeVideo, "":
|
||||
return true
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
const (
|
||||
BillingModelSourceRequested = "requested"
|
||||
BillingModelSourceUpstream = "upstream"
|
||||
BillingModelSourceChannelMapped = "channel_mapped"
|
||||
// BillingModelSourceResponse bills by a trusted model declaration observed
|
||||
// in the successful upstream response. It is deliberately distinct from
|
||||
// "upstream", which means the model sent to the provider.
|
||||
BillingModelSourceResponse = "response_model"
|
||||
)
|
||||
|
||||
// Channel 渠道实体
|
||||
type Channel struct {
|
||||
ID int64
|
||||
Name string
|
||||
Description string
|
||||
Status string
|
||||
BillingModelSource string // "requested", "upstream", "channel_mapped", or "response_model"
|
||||
RestrictModels bool // 是否限制模型(仅允许定价列表中的模型)
|
||||
Features string // 渠道特性描述(JSON 数组),用于支付页面展示
|
||||
FeaturesConfig map[string]any // 渠道功能配置(如 web search emulation)
|
||||
CreatedAt time.Time
|
||||
UpdatedAt time.Time
|
||||
|
||||
// 关联的分组 ID 列表
|
||||
GroupIDs []int64
|
||||
// 模型定价列表(每条含 Platform 字段)
|
||||
ModelPricing []ChannelModelPricing
|
||||
// 渠道级模型映射(按平台分组:platform → {src→dst})
|
||||
ModelMapping map[string]map[string]string
|
||||
|
||||
// 账号统计定价
|
||||
ApplyPricingToAccountStats bool // 是否应用渠道模型定价到账号统计
|
||||
AccountStatsPricingRules []AccountStatsPricingRule // 自定义账号统计定价规则(按 SortOrder 排序,先命中为准)
|
||||
}
|
||||
|
||||
// AccountStatsPricingRule 账号统计定价规则
|
||||
// 每条规则包含匹配条件(分组/账号)和独立的模型定价。
|
||||
// 多条规则按 SortOrder 排序,先命中为准。
|
||||
type AccountStatsPricingRule struct {
|
||||
ID int64
|
||||
ChannelID int64
|
||||
Name string
|
||||
GroupIDs []int64
|
||||
AccountIDs []int64
|
||||
SortOrder int
|
||||
Pricing []ChannelModelPricing // 规则内的模型定价(复用现有定价结构)
|
||||
CreatedAt time.Time
|
||||
UpdatedAt time.Time
|
||||
}
|
||||
|
||||
// ChannelModelPricing 渠道模型定价条目
|
||||
type ChannelModelPricing struct {
|
||||
ID int64 `json:"id,omitempty"`
|
||||
ChannelID int64 `json:"channel_id,omitempty"`
|
||||
Platform string `json:"platform"` // 所属平台(anthropic/openai/gemini/...)
|
||||
Models []string `json:"models"`
|
||||
BillingMode BillingMode `json:"billing_mode"`
|
||||
InputPrice *float64 `json:"input_price"`
|
||||
OutputPrice *float64 `json:"output_price"`
|
||||
CacheWritePrice *float64 `json:"cache_write_price"`
|
||||
CacheReadPrice *float64 `json:"cache_read_price"`
|
||||
FastMultiplier *float64 `json:"fast_multiplier"`
|
||||
FlexMultiplier *float64 `json:"flex_multiplier"`
|
||||
ImageInputPrice *float64 `json:"image_input_price"`
|
||||
ImageOutputPrice *float64 `json:"image_output_price"`
|
||||
PerRequestPrice *float64 `json:"per_request_price"`
|
||||
Intervals []PricingInterval `json:"intervals"`
|
||||
TimePricing *ChannelTimePricing `json:"time_pricing,omitempty"`
|
||||
CreatedAt time.Time `json:"created_at,omitempty"`
|
||||
UpdatedAt time.Time `json:"updated_at,omitempty"`
|
||||
}
|
||||
|
||||
// ChannelTimePricing 渠道模型定价的分时倍率配置。
|
||||
type ChannelTimePricing struct {
|
||||
Timezone string `json:"timezone"`
|
||||
Periods []ChannelTimePricingPeriod `json:"periods"`
|
||||
}
|
||||
|
||||
// ChannelTimePricingPeriod 是秒级的左闭右开分时倍率区间,并兼容历史 HH:mm 数据。
|
||||
type ChannelTimePricingPeriod struct {
|
||||
StartTime string `json:"start_time"`
|
||||
EndTime string `json:"end_time"`
|
||||
Multiplier float64 `json:"multiplier"`
|
||||
}
|
||||
|
||||
// PricingInterval 定价区间(token 区间 / 按次分层 / 图片分辨率分层)
|
||||
type PricingInterval struct {
|
||||
ID int64 `json:"id,omitempty"`
|
||||
PricingID int64 `json:"pricing_id,omitempty"`
|
||||
MinTokens int `json:"min_tokens"`
|
||||
MaxTokens *int `json:"max_tokens"`
|
||||
TierLabel string `json:"tier_label"`
|
||||
InputPrice *float64 `json:"input_price"`
|
||||
OutputPrice *float64 `json:"output_price"`
|
||||
CacheWritePrice *float64 `json:"cache_write_price"`
|
||||
CacheReadPrice *float64 `json:"cache_read_price"`
|
||||
InputMultiplier *float64 `json:"input_multiplier"`
|
||||
OutputMultiplier *float64 `json:"output_multiplier"`
|
||||
CacheWriteMultiplier *float64 `json:"cache_write_multiplier"`
|
||||
CacheReadMultiplier *float64 `json:"cache_read_multiplier"`
|
||||
PerRequestPrice *float64 `json:"per_request_price"`
|
||||
SortOrder int `json:"sort_order"`
|
||||
CreatedAt time.Time `json:"created_at,omitempty"`
|
||||
UpdatedAt time.Time `json:"updated_at,omitempty"`
|
||||
}
|
||||
|
||||
// IsActive 判断渠道是否启用
|
||||
func (c *Channel) IsActive() bool {
|
||||
return c.Status == StatusActive
|
||||
}
|
||||
|
||||
// normalizeBillingModelSource 若 BillingModelSource 为空则回填默认值 ChannelMapped。
|
||||
// 作为 *Channel 的实体方法集中管理默认值,service 层只需在 Channel 进入内存
|
||||
// (缓存装填、repo 读出)时调用一次,下游读路径就无需重复兜底。
|
||||
func (c *Channel) normalizeBillingModelSource() {
|
||||
if c == nil {
|
||||
return
|
||||
}
|
||||
if c.BillingModelSource == "" {
|
||||
c.BillingModelSource = BillingModelSourceChannelMapped
|
||||
}
|
||||
}
|
||||
|
||||
// GetModelPricing 根据模型名查找渠道定价,未找到返回 nil。
|
||||
// 精确匹配,大小写不敏感。返回值拷贝,不污染缓存。
|
||||
func (c *Channel) GetModelPricing(model string) *ChannelModelPricing {
|
||||
modelLower := strings.ToLower(model)
|
||||
|
||||
for i := range c.ModelPricing {
|
||||
for _, m := range c.ModelPricing[i].Models {
|
||||
if strings.ToLower(m) == modelLower {
|
||||
cp := c.ModelPricing[i].Clone()
|
||||
return &cp
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// FindMatchingInterval 在区间列表中查找匹配 totalTokens 的区间。
|
||||
// 区间为左开右闭 (min, max]:min 不含,max 包含。
|
||||
// 第一个区间 min=0 时,0 token 不匹配任何区间(回退到默认价格)。
|
||||
func FindMatchingInterval(intervals []PricingInterval, totalTokens int) *PricingInterval {
|
||||
for i := range intervals {
|
||||
iv := &intervals[i]
|
||||
if totalTokens > iv.MinTokens && (iv.MaxTokens == nil || totalTokens <= *iv.MaxTokens) {
|
||||
return iv
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// GetIntervalForContext 根据总 context token 数查找匹配的区间。
|
||||
func (p *ChannelModelPricing) GetIntervalForContext(totalTokens int) *PricingInterval {
|
||||
return FindMatchingInterval(p.Intervals, totalTokens)
|
||||
}
|
||||
|
||||
// GetTierByLabel 根据标签查找层级(用于 per_request / image 模式)
|
||||
func (p *ChannelModelPricing) GetTierByLabel(label string) *PricingInterval {
|
||||
labelLower := strings.ToLower(label)
|
||||
for i := range p.Intervals {
|
||||
if strings.ToLower(p.Intervals[i].TierLabel) == labelLower {
|
||||
return &p.Intervals[i]
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// Clone 返回 ChannelModelPricing 的拷贝(切片独立,指针字段共享,调用方只读安全)
|
||||
func (p ChannelModelPricing) Clone() ChannelModelPricing {
|
||||
cp := p
|
||||
if p.Models != nil {
|
||||
cp.Models = make([]string, len(p.Models))
|
||||
copy(cp.Models, p.Models)
|
||||
}
|
||||
if p.Intervals != nil {
|
||||
cp.Intervals = make([]PricingInterval, len(p.Intervals))
|
||||
copy(cp.Intervals, p.Intervals)
|
||||
}
|
||||
if p.TimePricing != nil {
|
||||
cp.TimePricing = &ChannelTimePricing{Timezone: p.TimePricing.Timezone}
|
||||
if p.TimePricing.Periods != nil {
|
||||
cp.TimePricing.Periods = append([]ChannelTimePricingPeriod(nil), p.TimePricing.Periods...)
|
||||
}
|
||||
}
|
||||
return cp
|
||||
}
|
||||
|
||||
// Clone 返回 Channel 的深拷贝
|
||||
func (c *Channel) Clone() *Channel {
|
||||
if c == nil {
|
||||
return nil
|
||||
}
|
||||
cp := *c
|
||||
if c.GroupIDs != nil {
|
||||
cp.GroupIDs = make([]int64, len(c.GroupIDs))
|
||||
copy(cp.GroupIDs, c.GroupIDs)
|
||||
}
|
||||
if c.ModelPricing != nil {
|
||||
cp.ModelPricing = make([]ChannelModelPricing, len(c.ModelPricing))
|
||||
for i := range c.ModelPricing {
|
||||
cp.ModelPricing[i] = c.ModelPricing[i].Clone()
|
||||
}
|
||||
}
|
||||
if c.ModelMapping != nil {
|
||||
cp.ModelMapping = make(map[string]map[string]string, len(c.ModelMapping))
|
||||
for platform, mapping := range c.ModelMapping {
|
||||
inner := make(map[string]string, len(mapping))
|
||||
for k, v := range mapping {
|
||||
inner[k] = v
|
||||
}
|
||||
cp.ModelMapping[platform] = inner
|
||||
}
|
||||
}
|
||||
if c.FeaturesConfig != nil {
|
||||
cp.FeaturesConfig = deepCopyFeaturesConfig(c.FeaturesConfig)
|
||||
}
|
||||
if c.AccountStatsPricingRules != nil {
|
||||
cp.AccountStatsPricingRules = make([]AccountStatsPricingRule, len(c.AccountStatsPricingRules))
|
||||
for i, rule := range c.AccountStatsPricingRules {
|
||||
cp.AccountStatsPricingRules[i] = rule
|
||||
if rule.GroupIDs != nil {
|
||||
cp.AccountStatsPricingRules[i].GroupIDs = make([]int64, len(rule.GroupIDs))
|
||||
copy(cp.AccountStatsPricingRules[i].GroupIDs, rule.GroupIDs)
|
||||
}
|
||||
if rule.AccountIDs != nil {
|
||||
cp.AccountStatsPricingRules[i].AccountIDs = make([]int64, len(rule.AccountIDs))
|
||||
copy(cp.AccountStatsPricingRules[i].AccountIDs, rule.AccountIDs)
|
||||
}
|
||||
if rule.Pricing != nil {
|
||||
cp.AccountStatsPricingRules[i].Pricing = make([]ChannelModelPricing, len(rule.Pricing))
|
||||
for j := range rule.Pricing {
|
||||
cp.AccountStatsPricingRules[i].Pricing[j] = rule.Pricing[j].Clone()
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
return &cp
|
||||
}
|
||||
|
||||
// IsWebSearchEmulationEnabled 返回该渠道是否为指定平台启用了 web search 模拟。
|
||||
func (c *Channel) IsWebSearchEmulationEnabled(platform string) bool {
|
||||
if c == nil || c.FeaturesConfig == nil {
|
||||
return false
|
||||
}
|
||||
wse, ok := c.FeaturesConfig[featureKeyWebSearchEmulation].(map[string]any)
|
||||
if !ok {
|
||||
return false
|
||||
}
|
||||
enabled, ok := wse[platform].(bool)
|
||||
return ok && enabled
|
||||
}
|
||||
|
||||
// IsBedrockCCCompatEnabled 返回该渠道是否启用了 Bedrock CC 兼容模式。
|
||||
// 一旦启用,该渠道下所有请求都会应用 CC 兼容转换,不区分账号 platform。
|
||||
func (c *Channel) IsBedrockCCCompatEnabled(platform string) bool {
|
||||
if c == nil || c.FeaturesConfig == nil {
|
||||
return false
|
||||
}
|
||||
// 直接检查 bedrock_cc_compat 开关,不再检查 platform 子字段
|
||||
enabled, ok := c.FeaturesConfig[featureKeyBedrockCCCompat].(bool)
|
||||
return ok && enabled
|
||||
}
|
||||
|
||||
// deepCopyFeaturesConfig creates a deep copy of FeaturesConfig to prevent cache pollution.
|
||||
func deepCopyFeaturesConfig(src map[string]any) map[string]any {
|
||||
dst := make(map[string]any, len(src))
|
||||
for k, v := range src {
|
||||
if inner, ok := v.(map[string]any); ok {
|
||||
dst[k] = deepCopyFeaturesConfig(inner)
|
||||
} else {
|
||||
dst[k] = v
|
||||
}
|
||||
}
|
||||
return dst
|
||||
}
|
||||
|
||||
// ValidateIntervals 校验区间列表的合法性。
|
||||
//
|
||||
// mode 决定区间语义:
|
||||
// - BillingModeToken(含空值):区间是上下文 token 数分段 (min, max],
|
||||
// 按 MinTokens 排序后无重叠,无界区间(MaxTokens=nil)必须是最后一个。
|
||||
// - BillingModePerRequest / BillingModeImage:区间是按 tier_label
|
||||
// (1K/2K/4K 等) 分层,匹配走 label 不依赖 min/max,因此跳过区间重叠
|
||||
// 与 last-unlimited 校验,仅做单条字段自洽(min/max/价格非负)检查。
|
||||
//
|
||||
// 通用规则:MinTokens >= 0;MaxTokens 若非 nil 则 > 0 且 > MinTokens;
|
||||
// 所有价格字段 >= 0。
|
||||
func ValidateIntervals(intervals []PricingInterval, mode BillingMode) error {
|
||||
if len(intervals) == 0 {
|
||||
return nil
|
||||
}
|
||||
sorted := make([]PricingInterval, len(intervals))
|
||||
copy(sorted, intervals)
|
||||
sort.Slice(sorted, func(i, j int) bool {
|
||||
return sorted[i].MinTokens < sorted[j].MinTokens
|
||||
})
|
||||
|
||||
for i := range sorted {
|
||||
if err := validateSingleInterval(&sorted[i], i); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
|
||||
// per_request / image 模式按 tier_label 匹配,不做 token 区间重叠校验
|
||||
if mode == BillingModePerRequest || mode == BillingModeImage || mode == BillingModeVideo {
|
||||
return nil
|
||||
}
|
||||
return validateIntervalOverlap(sorted)
|
||||
}
|
||||
|
||||
// validateSingleInterval 校验单个区间的字段合法性
|
||||
func validateSingleInterval(iv *PricingInterval, idx int) error {
|
||||
if iv.MinTokens < 0 {
|
||||
return fmt.Errorf("interval #%d: min_tokens (%d) must be >= 0", idx+1, iv.MinTokens)
|
||||
}
|
||||
if iv.MaxTokens != nil {
|
||||
if *iv.MaxTokens <= 0 {
|
||||
return fmt.Errorf("interval #%d: max_tokens (%d) must be > 0", idx+1, *iv.MaxTokens)
|
||||
}
|
||||
if *iv.MaxTokens <= iv.MinTokens {
|
||||
return fmt.Errorf("interval #%d: max_tokens (%d) must be > min_tokens (%d)",
|
||||
idx+1, *iv.MaxTokens, iv.MinTokens)
|
||||
}
|
||||
}
|
||||
return validateIntervalPrices(iv, idx)
|
||||
}
|
||||
|
||||
// validateIntervalPrices 校验区间价格 >= 0、倍率 > 0。
|
||||
func validateIntervalPrices(iv *PricingInterval, idx int) error {
|
||||
prices := []struct {
|
||||
name string
|
||||
val *float64
|
||||
}{
|
||||
{"input_price", iv.InputPrice},
|
||||
{"output_price", iv.OutputPrice},
|
||||
{"cache_write_price", iv.CacheWritePrice},
|
||||
{"cache_read_price", iv.CacheReadPrice},
|
||||
{"per_request_price", iv.PerRequestPrice},
|
||||
}
|
||||
for _, p := range prices {
|
||||
if p.val != nil && *p.val < 0 {
|
||||
return fmt.Errorf("interval #%d: %s must be >= 0", idx+1, p.name)
|
||||
}
|
||||
}
|
||||
multipliers := []struct {
|
||||
name string
|
||||
val *float64
|
||||
}{
|
||||
{"input_multiplier", iv.InputMultiplier},
|
||||
{"output_multiplier", iv.OutputMultiplier},
|
||||
{"cache_write_multiplier", iv.CacheWriteMultiplier},
|
||||
{"cache_read_multiplier", iv.CacheReadMultiplier},
|
||||
}
|
||||
for _, multiplier := range multipliers {
|
||||
if multiplier.val != nil && *multiplier.val <= 0 {
|
||||
return fmt.Errorf("interval #%d: %s must be > 0", idx+1, multiplier.name)
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// validateIntervalOverlap 校验排序后的区间列表无重叠,且无界区间在最后
|
||||
func validateIntervalOverlap(sorted []PricingInterval) error {
|
||||
for i, iv := range sorted {
|
||||
// 无界区间必须是最后一个
|
||||
if iv.MaxTokens == nil && i < len(sorted)-1 {
|
||||
return fmt.Errorf("interval #%d: unbounded interval (max_tokens=null) must be the last one",
|
||||
i+1)
|
||||
}
|
||||
if i == 0 {
|
||||
continue
|
||||
}
|
||||
prev := sorted[i-1]
|
||||
// 检查重叠:前一个区间的上界 > 当前区间的下界则重叠
|
||||
// (min, max] 语义:prev 覆盖 (prev.Min, prev.Max],cur 覆盖 (cur.Min, cur.Max]
|
||||
if prev.MaxTokens == nil || *prev.MaxTokens > iv.MinTokens {
|
||||
return fmt.Errorf("interval #%d and #%d overlap: prev max=%s > cur min=%d",
|
||||
i, i+1, formatMaxTokensLabel(prev.MaxTokens), iv.MinTokens)
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func formatMaxTokensLabel(max *int) string {
|
||||
if max == nil {
|
||||
return "∞"
|
||||
}
|
||||
return fmt.Sprintf("%d", *max)
|
||||
}
|
||||
|
||||
// ChannelUsageFields 渠道相关的使用记录字段(嵌入到各平台的 RecordUsageInput 中)
|
||||
type ChannelUsageFields struct {
|
||||
ChannelID int64 // 渠道 ID(0 = 无渠道)
|
||||
OriginalModel string // 用户原始请求模型(渠道映射前)
|
||||
ChannelMappedModel string // 渠道映射后的模型名(无映射时等于 OriginalModel)
|
||||
BillingModelSource string // 计费模型来源:"requested" / "upstream" / "channel_mapped"
|
||||
ModelMappingChain string // 映射链描述,如 "a→b→c"
|
||||
}
|
||||
|
||||
// SupportedModel 渠道的一个支持模型条目(无通配符、可直接展示给用户)
|
||||
type SupportedModel struct {
|
||||
Name string // 用户侧模型名
|
||||
Platform string // 所属平台
|
||||
Pricing *ChannelModelPricing // 定价详情(nil 表示未配置定价)
|
||||
}
|
||||
|
||||
// wildcardSuffix 是模型模式中的通配符后缀标记(仅支持尾部匹配)。
|
||||
const wildcardSuffix = "*"
|
||||
|
||||
// splitWildcardSuffix 将模型模式拆分为 (prefix, isWildcard)。
|
||||
//
|
||||
// "claude-opus-*" → ("claude-opus-", true)
|
||||
// "claude-opus-4" → ("claude-opus-4", false)
|
||||
// "*" → ("", true)
|
||||
//
|
||||
// 注意:返回的 prefix 保持原始大小写,由调用方按需 ToLower。
|
||||
func splitWildcardSuffix(pattern string) (prefix string, isWildcard bool) {
|
||||
if strings.HasSuffix(pattern, wildcardSuffix) {
|
||||
return strings.TrimSuffix(pattern, wildcardSuffix), true
|
||||
}
|
||||
return pattern, false
|
||||
}
|
||||
|
||||
// GetModelPricingByPlatform 在指定平台下查找精确模型的定价,未找到返回 nil。
|
||||
// 与 GetModelPricing 的区别:按 Platform 隔离,避免跨平台同名模型误匹配。
|
||||
func (c *Channel) GetModelPricingByPlatform(platform, model string) *ChannelModelPricing {
|
||||
if c == nil {
|
||||
return nil
|
||||
}
|
||||
modelLower := strings.ToLower(model)
|
||||
for i := range c.ModelPricing {
|
||||
if c.ModelPricing[i].Platform != platform {
|
||||
continue
|
||||
}
|
||||
for _, m := range c.ModelPricing[i].Models {
|
||||
if strings.ToLower(m) == modelLower {
|
||||
cp := c.ModelPricing[i].Clone()
|
||||
return &cp
|
||||
}
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// platformPricingIndex 是单个平台下定价信息的复合索引。
|
||||
// 一次扫描即可同时支持精确查找(exact 分支)与有序遍历(wildcard 分支),
|
||||
// 避免 SupportedModels 对每个平台重复扫描定价列表。
|
||||
//
|
||||
// byLower 与 names/originalCase 共享同一套去重规则:以 lower-case 模型名为 key,
|
||||
// 首个命中保留其原始大小写。names 维持按定价行扫描顺序的稳定迭代。
|
||||
type platformPricingIndex struct {
|
||||
byLower map[string]*ChannelModelPricing // lowercased model name → pricing (Clone'd)
|
||||
originalCase map[string]string // lowercased model name → original-case model name
|
||||
names []string // priced model names in their ORIGINAL case, insertion-ordered, deduped case-insensitively (first wins)
|
||||
}
|
||||
|
||||
// buildPricingIndex 对渠道的定价列表做一次扫描,按 platform 聚合为查找索引。
|
||||
// 索引值是定价条目的 Clone 指针,调用方可安全按需返回副本而不污染缓存。
|
||||
// 通配符后缀条目(如 "claude-*")不被索引(它们是模式,不是具体模型名)。
|
||||
// 同一平台中以大小写不敏感方式去重,先出现者保留原始大小写。
|
||||
func buildPricingIndex(pricings []ChannelModelPricing) map[string]*platformPricingIndex {
|
||||
idx := make(map[string]*platformPricingIndex)
|
||||
for i := range pricings {
|
||||
p := pricings[i]
|
||||
pidx, ok := idx[p.Platform]
|
||||
if !ok {
|
||||
pidx = &platformPricingIndex{
|
||||
byLower: make(map[string]*ChannelModelPricing),
|
||||
originalCase: make(map[string]string),
|
||||
names: make([]string, 0),
|
||||
}
|
||||
idx[p.Platform] = pidx
|
||||
}
|
||||
for _, m := range p.Models {
|
||||
if _, wild := splitWildcardSuffix(m); wild {
|
||||
continue
|
||||
}
|
||||
lower := strings.ToLower(m)
|
||||
if _, exists := pidx.byLower[lower]; exists {
|
||||
continue // 首个命中胜出(case-insensitive 去重后第一个定价 / 第一个原始大小写)
|
||||
}
|
||||
cp := pricings[i].Clone()
|
||||
pidx.byLower[lower] = &cp
|
||||
pidx.originalCase[lower] = m
|
||||
pidx.names = append(pidx.names, m)
|
||||
}
|
||||
}
|
||||
return idx
|
||||
}
|
||||
|
||||
// SupportedModels 计算渠道的支持模型列表,结果保证不含通配符。
|
||||
//
|
||||
// 算法(mapping ∪ pricing 并联):
|
||||
//
|
||||
// - Pass A(mapping):遍历 ModelMapping
|
||||
// - 精确 src → target:显示名 = src(用户视角),定价用 target 在同 platform 定价里查
|
||||
// (mapping 改写后实际计费的是 target;这是用户感知的"实际花费")。
|
||||
// target 为空或为通配符时退化为按 src 自查。
|
||||
// - 通配符 src(如 "claude-3-*"):用同 platform 定价里前缀匹配的模型作为候选展开,
|
||||
// 每个候选用自身定价(通配符场景一般是 passthrough,target 通常也是通配符)。
|
||||
// - "*" 单独 mapping key 走通配符分支(前缀为空 → 全展开)。
|
||||
// - Pass B(pricing-only):遍历 ModelPricing 中所有非通配符模型,对未在 Pass A 添加过的
|
||||
// 补齐——显示名 = 定价模型名,定价 = 自身(这是关键修复:定价存在即代表渠道支持该模型,
|
||||
// 即使没配映射)。
|
||||
//
|
||||
// 显示名命中定价时使用**定价的原始大小写**(定价是模型身份的事实来源)。
|
||||
// 按 (Platform, Name) 稳定排序,按 (Platform, lowercase(Name)) 去重,先到者胜出。
|
||||
//
|
||||
// 注意:定价仅在 channel.ModelPricing 内查找——全局 LiteLLM 回落由调用方
|
||||
// (`ChannelService.ListAvailable`)在合成展示数据时叠加。
|
||||
func (c *Channel) SupportedModels() []SupportedModel {
|
||||
if c == nil {
|
||||
return nil
|
||||
}
|
||||
if len(c.ModelMapping) == 0 && len(c.ModelPricing) == 0 {
|
||||
return nil
|
||||
}
|
||||
|
||||
idx := buildPricingIndex(c.ModelPricing)
|
||||
|
||||
type dedupKey struct {
|
||||
platform string
|
||||
name string
|
||||
}
|
||||
seen := make(map[dedupKey]struct{})
|
||||
result := make([]SupportedModel, 0)
|
||||
|
||||
// lookup 在 platform pricing index 中按精确名查定价,命中时返回定价大小写。
|
||||
lookup := func(pidx *platformPricingIndex, name string) (display string, pricing *ChannelModelPricing) {
|
||||
if pidx == nil || name == "" {
|
||||
return name, nil
|
||||
}
|
||||
lower := strings.ToLower(name)
|
||||
if p, ok := pidx.byLower[lower]; ok {
|
||||
return pidx.originalCase[lower], p
|
||||
}
|
||||
return name, nil
|
||||
}
|
||||
|
||||
add := func(platform, displayName string, pricing *ChannelModelPricing) {
|
||||
key := dedupKey{platform: platform, name: strings.ToLower(displayName)}
|
||||
if _, ok := seen[key]; ok {
|
||||
return
|
||||
}
|
||||
seen[key] = struct{}{}
|
||||
result = append(result, SupportedModel{
|
||||
Name: displayName,
|
||||
Platform: platform,
|
||||
Pricing: pricing,
|
||||
})
|
||||
}
|
||||
|
||||
// Pass A:从 mapping 展开
|
||||
for platform, mapping := range c.ModelMapping {
|
||||
if len(mapping) == 0 {
|
||||
continue
|
||||
}
|
||||
pidx := idx[platform]
|
||||
for src, target := range mapping {
|
||||
prefix, isWild := splitWildcardSuffix(src)
|
||||
if isWild {
|
||||
if pidx == nil {
|
||||
continue
|
||||
}
|
||||
prefixLower := strings.ToLower(prefix)
|
||||
for _, candidate := range pidx.names {
|
||||
if strings.HasPrefix(strings.ToLower(candidate), prefixLower) {
|
||||
display, pricing := lookup(pidx, candidate)
|
||||
add(platform, display, pricing)
|
||||
}
|
||||
}
|
||||
continue
|
||||
}
|
||||
// 精确 mapping:定价按 target 查;target 缺失/通配则退化按 src 查
|
||||
pricingKey := target
|
||||
if pricingKey == "" {
|
||||
pricingKey = src
|
||||
}
|
||||
if _, targetWild := splitWildcardSuffix(pricingKey); targetWild {
|
||||
pricingKey = src
|
||||
}
|
||||
_, pricing := lookup(pidx, pricingKey)
|
||||
// 显示名优先用 src 在定价里的原始大小写(若 src 本身是个定价模型名)
|
||||
displayName, _ := lookup(pidx, src)
|
||||
add(platform, displayName, pricing)
|
||||
}
|
||||
}
|
||||
|
||||
// Pass B:从 pricing 补齐 mapping 未覆盖的具体模型(修复"定价存在但没配映射 → 不显示")
|
||||
for platform, pidx := range idx {
|
||||
for _, name := range pidx.names {
|
||||
display, pricing := lookup(pidx, name)
|
||||
add(platform, display, pricing)
|
||||
}
|
||||
}
|
||||
|
||||
sort.SliceStable(result, func(i, j int) bool {
|
||||
if result[i].Platform != result[j].Platform {
|
||||
return result[i].Platform < result[j].Platform
|
||||
}
|
||||
return result[i].Name < result[j].Name
|
||||
})
|
||||
return result
|
||||
}
|
||||
Reference in New Issue
Block a user