From ba69736f554306a5a1c01cc6d24be628e7025898 Mon Sep 17 00:00:00 2001 From: cagedbird043 Date: Wed, 25 Feb 2026 21:02:40 +0800 Subject: [PATCH] =?UTF-8?q?refactor(admin):=20=E6=B5=8B=E8=AF=95=E8=BF=9E?= =?UTF-8?q?=E6=8E=A5=E6=A8=A1=E5=9E=8B=E5=88=97=E8=A1=A8=E6=94=B9=E4=B8=BA?= =?UTF-8?q?=E5=A4=8D=E7=94=A8=20antigravity.DefaultModels=EF=BC=8C?= =?UTF-8?q?=E6=B6=88=E9=99=A4=E7=A1=AC=E7=BC=96=E7=A0=81=E9=87=8D=E5=A4=8D?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../internal/handler/admin/account_handler.go | 33 ++----------------- 1 file changed, 3 insertions(+), 30 deletions(-) diff --git a/backend/internal/handler/admin/account_handler.go b/backend/internal/handler/admin/account_handler.go index 45867527..0a012b8f 100644 --- a/backend/internal/handler/admin/account_handler.go +++ b/backend/internal/handler/admin/account_handler.go @@ -16,6 +16,7 @@ import ( "github.com/Wei-Shaw/sub2api/internal/domain" "github.com/Wei-Shaw/sub2api/internal/handler/dto" + "github.com/Wei-Shaw/sub2api/internal/pkg/antigravity" "github.com/Wei-Shaw/sub2api/internal/pkg/claude" "github.com/Wei-Shaw/sub2api/internal/pkg/geminicli" "github.com/Wei-Shaw/sub2api/internal/pkg/openai" @@ -1459,36 +1460,8 @@ func (h *AccountHandler) GetAvailableModels(c *gin.Context) { // Handle Antigravity accounts: return Claude + Gemini models if account.Platform == service.PlatformAntigravity { - // Antigravity 支持 Claude 和部分 Gemini 模型 - type UnifiedModel struct { - ID string `json:"id"` - Type string `json:"type"` - DisplayName string `json:"display_name"` - } - - var models []UnifiedModel - - // 添加 Claude 模型 - for _, m := range claude.DefaultModels { - models = append(models, UnifiedModel{ - ID: m.ID, - Type: m.Type, - DisplayName: m.DisplayName, - }) - } - - // 添加 Gemini 3 系列模型用于测试 - geminiTestModels := []UnifiedModel{ - {ID: "gemini-3-flash", Type: "model", DisplayName: "Gemini 3 Flash"}, - {ID: "gemini-3-pro-low", Type: "model", DisplayName: "Gemini 3 Pro Low"}, - {ID: "gemini-3-pro-high", Type: "model", DisplayName: "Gemini 3 Pro High"}, - {ID: "gemini-3.1-pro-low", Type: "model", DisplayName: "Gemini 3.1 Pro Low"}, - {ID: "gemini-3.1-pro-high", Type: "model", DisplayName: "Gemini 3.1 Pro High"}, - {ID: "gemini-3-pro-preview", Type: "model", DisplayName: "Gemini 3 Pro Preview"}, - } - models = append(models, geminiTestModels...) - - response.Success(c, models) + // 直接复用 antigravity.DefaultModels()，与 /v1/models 端点保持同步 + response.Success(c, antigravity.DefaultModels()) return }