first

2026-07-12 21:26:08 +08:00
commit 9dd41afd48
502 changed files with 129901 additions and 0 deletions
@@ -0,0 +1,172 @@
+import { describe, it } from "node:test";
+import assert from "node:assert/strict";
+
+import {
+	graphCommunityDrawerViewModel,
+	graphGroupDrawerPromptAction,
+	graphSelectionGroupDrawerViewModel,
+	groupDrawerActionById,
+} from "../src/lib/graph-group-drawer";
+import type { GraphCommunitySummaryPayload, Selection } from "@llm-wiki/graph-engine";
+
+describe("graph group drawer view model", () => {
+	it("keeps normal community actions stable and enter-community available", () => {
+		const view = graphCommunityDrawerViewModel(summaryFixture({
+			coreNodeIds: ["a", "b", "c", "d"],
+			coreNodes: [
+				{ nodeId: "a", label: "Alpha", type: "topic", role: "核心" },
+				{ nodeId: "b", label: "Beta", type: "entity", role: "相关" },
+				{ nodeId: "c", label: "Gamma", type: "source", role: "相关" },
+				{ nodeId: "d", label: "Delta", type: "entity", role: "相关" },
+			],
+		}));
+
+		assert.equal(view.kicker, "社区");
+		assert.equal(view.title, "Knowledge Build");
+		assert.equal(view.canEnterCommunity, true);
+		assert.equal(view.recommendedActionId, "summarize_cluster");
+		assert.equal(view.nodeListExpandable, true);
+		assert.equal(view.nodeListKey, JSON.stringify(["community", "build", ["a", "b", "c", "d"]]));
+		assert.equal(view.dialogueHint, "当前社区会带入对话");
+		assert.deepEqual(view.facts, [
+			{ label: "页", value: 6 },
+			{ label: "链接", value: 5 },
+			{ label: "核心", value: 4 },
+			{ label: "孤立", value: 0 }
+		]);
+		assert.deepEqual(view.nodes.map((node) => node.nodeId), ["a", "b", "c", "d"]);
+		assert.deepEqual(view.actions.map((action) => action.label), [
+			"总结这一簇",
+			"找知识缺口",
+			"生成主题页",
+			"探索潜在关系"
+		]);
+		assert.equal(view.actions.find((action) => action.id === "explore_potential_links")?.recommended, false);
+		assert.equal(view.tags.includes("结构清晰"), true);
+		assert.equal(view.tags.includes("无搜索命中"), false);
+	});
+
+	it("recommends potential relation exploration for ungrouped community", () => {
+		const view = graphCommunityDrawerViewModel(summaryFixture({
+			communityId: "_none",
+			label: "未分组",
+			structureState: "ungrouped",
+			canEnterCommunity: false,
+			facts: { pageCount: 2, internalLinkCount: 0, communityCount: 1, isolatedCount: 2 },
+		}));
+
+		assert.equal(view.canEnterCommunity, false);
+		assert.equal(view.recommendedActionId, "explore_potential_links");
+		assert.equal(view.actions.find((action) => action.id === "explore_potential_links")?.recommended, true);
+		assert.equal(view.tags.includes("暂未成组"), true);
+	});
+
+	it("recommends find_knowledge_gaps for loose-structure community", () => {
+		const view = graphCommunityDrawerViewModel(summaryFixture({
+			structureState: "loose",
+			facts: { pageCount: 8, internalLinkCount: 1, communityCount: 1, isolatedCount: 5 },
+		}));
+		assert.equal(view.recommendedActionId, "find_knowledge_gaps");
+		assert.equal(view.actions.find((a) => a.id === "find_knowledge_gaps")?.recommended, true);
+		// 锁死"每屏只有一个推荐高亮"不变量
+		assert.equal(view.actions.filter((a) => a.recommended).length, 1);
+	});
+
+	it("uses the same skeleton for manual multi-node selections", () => {
+		const view = graphSelectionGroupDrawerViewModel("选区", selectionFixture({
+			id: "nodes:a,b,c,d",
+			nodeIds: ["a", "b", "c", "d"],
+		}));
+
+		assert.equal(view.kicker, "选区");
+		assert.equal(view.title, "选区");
+		assert.equal(view.canEnterCommunity, false);
+		assert.equal(view.recommendedActionId, "explore_potential_links");
+		assert.equal(view.nodeListExpandable, false);
+		assert.equal(view.nodeListKey, JSON.stringify(["selection", "nodes:a,b,c,d"]));
+		assert.equal(view.dialogueHint, "当前选区会带入对话");
+		assert.deepEqual(view.nodes.map((node) => node.nodeId), ["a", "b", "c"]);
+		assert.deepEqual(view.facts, [
+			{ label: "页", value: 3 },
+			{ label: "链接", value: 0 },
+			{ label: "社区", value: 2 },
+			{ label: "孤立", value: 1 }
+		]);
+		assert.deepEqual(view.actions.map((action) => action.label), [
+			"总结这一簇",
+			"找知识缺口",
+			"生成主题页",
+			"探索潜在关系"
+		]);
+	});
+
+	it("finds fixed actions by id for prompt dispatch", () => {
+		assert.equal(groupDrawerActionById("find_knowledge_gaps")?.label, "找知识缺口");
+		assert.equal(groupDrawerActionById("missing"), null);
+		assert.equal(groupDrawerActionById(null), null);
+	});
+
+	it("uses unambiguous community node-list keys when node ids contain separators", () => {
+		const first = graphCommunityDrawerViewModel(summaryFixture({
+			coreNodeIds: ["wiki/a,b.md", "wiki/c.md"],
+		}));
+		const second = graphCommunityDrawerViewModel(summaryFixture({
+			coreNodeIds: ["wiki/a.md", "wiki/b,c.md"],
+		}));
+
+		assert.notEqual(first.nodeListKey, second.nodeListKey);
+	});
+
+	it("keeps free-text sends free and uses the recommended action only for empty new conversations", () => {
+		assert.equal(graphGroupDrawerPromptAction(null, "summarize_cluster", "请只看风险", false), null);
+		assert.equal(graphGroupDrawerPromptAction(null, "summarize_cluster", "请只看风险", true), null);
+		assert.equal(graphGroupDrawerPromptAction(null, "summarize_cluster", "", true)?.id, "summarize_cluster");
+		assert.equal(graphGroupDrawerPromptAction("create_topic_page", "summarize_cluster", "", false)?.id, "create_topic_page");
+	});
+});
+
+function summaryFixture(overrides: Partial<GraphCommunitySummaryPayload> = {}): GraphCommunitySummaryPayload {
+	return {
+		kind: "community-summary",
+		object: { kind: "community", communityId: "build" },
+		communityId: "build",
+		label: "Knowledge Build",
+		nodeCount: 6,
+		facts: { pageCount: 6, internalLinkCount: 5, communityCount: 1, isolatedCount: 0 },
+		structureState: "clear",
+		description: "这组页面围绕同一主题聚在一起。你可以先看结构，也可以直接让 agent 基于这一组页面继续工作。",
+		canEnterCommunity: true,
+		coreNodeIds: ["a", "b", "c"],
+		coreNodes: [
+			{ nodeId: "a", label: "Alpha", type: "topic", role: "核心" },
+			{ nodeId: "b", label: "Beta", type: "entity", role: "相关" },
+			{ nodeId: "c", label: "Gamma", type: "source", role: "相关" }
+		],
+		searchResultIds: [],
+		pinHints: [],
+		selection: {
+			input: { kind: "community", id: "build" },
+			selectionId: "community:a,b,c",
+			selectedNodeIds: ["a", "b", "c"],
+			selectedCommunityIds: ["build"],
+			containsCurrentObject: true
+		},
+		strongestRelations: [],
+		bridgeRelations: [],
+		aggregationMarkers: [],
+		commands: [{ kind: "enter-community", communityId: "build", label: "进入社区" }],
+		...overrides
+	};
+}
+
+function selectionFixture(overrides: Partial<Selection> = {}): Selection {
+	return {
+		id: "nodes:a,b,c",
+		nodeIds: ["a", "b", "c"],
+		communityIds: ["alpha", "beta"],
+		facts: { pageCount: 3, internalLinkCount: 0, communityCount: 2, isolatedCount: 1 },
+		input: { kind: "nodes", ids: ["a", "b", "c"] },
+		actions: [{ id: "explore_potential_links", label: "探索潜在关系", tone: "bridge" }],
+		...overrides
+	};
+}