@dewtech/dare-cli 3.4.0 → 3.5.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/__tests__/dag-runner/buildLocateContext.test.d.ts +2 -0
- package/dist/__tests__/dag-runner/buildLocateContext.test.d.ts.map +1 -0
- package/dist/__tests__/dag-runner/buildLocateContext.test.js +53 -0
- package/dist/__tests__/dag-runner/buildLocateContext.test.js.map +1 -0
- package/dist/__tests__/dual-graph.test.d.ts +2 -0
- package/dist/__tests__/dual-graph.test.d.ts.map +1 -0
- package/dist/__tests__/dual-graph.test.js +141 -0
- package/dist/__tests__/dual-graph.test.js.map +1 -0
- package/dist/__tests__/graphrag/contract/traverse.contract.d.ts +9 -0
- package/dist/__tests__/graphrag/contract/traverse.contract.d.ts.map +1 -0
- package/dist/__tests__/graphrag/contract/traverse.contract.js +58 -0
- package/dist/__tests__/graphrag/contract/traverse.contract.js.map +1 -0
- package/dist/__tests__/graphrag/contract/traverse.contract.test.d.ts +2 -0
- package/dist/__tests__/graphrag/contract/traverse.contract.test.d.ts.map +1 -0
- package/dist/__tests__/graphrag/contract/traverse.contract.test.js +100 -0
- package/dist/__tests__/graphrag/contract/traverse.contract.test.js.map +1 -0
- package/dist/__tests__/graphrag/factory.test.js +10 -1
- package/dist/__tests__/graphrag/factory.test.js.map +1 -1
- package/dist/__tests__/graphrag/fixtures/dual-graph/build-fixture-graph.d.ts +11 -0
- package/dist/__tests__/graphrag/fixtures/dual-graph/build-fixture-graph.d.ts.map +1 -0
- package/dist/__tests__/graphrag/fixtures/dual-graph/build-fixture-graph.js +34 -0
- package/dist/__tests__/graphrag/fixtures/dual-graph/build-fixture-graph.js.map +1 -0
- package/dist/__tests__/graphrag/fixtures/dual-graph/fixtures.test.d.ts +2 -0
- package/dist/__tests__/graphrag/fixtures/dual-graph/fixtures.test.d.ts.map +1 -0
- package/dist/__tests__/graphrag/fixtures/dual-graph/fixtures.test.js +46 -0
- package/dist/__tests__/graphrag/fixtures/dual-graph/fixtures.test.js.map +1 -0
- package/dist/__tests__/graphrag/graph-rag.test.js +31 -0
- package/dist/__tests__/graphrag/graph-rag.test.js.map +1 -1
- package/dist/__tests__/graphrag/json-graph.test.js +57 -0
- package/dist/__tests__/graphrag/json-graph.test.js.map +1 -1
- package/dist/__tests__/graphrag/neo4j-graph.test.d.ts +2 -0
- package/dist/__tests__/graphrag/neo4j-graph.test.d.ts.map +1 -0
- package/dist/__tests__/graphrag/neo4j-graph.test.js +104 -0
- package/dist/__tests__/graphrag/neo4j-graph.test.js.map +1 -0
- package/dist/__tests__/graphrag/neo4j-persistence.test.d.ts +2 -0
- package/dist/__tests__/graphrag/neo4j-persistence.test.d.ts.map +1 -0
- package/dist/__tests__/graphrag/neo4j-persistence.test.js +110 -0
- package/dist/__tests__/graphrag/neo4j-persistence.test.js.map +1 -0
- package/dist/commands/__tests__/execute.telemetry.spec.js +9 -0
- package/dist/commands/__tests__/execute.telemetry.spec.js.map +1 -1
- package/dist/commands/__tests__/graph-viz-layers.test.d.ts +2 -0
- package/dist/commands/__tests__/graph-viz-layers.test.d.ts.map +1 -0
- package/dist/commands/__tests__/graph-viz-layers.test.js +36 -0
- package/dist/commands/__tests__/graph-viz-layers.test.js.map +1 -0
- package/dist/commands/__tests__/impact.test.d.ts +2 -0
- package/dist/commands/__tests__/impact.test.d.ts.map +1 -0
- package/dist/commands/__tests__/impact.test.js +26 -0
- package/dist/commands/__tests__/impact.test.js.map +1 -0
- package/dist/commands/__tests__/locate.test.d.ts +2 -0
- package/dist/commands/__tests__/locate.test.d.ts.map +1 -0
- package/dist/commands/__tests__/locate.test.js +35 -0
- package/dist/commands/__tests__/locate.test.js.map +1 -0
- package/dist/commands/__tests__/owners.test.d.ts +2 -0
- package/dist/commands/__tests__/owners.test.d.ts.map +1 -0
- package/dist/commands/__tests__/owners.test.js +34 -0
- package/dist/commands/__tests__/owners.test.js.map +1 -0
- package/dist/commands/__tests__/trace.test.d.ts +2 -0
- package/dist/commands/__tests__/trace.test.d.ts.map +1 -0
- package/dist/commands/__tests__/trace.test.js +29 -0
- package/dist/commands/__tests__/trace.test.js.map +1 -0
- package/dist/commands/execute.d.ts.map +1 -1
- package/dist/commands/execute.js +11 -6
- package/dist/commands/execute.js.map +1 -1
- package/dist/commands/graph-queries.d.ts +48 -0
- package/dist/commands/graph-queries.d.ts.map +1 -0
- package/dist/commands/graph-queries.js +204 -0
- package/dist/commands/graph-queries.js.map +1 -0
- package/dist/commands/graph.d.ts +13 -0
- package/dist/commands/graph.d.ts.map +1 -1
- package/dist/commands/graph.js +212 -8
- package/dist/commands/graph.js.map +1 -1
- package/dist/dag-runner/__tests__/fixtures/src/math.d.ts +3 -0
- package/dist/dag-runner/__tests__/fixtures/src/math.d.ts.map +1 -0
- package/dist/dag-runner/__tests__/fixtures/src/math.js +7 -0
- package/dist/dag-runner/__tests__/fixtures/src/math.js.map +1 -0
- package/dist/dag-runner/__tests__/graph-ingest-symbols.test.d.ts +2 -0
- package/dist/dag-runner/__tests__/graph-ingest-symbols.test.d.ts.map +1 -0
- package/dist/dag-runner/__tests__/graph-ingest-symbols.test.js +81 -0
- package/dist/dag-runner/__tests__/graph-ingest-symbols.test.js.map +1 -0
- package/dist/dag-runner/graph-ingest.d.ts.map +1 -1
- package/dist/dag-runner/graph-ingest.js +46 -0
- package/dist/dag-runner/graph-ingest.js.map +1 -1
- package/dist/dag-runner/graph-locate.d.ts +11 -0
- package/dist/dag-runner/graph-locate.d.ts.map +1 -0
- package/dist/dag-runner/graph-locate.js +74 -0
- package/dist/dag-runner/graph-locate.js.map +1 -0
- package/dist/dag-runner/run_dag.d.ts +3 -1
- package/dist/dag-runner/run_dag.d.ts.map +1 -1
- package/dist/dag-runner/run_dag.js +2 -1
- package/dist/dag-runner/run_dag.js.map +1 -1
- package/dist/dag-runner/utils/stitch-context.d.ts +3 -1
- package/dist/dag-runner/utils/stitch-context.d.ts.map +1 -1
- package/dist/dag-runner/utils/stitch-context.js +8 -2
- package/dist/dag-runner/utils/stitch-context.js.map +1 -1
- package/dist/graphrag/__tests__/code-index.test.d.ts +2 -0
- package/dist/graphrag/__tests__/code-index.test.d.ts.map +1 -0
- package/dist/graphrag/__tests__/code-index.test.js +75 -0
- package/dist/graphrag/__tests__/code-index.test.js.map +1 -0
- package/dist/graphrag/__tests__/fixtures/code-index/sample.d.ts +5 -0
- package/dist/graphrag/__tests__/fixtures/code-index/sample.d.ts.map +1 -0
- package/dist/graphrag/__tests__/fixtures/code-index/sample.js +10 -0
- package/dist/graphrag/__tests__/fixtures/code-index/sample.js.map +1 -0
- package/dist/graphrag/__tests__/locate.test.d.ts +2 -0
- package/dist/graphrag/__tests__/locate.test.d.ts.map +1 -0
- package/dist/graphrag/__tests__/locate.test.js +43 -0
- package/dist/graphrag/__tests__/locate.test.js.map +1 -0
- package/dist/graphrag/__tests__/requirement-ingest.test.d.ts +2 -0
- package/dist/graphrag/__tests__/requirement-ingest.test.d.ts.map +1 -0
- package/dist/graphrag/__tests__/requirement-ingest.test.js +66 -0
- package/dist/graphrag/__tests__/requirement-ingest.test.js.map +1 -0
- package/dist/graphrag/__tests__/traverse.test.d.ts +2 -0
- package/dist/graphrag/__tests__/traverse.test.d.ts.map +1 -0
- package/dist/graphrag/__tests__/traverse.test.js +71 -0
- package/dist/graphrag/__tests__/traverse.test.js.map +1 -0
- package/dist/graphrag/__tests__/types.test.d.ts +2 -0
- package/dist/graphrag/__tests__/types.test.d.ts.map +1 -0
- package/dist/graphrag/__tests__/types.test.js +52 -0
- package/dist/graphrag/__tests__/types.test.js.map +1 -0
- package/dist/graphrag/code-index.d.ts +15 -0
- package/dist/graphrag/code-index.d.ts.map +1 -0
- package/dist/graphrag/code-index.js +205 -0
- package/dist/graphrag/code-index.js.map +1 -0
- package/dist/graphrag/factory.d.ts.map +1 -1
- package/dist/graphrag/factory.js +5 -0
- package/dist/graphrag/factory.js.map +1 -1
- package/dist/graphrag/graph-rag.d.ts +6 -2
- package/dist/graphrag/graph-rag.d.ts.map +1 -1
- package/dist/graphrag/graph-rag.js +40 -2
- package/dist/graphrag/graph-rag.js.map +1 -1
- package/dist/graphrag/index.d.ts +8 -2
- package/dist/graphrag/index.d.ts.map +1 -1
- package/dist/graphrag/index.js +5 -1
- package/dist/graphrag/index.js.map +1 -1
- package/dist/graphrag/json-graph.d.ts +8 -0
- package/dist/graphrag/json-graph.d.ts.map +1 -1
- package/dist/graphrag/json-graph.js +50 -4
- package/dist/graphrag/json-graph.js.map +1 -1
- package/dist/graphrag/knowledge-graph.d.ts +8 -3
- package/dist/graphrag/knowledge-graph.d.ts.map +1 -1
- package/dist/graphrag/neo4j-graph.d.ts +19 -18
- package/dist/graphrag/neo4j-graph.d.ts.map +1 -1
- package/dist/graphrag/neo4j-graph.js +164 -38
- package/dist/graphrag/neo4j-graph.js.map +1 -1
- package/dist/graphrag/requirement-ingest.d.ts +15 -0
- package/dist/graphrag/requirement-ingest.d.ts.map +1 -0
- package/dist/graphrag/requirement-ingest.js +158 -0
- package/dist/graphrag/requirement-ingest.js.map +1 -0
- package/dist/graphrag/traverse.d.ts +5 -0
- package/dist/graphrag/traverse.d.ts.map +1 -0
- package/dist/graphrag/traverse.js +139 -0
- package/dist/graphrag/traverse.js.map +1 -0
- package/dist/graphrag/types.d.ts +60 -2
- package/dist/graphrag/types.d.ts.map +1 -1
- package/dist/graphrag/types.js +34 -1
- package/dist/graphrag/types.js.map +1 -1
- package/dist/mcp-server/__tests__/mcp-graph.test.d.ts +2 -0
- package/dist/mcp-server/__tests__/mcp-graph.test.d.ts.map +1 -0
- package/dist/mcp-server/__tests__/mcp-graph.test.js +85 -0
- package/dist/mcp-server/__tests__/mcp-graph.test.js.map +1 -0
- package/dist/mcp-server/server.d.ts.map +1 -1
- package/dist/mcp-server/server.js +106 -1
- package/dist/mcp-server/server.js.map +1 -1
- package/dist/utils/__tests__/static-analyzer-exports.test.d.ts +2 -0
- package/dist/utils/__tests__/static-analyzer-exports.test.d.ts.map +1 -0
- package/dist/utils/__tests__/static-analyzer-exports.test.js +21 -0
- package/dist/utils/__tests__/static-analyzer-exports.test.js.map +1 -0
- package/dist/utils/project-generator.js +12 -0
- package/dist/utils/project-generator.js.map +1 -1
- package/dist/utils/static-analyzer.d.ts +9 -0
- package/dist/utils/static-analyzer.d.ts.map +1 -1
- package/dist/utils/static-analyzer.js +2 -2
- package/dist/utils/static-analyzer.js.map +1 -1
- package/dist/verification/__tests__/telemetry.test.js +9 -0
- package/dist/verification/__tests__/telemetry.test.js.map +1 -1
- package/package.json +1 -1
|
@@ -0,0 +1,158 @@
|
|
|
1
|
+
import fs from 'node:fs';
|
|
2
|
+
import path from 'node:path';
|
|
3
|
+
const RF_RE = /^\|\s*(RF-\d+)\s*\|([^|]*)/;
|
|
4
|
+
const O_RE = /^\|\s*(O-\d+)\s*\|([^|]*)/;
|
|
5
|
+
const TASK_TABLE_RE = /^\|\s*(task-\d+)\s*\|([^|]*)/i;
|
|
6
|
+
const TASK_HEADING_RE = /\btask-(\d+)\s*:/i;
|
|
7
|
+
const PHASE_RE = /^###\s+(?:Phase|Fase)\s+(\d+)/i;
|
|
8
|
+
const PRIORITY_RE = /\b(MUST|SHOULD|COULD)\b/i;
|
|
9
|
+
function parsePriority(text) {
|
|
10
|
+
const m = PRIORITY_RE.exec(text);
|
|
11
|
+
if (!m)
|
|
12
|
+
return undefined;
|
|
13
|
+
return m[1].toUpperCase();
|
|
14
|
+
}
|
|
15
|
+
function cleanTitle(raw) {
|
|
16
|
+
return raw.replace(/\|/g, '').trim();
|
|
17
|
+
}
|
|
18
|
+
export function parseRequirementsFromMarkdown(content, source) {
|
|
19
|
+
const lines = content.split(/\r?\n/);
|
|
20
|
+
const out = [];
|
|
21
|
+
let currentParent;
|
|
22
|
+
for (const line of lines) {
|
|
23
|
+
const phase = PHASE_RE.exec(line);
|
|
24
|
+
if (phase) {
|
|
25
|
+
const phaseId = `phase-${phase[1]}`;
|
|
26
|
+
currentParent = phaseId;
|
|
27
|
+
out.push({
|
|
28
|
+
reqId: phaseId,
|
|
29
|
+
title: `Phase ${phase[1]}`,
|
|
30
|
+
source,
|
|
31
|
+
parentId: undefined,
|
|
32
|
+
});
|
|
33
|
+
continue;
|
|
34
|
+
}
|
|
35
|
+
let match = RF_RE.exec(line);
|
|
36
|
+
if (match) {
|
|
37
|
+
const title = cleanTitle(match[2] ?? '');
|
|
38
|
+
out.push({
|
|
39
|
+
reqId: match[1],
|
|
40
|
+
title,
|
|
41
|
+
priority: parsePriority(title),
|
|
42
|
+
parentId: currentParent,
|
|
43
|
+
source,
|
|
44
|
+
});
|
|
45
|
+
continue;
|
|
46
|
+
}
|
|
47
|
+
match = O_RE.exec(line);
|
|
48
|
+
if (match) {
|
|
49
|
+
const title = cleanTitle(match[2] ?? '');
|
|
50
|
+
out.push({
|
|
51
|
+
reqId: match[1],
|
|
52
|
+
title,
|
|
53
|
+
priority: parsePriority(title),
|
|
54
|
+
parentId: currentParent,
|
|
55
|
+
source,
|
|
56
|
+
});
|
|
57
|
+
continue;
|
|
58
|
+
}
|
|
59
|
+
match = TASK_TABLE_RE.exec(line);
|
|
60
|
+
if (match) {
|
|
61
|
+
const reqId = match[1].toLowerCase();
|
|
62
|
+
const title = cleanTitle(match[2] ?? reqId);
|
|
63
|
+
out.push({
|
|
64
|
+
reqId,
|
|
65
|
+
title,
|
|
66
|
+
priority: parsePriority(title),
|
|
67
|
+
parentId: currentParent,
|
|
68
|
+
source,
|
|
69
|
+
});
|
|
70
|
+
continue;
|
|
71
|
+
}
|
|
72
|
+
const taskHeading = TASK_HEADING_RE.exec(line);
|
|
73
|
+
if (taskHeading) {
|
|
74
|
+
const reqId = `task-${taskHeading[1]}`;
|
|
75
|
+
const title = line.replace(/^#+\s*/, '').trim();
|
|
76
|
+
out.push({
|
|
77
|
+
reqId,
|
|
78
|
+
title,
|
|
79
|
+
priority: parsePriority(title),
|
|
80
|
+
parentId: currentParent,
|
|
81
|
+
source,
|
|
82
|
+
});
|
|
83
|
+
}
|
|
84
|
+
}
|
|
85
|
+
return out;
|
|
86
|
+
}
|
|
87
|
+
function listMarkdownFiles(dir, prefix) {
|
|
88
|
+
if (!fs.existsSync(dir))
|
|
89
|
+
return [];
|
|
90
|
+
return fs
|
|
91
|
+
.readdirSync(dir)
|
|
92
|
+
.filter((f) => f.startsWith(prefix) && f.endsWith('.md'))
|
|
93
|
+
.map((f) => path.join(dir, f));
|
|
94
|
+
}
|
|
95
|
+
function sourceForFile(name) {
|
|
96
|
+
if (name.startsWith('DESIGN'))
|
|
97
|
+
return 'design';
|
|
98
|
+
if (name.startsWith('BLUEPRINT'))
|
|
99
|
+
return 'blueprint';
|
|
100
|
+
if (name.startsWith('TASKS'))
|
|
101
|
+
return 'tasks';
|
|
102
|
+
return null;
|
|
103
|
+
}
|
|
104
|
+
export function ingestRequirements(graph, projectRoot) {
|
|
105
|
+
const dareDir = path.join(projectRoot, 'DARE');
|
|
106
|
+
let nodes = 0;
|
|
107
|
+
let edges = 0;
|
|
108
|
+
const files = [];
|
|
109
|
+
for (const f of listMarkdownFiles(dareDir, 'DESIGN')) {
|
|
110
|
+
files.push({ file: f, source: 'design' });
|
|
111
|
+
}
|
|
112
|
+
for (const f of listMarkdownFiles(dareDir, 'BLUEPRINT')) {
|
|
113
|
+
files.push({ file: f, source: 'blueprint' });
|
|
114
|
+
}
|
|
115
|
+
for (const f of listMarkdownFiles(dareDir, 'TASKS')) {
|
|
116
|
+
files.push({ file: f, source: 'tasks' });
|
|
117
|
+
}
|
|
118
|
+
const parsed = [];
|
|
119
|
+
for (const { file, source } of files) {
|
|
120
|
+
const content = fs.readFileSync(file, 'utf-8');
|
|
121
|
+
parsed.push(...parseRequirementsFromMarkdown(content, source));
|
|
122
|
+
}
|
|
123
|
+
const seenNodes = new Set();
|
|
124
|
+
const seenEdges = new Set();
|
|
125
|
+
for (const req of parsed) {
|
|
126
|
+
const nodeId = `requirement:${req.reqId}`;
|
|
127
|
+
graph.addNode({
|
|
128
|
+
id: nodeId,
|
|
129
|
+
type: 'requirement',
|
|
130
|
+
label: req.title || req.reqId,
|
|
131
|
+
metadata: {
|
|
132
|
+
reqId: req.reqId,
|
|
133
|
+
source: req.source,
|
|
134
|
+
priority: req.priority,
|
|
135
|
+
title: req.title,
|
|
136
|
+
},
|
|
137
|
+
});
|
|
138
|
+
if (!seenNodes.has(nodeId)) {
|
|
139
|
+
seenNodes.add(nodeId);
|
|
140
|
+
nodes++;
|
|
141
|
+
}
|
|
142
|
+
if (req.parentId) {
|
|
143
|
+
const edgeId = `derives_from:${req.reqId}->${req.parentId}`;
|
|
144
|
+
graph.addEdge({
|
|
145
|
+
id: edgeId,
|
|
146
|
+
sourceId: nodeId,
|
|
147
|
+
targetId: `requirement:${req.parentId}`,
|
|
148
|
+
type: 'derives_from',
|
|
149
|
+
});
|
|
150
|
+
if (!seenEdges.has(edgeId)) {
|
|
151
|
+
seenEdges.add(edgeId);
|
|
152
|
+
edges++;
|
|
153
|
+
}
|
|
154
|
+
}
|
|
155
|
+
}
|
|
156
|
+
return { nodes, edges };
|
|
157
|
+
}
|
|
158
|
+
//# sourceMappingURL=requirement-ingest.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"requirement-ingest.js","sourceRoot":"","sources":["../../src/graphrag/requirement-ingest.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,IAAI,MAAM,WAAW,CAAC;AAY7B,MAAM,KAAK,GAAG,4BAA4B,CAAC;AAC3C,MAAM,IAAI,GAAG,2BAA2B,CAAC;AACzC,MAAM,aAAa,GAAG,+BAA+B,CAAC;AACtD,MAAM,eAAe,GAAG,mBAAmB,CAAC;AAC5C,MAAM,QAAQ,GAAG,gCAAgC,CAAC;AAClD,MAAM,WAAW,GAAG,0BAA0B,CAAC;AAE/C,SAAS,aAAa,CAAC,IAAY;IACjC,MAAM,CAAC,GAAG,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACjC,IAAI,CAAC,CAAC;QAAE,OAAO,SAAS,CAAC;IACzB,OAAO,CAAC,CAAC,CAAC,CAAE,CAAC,WAAW,EAAiC,CAAC;AAC5D,CAAC;AAED,SAAS,UAAU,CAAC,GAAW;IAC7B,OAAO,GAAG,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;AACvC,CAAC;AAED,MAAM,UAAU,6BAA6B,CAC3C,OAAe,EACf,MAAiC;IAEjC,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IACrC,MAAM,GAAG,GAAwB,EAAE,CAAC;IACpC,IAAI,aAAiC,CAAC;IAEtC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,MAAM,KAAK,GAAG,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClC,IAAI,KAAK,EAAE,CAAC;YACV,MAAM,OAAO,GAAG,SAAS,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;YACpC,aAAa,GAAG,OAAO,CAAC;YACxB,GAAG,CAAC,IAAI,CAAC;gBACP,KAAK,EAAE,OAAO;gBACd,KAAK,EAAE,SAAS,KAAK,CAAC,CAAC,CAAC,EAAE;gBAC1B,MAAM;gBACN,QAAQ,EAAE,SAAS;aACpB,CAAC,CAAC;YACH,SAAS;QACX,CAAC;QAED,IAAI,KAAK,GAA2B,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACrD,IAAI,KAAK,EAAE,CAAC;YACV,MAAM,KAAK,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;YACzC,GAAG,CAAC,IAAI,CAAC;gBACP,KAAK,EAAE,KAAK,CAAC,CAAC,CAAE;gBAChB,KAAK;gBACL,QAAQ,EAAE,aAAa,CAAC,KAAK,CAAC;gBAC9B,QAAQ,EAAE,aAAa;gBACvB,MAAM;aACP,CAAC,CAAC;YACH,SAAS;QACX,CAAC;QAED,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACxB,IAAI,KAAK,EAAE,CAAC;YACV,MAAM,KAAK,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;YACzC,GAAG,CAAC,IAAI,CAAC;gBACP,KAAK,EAAE,KAAK,CAAC,CAAC,CAAE;gBAChB,KAAK;gBACL,QAAQ,EAAE,aAAa,CAAC,KAAK,CAAC;gBAC9B,QAAQ,EAAE,aAAa;gBACvB,MAAM;aACP,CAAC,CAAC;YACH,SAAS;QACX,CAAC;QAED,KAAK,GAAG,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACjC,IAAI,KAAK,EAAE,CAAC;YACV,MAAM,KAAK,GAAG,KAAK,CAAC,CAAC,CAAE,CAAC,WAAW,EAAE,CAAC;YACtC,MAAM,KAAK,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC;YAC5C,GAAG,CAAC,IAAI,CAAC;gBACP,KAAK;gBACL,KAAK;gBACL,QAAQ,EAAE,aAAa,CAAC,KAAK,CAAC;gBAC9B,QAAQ,EAAE,aAAa;gBACvB,MAAM;aACP,CAAC,CAAC;YACH,SAAS;QACX,CAAC;QAED,MAAM,WAAW,GAAG,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC/C,IAAI,WAAW,EAAE,CAAC;YAChB,MAAM,KAAK,GAAG,QAAQ,WAAW,CAAC,CAAC,CAAC,EAAE,CAAC;YACvC,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;YAChD,GAAG,CAAC,IAAI,CAAC;gBACP,KAAK;gBACL,KAAK;gBACL,QAAQ,EAAE,aAAa,CAAC,KAAK,CAAC;gBAC9B,QAAQ,EAAE,aAAa;gBACvB,MAAM;aACP,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,OAAO,GAAG,CAAC;AACb,CAAC;AAED,SAAS,iBAAiB,CAAC,GAAW,EAAE,MAAc;IACpD,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,GAAG,CAAC;QAAE,OAAO,EAAE,CAAC;IACnC,OAAO,EAAE;SACN,WAAW,CAAC,GAAG,CAAC;SAChB,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;SACxD,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC,CAAC;AACnC,CAAC;AAED,SAAS,aAAa,CAAC,IAAY;IACjC,IAAI,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC;QAAE,OAAO,QAAQ,CAAC;IAC/C,IAAI,IAAI,CAAC,UAAU,CAAC,WAAW,CAAC;QAAE,OAAO,WAAW,CAAC;IACrD,IAAI,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC;QAAE,OAAO,OAAO,CAAC;IAC7C,OAAO,IAAI,CAAC;AACd,CAAC;AAED,MAAM,UAAU,kBAAkB,CAChC,KAAqB,EACrB,WAAmB;IAEnB,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,MAAM,CAAC,CAAC;IAC/C,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,IAAI,KAAK,GAAG,CAAC,CAAC;IAEd,MAAM,KAAK,GAA+D,EAAE,CAAC;IAC7E,KAAK,MAAM,CAAC,IAAI,iBAAiB,CAAC,OAAO,EAAE,QAAQ,CAAC,EAAE,CAAC;QACrD,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC,CAAC;IAC5C,CAAC;IACD,KAAK,MAAM,CAAC,IAAI,iBAAiB,CAAC,OAAO,EAAE,WAAW,CAAC,EAAE,CAAC;QACxD,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,MAAM,EAAE,WAAW,EAAE,CAAC,CAAC;IAC/C,CAAC;IACD,KAAK,MAAM,CAAC,IAAI,iBAAiB,CAAC,OAAO,EAAE,OAAO,CAAC,EAAE,CAAC;QACpD,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,CAAC;IAC3C,CAAC;IAED,MAAM,MAAM,GAAwB,EAAE,CAAC;IACvC,KAAK,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,KAAK,EAAE,CAAC;QACrC,MAAM,OAAO,GAAG,EAAE,CAAC,YAAY,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;QAC/C,MAAM,CAAC,IAAI,CAAC,GAAG,6BAA6B,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC,CAAC;IACjE,CAAC;IAED,MAAM,SAAS,GAAG,IAAI,GAAG,EAAU,CAAC;IACpC,MAAM,SAAS,GAAG,IAAI,GAAG,EAAU,CAAC;IAEpC,KAAK,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;QACzB,MAAM,MAAM,GAAG,eAAe,GAAG,CAAC,KAAK,EAAE,CAAC;QAC1C,KAAK,CAAC,OAAO,CAAC;YACZ,EAAE,EAAE,MAAM;YACV,IAAI,EAAE,aAAa;YACnB,KAAK,EAAE,GAAG,CAAC,KAAK,IAAI,GAAG,CAAC,KAAK;YAC7B,QAAQ,EAAE;gBACR,KAAK,EAAE,GAAG,CAAC,KAAK;gBAChB,MAAM,EAAE,GAAG,CAAC,MAAM;gBAClB,QAAQ,EAAE,GAAG,CAAC,QAAQ;gBACtB,KAAK,EAAE,GAAG,CAAC,KAAK;aACjB;SACF,CAAC,CAAC;QACH,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC;YAC3B,SAAS,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;YACtB,KAAK,EAAE,CAAC;QACV,CAAC;QAED,IAAI,GAAG,CAAC,QAAQ,EAAE,CAAC;YACjB,MAAM,MAAM,GAAG,gBAAgB,GAAG,CAAC,KAAK,KAAK,GAAG,CAAC,QAAQ,EAAE,CAAC;YAC5D,KAAK,CAAC,OAAO,CAAC;gBACZ,EAAE,EAAE,MAAM;gBACV,QAAQ,EAAE,MAAM;gBAChB,QAAQ,EAAE,eAAe,GAAG,CAAC,QAAQ,EAAE;gBACvC,IAAI,EAAE,cAAc;aACrB,CAAC,CAAC;YACH,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC;gBAC3B,SAAS,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;gBACtB,KAAK,EAAE,CAAC;YACV,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC;AAC1B,CAAC"}
|
|
@@ -0,0 +1,5 @@
|
|
|
1
|
+
import type { KnowledgeGraph } from './knowledge-graph.js';
|
|
2
|
+
import type { LocateOptions, LocateResult, TraverseOptions, TraverseResult } from './types.js';
|
|
3
|
+
export declare function traverse(graph: KnowledgeGraph, opts: TraverseOptions): TraverseResult;
|
|
4
|
+
export declare function locate(graph: KnowledgeGraph, seedQuery: string, opts?: LocateOptions): LocateResult;
|
|
5
|
+
//# sourceMappingURL=traverse.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"traverse.d.ts","sourceRoot":"","sources":["../../src/graphrag/traverse.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAC3D,OAAO,KAAK,EAGV,aAAa,EACb,YAAY,EACZ,eAAe,EACf,cAAc,EACf,MAAM,YAAY,CAAC;AA4BpB,wBAAgB,QAAQ,CAAC,KAAK,EAAE,cAAc,EAAE,IAAI,EAAE,eAAe,GAAG,cAAc,CA+CrF;AAQD,wBAAgB,MAAM,CACpB,KAAK,EAAE,cAAc,EACrB,SAAS,EAAE,MAAM,EACjB,IAAI,GAAE,aAAkB,GACvB,YAAY,CA2Ed"}
|
|
@@ -0,0 +1,139 @@
|
|
|
1
|
+
import { assertRelativeSafe } from '../utils/path-safety.js';
|
|
2
|
+
const QN_RE = /^[\w./-]+::\w+$/;
|
|
3
|
+
const SUPPORTED_EXT = /\.(ts|tsx|js|jsx|mjs|cjs|py|go|rs|php|rb|java|kt|cs)$/i;
|
|
4
|
+
function findByQualifiedName(graph, qn) {
|
|
5
|
+
const directId = qn.startsWith('code_symbol:') ? qn : `code_symbol:${qn}`;
|
|
6
|
+
const direct = graph.getNode(directId);
|
|
7
|
+
if (direct)
|
|
8
|
+
return direct;
|
|
9
|
+
const { nodes } = graph.exportToJson();
|
|
10
|
+
return (nodes.find((n) => n.type === 'code_symbol' &&
|
|
11
|
+
(n.metadata?.qualifiedName === qn || n.id === directId)) ?? null);
|
|
12
|
+
}
|
|
13
|
+
function isPathLike(seed) {
|
|
14
|
+
return seed.includes('/') || seed.includes('::') || SUPPORTED_EXT.test(seed);
|
|
15
|
+
}
|
|
16
|
+
function pathPartFromSeed(seed) {
|
|
17
|
+
if (seed.includes('::'))
|
|
18
|
+
return seed.split('::')[0];
|
|
19
|
+
return seed;
|
|
20
|
+
}
|
|
21
|
+
export function traverse(graph, opts) {
|
|
22
|
+
const maxHops = Math.min(opts.maxHops ?? 3, 5);
|
|
23
|
+
const maxFanout = Math.min(opts.maxFanout ?? 50, 200);
|
|
24
|
+
const direction = opts.direction ?? 'both';
|
|
25
|
+
const edgeFilter = opts.edgeTypes ? new Set(opts.edgeTypes) : null;
|
|
26
|
+
const nodeFilter = opts.nodeTypes ? new Set(opts.nodeTypes) : null;
|
|
27
|
+
const visitedNodes = new Map();
|
|
28
|
+
const visitedEdges = new Map();
|
|
29
|
+
for (const seedId of opts.seedNodeIds) {
|
|
30
|
+
const seed = graph.getNode(seedId);
|
|
31
|
+
if (seed)
|
|
32
|
+
visitedNodes.set(seedId, seed);
|
|
33
|
+
}
|
|
34
|
+
let frontier = opts.seedNodeIds.filter((id) => graph.getNode(id));
|
|
35
|
+
let hopsReached = 0;
|
|
36
|
+
for (let depth = 0; depth < maxHops && frontier.length > 0; depth++) {
|
|
37
|
+
const nextFrontier = [];
|
|
38
|
+
hopsReached = depth + 1;
|
|
39
|
+
for (const nodeId of frontier) {
|
|
40
|
+
let edges = graph.getEdges(nodeId, direction);
|
|
41
|
+
edges = edges.sort((a, b) => a.id.localeCompare(b.id)).slice(0, maxFanout);
|
|
42
|
+
if (edgeFilter)
|
|
43
|
+
edges = edges.filter((e) => edgeFilter.has(e.type));
|
|
44
|
+
for (const edge of edges) {
|
|
45
|
+
visitedEdges.set(edge.id, edge);
|
|
46
|
+
const neighborId = edge.sourceId === nodeId ? edge.targetId : edge.sourceId;
|
|
47
|
+
if (visitedNodes.has(neighborId))
|
|
48
|
+
continue;
|
|
49
|
+
const node = graph.getNode(neighborId);
|
|
50
|
+
if (!node)
|
|
51
|
+
continue;
|
|
52
|
+
if (nodeFilter && !nodeFilter.has(node.type))
|
|
53
|
+
continue;
|
|
54
|
+
visitedNodes.set(neighborId, node);
|
|
55
|
+
nextFrontier.push(neighborId);
|
|
56
|
+
}
|
|
57
|
+
}
|
|
58
|
+
frontier = nextFrontier;
|
|
59
|
+
}
|
|
60
|
+
return {
|
|
61
|
+
nodes: [...visitedNodes.values()].sort((a, b) => a.id.localeCompare(b.id)),
|
|
62
|
+
edges: [...visitedEdges.values()].sort((a, b) => a.id.localeCompare(b.id)),
|
|
63
|
+
hops: hopsReached,
|
|
64
|
+
};
|
|
65
|
+
}
|
|
66
|
+
export function locate(graph, seedQuery, opts = {}) {
|
|
67
|
+
const hops = Math.min(opts.hops ?? 3, 5);
|
|
68
|
+
const limit = Math.min(opts.limit ?? 10, 50);
|
|
69
|
+
const edgeTypes = opts.edgeTypes ?? ['implements', 'contains', 'depends_on'];
|
|
70
|
+
const nodeTypes = opts.nodeTypes ?? ['code_symbol', 'file', 'task'];
|
|
71
|
+
if (isPathLike(seedQuery)) {
|
|
72
|
+
assertRelativeSafe(pathPartFromSeed(seedQuery));
|
|
73
|
+
}
|
|
74
|
+
const seeds = [];
|
|
75
|
+
if (QN_RE.test(seedQuery)) {
|
|
76
|
+
const node = findByQualifiedName(graph, seedQuery);
|
|
77
|
+
if (node)
|
|
78
|
+
seeds.push({ node, score: 1.0, path: [node.id] });
|
|
79
|
+
}
|
|
80
|
+
else if (isPathLike(seedQuery) && !seedQuery.includes('::')) {
|
|
81
|
+
const prefix = seedQuery.replace(/\\/g, '/');
|
|
82
|
+
const { nodes } = graph.exportToJson();
|
|
83
|
+
for (const n of nodes) {
|
|
84
|
+
if (n.type !== 'file' && n.type !== 'code_symbol')
|
|
85
|
+
continue;
|
|
86
|
+
const p = String(n.metadata?.path ?? n.metadata?.qualifiedName ?? '');
|
|
87
|
+
if (p.startsWith(prefix) || n.id.includes(prefix)) {
|
|
88
|
+
seeds.push({ node: n, score: 1.0, path: [n.id] });
|
|
89
|
+
}
|
|
90
|
+
}
|
|
91
|
+
}
|
|
92
|
+
else {
|
|
93
|
+
for (const hit of graph.searchNodes(seedQuery, 5)) {
|
|
94
|
+
seeds.push({ node: hit.node, score: 0.7, path: [hit.node.id] });
|
|
95
|
+
}
|
|
96
|
+
}
|
|
97
|
+
const candidates = new Map();
|
|
98
|
+
for (const seed of seeds) {
|
|
99
|
+
const key = seed.node.id;
|
|
100
|
+
const existing = candidates.get(key);
|
|
101
|
+
if (!existing || seed.score > existing.score) {
|
|
102
|
+
candidates.set(key, seed);
|
|
103
|
+
}
|
|
104
|
+
const walked = traverse(graph, {
|
|
105
|
+
seedNodeIds: [seed.node.id],
|
|
106
|
+
maxHops: hops,
|
|
107
|
+
edgeTypes,
|
|
108
|
+
nodeTypes,
|
|
109
|
+
direction: 'both',
|
|
110
|
+
});
|
|
111
|
+
for (const node of walked.nodes) {
|
|
112
|
+
const hopDist = walked.edges.length > 0 ? 1 : 0;
|
|
113
|
+
let score = seed.score - hopDist * 0.15;
|
|
114
|
+
if (node.type === 'code_symbol')
|
|
115
|
+
score += 0.1;
|
|
116
|
+
const prev = candidates.get(node.id);
|
|
117
|
+
if (!prev || score > prev.score) {
|
|
118
|
+
candidates.set(node.id, {
|
|
119
|
+
node,
|
|
120
|
+
score,
|
|
121
|
+
path: [seed.node.id, node.id],
|
|
122
|
+
});
|
|
123
|
+
}
|
|
124
|
+
}
|
|
125
|
+
}
|
|
126
|
+
const sorted = [...candidates.values()].sort((a, b) => {
|
|
127
|
+
if (b.score !== a.score)
|
|
128
|
+
return b.score - a.score;
|
|
129
|
+
return a.node.id.localeCompare(b.node.id);
|
|
130
|
+
});
|
|
131
|
+
return {
|
|
132
|
+
candidates: sorted.slice(0, limit).map((c) => ({
|
|
133
|
+
node: c.node,
|
|
134
|
+
score: c.score,
|
|
135
|
+
path: c.path,
|
|
136
|
+
})),
|
|
137
|
+
};
|
|
138
|
+
}
|
|
139
|
+
//# sourceMappingURL=traverse.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"traverse.js","sourceRoot":"","sources":["../../src/graphrag/traverse.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAW7D,MAAM,KAAK,GAAG,iBAAiB,CAAC;AAChC,MAAM,aAAa,GAAG,wDAAwD,CAAC;AAE/E,SAAS,mBAAmB,CAAC,KAAqB,EAAE,EAAU;IAC5D,MAAM,QAAQ,GAAG,EAAE,CAAC,UAAU,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,eAAe,EAAE,EAAE,CAAC;IAC1E,MAAM,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IACvC,IAAI,MAAM;QAAE,OAAO,MAAM,CAAC;IAC1B,MAAM,EAAE,KAAK,EAAE,GAAG,KAAK,CAAC,YAAY,EAAE,CAAC;IACvC,OAAO,CACL,KAAK,CAAC,IAAI,CACR,CAAC,CAAC,EAAE,EAAE,CACJ,CAAC,CAAC,IAAI,KAAK,aAAa;QACxB,CAAC,CAAC,CAAC,QAAQ,EAAE,aAAa,KAAK,EAAE,IAAI,CAAC,CAAC,EAAE,KAAK,QAAQ,CAAC,CAC1D,IAAI,IAAI,CACV,CAAC;AACJ,CAAC;AAED,SAAS,UAAU,CAAC,IAAY;IAC9B,OAAO,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC/E,CAAC;AAED,SAAS,gBAAgB,CAAC,IAAY;IACpC,IAAI,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;QAAE,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAE,CAAC;IACrD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,MAAM,UAAU,QAAQ,CAAC,KAAqB,EAAE,IAAqB;IACnE,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC;IAC/C,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,IAAI,EAAE,EAAE,GAAG,CAAC,CAAC;IACtD,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,IAAI,MAAM,CAAC;IAC3C,MAAM,UAAU,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;IACnE,MAAM,UAAU,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;IAEnE,MAAM,YAAY,GAAG,IAAI,GAAG,EAAqB,CAAC;IAClD,MAAM,YAAY,GAAG,IAAI,GAAG,EAAqB,CAAC;IAElD,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;QACtC,MAAM,IAAI,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QACnC,IAAI,IAAI;YAAE,YAAY,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;IAC3C,CAAC;IAED,IAAI,QAAQ,GAAG,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC;IAClE,IAAI,WAAW,GAAG,CAAC,CAAC;IAEpB,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,OAAO,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,KAAK,EAAE,EAAE,CAAC;QACpE,MAAM,YAAY,GAAa,EAAE,CAAC;QAClC,WAAW,GAAG,KAAK,GAAG,CAAC,CAAC;QAExB,KAAK,MAAM,MAAM,IAAI,QAAQ,EAAE,CAAC;YAC9B,IAAI,KAAK,GAAG,KAAK,CAAC,QAAQ,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;YAC9C,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC;YAC3E,IAAI,UAAU;gBAAE,KAAK,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;YAEpE,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBACzB,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC;gBAChC,MAAM,UAAU,GACd,IAAI,CAAC,QAAQ,KAAK,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC;gBAC3D,IAAI,YAAY,CAAC,GAAG,CAAC,UAAU,CAAC;oBAAE,SAAS;gBAC3C,MAAM,IAAI,GAAG,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;gBACvC,IAAI,CAAC,IAAI;oBAAE,SAAS;gBACpB,IAAI,UAAU,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC;oBAAE,SAAS;gBACvD,YAAY,CAAC,GAAG,CAAC,UAAU,EAAE,IAAI,CAAC,CAAC;gBACnC,YAAY,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YAChC,CAAC;QACH,CAAC;QACD,QAAQ,GAAG,YAAY,CAAC;IAC1B,CAAC;IAED,OAAO;QACL,KAAK,EAAE,CAAC,GAAG,YAAY,CAAC,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QAC1E,KAAK,EAAE,CAAC,GAAG,YAAY,CAAC,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QAC1E,IAAI,EAAE,WAAW;KAClB,CAAC;AACJ,CAAC;AAQD,MAAM,UAAU,MAAM,CACpB,KAAqB,EACrB,SAAiB,EACjB,OAAsB,EAAE;IAExB,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC;IACzC,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC;IAC7C,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,IAAI,CAAC,YAAY,EAAE,UAAU,EAAE,YAAY,CAAC,CAAC;IAC7E,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,IAAI,CAAC,aAAa,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC;IAEpE,IAAI,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;QAC1B,kBAAkB,CAAC,gBAAgB,CAAC,SAAS,CAAC,CAAC,CAAC;IAClD,CAAC;IAED,MAAM,KAAK,GAAsB,EAAE,CAAC;IAEpC,IAAI,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC;QAC1B,MAAM,IAAI,GAAG,mBAAmB,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;QACnD,IAAI,IAAI;YAAE,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,EAAE,IAAI,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;IAC9D,CAAC;SAAM,IAAI,UAAU,CAAC,SAAS,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;QAC9D,MAAM,MAAM,GAAG,SAAS,CAAC,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;QAC7C,MAAM,EAAE,KAAK,EAAE,GAAG,KAAK,CAAC,YAAY,EAAE,CAAC;QACvC,KAAK,MAAM,CAAC,IAAI,KAAK,EAAE,CAAC;YACtB,IAAI,CAAC,CAAC,IAAI,KAAK,MAAM,IAAI,CAAC,CAAC,IAAI,KAAK,aAAa;gBAAE,SAAS;YAC5D,MAAM,CAAC,GAAG,MAAM,CAAC,CAAC,CAAC,QAAQ,EAAE,IAAI,IAAI,CAAC,CAAC,QAAQ,EAAE,aAAa,IAAI,EAAE,CAAC,CAAC;YACtE,IAAI,CAAC,CAAC,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;gBAClD,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;YACpD,CAAC;QACH,CAAC;IACH,CAAC;SAAM,CAAC;QACN,KAAK,MAAM,GAAG,IAAI,KAAK,CAAC,WAAW,CAAC,SAAS,EAAE,CAAC,CAAC,EAAE,CAAC;YAClD,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,GAAG,CAAC,IAAI,EAAE,KAAK,EAAE,GAAG,EAAE,IAAI,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;QAClE,CAAC;IACH,CAAC;IAED,MAAM,UAAU,GAAG,IAAI,GAAG,EAA2B,CAAC;IAEtD,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,MAAM,GAAG,GAAG,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;QACzB,MAAM,QAAQ,GAAG,UAAU,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACrC,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC,KAAK,GAAG,QAAQ,CAAC,KAAK,EAAE,CAAC;YAC7C,UAAU,CAAC,GAAG,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;QAC5B,CAAC;QAED,MAAM,MAAM,GAAG,QAAQ,CAAC,KAAK,EAAE;YAC7B,WAAW,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;YAC3B,OAAO,EAAE,IAAI;YACb,SAAS;YACT,SAAS;YACT,SAAS,EAAE,MAAM;SAClB,CAAC,CAAC;QAEH,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;YAChC,MAAM,OAAO,GAAG,MAAM,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAChD,IAAI,KAAK,GAAG,IAAI,CAAC,KAAK,GAAG,OAAO,GAAG,IAAI,CAAC;YACxC,IAAI,IAAI,CAAC,IAAI,KAAK,aAAa;gBAAE,KAAK,IAAI,GAAG,CAAC;YAC9C,MAAM,IAAI,GAAG,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACrC,IAAI,CAAC,IAAI,IAAI,KAAK,GAAG,IAAI,CAAC,KAAK,EAAE,CAAC;gBAChC,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE;oBACtB,IAAI;oBACJ,KAAK;oBACL,IAAI,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,EAAE,CAAC;iBAC9B,CAAC,CAAC;YACL,CAAC;QACH,CAAC;IACH,CAAC;IAED,MAAM,MAAM,GAAG,CAAC,GAAG,UAAU,CAAC,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;QACpD,IAAI,CAAC,CAAC,KAAK,KAAK,CAAC,CAAC,KAAK;YAAE,OAAO,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC;QAClD,OAAO,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAC5C,CAAC,CAAC,CAAC;IAEH,OAAO;QACL,UAAU,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YAC7C,IAAI,EAAE,CAAC,CAAC,IAAI;YACZ,KAAK,EAAE,CAAC,CAAC,KAAK;YACd,IAAI,EAAE,CAAC,CAAC,IAAI;SACb,CAAC,CAAC;KACJ,CAAC;AACJ,CAAC"}
|
package/dist/graphrag/types.d.ts
CHANGED
|
@@ -1,5 +1,13 @@
|
|
|
1
|
-
export type NodeType = 'task' | 'file' | 'schema' | 'endpoint' | 'component' | 'entity' | 'concept' | 'gate';
|
|
2
|
-
export type EdgeType = 'depends_on' | 'implements' | 'uses' | 'references' | 'related_to' | 'contains' | 'extends' | 'verified_by';
|
|
1
|
+
export type NodeType = 'task' | 'file' | 'schema' | 'endpoint' | 'component' | 'entity' | 'concept' | 'gate' | 'code_symbol' | 'requirement';
|
|
2
|
+
export type EdgeType = 'depends_on' | 'implements' | 'uses' | 'references' | 'related_to' | 'contains' | 'extends' | 'verified_by' | 'affects' | 'derives_from';
|
|
3
|
+
/** All members of {@link NodeType} — use for zero-initialized statistics (RNF-05). */
|
|
4
|
+
export declare const ALL_NODE_TYPES: readonly ["task", "file", "schema", "endpoint", "component", "entity", "concept", "gate", "code_symbol", "requirement"];
|
|
5
|
+
/** All members of {@link EdgeType} — use for zero-initialized statistics (RNF-05). */
|
|
6
|
+
export declare const ALL_EDGE_TYPES: readonly ["depends_on", "implements", "uses", "references", "related_to", "contains", "extends", "verified_by", "affects", "derives_from"];
|
|
7
|
+
/** Zero-filled `nodesByType` — absent types stay `0`, never `NaN` (RNF-05). */
|
|
8
|
+
export declare function emptyNodesByType(): Record<NodeType, number>;
|
|
9
|
+
/** Zero-filled `edgesByType` — absent types stay `0`, never `NaN` (RNF-05). */
|
|
10
|
+
export declare function emptyEdgesByType(): Record<EdgeType, number>;
|
|
3
11
|
export interface GraphNode {
|
|
4
12
|
id: string;
|
|
5
13
|
type: NodeType;
|
|
@@ -45,6 +53,56 @@ export interface ComponentNode extends GraphNode {
|
|
|
45
53
|
framework?: string;
|
|
46
54
|
props?: string[];
|
|
47
55
|
}
|
|
56
|
+
export type CodeSymbolKind = 'function' | 'class' | 'method';
|
|
57
|
+
/**
|
|
58
|
+
* Canonical id: `code_symbol:{qualifiedName}` — e.g. `code_symbol:src/math.ts::add`
|
|
59
|
+
*/
|
|
60
|
+
export interface CodeSymbolNode extends GraphNode {
|
|
61
|
+
type: 'code_symbol';
|
|
62
|
+
path: string;
|
|
63
|
+
symbol: string;
|
|
64
|
+
kind: CodeSymbolKind;
|
|
65
|
+
qualifiedName: string;
|
|
66
|
+
line?: number;
|
|
67
|
+
}
|
|
68
|
+
/**
|
|
69
|
+
* Canonical id: `requirement:{reqId}` — e.g. `requirement:RF-01`
|
|
70
|
+
* (`task:{taskId}` e `file:{posixPath}` permanecem inalterados.)
|
|
71
|
+
*/
|
|
72
|
+
export interface RequirementNode extends GraphNode {
|
|
73
|
+
type: 'requirement';
|
|
74
|
+
reqId: string;
|
|
75
|
+
source: 'design' | 'blueprint' | 'tasks' | 'dag';
|
|
76
|
+
title: string;
|
|
77
|
+
priority?: 'MUST' | 'SHOULD' | 'COULD';
|
|
78
|
+
}
|
|
79
|
+
export interface TraverseOptions {
|
|
80
|
+
readonly seedNodeIds: readonly string[];
|
|
81
|
+
readonly maxHops?: number;
|
|
82
|
+
readonly maxFanout?: number;
|
|
83
|
+
readonly nodeTypes?: readonly NodeType[];
|
|
84
|
+
readonly edgeTypes?: readonly EdgeType[];
|
|
85
|
+
readonly direction?: 'out' | 'in' | 'both';
|
|
86
|
+
}
|
|
87
|
+
export interface TraverseResult {
|
|
88
|
+
readonly nodes: readonly GraphNode[];
|
|
89
|
+
readonly edges: readonly GraphEdge[];
|
|
90
|
+
readonly hops: number;
|
|
91
|
+
}
|
|
92
|
+
export interface LocateOptions {
|
|
93
|
+
readonly hops?: number;
|
|
94
|
+
readonly nodeTypes?: readonly NodeType[];
|
|
95
|
+
readonly edgeTypes?: readonly EdgeType[];
|
|
96
|
+
readonly limit?: number;
|
|
97
|
+
}
|
|
98
|
+
export interface LocateCandidate {
|
|
99
|
+
readonly node: GraphNode;
|
|
100
|
+
readonly score: number;
|
|
101
|
+
readonly path: readonly string[];
|
|
102
|
+
}
|
|
103
|
+
export interface LocateResult {
|
|
104
|
+
readonly candidates: ReadonlyArray<LocateCandidate>;
|
|
105
|
+
}
|
|
48
106
|
export interface SearchResult {
|
|
49
107
|
node: GraphNode;
|
|
50
108
|
score: number;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/graphrag/types.ts"],"names":[],"mappings":"AAAA,MAAM,MAAM,QAAQ,GAChB,MAAM,GACN,MAAM,GACN,QAAQ,GACR,UAAU,GACV,WAAW,GACX,QAAQ,GACR,SAAS,GACT,MAAM,CAAC;
|
|
1
|
+
{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/graphrag/types.ts"],"names":[],"mappings":"AAAA,MAAM,MAAM,QAAQ,GAChB,MAAM,GACN,MAAM,GACN,QAAQ,GACR,UAAU,GACV,WAAW,GACX,QAAQ,GACR,SAAS,GACT,MAAM,GACN,aAAa,GACb,aAAa,CAAC;AAElB,MAAM,MAAM,QAAQ,GAChB,YAAY,GACZ,YAAY,GACZ,MAAM,GACN,YAAY,GACZ,YAAY,GACZ,UAAU,GACV,SAAS,GACT,aAAa,GACb,SAAS,GACT,cAAc,CAAC;AAEnB,sFAAsF;AACtF,eAAO,MAAM,cAAc,yHAWa,CAAC;AAEzC,sFAAsF;AACtF,eAAO,MAAM,cAAc,4IAWa,CAAC;AAEzC,+EAA+E;AAC/E,wBAAgB,gBAAgB,IAAI,MAAM,CAAC,QAAQ,EAAE,MAAM,CAAC,CAE3D;AAED,+EAA+E;AAC/E,wBAAgB,gBAAgB,IAAI,MAAM,CAAC,QAAQ,EAAE,MAAM,CAAC,CAE3D;AAED,MAAM,WAAW,SAAS;IACxB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,QAAQ,CAAC;IACf,KAAK,EAAE,MAAM,CAAC;IACd,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACnC,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,SAAS;IACxB,EAAE,EAAE,MAAM,CAAC;IACX,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,EAAE,QAAQ,CAAC;IACf,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACpC;AAED,MAAM,WAAW,QAAS,SAAQ,SAAS;IACzC,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,SAAS,GAAG,aAAa,GAAG,MAAM,GAAG,QAAQ,GAAG,SAAS,CAAC;IAClE,UAAU,EAAE,KAAK,GAAG,KAAK,GAAG,MAAM,CAAC;IACnC,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,QAAS,SAAQ,SAAS;IACzC,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,IAAI,CAAC,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,UAAW,SAAQ,SAAS;IAC3C,IAAI,EAAE,QAAQ,CAAC;IACf,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,OAAO,CAAC,EAAE,MAAM,EAAE,CAAC;CACpB;AAED,MAAM,WAAW,YAAa,SAAQ,SAAS;IAC7C,IAAI,EAAE,UAAU,CAAC;IACjB,MAAM,EAAE,KAAK,GAAG,MAAM,GAAG,KAAK,GAAG,OAAO,GAAG,QAAQ,CAAC;IACpD,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,aAAc,SAAQ,SAAS;IAC9C,IAAI,EAAE,WAAW,CAAC;IAClB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,KAAK,CAAC,EAAE,MAAM,EAAE,CAAC;CAClB;AAED,MAAM,MAAM,cAAc,GAAG,UAAU,GAAG,OAAO,GAAG,QAAQ,CAAC;AAE7D;;GAEG;AACH,MAAM,WAAW,cAAe,SAAQ,SAAS;IAC/C,IAAI,EAAE,aAAa,CAAC;IACpB,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,EAAE,cAAc,CAAC;IACrB,aAAa,EAAE,MAAM,CAAC;IACtB,IAAI,CAAC,EAAE,MAAM,CAAC;CACf;AAED;;;GAGG;AACH,MAAM,WAAW,eAAgB,SAAQ,SAAS;IAChD,IAAI,EAAE,aAAa,CAAC;IACpB,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,QAAQ,GAAG,WAAW,GAAG,OAAO,GAAG,KAAK,CAAC;IACjD,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,CAAC,EAAE,MAAM,GAAG,QAAQ,GAAG,OAAO,CAAC;CACxC;AAED,MAAM,WAAW,eAAe;IAC9B,QAAQ,CAAC,WAAW,EAAE,SAAS,MAAM,EAAE,CAAC;IACxC,QAAQ,CAAC,OAAO,CAAC,EAAE,MAAM,CAAC;IAC1B,QAAQ,CAAC,SAAS,CAAC,EAAE,MAAM,CAAC;IAC5B,QAAQ,CAAC,SAAS,CAAC,EAAE,SAAS,QAAQ,EAAE,CAAC;IACzC,QAAQ,CAAC,SAAS,CAAC,EAAE,SAAS,QAAQ,EAAE,CAAC;IACzC,QAAQ,CAAC,SAAS,CAAC,EAAE,KAAK,GAAG,IAAI,GAAG,MAAM,CAAC;CAC5C;AAED,MAAM,WAAW,cAAc;IAC7B,QAAQ,CAAC,KAAK,EAAE,SAAS,SAAS,EAAE,CAAC;IACrC,QAAQ,CAAC,KAAK,EAAE,SAAS,SAAS,EAAE,CAAC;IACrC,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;CACvB;AAED,MAAM,WAAW,aAAa;IAC5B,QAAQ,CAAC,IAAI,CAAC,EAAE,MAAM,CAAC;IACvB,QAAQ,CAAC,SAAS,CAAC,EAAE,SAAS,QAAQ,EAAE,CAAC;IACzC,QAAQ,CAAC,SAAS,CAAC,EAAE,SAAS,QAAQ,EAAE,CAAC;IACzC,QAAQ,CAAC,KAAK,CAAC,EAAE,MAAM,CAAC;CACzB;AAED,MAAM,WAAW,eAAe;IAC9B,QAAQ,CAAC,IAAI,EAAE,SAAS,CAAC;IACzB,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAC;IACvB,QAAQ,CAAC,IAAI,EAAE,SAAS,MAAM,EAAE,CAAC;CAClC;AAED,MAAM,WAAW,YAAY;IAC3B,QAAQ,CAAC,UAAU,EAAE,aAAa,CAAC,eAAe,CAAC,CAAC;CACrD;AAED,MAAM,WAAW,YAAY;IAC3B,IAAI,EAAE,SAAS,CAAC;IAChB,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,WAAW,eAAe;IAC9B,UAAU,EAAE,MAAM,CAAC;IACnB,UAAU,EAAE,MAAM,CAAC;IACnB,WAAW,EAAE,MAAM,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;IACtC,WAAW,EAAE,MAAM,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;CACvC"}
|
package/dist/graphrag/types.js
CHANGED
|
@@ -1,2 +1,35 @@
|
|
|
1
|
-
|
|
1
|
+
/** All members of {@link NodeType} — use for zero-initialized statistics (RNF-05). */
|
|
2
|
+
export const ALL_NODE_TYPES = [
|
|
3
|
+
'task',
|
|
4
|
+
'file',
|
|
5
|
+
'schema',
|
|
6
|
+
'endpoint',
|
|
7
|
+
'component',
|
|
8
|
+
'entity',
|
|
9
|
+
'concept',
|
|
10
|
+
'gate',
|
|
11
|
+
'code_symbol',
|
|
12
|
+
'requirement',
|
|
13
|
+
];
|
|
14
|
+
/** All members of {@link EdgeType} — use for zero-initialized statistics (RNF-05). */
|
|
15
|
+
export const ALL_EDGE_TYPES = [
|
|
16
|
+
'depends_on',
|
|
17
|
+
'implements',
|
|
18
|
+
'uses',
|
|
19
|
+
'references',
|
|
20
|
+
'related_to',
|
|
21
|
+
'contains',
|
|
22
|
+
'extends',
|
|
23
|
+
'verified_by',
|
|
24
|
+
'affects',
|
|
25
|
+
'derives_from',
|
|
26
|
+
];
|
|
27
|
+
/** Zero-filled `nodesByType` — absent types stay `0`, never `NaN` (RNF-05). */
|
|
28
|
+
export function emptyNodesByType() {
|
|
29
|
+
return Object.fromEntries(ALL_NODE_TYPES.map((t) => [t, 0]));
|
|
30
|
+
}
|
|
31
|
+
/** Zero-filled `edgesByType` — absent types stay `0`, never `NaN` (RNF-05). */
|
|
32
|
+
export function emptyEdgesByType() {
|
|
33
|
+
return Object.fromEntries(ALL_EDGE_TYPES.map((t) => [t, 0]));
|
|
34
|
+
}
|
|
2
35
|
//# sourceMappingURL=types.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"types.js","sourceRoot":"","sources":["../../src/graphrag/types.ts"],"names":[],"mappings":""}
|
|
1
|
+
{"version":3,"file":"types.js","sourceRoot":"","sources":["../../src/graphrag/types.ts"],"names":[],"mappings":"AAwBA,sFAAsF;AACtF,MAAM,CAAC,MAAM,cAAc,GAAG;IAC5B,MAAM;IACN,MAAM;IACN,QAAQ;IACR,UAAU;IACV,WAAW;IACX,QAAQ;IACR,SAAS;IACT,MAAM;IACN,aAAa;IACb,aAAa;CACyB,CAAC;AAEzC,sFAAsF;AACtF,MAAM,CAAC,MAAM,cAAc,GAAG;IAC5B,YAAY;IACZ,YAAY;IACZ,MAAM;IACN,YAAY;IACZ,YAAY;IACZ,UAAU;IACV,SAAS;IACT,aAAa;IACb,SAAS;IACT,cAAc;CACwB,CAAC;AAEzC,+EAA+E;AAC/E,MAAM,UAAU,gBAAgB;IAC9B,OAAO,MAAM,CAAC,WAAW,CAAC,cAAc,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAA6B,CAAC;AAC3F,CAAC;AAED,+EAA+E;AAC/E,MAAM,UAAU,gBAAgB;IAC9B,OAAO,MAAM,CAAC,WAAW,CAAC,cAAc,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAA6B,CAAC;AAC3F,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"mcp-graph.test.d.ts","sourceRoot":"","sources":["../../../src/mcp-server/__tests__/mcp-graph.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,85 @@
|
|
|
1
|
+
import { describe, it, expect, beforeEach, afterEach } from 'vitest';
|
|
2
|
+
import fs from 'fs-extra';
|
|
3
|
+
import os from 'node:os';
|
|
4
|
+
import path from 'node:path';
|
|
5
|
+
import request from 'supertest';
|
|
6
|
+
import { createMcpServer } from '../server.js';
|
|
7
|
+
import { buildGraphFromFixture, loadFixture } from '../../__tests__/graphrag/fixtures/dual-graph/build-fixture-graph.js';
|
|
8
|
+
import { JsonGraph } from '../../graphrag/json-graph.js';
|
|
9
|
+
describe('MCP graph tools', () => {
|
|
10
|
+
let projectRoot;
|
|
11
|
+
beforeEach(async () => {
|
|
12
|
+
projectRoot = await fs.mkdtemp(path.join(os.tmpdir(), 'mcp-graph-'));
|
|
13
|
+
await fs.ensureDir(path.join(projectRoot, 'DARE'));
|
|
14
|
+
await fs.writeJson(path.join(projectRoot, 'dare.config.json'), { name: 'mcp-graph-test' });
|
|
15
|
+
await fs.writeFile(path.join(projectRoot, 'dare-graph.yml'), 'backend: json\njson:\n path: .dare/graph.json\n');
|
|
16
|
+
const graphPath = path.join(projectRoot, '.dare', 'graph.json');
|
|
17
|
+
const graph = new JsonGraph(graphPath);
|
|
18
|
+
await graph.init();
|
|
19
|
+
buildGraphFromFixture(graph, loadFixture('impact-chain'));
|
|
20
|
+
graph.close();
|
|
21
|
+
});
|
|
22
|
+
afterEach(async () => {
|
|
23
|
+
await fs.remove(projectRoot).catch(() => undefined);
|
|
24
|
+
});
|
|
25
|
+
function app() {
|
|
26
|
+
return createMcpServer(projectRoot, { allowLoopbackWithoutToken: true });
|
|
27
|
+
}
|
|
28
|
+
it('lists graph tools in GET /tools', async () => {
|
|
29
|
+
const res = await request(app()).get('/tools');
|
|
30
|
+
const names = res.body.tools.map((t) => t.name);
|
|
31
|
+
expect(names).toContain('graph_locate');
|
|
32
|
+
expect(names).toContain('graph_map_requirement');
|
|
33
|
+
expect(names).toContain('graph_traverse');
|
|
34
|
+
});
|
|
35
|
+
it('POST /graph/locate returns candidates', async () => {
|
|
36
|
+
const res = await request(app())
|
|
37
|
+
.post('/graph/locate')
|
|
38
|
+
.send({ seed: 'math', hops: 3, limit: 5 });
|
|
39
|
+
expect(res.status).toBe(200);
|
|
40
|
+
expect(res.body.success).toBe(true);
|
|
41
|
+
expect(res.body.candidates.length).toBeGreaterThan(0);
|
|
42
|
+
});
|
|
43
|
+
it('POST /graph/locate rejects missing seed', async () => {
|
|
44
|
+
const res = await request(app()).post('/graph/locate').send({});
|
|
45
|
+
expect(res.status).toBe(400);
|
|
46
|
+
expect(res.body.error).toBe('seed is required');
|
|
47
|
+
});
|
|
48
|
+
it('POST /graph/locate rejects path escape', async () => {
|
|
49
|
+
const res = await request(app()).post('/graph/locate').send({ seed: '../etc/passwd' });
|
|
50
|
+
expect(res.status).toBe(403);
|
|
51
|
+
});
|
|
52
|
+
it('POST /graph/map-requirement maps RF-03', async () => {
|
|
53
|
+
const res = await request(app()).post('/graph/map-requirement').send({ reqId: 'RF-03' });
|
|
54
|
+
expect(res.status).toBe(200);
|
|
55
|
+
expect(res.body.success).toBe(true);
|
|
56
|
+
expect(res.body.symbols).toContain('src/math.ts::add');
|
|
57
|
+
expect(res.body.tasks).toContain('task-201');
|
|
58
|
+
});
|
|
59
|
+
it('POST /graph/map-requirement rejects invalid reqId', async () => {
|
|
60
|
+
const res = await request(app()).post('/graph/map-requirement').send({ reqId: 'BAD' });
|
|
61
|
+
expect(res.status).toBe(400);
|
|
62
|
+
expect(res.body.error).toBe('invalid reqId');
|
|
63
|
+
});
|
|
64
|
+
it('POST /graph/traverse returns nodes and edges', async () => {
|
|
65
|
+
const res = await request(app()).post('/graph/traverse').send({
|
|
66
|
+
seedNodeIds: ['requirement:RF-03'],
|
|
67
|
+
maxHops: 3,
|
|
68
|
+
edgeTypes: ['depends_on', 'implements', 'affects'],
|
|
69
|
+
});
|
|
70
|
+
expect(res.status).toBe(200);
|
|
71
|
+
expect(res.body.success).toBe(true);
|
|
72
|
+
expect(res.body.nodes.length).toBeGreaterThan(0);
|
|
73
|
+
});
|
|
74
|
+
it('POST /graph/traverse requires seedNodeIds', async () => {
|
|
75
|
+
const res = await request(app()).post('/graph/traverse').send({});
|
|
76
|
+
expect(res.status).toBe(400);
|
|
77
|
+
expect(res.body.error).toBe('seedNodeIds is required');
|
|
78
|
+
});
|
|
79
|
+
it('deterministic locate responses', async () => {
|
|
80
|
+
const a = await request(app()).post('/graph/locate').send({ seed: 'math' });
|
|
81
|
+
const b = await request(app()).post('/graph/locate').send({ seed: 'math' });
|
|
82
|
+
expect(a.body).toEqual(b.body);
|
|
83
|
+
});
|
|
84
|
+
});
|
|
85
|
+
//# sourceMappingURL=mcp-graph.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"mcp-graph.test.js","sourceRoot":"","sources":["../../../src/mcp-server/__tests__/mcp-graph.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACrE,OAAO,EAAE,MAAM,UAAU,CAAC;AAC1B,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,OAAO,MAAM,WAAW,CAAC;AAChC,OAAO,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAC/C,OAAO,EAAE,qBAAqB,EAAE,WAAW,EAAE,MAAM,qEAAqE,CAAC;AACzH,OAAO,EAAE,SAAS,EAAE,MAAM,8BAA8B,CAAC;AAEzD,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,IAAI,WAAmB,CAAC;IAExB,UAAU,CAAC,KAAK,IAAI,EAAE;QACpB,WAAW,GAAG,MAAM,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,YAAY,CAAC,CAAC,CAAC;QACrE,MAAM,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,MAAM,CAAC,CAAC,CAAC;QACnD,MAAM,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,kBAAkB,CAAC,EAAE,EAAE,IAAI,EAAE,gBAAgB,EAAE,CAAC,CAAC;QAC3F,MAAM,EAAE,CAAC,SAAS,CAChB,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,gBAAgB,CAAC,EACxC,kDAAkD,CACnD,CAAC;QAEF,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,OAAO,EAAE,YAAY,CAAC,CAAC;QAChE,MAAM,KAAK,GAAG,IAAI,SAAS,CAAC,SAAS,CAAC,CAAC;QACvC,MAAM,KAAK,CAAC,IAAI,EAAE,CAAC;QACnB,qBAAqB,CAAC,KAAK,EAAE,WAAW,CAAC,cAAc,CAAC,CAAC,CAAC;QAC1D,KAAK,CAAC,KAAK,EAAE,CAAC;IAChB,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,MAAM,EAAE,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;IACtD,CAAC,CAAC,CAAC;IAEH,SAAS,GAAG;QACV,OAAO,eAAe,CAAC,WAAW,EAAE,EAAE,yBAAyB,EAAE,IAAI,EAAE,CAAC,CAAC;IAC3E,CAAC;IAED,EAAE,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;QAC/C,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,GAAG,EAAE,CAAC,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC/C,MAAM,KAAK,GAAI,GAAG,CAAC,IAAI,CAAC,KAAiC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QAC7E,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;QACxC,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,uBAAuB,CAAC,CAAC;QACjD,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;IAC5C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;QACrD,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,GAAG,EAAE,CAAC;aAC7B,IAAI,CAAC,eAAe,CAAC;aACrB,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC;QAC7C,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC7B,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACpC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;QACvD,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,GAAG,EAAE,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAChE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC7B,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;QACtD,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,GAAG,EAAE,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,eAAe,EAAE,CAAC,CAAC;QACvF,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAC/B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;QACtD,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,GAAG,EAAE,CAAC,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC,CAAC;QACzF,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC7B,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACpC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,kBAAkB,CAAC,CAAC;QACvD,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;QACjE,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,GAAG,EAAE,CAAC,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC;QACvF,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC7B,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;QAC5D,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,GAAG,EAAE,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC,IAAI,CAAC;YAC5D,WAAW,EAAE,CAAC,mBAAmB,CAAC;YAClC,OAAO,EAAE,CAAC;YACV,SAAS,EAAE,CAAC,YAAY,EAAE,YAAY,EAAE,SAAS,CAAC;SACnD,CAAC,CAAC;QACH,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC7B,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACpC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;IACnD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;QACzD,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,GAAG,EAAE,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAClE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC7B,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;IACzD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;QAC9C,MAAM,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,EAAE,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC;QAC5E,MAAM,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,EAAE,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC;QAC5E,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;IACjC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"server.d.ts","sourceRoot":"","sources":["../../src/mcp-server/server.ts"],"names":[],"mappings":"AAEA,OAAgB,EAAE,KAAK,OAAO,EAAkD,MAAM,SAAS,CAAC;
|
|
1
|
+
{"version":3,"file":"server.d.ts","sourceRoot":"","sources":["../../src/mcp-server/server.ts"],"names":[],"mappings":"AAEA,OAAgB,EAAE,KAAK,OAAO,EAAkD,MAAM,SAAS,CAAC;AAiBhG,MAAM,WAAW,YAAY;IAC3B,IAAI,EAAE,MAAM,GAAG,MAAM,GAAG,YAAY,GAAG,cAAc,GAAG,QAAQ,GAAG,UAAU,CAAC;IAC9E,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,qEAAqE;IACrE,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB;AAED,MAAM,WAAW,aAAa;IAC5B,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,EAAE,MAAM,CAAC;CAChB;AAED,MAAM,WAAW,UAAU;IACzB,EAAE,EAAE,MAAM,CAAC;IACX,MAAM,EAAE,SAAS,GAAG,aAAa,GAAG,MAAM,GAAG,QAAQ,GAAG,SAAS,CAAC;IAClE,SAAS,EAAE,MAAM,CAAC;CACnB;AAuBD,MAAM,WAAW,gBAAgB;IAC/B,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,yBAAyB,CAAC,EAAE,OAAO,CAAC;CACrC;AA0DD,wBAAgB,eAAe,CAC7B,WAAW,GAAE,MAAuD,EACpE,OAAO,GAAE,gBAAqB,GAC7B,OAAO,CAkWT"}
|