mirror of
https://github.com/Wei-Shaw/sub2api.git
synced 2026-10-07 16:48:45 +08:00
shaw
6c9b84cc7a
模型登记：/v1/models 清单、Bedrock 默认映射（us.anthropic.claude-opus-5-v1）、 定价条目（$5/$25 per MTok、1M 上下文、128K 输出）、前端模型清单与 Anthropic/Bedrock 预设映射、限流 scope 简称。 同时修复两个会静默出错的问题： - 定价家族兜底 3 倍超收：定价数据缺 claude-opus-5 时，matchByModelFamily 的 Phase 2 关键字兜底会落到 opus-4 系列、getFallbackPricing 会落到 claude-3-opus，两条路都按 $15/$75 计费（官方 $5/$25），输入输出双双 3 倍超收且无任何报错。两处补 opus-5 家族并回退到同价的 4.8；判断用 opus-5/opus5 子串而非裸 "5"，避免误伤 claude-opus-4-5。顺带补齐兜底表 缺失的 claude-opus-4.8（此前同样会掉到 claude-3-opus）。 - Bedrock 版本闸门降级：claudeVersionRe 强制要求 major-minor 两段版本号， 只有主版本号的 claude-opus-5 / claude-sonnet-5 完全不匹配，被当成旧模型： isBedrockOpus47OrNewer 假导致 thinking.enabled 不转 adaptive（Opus 5 上游 已移除 budget_tokens，透传直接 400）、isBedrockClaude45OrNewer 假导致 cache_control.ttl 被剥离、bedrockModelSupportsToolSearch 假导致 tool search 被过滤。改为 minor 可选（缺省 minor=0），claude-sonnet-5 的同一问题一并修复。 Vertex 无需改动：normalizeVertexAnthropicModelID 只处理 -YYYYMMDD→@YYYYMMDD， 无日期后缀的裸 ID 原样透传即正确。context-1m-2025-08-07 白名单不动：Opus 系 上游不接受该 beta，且 Opus 5 的 1M 上下文是默认能力。 Antigravity 暂不接入：无上游支持证据，mapAntigravityModel 对未映射模型返回 空字符串即"该账号不支持"，fails closed 安全。 回归测试 internal/service/claude_opus5_test.go 覆盖定价两层兜底、Bedrock 三个闸门、thinking 转换与模型清单；逐个回退上述修复已确认测试会红。
157 lines
6.6 KiB
Go
157 lines
6.6 KiB
Go
package domain
|
||
|
||
// Status constants
|
||
const (
|
||
StatusActive = "active"
|
||
StatusDisabled = "disabled"
|
||
StatusError = "error"
|
||
StatusUnused = "unused"
|
||
StatusUsed = "used"
|
||
StatusExpired = "expired"
|
||
)
|
||
|
||
// Role constants
|
||
const (
|
||
RoleAdmin = "admin"
|
||
RoleUser = "user"
|
||
)
|
||
|
||
// Platform constants
|
||
const (
|
||
PlatformAnthropic = "anthropic"
|
||
PlatformOpenAI = "openai"
|
||
PlatformGemini = "gemini"
|
||
PlatformAntigravity = "antigravity"
|
||
PlatformGrok = "grok"
|
||
PlatformComposite = "composite"
|
||
)
|
||
|
||
// Account type constants
|
||
const (
|
||
AccountTypeOAuth = "oauth" // OAuth类型账号（full scope: profile + inference）
|
||
AccountTypeSetupToken = "setup-token" // Setup Token类型账号（inference only scope）
|
||
AccountTypeAPIKey = "apikey" // API Key类型账号
|
||
AccountTypeUpstream = "upstream" // 上游透传类型账号（通过 Base URL + API Key 连接上游）
|
||
AccountTypeBedrock = "bedrock" // AWS Bedrock 类型账号（通过 SigV4 签名或 API Key 连接 Bedrock，由 credentials.auth_mode 区分）
|
||
AccountTypeServiceAccount = "service_account" // Google Service Account 类型账号（用于 Vertex AI）
|
||
)
|
||
|
||
// Redeem type constants
|
||
const (
|
||
RedeemTypeBalance = "balance"
|
||
RedeemTypeConcurrency = "concurrency"
|
||
RedeemTypeSubscription = "subscription"
|
||
RedeemTypeInvitation = "invitation"
|
||
)
|
||
|
||
// PromoCode status constants
|
||
const (
|
||
PromoCodeStatusActive = "active"
|
||
PromoCodeStatusDisabled = "disabled"
|
||
)
|
||
|
||
// Admin adjustment type constants
|
||
const (
|
||
AdjustmentTypeAdminBalance = "admin_balance" // 管理员调整余额
|
||
AdjustmentTypeAdminConcurrency = "admin_concurrency" // 管理员调整并发数
|
||
)
|
||
|
||
// Group subscription type constants
|
||
const (
|
||
SubscriptionTypeStandard = "standard" // 标准计费模式（按余额扣费）
|
||
SubscriptionTypeSubscription = "subscription" // 订阅模式（按限额控制）
|
||
)
|
||
|
||
// Subscription status constants
|
||
const (
|
||
SubscriptionStatusActive = "active"
|
||
SubscriptionStatusExpired = "expired"
|
||
SubscriptionStatusSuspended = "suspended"
|
||
)
|
||
|
||
// AntigravityGemini31ProAgentModel is the upstream route for Gemini 3.1 Pro High.
|
||
const AntigravityGemini31ProAgentModel = "gemini-pro-agent"
|
||
|
||
// DefaultAntigravityModelMapping 是 Antigravity 平台的默认模型映射
|
||
// 当账号未配置 model_mapping 时使用此默认值
|
||
// 与前端 useModelWhitelist.ts 中的 antigravityDefaultMappings 保持一致
|
||
var DefaultAntigravityModelMapping = map[string]string{
|
||
// Claude 白名单
|
||
"claude-fable-5": "claude-fable-5", // 官方模型
|
||
"claude-opus-4-8": "claude-opus-4-8", // 官方模型
|
||
"claude-opus-4-7": "claude-opus-4-7", // 官方模型
|
||
"claude-opus-4-6-thinking": "claude-opus-4-6-thinking", // 官方模型
|
||
"claude-opus-4-6": "claude-opus-4-6-thinking", // 简称映射
|
||
"claude-opus-4-5-thinking": "claude-opus-4-6-thinking", // 迁移旧模型
|
||
"claude-sonnet-4-6": "claude-sonnet-4-6",
|
||
"claude-sonnet-4-5": "claude-sonnet-4-5",
|
||
"claude-sonnet-4-5-thinking": "claude-sonnet-4-5-thinking",
|
||
// Claude 详细版本 ID 映射
|
||
"claude-opus-4-5-20251101": "claude-opus-4-6-thinking", // 迁移旧模型
|
||
"claude-sonnet-4-5-20250929": "claude-sonnet-4-5",
|
||
// Claude Haiku → Sonnet（无 Haiku 支持）
|
||
"claude-haiku-4-5": "claude-sonnet-4-6",
|
||
"claude-haiku-4-5-20251001": "claude-sonnet-4-6",
|
||
// Gemini 2.5 白名单
|
||
"gemini-2.5-flash": "gemini-2.5-flash",
|
||
"gemini-2.5-flash-image": "gemini-2.5-flash-image",
|
||
"gemini-2.5-flash-image-preview": "gemini-2.5-flash-image",
|
||
"gemini-2.5-flash-lite": "gemini-2.5-flash-lite",
|
||
"gemini-2.5-flash-thinking": "gemini-2.5-flash-thinking",
|
||
"gemini-2.5-pro": "gemini-2.5-pro",
|
||
// Gemini 3 白名单
|
||
"gemini-3-flash": "gemini-3-flash",
|
||
"gemini-3-pro-high": "gemini-3-pro-high",
|
||
"gemini-3-pro-low": "gemini-3-pro-low",
|
||
// Gemini 3 preview 映射
|
||
"gemini-3-flash-preview": "gemini-3-flash",
|
||
"gemini-3-pro-preview": "gemini-3-pro-high",
|
||
// Gemini 3.1 白名单
|
||
AntigravityGemini31ProAgentModel: AntigravityGemini31ProAgentModel,
|
||
"gemini-3.1-pro": AntigravityGemini31ProAgentModel,
|
||
"gemini-3.1-pro-high": AntigravityGemini31ProAgentModel,
|
||
"gemini-3.1-pro-low": "gemini-3.1-pro-low",
|
||
// Gemini 3.1 preview 映射
|
||
"gemini-3.1-pro-preview": AntigravityGemini31ProAgentModel,
|
||
// Gemini 3.1 image 白名单
|
||
"gemini-3.1-flash-image": "gemini-3.1-flash-image",
|
||
// Gemini 3.1 image preview 映射
|
||
"gemini-3.1-flash-image-preview": "gemini-3.1-flash-image",
|
||
// Gemini 3 image 兼容映射（向 3.1 image 迁移）
|
||
"gemini-3-pro-image": "gemini-3.1-flash-image",
|
||
"gemini-3-pro-image-preview": "gemini-3.1-flash-image",
|
||
// 其他官方模型
|
||
"gpt-oss-120b-medium": "gpt-oss-120b-medium",
|
||
"tab_flash_lite_preview": "tab_flash_lite_preview",
|
||
}
|
||
|
||
// DefaultBedrockModelMapping 是 AWS Bedrock 平台的默认模型映射
|
||
// 将 Anthropic 标准模型名映射到 Bedrock 模型 ID
|
||
// 注意：此处的 "us." 前缀仅为默认值，ResolveBedrockModelID 会根据账号配置的
|
||
// aws_region 自动调整为匹配的区域前缀（如 eu.、apac.、jp. 等）
|
||
var DefaultBedrockModelMapping = map[string]string{
|
||
// Claude Fable
|
||
"claude-fable-5": "anthropic.claude-fable-5",
|
||
// Claude Opus
|
||
"claude-opus-5": "us.anthropic.claude-opus-5-v1",
|
||
"claude-opus-4-8": "us.anthropic.claude-opus-4-8-v1",
|
||
"claude-opus-4-7": "us.anthropic.claude-opus-4-7-v1",
|
||
"claude-opus-4-6-thinking": "us.anthropic.claude-opus-4-6-v1",
|
||
"claude-opus-4-6": "us.anthropic.claude-opus-4-6-v1",
|
||
"claude-opus-4-5-thinking": "us.anthropic.claude-opus-4-5-20251101-v1:0",
|
||
"claude-opus-4-5-20251101": "us.anthropic.claude-opus-4-5-20251101-v1:0",
|
||
"claude-opus-4-1": "us.anthropic.claude-opus-4-1-20250805-v1:0",
|
||
"claude-opus-4-20250514": "us.anthropic.claude-opus-4-20250514-v1:0",
|
||
// Claude Sonnet
|
||
"claude-sonnet-5": "us.anthropic.claude-sonnet-5-v1",
|
||
"claude-sonnet-4-6-thinking": "us.anthropic.claude-sonnet-4-6",
|
||
"claude-sonnet-4-6": "us.anthropic.claude-sonnet-4-6",
|
||
"claude-sonnet-4-5": "us.anthropic.claude-sonnet-4-5-20250929-v1:0",
|
||
"claude-sonnet-4-5-thinking": "us.anthropic.claude-sonnet-4-5-20250929-v1:0",
|
||
"claude-sonnet-4-5-20250929": "us.anthropic.claude-sonnet-4-5-20250929-v1:0",
|
||
"claude-sonnet-4-20250514": "us.anthropic.claude-sonnet-4-20250514-v1:0",
|
||
// Claude Haiku
|
||
"claude-haiku-4-5": "us.anthropic.claude-haiku-4-5-20251001-v1:0",
|
||
"claude-haiku-4-5-20251001": "us.anthropic.claude-haiku-4-5-20251001-v1:0",
|
||
}
|