magic-agent 1.0.49 → 1.0.109

package/internal/cli/team_test.go

@@ -0,0 +1,360 @@
+package cli
+
+// team_test.go - `--team` 的 CLI 层接线测试。
+//
+// 边界：这里**只验接线**（分发、参数校验、事件形状、收尾 envelope、会话登记）。
+// 团队模式自己的逻辑（计划解析 / 分层 / 路由 / 降级）在 internal/team 里有单测，
+// 这里不重复覆盖 —— 用一个会回固定计划的假引擎把流程走通即可。
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+	"os"
+	"path/filepath"
+	"strings"
+	"sync"
+	"sync/atomic"
+	"testing"
+
+	"github.com/darren/magic-agent/internal/agent"
+	"github.com/darren/magic-agent/internal/session"
+)
+
+/* uniqueTeamEngine 给每个用例一个**全局唯一**的假引擎名。
+ *
+ * 为什么必须唯一：`agent.Register` 只 append，`agent.Lookup` 按名返回**第一个**命中
+ * （见 engine.go 的 registry 注释 —— 内置引擎的惰性注册挂在「registry 为空」上，
+ * 改成覆盖/无条件注册会打断另一批用例）。`-count=2` 时同一个用例跑两遍，
+ * 第二遍 `Lookup` 拿到的是第一遍那个实例，而 `teamPlanEngine.calls` 早已 >1 →
+ * 它把「正文」当成计划返回 → 计划解析失败降级成单任务，断言全线崩
+ * （2026-10-21 实测：`result.tasks = 1, want 2`、`TASK-t2.md` 不存在）。
+ * 这在 `-count=1` 下看不出来，属于**测试隔离**问题而非产品行为。 */
+var teamEngineSeq int64
+
+func uniqueTeamEngine(base string) string {
+	return fmt.Sprintf("%s-%d", base, atomic.AddInt64(&teamEngineSeq, 1))
+}
+
+// teamPlanEngine 假引擎：第一次调用（计划阶段）回一段计划 JSON，之后回正文。
+type teamPlanEngine struct {
+	name string
+	plan string
+	text string
+
+	mu    sync.Mutex
+	calls int
+}
+
+func (e *teamPlanEngine) Name() string           { return e.name }
+func (e *teamPlanEngine) Detect() (bool, string) { return true, "fake://" + e.name }
+func (e *teamPlanEngine) Complete(ctx context.Context, req agent.Request) (agent.Response, error) {
+	t, plan := e.next()
+	if !plan {
+		writeDesignDoc(req.Workspace)
+	}
+	return agent.Response{Engine: e.name, Text: t}, nil
+}
+func (e *teamPlanEngine) Stream(ctx context.Context, req agent.Request, onEvent func(agent.StreamEvent)) (agent.StreamResult, error) {
+	t, plan := e.next()
+	if !plan {
+		writeDesignDoc(req.Workspace)
+	}
+	onEvent(agent.StreamEvent{Kind: agent.KindText, Text: t})
+	return agent.StreamResult{Response: agent.Response{Engine: e.name, Text: t}}, nil
+}
+func (e *teamPlanEngine) next() (string, bool) {
+	e.mu.Lock()
+	defer e.mu.Unlock()
+	e.calls++
+	if e.calls == 1 {
+		return e.plan, true
+	}
+	return e.text, false
+}
+
+// writeDesignDoc 按计划里 t1 的 deliverables 落一个真文件。
+//
+// 团队模式会在收尾时核对交付物（缺了会重试一次，仍缺则 t1 失败、t2 被级联跳过），
+// 假引擎不写文件就只剩预开卡、看不到「开跑」那一条。
+func writeDesignDoc(dir string) {
+	if dir == "" {
+		return
+	}
+	if err := os.MkdirAll(dir, 0o755); err != nil {
+		return
+	}
+	_ = os.WriteFile(filepath.Join(dir, "design.md"), []byte("design"), 0o644)
+}
+
+const twoTaskPlan = `{"goal":"做一个小游戏","tasks":[
+  {"id":"t1","kind":"text","title":"设计文档","prompt":"写一份设计文档","deliverables":["design.md"]},
+  {"id":"t2","kind":"text","title":"实现","prompt":"按文档实现","deps":["t1"]}
+]}`
+
+// TestTeamAskEndToEnd 跑一整轮 `--team`：必须有 ready → 委派卡 → 收尾 result(mode=team)，
+// 并且运行目录里落下了 PLAN.md / TASK-*.md。
+func TestTeamAskEndToEnd(t *testing.T) {
+	name := uniqueTeamEngine("fake-team")
+	eng := &teamPlanEngine{name: name, plan: twoTaskPlan, text: "做完了"}
+	registerFake(eng)
+
+	runDir := t.TempDir()
+	stdout, stderr, err := runAskCmd(t, "",
+		"-e", name, "--stream", "--events", "--team",
+		"--team-run-dir", runDir,
+		"--save-registry",
+		"做一个小游戏")
+	if err != nil {
+		t.Fatalf("--team 失败: %v\nstderr=%s", err, stderr)
+	}
+
+	lines := nonEmptyLines(stdout)
+	if len(lines) < 3 {
+		t.Fatalf("事件太少（%d 行）：\n%s", len(lines), stdout)
+	}
+	// 首行必须是 ready（--events 约定）。
+	if typ := eventType(t, lines[0]); typ != "ready" {
+		t.Errorf("首行 type = %q, want ready", typ)
+	}
+	// 收尾必须是 result，且带 mode/plan_path/run_dir/tasks。
+	last := decodeEventLine(t, lines[len(lines)-1])
+	if last["type"] != "result" {
+		t.Fatalf("末行 type = %v, want result（每轮以 result 或 error 收尾）", last["type"])
+	}
+	if last["mode"] != "team" {
+		t.Errorf("result.mode = %v, want team", last["mode"])
+	}
+	if s, _ := last["plan_path"].(string); !strings.HasSuffix(s, "PLAN.md") {
+		t.Errorf("result.plan_path = %v", last["plan_path"])
+	}
+	tasks, _ := last["tasks"].([]any)
+	if len(tasks) != 2 {
+		t.Errorf("result.tasks 数量 = %d, want 2", len(tasks))
+	}
+
+	// 主会话只有**一张任务列表卡**：计划落定先发一份（两行都在），每个任务开跑 /
+	// 收尾再整份重发；子任务的过程事件写各自的会话日志，不转发到主会话。
+	var snaps [][]map[string]any
+	var parented, taskCards int
+	for _, ln := range lines {
+		m := decodeEventLine(t, ln)
+		switch m["type"] {
+		case "tasks":
+			text, _ := m["text"].(string)
+			var snap struct {
+				Tasks []map[string]any `json:"tasks"`
+			}
+			if err := json.Unmarshal([]byte(text), &snap); err != nil {
+				t.Fatalf("tasks 快照不是合法 JSON：%v\n%s", err, text)
+			}
+			snaps = append(snaps, snap.Tasks)
+		case "tool_use":
+			if m["name"] == "Task" {
+				taskCards++
+			}
+		}
+		if p, _ := m["parent"].(string); p != "" {
+			parented++
+		}
+	}
+	if len(snaps) == 0 {
+		t.Fatal("没有 tasks 快照：主会话里只会显示任务列表卡")
+	}
+	if len(snaps[0]) != 2 {
+		t.Errorf("首份快照应当有两行（计划确定即全部登记），实际 %d 行", len(snaps[0]))
+	}
+	lastSnap := snaps[len(snaps)-1]
+	if len(lastSnap) != 2 {
+		t.Errorf("末份快照应当仍是两行全量，实际 %d 行", len(lastSnap))
+	}
+	for _, row := range lastSnap {
+		if row["status"] != "done" {
+			t.Errorf("末份快照里 %v 应当 done，实际 %v", row["id"], row["status"])
+		}
+		// 子任务 = 普通会话：快照里带它的登记 run_id，列表据此挂到主会话下面。
+		if id, _ := row["run_id"].(string); id == "" {
+			t.Errorf("%v 缺少 run_id：子会话没登记", row["id"])
+		}
+	}
+	if taskCards != 0 {
+		t.Errorf("不该再有委派卡（tool_use Task = %d）", taskCards)
+	}
+	if parented != 0 {
+		t.Errorf("子任务事件不该转发到主会话（parent = %d 条）", parented)
+	}
+
+	// 运行目录的文档。⚠️ `--team-run-dir` 只是**父目录**：真正的运行目录是
+	// `<父>/<run_id>`（同一台机器跑多轮团队会话时互不覆盖）—— 所以这里要下钻一层。
+	runs, _ := os.ReadDir(runDir)
+	if len(runs) != 1 {
+		t.Fatalf("父目录下应当恰好有一个运行目录，实际 %d 个", len(runs))
+	}
+	actual := filepath.Join(runDir, runs[0].Name())
+	for _, name := range []string{"PLAN.md", "TASK-t1.md", "TASK-t2.md", "OUT-t1.md", "OUT-t2.md"} {
+		if _, err := os.Stat(filepath.Join(actual, name)); err != nil {
+			t.Errorf("运行目录缺少 %s：%v", name, err)
+		}
+	}
+	// PLAN.md 是只读的（写权是机制保证的）。
+	if st, err := os.Stat(filepath.Join(actual, "PLAN.md")); err == nil {
+		if st.Mode().Perm()&0o200 != 0 {
+			t.Errorf("PLAN.md 应当只读，实际权限 %v", st.Mode().Perm())
+		}
+	}
+}
+
+// TestTeamAskRecordsMode 团队会话在登记表里带 mode=team 与 agents。
+func TestTeamAskRecordsMode(t *testing.T) {
+	name := uniqueTeamEngine("fake-team-reg")
+	eng := &teamPlanEngine{name: name, plan: twoTaskPlan, text: "ok"}
+	registerFake(eng)
+
+	if _, _, err := runAskCmd(t, "",
+		"-e", name, "--stream", "--events", "--team",
+		"--team-run-dir", t.TempDir(), "--save-registry",
+		"做一个小游戏"); err != nil {
+		t.Fatalf("--team 失败: %v", err)
+	}
+
+	recs, err := session.List()
+	if err != nil {
+		t.Fatalf("读登记表: %v", err)
+	}
+	/* 子任务是**普通会话**，和主会话同引擎，所以不能只按 Engine 认主记录 ——
+	   得认 mode=team 且没有父指针的那条。 */
+	var found *session.Record
+	var children []session.Record
+	for i := range recs {
+		if recs[i].Engine != name {
+			continue
+		}
+		if recs[i].Mode == session.ModeTeam && recs[i].ParentRunID == "" {
+			found = &recs[i]
+		}
+		if recs[i].ParentRunID != "" {
+			children = append(children, recs[i])
+		}
+	}
+	if found == nil {
+		t.Fatal("登记表里没有这条团队会话")
+	}
+	if found.Mode != session.ModeTeam {
+		t.Errorf("record.mode = %q, want team", found.Mode)
+	}
+	if len(found.Agents) == 0 {
+		t.Error("record.agents 为空：列表页看不出「谁在干活」")
+	}
+	if found.State != session.StateDone {
+		t.Errorf("record.state = %q, want done", found.State)
+	}
+	// 每个子任务都是一条**普通会话**（独立 run/session、可点进去看过程），
+	// 靠父指针挂在主会话下面 —— 不是非交互的一次性运行。
+	if len(children) != 2 {
+		t.Fatalf("应当为 2 个子任务各登记一条子会话，得到 %d 条：%+v", len(children), children)
+	}
+	for _, c := range children {
+		if c.ParentRunID != found.RunID {
+			t.Errorf("子会话 %s 的 parent_run_id = %q，want %q", c.RunID, c.ParentRunID, found.RunID)
+		}
+		if c.State != session.StateDone {
+			t.Errorf("子会话 %s 的状态 = %q，want done", c.RunID, c.State)
+		}
+	}
+
+	/* 还要从**列表接口**再看一眼：登记表里有了不代表用户看得到 ——
+	   `--sessions` 是把 Record 投影成另一套 row 结构（internal/cli/session.go 的
+	   匿名 struct，字段漏写就是静默消失）。第一版就漏了 mode / agents：
+	   会话是对的，列表上却是一条普通会话。这里钉住投影本身。 */
+	stdout, _, err := runAskCmd(t, "", "--sessions")
+	if err != nil {
+		t.Fatalf("--sessions: %v", err)
+	}
+	var rows []map[string]any
+	if err := json.Unmarshal([]byte(stdout), &rows); err != nil {
+		t.Fatalf("--sessions 输出不是 JSON：%v", err)
+	}
+	var row map[string]any
+	for _, r := range rows {
+		if r["run_id"] == found.RunID {
+			row = r
+			break
+		}
+	}
+	if row == nil {
+		t.Fatalf("--sessions 里没有 run_id=%s", found.RunID)
+	}
+	if row["mode"] != session.ModeTeam {
+		t.Errorf("--sessions 的 mode = %v，want %q（列表页靠它区分团队会话）", row["mode"], session.ModeTeam)
+	}
+	if ags, _ := row["agents"].([]any); len(ags) == 0 {
+		t.Errorf("--sessions 的 agents 为空/缺失：%v", row["agents"])
+	}
+	// 子会话行必须带父指针，否则列表页没法把它缩进挂到主会话下面。
+	for _, c := range children {
+		var kid map[string]any
+		for _, r := range rows {
+			if r["run_id"] == c.RunID {
+				kid = r
+				break
+			}
+		}
+		if kid == nil {
+			t.Errorf("--sessions 里没有子会话 run_id=%s", c.RunID)
+			continue
+		}
+		if kid["parent_run_id"] != found.RunID {
+			t.Errorf("子会话 %s 的 parent_run_id = %v，want %q", c.RunID, kid["parent_run_id"], found.RunID)
+		}
+	}
+}
+
+// TestTeamRequiresStream `--team` 不配 --stream 必须明确拒绝（exit 2），
+// 而不是悄悄忽略开关 —— 后者会让用户以为开了团队模式却没开。
+func TestTeamRequiresStream(t *testing.T) {
+	registerFake(&stringEngine{name: "fake-nostream", text: "x"})
+	_, stderr, err := runAskCmd(t, "", "-e", "fake-nostream", "--team", "hi")
+	if err == nil {
+		t.Fatal("--team 没有 --stream 时应当报错")
+	}
+	if !strings.Contains(err.Error(), "--stream") && !strings.Contains(stderr, "--stream") {
+		t.Errorf("报错应当提到 --stream：err=%v stderr=%q", err, stderr)
+	}
+}
+
+// TestTeamRouteParsingInvalid 非法 --team-route 要在开跑前就报（exit 2）。
+func TestTeamRouteParsingInvalid(t *testing.T) {
+	// 必须用**流式**引擎：团队路径先校验引擎有没有 Stream，再看路由 ——
+	// 拿 stringEngine 会先撞「不支持流式」，测的就不是路由校验了。
+	name := uniqueTeamEngine("fake-badroute")
+	registerFake(&teamPlanEngine{name: name, plan: twoTaskPlan, text: "x"})
+	_, _, err := runAskCmd(t, "",
+		"-e", name, "--stream", "--team", "--team-route", "audio=llm", "hi")
+	if err == nil {
+		t.Fatal("未知类别应当报错")
+	}
+	if !strings.Contains(err.Error(), "类别") {
+		t.Errorf("err = %v，应提到类别", err)
+	}
+}
+
+// nonEmptyLines 去掉空行。
+func nonEmptyLines(s string) []string {
+	out := make([]string, 0, 64)
+	for _, ln := range strings.Split(s, "\n") {
+		if strings.TrimSpace(ln) != "" {
+			out = append(out, ln)
+		}
+	}
+	return out
+}
+
+// eventType 取一行的 type。
+func eventType(t *testing.T, line string) string {
+	t.Helper()
+	typ, _ := decodeEventLine(t, line)["type"].(string)
+	return typ
+}
+
+// 保持 json import 有用（decodeEventLine 在 events_test.go，这里只做断言辅助）。
+var _ = json.Marshal

package/internal/client/plugins.go

@@ -70,9 +70,11 @@ func pluginSpecs() []pluginSpec {
 			ID:      "magic-shorts",
 			Name:    "magic-shorts",
 			CLI:     "magic-shorts",
-			Tagline: "知识短视频一条龙：选题 → 脚本 → 配音 → 素材 → 设计 → 成片",
+			Tagline: "擅长知识短视频：选题 → 脚本 → 配音 → 素材 → 设计 → 成片",
 			Description: "把一句话主题做成可直接发布的横版/竖版短视频：先扫爆款定选题，再成稿、评审、分段配音、" +
-				"按设计规范合成成片。工程自带工作台（magic-shorts serve，端口 8899）与 macOS 菜单栏 App；" +
+				"按设计规范合成成片。这是一套**成片能力**（选题、配音、设计规范都在里面），" +
+				"不只是转码工具 —— 选它就是让这一轮由擅长做短视频的 CLI 来主导。" +
+				"工程自带工作台（magic-shorts serve，端口 8899）与 macOS 菜单栏 App；" +
 				"数据落在 ~/.magic-shorts/。",
 			Requirements: "Node ≥18、Python ≥3.10（skillflow 依赖 click）",
 			Homepage:     "https://github.com/javacn/magic-shorts",
@@ -98,9 +100,10 @@ func pluginSpecs() []pluginSpec {
 			ID:      "magic-video",
 			Name:    "magic-video",
 			CLI:     "magic-video",
-			Tagline: "短剧剧本 / 小说 / 视频分镜：三域一体的内容生产 CLI",
+			Tagline: "擅长短剧剧本 / 小说 / 视频分镜：三域一体的内容 CLI",
 			Description: "一个入口管三条内容线：novel（长篇/短篇小说）、script（180–210 秒一集的短剧剧本）、" +
-				"video（分镜与媒体）。另有 workbench 本地工作台、doctor 健康检查与 cost 用量统计；" +
+				"video（分镜与媒体）。它的强项是**写与拆**（剧本骨架、分集钩子、分镜），" +
+				"再往下才是画面与媒体处理。另有 workbench 本地工作台、doctor 健康检查与 cost 用量统计；" +
 				"数据落在 ~/.magic-video/。",
 			Requirements: "Node ≥18（npm 包内自带二进制，无需 Go 环境）",
 			Homepage:     "https://github.com/javacn/magic-video",