shadow_routing.go 2.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051
  1. package service
  2. // parentHealthyForShadow 报告 spark 影子账号的母账号凭据是否可用(影子据此可被调度)。
  3. //
  4. // 非影子账号直接返回 true(不受此检查约束)。
  5. // lookup 将母账号 ID 解析为当前 Account(来自调度快照 map 或 repo)。
  6. //
  7. // 关键语义(F1 决策 A + 外审 D):母账号须仍是 OpenAI OAuth(fail-closed——否则透传凭据解析必失败,
  8. // 影子不应进调度候选),且凭据「可用」。IsCredentialUsableForShadow 检查:账号 active、OAuth token
  9. // 未过期、且**未处于 TempUnschedulableUntil 冷却期**——对 OpenAI 账号该字段由 401/token 刷新耗尽/
  10. // transport·proxy 故障写入,代表共享凭据或传输坏死,故**连坐**影子。
  11. //
  12. // **刻意排除** global 维度的 RateLimitResetAt/OverloadUntil 与母账号手动 Schedulable 开关:
  13. // 母账号 global 429 不得连坐 spark 影子,否则会重新耦合影子架构本应解耦的两条 429 道。
  14. // 母账号未找到(nil)、非 OpenAI OAuth、或凭据不可用时影子被挡。
  15. func parentHealthyForShadow(account *Account, lookup func(int64) *Account) bool {
  16. if account == nil || !account.IsShadow() {
  17. return true
  18. }
  19. parent := lookup(*account.ParentAccountID)
  20. if parent == nil {
  21. return false
  22. }
  23. return parent.IsOpenAIOAuth() && parent.IsCredentialUsableForShadow()
  24. }
  25. // sparkModelVariants 返回所有归一到 spark 的模型 ID(当前仅 base:spark 无 effort 变体)。
  26. // 从 codexModelMap 派生,使集合与别名表单一来源、不漂移;若上游将来新增 spark 变体,
  27. // 在 codexModelMap 注册后此处自动跟随。
  28. func sparkModelVariants() []string {
  29. out := make([]string, 0, 1)
  30. for alias, target := range codexModelMap {
  31. if target == "gpt-5.3-codex-spark" {
  32. out = append(out, alias)
  33. }
  34. }
  35. return out
  36. }
  37. // defaultSparkShadowModelMapping 返回 spark 影子账号的默认 model_mapping。
  38. //
  39. // 恒等映射(key 映射到自身)把「只接 spark」限制落在 key 白名单上,模型零改写、
  40. // 与空 mapping 透传行为一致。当前 spark 仅 base 一个模型(无 effort 变体)。
  41. func defaultSparkShadowModelMapping() map[string]any {
  42. variants := sparkModelVariants()
  43. mapping := make(map[string]any, len(variants))
  44. for _, m := range variants {
  45. mapping[m] = m
  46. }
  47. return mapping
  48. }