magic-agent 1.0.3 → 1.0.49

package/internal/agent/codex_test.go

@@ -63,6 +63,30 @@ func TestCodexBuildArgsModel(t *testing.T) {
 	}
 }
 
+// 续接轮的模型不能走 `-m`：`codex exec resume` 的参数表里没有它（同 `-C` 一样会被 clap
+// 以 `unexpected argument` 拒掉、exit 2、stdout 全空 → core 报「empty result · 0.0s」）。
+// 改为配置覆写 `-c model="<m>"` —— codex 帮助里正是拿 model 当 `-c` 的例子。
+func TestCodexResumeModelViaConfigOverride(t *testing.T) {
+	dir := t.TempDir()
+	w, log := argsCaptureCLI(t, dir, "codex", codexStubWithText("ok"))
+	e := &CodexEngine{BinPath: w.bin}
+	if _, err := e.Complete(context.Background(), Request{
+		SessionID: "th-1",
+		Model:     "codex/MiniMax-M3",
+		Messages:  []Message{{Role: "user", Content: "继续"}},
+	}); err != nil {
+		t.Fatalf("Complete: %v", err)
+	}
+	data, _ := readFile(log)
+	args := string(data)
+	if strings.Contains(args, "-m MiniMax-M3") {
+		t.Errorf("续接轮不该用 -m（resume 子命令没有这个 flag）：%q", args)
+	}
+	if !strings.Contains(args, `-c model="MiniMax-M3"`) {
+		t.Errorf("续接轮的模型应走 -c model=…：%q", args)
+	}
+}
+
 func TestCodexSystemPrompt(t *testing.T) {
 	dir := t.TempDir()
 	w, log := argsCaptureCLI(t, dir, "codex", codexStubWithText("ok"))

package/internal/agent/custom_models.go

@@ -46,6 +46,31 @@ const customModelsBackupSuffix = ".magic-agent.bak"
 // customModelIDPrefix 自定义模型 id 前缀（App 与 CLI 共用这个约定）。
 const customModelIDPrefix = "custom-local:"
 
+// StripCustomLocalPrefix 去掉模型 id 的 "custom-local:" 前缀（仅用于 --engines / --contract
+// 展示：codebuddy 系模型的噪点前缀，给客户端渲染用）。-m 实际传值时由
+// codebuddyCore.modelArg 按裸名补回前缀，自定义模型依旧可用（见该函数的说明）。
+// 不修改入参，返回新切片。
+func StripCustomLocalPrefix(models []string) []string {
+	out := make([]string, 0, len(models))
+	for _, m := range models {
+		out = append(out, strings.TrimPrefix(m, customModelIDPrefix))
+	}
+	return out
+}
+
+// StripCustomLocalPrefixCredits 同 StripCustomLocalPrefix，但作用于「模型 → 倍率」表。
+// custom-local 模型本来就没有倍率（不进表），这里只为键名形态与 models 一致。
+func StripCustomLocalPrefixCredits(credits map[string]string) map[string]string {
+	if len(credits) == 0 {
+		return credits
+	}
+	out := make(map[string]string, len(credits))
+	for k, v := range credits {
+		out[strings.TrimPrefix(k, customModelIDPrefix)] = v
+	}
+	return out
+}
+
 // syncCustomModelsFromApp 把 App 合并配置缓存（acc-product-config）里的
 // `custom-local:*` 模型合并进 `<configDir>/models.json`，返回新增条数。
 //
@@ -207,15 +232,96 @@ func collectCustomModels(node any, out map[string]map[string]any) {
 // ensureCustomModelEndpoint 仅当本次请求用的是一个自定义模型时，才补一次端点。
 //
 // 为什么不止在 listModels 里同步：用户可能在 App 里刚加了一个自定义模型，就直接
-// `-m custom-local:<新>` 调用（没打开过模型选择器）。这条兜底让那种路径也能用上。
-// best-effort：不同步失败也不影响正常模型（官方模型根本不读 models.json）。
+// `-m custom-local:<新>` 或 `-m <裸名>` 调用（没打开过模型选择器）。这条兜底让那种
+// 路径也能用上。best-effort：不同步失败也不影响正常模型（官方模型根本不读 models.json）。
 func (c codebuddyCore) ensureCustomModelEndpoint(model string) {
-	if !strings.HasPrefix(model, customModelIDPrefix) {
+	m := stripModelPrefix(model)
+	// 已是 custom-local: 前缀（用户直传全 id），或裸名命中自定义模型 → 需要补端点。
+	if !strings.HasPrefix(m, customModelIDPrefix) && c.resolveCustomModelID(model) == "" {
 		return
 	}
 	_, _ = syncCustomModelsFromApp(c.accConfigPath, c.effectiveConfigDir())
 }
 
+// resolveCustomModelID 把可能带 "engine/" 或 "custom-local:" 前缀的模型名归一成
+// 自定义模型 id（"custom-local:<裸名>"）。裸名命中 acc 缓存里的自定义模型、且**不是**
+// 同缓存里的官方模型时才补回前缀 —— 避免把官方模型（如 glm-5.3）误改写成
+// custom-local:glm-5.3（用户自定义了一个同名模型时的退化处理：优先官方）。
+// 命中返回该 id，否则返回 ""（不是自定义模型，不补前缀）。
+//
+// 用途：--engines 展示时剥掉了 custom-local: 前缀（见 StripCustomLocalPrefix），用户
+// 选了之后回传的是裸名；-m 需要带前缀才能真正命中自定义模型端点，故在此补回。
+func (c codebuddyCore) resolveCustomModelID(model string) string {
+	name := stripModelPrefix(model)
+	name = strings.TrimPrefix(name, customModelIDPrefix)
+	if name == "" {
+		return ""
+	}
+	customID := customModelIDPrefix + name
+	// 同份 acc 缓存里递归收集全部 id（含嵌套、官方与 custom-local:*）：裸名若也是官方模型，
+	// 优先官方、不补前缀；仅当它是「纯自定义」（无同名官方）时才补回 custom-local:。
+	all := readAllAccModelIDs(c.accConfigPath)
+	if all[customID] && !all[name] {
+		return customID
+	}
+	return ""
+}
+
+// collectAllModelIDs 递归遍历 acc 配置缓存，收集所有模型的 id（官方与 custom-local:*
+// 都收，先到先得）。acc 缓存里官方模型常常嵌在嵌套结构里，故必须递归而非只读顶层。
+func collectAllModelIDs(node any, out map[string]bool) {
+	switch v := node.(type) {
+	case map[string]any:
+		if id, ok := v["id"].(string); ok && id != "" {
+			out[id] = true
+		}
+		for _, child := range v {
+			collectAllModelIDs(child, out)
+		}
+	case []any:
+		for _, child := range v {
+			collectAllModelIDs(child, out)
+		}
+	}
+}
+
+// readAllAccModelIDs 返回 acc 配置缓存里出现过的全部模型 id 集合（递归，不区分官方 / 自定义）。
+// 文件不存在 / 解析失败 → 空集。
+func readAllAccModelIDs(accConfigPath string) map[string]bool {
+	out := map[string]bool{}
+	if accConfigPath == "" {
+		return out
+	}
+	b, err := os.ReadFile(accConfigPath)
+	if err != nil {
+		return out
+	}
+	var doc any
+	if err := json.Unmarshal(b, &doc); err != nil {
+		return out
+	}
+	collectAllModelIDs(doc, out)
+	return out
+}
+
+// modelArg 返回传给 CLI --model 的值。
+//
+// --engines 展示时剥掉了 custom-local: 前缀（见 StripCustomLocalPrefix），用户选了
+// 自定义模型后回传裸名；这里命中自定义模型就补回 custom-local: 前缀，保证 -m 可用。
+func (c codebuddyCore) modelArg(model string) string {
+	m := stripModelPrefix(model)
+	if m == "" {
+		return ""
+	}
+	if strings.HasPrefix(m, customModelIDPrefix) {
+		return m
+	}
+	if id := c.resolveCustomModelID(m); id != "" {
+		return id
+	}
+	return m
+}
+
 // effectiveConfigDir 本次子进程实际使用的配置目录：用户显式设了 CODEBUDDY_CONFIG_DIR
 // 就跟着走（CLI 也这么读，见 codebuddyAccountEnv），否则用引擎自带的那个。
 func (c codebuddyCore) effectiveConfigDir() string {

package/internal/agent/custom_models_test.go

@@ -56,6 +56,80 @@ func appCacheFixture() map[string]any {
 	}
 }
 
+// StripCustomLocalPrefix 剥掉自定义模型 id 的 "custom-local:" 前缀（仅展示用）。
+func TestStripCustomLocalPrefix(t *testing.T) {
+	got := StripCustomLocalPrefix([]string{"hy3", "custom-local:MiniMax-M3", "custom-local:glm-5.3"})
+	want := []string{"hy3", "MiniMax-M3", "glm-5.3"}
+	if len(got) != len(want) {
+		t.Fatalf("StripCustomLocalPrefix = %v want %v", got, want)
+	}
+	for i := range want {
+		if got[i] != want[i] {
+			t.Errorf("StripCustomLocalPrefix[%d] = %q want %q (full %v)", i, got[i], want[i], got)
+		}
+	}
+	c := StripCustomLocalPrefixCredits(map[string]string{"hy3": "0.00", "custom-local:X": "0.34"})
+	if c["custom-local:X"] != "" || c["X"] != "0.34" || c["hy3"] != "0.00" {
+		t.Errorf("StripCustomLocalPrefixCredits = %v", c)
+	}
+}
+
+// resolveCustomModelID：裸名命中纯自定义模型时补回 custom-local: 前缀；
+// 裸名同时是官方模型时**不补**（优先官方，避免把 glm-5.3 误写成 custom-local:glm-5.3）。
+func TestResolveCustomModelID(t *testing.T) {
+	dir := t.TempDir()
+	acc := filepath.Join(dir, "acc.json")
+	writeJSON(t, acc, map[string]any{"models": []any{
+		map[string]any{"id": "glm-5.3"},
+		map[string]any{"id": "custom-local:MiniMax-M3"},
+		// 嵌套结构里的官方模型也要能识别（readAllAccModelIDs 递归收集）。
+		map[string]any{"wrapped": map[string]any{"id": "deepseek-v4.1-flash"}},
+	}})
+	c := codebuddyCore{accConfigPath: acc}
+
+	if got := c.resolveCustomModelID("MiniMax-M3"); got != "custom-local:MiniMax-M3" {
+		t.Errorf("resolve MiniMax-M3 = %q want custom-local:MiniMax-M3", got)
+	}
+	if got := c.resolveCustomModelID("glm-5.3"); got != "" {
+		t.Errorf("resolve glm-5.3 = %q want \"\"（优先官方模型，不补前缀）", got)
+	}
+	// 已带前缀：原样返回。
+	if got := c.resolveCustomModelID("custom-local:MiniMax-M3"); got != "custom-local:MiniMax-M3" {
+		t.Errorf("resolve custom-local:MiniMax-M3 = %q", got)
+	}
+	// 既不是官方也不是自定义 → 不补。
+	if got := c.resolveCustomModelID("nonexistent-model"); got != "" {
+		t.Errorf("resolve nonexistent = %q want \"\"", got)
+	}
+}
+
+// modelArg：传给 CLI --model 的值。展示时剥了前缀的自定义模型裸名要补回前缀。
+func TestCodeBuddyModelArg(t *testing.T) {
+	dir := t.TempDir()
+	acc := filepath.Join(dir, "acc.json")
+	writeJSON(t, acc, map[string]any{"models": []any{
+		map[string]any{"id": "glm-5.3"},
+		map[string]any{"id": "custom-local:MiniMax-M3"},
+	}})
+	c := codebuddyCore{accConfigPath: acc}
+
+	cases := []struct {
+		in, want string
+	}{
+		{"hy3", "hy3"},                                         // 官方：原样
+		{"codebuddy/hy3", "hy3"},                               // 带 engine/ 前缀：剥掉
+		{"glm-5.3", "glm-5.3"},                                 // 同名官方：不补前缀
+		{"MiniMax-M3", "custom-local:MiniMax-M3"},              // 自定义裸名：补前缀
+		{"custom-local:MiniMax-M3", "custom-local:MiniMax-M3"}, // 已带前缀：原样
+		{"", ""}, // 空：空
+	}
+	for _, tc := range cases {
+		if got := c.modelArg(tc.in); got != tc.want {
+			t.Errorf("modelArg(%q) = %q want %q", tc.in, got, tc.want)
+		}
+	}
+}
+
 func TestSyncCustomModelsFromApp_FirstSync(t *testing.T) {
 	dir := t.TempDir()
 	acc := filepath.Join(dir, "acc-product-config-v3.json")