mirror of
https://github.com/Wei-Shaw/sub2api.git
synced 2026-10-16 15:43:13 +08:00
DaydreamCodingandClaude Opus 4.8
819fda34d9
将 codex_cli_only 客户端识别从「单一 strict 开关 + 固定 OR 头集合」重构为 可逐项管理的引擎指纹信号列表，加固整条判定链，并补齐账号级 app-server 控制、 对齐前端设置文案。 判定链（每步可短路）： - 账号未开 codex_cli_only → 不限制；gateway.force_codex_cli → 旁路放行 - 全局黑名单命中（OR 宽 deny）→ 立即拒 - 身份候选：官方 UA（strict，仅前缀）/ 官方 originator（OR）/ 全局白名单（双因子 AND） / 全局 app-server 开关 OR 账号 app-server 开关；均不命中 → 拒 - 版本门（仅官方候选）：UA 须可解析引擎版本，再校验 [min,max] 区间 - 引擎指纹 AND 硬门：按信号列表逐条勾选 AND、每条行内变体 OR；无 Required 信号 → 放行 引擎指纹信号列表（唯一真源） - 新增 openai.EngineFingerprintSignal 类型 + EvaluateEngineFingerprint 求值器 （勾选 AND / 行内变体 OR / 无勾选 → 放行） - CodexRestrictionPolicy 增 EngineFingerprintSignals；信号列表单一决定是否启用指纹门， 不再保留独立「要求引擎指纹」总开关（与「信号全不选」语义重复） - 新设置键 codex_cli_only_engine_fingerprint_signals（默认只勾 x-codex- 前缀）； 旧 body 指纹开关幂等迁移并入信号列表；wire 接线 - 黑/白名单自由条目、命名预设、版本区间 全局设置管线 - gateway 缺 settingService（仅测试/误配可达）时指纹门回退默认种子信号、失败关闭， 不再因零值 policy（nil 信号）失败开放 账号级 Codex app-server（替换已失效的 ClaudeCode 放行机制） - account.IsCodexCLIOnlyAppServerAllowed() 读 extra.codex_cli_only_allow_app_server， 仅在 codex_cli_only 开启时生效；候选身份门「全局 OR 账号」，与旧系统双层控制对齐 - 移除已无入口的 claude_code 预设机制（allowedClientRegistry / MatchAllowedClients / 账号 GetCodexCLIOnlyAllowedClients / reason）；白名单 AllowedClientEntry / IsAllowedClientMatch 保留 门加固（反伪 + 写入校验） - 官方 UA 访问门改 strict：IsCodexOfficialClientRequestStrict 仅前缀匹配，收窄「浏览器前缀 + 中段 codex token」伪造面（strict 仍保留 Codex 家族前缀与 UA 尾部兜底，故对「任意前缀 + 官方尾部 (name;ver)」仍放行——与 UA 可伪造、真正反伪靠引擎指纹门的设计一致） - 官方客户端识别扩展：新增 codex-tui/、codex_vscode_copilot/ 前缀 + UA 尾部 (name;ver) 兜底 （恢复 CODEX_INTERNAL_ORIGINATOR_OVERRIDE 的真实 client，如 cccc→codex-tui），originator 改 精确集。该识别经 IsCodexOfficialClientByHeaders 被 passthrough 复用，故透传的官方判定一并 修正（codex-tui 等不再被误改写 UA）——非「行为不变」，属有意修正 - 白名单写入校验 ValidateCodexWhitelistEntriesJSON + AllowedClientEntry.IsWhitelistable： 双因子 AND 条目须可命中（非空 originator + 非空 ua_contains），拒绝写入会静默失效的死规则； 黑名单（OR 宽 deny，允许 originator-only）不受约束 管理端 / 前端 - handler / DTO / settings_view / 契约测试；gateway 接入判定链 - 信号列表编辑器（替换 body 开关）、api 类型、SettingsView；无勾选给常驻警告 - Create/Edit/Bulk 三弹窗「Codex Only」下新增 app-server 开关（OR 合并全局） - 文案：UA/Originator → User-Agent/Originator；黑/白名单重命名为 User-Agent/Originator 黑/白名单； 「允许 App Server 第三方客户端」→「Codex app-server」+ 简介示例；i18n zh/en 同步 - 移除死代码 HasCodex*Fingerprint helper 测试：引擎指纹求值器 / 账号 app-server（OR 语义）/ detector（含 N1 strict、失败关闭）/ 白名单写入校验 / BulkEdit spec 等；后端 build + service/openai/admin 单测全绿，前端 vue-tsc + vitest 全绿。 Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
546 lines
21 KiB
Go
546 lines
21 KiB
Go
package service
|
||
|
||
import "strings"
|
||
|
||
func firstNonEmpty(values ...string) string {
|
||
for _, value := range values {
|
||
if trimmed := strings.TrimSpace(value); trimmed != "" {
|
||
return trimmed
|
||
}
|
||
}
|
||
return ""
|
||
}
|
||
|
||
type SystemSettings struct {
|
||
RegistrationEnabled bool
|
||
EmailVerifyEnabled bool
|
||
RegistrationEmailSuffixWhitelist []string
|
||
PromoCodeEnabled bool
|
||
PasswordResetEnabled bool
|
||
FrontendURL string
|
||
InvitationCodeEnabled bool
|
||
TotpEnabled bool // TOTP 双因素认证
|
||
LoginAgreementEnabled bool
|
||
LoginAgreementMode string
|
||
LoginAgreementUpdatedAt string
|
||
LoginAgreementDocuments []LoginAgreementDocument
|
||
|
||
SMTPHost string
|
||
SMTPPort int
|
||
SMTPUsername string
|
||
SMTPPassword string
|
||
SMTPPasswordConfigured bool
|
||
SMTPFrom string
|
||
SMTPFromName string
|
||
SMTPUseTLS bool
|
||
|
||
TurnstileEnabled bool
|
||
TurnstileSiteKey string
|
||
TurnstileSecretKey string
|
||
TurnstileSecretKeyConfigured bool
|
||
APIKeyACLTrustForwardedIP bool
|
||
|
||
// LinuxDo Connect OAuth 登录
|
||
LinuxDoConnectEnabled bool
|
||
LinuxDoConnectClientID string
|
||
LinuxDoConnectClientSecret string
|
||
LinuxDoConnectClientSecretConfigured bool
|
||
LinuxDoConnectRedirectURL string
|
||
|
||
// DingTalk Connect OAuth 登录
|
||
DingTalkConnectEnabled bool
|
||
DingTalkConnectClientID string
|
||
DingTalkConnectClientSecret string
|
||
DingTalkConnectClientSecretConfigured bool
|
||
DingTalkConnectRedirectURL string
|
||
DingTalkConnectCorpRestrictionPolicy string
|
||
DingTalkConnectInternalCorpID string
|
||
DingTalkConnectBypassRegistration bool
|
||
DingTalkConnectSyncCorpEmail bool
|
||
DingTalkConnectSyncDisplayName bool
|
||
DingTalkConnectSyncDept bool
|
||
DingTalkConnectSyncCorpEmailAttrKey string
|
||
DingTalkConnectSyncDisplayNameAttrKey string
|
||
DingTalkConnectSyncDeptAttrKey string
|
||
DingTalkConnectSyncCorpEmailAttrName string
|
||
DingTalkConnectSyncDisplayNameAttrName string
|
||
DingTalkConnectSyncDeptAttrName string
|
||
|
||
// WeChat Connect OAuth 登录
|
||
WeChatConnectEnabled bool
|
||
WeChatConnectAppID string
|
||
WeChatConnectAppSecret string
|
||
WeChatConnectAppSecretConfigured bool
|
||
WeChatConnectOpenAppID string
|
||
WeChatConnectOpenAppSecret string
|
||
WeChatConnectOpenAppSecretConfigured bool
|
||
WeChatConnectMPAppID string
|
||
WeChatConnectMPAppSecret string
|
||
WeChatConnectMPAppSecretConfigured bool
|
||
WeChatConnectMobileAppID string
|
||
WeChatConnectMobileAppSecret string
|
||
WeChatConnectMobileAppSecretConfigured bool
|
||
WeChatConnectOpenEnabled bool
|
||
WeChatConnectMPEnabled bool
|
||
WeChatConnectMobileEnabled bool
|
||
WeChatConnectMode string
|
||
WeChatConnectScopes string
|
||
WeChatConnectRedirectURL string
|
||
WeChatConnectFrontendRedirectURL string
|
||
|
||
// Generic OIDC OAuth 登录
|
||
OIDCConnectEnabled bool
|
||
OIDCConnectProviderName string
|
||
OIDCConnectClientID string
|
||
OIDCConnectClientSecret string
|
||
OIDCConnectClientSecretConfigured bool
|
||
OIDCConnectIssuerURL string
|
||
OIDCConnectDiscoveryURL string
|
||
OIDCConnectAuthorizeURL string
|
||
OIDCConnectTokenURL string
|
||
OIDCConnectUserInfoURL string
|
||
OIDCConnectJWKSURL string
|
||
OIDCConnectScopes string
|
||
OIDCConnectRedirectURL string
|
||
OIDCConnectFrontendRedirectURL string
|
||
OIDCConnectTokenAuthMethod string
|
||
OIDCConnectUsePKCE bool
|
||
OIDCConnectValidateIDToken bool
|
||
OIDCConnectAllowedSigningAlgs string
|
||
OIDCConnectClockSkewSeconds int
|
||
OIDCConnectRequireEmailVerified bool
|
||
OIDCConnectUserInfoEmailPath string
|
||
OIDCConnectUserInfoIDPath string
|
||
OIDCConnectUserInfoUsernamePath string
|
||
|
||
// GitHub / Google 邮箱快捷登录
|
||
GitHubOAuthEnabled bool
|
||
GitHubOAuthClientID string
|
||
GitHubOAuthClientSecret string
|
||
GitHubOAuthClientSecretConfigured bool
|
||
GitHubOAuthRedirectURL string
|
||
GitHubOAuthFrontendRedirectURL string
|
||
GoogleOAuthEnabled bool
|
||
GoogleOAuthClientID string
|
||
GoogleOAuthClientSecret string
|
||
GoogleOAuthClientSecretConfigured bool
|
||
GoogleOAuthRedirectURL string
|
||
GoogleOAuthFrontendRedirectURL string
|
||
|
||
SiteName string
|
||
SiteLogo string
|
||
SiteSubtitle string
|
||
APIBaseURL string
|
||
ContactInfo string
|
||
DocURL string
|
||
HomeContent string
|
||
HideCcsImportButton bool
|
||
PurchaseSubscriptionEnabled bool
|
||
PurchaseSubscriptionURL string
|
||
TableDefaultPageSize int
|
||
TablePageSizeOptions []int
|
||
CustomMenuItems string // JSON array of custom menu items
|
||
CustomEndpoints string // JSON array of custom endpoints
|
||
|
||
DefaultConcurrency int
|
||
DefaultBalance float64
|
||
RiskControlEnabled bool
|
||
CyberSessionBlockEnabled bool
|
||
CyberSessionBlockTTLSeconds int
|
||
AffiliateEnabled bool
|
||
AffiliateRebateRate float64
|
||
AffiliateRebateFreezeHours int
|
||
AffiliateRebateDurationDays int
|
||
AffiliateRebatePerInviteeCap float64
|
||
DefaultUserRPMLimit int
|
||
DefaultSubscriptions []DefaultSubscriptionSetting
|
||
|
||
// Model fallback configuration
|
||
EnableModelFallback bool `json:"enable_model_fallback"`
|
||
FallbackModelAnthropic string `json:"fallback_model_anthropic"`
|
||
FallbackModelOpenAI string `json:"fallback_model_openai"`
|
||
FallbackModelGemini string `json:"fallback_model_gemini"`
|
||
FallbackModelAntigravity string `json:"fallback_model_antigravity"`
|
||
|
||
// Identity patch configuration (Claude -> Gemini)
|
||
EnableIdentityPatch bool `json:"enable_identity_patch"`
|
||
IdentityPatchPrompt string `json:"identity_patch_prompt"`
|
||
|
||
// Ops monitoring (vNext)
|
||
OpsMonitoringEnabled bool
|
||
OpsRealtimeMonitoringEnabled bool
|
||
OpsQueryModeDefault string
|
||
OpsMetricsIntervalSeconds int
|
||
|
||
// Channel Monitor feature
|
||
ChannelMonitorEnabled bool `json:"channel_monitor_enabled"`
|
||
ChannelMonitorDefaultIntervalSeconds int `json:"channel_monitor_default_interval_seconds"`
|
||
|
||
// Available Channels feature (user-facing aggregate view)
|
||
AvailableChannelsEnabled bool `json:"available_channels_enabled"`
|
||
|
||
// Claude Code version check
|
||
MinClaudeCodeVersion string
|
||
MaxClaudeCodeVersion string
|
||
|
||
// 分组隔离：允许未分组 Key 调度（默认 false → 403）
|
||
AllowUngroupedKeyScheduling bool
|
||
|
||
// Backend 模式：禁用用户注册和自助服务，仅管理员可登录
|
||
BackendModeEnabled bool
|
||
|
||
// Gateway forwarding behavior
|
||
EnableFingerprintUnification bool // 是否统一 OAuth 账号的指纹头（默认 true）
|
||
EnableMetadataPassthrough bool // 是否透传客户端原始 metadata（默认 false）
|
||
EnableCCHSigning bool // 已废弃 no-op：新版 CLI 取消 cch 签名后网关不再注入/签名 cch，开关无效果
|
||
EnableClaudeOAuthSystemPromptInjection bool // 是否对 Claude OAuth mimic 路径注入 Claude Code system blocks（默认 true）
|
||
ClaudeOAuthSystemPrompt string // Claude OAuth mimic 路径注入的通用扩展 system prompt；空值使用内置默认
|
||
ClaudeOAuthSystemPromptBlocks string // Claude OAuth mimic 路径注入的 system blocks JSON 配置；空值使用内置默认
|
||
EnableAnthropicCacheTTL1hInjection bool // 是否对 Anthropic OAuth/SetupToken 请求体注入 1h cache_control ttl（默认 false）
|
||
RewriteMessageCacheControl bool // 是否改写 messages[*].content[*].cache_control（默认 false）
|
||
AntigravityUserAgentVersion string // Antigravity 上游 User-Agent 版本号；空值使用配置/默认值
|
||
OpenAICodexUserAgent string // OpenAI Codex 上游完整 User-Agent；空值使用内置默认
|
||
MinCodexVersion string // codex_cli_only 最低 Codex 引擎版本；空=不检查
|
||
MaxCodexVersion string // codex_cli_only 最高 Codex 引擎版本；空=不检查
|
||
CodexCLIOnlyBlacklist string // codex_cli_only 全局黑名单 JSON（[]AllowedClientEntry，OR deny）
|
||
CodexCLIOnlyWhitelist string // codex_cli_only 全局白名单 JSON（[]AllowedClientEntry，AND allow）
|
||
CodexCLIOnlyAllowAppServerClients bool // codex_cli_only App Server 开关：对未列名客户端开闸（默认 false）
|
||
CodexCLIOnlyEngineFingerprintSignals string // codex_cli_only 引擎指纹门信号列表 JSON（[]EngineFingerprintSignal）
|
||
|
||
// Web Search Emulation
|
||
WebSearchEmulationEnabled bool // 是否启用 web search 模拟
|
||
|
||
// Payment visible method routing
|
||
PaymentVisibleMethodAlipaySource string
|
||
PaymentVisibleMethodWxpaySource string
|
||
PaymentVisibleMethodAlipayEnabled bool
|
||
PaymentVisibleMethodWxpayEnabled bool
|
||
|
||
// OpenAI 账号调度
|
||
OpenAIAdvancedSchedulerEnabled bool
|
||
|
||
// 余额不足提醒
|
||
BalanceLowNotifyEnabled bool
|
||
BalanceLowNotifyThreshold float64
|
||
BalanceLowNotifyRechargeURL string
|
||
|
||
// 订阅到期提醒
|
||
SubscriptionExpiryNotifyEnabled bool
|
||
|
||
// 账号限额通知
|
||
AccountQuotaNotifyEnabled bool
|
||
AccountQuotaNotifyEmails []NotifyEmailEntry
|
||
|
||
// 系统全局默认平台配额（key = platform，nil/缺省 = 不限制）
|
||
DefaultPlatformQuotas map[string]*DefaultPlatformQuotaSetting `json:"default_platform_quotas"`
|
||
|
||
// 允许终端用户在用量页查看自己的失败请求
|
||
AllowUserViewErrorRequests bool
|
||
}
|
||
|
||
type DefaultSubscriptionSetting struct {
|
||
GroupID int64 `json:"group_id"`
|
||
ValidityDays int `json:"validity_days"`
|
||
}
|
||
|
||
type PublicSettings struct {
|
||
RegistrationEnabled bool
|
||
EmailVerifyEnabled bool
|
||
ForceEmailOnThirdPartySignup bool
|
||
RegistrationEmailSuffixWhitelist []string
|
||
PromoCodeEnabled bool
|
||
PasswordResetEnabled bool
|
||
InvitationCodeEnabled bool
|
||
TotpEnabled bool // TOTP 双因素认证
|
||
LoginAgreementEnabled bool
|
||
LoginAgreementMode string
|
||
LoginAgreementUpdatedAt string
|
||
LoginAgreementRevision string
|
||
LoginAgreementDocuments []LoginAgreementDocument
|
||
TurnstileEnabled bool
|
||
TurnstileSiteKey string
|
||
SiteName string
|
||
SiteLogo string
|
||
SiteSubtitle string
|
||
APIBaseURL string
|
||
ContactInfo string
|
||
DocURL string
|
||
HomeContent string
|
||
HideCcsImportButton bool
|
||
|
||
PurchaseSubscriptionEnabled bool
|
||
PurchaseSubscriptionURL string
|
||
TableDefaultPageSize int
|
||
TablePageSizeOptions []int
|
||
CustomMenuItems string // JSON array of custom menu items
|
||
CustomEndpoints string // JSON array of custom endpoints
|
||
|
||
LinuxDoOAuthEnabled bool
|
||
DingTalkOAuthEnabled bool
|
||
WeChatOAuthEnabled bool
|
||
WeChatOAuthOpenEnabled bool
|
||
WeChatOAuthMPEnabled bool
|
||
WeChatOAuthMobileEnabled bool
|
||
BackendModeEnabled bool
|
||
PaymentEnabled bool
|
||
OIDCOAuthEnabled bool
|
||
OIDCOAuthProviderName string
|
||
GitHubOAuthEnabled bool
|
||
GoogleOAuthEnabled bool
|
||
Version string
|
||
|
||
BalanceLowNotifyEnabled bool
|
||
AccountQuotaNotifyEnabled bool
|
||
BalanceLowNotifyThreshold float64
|
||
BalanceLowNotifyRechargeURL string
|
||
|
||
// Channel Monitor feature
|
||
ChannelMonitorEnabled bool `json:"channel_monitor_enabled"`
|
||
ChannelMonitorDefaultIntervalSeconds int `json:"channel_monitor_default_interval_seconds"`
|
||
|
||
// Available Channels feature (user-facing aggregate view)
|
||
AvailableChannelsEnabled bool `json:"available_channels_enabled"`
|
||
|
||
// Affiliate (邀请返利) feature toggle
|
||
AffiliateEnabled bool `json:"affiliate_enabled"`
|
||
|
||
// 风控中心功能开关
|
||
RiskControlEnabled bool `json:"risk_control_enabled"`
|
||
|
||
// 允许终端用户在用量页查看自己的失败请求
|
||
AllowUserViewErrorRequests bool `json:"allow_user_view_error_requests"`
|
||
}
|
||
|
||
type LoginAgreementDocument struct {
|
||
ID string `json:"id"`
|
||
Title string `json:"title"`
|
||
ContentMD string `json:"content_md"`
|
||
}
|
||
|
||
type WeChatConnectOAuthConfig struct {
|
||
Enabled bool
|
||
LegacyAppID string
|
||
LegacyAppSecret string
|
||
OpenAppID string
|
||
OpenAppSecret string
|
||
MPAppID string
|
||
MPAppSecret string
|
||
MobileAppID string
|
||
MobileAppSecret string
|
||
OpenEnabled bool
|
||
MPEnabled bool
|
||
MobileEnabled bool
|
||
Mode string
|
||
Scopes string
|
||
RedirectURL string
|
||
FrontendRedirectURL string
|
||
}
|
||
|
||
func (cfg WeChatConnectOAuthConfig) SupportsMode(mode string) bool {
|
||
switch normalizeWeChatConnectModeSetting(mode) {
|
||
case "mp":
|
||
return cfg.MPEnabled
|
||
case "mobile":
|
||
return cfg.MobileEnabled
|
||
default:
|
||
return cfg.OpenEnabled
|
||
}
|
||
}
|
||
|
||
func (cfg WeChatConnectOAuthConfig) ScopeForMode(mode string) string {
|
||
switch normalizeWeChatConnectModeSetting(mode) {
|
||
case "mp":
|
||
return normalizeWeChatConnectScopeSetting(cfg.Scopes, "mp")
|
||
case "mobile":
|
||
return ""
|
||
}
|
||
return defaultWeChatConnectScopeForMode("open")
|
||
}
|
||
|
||
func (cfg WeChatConnectOAuthConfig) AppIDForMode(mode string) string {
|
||
switch normalizeWeChatConnectModeSetting(mode) {
|
||
case "mp":
|
||
return strings.TrimSpace(firstNonEmpty(cfg.MPAppID, cfg.LegacyAppID))
|
||
case "mobile":
|
||
return strings.TrimSpace(firstNonEmpty(cfg.MobileAppID, cfg.LegacyAppID))
|
||
}
|
||
return strings.TrimSpace(firstNonEmpty(cfg.OpenAppID, cfg.LegacyAppID))
|
||
}
|
||
|
||
func (cfg WeChatConnectOAuthConfig) AppSecretForMode(mode string) string {
|
||
switch normalizeWeChatConnectModeSetting(mode) {
|
||
case "mp":
|
||
return strings.TrimSpace(firstNonEmpty(cfg.MPAppSecret, cfg.LegacyAppSecret))
|
||
case "mobile":
|
||
return strings.TrimSpace(firstNonEmpty(cfg.MobileAppSecret, cfg.LegacyAppSecret))
|
||
}
|
||
return strings.TrimSpace(firstNonEmpty(cfg.OpenAppSecret, cfg.LegacyAppSecret))
|
||
}
|
||
|
||
// StreamTimeoutSettings 流超时处理配置（仅控制超时后的处理方式，超时判定由网关配置控制）
|
||
type StreamTimeoutSettings struct {
|
||
// Enabled 是否启用流超时处理
|
||
Enabled bool `json:"enabled"`
|
||
// Action 超时后的处理方式: "temp_unsched" | "error" | "none"
|
||
Action string `json:"action"`
|
||
// TempUnschedMinutes 临时不可调度持续时间（分钟）
|
||
TempUnschedMinutes int `json:"temp_unsched_minutes"`
|
||
// ThresholdCount 触发阈值次数（累计多少次超时才触发）
|
||
ThresholdCount int `json:"threshold_count"`
|
||
// ThresholdWindowMinutes 阈值窗口时间（分钟）
|
||
ThresholdWindowMinutes int `json:"threshold_window_minutes"`
|
||
}
|
||
|
||
// StreamTimeoutAction 流超时处理方式常量
|
||
const (
|
||
StreamTimeoutActionTempUnsched = "temp_unsched" // 临时不可调度
|
||
StreamTimeoutActionError = "error" // 标记为错误状态
|
||
StreamTimeoutActionNone = "none" // 不处理
|
||
)
|
||
|
||
// DefaultStreamTimeoutSettings 返回默认的流超时配置
|
||
func DefaultStreamTimeoutSettings() *StreamTimeoutSettings {
|
||
return &StreamTimeoutSettings{
|
||
Enabled: false,
|
||
Action: StreamTimeoutActionTempUnsched,
|
||
TempUnschedMinutes: 5,
|
||
ThresholdCount: 3,
|
||
ThresholdWindowMinutes: 10,
|
||
}
|
||
}
|
||
|
||
// RectifierSettings 请求整流器配置
|
||
type RectifierSettings struct {
|
||
Enabled bool `json:"enabled"` // 总开关
|
||
ThinkingSignatureEnabled bool `json:"thinking_signature_enabled"` // Thinking 签名整流
|
||
ThinkingBudgetEnabled bool `json:"thinking_budget_enabled"` // Thinking Budget 整流
|
||
APIKeySignatureEnabled bool `json:"apikey_signature_enabled"` // API Key 签名整流开关
|
||
APIKeySignaturePatterns []string `json:"apikey_signature_patterns"` // API Key 自定义匹配关键词
|
||
}
|
||
|
||
// DefaultRectifierSettings 返回默认的整流器配置（全部启用）
|
||
func DefaultRectifierSettings() *RectifierSettings {
|
||
return &RectifierSettings{
|
||
Enabled: true,
|
||
ThinkingSignatureEnabled: true,
|
||
ThinkingBudgetEnabled: true,
|
||
}
|
||
}
|
||
|
||
// Beta Policy 策略常量
|
||
const (
|
||
BetaPolicyActionPass = "pass" // 透传，不做任何处理
|
||
BetaPolicyActionFilter = "filter" // 过滤，从 beta header 中移除该 token
|
||
BetaPolicyActionBlock = "block" // 拦截，直接返回错误
|
||
|
||
BetaPolicyScopeAll = "all" // 所有账号类型
|
||
BetaPolicyScopeOAuth = "oauth" // 仅 OAuth 账号
|
||
BetaPolicyScopeAPIKey = "apikey" // 仅 API Key 账号
|
||
BetaPolicyScopeBedrock = "bedrock" // 仅 AWS Bedrock 账号
|
||
)
|
||
|
||
// BetaPolicyRule 单条 Beta 策略规则
|
||
type BetaPolicyRule struct {
|
||
BetaToken string `json:"beta_token"` // beta token 值
|
||
Action string `json:"action"` // "pass" | "filter" | "block"
|
||
Scope string `json:"scope"` // "all" | "oauth" | "apikey" | "bedrock"
|
||
ErrorMessage string `json:"error_message,omitempty"` // 自定义错误消息 (action=block 时生效)
|
||
ModelWhitelist []string `json:"model_whitelist,omitempty"` // 模型匹配模式列表（为空=对所有模型生效）
|
||
FallbackAction string `json:"fallback_action,omitempty"` // 未匹配白名单的模型的处理方式
|
||
FallbackErrorMessage string `json:"fallback_error_message,omitempty"` // 未匹配白名单时的自定义错误消息 (fallback_action=block 时生效)
|
||
}
|
||
|
||
// BetaPolicySettings Beta 策略配置
|
||
type BetaPolicySettings struct {
|
||
Rules []BetaPolicyRule `json:"rules"`
|
||
}
|
||
|
||
// OverloadCooldownSettings 529过载冷却配置
|
||
type OverloadCooldownSettings struct {
|
||
// Enabled 是否在收到529时暂停账号调度
|
||
Enabled bool `json:"enabled"`
|
||
// CooldownMinutes 冷却时长（分钟）
|
||
CooldownMinutes int `json:"cooldown_minutes"`
|
||
}
|
||
|
||
// RateLimit429CooldownSettings 429默认回避配置
|
||
type RateLimit429CooldownSettings struct {
|
||
// Enabled 是否在无法解析上游重置时间时应用默认429回避
|
||
Enabled bool `json:"enabled"`
|
||
// CooldownSeconds 默认回避时长（秒）
|
||
CooldownSeconds int `json:"cooldown_seconds"`
|
||
}
|
||
|
||
// DefaultOverloadCooldownSettings 返回默认的过载冷却配置（启用，10分钟）
|
||
func DefaultOverloadCooldownSettings() *OverloadCooldownSettings {
|
||
return &OverloadCooldownSettings{
|
||
Enabled: true,
|
||
CooldownMinutes: 10,
|
||
}
|
||
}
|
||
|
||
// DefaultRateLimit429CooldownSettings 返回默认的429回避配置（启用，5秒）
|
||
func DefaultRateLimit429CooldownSettings() *RateLimit429CooldownSettings {
|
||
return &RateLimit429CooldownSettings{
|
||
Enabled: true,
|
||
CooldownSeconds: 5,
|
||
}
|
||
}
|
||
|
||
// DefaultBetaPolicySettings 返回默认的 Beta 策略配置
|
||
func DefaultBetaPolicySettings() *BetaPolicySettings {
|
||
return &BetaPolicySettings{
|
||
Rules: []BetaPolicyRule{
|
||
{
|
||
BetaToken: "fast-mode-2026-02-01",
|
||
Action: BetaPolicyActionFilter,
|
||
Scope: BetaPolicyScopeAll,
|
||
},
|
||
{
|
||
BetaToken: "context-1m-2025-08-07",
|
||
Action: BetaPolicyActionFilter,
|
||
Scope: BetaPolicyScopeAll,
|
||
},
|
||
},
|
||
}
|
||
}
|
||
|
||
// OpenAI Fast Policy 策略常量
|
||
// OpenAI 的 "fast 模式" 通过请求体中的 service_tier 字段识别：
|
||
// - "priority"（客户端可传 "fast"，归一化为 "priority"）：fast 模式
|
||
// - "flex"：低优先级模式
|
||
// - 省略：normal 默认
|
||
//
|
||
// 本策略复用 BetaPolicyAction*/BetaPolicyScope* 常量语义，只是匹配键从
|
||
// anthropic-beta header 换成 body 的 service_tier 字段。
|
||
const (
|
||
OpenAIFastTierAny = "all" // 匹配任意已识别的 service_tier
|
||
OpenAIFastTierPriority = "priority" // 仅匹配 fast（priority）
|
||
OpenAIFastTierFlex = "flex" // 仅匹配 flex
|
||
)
|
||
|
||
// OpenAIFastPolicyRule 单条 OpenAI fast/flex 策略规则
|
||
type OpenAIFastPolicyRule struct {
|
||
ServiceTier string `json:"service_tier"` // "priority" | "flex" | "auto" | "default" | "scale" | "all"
|
||
Action string `json:"action"` // "pass" | "filter" | "block"
|
||
Scope string `json:"scope"` // "all" | "oauth" | "apikey" | "bedrock"
|
||
ErrorMessage string `json:"error_message,omitempty"` // 自定义错误消息 (action=block 时生效)
|
||
ModelWhitelist []string `json:"model_whitelist,omitempty"` // 模型匹配模式列表（为空=对所有模型生效）
|
||
FallbackAction string `json:"fallback_action,omitempty"` // 未匹配白名单的模型的处理方式
|
||
FallbackErrorMessage string `json:"fallback_error_message,omitempty"` // 未匹配白名单时的自定义错误消息 (fallback_action=block 时生效)
|
||
}
|
||
|
||
// OpenAIFastPolicySettings OpenAI fast 策略配置
|
||
type OpenAIFastPolicySettings struct {
|
||
Rules []OpenAIFastPolicyRule `json:"rules"`
|
||
}
|
||
|
||
// DefaultOpenAIFastPolicySettings 返回默认的 OpenAI fast 策略配置。
|
||
// 默认不配置任何规则，保留 OpenAI 上游 service_tier 语义；管理员如需
|
||
// 限制 priority/flex，可以在 admin UI 中显式配置 filter 或 block 规则。
|
||
func DefaultOpenAIFastPolicySettings() *OpenAIFastPolicySettings {
|
||
return &OpenAIFastPolicySettings{
|
||
Rules: []OpenAIFastPolicyRule{},
|
||
}
|
||
}
|