fix: show real project runtime agents

2026-05-25 23:40:52 +08:00
parent 4262191462
commit b6648f384d
6 changed files with 371 additions and 26 deletions
--- a/internal/runtime/store_test.go
+++ b/internal/runtime/store_test.go
@@ -66,6 +66,100 @@ func TestStoreReadsThreadsEdgesAndGoalsFromReadonlySQLite(t *testing.T) {
 	}
 }

+func TestStoreReadsCurrentCodexRuntimeSchema(t *testing.T) {
+	root := t.TempDir()
+	stateDB := openWritableSQLite(t, filepath.Join(root, "state_5.sqlite"))
+	defer stateDB.Close()
+	execSQL(t, stateDB, `CREATE TABLE threads (
+		id TEXT PRIMARY KEY,
+		rollout_path TEXT NOT NULL,
+		created_at INTEGER NOT NULL,
+		updated_at INTEGER NOT NULL,
+		source TEXT NOT NULL,
+		model_provider TEXT NOT NULL,
+		cwd TEXT NOT NULL,
+		title TEXT NOT NULL,
+		sandbox_policy TEXT NOT NULL,
+		approval_mode TEXT NOT NULL,
+		tokens_used INTEGER NOT NULL DEFAULT 0,
+		has_user_event INTEGER NOT NULL DEFAULT 0,
+		archived INTEGER NOT NULL DEFAULT 0,
+		archived_at INTEGER,
+		git_sha TEXT,
+		git_branch TEXT,
+		git_origin_url TEXT,
+		cli_version TEXT NOT NULL DEFAULT '',
+		first_user_message TEXT NOT NULL DEFAULT '',
+		agent_nickname TEXT,
+		agent_role TEXT,
+		memory_mode TEXT NOT NULL DEFAULT 'enabled',
+		model TEXT,
+		reasoning_effort TEXT,
+		agent_path TEXT,
+		created_at_ms INTEGER,
+		updated_at_ms INTEGER,
+		thread_source TEXT,
+		preview TEXT NOT NULL DEFAULT ''
+	)`)
+	execSQL(t, stateDB, `CREATE TABLE thread_spawn_edges (
+		parent_thread_id TEXT NOT NULL,
+		child_thread_id TEXT NOT NULL PRIMARY KEY,
+		status TEXT NOT NULL
+	)`)
+	execSQL(t, stateDB, `INSERT INTO threads (
+		id, rollout_path, created_at, updated_at, source, model_provider, cwd, title, sandbox_policy, approval_mode,
+		cli_version, agent_nickname, agent_role, agent_path, created_at_ms, updated_at_ms, thread_source, preview
+	) VALUES
+		('thread-a', '/tmp/a.jsonl', 1, 2, 'codex', 'openai', '/repo/a', '主线程', 'workspace-write', 'on-request',
+			'0.0.1', '主控', '智能体编排者', '/agents/orchestrator.toml', 1000, 2000, 'cli', '监管项目流程'),
+		('thread-b', '/tmp/b.jsonl', 3, 4, 'codex', 'openai', '/repo/b', '审查线程', 'workspace-write', 'on-request',
+			'0.0.1', '审查员', '代码审查员', '/agents/reviewer.toml', 3000, 4000, 'cli', '审查实现')`)
+	execSQL(t, stateDB, `INSERT INTO thread_spawn_edges (parent_thread_id, child_thread_id, status) VALUES
+		('thread-a', 'thread-b', 'spawned')`)
+
+	goalsDB := openWritableSQLite(t, filepath.Join(root, "goals_1.sqlite"))
+	defer goalsDB.Close()
+	execSQL(t, goalsDB, `CREATE TABLE thread_goals (
+		thread_id TEXT PRIMARY KEY NOT NULL,
+		goal_id TEXT NOT NULL,
+		objective TEXT NOT NULL,
+		status TEXT NOT NULL,
+		token_budget INTEGER,
+		tokens_used INTEGER NOT NULL DEFAULT 0,
+		time_used_seconds INTEGER NOT NULL DEFAULT 0,
+		created_at_ms INTEGER NOT NULL,
+		updated_at_ms INTEGER NOT NULL
+	)`)
+	execSQL(t, goalsDB, `INSERT INTO thread_goals (
+		thread_id, goal_id, objective, status, token_budget, tokens_used, time_used_seconds, created_at_ms, updated_at_ms
+	) VALUES
+		('thread-a', 'goal-a', '管理当前项目的智能体流程', 'active', 10000, 1200, 60, 1000, 2000)`)
+
+	snapshot, err := Store{CodexHome: root}.Snapshot()
+	if err != nil {
+		t.Fatalf("Snapshot returned error: %v", err)
+	}
+	if len(snapshot.Threads) != 2 {
+		t.Fatalf("threads = %#v", snapshot.Threads)
+	}
+	first := snapshot.Threads[0]
+	if first.ID != "thread-a" || first.CWD != "/repo/a" || first.Title != "主线程" {
+		t.Fatalf("unexpected first thread identity: %#v", first)
+	}
+	if first.AgentNickname != "主控" || first.AgentRole != "智能体编排者" || first.Role != "智能体编排者" {
+		t.Fatalf("unexpected first thread agent fields: %#v", first)
+	}
+	if first.AgentPath != "/agents/orchestrator.toml" || first.Preview != "监管项目流程" || first.CreatedAt != "1000" || first.UpdatedAt != "2000" {
+		t.Fatalf("unexpected first thread metadata: %#v", first)
+	}
+	if len(snapshot.SpawnEdges) != 1 || snapshot.SpawnEdges[0].FromThreadID != "thread-a" || snapshot.SpawnEdges[0].ToThreadID != "thread-b" || snapshot.SpawnEdges[0].Reason != "spawned" {
+		t.Fatalf("unexpected current-schema edges: %#v", snapshot.SpawnEdges)
+	}
+	if len(snapshot.Goals) != 1 || snapshot.Goals[0].ThreadID != "thread-a" || snapshot.Goals[0].Goal != "管理当前项目的智能体流程" || snapshot.Goals[0].Status != "active" || snapshot.Goals[0].UpdatedAt != "2000" {
+		t.Fatalf("unexpected current-schema goals: %#v", snapshot.Goals)
+	}
+}
+
 func TestStoreMarksStateMissingWhenOnlyGoalsSQLiteExists(t *testing.T) {
 	root := t.TempDir()
 	createRuntimeSQLite(t, root)