mirror of
https://github.com/Wei-Shaw/sub2api.git
synced 2026-09-24 16:05:44 +08:00
limingeandClaude Fable 5
13e773ef5e
背景：Codex CLI / Codex App 会从 provider 的 GET {base_url}/models?client_version=...
（自定义 provider 模式）或 GET /backend-api/codex/models（chatgpt_base_url 模式）
刷新模型选单，期望 ChatGPT Codex manifest 格式（{"models":[{slug,...}]}）。
sub2api 此前只提供 OpenAI 兼容格式的 /v1/models，Codex 客户端解析失败后静默
回落到本地缓存，导致指向 sub2api 的 Codex 客户端模型选单永久冻结在切换
provider 当天的状态，新模型（如 gpt-5.6 系列）永远不会出现。
方案：新增 manifest 透传——用组内可调度 OAuth 账号的凭据向
chatgpt.com/backend-api/codex/models 实时转发请求，响应体与 ETag 原样透传。
不在网关侧解析或维护模型清单：manifest schema 随 Codex 客户端版本演进，
透传保证网关无需跟进 schema 变化，且返回的始终是账号真实的模型权限
（区别于静态 DefaultModels 的"理论列表"）。
路由：
- GET /backend-api/codex/models（新增）
- GET /v1/models 带 client_version 查询参数且组平台为 openai 时分发到
manifest 透传（client_version 是 Codex 客户端的天然指纹，普通 OpenAI
客户端不携带）；其余请求保持原有 OpenAI 格式行为不变。
上游失败时按 fast-fail 返回错误，不伪造列表；Codex 客户端自身会回落缓存。
Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
54 lines
1.9 KiB
Go
54 lines
1.9 KiB
Go
package handler
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
|
|
infraerrors "github.com/Wei-Shaw/sub2api/internal/pkg/errors"
|
|
middleware2 "github.com/Wei-Shaw/sub2api/internal/server/middleware"
|
|
"github.com/Wei-Shaw/sub2api/internal/service"
|
|
)
|
|
|
|
// CodexModels serves the Codex models manifest for Codex clients.
|
|
//
|
|
// Codex CLI and the Codex desktop app refresh their model picker from
|
|
// GET {base_url}/models?client_version=... (custom provider mode) or
|
|
// GET /backend-api/codex/models (chatgpt_base_url mode). Both routes land
|
|
// here. The manifest is proxied verbatim from the ChatGPT backend with a
|
|
// schedulable OAuth account's credentials, so clients pointed at the gateway
|
|
// see the account's real, always-current model entitlements instead of a
|
|
// frozen local cache.
|
|
func (h *OpenAIGatewayHandler) CodexModels(c *gin.Context) {
|
|
apiKey, ok := middleware2.GetAPIKeyFromContext(c)
|
|
if !ok || apiKey.Group == nil {
|
|
h.errorResponse(c, http.StatusUnauthorized, "invalid_request_error", "API key group is required")
|
|
return
|
|
}
|
|
if apiKey.Group.Platform != service.PlatformOpenAI {
|
|
h.errorResponse(c, http.StatusNotFound, "not_found_error", "Codex models manifest is only available for OpenAI groups")
|
|
return
|
|
}
|
|
|
|
account, err := h.gatewayService.SelectAccountForModel(c.Request.Context(), apiKey.GroupID, "", "")
|
|
if err != nil {
|
|
h.errorResponse(c, http.StatusServiceUnavailable, "upstream_error", "No available OpenAI accounts")
|
|
return
|
|
}
|
|
|
|
manifest, err := h.gatewayService.FetchCodexModelsManifest(c.Request.Context(), account, c.Query("client_version"), c.GetHeader("If-None-Match"))
|
|
if err != nil {
|
|
h.errorResponse(c, infraerrors.Code(err), "upstream_error", infraerrors.Message(err))
|
|
return
|
|
}
|
|
|
|
if manifest.ETag != "" {
|
|
c.Header("ETag", manifest.ETag)
|
|
}
|
|
if manifest.NotModified {
|
|
c.Status(http.StatusNotModified)
|
|
return
|
|
}
|
|
c.Data(http.StatusOK, "application/json", manifest.Body)
|
|
}
|