108 lines
5.1 KiB
Bash
Executable File
108 lines
5.1 KiB
Bash
Executable File
#!/bin/bash
|
|
# Regression: focused community view renders as a lightweight node relationship map.
|
|
|
|
set -euo pipefail
|
|
|
|
REPO_ROOT="$(cd "$(dirname "$0")/.." && pwd)"
|
|
# shellcheck disable=SC1091
|
|
source "$REPO_ROOT/tests/lib/graph-html-engine-helpers.sh"
|
|
|
|
tmp_dir="$(mktemp -d)"
|
|
cleanup() {
|
|
rm -rf "$tmp_dir"
|
|
}
|
|
trap cleanup EXIT
|
|
|
|
npm run build -w @llm-wiki/graph-engine > /dev/null 2>&1 \
|
|
|| fail "graph-engine build should succeed before community node map regression"
|
|
|
|
build_graph_html_fixture "$tmp_dir"
|
|
|
|
node - "$tmp_dir/wiki/graph-data.json" <<'NODE'
|
|
const fs = require("fs");
|
|
const file = process.argv[2];
|
|
const graph = JSON.parse(fs.readFileSync(file, "utf8"));
|
|
const coreNodes = [
|
|
{ id: "A", label: "节点A", type: "topic", community: "t1", content: "# 节点A\n\n这是节点A。", source_path: "/fake/wiki/entities/A.md", x: 24, y: 42, weight: 90 },
|
|
{ id: "B", label: "节点B", type: "entity", community: "t1", content: "# 节点B\n\n这是节点B。", source_path: "/fake/wiki/entities/B.md", x: 36, y: 32, weight: 70 },
|
|
{ id: "C", label: "节点C", type: "source", community: "t1", content: "# 节点C\n\n这是节点C。", source_path: "/fake/wiki/entities/C.md", x: 38, y: 56, weight: 60 },
|
|
{ id: "D", label: "节点D", type: "entity", community: "t1", content: "# 节点D\n\n这是节点D。", source_path: "/fake/wiki/entities/D.md", x: 55, y: 34, weight: 50 },
|
|
{ id: "E", label: "节点E", type: "entity", community: "t1", content: "# 节点E\n\n这是节点E。", source_path: "/fake/wiki/entities/E.md", x: 68, y: 38, weight: 40 },
|
|
{ id: "F", label: "节点F", type: "comparison", community: "t1", content: "# 节点F\n\n这是节点F。", source_path: "/fake/wiki/entities/F.md", x: 80, y: 46, weight: 30 }
|
|
];
|
|
const fillerNodes = Array.from({ length: 18 }, (_, index) => {
|
|
const id = `N${index + 1}`;
|
|
const column = index % 6;
|
|
const row = Math.floor(index / 6);
|
|
return {
|
|
id,
|
|
label: `普通节点${index + 1}`,
|
|
type: index % 7 === 0 ? "topic" : index % 5 === 0 ? "source" : "entity",
|
|
community: "t1",
|
|
content: `# 普通节点${index + 1}\n\n这是用于验证默认标签稀疏度的普通节点。`,
|
|
source_path: `/fake/wiki/entities/${id}.md`,
|
|
x: 18 + column * 12,
|
|
y: 18 + row * 21,
|
|
weight: Math.max(5, 28 - index)
|
|
};
|
|
});
|
|
graph.nodes = [...coreNodes, ...fillerNodes];
|
|
const aNeighborEdges = Array.from({ length: 10 }, (_, index) => ({
|
|
id: `eAN${index + 1}`,
|
|
from: "A",
|
|
to: `N${index + 1}`,
|
|
type: "EXTRACTED",
|
|
confidence: "EXTRACTED",
|
|
relation_type: index % 2 === 0 ? "实现" : "依赖",
|
|
weight: 0.6
|
|
}));
|
|
graph.edges = [
|
|
{ id: "eAB", from: "A", to: "B", type: "EXTRACTED", confidence: "EXTRACTED", relation_type: "实现", weight: 1 },
|
|
{ id: "eAC", from: "A", to: "C", type: "INFERRED", confidence: "INFERRED", relation_type: "对比", weight: 0.8 },
|
|
{ id: "eBD", from: "B", to: "D", type: "EXTRACTED", confidence: "EXTRACTED", relation_type: "依赖", weight: 0.7 },
|
|
{ id: "eDE", from: "D", to: "E", type: "EXTRACTED", confidence: "EXTRACTED", relation_type: "衍生", weight: 0.5 },
|
|
{ id: "eEF", from: "E", to: "F", type: "AMBIGUOUS", confidence: "AMBIGUOUS", relation_type: "矛盾", weight: 0.4 },
|
|
...aNeighborEdges
|
|
];
|
|
graph.meta.total_nodes = graph.nodes.length;
|
|
graph.meta.total_edges = graph.edges.length;
|
|
graph.learning = {
|
|
version: 1,
|
|
entry: { recommended_start_node_id: "A", recommended_start_reason: "fixture", default_mode: "global" },
|
|
views: {
|
|
path: { enabled: false, start_node_id: null, node_ids: [], degraded: true },
|
|
community: { enabled: true, community_id: "t1", label: "测试社区", node_ids: graph.nodes.map((node) => node.id), is_weak: false, degraded: false },
|
|
global: { enabled: true, node_ids: graph.nodes.map((node) => node.id), degraded: false }
|
|
},
|
|
communities: [
|
|
{ id: "t1", label: "测试社区", node_count: graph.nodes.length, color_index: 0, recommended_start_node_id: "A" }
|
|
]
|
|
};
|
|
fs.writeFileSync(file, `${JSON.stringify(graph, null, 2)}\n`);
|
|
NODE
|
|
|
|
bash "$REPO_ROOT/scripts/build-graph-html.sh" "$tmp_dir" > /dev/null 2>&1 \
|
|
|| fail "build-graph-html.sh should succeed on community node map fixture"
|
|
|
|
playwright_node_path="$(
|
|
npx --yes -p playwright -c 'node -e "const path=require(\"path\"); console.log(path.dirname(process.env.PATH.split(\":\")[0]))"'
|
|
)"
|
|
|
|
chrome_executable="${GRAPH_COMMUNITY_NODE_MAP_CHROME_EXECUTABLE:-}"
|
|
if [ -z "$chrome_executable" ] && [ -x "/Applications/Google Chrome.app/Contents/MacOS/Google Chrome" ]; then
|
|
chrome_executable="/Applications/Google Chrome.app/Contents/MacOS/Google Chrome"
|
|
fi
|
|
|
|
artifact_dir="${GRAPH_COMMUNITY_NODE_MAP_ARTIFACT_DIR:-$(mktemp -d "${TMPDIR:-/tmp}/llm-wiki-community-node-map.XXXXXX")}"
|
|
screenshot_path="${GRAPH_COMMUNITY_NODE_MAP_SCREENSHOT:-$artifact_dir/community-node-map.png}"
|
|
|
|
GRAPH_COMMUNITY_NODE_MAP_HTML="$tmp_dir/wiki/knowledge-graph.html" \
|
|
GRAPH_COMMUNITY_NODE_MAP_CHROME_EXECUTABLE="$chrome_executable" \
|
|
GRAPH_COMMUNITY_NODE_MAP_SCREENSHOT="$screenshot_path" \
|
|
NODE_PATH="$playwright_node_path" \
|
|
node "$REPO_ROOT/tests/browser/graph-community-node-map.mjs" \
|
|
|| fail "community node map browser regression should pass"
|
|
|
|
echo "PASS: graph community node map regression"
|
|
echo "SCREENSHOT: $screenshot_path"
|