173 lines
6.8 KiB
TypeScript
173 lines
6.8 KiB
TypeScript
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
|
|
};
|
|
}
|