magic-agent 2.4.3 → 2.6.3

package/internal/agent/credits_minimax_test.go

@@ -0,0 +1,97 @@
+package agent
+
+// credits_minimax_test.go - llm 引擎的 MiniMax 套餐额度（剩余百分比）解析。
+
+import (
+	"context"
+	"net/http"
+	"net/http/httptest"
+	"testing"
+)
+
+func TestMiniMaxQuotaAt(t *testing.T) {
+	srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		if got := r.Header.Get("Authorization"); got != "Bearer sk-cp-test" {
+			t.Errorf("Authorization = %q", got)
+		}
+		_, _ = w.Write([]byte(`{"model_remains":[
+			{"model_name":"video","current_interval_remaining_percent":100,"current_weekly_remaining_percent":100},
+			{"model_name":"general","current_interval_remaining_percent":98,"current_weekly_remaining_percent":53}
+		],"base_resp":{"status_code":0,"status_msg":"success"}}`))
+	}))
+	defer srv.Close()
+
+	q, ok := minimaxQuotaAt(context.Background(), srv.URL, "sk-cp-test")
+	if !ok || q.IntervalLeft != "98" || q.WeeklyLeft != "53" {
+		t.Fatalf("got (%+v, %v), want {98 53}", q, ok)
+	}
+}
+
+func TestMiniMaxQuotaAtBusinessError(t *testing.T) {
+	srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		_, _ = w.Write([]byte(`{"base_resp":{"status_code":1004,"status_msg":"invalid api key"}}`))
+	}))
+	defer srv.Close()
+
+	if q, ok := minimaxQuotaAt(context.Background(), srv.URL, "sk-bad"); ok {
+		t.Fatalf("业务失败应返回 ok=false，实际 %+v", q)
+	}
+}
+
+func TestModelQuotaPerKey(t *testing.T) {
+	list := []LLMConfigModel{
+		{ID: "minimax-a", URL: "https://api.minimaxi.com/v1/chat/completions", APIKey: "sk-a"},
+		{ID: "minimax-a-variant", URL: "https://api.minimaxi.com/v1", APIKey: "sk-a"},
+		{ID: "minimax-b", URL: "https://api.minimaxi.com/v1/chat/completions", APIKey: "sk-b"},
+		{ID: "ark-glm", URL: "https://ark.example.com/v1", APIKey: "sk-other"},
+		{ID: "minimax-nokey", URL: "https://api.minimaxi.com/v1"},
+	}
+	calls := 0
+	query := func(_ context.Context, host, key string) (QuotaWindows, bool) {
+		calls++
+		if host != "https://www.minimaxi.com" {
+			t.Errorf("host = %q", host)
+		}
+		switch key {
+		case "sk-a":
+			return QuotaWindows{IntervalLeft: "98", WeeklyLeft: "53"}, true
+		case "sk-b":
+			return QuotaWindows{IntervalLeft: "12", WeeklyLeft: "7"}, true
+		}
+		return QuotaWindows{}, false
+	}
+	got := modelQuotaMap(context.Background(), list, query)
+
+	if calls != 2 {
+		t.Fatalf("同一把 key 应只查一次，实际 %d 次", calls)
+	}
+	if q := got["minimax-a"]; q.IntervalLeft != "98" || q.WeeklyLeft != "53" {
+		t.Fatalf("minimax-a = %+v", q)
+	}
+	if q := got["minimax-a-variant"]; q.IntervalLeft != "98" {
+		t.Fatalf("同 key 的变体应与 sk-a 一致，实际 %+v", q)
+	}
+	if q := got["minimax-b"]; q.IntervalLeft != "12" || q.WeeklyLeft != "7" {
+		t.Fatalf("minimax-b 应按自己的 key 取额度，实际 %+v", q)
+	}
+	if _, ok := got["ark-glm"]; ok {
+		t.Fatal("非 MiniMax 模型不应有额度")
+	}
+	if _, ok := got["minimax-nokey"]; ok {
+		t.Fatal("没有 key 的模型不应有额度")
+	}
+}
+
+func TestMinimaxHost(t *testing.T) {
+	cases := map[string]string{
+		"https://api.minimaxi.com/v1/chat/completions": "https://www.minimaxi.com",
+		"https://api.minimax.io/v1":                    "https://www.minimax.io",
+		"https://ark.example.com/v1":                   "",
+		"":                                             "",
+	}
+	for in, want := range cases {
+		if got := minimaxHost(in); got != want {
+			t.Errorf("minimaxHost(%q) = %q, want %q", in, got, want)
+		}
+	}
+}

package/internal/agent/credits_test.go

@@ -0,0 +1,211 @@
+package agent
+
+// credits_test.go - codebuddy 「剩余积分」查询的纯解析逻辑（不碰网络）。
+
+import (
+	"os"
+	"path/filepath"
+	"testing"
+	"time"
+)
+
+func TestFormatCredit(t *testing.T) {
+	cases := map[float64]string{
+		5939.20000022: "5939.2",
+		500:           "500",
+		0:             "0",
+		0.004:         "0",
+		12.345:        "12.35",
+	}
+	for in, want := range cases {
+		if got := formatCredit(in); got != want {
+			t.Errorf("formatCredit(%v) = %q, want %q", in, got, want)
+		}
+	}
+}
+
+func TestPersonalCreditLeftFromJSON(t *testing.T) {
+	// ① 接口给 TotalCredits（数字或字符串都出现过）
+	for _, body := range []string{
+		`{"code":0,"data":{"TotalCredits":5939.20000022}}`,
+		`{"code":0,"data":{"TotalCredits":"5939.20000022"}}`,
+	} {
+		if got := personalCreditLeftFromJSON([]byte(body)); got != "5939.2" {
+			t.Errorf("TotalCredits body %s → %q, want 5939.2", body, got)
+		}
+	}
+	// ② 没有 TotalCredits 时累加 Packages[].CycleRemainCapacity
+	body := `{"code":0,"data":{"Packages":[
+	  {"CycleRemainCapacity":"5339.20000022"},
+	  {"CycleRemainCapacity":"500"},
+	  {"CycleRemainCapacity":100}]}}`
+	if got := personalCreditLeftFromJSON([]byte(body)); got != "5939.2" {
+		t.Errorf("包剩余累加 → %q, want 5939.2", got)
+	}
+	// ③ 业务错误 / 空数据 → 不给数字
+	for _, body := range []string{`{"code":14001,"msg":"no credit"}`, `{"code":0,"data":{}}`, `not-json`} {
+		if got := personalCreditLeftFromJSON([]byte(body)); got != "" {
+			t.Errorf("bad body %s → %q, want empty", body, got)
+		}
+	}
+}
+
+func TestEnterpriseCreditLeftFromJSON(t *testing.T) {
+	body := `{"code":0,"data":{"limit_num":1000,"used_num":321.5}}`
+	if got := enterpriseCreditLeftFromJSON([]byte(body)); got != "678.5" {
+		t.Errorf("enterprise → %q, want 678.5", got)
+	}
+	// limit_num = -1：不限量，不给数字
+	if got := enterpriseCreditLeftFromJSON([]byte(`{"code":0,"data":{"limit_num":-1,"used_num":10}}`)); got != "" {
+		t.Errorf("unlimited → %q, want empty", got)
+	}
+	if got := enterpriseCreditLeftFromJSON([]byte(`{"code":1}`)); got != "" {
+		t.Errorf("code!=0 → %q, want empty", got)
+	}
+}
+
+// 明文票据可读；桌面 App 的 $wbEncrypted token 必须被跳过（独立进程解不开）。
+func TestParseCodebuddyTicket(t *testing.T) {
+	dir := t.TempDir()
+	plain := filepath.Join(dir, "codebuddy.info")
+	write := func(p, s string) {
+		t.Helper()
+		if err := os.WriteFile(p, []byte(s), 0o600); err != nil {
+			t.Fatal(err)
+		}
+	}
+	write(plain, `{"account":{"uid":"u-1","nickname":"Darren"},"auth":{"accessToken":"tok-1","domain":"www.codebuddy.cn"}}`)
+	got, ok := parseCodebuddyTicket(plain)
+	if !ok || got.Token != "tok-1" || got.UID != "u-1" || got.Nickname != "Darren" || got.Domain != "www.codebuddy.cn" {
+		t.Fatalf("plain ticket = %+v ok=%v", got, ok)
+	}
+
+	enc := filepath.Join(dir, "workbuddy-desktop.info")
+	write(enc, `{"account":{"uid":"u-2"},"auth":{"accessToken":{"$wbEncrypted":1,"envelope":"x"},"domain":"www.workbuddy.cn"}}`)
+	if _, ok := parseCodebuddyTicket(enc); ok {
+		t.Error("$wbEncrypted 票据不应被当成可用票据")
+	}
+}
+
+// 候选顺序：本引擎 id → base product id → 其余最新明文票据。
+func TestReadCodebuddyTicketOrder(t *testing.T) {
+	dir := t.TempDir()
+	now := time.Now()
+	writeAt := func(name, body string, mod time.Time) {
+		t.Helper()
+		p := filepath.Join(dir, name)
+		if err := os.WriteFile(p, []byte(body), 0o600); err != nil {
+			t.Fatal(err)
+		}
+		if err := os.Chtimes(p, mod, mod); err != nil {
+			t.Fatal(err)
+		}
+	}
+	writeAt("codebuddy.info", `{"account":{"uid":"mine"},"auth":{"accessToken":"t1","domain":"d"}}`, now.Add(-time.Hour))
+	writeAt(baseProductAuthID+".info", `{"account":{"uid":"base"},"auth":{"accessToken":"t2","domain":"d"}}`, now)
+	writeAt("newer-other.info", `{"account":{"uid":"other"},"auth":{"accessToken":"t3","domain":"d"}}`, now.Add(time.Hour))
+
+	if got, ok := readCodebuddyTicket(dir, "codebuddy"); !ok || got.UID != "mine" {
+		t.Errorf("应优先本引擎 id，实际 %+v ok=%v", got, ok)
+	}
+	if err := os.Remove(filepath.Join(dir, "codebuddy.info")); err != nil {
+		t.Fatal(err)
+	}
+	if got, ok := readCodebuddyTicket(dir, "codebuddy"); !ok || got.UID != "base" {
+		t.Errorf("本引擎 id 缺失时应回退 base product id，实际 %+v ok=%v", got, ok)
+	}
+	if err := os.Remove(filepath.Join(dir, baseProductAuthID+".info")); err != nil {
+		t.Fatal(err)
+	}
+	if got, ok := readCodebuddyTicket(dir, "codebuddy"); !ok || got.UID != "other" {
+		t.Errorf("两者都缺失时应取最新明文票据，实际 %+v ok=%v", got, ok)
+	}
+}
+
+func TestCheckinStatusFromJSON(t *testing.T) {
+	body := `{"code":0,"data":{"active":true,"today_checked_in":false,"daily_credit":100,"streak_days":19}}`
+	info, ok := checkinStatusFromJSON([]byte(body))
+	if !ok || info.CheckedToday || info.DailyCredit != "100" || info.StreakDays != 19 {
+		t.Fatalf("status = %+v ok=%v", info, ok)
+	}
+	// 活动未开 / 业务错误 / 坏 JSON → 拿不到状态（--engines 省略 checkin_today）
+	for _, bad := range []string{
+		`{"code":0,"data":{"active":false,"today_checked_in":true}}`,
+		`{"code":14001,"msg":"x"}`,
+		`not-json`,
+	} {
+		if _, ok := checkinStatusFromJSON([]byte(bad)); ok {
+			t.Errorf("bad body %s 应返回 ok=false", bad)
+		}
+	}
+}
+
+func TestClaimCheckinFromJSON(t *testing.T) {
+	// 顶层 / data 两种成功形状都要认
+	for _, body := range []string{
+		`{"code":0,"credit":100,"streak_days":20}`,
+		`{"code":0,"data":{"credit":100,"streak_days":20,"today_checked_in":true}}`,
+	} {
+		info, err := claimCheckinFromJSON([]byte(body))
+		if err != nil || !info.CheckedToday || info.Credit != "100" || info.StreakDays != 20 {
+			t.Errorf("claim body %s → %+v err=%v", body, info, err)
+		}
+	}
+	// 业务失败：error 带服务端原因
+	info, err := claimCheckinFromJSON([]byte(`{"code":14001,"msg":"今天已经领过了"}`))
+	if err == nil || info.Message != "今天已经领过了" {
+		t.Errorf("业务失败应返回 error 与 message，实际 %+v err=%v", info, err)
+	}
+	if _, err := claimCheckinFromJSON([]byte(`not-json`)); err == nil {
+		t.Error("坏 JSON 应报错")
+	}
+}
+
+// 多账号槽位的登录态：各自看自己的票据目录，互不影响（「登录几个账号显示几个」）。
+func TestCodeBuddyLoggedIn(t *testing.T) {
+	home := t.TempDir()
+	// Windows 的 os.UserHomeDir() 读 USERPROFILE；两边都设，跨平台都能隔离。
+	t.Setenv("HOME", home)
+	t.Setenv("USERPROFILE", home)
+
+	main, ai := &CodeBuddyEngine{}, &CodeBuddyAIEngine{}
+	if main.LoggedIn() || ai.LoggedIn() {
+		t.Fatal("没有任何票据时应为未登录")
+	}
+	if main.AccountName() != "" || ai.AccountName() != "" {
+		t.Fatal("未登录时不应有账号昵称")
+	}
+
+	writeTicket := func(dir, id, nickname string) {
+		t.Helper()
+		if err := os.MkdirAll(dir, 0o700); err != nil {
+			t.Fatal(err)
+		}
+		body := `{"account":{"uid":"u-1","nickname":"` + nickname + `"},"auth":{"accessToken":"tok","domain":"www.codebuddy.cn"}}`
+		if err := os.WriteFile(filepath.Join(dir, id+".info"), []byte(body), 0o600); err != nil {
+			t.Fatal(err)
+		}
+	}
+
+	writeTicket(codebuddyAuthDir(home), codebuddyAuthID, "Darren")
+	if !main.LoggedIn() {
+		t.Fatal("主账号票据存在时应为已登录")
+	}
+	if got := main.AccountName(); got != "Darren" {
+		t.Fatalf("主账号昵称 = %q，want Darren", got)
+	}
+	if ai.LoggedIn() {
+		t.Fatal("主账号的票据不应让隔离槽位也变成已登录")
+	}
+
+	writeTicket(codebuddyAuthDir(filepath.Join(home, ".codebuddy-ai")), codebuddyAIAuthID, "玲")
+	if !ai.LoggedIn() {
+		t.Fatal("隔离账号票据存在时应为已登录")
+	}
+	if got := ai.AccountName(); got != "玲" {
+		t.Fatalf("隔离账号昵称 = %q，want 玲", got)
+	}
+	if got := main.AccountName(); got != "Darren" {
+		t.Fatalf("另一个槽位的昵称被带偏: %q", got)
+	}
+}

package/internal/agent/custom_models.go

@@ -71,6 +71,149 @@ func StripCustomLocalPrefixCredits(credits map[string]string) map[string]string
 	return out
 }
 
+// CLICustomModel 一条本地自定义模型在**清单/选择器**里的展示形态。
+type CLICustomModel struct {
+	// ID 「选中时下发给 -m 的值」：优先取 models.json 里 `aliases[0]`
+	// （CLI 的模型查找按 id / name / aliases 命中，别名可直接 -m），
+	// 没有别名就取剥掉 custom-local: 前缀的 id（= 实际转发给端点的模型名）。
+	ID string
+	// Wire 去掉 custom-local: 前缀后的真实 id（= CLI 转发给端点的模型名）。
+	// 用于把「已经从 models.json 删掉、但桌面端缓存里还留着」的旧条目从清单里剔除。
+	Wire string
+	// Vendor 供应商展示名（models.json 的 vendor；空 / "Custom" 时为空）。
+	Vendor string
+	// URL / APIKey 该自定义模型的端点与密钥（仅用于按账号查套餐额度，不外泄到清单）。
+	URL    string
+	APIKey string
+}
+
+// CLICustomModels 读 <configDir>/models.json 里**本地自定义**模型（custom-local:*）条目。
+//
+// 为什么只认 custom-local:：models.json 里还有官方模型条目（id 是裸名），它们不是
+// 「用户自己接的端点」，混进清单会把官方模型重复列一遍。只读**顶层对象**格式
+// （CLI 自己的 models.json）；桌面端数组格式不碰。
+func CLICustomModels(configDir string) []CLICustomModel {
+	if strings.TrimSpace(configDir) == "" {
+		return nil
+	}
+	b, err := os.ReadFile(filepath.Join(configDir, customModelsFileName))
+	if err != nil {
+		return nil
+	}
+	var root map[string]any
+	if err := json.Unmarshal(b, &root); err != nil {
+		return nil
+	}
+	raw, _ := root["models"].([]any)
+	var out []CLICustomModel
+	seen := map[string]bool{}
+	for _, it := range raw {
+		m, ok := it.(map[string]any)
+		if !ok {
+			continue
+		}
+		rawID, _ := m["id"].(string)
+		rawID = strings.TrimSpace(rawID)
+		if !strings.HasPrefix(rawID, customModelIDPrefix) {
+			continue
+		}
+		wire := strings.TrimSpace(strings.TrimPrefix(rawID, customModelIDPrefix))
+		if wire == "" {
+			continue
+		}
+		listID := wire
+		if alias := firstString(m["aliases"]); alias != "" {
+			listID = alias // 同名模型（如官方 glm-5.3-flash）靠别名区分
+		}
+		if seen[listID] {
+			continue
+		}
+		seen[listID] = true
+		vendor, _ := m["vendor"].(string)
+		vendor = strings.TrimSpace(vendor)
+		if strings.EqualFold(vendor, "custom") {
+			vendor = ""
+		}
+		url, _ := m["url"].(string)
+		apiKey, _ := m["apiKey"].(string)
+		out = append(out, CLICustomModel{
+			ID:     listID,
+			Wire:   wire,
+			Vendor: vendor,
+			URL:    strings.TrimSpace(url),
+			APIKey: strings.TrimSpace(apiKey),
+		})
+	}
+	return out
+}
+
+// CLIModelVendors 模型 id（与 CLICustomModels 的 ID 同形）→ 供应商展示名。
+// 没有供应商信息的条目不出现在 map 里；一条都没有返回 nil。
+func CLIModelVendors(configDir string) map[string]string {
+	models := CLICustomModels(configDir)
+	out := map[string]string{}
+	for _, m := range models {
+		if m.Vendor != "" {
+			out[m.ID] = m.Vendor
+		}
+	}
+	if len(out) == 0 {
+		return nil
+	}
+	return out
+}
+
+// firstString 取 []any 里第一个非空字符串（models.json 的 aliases 列表）。
+func firstString(v any) string {
+	list, _ := v.([]any)
+	for _, it := range list {
+		if s, ok := it.(string); ok {
+			if s = strings.TrimSpace(s); s != "" {
+				return s
+			}
+		}
+	}
+	return ""
+}
+
+// MergeCLICustomModels 把 models.json 里的自定义模型并入清单（追加在官方模型之后），
+// 并剔除「已从 models.json 删掉、但桌面端 acc 缓存里还留着」的 custom-local 条目。
+//
+// models.json 不存在时**原样返回**：没有权威来源时不动别人的清单（否则会把 App 里
+// 已配好、只是还没同步到 models.json 的自定义模型一并隐藏）。
+func MergeCLICustomModels(configDir string, ids []string) []string {
+	if strings.TrimSpace(configDir) == "" {
+		return ids
+	}
+	if _, err := os.Stat(filepath.Join(configDir, customModelsFileName)); err != nil {
+		return ids
+	}
+	customs := CLICustomModels(configDir)
+	// 自定义条目统一在末尾追加（顺序由 UI 再按倍率排）；清单里原有的 custom-local:*
+	// 一律先剔除 —— 既去掉「已从 models.json 删掉、缓存里还在」的旧条目，也避免
+	// 「别名条目 + 缓存里的原名」让同一模型出现两次。
+	out := make([]string, 0, len(ids)+len(customs))
+	seen := make(map[string]bool, len(ids)+len(customs))
+	for _, id := range ids {
+		if strings.HasPrefix(id, customModelIDPrefix) {
+			continue
+		}
+		if id == "" || seen[id] {
+			continue
+		}
+		seen[id] = true
+		out = append(out, id)
+	}
+	for _, m := range customs {
+		if m.ID == "" || seen[m.ID] {
+			continue
+		}
+		seen[m.ID] = true
+		out = append(out, m.ID)
+	}
+	return out
+}
+
 // syncCustomModelsFromApp 把 App 合并配置缓存（acc-product-config）里的
 // `custom-local:*` 模型合并进 `<configDir>/models.json`，返回新增条数。
 //
@@ -261,6 +404,18 @@ func (c codebuddyCore) resolveCustomModelID(model string) string {
 	// 同份 acc 缓存里递归收集全部 id（含嵌套、官方与 custom-local:*）：裸名若也是官方模型，
 	// 优先官方、不补前缀；仅当它是「纯自定义」（无同名官方）时才补回 custom-local:。
 	all := readAllAccModelIDs(c.accConfigPath)
+	/* CLI 自己的 models.json 也是权威来源（2026-10-25）：用户直接往那份文件里加的自定义
+	   模型不一定进了 App 的 acc 缓存，只查 acc 缓存就补不回前缀，`-m 裸名` 会找不到端点。
+	   按 id / aliases 命中，统一映射回真正的 `custom-local:<wire>`（wire 才是发给端点
+	   的模型名，见 CLI 的 stripCustomLocalModelPrefix）。 */
+	for _, cm := range CLICustomModels(c.effectiveConfigDir()) {
+		if cm.Wire == "" || all[name] {
+			continue
+		}
+		if name == cm.ID || name == cm.Wire {
+			return customModelIDPrefix + cm.Wire
+		}
+	}
 	if all[customID] && !all[name] {
 		return customID
 	}

package/internal/agent/custom_models_test.go

@@ -130,6 +130,40 @@ func TestCodeBuddyModelArg(t *testing.T) {
 	}
 }
 
+// 只出现在 CLI models.json（不在 App acc 缓存）的自定义模型，裸名也要能补回前缀。
+// 2026-10-25：用户直接往 models.json 里加两个 deepseek-v4.1-flash-1017 变体后报「选了
+// 就报没有 endpoint」，根因就是 resolve 只查 acc 缓存。
+func TestResolveCustomModelIDFromCLIModelsJSON(t *testing.T) {
+	dir := t.TempDir()
+	cfg := filepath.Join(dir, "cfg")
+	if err := os.MkdirAll(cfg, 0o700); err != nil {
+		t.Fatal(err)
+	}
+	writeJSON(t, filepath.Join(cfg, "models.json"), map[string]any{
+		"availableModels": []any{
+			"custom-local:deepseek-v4.1-flash-1017",
+			"custom-local:deepseek-v4.1-flash-1017-ark",
+		},
+		"models": []any{
+			map[string]any{"id": "custom-local:deepseek-v4.1-flash-1017", "vendor": "Modelverse"},
+			map[string]any{"id": "custom-local:deepseek-v4.1-flash-1017-ark", "vendor": "Volcengine"},
+			// aliases：列表用别名，但 -m 必须映射回 wire 名。
+			map[string]any{"id": "custom-local:glm-5.3-flash", "aliases": []any{"glm-5.3-flash-volc"}},
+		},
+	})
+	c := codebuddyCore{accConfigPath: filepath.Join(dir, "missing.json"), configDir: cfg}
+
+	if got := c.modelArg("deepseek-v4.1-flash-1017"); got != "custom-local:deepseek-v4.1-flash-1017" {
+		t.Errorf("modelverse 裸名 = %q", got)
+	}
+	if got := c.modelArg("deepseek-v4.1-flash-1017-ark"); got != "custom-local:deepseek-v4.1-flash-1017-ark" {
+		t.Errorf("ark 裸名 = %q", got)
+	}
+	if got := c.modelArg("glm-5.3-flash-volc"); got != "custom-local:glm-5.3-flash" {
+		t.Errorf("别名应映射回 wire 名，实际 %q", got)
+	}
+}
+
 func TestSyncCustomModelsFromApp_FirstSync(t *testing.T) {
 	dir := t.TempDir()
 	acc := filepath.Join(dir, "acc-product-config-v3.json")
@@ -483,3 +517,94 @@ func hasStringInList(list []any, want string) bool {
 	}
 	return false
 }
+
+// CLIModelVendors：只取有意义的 vendor（空 / "Custom" 过滤），键剥 custom-local: 前缀，
+// 桌面端数组格式（不认识）返回 nil —— 不能把别人的文件当 CLI 格式解析。
+func TestCLIModelVendors(t *testing.T) {
+	dir := t.TempDir()
+	writeJSON(t, filepath.Join(dir, customModelsFileName), map[string]any{
+		"models": []any{
+			map[string]any{
+				"id":      "custom-local:glm-5.3-flash",
+				"vendor":  "Volcengine",
+				"aliases": []any{"ark-glm-5.3-flash"},
+			},
+			map[string]any{"id": "custom-local:deepseek-v4.1-flash-1017", "vendor": "Volcengine"},
+			map[string]any{"id": "custom-local:MiniMax-M3", "vendor": "Custom"},
+			map[string]any{"id": "plain-model", "vendor": ""},
+		},
+	})
+	got := CLIModelVendors(dir)
+	want := map[string]string{
+		"ark-glm-5.3-flash":        "Volcengine", // 有别名：清单/供应商表都用别名（区分同名官方模型）
+		"deepseek-v4.1-flash-1017": "Volcengine",
+	}
+	if len(got) != len(want) {
+		t.Fatalf("vendors = %v, want %v", got, want)
+	}
+	for k, v := range want {
+		if got[k] != v {
+			t.Errorf("vendors[%q] = %q, want %q（全量 %v）", k, got[k], v, got)
+		}
+	}
+
+	// 桌面端数组格式：不是 CLI 的 models.json，返回 nil。
+	arrDir := t.TempDir()
+	if err := os.WriteFile(filepath.Join(arrDir, customModelsFileName),
+		[]byte(`[{"id":"custom-local:x","vendor":"Volcengine"}]`), 0o644); err != nil {
+		t.Fatal(err)
+	}
+	if got := CLIModelVendors(arrDir); got != nil {
+		t.Errorf("桌面端数组格式应返回 nil，得到 %v", got)
+	}
+}
+
+// MergeCLICustomModels：自定义条目统一追加到末尾（有别名用别名）、缓存里已被
+// 删除/重复的 custom-local 条目一律剔除；models.json 不存在时原样返回。
+func TestMergeCLICustomModels(t *testing.T) {
+	dir := t.TempDir()
+	writeJSON(t, filepath.Join(dir, customModelsFileName), map[string]any{
+		"models": []any{
+			map[string]any{
+				"id":      "custom-local:glm-5.3-flash",
+				"aliases": []any{"ark-glm-5.3-flash"},
+				"vendor":  "Volcengine",
+			},
+			map[string]any{"id": "custom-local:MiniMax-M3.1-Flash", "vendor": "MiniMax"},
+		},
+	})
+	ids := []string{
+		"hy3",
+		"custom-local:MiniMax-M3",    // 已从 models.json 删除 → 应剔除
+		"custom-local:glm-5.3-flash", // 有别名 → 用别名追加，原名不重复
+		"custom-local:MiniMax-M3.1-Flash",
+	}
+	want := []string{"hy3", "ark-glm-5.3-flash", "MiniMax-M3.1-Flash"}
+	if got := MergeCLICustomModels(dir, ids); !equalStrings(got, want) {
+		t.Errorf("MergeCLICustomModels = %v, want %v", got, want)
+	}
+
+	// 没有 models.json：没有权威来源，原样返回（不能隐藏 App 侧刚配好的自定义模型）。
+	empty := t.TempDir()
+	if got := MergeCLICustomModels(empty, ids); !equalStrings(got, ids) {
+		t.Errorf("models.json 缺失时应原样返回，得到 %v", got)
+	}
+}
+
+// CLIModelQuotas：只对 MiniMax 端点发查询；非 MiniMax 的自定义模型直接跳过
+// （这里用 Ark 端点验证「不发请求、返回 nil」，不需要真实网络）。
+func TestCLIModelQuotas_NonMiniMax(t *testing.T) {
+	dir := t.TempDir()
+	writeJSON(t, filepath.Join(dir, customModelsFileName), map[string]any{
+		"models": []any{
+			map[string]any{
+				"id":     "custom-local:glm-5.3-flash",
+				"url":    "https://ark.cn-beijing.volces.com/api/coding/v3",
+				"apiKey": "ark-test",
+			},
+		},
+	})
+	if got := CLIModelQuotas(t.Context(), dir); got != nil {
+		t.Errorf("非 MiniMax 端点不该查额度，得到 %v", got)
+	}
+}