1package provider
2
3import (
4 "encoding/json"
5 "os"
6)
7
8// UserModelsFile is the JSON format for user-defined models.
9// Place a models.json in $ZOT_HOME to add models that aren't in the
10// baked-in catalog or to override catalog entries.
11//
12// Example:
13//
14// {
15// "providers": {
16// "openai": {
17// "models": [
18// {
19// "id": "gpt-5.5",
20// "name": "GPT-5.5",
21// "reasoning": true,
22// "contextWindow": 400000,
23// "maxTokens": 128000,
24// "priceInput": 2.50,
25// "priceOutput": 15.00,
26// "priceCacheRead": 0.25
27// }
28// ]
29// }
30// }
31// }
32type UserModelsFile struct {
33 Providers map[string]UserProvider `json:"providers"`
34}
35
36// UserProvider groups models under a provider key.
37type UserProvider struct {
38 Models []UserModel `json:"models"`
39}
40
41// UserModel is a single model entry in the user's models.json.
42type UserModel struct {
43 ID string `json:"id"`
44 Name string `json:"name"`
45 Reasoning bool `json:"reasoning"`
46 ContextWindow int `json:"contextWindow"`
47 MaxTokens int `json:"maxTokens"`
48 PriceInput float64 `json:"priceInput"`
49 PriceOutput float64 `json:"priceOutput"`
50 PriceCacheRead float64 `json:"priceCacheRead"`
51 PriceCacheWrite float64 `json:"priceCacheWrite"`
52 BaseURL string `json:"baseUrl,omitempty"`
53 Input []string `json:"input"` // informational only
54 API string `json:"api"` // informational only
55}
56
57// LoadUserModels reads a models.json file and returns the models
58// converted to the internal Model type. Returns nil on any error
59// (missing file, bad JSON, etc.) so the caller can treat it as
60// optional without error handling.
61func LoadUserModels(path string) []Model {
62 data, err := os.ReadFile(path)
63 if err != nil {
64 return nil
65 }
66 var file UserModelsFile
67 if err := json.Unmarshal(data, &file); err != nil {
68 return nil
69 }
70
71 var out []Model
72 for providerName, prov := range file.Providers {
73 // Normalize provider name: strip suffixes like "-codex" so
74 // "openai-codex" maps to "openai".
75 normalized := providerName
76 switch providerName {
77 case "openai-codex", "openai-responses":
78 normalized = "openai"
79 case "anthropic-messages":
80 normalized = "anthropic"
81 case "moonshot", "moonshot-ai", "kimi-code":
82 normalized = "kimi"
83 case "deepseek-chat", "deepseek-ai":
84 normalized = "deepseek"
85 }
86
87 for _, um := range prov.Models {
88 m := Model{
89 Provider: normalized,
90 ID: um.ID,
91 DisplayName: um.Name,
92 ContextWindow: um.ContextWindow,
93 MaxOutput: um.MaxTokens,
94 Reasoning: um.Reasoning,
95 PriceInput: um.PriceInput,
96 PriceOutput: um.PriceOutput,
97 PriceCacheRead: um.PriceCacheRead,
98 PriceCacheWrite: um.PriceCacheWrite,
99 BaseURL: um.BaseURL,
100 Source: "user",
101 }
102 if m.DisplayName == "" {
103 m.DisplayName = m.ID
104 }
105 out = append(out, m)
106 }
107 }
108 return out
109}
110
111// SetUserModels merges user-defined models into the active catalog.
112// User models take precedence over both the baked-in catalog and
113// live-discovered models.
114func SetUserModels(models []Model) {
115 if len(models) == 0 {
116 return
117 }
118 activeMu.Lock()
119 defer activeMu.Unlock()
120
121 // Build index of current active models.
122 byKey := func(p, id string) string { return p + "/" + id }
123 index := make(map[string]int, len(active))
124 for i, m := range active {
125 index[byKey(m.Provider, m.ID)] = i
126 }
127
128 for _, um := range models {
129 k := byKey(um.Provider, um.ID)
130 if idx, ok := index[k]; ok {
131 // Override existing entry but keep prices from user if
132 // they provided them, otherwise keep catalog prices.
133 existing := active[idx]
134 if um.PriceInput > 0 {
135 existing.PriceInput = um.PriceInput
136 }
137 if um.PriceOutput > 0 {
138 existing.PriceOutput = um.PriceOutput
139 }
140 if um.PriceCacheRead > 0 {
141 existing.PriceCacheRead = um.PriceCacheRead
142 }
143 if um.PriceCacheWrite > 0 {
144 existing.PriceCacheWrite = um.PriceCacheWrite
145 }
146 if um.DisplayName != "" {
147 existing.DisplayName = um.DisplayName
148 }
149 if um.ContextWindow > 0 {
150 existing.ContextWindow = um.ContextWindow
151 }
152 if um.MaxOutput > 0 {
153 existing.MaxOutput = um.MaxOutput
154 }
155 existing.Reasoning = um.Reasoning
156 if um.BaseURL != "" {
157 existing.BaseURL = um.BaseURL
158 }
159 existing.Source = "user"
160 existing.Speculative = false
161 active[idx] = existing
162 } else {
163 // New model not in catalog.
164 active = append(active, um)
165 }
166 }
167}