li
02fbcbe3ad
上一版守卫只看模型类型，不区分请求从哪个端点进来。OAuth 账号的 /v1/images/* 上游同样是 Codex Responses（openai_images_responses.go → handleOpenAIImagesErrorResponse → handleOpenAIAccountUpstreamError → HandleUpstreamModelNotFound），所以专用生图 端点也会命中 plan-gated 分支。账号确实不具备生图能力时跳过冷却，会让调度层失去 唯一的刹车：每个请求都完整走一遍号池，对上游形成无上界的 400 放大。 改动： - 新增 ctxkey.OpenAIImagesEndpoint 与 WithOpenAIImagesEndpoint / OpenAIImagesEndpointFromContext，在 handler/openai_images.go 入口置位； 与 OpenAIImageGenerationIntent 区分——后者在 /v1/responses 带图片模型时也会置位。 - 守卫下移到 modelKey 计算之后，抽成 shouldSkipCodexPlanGatedImageModelCooldown， 仅在 plan-gated 分支、且非 /v1/images/* 入站时生效。 - 同时判断 requestedModel 与最终 modelKey：冷却键走 account.GetMappedModel， 账号可以把文本别名映射到 gpt-image-*，只判请求模型会漏掉这种形态。
85 lines
4.1 KiB
Go
85 lines
4.1 KiB
Go
// Package ctxkey 定义用于 context.Value 的类型安全 key
|
||
package ctxkey
|
||
|
||
// Key 定义 context key 的类型，避免使用内置 string 类型（staticcheck SA1029）
|
||
type Key string
|
||
|
||
const (
|
||
// ForcePlatform 强制平台（用于 /antigravity 路由），由 middleware.ForcePlatform 设置
|
||
ForcePlatform Key = "ctx_force_platform"
|
||
|
||
// ResolvedTargetPlatform 是 composite 分组按请求模型解析出的真实目标平台。
|
||
ResolvedTargetPlatform Key = "ctx_resolved_target_platform"
|
||
|
||
// ResolvedUpstreamModel 是 composite 分组将公开模型名解析到的上游模型名。
|
||
ResolvedUpstreamModel Key = "ctx_resolved_upstream_model"
|
||
|
||
// RequestedPublicModel 是客户端原始请求中的公开模型名。
|
||
RequestedPublicModel Key = "ctx_requested_public_model"
|
||
|
||
// CompositeRouteSource 标识 composite 解析结果来自显式路由还是内置模型探测。
|
||
CompositeRouteSource Key = "ctx_composite_route_source"
|
||
|
||
// RequestID 为服务端生成/透传的请求 ID。
|
||
RequestID Key = "ctx_request_id"
|
||
|
||
// ClientRequestID 客户端请求的唯一标识，用于追踪请求全生命周期（用于 Ops 监控与排障）。
|
||
ClientRequestID Key = "ctx_client_request_id"
|
||
|
||
// Model 请求模型标识（用于统一请求链路日志字段）。
|
||
Model Key = "ctx_model"
|
||
|
||
// Platform 当前请求最终命中的平台（用于统一请求链路日志字段）。
|
||
Platform Key = "ctx_platform"
|
||
|
||
// AccountID 当前请求最终命中的账号 ID（用于统一请求链路日志字段）。
|
||
AccountID Key = "ctx_account_id"
|
||
|
||
// RetryCount 表示当前请求在网关层的重试次数（用于 Ops 记录与排障）。
|
||
RetryCount Key = "ctx_retry_count"
|
||
|
||
// AccountSwitchCount 表示请求过程中发生的账号切换次数
|
||
AccountSwitchCount Key = "ctx_account_switch_count"
|
||
|
||
// IsClaudeCodeClient 标识当前请求是否来自 Claude Code 客户端
|
||
IsClaudeCodeClient Key = "ctx_is_claude_code_client"
|
||
|
||
// ThinkingEnabled 标识当前请求是否开启 thinking（用于 Antigravity 最终模型名推导与模型维度限流）
|
||
ThinkingEnabled Key = "ctx_thinking_enabled"
|
||
|
||
// OpenAIImageGenerationIntent 标识 OpenAI 请求会触发生图能力（用于图片能力维度限流）
|
||
OpenAIImageGenerationIntent Key = "ctx_openai_image_generation_intent"
|
||
|
||
// OpenAIImagesEndpoint 标识请求是从 /v1/images/* 入站的。
|
||
// 与 OpenAIImageGenerationIntent 的区别：后者只表示"这次请求会生图"，
|
||
// /v1/responses 带图片模型时也会置位；本 key 只在专用生图端点置位，
|
||
// 用于区分"用错端点"与"端点用对了但账号没能力"。
|
||
OpenAIImagesEndpoint Key = "ctx_openai_images_endpoint"
|
||
|
||
// Group 认证后的分组信息，由 API Key 认证中间件设置
|
||
Group Key = "ctx_group"
|
||
|
||
// UserID 认证后的 Sub2API 用户 ID，由 API Key 认证中间件设置。
|
||
// 供 service 层执行用户级策略，不能使用客户端请求体中的 user 标识替代。
|
||
UserID Key = "ctx_user_id"
|
||
|
||
// IsMaxTokensOneHaikuRequest 标识当前请求是否为 max_tokens=1 + haiku 模型的探测请求
|
||
// 用于 ClaudeCodeOnly 验证绕过（绕过 system prompt 检查，但仍需验证 User-Agent）
|
||
IsMaxTokensOneHaikuRequest Key = "ctx_is_max_tokens_one_haiku"
|
||
|
||
// SingleAccountRetry 标识当前请求处于单账号 503 退避重试模式。
|
||
// 在此模式下，Service 层的模型限流预检查将等待限流过期而非直接切换账号。
|
||
SingleAccountRetry Key = "ctx_single_account_retry"
|
||
|
||
// PrefetchedStickyAccountID 标识上游（通常 handler）预取到的 sticky session 账号 ID。
|
||
// Service 层可复用该值，避免同请求链路重复读取 Redis。
|
||
PrefetchedStickyAccountID Key = "ctx_prefetched_sticky_account_id"
|
||
|
||
// PrefetchedStickyGroupID 标识上游预取 sticky session 时所使用的分组 ID。
|
||
// Service 层仅在分组匹配时复用 PrefetchedStickyAccountID，避免分组切换重试误用旧 sticky。
|
||
PrefetchedStickyGroupID Key = "ctx_prefetched_sticky_group_id"
|
||
|
||
// ClaudeCodeVersion stores the extracted Claude Code version from User-Agent (e.g. "2.1.22")
|
||
ClaudeCodeVersion Key = "ctx_claude_code_version"
|
||
)
|