mirror of
https://github.com/Wei-Shaw/sub2api.git
synced 2026-10-21 03:32:36 +00:00
li
b6eb6c1efa
/v1beta/models/{model}:generateContent 与 Anthropic→Gemini 兼容路径的
ImageCount 只由 isImageGenerationModel(originalModel) 决定，而该白名单是
按 Google 官方模型名精确/前缀匹配写死的（antigravity_image_test.go 里
明确断言 my-gemini-3-pro-image-test 这类自定义名返回 false）。
GeminiMessagesCompatService 服务的却主要是 API Key + 自定义模型映射的账号：
客户端请求名和 GetMappedModel 后的上游名都可能是站长自取的别名，白名单必然
判不出来 → ImageCount=0 → calculateRecordUsageCost 里 `if result.ImageCount > 0`
的按次计费分支整条不触发 → 生图请求全部记 $0（issue #5358）。
改为优先按上游响应里真实的 inlineData 图片 part 计数：
- 新增请求级计数器，挂在 gin.Context 上，与既有的
upstreamResponseModelObserver 同一批调用点取解包后的响应体；
- 取「单个 payload 内的最大值」而非累加：Gemini 兼容上游的 SSE 分片可能是
累积式的（computeGeminiTextDelta 正是为此存在），逐 chunk 累加会把同一张图
重复计费。max 保证累积式流与非流式都得到真实张数，增量式多图流最差退化到 1，
与改动前同值，不构成回退；
- 每次 Forward 开头重置计数器，避免 failover 复用同一个 gin.Context 时
把失败账号已回吐的图叠加到成功账号账单上；
- 响应里数不出图时（fileData 引用式回图等）退回原有模型名启发式，并额外认
映射后的上游模型名，与 shouldSkipCodexPlanGatedImageModelCooldown 同时取
requestedModel / modelKey 的口径一致。
inlineData / inline_data 两种字段风格都认（官方 SDK 与部分中转回 snake_case），
只统计带 base64 数据且 MIME 为图片的 part。
Fixes #5358