first

2026-07-12 21:26:08 +08:00
commit 9dd41afd48
502 changed files with 129901 additions and 0 deletions
@@ -0,0 +1,98 @@
+import { describe, it } from "node:test";
+import assert from "node:assert/strict";
+
+import { resolveSelection, UNGROUPED_COMMUNITY_ID, type GraphData } from "@llm-wiki/graph-engine";
+import { buildSelectionPromptPayload } from "../src/lib/graph-selection";
+
+describe("graph selection prompt payload", () => {
+	it("expands a community selection into page list, links, and communities", () => {
+		const data = fixtureGraph();
+		const selection = resolveSelection(data, { kind: "community", id: "alpha" });
+		const action = selection.actions?.find((item) => item.id === "summarize_cluster") ?? null;
+		const payload = buildSelectionPromptPayload(data, selection, action);
+
+		assert.match(payload.displayText, /@\[选区:Alpha Group · 2页\] 总结这一簇/);
+		assert.match(payload.expandedText, /动作：总结这一簇/);
+		assert.match(payload.expandedText, /页面清单：/);
+		assert.match(payload.expandedText, /1\. \[\[wiki\/topics\/alpha-a\.md\]\] - Alpha A - 社区 alpha/);
+		assert.match(payload.expandedText, /2\. \[\[wiki\/entities\/alpha-b\.md\]\] - Alpha B - 社区 alpha/);
+		assert.match(payload.expandedText, /链接关系：/);
+		assert.match(payload.expandedText, /- a -> b \(EXTRACTED\)/);
+		assert.match(payload.expandedText, /请基于上面的选区信息回答/);
+	});
+
+	it("uses the shared ungrouped community label when formatting selection prompts", () => {
+		const data = fixtureGraph();
+		data.nodes = [
+			{ id: "loose-a", label: "Loose A", type: "topic", community: null, source_path: "wiki/loose/a.md" },
+			{ id: "loose-b", label: "Loose B", type: "entity", source_path: "wiki/loose/b.md" },
+		];
+		data.edges = [];
+		data.learning = { ...data.learning!, communities: [] };
+		const selection = resolveSelection(data, { kind: "community", id: UNGROUPED_COMMUNITY_ID });
+		const payload = buildSelectionPromptPayload(data, selection, null);
+		assert.match(payload.displayText, /@\[选区:未分组 · 2页\]/);
+		assert.match(payload.expandedText, /Loose A - 社区 未分组/);
+		assert.doesNotMatch(payload.expandedText, /社区 _none/);
+	});
+
+	it("uses stored synthesis session paths when sending graph nodes into chat", () => {
+		const data = fixtureGraph();
+		data.nodes = [
+			{
+				id: "session-a",
+				label: "Session A",
+				type: "synthesis",
+				community: "alpha",
+				source_path: "wiki/synthesis/sessions/session-a.md",
+			},
+		];
+		data.edges = [];
+		const learning = data.learning;
+		assert.ok(learning);
+		data.learning = {
+			...learning,
+			views: {
+				...learning.views,
+				global: { enabled: true, node_ids: ["session-a"], degraded: false },
+			},
+			communities: [{ id: "alpha", label: "Alpha Group", node_count: 1, color_index: 0 }],
+		};
+
+		const selection = resolveSelection(data, { kind: "node", id: "session-a" });
+		const payload = buildSelectionPromptPayload(data, selection, null);
+
+		assert.match(payload.expandedText, /1\. \[\[wiki\/synthesis\/sessions\/session-a\.md\]\] - Session A - 社区 alpha/);
+		assert.doesNotMatch(payload.expandedText, /wiki\/synthesis\/session-a\.md/);
+	});
+});
+
+function fixtureGraph(): GraphData {
+	return {
+		meta: {
+			build_date: "2026-06-12T00:00:00.000Z",
+			wiki_title: "Selection Fixture",
+			total_nodes: 2,
+			total_edges: 1
+		},
+		nodes: [
+			{ id: "a", label: "Alpha A", type: "topic", community: "alpha", source_path: "wiki/topics/alpha-a.md" },
+			{ id: "b", label: "Alpha B", type: "entity", community: "alpha", source_path: "wiki/entities/alpha-b.md" }
+		],
+		edges: [
+			{ id: "a-b", from: "a", to: "b", type: "EXTRACTED", weight: 1 }
+		],
+		learning: {
+			version: 1,
+			entry: { recommended_start_node_id: "a", recommended_start_reason: "community_hub", default_mode: "global" },
+			views: {
+				path: { enabled: false, start_node_id: null, node_ids: [], degraded: true },
+				community: { enabled: false, community_id: null, label: null, node_ids: [], is_weak: false, degraded: true },
+				global: { enabled: true, node_ids: ["a", "b"], degraded: false }
+			},
+			communities: [
+				{ id: "alpha", label: "Alpha Group", node_count: 2, color_index: 0 }
+			]
+		}
+	};
+}