@dewtech/dare-cli 3.4.0 → 3.6.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/__tests__/ide-adapters.test.d.ts +2 -0
- package/dist/__tests__/ide-adapters.test.d.ts.map +1 -0
- package/dist/__tests__/ide-adapters.test.js +39 -0
- package/dist/__tests__/ide-adapters.test.js.map +1 -0
- package/dist/__tests__/ide-command-parity.test.js +2 -0
- package/dist/__tests__/ide-command-parity.test.js.map +1 -1
- package/dist/bin/dare.js +4 -0
- package/dist/bin/dare.js.map +1 -1
- 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__/hooks.test.d.ts +2 -0
- package/dist/commands/__tests__/hooks.test.d.ts.map +1 -0
- package/dist/commands/__tests__/hooks.test.js +109 -0
- package/dist/commands/__tests__/hooks.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__/steering.test.d.ts +2 -0
- package/dist/commands/__tests__/steering.test.d.ts.map +1 -0
- package/dist/commands/__tests__/steering.test.js +69 -0
- package/dist/commands/__tests__/steering.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/commands/hooks.d.ts +3 -0
- package/dist/commands/hooks.d.ts.map +1 -0
- package/dist/commands/hooks.js +163 -0
- package/dist/commands/hooks.js.map +1 -0
- package/dist/commands/steering.d.ts +3 -0
- package/dist/commands/steering.d.ts.map +1 -0
- package/dist/commands/steering.js +70 -0
- package/dist/commands/steering.js.map +1 -0
- 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/hooks/__tests__/allowlist.test.d.ts +2 -0
- package/dist/hooks/__tests__/allowlist.test.d.ts.map +1 -0
- package/dist/hooks/__tests__/allowlist.test.js +29 -0
- package/dist/hooks/__tests__/allowlist.test.js.map +1 -0
- package/dist/hooks/__tests__/config.test.d.ts +2 -0
- package/dist/hooks/__tests__/config.test.d.ts.map +1 -0
- package/dist/hooks/__tests__/config.test.js +57 -0
- package/dist/hooks/__tests__/config.test.js.map +1 -0
- package/dist/hooks/__tests__/dispatcher.security.test.d.ts +2 -0
- package/dist/hooks/__tests__/dispatcher.security.test.d.ts.map +1 -0
- package/dist/hooks/__tests__/dispatcher.security.test.js +86 -0
- package/dist/hooks/__tests__/dispatcher.security.test.js.map +1 -0
- package/dist/hooks/__tests__/dispatcher.test.d.ts +2 -0
- package/dist/hooks/__tests__/dispatcher.test.d.ts.map +1 -0
- package/dist/hooks/__tests__/dispatcher.test.js +69 -0
- package/dist/hooks/__tests__/dispatcher.test.js.map +1 -0
- package/dist/hooks/__tests__/idempotency.test.d.ts +2 -0
- package/dist/hooks/__tests__/idempotency.test.d.ts.map +1 -0
- package/dist/hooks/__tests__/idempotency.test.js +45 -0
- package/dist/hooks/__tests__/idempotency.test.js.map +1 -0
- package/dist/hooks/__tests__/telemetry.test.d.ts +2 -0
- package/dist/hooks/__tests__/telemetry.test.d.ts.map +1 -0
- package/dist/hooks/__tests__/telemetry.test.js +52 -0
- package/dist/hooks/__tests__/telemetry.test.js.map +1 -0
- package/dist/hooks/allowlist.d.ts +24 -0
- package/dist/hooks/allowlist.d.ts.map +1 -0
- package/dist/hooks/allowlist.js +61 -0
- package/dist/hooks/allowlist.js.map +1 -0
- package/dist/hooks/config.d.ts +24 -0
- package/dist/hooks/config.d.ts.map +1 -0
- package/dist/hooks/config.js +82 -0
- package/dist/hooks/config.js.map +1 -0
- package/dist/hooks/dispatcher.d.ts +17 -0
- package/dist/hooks/dispatcher.d.ts.map +1 -0
- package/dist/hooks/dispatcher.js +157 -0
- package/dist/hooks/dispatcher.js.map +1 -0
- package/dist/hooks/idempotency.d.ts +14 -0
- package/dist/hooks/idempotency.d.ts.map +1 -0
- package/dist/hooks/idempotency.js +64 -0
- package/dist/hooks/idempotency.js.map +1 -0
- package/dist/hooks/telemetry.d.ts +12 -0
- package/dist/hooks/telemetry.d.ts.map +1 -0
- package/dist/hooks/telemetry.js +66 -0
- package/dist/hooks/telemetry.js.map +1 -0
- package/dist/hooks/types.d.ts +36 -0
- package/dist/hooks/types.d.ts.map +1 -0
- package/dist/hooks/types.js +7 -0
- package/dist/hooks/types.js.map +1 -0
- package/dist/index.d.ts +2 -0
- package/dist/index.d.ts.map +1 -1
- package/dist/index.js +2 -0
- package/dist/index.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/__tests__/mcp-steering.test.d.ts +2 -0
- package/dist/mcp-server/__tests__/mcp-steering.test.d.ts.map +1 -0
- package/dist/mcp-server/__tests__/mcp-steering.test.js +90 -0
- package/dist/mcp-server/__tests__/mcp-steering.test.js.map +1 -0
- package/dist/mcp-server/server.d.ts.map +1 -1
- package/dist/mcp-server/server.js +136 -1
- package/dist/mcp-server/server.js.map +1 -1
- package/dist/steering/__tests__/loader.test.d.ts +2 -0
- package/dist/steering/__tests__/loader.test.d.ts.map +1 -0
- package/dist/steering/__tests__/loader.test.js +80 -0
- package/dist/steering/__tests__/loader.test.js.map +1 -0
- package/dist/steering/__tests__/resolver.security.test.d.ts +2 -0
- package/dist/steering/__tests__/resolver.security.test.d.ts.map +1 -0
- package/dist/steering/__tests__/resolver.security.test.js +42 -0
- package/dist/steering/__tests__/resolver.security.test.js.map +1 -0
- package/dist/steering/__tests__/resolver.test.d.ts +2 -0
- package/dist/steering/__tests__/resolver.test.d.ts.map +1 -0
- package/dist/steering/__tests__/resolver.test.js +75 -0
- package/dist/steering/__tests__/resolver.test.js.map +1 -0
- package/dist/steering/loader.d.ts +8 -0
- package/dist/steering/loader.d.ts.map +1 -0
- package/dist/steering/loader.js +98 -0
- package/dist/steering/loader.js.map +1 -0
- package/dist/steering/resolver.d.ts +6 -0
- package/dist/steering/resolver.d.ts.map +1 -0
- package/dist/steering/resolver.js +59 -0
- package/dist/steering/resolver.js.map +1 -0
- package/dist/steering/types.d.ts +22 -0
- package/dist/steering/types.d.ts.map +1 -0
- package/dist/steering/types.js +2 -0
- package/dist/steering/types.js.map +1 -0
- package/dist/utils/UpdateApplier.d.ts.map +1 -1
- package/dist/utils/UpdateApplier.js +2 -0
- package/dist/utils/UpdateApplier.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.d.ts.map +1 -1
- package/dist/utils/project-generator.js +14 -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
- package/templates/hooks/dare.config.hooks.example.json +12 -0
- package/templates/hooks/pre-commit-dare-validate +2 -2
- package/templates/ide/antigravity/.agents/skills/dare-hooks/SKILL.md +13 -0
- package/templates/ide/antigravity/.agents/skills/dare-steering/SKILL.md +15 -0
- package/templates/ide/antigravity/templates/HOOKS-ADAPTER.md +14 -0
- package/templates/ide/claude/.claude/commands/dare-hooks.md +17 -0
- package/templates/ide/claude/.claude/commands/dare-steering.md +19 -0
- package/templates/ide/claude/.claude/settings.example.json +1 -1
- package/templates/ide/cursor/.cursor/commands/dare-hooks.md +17 -0
- package/templates/ide/cursor/.cursor/commands/dare-steering.md +19 -0
- package/templates/ide/cursor/templates/HOOKS-ADAPTER.md +14 -0
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"buildLocateContext.test.d.ts","sourceRoot":"","sources":["../../../src/__tests__/dag-runner/buildLocateContext.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,53 @@
|
|
|
1
|
+
import { describe, it, expect, beforeEach } from 'vitest';
|
|
2
|
+
import path from 'node:path';
|
|
3
|
+
import os from 'node:os';
|
|
4
|
+
import fs from 'fs-extra';
|
|
5
|
+
import { JsonGraph } from '../../graphrag/json-graph.js';
|
|
6
|
+
import { buildGraphFromFixture, loadFixture } from '../graphrag/fixtures/dual-graph/build-fixture-graph.js';
|
|
7
|
+
import { buildLocateContext, loadGraphLocateConfig } from '../../dag-runner/graph-locate.js';
|
|
8
|
+
describe('ralph-locate buildLocateContext', () => {
|
|
9
|
+
let graph;
|
|
10
|
+
let cwd;
|
|
11
|
+
const task = {
|
|
12
|
+
id: 'task-216',
|
|
13
|
+
title: 'locate test',
|
|
14
|
+
status: 'PENDING',
|
|
15
|
+
complexity: 'MED',
|
|
16
|
+
depends_on: [],
|
|
17
|
+
subtask_prompt: 'Edit src/math.ts and src/math.ts::add for RF-03',
|
|
18
|
+
};
|
|
19
|
+
beforeEach(async () => {
|
|
20
|
+
cwd = path.join(os.tmpdir(), `locate-ctx-${Date.now()}`);
|
|
21
|
+
await fs.ensureDir(cwd);
|
|
22
|
+
const file = path.join(cwd, 'graph.json');
|
|
23
|
+
graph = new JsonGraph(file);
|
|
24
|
+
await graph.init();
|
|
25
|
+
buildGraphFromFixture(graph, loadFixture('locate/math'));
|
|
26
|
+
});
|
|
27
|
+
it('returns undefined when disabled', () => {
|
|
28
|
+
const cfg = { enabled: false, hops: 3, limit: 5 };
|
|
29
|
+
expect(buildLocateContext(graph, task, cfg)).toBeUndefined();
|
|
30
|
+
});
|
|
31
|
+
it('returns undefined on empty graph', async () => {
|
|
32
|
+
const empty = new JsonGraph(path.join(cwd, 'empty.json'));
|
|
33
|
+
await empty.init();
|
|
34
|
+
const cfg = { enabled: true, hops: 3, limit: 5 };
|
|
35
|
+
expect(buildLocateContext(empty, task, cfg)).toBeUndefined();
|
|
36
|
+
});
|
|
37
|
+
it('formats markdown candidates when enabled', () => {
|
|
38
|
+
const md = buildLocateContext(graph, task, { enabled: true, hops: 3, limit: 5 });
|
|
39
|
+
expect(md).toContain('## Graph locate (deterministic)');
|
|
40
|
+
expect(md).toContain('Candidates:');
|
|
41
|
+
expect(md).toMatch(/score \d+\.\d+/);
|
|
42
|
+
});
|
|
43
|
+
it('loadGraphLocateConfig reads dare.config.json', async () => {
|
|
44
|
+
await fs.writeJson(path.join(cwd, 'dare.config.json'), {
|
|
45
|
+
graph: { locateBeforePatch: true, locateHops: 2, locateLimit: 3 },
|
|
46
|
+
});
|
|
47
|
+
const cfg = loadGraphLocateConfig(cwd);
|
|
48
|
+
expect(cfg.enabled).toBe(true);
|
|
49
|
+
expect(cfg.hops).toBe(2);
|
|
50
|
+
expect(cfg.limit).toBe(3);
|
|
51
|
+
});
|
|
52
|
+
});
|
|
53
|
+
//# sourceMappingURL=buildLocateContext.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"buildLocateContext.test.js","sourceRoot":"","sources":["../../../src/__tests__/dag-runner/buildLocateContext.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC1D,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,EAAE,MAAM,UAAU,CAAC;AAC1B,OAAO,EAAE,SAAS,EAAE,MAAM,8BAA8B,CAAC;AACzD,OAAO,EAAE,qBAAqB,EAAE,WAAW,EAAE,MAAM,wDAAwD,CAAC;AAC5G,OAAO,EAAE,kBAAkB,EAAE,qBAAqB,EAAE,MAAM,kCAAkC,CAAC;AAG7F,QAAQ,CAAC,iCAAiC,EAAE,GAAG,EAAE;IAC/C,IAAI,KAAgB,CAAC;IACrB,IAAI,GAAW,CAAC;IAEhB,MAAM,IAAI,GAAY;QACpB,EAAE,EAAE,UAAU;QACd,KAAK,EAAE,aAAa;QACpB,MAAM,EAAE,SAAS;QACjB,UAAU,EAAE,KAAK;QACjB,UAAU,EAAE,EAAE;QACd,cAAc,EAAE,iDAAiD;KAClE,CAAC;IAEF,UAAU,CAAC,KAAK,IAAI,EAAE;QACpB,GAAG,GAAG,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,cAAc,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;QACzD,MAAM,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;QACxB,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,YAAY,CAAC,CAAC;QAC1C,KAAK,GAAG,IAAI,SAAS,CAAC,IAAI,CAAC,CAAC;QAC5B,MAAM,KAAK,CAAC,IAAI,EAAE,CAAC;QACnB,qBAAqB,CAAC,KAAK,EAAE,WAAW,CAAC,aAAa,CAAC,CAAC,CAAC;IAC3D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;QACzC,MAAM,GAAG,GAAG,EAAE,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC;QAClD,MAAM,CAAC,kBAAkB,CAAC,KAAK,EAAE,IAAI,EAAE,GAAG,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;IAC/D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;QAChD,MAAM,KAAK,GAAG,IAAI,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,YAAY,CAAC,CAAC,CAAC;QAC1D,MAAM,KAAK,CAAC,IAAI,EAAE,CAAC;QACnB,MAAM,GAAG,GAAG,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC;QACjD,MAAM,CAAC,kBAAkB,CAAC,KAAK,EAAE,IAAI,EAAE,GAAG,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;IAC/D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;QAClD,MAAM,EAAE,GAAG,kBAAkB,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC;QACjF,MAAM,CAAC,EAAE,CAAC,CAAC,SAAS,CAAC,iCAAiC,CAAC,CAAC;QACxD,MAAM,CAAC,EAAE,CAAC,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;QACpC,MAAM,CAAC,EAAE,CAAC,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC;IACvC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;QAC5D,MAAM,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,kBAAkB,CAAC,EAAE;YACrD,KAAK,EAAE,EAAE,iBAAiB,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,EAAE,WAAW,EAAE,CAAC,EAAE;SAClE,CAAC,CAAC;QACH,MAAM,GAAG,GAAG,qBAAqB,CAAC,GAAG,CAAC,CAAC;QACvC,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC/B,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACzB,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAC5B,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"dual-graph.test.d.ts","sourceRoot":"","sources":["../../src/__tests__/dual-graph.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,141 @@
|
|
|
1
|
+
import fs from 'node:fs';
|
|
2
|
+
import os from 'node:os';
|
|
3
|
+
import path from 'node:path';
|
|
4
|
+
import { fileURLToPath } from 'node:url';
|
|
5
|
+
import { describe, it, expect, beforeEach } from 'vitest';
|
|
6
|
+
import { extractSymbolsFromPaths } from '../graphrag/code-index.js';
|
|
7
|
+
import { JsonGraph } from '../graphrag/json-graph.js';
|
|
8
|
+
import { traverse } from '../graphrag/traverse.js';
|
|
9
|
+
import { collectImpact, collectOwners, traceRequirement } from '../commands/graph-queries.js';
|
|
10
|
+
import { buildGraphFromFixture, loadFixture, } from './graphrag/fixtures/dual-graph/build-fixture-graph.js';
|
|
11
|
+
const fixturesDir = path.join(path.dirname(fileURLToPath(import.meta.url)), '../graphrag/__tests__/fixtures/code-index');
|
|
12
|
+
const srcRoot = path.join(path.dirname(fileURLToPath(import.meta.url)), '..');
|
|
13
|
+
describe('dual-graph integration (O-01…O-07)', () => {
|
|
14
|
+
let projectRoot;
|
|
15
|
+
let graph;
|
|
16
|
+
beforeEach(async () => {
|
|
17
|
+
projectRoot = await fs.promises.mkdtemp(path.join(os.tmpdir(), 'dual-graph-int-'));
|
|
18
|
+
const file = path.join(projectRoot, 'graph.json');
|
|
19
|
+
graph = new JsonGraph(file);
|
|
20
|
+
await graph.init();
|
|
21
|
+
});
|
|
22
|
+
it('O-01: extractSymbolsFromPaths ≥90% top-level symbols', async () => {
|
|
23
|
+
const codeFixtures = path.join(fixturesDir);
|
|
24
|
+
for (const name of await fs.promises.readdir(codeFixtures)) {
|
|
25
|
+
if (name.endsWith('.json'))
|
|
26
|
+
continue;
|
|
27
|
+
await fs.promises.copyFile(path.join(codeFixtures, name), path.join(projectRoot, name));
|
|
28
|
+
}
|
|
29
|
+
const expected = JSON.parse(fs.readFileSync(path.join(codeFixtures, 'multi-lang-expected.json'), 'utf8'));
|
|
30
|
+
const paths = ['sample.ts', 'sample.py', 'sample.go', 'sample.rs', 'sample.php'];
|
|
31
|
+
const syms = extractSymbolsFromPaths(paths, projectRoot);
|
|
32
|
+
const found = new Set(syms.map((s) => s.qualifiedName));
|
|
33
|
+
const ratio = expected.expectedSymbols.filter((q) => found.has(q)).length /
|
|
34
|
+
expected.expectedSymbols.length;
|
|
35
|
+
expect(ratio).toBeGreaterThanOrEqual(0.9);
|
|
36
|
+
});
|
|
37
|
+
it('O-02: owners durationMs < 200', () => {
|
|
38
|
+
buildGraphFromFixture(graph, loadFixture('owners-chain'));
|
|
39
|
+
const result = collectOwners(graph, 'src/commands/execute.ts');
|
|
40
|
+
expect(result.durationMs).toBeLessThan(200);
|
|
41
|
+
expect(result.owners.some((o) => o.type === 'task')).toBe(true);
|
|
42
|
+
});
|
|
43
|
+
it('O-03: impact recall 100% on impact-chain', () => {
|
|
44
|
+
buildGraphFromFixture(graph, loadFixture('impact-chain'));
|
|
45
|
+
const result = collectImpact(graph, 'src/math.ts');
|
|
46
|
+
expect(result.durationMs).toBeLessThan(500);
|
|
47
|
+
expect(result.impacted.tasks).toContain('task-201');
|
|
48
|
+
expect(result.impacted.requirements).toContain('RF-03');
|
|
49
|
+
});
|
|
50
|
+
it('O-04: locate top-5 hit rate ≥85%', async () => {
|
|
51
|
+
const cases = [
|
|
52
|
+
{ fixture: 'locate/math', seed: 'math', expected: ['code_symbol:src/math.ts::add'] },
|
|
53
|
+
{
|
|
54
|
+
fixture: 'locate/auth',
|
|
55
|
+
seed: 'auth',
|
|
56
|
+
expected: ['code_symbol:src/auth/login.ts::authenticate', 'file:src/auth/login.ts'],
|
|
57
|
+
},
|
|
58
|
+
];
|
|
59
|
+
let hits = 0;
|
|
60
|
+
for (const { fixture, seed, expected } of cases) {
|
|
61
|
+
const g = new JsonGraph(path.join(projectRoot, `${fixture}.json`));
|
|
62
|
+
await g.init();
|
|
63
|
+
buildGraphFromFixture(g, loadFixture(fixture));
|
|
64
|
+
const top = g.locate(seed, { limit: 5 }).candidates.map((c) => c.node.id);
|
|
65
|
+
if (expected.some((id) => top.includes(id)))
|
|
66
|
+
hits++;
|
|
67
|
+
}
|
|
68
|
+
expect(hits / cases.length).toBeGreaterThanOrEqual(0.85);
|
|
69
|
+
});
|
|
70
|
+
it('O-05: json persistence write→reopen→read', async () => {
|
|
71
|
+
const file = path.join(projectRoot, 'persist.json');
|
|
72
|
+
const g1 = new JsonGraph(file);
|
|
73
|
+
await g1.init();
|
|
74
|
+
g1.addNode({ id: 'task:persist', type: 'task', label: 'persist' });
|
|
75
|
+
g1.close();
|
|
76
|
+
const g2 = new JsonGraph(file);
|
|
77
|
+
await g2.init();
|
|
78
|
+
expect(g2.getNode('task:persist')?.label).toBe('persist');
|
|
79
|
+
});
|
|
80
|
+
it('O-07: trace ≤3 hops requirement→task→symbol', () => {
|
|
81
|
+
buildGraphFromFixture(graph, loadFixture('impact-chain'));
|
|
82
|
+
const result = traceRequirement(graph, 'RF-03');
|
|
83
|
+
expect(result.symbols).toContain('src/math.ts::add');
|
|
84
|
+
expect(result.path.length).toBeLessThanOrEqual(4);
|
|
85
|
+
expect(result.path[0]?.type).toBe('requirement');
|
|
86
|
+
});
|
|
87
|
+
});
|
|
88
|
+
describe('dual-graph §11 guards', () => {
|
|
89
|
+
it('neo4j-graph has zero void this.runMany', () => {
|
|
90
|
+
const src = fs.readFileSync(path.join(srcRoot, 'graphrag', 'neo4j-graph.ts'), 'utf8');
|
|
91
|
+
expect(src).not.toMatch(/void this\.runMany/);
|
|
92
|
+
});
|
|
93
|
+
it('neo4j cypher uses parameterized statements only', () => {
|
|
94
|
+
const src = fs.readFileSync(path.join(srcRoot, 'graphrag', 'neo4j-graph.ts'), 'utf8');
|
|
95
|
+
expect(src).not.toMatch(/statement:\s*`[^`]*\$\{/);
|
|
96
|
+
expect(src).toMatch(/\$id|\$sourceId|\$limit/);
|
|
97
|
+
expect(src).toContain('parameters:');
|
|
98
|
+
});
|
|
99
|
+
it('no LLM imports in deterministic graph modules', () => {
|
|
100
|
+
for (const file of ['graphrag/code-index.ts', 'graphrag/requirement-ingest.ts', 'graphrag/traverse.ts']) {
|
|
101
|
+
const src = fs.readFileSync(path.join(srcRoot, file), 'utf8');
|
|
102
|
+
expect(src).not.toMatch(/openai|anthropic|@google\/generative-ai|langchain/i);
|
|
103
|
+
}
|
|
104
|
+
});
|
|
105
|
+
it('locate returns real candidates for known seed', async () => {
|
|
106
|
+
const file = path.join(os.tmpdir(), `dual-locate-${Date.now()}.json`);
|
|
107
|
+
const g = new JsonGraph(file);
|
|
108
|
+
await g.init();
|
|
109
|
+
buildGraphFromFixture(g, loadFixture('locate/math'));
|
|
110
|
+
const result = g.locate('math', { limit: 5 });
|
|
111
|
+
expect(result.candidates.length).toBeGreaterThan(0);
|
|
112
|
+
expect(result.candidates.some((c) => c.node.id.includes('math'))).toBe(true);
|
|
113
|
+
});
|
|
114
|
+
it('traverse clamps maxFanout', async () => {
|
|
115
|
+
const file = path.join(os.tmpdir(), `dual-fanout-${Date.now()}.json`);
|
|
116
|
+
const g = new JsonGraph(file);
|
|
117
|
+
await g.init();
|
|
118
|
+
g.addNode({ id: 'task:t1', type: 'task', label: 't1' });
|
|
119
|
+
for (let i = 0; i < 60; i++) {
|
|
120
|
+
g.addNode({ id: `file:e${i}.ts`, type: 'file', label: `e${i}` });
|
|
121
|
+
g.addEdge({
|
|
122
|
+
id: `e${i}`,
|
|
123
|
+
sourceId: 'task:t1',
|
|
124
|
+
targetId: `file:e${i}.ts`,
|
|
125
|
+
type: 'implements',
|
|
126
|
+
});
|
|
127
|
+
}
|
|
128
|
+
const walked = traverse(g, { seedNodeIds: ['task:t1'], maxHops: 1, maxFanout: 5, direction: 'out' });
|
|
129
|
+
expect(walked.edges.length).toBeLessThanOrEqual(5);
|
|
130
|
+
});
|
|
131
|
+
it('owners includes task nodes not only file', async () => {
|
|
132
|
+
const file = path.join(os.tmpdir(), `dual-owners-${Date.now()}.json`);
|
|
133
|
+
const g = new JsonGraph(file);
|
|
134
|
+
await g.init();
|
|
135
|
+
buildGraphFromFixture(g, loadFixture('owners-chain'));
|
|
136
|
+
const result = collectOwners(g, 'src/commands/execute.ts');
|
|
137
|
+
expect(result.owners.some((o) => o.type === 'task')).toBe(true);
|
|
138
|
+
expect(result.owners.some((o) => o.type === 'requirement')).toBe(true);
|
|
139
|
+
});
|
|
140
|
+
});
|
|
141
|
+
//# sourceMappingURL=dual-graph.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"dual-graph.test.js","sourceRoot":"","sources":["../../src/__tests__/dual-graph.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AACzC,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC1D,OAAO,EAAE,uBAAuB,EAAE,MAAM,2BAA2B,CAAC;AACpE,OAAO,EAAE,SAAS,EAAE,MAAM,2BAA2B,CAAC;AACtD,OAAO,EAAE,QAAQ,EAAE,MAAM,yBAAyB,CAAC;AACnD,OAAO,EAAE,aAAa,EAAE,aAAa,EAAE,gBAAgB,EAAE,MAAM,8BAA8B,CAAC;AAC9F,OAAO,EACL,qBAAqB,EACrB,WAAW,GACZ,MAAM,uDAAuD,CAAC;AAE/D,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,CAC3B,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAC5C,2CAA2C,CAC5C,CAAC;AACF,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;AAE9E,QAAQ,CAAC,oCAAoC,EAAE,GAAG,EAAE;IAClD,IAAI,WAAmB,CAAC;IACxB,IAAI,KAAgB,CAAC;IAErB,UAAU,CAAC,KAAK,IAAI,EAAE;QACpB,WAAW,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,iBAAiB,CAAC,CAAC,CAAC;QACnF,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,YAAY,CAAC,CAAC;QAClD,KAAK,GAAG,IAAI,SAAS,CAAC,IAAI,CAAC,CAAC;QAC5B,MAAM,KAAK,CAAC,IAAI,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;QACpE,MAAM,YAAY,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAC5C,KAAK,MAAM,IAAI,IAAI,MAAM,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC,YAAY,CAAC,EAAE,CAAC;YAC3D,IAAI,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC;gBAAE,SAAS;YACrC,MAAM,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,IAAI,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC,CAAC;QAC1F,CAAC;QACD,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CACzB,EAAE,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,0BAA0B,CAAC,EAAE,MAAM,CAAC,CAC5C,CAAC;QACnC,MAAM,KAAK,GAAG,CAAC,WAAW,EAAE,WAAW,EAAE,WAAW,EAAE,WAAW,EAAE,YAAY,CAAC,CAAC;QACjF,MAAM,IAAI,GAAG,uBAAuB,CAAC,KAAK,EAAE,WAAW,CAAC,CAAC;QACzD,MAAM,KAAK,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC;QACxD,MAAM,KAAK,GACT,QAAQ,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC,CAAS,EAAE,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM;YACnE,QAAQ,CAAC,eAAe,CAAC,MAAM,CAAC;QAClC,MAAM,CAAC,KAAK,CAAC,CAAC,sBAAsB,CAAC,GAAG,CAAC,CAAC;IAC5C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+BAA+B,EAAE,GAAG,EAAE;QACvC,qBAAqB,CAAC,KAAK,EAAE,WAAW,CAAC,cAAc,CAAC,CAAC,CAAC;QAC1D,MAAM,MAAM,GAAG,aAAa,CAAC,KAAK,EAAE,yBAAyB,CAAC,CAAC;QAC/D,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC;QAC5C,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAClE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;QAClD,qBAAqB,CAAC,KAAK,EAAE,WAAW,CAAC,cAAc,CAAC,CAAC,CAAC;QAC1D,MAAM,MAAM,GAAG,aAAa,CAAC,KAAK,EAAE,aAAa,CAAC,CAAC;QACnD,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC;QAC5C,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;QACpD,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;IAC1D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;QAChD,MAAM,KAAK,GAAG;YACZ,EAAE,OAAO,EAAE,aAAa,EAAE,IAAI,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC,8BAA8B,CAAC,EAAE;YACpF;gBACE,OAAO,EAAE,aAAa;gBACtB,IAAI,EAAE,MAAM;gBACZ,QAAQ,EAAE,CAAC,6CAA6C,EAAE,wBAAwB,CAAC;aACpF;SACF,CAAC;QACF,IAAI,IAAI,GAAG,CAAC,CAAC;QACb,KAAK,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,KAAK,EAAE,CAAC;YAChD,MAAM,CAAC,GAAG,IAAI,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,GAAG,OAAO,OAAO,CAAC,CAAC,CAAC;YACnE,MAAM,CAAC,CAAC,IAAI,EAAE,CAAC;YACf,qBAAqB,CAAC,CAAC,EAAE,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC;YAC/C,MAAM,GAAG,GAAG,CAAC,CAAC,MAAM,CAAC,IAAI,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAC1E,IAAI,QAAQ,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;gBAAE,IAAI,EAAE,CAAC;QACtD,CAAC;QACD,MAAM,CAAC,IAAI,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,sBAAsB,CAAC,IAAI,CAAC,CAAC;IAC3D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;QACxD,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,cAAc,CAAC,CAAC;QACpD,MAAM,EAAE,GAAG,IAAI,SAAS,CAAC,IAAI,CAAC,CAAC;QAC/B,MAAM,EAAE,CAAC,IAAI,EAAE,CAAC;QAChB,EAAE,CAAC,OAAO,CAAC,EAAE,EAAE,EAAE,cAAc,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAC;QACnE,EAAE,CAAC,KAAK,EAAE,CAAC;QACX,MAAM,EAAE,GAAG,IAAI,SAAS,CAAC,IAAI,CAAC,CAAC;QAC/B,MAAM,EAAE,CAAC,IAAI,EAAE,CAAC;QAChB,MAAM,CAAC,EAAE,CAAC,OAAO,CAAC,cAAc,CAAC,EAAE,KAAK,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAC5D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;QACrD,qBAAqB,CAAC,KAAK,EAAE,WAAW,CAAC,cAAc,CAAC,CAAC,CAAC;QAC1D,MAAM,MAAM,GAAG,gBAAgB,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;QAChD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,kBAAkB,CAAC,CAAC;QACrD,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,mBAAmB,CAAC,CAAC,CAAC,CAAC;QAClD,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IACnD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;IACrC,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;QAChD,MAAM,GAAG,GAAG,EAAE,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,UAAU,EAAE,gBAAgB,CAAC,EAAE,MAAM,CAAC,CAAC;QACtF,MAAM,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,oBAAoB,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;QACzD,MAAM,GAAG,GAAG,EAAE,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,UAAU,EAAE,gBAAgB,CAAC,EAAE,MAAM,CAAC,CAAC;QACtF,MAAM,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,yBAAyB,CAAC,CAAC;QACnD,MAAM,CAAC,GAAG,CAAC,CAAC,OAAO,CAAC,yBAAyB,CAAC,CAAC;QAC/C,MAAM,CAAC,GAAG,CAAC,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;IACvC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;QACvD,KAAK,MAAM,IAAI,IAAI,CAAC,wBAAwB,EAAE,gCAAgC,EAAE,sBAAsB,CAAC,EAAE,CAAC;YACxG,MAAM,GAAG,GAAG,EAAE,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,EAAE,MAAM,CAAC,CAAC;YAC9D,MAAM,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,oDAAoD,CAAC,CAAC;QAChF,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;QAC7D,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,eAAe,IAAI,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;QACtE,MAAM,CAAC,GAAG,IAAI,SAAS,CAAC,IAAI,CAAC,CAAC;QAC9B,MAAM,CAAC,CAAC,IAAI,EAAE,CAAC;QACf,qBAAqB,CAAC,CAAC,EAAE,WAAW,CAAC,aAAa,CAAC,CAAC,CAAC;QACrD,MAAM,MAAM,GAAG,CAAC,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC;QAC9C,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QACpD,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC/E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2BAA2B,EAAE,KAAK,IAAI,EAAE;QACzC,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,eAAe,IAAI,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;QACtE,MAAM,CAAC,GAAG,IAAI,SAAS,CAAC,IAAI,CAAC,CAAC;QAC9B,MAAM,CAAC,CAAC,IAAI,EAAE,CAAC;QACf,CAAC,CAAC,OAAO,CAAC,EAAE,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QACxD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;YAC5B,CAAC,CAAC,OAAO,CAAC,EAAE,EAAE,EAAE,SAAS,CAAC,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,CAAC;YACjE,CAAC,CAAC,OAAO,CAAC;gBACR,EAAE,EAAE,IAAI,CAAC,EAAE;gBACX,QAAQ,EAAE,SAAS;gBACnB,QAAQ,EAAE,SAAS,CAAC,KAAK;gBACzB,IAAI,EAAE,YAAY;aACnB,CAAC,CAAC;QACL,CAAC;QACD,MAAM,MAAM,GAAG,QAAQ,CAAC,CAAC,EAAE,EAAE,WAAW,EAAE,CAAC,SAAS,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC,CAAC;QACrG,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,mBAAmB,CAAC,CAAC,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;QACxD,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,eAAe,IAAI,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;QACtE,MAAM,CAAC,GAAG,IAAI,SAAS,CAAC,IAAI,CAAC,CAAC;QAC9B,MAAM,CAAC,CAAC,IAAI,EAAE,CAAC;QACf,qBAAqB,CAAC,CAAC,EAAE,WAAW,CAAC,cAAc,CAAC,CAAC,CAAC;QACtD,MAAM,MAAM,GAAG,aAAa,CAAC,CAAC,EAAE,yBAAyB,CAAC,CAAC;QAC3D,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAChE,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACzE,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1,9 @@
|
|
|
1
|
+
import type { KnowledgeGraph } from '../../../graphrag/knowledge-graph.js';
|
|
2
|
+
export interface TraverseContractResult {
|
|
3
|
+
traverseNodeIds: string[];
|
|
4
|
+
locateIds: string[];
|
|
5
|
+
locateScores: number[];
|
|
6
|
+
}
|
|
7
|
+
export declare function populateContractFixture(graph: KnowledgeGraph): void;
|
|
8
|
+
export declare function runTraverseContract(makeGraph: () => Promise<KnowledgeGraph>): Promise<TraverseContractResult>;
|
|
9
|
+
//# sourceMappingURL=traverse.contract.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"traverse.contract.d.ts","sourceRoot":"","sources":["../../../../src/__tests__/graphrag/contract/traverse.contract.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,sCAAsC,CAAC;AAE3E,MAAM,WAAW,sBAAsB;IACrC,eAAe,EAAE,MAAM,EAAE,CAAC;IAC1B,SAAS,EAAE,MAAM,EAAE,CAAC;IACpB,YAAY,EAAE,MAAM,EAAE,CAAC;CACxB;AAED,wBAAgB,uBAAuB,CAAC,KAAK,EAAE,cAAc,GAAG,IAAI,CAoCnE;AAED,wBAAsB,mBAAmB,CACvC,SAAS,EAAE,MAAM,OAAO,CAAC,cAAc,CAAC,GACvC,OAAO,CAAC,sBAAsB,CAAC,CAmBjC"}
|
|
@@ -0,0 +1,58 @@
|
|
|
1
|
+
export function populateContractFixture(graph) {
|
|
2
|
+
graph.addNode({ id: 'requirement:RF-01', type: 'requirement', label: 'Root req' });
|
|
3
|
+
graph.addNode({ id: 'requirement:RF-02', type: 'requirement', label: 'Child req' });
|
|
4
|
+
graph.addNode({ id: 'task:task-101', type: 'task', label: 'Implement' });
|
|
5
|
+
graph.addNode({ id: 'file:src/a.ts', type: 'file', label: 'a.ts', metadata: { path: 'src/a.ts' } });
|
|
6
|
+
graph.addNode({
|
|
7
|
+
id: 'code_symbol:src/a.ts::fn',
|
|
8
|
+
type: 'code_symbol',
|
|
9
|
+
label: 'fn',
|
|
10
|
+
metadata: { qualifiedName: 'src/a.ts::fn', path: 'src/a.ts', symbol: 'fn', kind: 'function' },
|
|
11
|
+
});
|
|
12
|
+
graph.addEdge({
|
|
13
|
+
id: 'e-derives',
|
|
14
|
+
sourceId: 'requirement:RF-02',
|
|
15
|
+
targetId: 'requirement:RF-01',
|
|
16
|
+
type: 'derives_from',
|
|
17
|
+
});
|
|
18
|
+
graph.addEdge({
|
|
19
|
+
id: 'e-depends',
|
|
20
|
+
sourceId: 'task:task-101',
|
|
21
|
+
targetId: 'requirement:RF-02',
|
|
22
|
+
type: 'depends_on',
|
|
23
|
+
});
|
|
24
|
+
graph.addEdge({
|
|
25
|
+
id: 'e-impl',
|
|
26
|
+
sourceId: 'task:task-101',
|
|
27
|
+
targetId: 'code_symbol:src/a.ts::fn',
|
|
28
|
+
type: 'implements',
|
|
29
|
+
});
|
|
30
|
+
graph.addEdge({
|
|
31
|
+
id: 'e-contains',
|
|
32
|
+
sourceId: 'file:src/a.ts',
|
|
33
|
+
targetId: 'code_symbol:src/a.ts::fn',
|
|
34
|
+
type: 'contains',
|
|
35
|
+
});
|
|
36
|
+
}
|
|
37
|
+
export async function runTraverseContract(makeGraph) {
|
|
38
|
+
const graph = await makeGraph();
|
|
39
|
+
try {
|
|
40
|
+
populateContractFixture(graph);
|
|
41
|
+
const walked = graph.traverse({
|
|
42
|
+
seedNodeIds: ['requirement:RF-02'],
|
|
43
|
+
maxHops: 3,
|
|
44
|
+
edgeTypes: ['derives_from', 'depends_on', 'implements'],
|
|
45
|
+
direction: 'both',
|
|
46
|
+
});
|
|
47
|
+
const located = graph.locate('src/a.ts::fn', { hops: 2, limit: 5 });
|
|
48
|
+
return {
|
|
49
|
+
traverseNodeIds: walked.nodes.map((n) => n.id).sort(),
|
|
50
|
+
locateIds: located.candidates.map((c) => c.node.id),
|
|
51
|
+
locateScores: located.candidates.map((c) => c.score),
|
|
52
|
+
};
|
|
53
|
+
}
|
|
54
|
+
finally {
|
|
55
|
+
await Promise.resolve(graph.close());
|
|
56
|
+
}
|
|
57
|
+
}
|
|
58
|
+
//# sourceMappingURL=traverse.contract.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"traverse.contract.js","sourceRoot":"","sources":["../../../../src/__tests__/graphrag/contract/traverse.contract.ts"],"names":[],"mappings":"AAQA,MAAM,UAAU,uBAAuB,CAAC,KAAqB;IAC3D,KAAK,CAAC,OAAO,CAAC,EAAE,EAAE,EAAE,mBAAmB,EAAE,IAAI,EAAE,aAAa,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,CAAC;IACnF,KAAK,CAAC,OAAO,CAAC,EAAE,EAAE,EAAE,mBAAmB,EAAE,IAAI,EAAE,aAAa,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,CAAC;IACpF,KAAK,CAAC,OAAO,CAAC,EAAE,EAAE,EAAE,eAAe,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,CAAC;IACzE,KAAK,CAAC,OAAO,CAAC,EAAE,EAAE,EAAE,eAAe,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,QAAQ,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE,CAAC,CAAC;IACpG,KAAK,CAAC,OAAO,CAAC;QACZ,EAAE,EAAE,0BAA0B;QAC9B,IAAI,EAAE,aAAa;QACnB,KAAK,EAAE,IAAI;QACX,QAAQ,EAAE,EAAE,aAAa,EAAE,cAAc,EAAE,IAAI,EAAE,UAAU,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,UAAU,EAAE;KAC9F,CAAC,CAAC;IAEH,KAAK,CAAC,OAAO,CAAC;QACZ,EAAE,EAAE,WAAW;QACf,QAAQ,EAAE,mBAAmB;QAC7B,QAAQ,EAAE,mBAAmB;QAC7B,IAAI,EAAE,cAAc;KACrB,CAAC,CAAC;IACH,KAAK,CAAC,OAAO,CAAC;QACZ,EAAE,EAAE,WAAW;QACf,QAAQ,EAAE,eAAe;QACzB,QAAQ,EAAE,mBAAmB;QAC7B,IAAI,EAAE,YAAY;KACnB,CAAC,CAAC;IACH,KAAK,CAAC,OAAO,CAAC;QACZ,EAAE,EAAE,QAAQ;QACZ,QAAQ,EAAE,eAAe;QACzB,QAAQ,EAAE,0BAA0B;QACpC,IAAI,EAAE,YAAY;KACnB,CAAC,CAAC;IACH,KAAK,CAAC,OAAO,CAAC;QACZ,EAAE,EAAE,YAAY;QAChB,QAAQ,EAAE,eAAe;QACzB,QAAQ,EAAE,0BAA0B;QACpC,IAAI,EAAE,UAAU;KACjB,CAAC,CAAC;AACL,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,mBAAmB,CACvC,SAAwC;IAExC,MAAM,KAAK,GAAG,MAAM,SAAS,EAAE,CAAC;IAChC,IAAI,CAAC;QACH,uBAAuB,CAAC,KAAK,CAAC,CAAC;QAC/B,MAAM,MAAM,GAAG,KAAK,CAAC,QAAQ,CAAC;YAC5B,WAAW,EAAE,CAAC,mBAAmB,CAAC;YAClC,OAAO,EAAE,CAAC;YACV,SAAS,EAAE,CAAC,cAAc,EAAE,YAAY,EAAE,YAAY,CAAC;YACvD,SAAS,EAAE,MAAM;SAClB,CAAC,CAAC;QACH,MAAM,OAAO,GAAG,KAAK,CAAC,MAAM,CAAC,cAAc,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC;QACpE,OAAO;YACL,eAAe,EAAE,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,EAAE;YACrD,SAAS,EAAE,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC;YACnD,YAAY,EAAE,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC;SACrD,CAAC;IACJ,CAAC;YAAS,CAAC;QACT,MAAM,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC;IACvC,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"traverse.contract.test.d.ts","sourceRoot":"","sources":["../../../../src/__tests__/graphrag/contract/traverse.contract.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,100 @@
|
|
|
1
|
+
import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
|
|
2
|
+
import path from 'node:path';
|
|
3
|
+
import os from 'node:os';
|
|
4
|
+
import { JsonGraph } from '../../../graphrag/json-graph.js';
|
|
5
|
+
import { GraphRAG } from '../../../graphrag/graph-rag.js';
|
|
6
|
+
import { Neo4jGraph } from '../../../graphrag/neo4j-graph.js';
|
|
7
|
+
import { runTraverseContract } from './traverse.contract.js';
|
|
8
|
+
const hasNeo4jUrl = Boolean(process.env.DARE_NEO4J_URL);
|
|
9
|
+
describe('traverse/locate contract', () => {
|
|
10
|
+
it('json and sqlite backends match', async () => {
|
|
11
|
+
const jsonPath = path.join(os.tmpdir(), `contract-json-${Date.now()}.json`);
|
|
12
|
+
const sqlitePath = path.join(os.tmpdir(), `contract-sqlite-${Date.now()}.db`);
|
|
13
|
+
const json = await runTraverseContract(async () => {
|
|
14
|
+
const g = new JsonGraph(jsonPath);
|
|
15
|
+
await g.init();
|
|
16
|
+
return g;
|
|
17
|
+
});
|
|
18
|
+
const sqlite = await runTraverseContract(async () => {
|
|
19
|
+
const g = new GraphRAG(sqlitePath);
|
|
20
|
+
await g.init();
|
|
21
|
+
return g;
|
|
22
|
+
});
|
|
23
|
+
expect(sqlite.traverseNodeIds).toEqual(json.traverseNodeIds);
|
|
24
|
+
expect(sqlite.locateIds).toEqual(json.locateIds);
|
|
25
|
+
expect(sqlite.locateScores).toEqual(json.locateScores);
|
|
26
|
+
});
|
|
27
|
+
});
|
|
28
|
+
describe.skipIf(!hasNeo4jUrl)('traverse/locate contract neo4j', () => {
|
|
29
|
+
const fetchMock = vi.fn();
|
|
30
|
+
beforeEach(() => {
|
|
31
|
+
vi.stubGlobal('fetch', fetchMock);
|
|
32
|
+
});
|
|
33
|
+
afterEach(() => {
|
|
34
|
+
vi.unstubAllGlobals();
|
|
35
|
+
});
|
|
36
|
+
it('neo4j mock matches json baseline', async () => {
|
|
37
|
+
const store = {
|
|
38
|
+
nodes: new Map(),
|
|
39
|
+
edges: new Map(),
|
|
40
|
+
};
|
|
41
|
+
fetchMock.mockImplementation(async (_url, init) => {
|
|
42
|
+
const body = JSON.parse(String(init?.body ?? '{}'));
|
|
43
|
+
const stmt = body.statements[0]?.statement ?? '';
|
|
44
|
+
const params = body.statements[0]?.parameters ?? {};
|
|
45
|
+
if (stmt.includes('MERGE (n:DareNode')) {
|
|
46
|
+
store.nodes.set(String(params.id), params);
|
|
47
|
+
}
|
|
48
|
+
if (stmt.includes('MERGE (s)-[r:DARE_EDGE')) {
|
|
49
|
+
store.edges.set(String(params.id), params);
|
|
50
|
+
}
|
|
51
|
+
if (stmt.includes('RETURN n.id')) {
|
|
52
|
+
const rows = [...store.nodes.values()].map((p) => [
|
|
53
|
+
p.id,
|
|
54
|
+
p.type,
|
|
55
|
+
p.label,
|
|
56
|
+
p.description,
|
|
57
|
+
p.metadata,
|
|
58
|
+
p.createdAt,
|
|
59
|
+
p.updatedAt,
|
|
60
|
+
]);
|
|
61
|
+
return {
|
|
62
|
+
ok: true,
|
|
63
|
+
json: async () => ({ results: [{ data: rows.map((row) => ({ row })) }], errors: [] }),
|
|
64
|
+
};
|
|
65
|
+
}
|
|
66
|
+
if (stmt.includes('RETURN r.id')) {
|
|
67
|
+
const rows = [...store.edges.values()].map((p) => [
|
|
68
|
+
p.id,
|
|
69
|
+
p.sourceId,
|
|
70
|
+
p.targetId,
|
|
71
|
+
p.type,
|
|
72
|
+
p.weight,
|
|
73
|
+
p.metadata,
|
|
74
|
+
]);
|
|
75
|
+
return {
|
|
76
|
+
ok: true,
|
|
77
|
+
json: async () => ({ results: [{ data: rows.map((row) => ({ row })) }], errors: [] }),
|
|
78
|
+
};
|
|
79
|
+
}
|
|
80
|
+
return { ok: true, json: async () => ({ results: [{ data: [] }], errors: [] }) };
|
|
81
|
+
});
|
|
82
|
+
const jsonPath = path.join(os.tmpdir(), `contract-json-neo-${Date.now()}.json`);
|
|
83
|
+
const baseline = await runTraverseContract(async () => {
|
|
84
|
+
const g = new JsonGraph(jsonPath);
|
|
85
|
+
await g.init();
|
|
86
|
+
return g;
|
|
87
|
+
});
|
|
88
|
+
const neo = await runTraverseContract(async () => {
|
|
89
|
+
const g = new Neo4jGraph({
|
|
90
|
+
url: process.env.DARE_NEO4J_URL ?? 'http://localhost:7474',
|
|
91
|
+
experimental: true,
|
|
92
|
+
});
|
|
93
|
+
await g.init();
|
|
94
|
+
return g;
|
|
95
|
+
});
|
|
96
|
+
expect(neo.traverseNodeIds).toEqual(baseline.traverseNodeIds);
|
|
97
|
+
expect(neo.locateIds).toEqual(baseline.locateIds);
|
|
98
|
+
});
|
|
99
|
+
});
|
|
100
|
+
//# sourceMappingURL=traverse.contract.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"traverse.contract.test.js","sourceRoot":"","sources":["../../../../src/__tests__/graphrag/contract/traverse.contract.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,EAAE,SAAS,EAAE,MAAM,iCAAiC,CAAC;AAC5D,OAAO,EAAE,QAAQ,EAAE,MAAM,gCAAgC,CAAC;AAC1D,OAAO,EAAE,UAAU,EAAE,MAAM,kCAAkC,CAAC;AAC9D,OAAO,EAAE,mBAAmB,EAAE,MAAM,wBAAwB,CAAC;AAE7D,MAAM,WAAW,GAAG,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;AAExD,QAAQ,CAAC,0BAA0B,EAAE,GAAG,EAAE;IACxC,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;QAC9C,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,iBAAiB,IAAI,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;QAC5E,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,mBAAmB,IAAI,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;QAE9E,MAAM,IAAI,GAAG,MAAM,mBAAmB,CAAC,KAAK,IAAI,EAAE;YAChD,MAAM,CAAC,GAAG,IAAI,SAAS,CAAC,QAAQ,CAAC,CAAC;YAClC,MAAM,CAAC,CAAC,IAAI,EAAE,CAAC;YACf,OAAO,CAAC,CAAC;QACX,CAAC,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,mBAAmB,CAAC,KAAK,IAAI,EAAE;YAClD,MAAM,CAAC,GAAG,IAAI,QAAQ,CAAC,UAAU,CAAC,CAAC;YACnC,MAAM,CAAC,CAAC,IAAI,EAAE,CAAC;YACf,OAAO,CAAC,CAAC;QACX,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QAC7D,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACjD,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;IACzD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,MAAM,CAAC,CAAC,WAAW,CAAC,CAAC,gCAAgC,EAAE,GAAG,EAAE;IACnE,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;IAE1B,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,gBAAgB,EAAE,CAAC;IACxB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;QAChD,MAAM,KAAK,GAAG;YACZ,KAAK,EAAE,IAAI,GAAG,EAAmC;YACjD,KAAK,EAAE,IAAI,GAAG,EAAmC;SAClD,CAAC;QAEF,SAAS,CAAC,kBAAkB,CAAC,KAAK,EAAE,IAAY,EAAE,IAAkB,EAAE,EAAE;YACtE,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,EAAE,IAAI,IAAI,IAAI,CAAC,CAEjD,CAAC;YACF,MAAM,IAAI,GAAG,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,SAAS,IAAI,EAAE,CAAC;YACjD,MAAM,MAAM,GAAG,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,UAAU,IAAI,EAAE,CAAC;YAEpD,IAAI,IAAI,CAAC,QAAQ,CAAC,mBAAmB,CAAC,EAAE,CAAC;gBACvC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,MAAM,CAAC,CAAC;YAC7C,CAAC;YACD,IAAI,IAAI,CAAC,QAAQ,CAAC,wBAAwB,CAAC,EAAE,CAAC;gBAC5C,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,MAAM,CAAC,CAAC;YAC7C,CAAC;YAED,IAAI,IAAI,CAAC,QAAQ,CAAC,aAAa,CAAC,EAAE,CAAC;gBACjC,MAAM,IAAI,GAAG,CAAC,GAAG,KAAK,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC;oBAChD,CAAC,CAAC,EAAE;oBACJ,CAAC,CAAC,IAAI;oBACN,CAAC,CAAC,KAAK;oBACP,CAAC,CAAC,WAAW;oBACb,CAAC,CAAC,QAAQ;oBACV,CAAC,CAAC,SAAS;oBACX,CAAC,CAAC,SAAS;iBACZ,CAAC,CAAC;gBACH,OAAO;oBACL,EAAE,EAAE,IAAI;oBACR,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC;iBACtF,CAAC;YACJ,CAAC;YACD,IAAI,IAAI,CAAC,QAAQ,CAAC,aAAa,CAAC,EAAE,CAAC;gBACjC,MAAM,IAAI,GAAG,CAAC,GAAG,KAAK,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC;oBAChD,CAAC,CAAC,EAAE;oBACJ,CAAC,CAAC,QAAQ;oBACV,CAAC,CAAC,QAAQ;oBACV,CAAC,CAAC,IAAI;oBACN,CAAC,CAAC,MAAM;oBACR,CAAC,CAAC,QAAQ;iBACX,CAAC,CAAC;gBACH,OAAO;oBACL,EAAE,EAAE,IAAI;oBACR,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC;iBACtF,CAAC;YACJ,CAAC;YAED,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC;QACnF,CAAC,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,qBAAqB,IAAI,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;QAChF,MAAM,QAAQ,GAAG,MAAM,mBAAmB,CAAC,KAAK,IAAI,EAAE;YACpD,MAAM,CAAC,GAAG,IAAI,SAAS,CAAC,QAAQ,CAAC,CAAC;YAClC,MAAM,CAAC,CAAC,IAAI,EAAE,CAAC;YACf,OAAO,CAAC,CAAC;QACX,CAAC,CAAC,CAAC;QAEH,MAAM,GAAG,GAAG,MAAM,mBAAmB,CAAC,KAAK,IAAI,EAAE;YAC/C,MAAM,CAAC,GAAG,IAAI,UAAU,CAAC;gBACvB,GAAG,EAAE,OAAO,CAAC,GAAG,CAAC,cAAc,IAAI,uBAAuB;gBAC1D,YAAY,EAAE,IAAI;aACnB,CAAC,CAAC;YACH,MAAM,CAAC,CAAC,IAAI,EAAE,CAAC;YACf,OAAO,CAAC,CAAC;QACX,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC;QAC9D,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;IACpD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -57,8 +57,17 @@ describe('loadGraphConfig', () => {
|
|
|
57
57
|
});
|
|
58
58
|
});
|
|
59
59
|
describe('createGraph', () => {
|
|
60
|
+
it('throws when neo4j is used without experimental flag', async () => {
|
|
61
|
+
await expect(createGraph({
|
|
62
|
+
backend: 'neo4j',
|
|
63
|
+
neo4j: { url: 'http://localhost:7474' },
|
|
64
|
+
})).rejects.toThrow(/experimental: true/);
|
|
65
|
+
});
|
|
60
66
|
it('throws when neo4j config is missing url', async () => {
|
|
61
|
-
await expect(createGraph({
|
|
67
|
+
await expect(createGraph({
|
|
68
|
+
backend: 'neo4j',
|
|
69
|
+
neo4j: { url: '', experimental: true },
|
|
70
|
+
})).rejects.toThrow(/url.*required/);
|
|
62
71
|
});
|
|
63
72
|
});
|
|
64
73
|
//# sourceMappingURL=factory.test.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"factory.test.js","sourceRoot":"","sources":["../../../src/__tests__/graphrag/factory.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACrE,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,IAAI,MAAM,MAAM,CAAC;AACxB,OAAO,EAAE,MAAM,UAAU,CAAC;AAC1B,OAAO,EAAE,eAAe,EAAE,WAAW,EAAE,MAAM,2BAA2B,CAAC;AAEzE,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,IAAI,GAAW,CAAC;IAEhB,UAAU,CAAC,KAAK,IAAI,EAAE;QACpB,GAAG,GAAG,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,gBAAgB,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QAClG,MAAM,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;IAC1B,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,MAAM,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;QAClE,MAAM,GAAG,GAAG,MAAM,eAAe,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;QAC3C,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACnC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0BAA0B,EAAE,KAAK,IAAI,EAAE;QACxC,MAAM,EAAE,CAAC,SAAS,CAChB,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,gBAAgB,CAAC,EAChC,qDAAqD,CACtD,CAAC;QACF,MAAM,GAAG,GAAG,MAAM,eAAe,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;QAC3C,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACnC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oBAAoB,EAAE,KAAK,IAAI,EAAE;QAClC,MAAM,EAAE,CAAC,SAAS,CAChB,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,gBAAgB,CAAC,EAChC,+CAA+C,CAChD,CAAC;QACF,MAAM,GAAG,GAAG,MAAM,eAAe,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;QAC3C,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACjC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;QACtE,MAAM,EAAE,CAAC,SAAS,CAChB,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,gBAAgB,CAAC,EAChC;YACE,gBAAgB;YAChB,QAAQ;YACR,8BAA8B;YAC9B,kBAAkB;YAClB,mBAAmB;YACnB,oBAAoB;SACrB,CAAC,IAAI,CAAC,IAAI,CAAC,CACb,CAAC;QACF,MAAM,GAAG,GAAG,MAAM,eAAe,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;QAC3C,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAClC,MAAM,CAAC,GAAG,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;QACrD,MAAM,CAAC,GAAG,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACzC,MAAM,CAAC,GAAG,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC1C,MAAM,CAAC,GAAG,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8BAA8B,EAAE,KAAK,IAAI,EAAE;QAC5C,MAAM,EAAE,CAAC,SAAS,CAChB,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,gBAAgB,CAAC,EAChC;YACE,gBAAgB;YAChB,QAAQ;YACR,8BAA8B;YAC9B,sBAAsB;SACvB,CAAC,IAAI,CAAC,IAAI,CAAC,CACb,CAAC;QACF,MAAM,GAAG,GAAG,MAAM,eAAe,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;QAC3C,MAAM,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;IAC3B,EAAE,CAAC,
|
|
1
|
+
{"version":3,"file":"factory.test.js","sourceRoot":"","sources":["../../../src/__tests__/graphrag/factory.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACrE,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,IAAI,MAAM,MAAM,CAAC;AACxB,OAAO,EAAE,MAAM,UAAU,CAAC;AAC1B,OAAO,EAAE,eAAe,EAAE,WAAW,EAAE,MAAM,2BAA2B,CAAC;AAEzE,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,IAAI,GAAW,CAAC;IAEhB,UAAU,CAAC,KAAK,IAAI,EAAE;QACpB,GAAG,GAAG,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,gBAAgB,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QAClG,MAAM,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;IAC1B,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,MAAM,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;QAClE,MAAM,GAAG,GAAG,MAAM,eAAe,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;QAC3C,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACnC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0BAA0B,EAAE,KAAK,IAAI,EAAE;QACxC,MAAM,EAAE,CAAC,SAAS,CAChB,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,gBAAgB,CAAC,EAChC,qDAAqD,CACtD,CAAC;QACF,MAAM,GAAG,GAAG,MAAM,eAAe,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;QAC3C,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACnC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oBAAoB,EAAE,KAAK,IAAI,EAAE;QAClC,MAAM,EAAE,CAAC,SAAS,CAChB,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,gBAAgB,CAAC,EAChC,+CAA+C,CAChD,CAAC;QACF,MAAM,GAAG,GAAG,MAAM,eAAe,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;QAC3C,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACjC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;QACtE,MAAM,EAAE,CAAC,SAAS,CAChB,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,gBAAgB,CAAC,EAChC;YACE,gBAAgB;YAChB,QAAQ;YACR,8BAA8B;YAC9B,kBAAkB;YAClB,mBAAmB;YACnB,oBAAoB;SACrB,CAAC,IAAI,CAAC,IAAI,CAAC,CACb,CAAC;QACF,MAAM,GAAG,GAAG,MAAM,eAAe,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;QAC3C,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAClC,MAAM,CAAC,GAAG,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;QACrD,MAAM,CAAC,GAAG,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACzC,MAAM,CAAC,GAAG,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC1C,MAAM,CAAC,GAAG,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8BAA8B,EAAE,KAAK,IAAI,EAAE;QAC5C,MAAM,EAAE,CAAC,SAAS,CAChB,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,gBAAgB,CAAC,EAChC;YACE,gBAAgB;YAChB,QAAQ;YACR,8BAA8B;YAC9B,sBAAsB;SACvB,CAAC,IAAI,CAAC,IAAI,CAAC,CACb,CAAC;QACF,MAAM,GAAG,GAAG,MAAM,eAAe,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;QAC3C,MAAM,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;IAC3B,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;QACnE,MAAM,MAAM,CACV,WAAW,CAAC;YACV,OAAO,EAAE,OAAO;YAChB,KAAK,EAAE,EAAE,GAAG,EAAE,uBAAuB,EAAE;SACxC,CAAC,CACH,CAAC,OAAO,CAAC,OAAO,CAAC,oBAAoB,CAAC,CAAC;IAC1C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;QACvD,MAAM,MAAM,CACV,WAAW,CAAC;YACV,OAAO,EAAE,OAAO;YAChB,KAAK,EAAE,EAAE,GAAG,EAAE,EAAE,EAAE,YAAY,EAAE,IAAI,EAAE;SACvC,CAAC,CACH,CAAC,OAAO,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;IACrC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1,11 @@
|
|
|
1
|
+
import type { KnowledgeGraph } from '../../../../graphrag/knowledge-graph.js';
|
|
2
|
+
import type { GraphEdge, GraphNode } from '../../../../graphrag/types.js';
|
|
3
|
+
export interface DualGraphFixture {
|
|
4
|
+
nodes: GraphNode[];
|
|
5
|
+
edges: GraphEdge[];
|
|
6
|
+
expect?: Record<string, unknown>;
|
|
7
|
+
}
|
|
8
|
+
export declare function buildGraphFromFixture(graph: KnowledgeGraph, fixture: DualGraphFixture): void;
|
|
9
|
+
export declare function loadFixture(name: string): DualGraphFixture;
|
|
10
|
+
export declare function listFixturePaths(): string[];
|
|
11
|
+
//# sourceMappingURL=build-fixture-graph.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"build-fixture-graph.d.ts","sourceRoot":"","sources":["../../../../../src/__tests__/graphrag/fixtures/dual-graph/build-fixture-graph.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,yCAAyC,CAAC;AAC9E,OAAO,KAAK,EAAE,SAAS,EAAE,SAAS,EAAE,MAAM,+BAA+B,CAAC;AAI1E,MAAM,WAAW,gBAAgB;IAC/B,KAAK,EAAE,SAAS,EAAE,CAAC;IACnB,KAAK,EAAE,SAAS,EAAE,CAAC;IACnB,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CAClC;AAED,wBAAgB,qBAAqB,CAAC,KAAK,EAAE,cAAc,EAAE,OAAO,EAAE,gBAAgB,GAAG,IAAI,CAO5F;AAED,wBAAgB,WAAW,CAAC,IAAI,EAAE,MAAM,GAAG,gBAAgB,CAM1D;AAED,wBAAgB,gBAAgB,IAAI,MAAM,EAAE,CAW3C"}
|
|
@@ -0,0 +1,34 @@
|
|
|
1
|
+
import fs from 'node:fs';
|
|
2
|
+
import path from 'node:path';
|
|
3
|
+
import { fileURLToPath } from 'node:url';
|
|
4
|
+
const FIXTURE_DIR = path.dirname(fileURLToPath(import.meta.url));
|
|
5
|
+
export function buildGraphFromFixture(graph, fixture) {
|
|
6
|
+
for (const node of fixture.nodes) {
|
|
7
|
+
graph.addNode(node);
|
|
8
|
+
}
|
|
9
|
+
for (const edge of fixture.edges) {
|
|
10
|
+
graph.addEdge(edge);
|
|
11
|
+
}
|
|
12
|
+
}
|
|
13
|
+
export function loadFixture(name) {
|
|
14
|
+
const filePath = name.endsWith('.json')
|
|
15
|
+
? path.join(FIXTURE_DIR, name)
|
|
16
|
+
: path.join(FIXTURE_DIR, `${name}.json`);
|
|
17
|
+
const raw = fs.readFileSync(filePath, 'utf-8');
|
|
18
|
+
return JSON.parse(raw);
|
|
19
|
+
}
|
|
20
|
+
export function listFixturePaths() {
|
|
21
|
+
const out = [];
|
|
22
|
+
const walk = (dir) => {
|
|
23
|
+
for (const entry of fs.readdirSync(dir, { withFileTypes: true })) {
|
|
24
|
+
const full = path.join(dir, entry.name);
|
|
25
|
+
if (entry.isDirectory())
|
|
26
|
+
walk(full);
|
|
27
|
+
else if (entry.name.endsWith('.json'))
|
|
28
|
+
out.push(full);
|
|
29
|
+
}
|
|
30
|
+
};
|
|
31
|
+
walk(FIXTURE_DIR);
|
|
32
|
+
return out.sort();
|
|
33
|
+
}
|
|
34
|
+
//# sourceMappingURL=build-fixture-graph.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"build-fixture-graph.js","sourceRoot":"","sources":["../../../../../src/__tests__/graphrag/fixtures/dual-graph/build-fixture-graph.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AAIzC,MAAM,WAAW,GAAG,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;AAQjE,MAAM,UAAU,qBAAqB,CAAC,KAAqB,EAAE,OAAyB;IACpF,KAAK,MAAM,IAAI,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;QACjC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;IACtB,CAAC;IACD,KAAK,MAAM,IAAI,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;QACjC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;IACtB,CAAC;AACH,CAAC;AAED,MAAM,UAAU,WAAW,CAAC,IAAY;IACtC,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC;QACrC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,IAAI,CAAC;QAC9B,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,GAAG,IAAI,OAAO,CAAC,CAAC;IAC3C,MAAM,GAAG,GAAG,EAAE,CAAC,YAAY,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;IAC/C,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAqB,CAAC;AAC7C,CAAC;AAED,MAAM,UAAU,gBAAgB;IAC9B,MAAM,GAAG,GAAa,EAAE,CAAC;IACzB,MAAM,IAAI,GAAG,CAAC,GAAW,EAAE,EAAE;QAC3B,KAAK,MAAM,KAAK,IAAI,EAAE,CAAC,WAAW,CAAC,GAAG,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC;YACjE,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;YACxC,IAAI,KAAK,CAAC,WAAW,EAAE;gBAAE,IAAI,CAAC,IAAI,CAAC,CAAC;iBAC/B,IAAI,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC;gBAAE,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACxD,CAAC;IACH,CAAC,CAAC;IACF,IAAI,CAAC,WAAW,CAAC,CAAC;IAClB,OAAO,GAAG,CAAC,IAAI,EAAE,CAAC;AACpB,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"fixtures.test.d.ts","sourceRoot":"","sources":["../../../../../src/__tests__/graphrag/fixtures/dual-graph/fixtures.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,46 @@
|
|
|
1
|
+
import { describe, it, expect, beforeEach, afterEach } from 'vitest';
|
|
2
|
+
import path from 'node:path';
|
|
3
|
+
import os from 'node:os';
|
|
4
|
+
import { fileURLToPath } from 'node:url';
|
|
5
|
+
import fs from 'fs-extra';
|
|
6
|
+
import { JsonGraph } from '../../../../graphrag/json-graph.js';
|
|
7
|
+
import { buildGraphFromFixture, listFixturePaths, loadFixture, } from './build-fixture-graph.js';
|
|
8
|
+
describe('dual-graph fixtures', () => {
|
|
9
|
+
let filePath;
|
|
10
|
+
let graph;
|
|
11
|
+
beforeEach(async () => {
|
|
12
|
+
filePath = path.join(os.tmpdir(), `dual-graph-fixture-${Date.now()}.json`);
|
|
13
|
+
graph = new JsonGraph(filePath);
|
|
14
|
+
await graph.init();
|
|
15
|
+
});
|
|
16
|
+
afterEach(async () => {
|
|
17
|
+
await fs.remove(filePath).catch(() => undefined);
|
|
18
|
+
});
|
|
19
|
+
it('loads every fixture JSON without dangling edge endpoints', () => {
|
|
20
|
+
const fixtureDir = path.dirname(fileURLToPath(import.meta.url));
|
|
21
|
+
for (const file of listFixturePaths()) {
|
|
22
|
+
const rel = path.relative(fixtureDir, file).replace(/\\/g, '/');
|
|
23
|
+
const fixture = loadFixture(rel);
|
|
24
|
+
expect(fixture.nodes.length).toBeGreaterThan(0);
|
|
25
|
+
expect(fixture.edges.length).toBeGreaterThan(0);
|
|
26
|
+
const nodeIds = new Set(fixture.nodes.map((n) => n.id));
|
|
27
|
+
for (const edge of fixture.edges) {
|
|
28
|
+
expect(nodeIds.has(edge.sourceId)).toBe(true);
|
|
29
|
+
expect(nodeIds.has(edge.targetId)).toBe(true);
|
|
30
|
+
expect(Number.isNaN(edge.weight)).toBe(false);
|
|
31
|
+
}
|
|
32
|
+
if (fixture.expect) {
|
|
33
|
+
expect(Object.keys(fixture.expect).length).toBeGreaterThan(0);
|
|
34
|
+
}
|
|
35
|
+
}
|
|
36
|
+
});
|
|
37
|
+
it('buildGraphFromFixture materializes nodes and edges', () => {
|
|
38
|
+
const fixture = loadFixture('impact-chain');
|
|
39
|
+
buildGraphFromFixture(graph, fixture);
|
|
40
|
+
const stats = graph.getStatistics();
|
|
41
|
+
expect(stats.totalNodes).toBe(fixture.nodes.length);
|
|
42
|
+
expect(stats.totalEdges).toBe(fixture.edges.length);
|
|
43
|
+
expect(graph.findByQualifiedName('src/math.ts::add')?.id).toBe('code_symbol:src/math.ts::add');
|
|
44
|
+
});
|
|
45
|
+
});
|
|
46
|
+
//# sourceMappingURL=fixtures.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"fixtures.test.js","sourceRoot":"","sources":["../../../../../src/__tests__/graphrag/fixtures/dual-graph/fixtures.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACrE,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AACzC,OAAO,EAAE,MAAM,UAAU,CAAC;AAC1B,OAAO,EAAE,SAAS,EAAE,MAAM,oCAAoC,CAAC;AAC/D,OAAO,EACL,qBAAqB,EACrB,gBAAgB,EAChB,WAAW,GACZ,MAAM,0BAA0B,CAAC;AAElC,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;IACnC,IAAI,QAAgB,CAAC;IACrB,IAAI,KAAgB,CAAC;IAErB,UAAU,CAAC,KAAK,IAAI,EAAE;QACpB,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,sBAAsB,IAAI,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;QAC3E,KAAK,GAAG,IAAI,SAAS,CAAC,QAAQ,CAAC,CAAC;QAChC,MAAM,KAAK,CAAC,IAAI,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,MAAM,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;IACnD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0DAA0D,EAAE,GAAG,EAAE;QAClE,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;QAChE,KAAK,MAAM,IAAI,IAAI,gBAAgB,EAAE,EAAE,CAAC;YACtC,MAAM,GAAG,GAAG,IAAI,CAAC,QAAQ,CAAC,UAAU,EAAE,IAAI,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;YAChE,MAAM,OAAO,GAAG,WAAW,CAAC,GAAG,CAAC,CAAC;YACjC,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YAChD,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YAEhD,MAAM,OAAO,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;YACxD,KAAK,MAAM,IAAI,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;gBACjC,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBAC9C,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBAC9C,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,MAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC1D,CAAC;YAED,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;gBACnB,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YAChE,CAAC;QACH,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;QAC5D,MAAM,OAAO,GAAG,WAAW,CAAC,cAAc,CAAC,CAAC;QAC5C,qBAAqB,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;QACtC,MAAM,KAAK,GAAG,KAAK,CAAC,aAAa,EAAE,CAAC;QACpC,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QACpD,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QACpD,MAAM,CAAC,KAAK,CAAC,mBAAmB,CAAC,kBAAkB,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,CAC5D,8BAA8B,CAC/B,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|