Merge pull request router-for-me#2013 from router-for-me/model

Fetch model catalog from network

Author: luispater

.github/workflows/docker-image.yml

@@ -15,6 +15,8 @@ jobs:
  steps:
  - name: Checkout
  uses: actions/checkout@v4
+ - name: Refresh models catalog
+ run: curl -fsSL https://raw.githubusercontent.com/router-for-me/models/refs/heads/main/models.json -o internal/registry/models/models.json
  - name: Set up Docker Buildx
  uses: docker/setup-buildx-action@v3
  - name: Login to DockerHub
@@ -46,6 +48,8 @@ jobs:
  steps:
  - name: Checkout
  uses: actions/checkout@v4
+ - name: Refresh models catalog
+ run: curl -fsSL https://raw.githubusercontent.com/router-for-me/models/refs/heads/main/models.json -o internal/registry/models/models.json
  - name: Set up Docker Buildx
  uses: docker/setup-buildx-action@v3
  - name: Login to DockerHub

.github/workflows/pr-test-build.yml

@@ -12,6 +12,8 @@ jobs:
  steps:
  - name: Checkout
  uses: actions/checkout@v4
+ - name: Refresh models catalog
+ run: curl -fsSL https://raw.githubusercontent.com/router-for-me/models/refs/heads/main/models.json -o internal/registry/models/models.json
  - name: Set up Go
  uses: actions/setup-go@v5
  with:

.github/workflows/release.yaml

@@ -16,6 +16,8 @@ jobs:
  - uses: actions/checkout@v4
  with:
  fetch-depth: 0
+ - name: Refresh models catalog
+ run: curl -fsSL https://raw.githubusercontent.com/router-for-me/models/refs/heads/main/models.json -o internal/registry/models/models.json
  - run: git fetch --force --tags
  - uses: actions/setup-go@v4
  with:

cmd/server/main.go

@@ -24,6 +24,7 @@ import (
 "github.com/router-for-me/CLIProxyAPI/v6/internal/logging"
 "github.com/router-for-me/CLIProxyAPI/v6/internal/managementasset"
 "github.com/router-for-me/CLIProxyAPI/v6/internal/misc"
+"github.com/router-for-me/CLIProxyAPI/v6/internal/registry"
 "github.com/router-for-me/CLIProxyAPI/v6/internal/store"
 _ "github.com/router-for-me/CLIProxyAPI/v6/internal/translator"
 "github.com/router-for-me/CLIProxyAPI/v6/internal/tui"
@@ -494,6 +495,7 @@ func main() {
 if standalone {
 // Standalone mode: start an embedded local server and connect TUI client to it.
 managementasset.StartAutoUpdater(context.Background(), configFilePath)
+registry.StartModelsUpdater(context.Background())
 hook := tui.NewLogHook(2000)
 hook.SetFormatter(&logging.LogFormatter{})
 log.AddHook(hook)
@@ -566,6 +568,7 @@ func main() {
 } else {
 // Start the main proxy service
 managementasset.StartAutoUpdater(context.Background(), configFilePath)
+registry.StartModelsUpdater(context.Background())
 cmd.StartService(cfg, configFilePath, password)
 }
 }

internal/registry/model_definitions.go

@@ -1,12 +1,137 @@
 // Package registry provides model definitions and lookup helpers for various AI providers.
-// Static model metadata is stored in model_definitions_static_data.go.
+// Static model metadata is loaded from the embedded models.json file and can be refreshed from network.
 package registry
 
 import (
 "sort"
 "strings"
 )
 
+// AntigravityModelConfig captures static antigravity model overrides, including
+// Thinking budget limits and provider max completion tokens.
+type AntigravityModelConfig struct {
+Thinking *ThinkingSupport `json:"thinking,omitempty"`
+MaxCompletionTokens int `json:"max_completion_tokens,omitempty"`
+}
+
+// staticModelsJSON mirrors the top-level structure of models.json.
+type staticModelsJSON struct {
+Claude []*ModelInfo `json:"claude"`
+Gemini []*ModelInfo `json:"gemini"`
+Vertex []*ModelInfo `json:"vertex"`
+GeminiCLI []*ModelInfo `json:"gemini-cli"`
+AIStudio []*ModelInfo `json:"aistudio"`
+CodexFree []*ModelInfo `json:"codex-free"`
+CodexTeam []*ModelInfo `json:"codex-team"`
+CodexPlus []*ModelInfo `json:"codex-plus"`
+CodexPro []*ModelInfo `json:"codex-pro"`
+Qwen []*ModelInfo `json:"qwen"`
+IFlow []*ModelInfo `json:"iflow"`
+Kimi []*ModelInfo `json:"kimi"`
+Antigravity map[string]*AntigravityModelConfig `json:"antigravity"`
+}
+
+// GetClaudeModels returns the standard Claude model definitions.
+func GetClaudeModels() []*ModelInfo {
+return cloneModelInfos(getModels().Claude)
+}
+
+// GetGeminiModels returns the standard Gemini model definitions.
+func GetGeminiModels() []*ModelInfo {
+return cloneModelInfos(getModels().Gemini)
+}
+
+// GetGeminiVertexModels returns Gemini model definitions for Vertex AI.
+func GetGeminiVertexModels() []*ModelInfo {
+return cloneModelInfos(getModels().Vertex)
+}
+
+// GetGeminiCLIModels returns Gemini model definitions for the Gemini CLI.
+func GetGeminiCLIModels() []*ModelInfo {
+return cloneModelInfos(getModels().GeminiCLI)
+}
+
+// GetAIStudioModels returns model definitions for AI Studio.
+func GetAIStudioModels() []*ModelInfo {
+return cloneModelInfos(getModels().AIStudio)
+}
+
+// GetCodexFreeModels returns model definitions for the Codex free plan tier.
+func GetCodexFreeModels() []*ModelInfo {
+return cloneModelInfos(getModels().CodexFree)
+}
+
+// GetCodexTeamModels returns model definitions for the Codex team plan tier.
+func GetCodexTeamModels() []*ModelInfo {
+return cloneModelInfos(getModels().CodexTeam)
+}
+
+// GetCodexPlusModels returns model definitions for the Codex plus plan tier.
+func GetCodexPlusModels() []*ModelInfo {
+return cloneModelInfos(getModels().CodexPlus)
+}
+
+// GetCodexProModels returns model definitions for the Codex pro plan tier.
+func GetCodexProModels() []*ModelInfo {
+return cloneModelInfos(getModels().CodexPro)
+}
+
+// GetQwenModels returns the standard Qwen model definitions.
+func GetQwenModels() []*ModelInfo {
+return cloneModelInfos(getModels().Qwen)
+}
+
+// GetIFlowModels returns the standard iFlow model definitions.
+func GetIFlowModels() []*ModelInfo {
+return cloneModelInfos(getModels().IFlow)
+}
+
+// GetKimiModels returns the standard Kimi (Moonshot AI) model definitions.
+func GetKimiModels() []*ModelInfo {
+return cloneModelInfos(getModels().Kimi)
+}
+
+// GetAntigravityModelConfig returns static configuration for antigravity models.
+// Keys use upstream model names returned by the Antigravity models endpoint.
+func GetAntigravityModelConfig() map[string]*AntigravityModelConfig {
+data := getModels()
+if len(data.Antigravity) == 0 {
+return nil
+}
+out := make(map[string]*AntigravityModelConfig, len(data.Antigravity))
+for k, v := range data.Antigravity {
+out[k] = cloneAntigravityModelConfig(v)
+}
+return out
+}
+
+func cloneAntigravityModelConfig(cfg *AntigravityModelConfig) *AntigravityModelConfig {
+if cfg == nil {
+return nil
+}
+copyConfig := *cfg
+if cfg.Thinking != nil {
+copyThinking := *cfg.Thinking
+if len(cfg.Thinking.Levels) > 0 {
+copyThinking.Levels = append([]string(nil), cfg.Thinking.Levels...)
+}
+copyConfig.Thinking = &copyThinking
+}
+return &copyConfig
+}
+
+// cloneModelInfos returns a shallow copy of the slice with each element deep-cloned.
+func cloneModelInfos(models []*ModelInfo) []*ModelInfo {
+if len(models) == 0 {
+return nil
+}
+out := make([]*ModelInfo, len(models))
+for i, m := range models {
+out[i] = cloneModelInfo(m)
+}
+return out
+}
+
 // GetStaticModelDefinitionsByChannel returns static model definitions for a given channel/provider.
 // It returns nil when the channel is unknown.
 //
@@ -35,7 +160,7 @@ func GetStaticModelDefinitionsByChannel(channel string) []*ModelInfo {
 case "aistudio":
 return GetAIStudioModels()
 case "codex":
-return GetOpenAIModels()
+return GetCodexProModels()
 case "qwen":
 return GetQwenModels()
 case "iflow":
@@ -77,27 +202,28 @@ func LookupStaticModelInfo(modelID string) *ModelInfo {
 return nil
 }
 
+data := getModels()
 allModels := [][]*ModelInfo{
-GetClaudeModels(),
-GetGeminiModels(),
-GetGeminiVertexModels(),
-GetGeminiCLIModels(),
-GetAIStudioModels(),
-GetOpenAIModels(),
-GetQwenModels(),
-GetIFlowModels(),
-GetKimiModels(),
+data.Claude,
+data.Gemini,
+data.Vertex,
+data.GeminiCLI,
+data.AIStudio,
+data.CodexPro,
+data.Qwen,
+data.IFlow,
+data.Kimi,
 }
 for _, models := range allModels {
 for _, m := range models {
 if m != nil && m.ID == modelID {
-return m
+return cloneModelInfo(m)
 }
 }
 }
 
 // Check Antigravity static config
-if cfg := GetAntigravityModelConfig()[modelID]; cfg != nil {
+if cfg := cloneAntigravityModelConfig(data.Antigravity[modelID]); cfg != nil {
 return &ModelInfo{
 ID: modelID,
 Thinking: cfg.Thinking,