@dewtech/dare-cli 3.9.0 → 3.11.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/README.md +5 -0
- package/dist/__tests__/ci-pr-regression.test.d.ts +2 -0
- package/dist/__tests__/ci-pr-regression.test.d.ts.map +1 -0
- package/dist/__tests__/ci-pr-regression.test.js +76 -0
- package/dist/__tests__/ci-pr-regression.test.js.map +1 -0
- package/dist/__tests__/dashboard-front.test.d.ts +2 -0
- package/dist/__tests__/dashboard-front.test.d.ts.map +1 -0
- package/dist/__tests__/dashboard-front.test.js +37 -0
- package/dist/__tests__/dashboard-front.test.js.map +1 -0
- package/dist/__tests__/dashboard-regression.test.d.ts +2 -0
- package/dist/__tests__/dashboard-regression.test.d.ts.map +1 -0
- package/dist/__tests__/dashboard-regression.test.js +68 -0
- package/dist/__tests__/dashboard-regression.test.js.map +1 -0
- package/dist/__tests__/drift-regression.test.d.ts +2 -0
- package/dist/__tests__/drift-regression.test.d.ts.map +1 -0
- package/dist/__tests__/drift-regression.test.js +118 -0
- package/dist/__tests__/drift-regression.test.js.map +1 -0
- package/dist/__tests__/dynamic-dag-regression.test.d.ts +2 -0
- package/dist/__tests__/dynamic-dag-regression.test.d.ts.map +1 -0
- package/dist/__tests__/dynamic-dag-regression.test.js +464 -0
- package/dist/__tests__/dynamic-dag-regression.test.js.map +1 -0
- package/dist/__tests__/ide-command-parity.test.js +1 -0
- package/dist/__tests__/ide-command-parity.test.js.map +1 -1
- package/dist/__tests__/project-generator.test.js +10 -1
- package/dist/__tests__/project-generator.test.js.map +1 -1
- package/dist/__tests__/semantic-regression.test.d.ts +2 -0
- package/dist/__tests__/semantic-regression.test.d.ts.map +1 -0
- package/dist/__tests__/semantic-regression.test.js +248 -0
- package/dist/__tests__/semantic-regression.test.js.map +1 -0
- package/dist/bin/dare.js +2 -0
- package/dist/bin/dare.js.map +1 -1
- package/dist/commands/__tests__/dashboard-command.test.d.ts +2 -0
- package/dist/commands/__tests__/dashboard-command.test.d.ts.map +1 -0
- package/dist/commands/__tests__/dashboard-command.test.js +62 -0
- package/dist/commands/__tests__/dashboard-command.test.js.map +1 -0
- package/dist/commands/__tests__/execute.telemetry.spec.js +3 -0
- package/dist/commands/__tests__/execute.telemetry.spec.js.map +1 -1
- package/dist/commands/__tests__/gate-flags.test.d.ts +2 -0
- package/dist/commands/__tests__/gate-flags.test.d.ts.map +1 -0
- package/dist/commands/__tests__/gate-flags.test.js +58 -0
- package/dist/commands/__tests__/gate-flags.test.js.map +1 -0
- package/dist/commands/__tests__/graph-drift.test.d.ts +2 -0
- package/dist/commands/__tests__/graph-drift.test.d.ts.map +1 -0
- package/dist/commands/__tests__/graph-drift.test.js +169 -0
- package/dist/commands/__tests__/graph-drift.test.js.map +1 -0
- package/dist/commands/__tests__/persist-viz.test.d.ts +2 -0
- package/dist/commands/__tests__/persist-viz.test.d.ts.map +1 -0
- package/dist/commands/__tests__/persist-viz.test.js +178 -0
- package/dist/commands/__tests__/persist-viz.test.js.map +1 -0
- package/dist/commands/__tests__/refine-apply.test.d.ts +2 -0
- package/dist/commands/__tests__/refine-apply.test.d.ts.map +1 -0
- package/dist/commands/__tests__/refine-apply.test.js +118 -0
- package/dist/commands/__tests__/refine-apply.test.js.map +1 -0
- package/dist/commands/__tests__/replan-splice.test.d.ts +2 -0
- package/dist/commands/__tests__/replan-splice.test.d.ts.map +1 -0
- package/dist/commands/__tests__/replan-splice.test.js +295 -0
- package/dist/commands/__tests__/replan-splice.test.js.map +1 -0
- package/dist/commands/dag.d.ts.map +1 -1
- package/dist/commands/dag.js +2 -0
- package/dist/commands/dag.js.map +1 -1
- package/dist/commands/dashboard.d.ts +17 -0
- package/dist/commands/dashboard.d.ts.map +1 -0
- package/dist/commands/dashboard.js +77 -0
- package/dist/commands/dashboard.js.map +1 -0
- package/dist/commands/execute.d.ts.map +1 -1
- package/dist/commands/execute.js +117 -2
- package/dist/commands/execute.js.map +1 -1
- package/dist/commands/graph.d.ts.map +1 -1
- package/dist/commands/graph.js +232 -1
- package/dist/commands/graph.js.map +1 -1
- package/dist/commands/guard.d.ts.map +1 -1
- package/dist/commands/guard.js +27 -3
- package/dist/commands/guard.js.map +1 -1
- package/dist/commands/refine.d.ts +2 -0
- package/dist/commands/refine.d.ts.map +1 -1
- package/dist/commands/refine.js +71 -20
- package/dist/commands/refine.js.map +1 -1
- package/dist/commands/review.d.ts.map +1 -1
- package/dist/commands/review.js +36 -5
- package/dist/commands/review.js.map +1 -1
- package/dist/dag-runner/__tests__/maxdepth-config.test.d.ts +2 -0
- package/dist/dag-runner/__tests__/maxdepth-config.test.d.ts.map +1 -0
- package/dist/dag-runner/__tests__/maxdepth-config.test.js +28 -0
- package/dist/dag-runner/__tests__/maxdepth-config.test.js.map +1 -0
- package/dist/dag-runner/__tests__/sub-dag.test.d.ts +2 -0
- package/dist/dag-runner/__tests__/sub-dag.test.d.ts.map +1 -0
- package/dist/dag-runner/__tests__/sub-dag.test.js +127 -0
- package/dist/dag-runner/__tests__/sub-dag.test.js.map +1 -0
- package/dist/dag-runner/graph-ingest.d.ts.map +1 -1
- package/dist/dag-runner/graph-ingest.js +13 -7
- package/dist/dag-runner/graph-ingest.js.map +1 -1
- package/dist/dag-runner/run_dag.d.ts +2 -0
- package/dist/dag-runner/run_dag.d.ts.map +1 -1
- package/dist/dag-runner/run_dag.js.map +1 -1
- package/dist/dag-runner/state-store.d.ts +2 -0
- package/dist/dag-runner/state-store.d.ts.map +1 -1
- package/dist/dag-runner/state-store.js +44 -8
- package/dist/dag-runner/state-store.js.map +1 -1
- package/dist/dag-runner/sub-dag.d.ts +23 -0
- package/dist/dag-runner/sub-dag.d.ts.map +1 -0
- package/dist/dag-runner/sub-dag.js +117 -0
- package/dist/dag-runner/sub-dag.js.map +1 -0
- package/dist/dashboard/__tests__/dashboard-routes.test.d.ts +2 -0
- package/dist/dashboard/__tests__/dashboard-routes.test.d.ts.map +1 -0
- package/dist/dashboard/__tests__/dashboard-routes.test.js +102 -0
- package/dist/dashboard/__tests__/dashboard-routes.test.js.map +1 -0
- package/dist/dashboard/routes.d.ts +8 -0
- package/dist/dashboard/routes.d.ts.map +1 -0
- package/dist/dashboard/routes.js +70 -0
- package/dist/dashboard/routes.js.map +1 -0
- package/dist/graphrag/__tests__/drift-config.test.d.ts +2 -0
- package/dist/graphrag/__tests__/drift-config.test.d.ts.map +1 -0
- package/dist/graphrag/__tests__/drift-config.test.js +33 -0
- package/dist/graphrag/__tests__/drift-config.test.js.map +1 -0
- package/dist/graphrag/__tests__/drift.test.d.ts +2 -0
- package/dist/graphrag/__tests__/drift.test.d.ts.map +1 -0
- package/dist/graphrag/__tests__/drift.test.js +152 -0
- package/dist/graphrag/__tests__/drift.test.js.map +1 -0
- package/dist/graphrag/__tests__/hybrid-integration.test.d.ts +2 -0
- package/dist/graphrag/__tests__/hybrid-integration.test.d.ts.map +1 -0
- package/dist/graphrag/__tests__/hybrid-integration.test.js +158 -0
- package/dist/graphrag/__tests__/hybrid-integration.test.js.map +1 -0
- package/dist/graphrag/__tests__/hybrid.test.d.ts +2 -0
- package/dist/graphrag/__tests__/hybrid.test.d.ts.map +1 -0
- package/dist/graphrag/__tests__/hybrid.test.js +71 -0
- package/dist/graphrag/__tests__/hybrid.test.js.map +1 -0
- package/dist/graphrag/__tests__/incremental-index.test.d.ts +2 -0
- package/dist/graphrag/__tests__/incremental-index.test.d.ts.map +1 -0
- package/dist/graphrag/__tests__/incremental-index.test.js +135 -0
- package/dist/graphrag/__tests__/incremental-index.test.js.map +1 -0
- package/dist/graphrag/__tests__/no-heavy-dep-in-core.test.d.ts +2 -0
- package/dist/graphrag/__tests__/no-heavy-dep-in-core.test.d.ts.map +1 -0
- package/dist/graphrag/__tests__/no-heavy-dep-in-core.test.js +48 -0
- package/dist/graphrag/__tests__/no-heavy-dep-in-core.test.js.map +1 -0
- package/dist/graphrag/__tests__/requirement-hash.test.d.ts +2 -0
- package/dist/graphrag/__tests__/requirement-hash.test.d.ts.map +1 -0
- package/dist/graphrag/__tests__/requirement-hash.test.js +74 -0
- package/dist/graphrag/__tests__/requirement-hash.test.js.map +1 -0
- package/dist/graphrag/__tests__/semantic-config.test.d.ts +2 -0
- package/dist/graphrag/__tests__/semantic-config.test.d.ts.map +1 -0
- package/dist/graphrag/__tests__/semantic-config.test.js +39 -0
- package/dist/graphrag/__tests__/semantic-config.test.js.map +1 -0
- package/dist/graphrag/__tests__/vector-persistence.test.d.ts +2 -0
- package/dist/graphrag/__tests__/vector-persistence.test.d.ts.map +1 -0
- package/dist/graphrag/__tests__/vector-persistence.test.js +112 -0
- package/dist/graphrag/__tests__/vector-persistence.test.js.map +1 -0
- package/dist/graphrag/__tests__/vector-search.test.d.ts +2 -0
- package/dist/graphrag/__tests__/vector-search.test.d.ts.map +1 -0
- package/dist/graphrag/__tests__/vector-search.test.js +36 -0
- package/dist/graphrag/__tests__/vector-search.test.js.map +1 -0
- package/dist/graphrag/drift.d.ts +22 -0
- package/dist/graphrag/drift.d.ts.map +1 -0
- package/dist/graphrag/drift.js +204 -0
- package/dist/graphrag/drift.js.map +1 -0
- package/dist/graphrag/embeddings.d.ts +15 -0
- package/dist/graphrag/embeddings.d.ts.map +1 -0
- package/dist/graphrag/embeddings.js +107 -0
- package/dist/graphrag/embeddings.js.map +1 -0
- package/dist/graphrag/graph-rag.d.ts +17 -1
- package/dist/graphrag/graph-rag.d.ts.map +1 -1
- package/dist/graphrag/graph-rag.js +170 -2
- package/dist/graphrag/graph-rag.js.map +1 -1
- package/dist/graphrag/hybrid.d.ts +8 -0
- package/dist/graphrag/hybrid.d.ts.map +1 -0
- package/dist/graphrag/hybrid.js +203 -0
- package/dist/graphrag/hybrid.js.map +1 -0
- package/dist/graphrag/incremental-index.d.ts +18 -0
- package/dist/graphrag/incremental-index.d.ts.map +1 -0
- package/dist/graphrag/incremental-index.js +159 -0
- package/dist/graphrag/incremental-index.js.map +1 -0
- package/dist/graphrag/index.d.ts +1 -1
- package/dist/graphrag/index.d.ts.map +1 -1
- package/dist/graphrag/json-graph.d.ts +2 -1
- package/dist/graphrag/json-graph.d.ts.map +1 -1
- package/dist/graphrag/json-graph.js +14 -0
- package/dist/graphrag/json-graph.js.map +1 -1
- package/dist/graphrag/knowledge-graph.d.ts +5 -0
- package/dist/graphrag/knowledge-graph.d.ts.map +1 -1
- package/dist/graphrag/neo4j-graph.d.ts +2 -1
- package/dist/graphrag/neo4j-graph.d.ts.map +1 -1
- package/dist/graphrag/neo4j-graph.js +40 -5
- package/dist/graphrag/neo4j-graph.js.map +1 -1
- package/dist/graphrag/requirement-ingest.d.ts +4 -1
- package/dist/graphrag/requirement-ingest.d.ts.map +1 -1
- package/dist/graphrag/requirement-ingest.js +18 -8
- package/dist/graphrag/requirement-ingest.js.map +1 -1
- package/dist/graphrag/types.d.ts +1 -0
- package/dist/graphrag/types.d.ts.map +1 -1
- package/dist/graphrag/vector-search.d.ts +10 -0
- package/dist/graphrag/vector-search.d.ts.map +1 -0
- package/dist/graphrag/vector-search.js +38 -0
- package/dist/graphrag/vector-search.js.map +1 -0
- package/dist/http/__tests__/shared-app.test.d.ts +2 -0
- package/dist/http/__tests__/shared-app.test.d.ts.map +1 -0
- package/dist/http/__tests__/shared-app.test.js +57 -0
- package/dist/http/__tests__/shared-app.test.js.map +1 -0
- package/dist/http/app.d.ts +15 -0
- package/dist/http/app.d.ts.map +1 -0
- package/dist/http/app.js +32 -0
- package/dist/http/app.js.map +1 -0
- package/dist/mcp-server/server.d.ts.map +1 -1
- package/dist/mcp-server/server.js +13 -16
- package/dist/mcp-server/server.js.map +1 -1
- package/dist/reporters/__tests__/github-reporter.test.d.ts +2 -0
- package/dist/reporters/__tests__/github-reporter.test.d.ts.map +1 -0
- package/dist/reporters/__tests__/github-reporter.test.js +93 -0
- package/dist/reporters/__tests__/github-reporter.test.js.map +1 -0
- package/dist/reporters/ci-gate.d.ts +27 -0
- package/dist/reporters/ci-gate.d.ts.map +1 -0
- package/dist/reporters/ci-gate.js +100 -0
- package/dist/reporters/ci-gate.js.map +1 -0
- package/dist/reporters/github.d.ts +36 -0
- package/dist/reporters/github.d.ts.map +1 -0
- package/dist/reporters/github.js +142 -0
- package/dist/reporters/github.js.map +1 -0
- package/dist/telemetry/__tests__/aggregator.test.d.ts +2 -0
- package/dist/telemetry/__tests__/aggregator.test.d.ts.map +1 -0
- package/dist/telemetry/__tests__/aggregator.test.js +164 -0
- package/dist/telemetry/__tests__/aggregator.test.js.map +1 -0
- package/dist/telemetry/aggregator.d.ts +43 -0
- package/dist/telemetry/aggregator.d.ts.map +1 -0
- package/dist/telemetry/aggregator.js +282 -0
- package/dist/telemetry/aggregator.js.map +1 -0
- package/dist/utils/UpdateApplier.d.ts.map +1 -1
- package/dist/utils/UpdateApplier.js +3 -1
- package/dist/utils/UpdateApplier.js.map +1 -1
- package/dist/utils/excalidraw-renderer.d.ts.map +1 -1
- package/dist/utils/excalidraw-renderer.js +1 -0
- package/dist/utils/excalidraw-renderer.js.map +1 -1
- package/dist/utils/graph-renderer.d.ts +2 -0
- package/dist/utils/graph-renderer.d.ts.map +1 -1
- package/dist/utils/graph-renderer.js +188 -14
- package/dist/utils/graph-renderer.js.map +1 -1
- package/dist/utils/project-generator.d.ts.map +1 -1
- package/dist/utils/project-generator.js +34 -5
- package/dist/utils/project-generator.js.map +1 -1
- package/dist/verification/__tests__/decay-policy.test.js +1 -0
- package/dist/verification/__tests__/decay-policy.test.js.map +1 -1
- package/dist/verification/__tests__/telemetry.test.js +3 -0
- package/dist/verification/__tests__/telemetry.test.js.map +1 -1
- package/dist/verification/config.d.ts +82 -0
- package/dist/verification/config.d.ts.map +1 -1
- package/dist/verification/config.js +148 -0
- package/dist/verification/config.js.map +1 -1
- package/dist/verification/types.d.ts +2 -0
- package/dist/verification/types.d.ts.map +1 -1
- package/package.json +3 -2
- package/templates/.github/workflows/dare-pr.yml +26 -0
- package/templates/dashboard/app.js +184 -0
- package/templates/dashboard/index.html +63 -0
- package/templates/dashboard/style.css +134 -0
- package/templates/ide/antigravity/.agents/skills/dare-dashboard/SKILL.md +18 -0
- package/templates/ide/claude/.claude/commands/dare-dashboard.md +18 -0
- package/templates/ide/cursor/.cursor/commands/dare-dashboard.md +18 -0
|
@@ -0,0 +1,107 @@
|
|
|
1
|
+
import { createHash } from 'node:crypto';
|
|
2
|
+
const DEFAULT_RUNTIME_PACKAGE = '@huggingface/transformers';
|
|
3
|
+
export class EmbeddingModelMissingError extends Error {
|
|
4
|
+
constructor(runtimePackage = DEFAULT_RUNTIME_PACKAGE) {
|
|
5
|
+
super(`Optional embedding runtime not installed. Run: npm i ${runtimePackage} — or disable graphrag.semantic.`);
|
|
6
|
+
this.code = 'EMBEDDING_MODEL_MISSING';
|
|
7
|
+
this.name = 'EmbeddingModelMissingError';
|
|
8
|
+
}
|
|
9
|
+
}
|
|
10
|
+
function isRecord(value) {
|
|
11
|
+
return typeof value === 'object' && value !== null;
|
|
12
|
+
}
|
|
13
|
+
function readModelHash(value) {
|
|
14
|
+
if (!isRecord(value))
|
|
15
|
+
return null;
|
|
16
|
+
const candidates = [value.modelHash, value.sha256, value.hash];
|
|
17
|
+
for (const candidate of candidates) {
|
|
18
|
+
if (typeof candidate === 'string' && candidate.trim().length > 0) {
|
|
19
|
+
return candidate.trim();
|
|
20
|
+
}
|
|
21
|
+
}
|
|
22
|
+
return null;
|
|
23
|
+
}
|
|
24
|
+
function toFloat32Array(value) {
|
|
25
|
+
if (value instanceof Float32Array) {
|
|
26
|
+
return value;
|
|
27
|
+
}
|
|
28
|
+
if (Array.isArray(value)) {
|
|
29
|
+
return new Float32Array(value.map((item) => Number(item)));
|
|
30
|
+
}
|
|
31
|
+
if (isRecord(value)) {
|
|
32
|
+
if (value.data instanceof Float32Array) {
|
|
33
|
+
return value.data;
|
|
34
|
+
}
|
|
35
|
+
if (Array.isArray(value.data)) {
|
|
36
|
+
return new Float32Array(value.data.map((item) => Number(item)));
|
|
37
|
+
}
|
|
38
|
+
}
|
|
39
|
+
throw new Error('Embedding runtime returned an unsupported vector shape.');
|
|
40
|
+
}
|
|
41
|
+
function assertModelHash(model, expectedHash, candidates) {
|
|
42
|
+
const discoveredHash = candidates.map(readModelHash).find((hash) => hash !== null);
|
|
43
|
+
const effectiveHash = discoveredHash ?? createHash('sha256').update(`model:${model}`).digest('hex');
|
|
44
|
+
if (effectiveHash !== expectedHash) {
|
|
45
|
+
throw new Error(`Embedding model hash mismatch for "${model}". Expected ${expectedHash}, got ${effectiveHash}.`);
|
|
46
|
+
}
|
|
47
|
+
}
|
|
48
|
+
function normalizeEmbedder(candidate) {
|
|
49
|
+
return {
|
|
50
|
+
get dim() {
|
|
51
|
+
const dim = candidate.dim ?? candidate.dimension;
|
|
52
|
+
if (typeof dim !== 'number' || !Number.isFinite(dim) || dim <= 0) {
|
|
53
|
+
throw new Error('Embedding runtime returned an invalid vector dimension.');
|
|
54
|
+
}
|
|
55
|
+
return Math.trunc(dim);
|
|
56
|
+
},
|
|
57
|
+
async embed(text) {
|
|
58
|
+
const vector = await candidate.embed(text);
|
|
59
|
+
return toFloat32Array(vector);
|
|
60
|
+
},
|
|
61
|
+
};
|
|
62
|
+
}
|
|
63
|
+
async function fromRuntimeFactory(rt, cfg) {
|
|
64
|
+
const factory = rt.loadEmbedder ?? rt.createEmbedder;
|
|
65
|
+
if (!factory) {
|
|
66
|
+
return null;
|
|
67
|
+
}
|
|
68
|
+
const candidate = await factory({ model: cfg.model, modelHash: cfg.modelHash });
|
|
69
|
+
assertModelHash(cfg.model, cfg.modelHash, [candidate]);
|
|
70
|
+
return normalizeEmbedder(candidate);
|
|
71
|
+
}
|
|
72
|
+
async function fromTransformersRuntime(rt, cfg) {
|
|
73
|
+
if (!rt.pipeline) {
|
|
74
|
+
throw new Error('Embedding runtime is missing a supported embedder factory.');
|
|
75
|
+
}
|
|
76
|
+
const extractor = await rt.pipeline('feature-extraction', cfg.model);
|
|
77
|
+
const probe = await extractor('shape probe', { pooling: 'mean', normalize: true });
|
|
78
|
+
const probeVector = toFloat32Array(probe);
|
|
79
|
+
assertModelHash(cfg.model, cfg.modelHash, [extractor, probe]);
|
|
80
|
+
return {
|
|
81
|
+
dim: probeVector.length,
|
|
82
|
+
async embed(text) {
|
|
83
|
+
const output = await extractor(text, { pooling: 'mean', normalize: true });
|
|
84
|
+
const vector = toFloat32Array(output);
|
|
85
|
+
if (vector.length !== probeVector.length) {
|
|
86
|
+
throw new Error(`Embedding dimension drift detected. Expected ${probeVector.length}, got ${vector.length}.`);
|
|
87
|
+
}
|
|
88
|
+
return vector;
|
|
89
|
+
},
|
|
90
|
+
};
|
|
91
|
+
}
|
|
92
|
+
export async function loadEmbedder(cfg) {
|
|
93
|
+
const runtimePackage = cfg.runtimePackage ?? DEFAULT_RUNTIME_PACKAGE;
|
|
94
|
+
let runtime;
|
|
95
|
+
try {
|
|
96
|
+
runtime = (await import(runtimePackage));
|
|
97
|
+
}
|
|
98
|
+
catch {
|
|
99
|
+
throw new EmbeddingModelMissingError(runtimePackage);
|
|
100
|
+
}
|
|
101
|
+
const embedderFromRuntime = await fromRuntimeFactory(runtime, cfg);
|
|
102
|
+
if (embedderFromRuntime) {
|
|
103
|
+
return embedderFromRuntime;
|
|
104
|
+
}
|
|
105
|
+
return fromTransformersRuntime(runtime, cfg);
|
|
106
|
+
}
|
|
107
|
+
//# sourceMappingURL=embeddings.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"embeddings.js","sourceRoot":"","sources":["../../src/graphrag/embeddings.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAEzC,MAAM,uBAAuB,GAAG,2BAA2B,CAAC;AAa5D,MAAM,OAAO,0BAA2B,SAAQ,KAAK;IAGnD,YAAY,cAAc,GAAG,uBAAuB;QAClD,KAAK,CACH,wDAAwD,cAAc,kCAAkC,CACzG,CAAC;QALK,SAAI,GAAG,yBAAkC,CAAC;QAMjD,IAAI,CAAC,IAAI,GAAG,4BAA4B,CAAC;IAC3C,CAAC;CACF;AAyBD,SAAS,QAAQ,CAAC,KAAc;IAC9B,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,CAAC;AACrD,CAAC;AAED,SAAS,aAAa,CAAC,KAAc;IACnC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC;QAAE,OAAO,IAAI,CAAC;IAClC,MAAM,UAAU,GAAG,CAAC,KAAK,CAAC,SAAS,EAAE,KAAK,CAAC,MAAM,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;IAC/D,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;QACnC,IAAI,OAAO,SAAS,KAAK,QAAQ,IAAI,SAAS,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACjE,OAAO,SAAS,CAAC,IAAI,EAAE,CAAC;QAC1B,CAAC;IACH,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,cAAc,CAAC,KAAc;IACpC,IAAI,KAAK,YAAY,YAAY,EAAE,CAAC;QAClC,OAAO,KAAK,CAAC;IACf,CAAC;IACD,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QACzB,OAAO,IAAI,YAAY,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAC7D,CAAC;IACD,IAAI,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;QACpB,IAAI,KAAK,CAAC,IAAI,YAAY,YAAY,EAAE,CAAC;YACvC,OAAO,KAAK,CAAC,IAAI,CAAC;QACpB,CAAC;QACD,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC;YAC9B,OAAO,IAAI,YAAY,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClE,CAAC;IACH,CAAC;IACD,MAAM,IAAI,KAAK,CAAC,yDAAyD,CAAC,CAAC;AAC7E,CAAC;AAED,SAAS,eAAe,CAAC,KAAa,EAAE,YAAoB,EAAE,UAAkC;IAC9F,MAAM,cAAc,GAAG,UAAU,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,KAAK,IAAI,CAAC,CAAC;IACnF,MAAM,aAAa,GACjB,cAAc,IAAI,UAAU,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,SAAS,KAAK,EAAE,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IAEhF,IAAI,aAAa,KAAK,YAAY,EAAE,CAAC;QACnC,MAAM,IAAI,KAAK,CACb,sCAAsC,KAAK,eAAe,YAAY,SAAS,aAAa,GAAG,CAChG,CAAC;IACJ,CAAC;AACH,CAAC;AAED,SAAS,iBAAiB,CAAC,SAAmC;IAC5D,OAAO;QACL,IAAI,GAAG;YACL,MAAM,GAAG,GAAG,SAAS,CAAC,GAAG,IAAI,SAAS,CAAC,SAAS,CAAC;YACjD,IAAI,OAAO,GAAG,KAAK,QAAQ,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC;gBACjE,MAAM,IAAI,KAAK,CAAC,yDAAyD,CAAC,CAAC;YAC7E,CAAC;YACD,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QACzB,CAAC;QACD,KAAK,CAAC,KAAK,CAAC,IAAY;YACtB,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YAC3C,OAAO,cAAc,CAAC,MAAM,CAAC,CAAC;QAChC,CAAC;KACF,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,kBAAkB,CAAC,EAA0B,EAAE,GAAmB;IAC/E,MAAM,OAAO,GAAG,EAAE,CAAC,YAAY,IAAI,EAAE,CAAC,cAAc,CAAC;IACrD,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,SAAS,GAAG,MAAM,OAAO,CAAC,EAAE,KAAK,EAAE,GAAG,CAAC,KAAK,EAAE,SAAS,EAAE,GAAG,CAAC,SAAS,EAAE,CAAC,CAAC;IAChF,eAAe,CAAC,GAAG,CAAC,KAAK,EAAE,GAAG,CAAC,SAAS,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC;IACvD,OAAO,iBAAiB,CAAC,SAAS,CAAC,CAAC;AACtC,CAAC;AAED,KAAK,UAAU,uBAAuB,CAAC,EAA0B,EAAE,GAAmB;IACpF,IAAI,CAAC,EAAE,CAAC,QAAQ,EAAE,CAAC;QACjB,MAAM,IAAI,KAAK,CAAC,4DAA4D,CAAC,CAAC;IAChF,CAAC;IAED,MAAM,SAAS,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,oBAAoB,EAAE,GAAG,CAAC,KAAK,CAAC,CAAC;IACrE,MAAM,KAAK,GAAG,MAAM,SAAS,CAAC,aAAa,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IACnF,MAAM,WAAW,GAAG,cAAc,CAAC,KAAK,CAAC,CAAC;IAC1C,eAAe,CAAC,GAAG,CAAC,KAAK,EAAE,GAAG,CAAC,SAAS,EAAE,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC,CAAC;IAE9D,OAAO;QACL,GAAG,EAAE,WAAW,CAAC,MAAM;QACvB,KAAK,CAAC,KAAK,CAAC,IAAY;YACtB,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,IAAI,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YAC3E,MAAM,MAAM,GAAG,cAAc,CAAC,MAAM,CAAC,CAAC;YACtC,IAAI,MAAM,CAAC,MAAM,KAAK,WAAW,CAAC,MAAM,EAAE,CAAC;gBACzC,MAAM,IAAI,KAAK,CACb,gDAAgD,WAAW,CAAC,MAAM,SAAS,MAAM,CAAC,MAAM,GAAG,CAC5F,CAAC;YACJ,CAAC;YACD,OAAO,MAAM,CAAC;QAChB,CAAC;KACF,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,YAAY,CAAC,GAAmB;IACpD,MAAM,cAAc,GAAG,GAAG,CAAC,cAAc,IAAI,uBAAuB,CAAC;IACrE,IAAI,OAA+B,CAAC;IACpC,IAAI,CAAC;QACH,OAAO,GAAG,CAAC,MAAM,MAAM,CAAC,cAAc,CAAC,CAA2B,CAAC;IACrE,CAAC;IAAC,MAAM,CAAC;QACP,MAAM,IAAI,0BAA0B,CAAC,cAAc,CAAC,CAAC;IACvD,CAAC;IAED,MAAM,mBAAmB,GAAG,MAAM,kBAAkB,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC;IACnE,IAAI,mBAAmB,EAAE,CAAC;QACxB,OAAO,mBAAmB,CAAC;IAC7B,CAAC;IAED,OAAO,uBAAuB,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC;AAC/C,CAAC"}
|
|
@@ -1,26 +1,35 @@
|
|
|
1
|
-
import type { KnowledgeGraph } from './knowledge-graph.js';
|
|
1
|
+
import type { KnowledgeGraph, VectorRow } from './knowledge-graph.js';
|
|
2
2
|
import type { GraphNode, GraphEdge, NodeType, SearchResult, GraphStatistics, CodeSymbolNode, TraverseOptions, TraverseResult, LocateOptions, LocateResult } from './types.js';
|
|
3
|
+
export type { VectorRow } from './knowledge-graph.js';
|
|
3
4
|
export declare class GraphRAG implements KnowledgeGraph {
|
|
4
5
|
private db;
|
|
5
6
|
private dbPath;
|
|
6
7
|
private SQL;
|
|
8
|
+
private semanticConfigCache;
|
|
9
|
+
private embedderPromise;
|
|
10
|
+
private semanticFallbackLogged;
|
|
7
11
|
constructor(dbPath: string);
|
|
8
12
|
init(): Promise<void>;
|
|
9
13
|
private save;
|
|
10
14
|
private run;
|
|
11
15
|
private query;
|
|
12
16
|
private queryOne;
|
|
17
|
+
private ensureVectorColumn;
|
|
13
18
|
addNode(node: GraphNode): void;
|
|
14
19
|
getNode(id: string): GraphNode | null;
|
|
15
20
|
queryNodes(type?: NodeType, limit?: number): GraphNode[];
|
|
16
21
|
searchNodes(queryStr: string, limit?: number): SearchResult[];
|
|
22
|
+
searchNodesHybrid(queryStr: string, limit?: number): Promise<SearchResult[]>;
|
|
23
|
+
private keywordSearchNodes;
|
|
17
24
|
deleteNode(id: string): void;
|
|
18
25
|
addEdge(edge: GraphEdge): void;
|
|
19
26
|
getEdges(nodeId: string, direction?: 'out' | 'in' | 'both'): GraphEdge[];
|
|
20
27
|
getNodeDependencies(nodeId: string, depth?: number): GraphNode[];
|
|
21
28
|
traverse(opts: TraverseOptions): TraverseResult;
|
|
22
29
|
locate(seedQuery: string, opts?: LocateOptions): LocateResult;
|
|
30
|
+
locateHybrid(seedQuery: string, opts?: LocateOptions): Promise<LocateResult>;
|
|
23
31
|
findByQualifiedName(qn: string): CodeSymbolNode | null;
|
|
32
|
+
loadVectors(): VectorRow[];
|
|
24
33
|
getStatistics(): GraphStatistics;
|
|
25
34
|
exportToJson(): {
|
|
26
35
|
nodes: GraphNode[];
|
|
@@ -33,5 +42,12 @@ export declare class GraphRAG implements KnowledgeGraph {
|
|
|
33
42
|
close(): void;
|
|
34
43
|
private rowToNode;
|
|
35
44
|
private rowToEdge;
|
|
45
|
+
private serializeVector;
|
|
46
|
+
private deserializeVector;
|
|
47
|
+
private createSearchOverrideView;
|
|
48
|
+
private getSemanticConfig;
|
|
49
|
+
private resolveProjectRoot;
|
|
50
|
+
private loadEmbedderLazy;
|
|
51
|
+
private logSemanticFallback;
|
|
36
52
|
}
|
|
37
53
|
//# sourceMappingURL=graph-rag.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"graph-rag.d.ts","sourceRoot":"","sources":["../../src/graphrag/graph-rag.ts"],"names":[],"mappings":"
|
|
1
|
+
{"version":3,"file":"graph-rag.d.ts","sourceRoot":"","sources":["../../src/graphrag/graph-rag.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAAE,cAAc,EAAE,SAAS,EAAE,MAAM,sBAAsB,CAAC;AACtE,OAAO,KAAK,EACV,SAAS,EAAE,SAAS,EAAE,QAAQ,EAC9B,YAAY,EAAE,eAAe,EAC7B,cAAc,EAAE,eAAe,EAAE,cAAc,EAC/C,aAAa,EAAE,YAAY,EAC5B,MAAM,YAAY,CAAC;AAOpB,YAAY,EAAE,SAAS,EAAE,MAAM,sBAAsB,CAAC;AAuCtD,qBAAa,QAAS,YAAW,cAAc;IAC7C,OAAO,CAAC,EAAE,CAAY;IACtB,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,GAAG,CAAe;IAC1B,OAAO,CAAC,mBAAmB,CAAuC;IAClE,OAAO,CAAC,eAAe,CAAyC;IAChE,OAAO,CAAC,sBAAsB,CAAS;gBAE3B,MAAM,EAAE,MAAM;IAIpB,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;IAgB3B,OAAO,CAAC,IAAI;IAKZ,OAAO,CAAC,GAAG;IAKX,OAAO,CAAC,KAAK;IAWb,OAAO,CAAC,QAAQ;IAKhB,OAAO,CAAC,kBAAkB;IAU1B,OAAO,CAAC,IAAI,EAAE,SAAS,GAAG,IAAI;IA8B9B,OAAO,CAAC,EAAE,EAAE,MAAM,GAAG,SAAS,GAAG,IAAI;IAMrC,UAAU,CAAC,IAAI,CAAC,EAAE,QAAQ,EAAE,KAAK,SAAK,GAAG,SAAS,EAAE;IAOpD,WAAW,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,SAAK,GAAG,YAAY,EAAE;IAInD,iBAAiB,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,SAAK,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;IAuB9E,OAAO,CAAC,kBAAkB;IAc1B,UAAU,CAAC,EAAE,EAAE,MAAM,GAAG,IAAI;IAO5B,OAAO,CAAC,IAAI,EAAE,SAAS,GAAG,IAAI;IAe9B,QAAQ,CAAC,MAAM,EAAE,MAAM,EAAE,SAAS,GAAE,KAAK,GAAG,IAAI,GAAG,MAAe,GAAG,SAAS,EAAE;IAWhF,mBAAmB,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,SAAI,GAAG,SAAS,EAAE;IA6B3D,QAAQ,CAAC,IAAI,EAAE,eAAe,GAAG,cAAc;IAI/C,MAAM,CAAC,SAAS,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,aAAa,GAAG,YAAY;IAIvD,YAAY,CAAC,SAAS,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,aAAa,GAAG,OAAO,CAAC,YAAY,CAAC;IAmBlF,mBAAmB,CAAC,EAAE,EAAE,MAAM,GAAG,cAAc,GAAG,IAAI;IAsBtD,WAAW,IAAI,SAAS,EAAE;IAa1B,aAAa,IAAI,eAAe;IAuBhC,YAAY,IAAI;QAAE,KAAK,EAAE,SAAS,EAAE,CAAC;QAAC,KAAK,EAAE,SAAS,EAAE,CAAA;KAAE;IAM1D,cAAc,CAAC,IAAI,EAAE;QAAE,KAAK,EAAE,SAAS,EAAE,CAAC;QAAC,KAAK,EAAE,SAAS,EAAE,CAAA;KAAE,GAAG,IAAI;IAKtE,KAAK,IAAI,IAAI;IAOb,OAAO,CAAC,SAAS;IAcjB,OAAO,CAAC,SAAS;IAWjB,OAAO,CAAC,eAAe;IAKvB,OAAO,CAAC,iBAAiB;IASzB,OAAO,CAAC,wBAAwB;IAQhC,OAAO,CAAC,iBAAiB;IAwBzB,OAAO,CAAC,kBAAkB;YAQZ,gBAAgB;IAc9B,OAAO,CAAC,mBAAmB;CAY5B"}
|
|
@@ -1,14 +1,19 @@
|
|
|
1
1
|
import initSqlJs from 'sql.js';
|
|
2
|
+
import { createHash } from 'node:crypto';
|
|
2
3
|
import path from 'path';
|
|
3
4
|
import fs from 'fs-extra';
|
|
4
5
|
import { emptyEdgesByType, emptyNodesByType } from './types.js';
|
|
5
6
|
import { traverse as traverseFn, locate as locateFn } from './traverse.js';
|
|
7
|
+
import { EmbeddingModelMissingError, loadEmbedder } from './embeddings.js';
|
|
8
|
+
import { hybridSearch } from './hybrid.js';
|
|
9
|
+
import { parseSemanticConfig } from '../verification/config.js';
|
|
6
10
|
const SCHEMA_SQL = `
|
|
7
11
|
CREATE TABLE IF NOT EXISTS nodes (
|
|
8
12
|
id TEXT PRIMARY KEY,
|
|
9
13
|
type TEXT NOT NULL,
|
|
10
14
|
label TEXT NOT NULL,
|
|
11
15
|
description TEXT,
|
|
16
|
+
vector BLOB,
|
|
12
17
|
metadata TEXT DEFAULT '{}',
|
|
13
18
|
created_at TEXT DEFAULT (datetime('now')),
|
|
14
19
|
updated_at TEXT DEFAULT (datetime('now'))
|
|
@@ -28,8 +33,12 @@ CREATE INDEX IF NOT EXISTS idx_edges_source ON edges(source_id);
|
|
|
28
33
|
CREATE INDEX IF NOT EXISTS idx_edges_target ON edges(target_id);
|
|
29
34
|
CREATE INDEX IF NOT EXISTS idx_edges_type ON edges(type);
|
|
30
35
|
`;
|
|
36
|
+
const PATH_SEED_HINT_RE = /[\\/]|::|\.(ts|tsx|js|jsx|mjs|cjs|py|go|rs|php|rb|java|kt|cs)$/i;
|
|
31
37
|
export class GraphRAG {
|
|
32
38
|
constructor(dbPath) {
|
|
39
|
+
this.semanticConfigCache = null;
|
|
40
|
+
this.embedderPromise = null;
|
|
41
|
+
this.semanticFallbackLogged = false;
|
|
33
42
|
this.dbPath = dbPath;
|
|
34
43
|
}
|
|
35
44
|
async init() {
|
|
@@ -43,6 +52,7 @@ export class GraphRAG {
|
|
|
43
52
|
this.db = new this.SQL.Database();
|
|
44
53
|
}
|
|
45
54
|
this.db.run(SCHEMA_SQL);
|
|
55
|
+
this.ensureVectorColumn();
|
|
46
56
|
this.save();
|
|
47
57
|
}
|
|
48
58
|
save() {
|
|
@@ -67,14 +77,34 @@ export class GraphRAG {
|
|
|
67
77
|
const results = this.query(sql, params);
|
|
68
78
|
return results[0] ?? null;
|
|
69
79
|
}
|
|
80
|
+
ensureVectorColumn() {
|
|
81
|
+
const columns = this.query('PRAGMA table_info(nodes)');
|
|
82
|
+
const hasVectorColumn = columns.some((column) => column.name === 'vector');
|
|
83
|
+
if (!hasVectorColumn) {
|
|
84
|
+
this.db.run('ALTER TABLE nodes ADD COLUMN vector BLOB');
|
|
85
|
+
}
|
|
86
|
+
}
|
|
70
87
|
// ─── Node Operations ───────────────────────────────────────────────────────
|
|
71
88
|
addNode(node) {
|
|
72
89
|
const existing = this.queryOne('SELECT id FROM nodes WHERE id = ?', [node.id]);
|
|
90
|
+
const vectorBlob = node.vector === undefined ? undefined : this.serializeVector(node.vector);
|
|
73
91
|
if (existing) {
|
|
74
|
-
|
|
92
|
+
if (vectorBlob === undefined) {
|
|
93
|
+
this.run(`UPDATE nodes SET label = ?, description = ?, metadata = ?, updated_at = datetime('now') WHERE id = ?`, [node.label, node.description || null, JSON.stringify(node.metadata || {}), node.id]);
|
|
94
|
+
}
|
|
95
|
+
else {
|
|
96
|
+
this.run(`UPDATE nodes SET label = ?, description = ?, vector = ?, metadata = ?, updated_at = datetime('now') WHERE id = ?`, [node.label, node.description || null, vectorBlob, JSON.stringify(node.metadata || {}), node.id]);
|
|
97
|
+
}
|
|
75
98
|
}
|
|
76
99
|
else {
|
|
77
|
-
this.run(`INSERT INTO nodes (id, type, label, description, metadata) VALUES (?, ?, ?, ?, ?)`, [
|
|
100
|
+
this.run(`INSERT INTO nodes (id, type, label, description, vector, metadata) VALUES (?, ?, ?, ?, ?, ?)`, [
|
|
101
|
+
node.id,
|
|
102
|
+
node.type,
|
|
103
|
+
node.label,
|
|
104
|
+
node.description || null,
|
|
105
|
+
vectorBlob ?? null,
|
|
106
|
+
JSON.stringify(node.metadata || {}),
|
|
107
|
+
]);
|
|
78
108
|
}
|
|
79
109
|
}
|
|
80
110
|
getNode(id) {
|
|
@@ -90,6 +120,28 @@ export class GraphRAG {
|
|
|
90
120
|
return rows.map((r) => this.rowToNode(r));
|
|
91
121
|
}
|
|
92
122
|
searchNodes(queryStr, limit = 10) {
|
|
123
|
+
return this.keywordSearchNodes(queryStr, limit);
|
|
124
|
+
}
|
|
125
|
+
async searchNodesHybrid(queryStr, limit = 10) {
|
|
126
|
+
const keywordRanked = this.keywordSearchNodes(queryStr, limit);
|
|
127
|
+
const semantic = this.getSemanticConfig();
|
|
128
|
+
if (!semantic.enabled) {
|
|
129
|
+
return keywordRanked;
|
|
130
|
+
}
|
|
131
|
+
const embedder = await this.loadEmbedderLazy(semantic);
|
|
132
|
+
if (!embedder) {
|
|
133
|
+
return keywordRanked;
|
|
134
|
+
}
|
|
135
|
+
try {
|
|
136
|
+
const keywordView = this.createSearchOverrideView((query, searchLimit = 10) => this.keywordSearchNodes(query, searchLimit));
|
|
137
|
+
return await hybridSearch(keywordView, embedder, queryStr, { k: limit, rrfK: semantic.rrfK });
|
|
138
|
+
}
|
|
139
|
+
catch (error) {
|
|
140
|
+
this.logSemanticFallback(error);
|
|
141
|
+
return keywordRanked;
|
|
142
|
+
}
|
|
143
|
+
}
|
|
144
|
+
keywordSearchNodes(queryStr, limit = 10) {
|
|
93
145
|
// Simple LIKE-based search (sql.js doesn't support FTS5)
|
|
94
146
|
const pattern = `%${queryStr.toLowerCase()}%`;
|
|
95
147
|
const rows = this.query(`SELECT * FROM nodes WHERE lower(label) LIKE ? OR lower(description) LIKE ? LIMIT ?`, [pattern, pattern, limit]);
|
|
@@ -158,6 +210,23 @@ export class GraphRAG {
|
|
|
158
210
|
locate(seedQuery, opts) {
|
|
159
211
|
return locateFn(this, seedQuery, opts);
|
|
160
212
|
}
|
|
213
|
+
async locateHybrid(seedQuery, opts) {
|
|
214
|
+
const semantic = this.getSemanticConfig();
|
|
215
|
+
if (!semantic.enabled || PATH_SEED_HINT_RE.test(seedQuery)) {
|
|
216
|
+
return locateFn(this, seedQuery, opts);
|
|
217
|
+
}
|
|
218
|
+
const hybridRanked = await this.searchNodesHybrid(seedQuery, 5);
|
|
219
|
+
const rankedForSeed = hybridRanked.slice(0, 5);
|
|
220
|
+
if (rankedForSeed.length === 0) {
|
|
221
|
+
return locateFn(this, seedQuery, opts);
|
|
222
|
+
}
|
|
223
|
+
const seededView = this.createSearchOverrideView((query, limit = 10) => {
|
|
224
|
+
if (query === seedQuery)
|
|
225
|
+
return rankedForSeed.slice(0, limit);
|
|
226
|
+
return this.keywordSearchNodes(query, limit);
|
|
227
|
+
});
|
|
228
|
+
return locateFn(seededView, seedQuery, opts);
|
|
229
|
+
}
|
|
161
230
|
findByQualifiedName(qn) {
|
|
162
231
|
const normalized = qn.startsWith('code_symbol:') ? qn.slice('code_symbol:'.length) : qn;
|
|
163
232
|
const directId = `code_symbol:${normalized}`;
|
|
@@ -178,6 +247,17 @@ export class GraphRAG {
|
|
|
178
247
|
}
|
|
179
248
|
return null;
|
|
180
249
|
}
|
|
250
|
+
loadVectors() {
|
|
251
|
+
const rows = this.query('SELECT id, vector FROM nodes WHERE vector IS NOT NULL');
|
|
252
|
+
const vectors = [];
|
|
253
|
+
for (const row of rows) {
|
|
254
|
+
const decoded = this.deserializeVector(row.vector);
|
|
255
|
+
if (!decoded || decoded.length === 0)
|
|
256
|
+
continue;
|
|
257
|
+
vectors.push({ id: row.id, v: decoded });
|
|
258
|
+
}
|
|
259
|
+
return vectors;
|
|
260
|
+
}
|
|
181
261
|
getStatistics() {
|
|
182
262
|
const totalNodesRow = this.queryOne('SELECT COUNT(*) as count FROM nodes');
|
|
183
263
|
const totalEdgesRow = this.queryOne('SELECT COUNT(*) as count FROM edges');
|
|
@@ -214,11 +294,13 @@ export class GraphRAG {
|
|
|
214
294
|
}
|
|
215
295
|
// ─── Private ───────────────────────────────────────────────────────────────
|
|
216
296
|
rowToNode(row) {
|
|
297
|
+
const vector = this.deserializeVector(row['vector']);
|
|
217
298
|
return {
|
|
218
299
|
id: row['id'],
|
|
219
300
|
type: row['type'],
|
|
220
301
|
label: row['label'],
|
|
221
302
|
description: row['description'],
|
|
303
|
+
vector: vector ? Array.from(vector) : undefined,
|
|
222
304
|
metadata: JSON.parse(row['metadata'] || '{}'),
|
|
223
305
|
createdAt: row['created_at'],
|
|
224
306
|
updatedAt: row['updated_at'],
|
|
@@ -234,5 +316,91 @@ export class GraphRAG {
|
|
|
234
316
|
metadata: JSON.parse(row['metadata'] || '{}'),
|
|
235
317
|
};
|
|
236
318
|
}
|
|
319
|
+
serializeVector(vector) {
|
|
320
|
+
const floatVector = new Float32Array(vector.map((value) => Number(value)));
|
|
321
|
+
return new Uint8Array(floatVector.buffer);
|
|
322
|
+
}
|
|
323
|
+
deserializeVector(value) {
|
|
324
|
+
const bytes = toUint8Array(value);
|
|
325
|
+
if (!bytes || bytes.byteLength === 0 || bytes.byteLength % Float32Array.BYTES_PER_ELEMENT !== 0) {
|
|
326
|
+
return null;
|
|
327
|
+
}
|
|
328
|
+
const copy = Uint8Array.from(bytes);
|
|
329
|
+
return new Float32Array(copy.buffer, copy.byteOffset, copy.byteLength / Float32Array.BYTES_PER_ELEMENT);
|
|
330
|
+
}
|
|
331
|
+
createSearchOverrideView(search) {
|
|
332
|
+
const view = Object.create(this);
|
|
333
|
+
view.searchNodes = search;
|
|
334
|
+
return view;
|
|
335
|
+
}
|
|
336
|
+
getSemanticConfig() {
|
|
337
|
+
if (this.semanticConfigCache) {
|
|
338
|
+
return this.semanticConfigCache;
|
|
339
|
+
}
|
|
340
|
+
const projectRoot = this.resolveProjectRoot();
|
|
341
|
+
const configPath = path.join(projectRoot, 'dare.config.json');
|
|
342
|
+
const rawConfig = fs.existsSync(configPath) ? fs.readJSONSync(configPath) : {};
|
|
343
|
+
const parsed = parseSemanticConfig(rawConfig);
|
|
344
|
+
const model = parsed.model.trim();
|
|
345
|
+
const modelHash = typeof parsed.modelHash === 'string' && parsed.modelHash.trim().length > 0
|
|
346
|
+
? parsed.modelHash.trim()
|
|
347
|
+
: createHash('sha256').update(`model:${model}`).digest('hex');
|
|
348
|
+
this.semanticConfigCache = {
|
|
349
|
+
enabled: parsed.enabled && model.length > 0,
|
|
350
|
+
model,
|
|
351
|
+
modelHash,
|
|
352
|
+
rrfK: parsed.rrfK,
|
|
353
|
+
};
|
|
354
|
+
return this.semanticConfigCache;
|
|
355
|
+
}
|
|
356
|
+
resolveProjectRoot() {
|
|
357
|
+
const dbDir = path.dirname(this.dbPath);
|
|
358
|
+
if (path.basename(dbDir) === '.dare') {
|
|
359
|
+
return path.dirname(dbDir);
|
|
360
|
+
}
|
|
361
|
+
return process.cwd();
|
|
362
|
+
}
|
|
363
|
+
async loadEmbedderLazy(semantic) {
|
|
364
|
+
if (this.embedderPromise) {
|
|
365
|
+
return this.embedderPromise;
|
|
366
|
+
}
|
|
367
|
+
this.embedderPromise = loadEmbedder({
|
|
368
|
+
model: semantic.model,
|
|
369
|
+
modelHash: semantic.modelHash,
|
|
370
|
+
}).catch((error) => {
|
|
371
|
+
this.logSemanticFallback(error);
|
|
372
|
+
return null;
|
|
373
|
+
});
|
|
374
|
+
return this.embedderPromise;
|
|
375
|
+
}
|
|
376
|
+
logSemanticFallback(error) {
|
|
377
|
+
if (this.semanticFallbackLogged)
|
|
378
|
+
return;
|
|
379
|
+
this.semanticFallbackLogged = true;
|
|
380
|
+
const reason = error instanceof EmbeddingModelMissingError
|
|
381
|
+
? error.message
|
|
382
|
+
: error instanceof Error
|
|
383
|
+
? error.message
|
|
384
|
+
: String(error);
|
|
385
|
+
console.info(`[graphrag] semantic disabled, using keyword fallback: ${reason}`);
|
|
386
|
+
}
|
|
387
|
+
}
|
|
388
|
+
function toUint8Array(value) {
|
|
389
|
+
if (!value)
|
|
390
|
+
return null;
|
|
391
|
+
if (value instanceof Uint8Array)
|
|
392
|
+
return value;
|
|
393
|
+
if (value instanceof ArrayBuffer)
|
|
394
|
+
return new Uint8Array(value);
|
|
395
|
+
if (ArrayBuffer.isView(value)) {
|
|
396
|
+
return new Uint8Array(value.buffer, value.byteOffset, value.byteLength);
|
|
397
|
+
}
|
|
398
|
+
if (Array.isArray(value)) {
|
|
399
|
+
const bytes = value.map((entry) => Number(entry));
|
|
400
|
+
if (bytes.some((entry) => !Number.isFinite(entry)))
|
|
401
|
+
return null;
|
|
402
|
+
return Uint8Array.from(bytes);
|
|
403
|
+
}
|
|
404
|
+
return null;
|
|
237
405
|
}
|
|
238
406
|
//# sourceMappingURL=graph-rag.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"graph-rag.js","sourceRoot":"","sources":["../../src/graphrag/graph-rag.ts"],"names":[],"mappings":"AAAA,OAAO,SAAoC,MAAM,QAAQ,CAAC;AAC1D,OAAO,IAAI,MAAM,MAAM,CAAC;AACxB,OAAO,EAAE,MAAM,UAAU,CAAC;AAQ1B,OAAO,EAAE,gBAAgB,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAChE,OAAO,EAAE,QAAQ,IAAI,UAAU,EAAE,MAAM,IAAI,QAAQ,EAAE,MAAM,eAAe,CAAC;AAE3E,MAAM,UAAU,GAAG;;;;;;;;;;;;;;;;;;;;;;;;CAwBlB,CAAC;AAEF,MAAM,OAAO,QAAQ;IAKnB,YAAY,MAAc;QACxB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAED,KAAK,CAAC,IAAI;QACR,IAAI,CAAC,GAAG,GAAG,MAAM,SAAS,EAAE,CAAC;QAC7B,EAAE,CAAC,aAAa,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC;QAE5C,IAAI,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;YAC/B,MAAM,UAAU,GAAG,EAAE,CAAC,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAChD,IAAI,CAAC,EAAE,GAAG,IAAI,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;QAC9C,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,EAAE,GAAG,IAAI,IAAI,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;QACpC,CAAC;QAED,IAAI,CAAC,EAAE,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QACxB,IAAI,CAAC,IAAI,EAAE,CAAC;IACd,CAAC;IAEO,IAAI;QACV,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC;QAC9B,EAAE,CAAC,aAAa,CAAC,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;IACnD,CAAC;IAEO,GAAG,CAAC,GAAW,EAAE,SAAqC,EAAE;QAC9D,IAAI,CAAC,EAAE,CAAC,GAAG,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QACzB,IAAI,CAAC,IAAI,EAAE,CAAC;IACd,CAAC;IAEO,KAAK,CAAI,GAAW,EAAE,SAAqC,EAAE;QACnE,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QAClC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAClB,MAAM,OAAO,GAAQ,EAAE,CAAC;QACxB,OAAO,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC;YACnB,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAO,CAAC,CAAC;QACxC,CAAC;QACD,IAAI,CAAC,IAAI,EAAE,CAAC;QACZ,OAAO,OAAO,CAAC;IACjB,CAAC;IAEO,QAAQ,CAAI,GAAW,EAAE,SAAqC,EAAE;QACtE,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAI,GAAG,EAAE,MAAM,CAAC,CAAC;QAC3C,OAAO,OAAO,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC;IAC5B,CAAC;IAED,8EAA8E;IAE9E,OAAO,CAAC,IAAe;QACrB,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,mCAAmC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC;QAC/E,IAAI,QAAQ,EAAE,CAAC;YACb,IAAI,CAAC,GAAG,CACN,sGAAsG,EACtG,CAAC,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,WAAW,IAAI,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,QAAQ,IAAI,EAAE,CAAC,EAAE,IAAI,CAAC,EAAE,CAAC,CACrF,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,GAAG,CACN,mFAAmF,EACnF,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,WAAW,IAAI,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,QAAQ,IAAI,EAAE,CAAC,CAAC,CAChG,CAAC;QACJ,CAAC;IACH,CAAC;IAED,OAAO,CAAC,EAAU;QAChB,MAAM,GAAG,GAAG,IAAI,CAAC,QAAQ,CAA0B,kCAAkC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;QAC7F,IAAI,CAAC,GAAG;YAAE,OAAO,IAAI,CAAC;QACtB,OAAO,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;IAC7B,CAAC;IAED,UAAU,CAAC,IAAe,EAAE,KAAK,GAAG,EAAE;QACpC,MAAM,IAAI,GAAG,IAAI;YACf,CAAC,CAAC,IAAI,CAAC,KAAK,CAA0B,4CAA4C,EAAE,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;YAClG,CAAC,CAAC,IAAI,CAAC,KAAK,CAA0B,6BAA6B,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC;QAChF,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC;IAC5C,CAAC;IAED,WAAW,CAAC,QAAgB,EAAE,KAAK,GAAG,EAAE;QACtC,yDAAyD;QACzD,MAAM,OAAO,GAAG,IAAI,QAAQ,CAAC,WAAW,EAAE,GAAG,CAAC;QAC9C,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CACrB,oFAAoF,EACpF,CAAC,OAAO,EAAE,OAAO,EAAE,KAAK,CAAC,CAC1B,CAAC;QACF,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;YACxB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC;YACzB,KAAK,EAAE,GAAG;YACV,OAAO,EAAG,GAAG,CAAC,aAAa,CAAY,EAAE,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC;SAC3D,CAAC,CAAC,CAAC;IACN,CAAC;IAED,UAAU,CAAC,EAAU;QACnB,IAAI,CAAC,GAAG,CAAC,wDAAwD,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;QAC7E,IAAI,CAAC,GAAG,CAAC,gCAAgC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;IACnD,CAAC;IAED,8EAA8E;IAE9E,OAAO,CAAC,IAAe;QACrB,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,mCAAmC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC;QAC/E,IAAI,QAAQ,EAAE,CAAC;YACb,IAAI,CAAC,GAAG,CACN,kEAAkE,EAClE,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,MAAM,IAAI,GAAG,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,QAAQ,IAAI,EAAE,CAAC,EAAE,IAAI,CAAC,EAAE,CAAC,CAC9E,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,GAAG,CACN,gGAAgG,EAChG,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,MAAM,IAAI,GAAG,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,QAAQ,IAAI,EAAE,CAAC,CAAC,CAC5G,CAAC;QACJ,CAAC;IACH,CAAC;IAED,QAAQ,CAAC,MAAc,EAAE,YAAmC,MAAM;QAChE,IAAI,GAAG,GAAG,EAAE,CAAC;QACb,IAAI,MAAgB,CAAC;QACrB,IAAI,SAAS,KAAK,KAAK,EAAE,CAAC;YAAC,GAAG,GAAG,yCAAyC,CAAC;YAAC,MAAM,GAAG,CAAC,MAAM,CAAC,CAAC;QAAC,CAAC;aAC3F,IAAI,SAAS,KAAK,IAAI,EAAE,CAAC;YAAC,GAAG,GAAG,yCAAyC,CAAC;YAAC,MAAM,GAAG,CAAC,MAAM,CAAC,CAAC;QAAC,CAAC;aAC/F,CAAC;YAAC,GAAG,GAAG,0DAA0D,CAAC;YAAC,MAAM,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAAC,CAAC;QAErG,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAA0B,GAAG,EAAE,MAAM,CAAC,CAAC;QAC9D,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC;IAC5C,CAAC;IAED,mBAAmB,CAAC,MAAc,EAAE,KAAK,GAAG,CAAC;QAC3C,MAAM,OAAO,GAAG,IAAI,GAAG,EAAU,CAAC;QAClC,MAAM,MAAM,GAAgB,EAAE,CAAC;QAE/B,MAAM,QAAQ,GAAG,CAAC,EAAU,EAAE,YAAoB,EAAE,EAAE;YACpD,IAAI,YAAY,KAAK,CAAC,IAAI,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC;gBAAE,OAAO;YAClD,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YAEhB,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CACtB,iEAAiE,EACjE,CAAC,EAAE,CAAC,CACL,CAAC;YAEF,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBACzB,MAAM,QAAQ,GAAG,IAAI,CAAC,WAAW,CAAW,CAAC;gBAC7C,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;gBACpC,IAAI,IAAI,EAAE,CAAC;oBACT,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;oBAClB,QAAQ,CAAC,QAAQ,EAAE,YAAY,GAAG,CAAC,CAAC,CAAC;gBACvC,CAAC;YACH,CAAC;QACH,CAAC,CAAC;QAEF,QAAQ,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;QACxB,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,8EAA8E;IAE9E,QAAQ,CAAC,IAAqB;QAC5B,OAAO,UAAU,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;IAChC,CAAC;IAED,MAAM,CAAC,SAAiB,EAAE,IAAoB;QAC5C,OAAO,QAAQ,CAAC,IAAI,EAAE,SAAS,EAAE,IAAI,CAAC,CAAC;IACzC,CAAC;IAED,mBAAmB,CAAC,EAAU;QAC5B,MAAM,UAAU,GAAG,EAAE,CAAC,UAAU,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QACxF,MAAM,QAAQ,GAAG,eAAe,UAAU,EAAE,CAAC;QAC7C,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QACtC,IAAI,MAAM,EAAE,IAAI,KAAK,aAAa;YAAE,OAAO,MAAwB,CAAC;QAEpE,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,IAAI,CAAC,QAAQ,CACvB,0GAA0G,EAC1G,CAAC,UAAU,CAAC,CACb,CAAC;YACF,IAAI,GAAG;gBAAE,OAAO,IAAI,CAAC,SAAS,CAAC,GAAG,CAAmB,CAAC;QACxD,CAAC;QAAC,MAAM,CAAC;YACP,kDAAkD;QACpD,CAAC;QAED,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,UAAU,CAAC,aAAa,EAAE,KAAM,CAAC,EAAE,CAAC;YAC1D,IAAI,IAAI,CAAC,QAAQ,EAAE,aAAa,KAAK,UAAU;gBAAE,OAAO,IAAsB,CAAC;QACjF,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,aAAa;QACX,MAAM,aAAa,GAAG,IAAI,CAAC,QAAQ,CAAoB,qCAAqC,CAAC,CAAC;QAC9F,MAAM,aAAa,GAAG,IAAI,CAAC,QAAQ,CAAoB,qCAAqC,CAAC,CAAC;QAC9F,MAAM,UAAU,GAAG,aAAa,EAAE,KAAK,IAAI,CAAC,CAAC;QAC7C,MAAM,UAAU,GAAG,aAAa,EAAE,KAAK,IAAI,CAAC,CAAC;QAE7C,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAkC,yDAAyD,CAAC,CAAC;QACxH,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAkC,yDAAyD,CAAC,CAAC;QAExH,MAAM,WAAW,GAAG,gBAAgB,EAAE,CAAC;QACvC,MAAM,WAAW,GAAG,gBAAgB,EAAE,CAAC;QACvC,KAAK,MAAM,CAAC,IAAI,QAAQ,EAAE,CAAC;YACzB,MAAM,CAAC,GAAG,CAAC,CAAC,IAAgB,CAAC;YAC7B,IAAI,CAAC,IAAI,WAAW;gBAAE,WAAW,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC;QACjD,CAAC;QACD,KAAK,MAAM,CAAC,IAAI,QAAQ,EAAE,CAAC;YACzB,MAAM,CAAC,GAAG,CAAC,CAAC,IAAgB,CAAC;YAC7B,IAAI,CAAC,IAAI,WAAW;gBAAE,WAAW,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC;QACjD,CAAC;QAED,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,WAAW,EAAE,WAAW,EAAE,CAAC;IAC9D,CAAC;IAED,YAAY;QACV,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAA0B,qBAAqB,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC;QACvG,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAA0B,qBAAqB,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC;QACvG,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC;IAC1B,CAAC;IAED,cAAc,CAAC,IAAgD;QAC7D,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC;QAC3C,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC;IAC7C,CAAC;IAED,KAAK;QACH,IAAI,CAAC,IAAI,EAAE,CAAC;QACZ,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC;IAClB,CAAC;IAED,8EAA8E;IAEtE,SAAS,CAAC,GAA4B;QAC5C,OAAO;YACL,EAAE,EAAE,GAAG,CAAC,IAAI,CAAW;YACvB,IAAI,EAAE,GAAG,CAAC,MAAM,CAAa;YAC7B,KAAK,EAAE,GAAG,CAAC,OAAO,CAAW;YAC7B,WAAW,EAAE,GAAG,CAAC,aAAa,CAAuB;YACrD,QAAQ,EAAE,IAAI,CAAC,KAAK,CAAE,GAAG,CAAC,UAAU,CAAY,IAAI,IAAI,CAAC;YACzD,SAAS,EAAE,GAAG,CAAC,YAAY,CAAW;YACtC,SAAS,EAAE,GAAG,CAAC,YAAY,CAAW;SACvC,CAAC;IACJ,CAAC;IAEO,SAAS,CAAC,GAA4B;QAC5C,OAAO;YACL,EAAE,EAAE,GAAG,CAAC,IAAI,CAAW;YACvB,QAAQ,EAAE,GAAG,CAAC,WAAW,CAAW;YACpC,QAAQ,EAAE,GAAG,CAAC,WAAW,CAAW;YACpC,IAAI,EAAE,GAAG,CAAC,MAAM,CAAa;YAC7B,MAAM,EAAE,GAAG,CAAC,QAAQ,CAAW;YAC/B,QAAQ,EAAE,IAAI,CAAC,KAAK,CAAE,GAAG,CAAC,UAAU,CAAY,IAAI,IAAI,CAAC;SAC1D,CAAC;IACJ,CAAC;CACF"}
|
|
1
|
+
{"version":3,"file":"graph-rag.js","sourceRoot":"","sources":["../../src/graphrag/graph-rag.ts"],"names":[],"mappings":"AAAA,OAAO,SAAoC,MAAM,QAAQ,CAAC;AAC1D,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,IAAI,MAAM,MAAM,CAAC;AACxB,OAAO,EAAE,MAAM,UAAU,CAAC;AAQ1B,OAAO,EAAE,gBAAgB,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAChE,OAAO,EAAE,QAAQ,IAAI,UAAU,EAAE,MAAM,IAAI,QAAQ,EAAE,MAAM,eAAe,CAAC;AAC3E,OAAO,EAAE,0BAA0B,EAAiB,YAAY,EAAE,MAAM,iBAAiB,CAAC;AAC1F,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAC3C,OAAO,EAAE,mBAAmB,EAAE,MAAM,2BAA2B,CAAC;AAIhE,MAAM,UAAU,GAAG;;;;;;;;;;;;;;;;;;;;;;;;;CAyBlB,CAAC;AAGF,MAAM,iBAAiB,GAAG,iEAAiE,CAAC;AAS5F,MAAM,OAAO,QAAQ;IAQnB,YAAY,MAAc;QAJlB,wBAAmB,GAAkC,IAAI,CAAC;QAC1D,oBAAe,GAAoC,IAAI,CAAC;QACxD,2BAAsB,GAAG,KAAK,CAAC;QAGrC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAED,KAAK,CAAC,IAAI;QACR,IAAI,CAAC,GAAG,GAAG,MAAM,SAAS,EAAE,CAAC;QAC7B,EAAE,CAAC,aAAa,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC;QAE5C,IAAI,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;YAC/B,MAAM,UAAU,GAAG,EAAE,CAAC,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAChD,IAAI,CAAC,EAAE,GAAG,IAAI,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;QAC9C,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,EAAE,GAAG,IAAI,IAAI,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;QACpC,CAAC;QAED,IAAI,CAAC,EAAE,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QACxB,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAC1B,IAAI,CAAC,IAAI,EAAE,CAAC;IACd,CAAC;IAEO,IAAI;QACV,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC;QAC9B,EAAE,CAAC,aAAa,CAAC,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;IACnD,CAAC;IAEO,GAAG,CAAC,GAAW,EAAE,SAAqB,EAAE;QAC9C,IAAI,CAAC,EAAE,CAAC,GAAG,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QACzB,IAAI,CAAC,IAAI,EAAE,CAAC;IACd,CAAC;IAEO,KAAK,CAAI,GAAW,EAAE,SAAqB,EAAE;QACnD,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QAClC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAClB,MAAM,OAAO,GAAQ,EAAE,CAAC;QACxB,OAAO,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC;YACnB,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAO,CAAC,CAAC;QACxC,CAAC;QACD,IAAI,CAAC,IAAI,EAAE,CAAC;QACZ,OAAO,OAAO,CAAC;IACjB,CAAC;IAEO,QAAQ,CAAI,GAAW,EAAE,SAAqB,EAAE;QACtD,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAI,GAAG,EAAE,MAAM,CAAC,CAAC;QAC3C,OAAO,OAAO,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC;IAC5B,CAAC;IAEO,kBAAkB;QACxB,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAmB,0BAA0B,CAAC,CAAC;QACzE,MAAM,eAAe,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC;QAC3E,IAAI,CAAC,eAAe,EAAE,CAAC;YACrB,IAAI,CAAC,EAAE,CAAC,GAAG,CAAC,0CAA0C,CAAC,CAAC;QAC1D,CAAC;IACH,CAAC;IAED,8EAA8E;IAE9E,OAAO,CAAC,IAAe;QACrB,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,mCAAmC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC;QAC/E,MAAM,UAAU,GAAG,IAAI,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC7F,IAAI,QAAQ,EAAE,CAAC;YACb,IAAI,UAAU,KAAK,SAAS,EAAE,CAAC;gBAC7B,IAAI,CAAC,GAAG,CACN,sGAAsG,EACtG,CAAC,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,WAAW,IAAI,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,QAAQ,IAAI,EAAE,CAAC,EAAE,IAAI,CAAC,EAAE,CAAC,CACrF,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACN,IAAI,CAAC,GAAG,CACN,kHAAkH,EAClH,CAAC,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,WAAW,IAAI,IAAI,EAAE,UAAU,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,QAAQ,IAAI,EAAE,CAAC,EAAE,IAAI,CAAC,EAAE,CAAC,CACjG,CAAC;YACJ,CAAC;QACH,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,GAAG,CACN,8FAA8F,EAC9F;gBACE,IAAI,CAAC,EAAE;gBACP,IAAI,CAAC,IAAI;gBACT,IAAI,CAAC,KAAK;gBACV,IAAI,CAAC,WAAW,IAAI,IAAI;gBACxB,UAAU,IAAI,IAAI;gBAClB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,QAAQ,IAAI,EAAE,CAAC;aACpC,CACF,CAAC;QACJ,CAAC;IACH,CAAC;IAED,OAAO,CAAC,EAAU;QAChB,MAAM,GAAG,GAAG,IAAI,CAAC,QAAQ,CAA0B,kCAAkC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;QAC7F,IAAI,CAAC,GAAG;YAAE,OAAO,IAAI,CAAC;QACtB,OAAO,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;IAC7B,CAAC;IAED,UAAU,CAAC,IAAe,EAAE,KAAK,GAAG,EAAE;QACpC,MAAM,IAAI,GAAG,IAAI;YACf,CAAC,CAAC,IAAI,CAAC,KAAK,CAA0B,4CAA4C,EAAE,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;YAClG,CAAC,CAAC,IAAI,CAAC,KAAK,CAA0B,6BAA6B,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC;QAChF,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC;IAC5C,CAAC;IAED,WAAW,CAAC,QAAgB,EAAE,KAAK,GAAG,EAAE;QACtC,OAAO,IAAI,CAAC,kBAAkB,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;IAClD,CAAC;IAED,KAAK,CAAC,iBAAiB,CAAC,QAAgB,EAAE,KAAK,GAAG,EAAE;QAClD,MAAM,aAAa,GAAG,IAAI,CAAC,kBAAkB,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QAC/D,MAAM,QAAQ,GAAG,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAC1C,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC;YACtB,OAAO,aAAa,CAAC;QACvB,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;QACvD,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,OAAO,aAAa,CAAC;QACvB,CAAC;QAED,IAAI,CAAC;YACH,MAAM,WAAW,GAAG,IAAI,CAAC,wBAAwB,CAAC,CAAC,KAAK,EAAE,WAAW,GAAG,EAAE,EAAE,EAAE,CAC5E,IAAI,CAAC,kBAAkB,CAAC,KAAK,EAAE,WAAW,CAAC,CAC5C,CAAC;YACF,OAAO,MAAM,YAAY,CAAC,WAAW,EAAE,QAAQ,EAAE,QAAQ,EAAE,EAAE,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC;QAChG,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,CAAC;YAChC,OAAO,aAAa,CAAC;QACvB,CAAC;IACH,CAAC;IAEO,kBAAkB,CAAC,QAAgB,EAAE,KAAK,GAAG,EAAE;QACrD,yDAAyD;QACzD,MAAM,OAAO,GAAG,IAAI,QAAQ,CAAC,WAAW,EAAE,GAAG,CAAC;QAC9C,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CACrB,oFAAoF,EACpF,CAAC,OAAO,EAAE,OAAO,EAAE,KAAK,CAAC,CAC1B,CAAC;QACF,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;YACxB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC;YACzB,KAAK,EAAE,GAAG;YACV,OAAO,EAAG,GAAG,CAAC,aAAa,CAAY,EAAE,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC;SAC3D,CAAC,CAAC,CAAC;IACN,CAAC;IAED,UAAU,CAAC,EAAU;QACnB,IAAI,CAAC,GAAG,CAAC,wDAAwD,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;QAC7E,IAAI,CAAC,GAAG,CAAC,gCAAgC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;IACnD,CAAC;IAED,8EAA8E;IAE9E,OAAO,CAAC,IAAe;QACrB,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,mCAAmC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC;QAC/E,IAAI,QAAQ,EAAE,CAAC;YACb,IAAI,CAAC,GAAG,CACN,kEAAkE,EAClE,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,MAAM,IAAI,GAAG,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,QAAQ,IAAI,EAAE,CAAC,EAAE,IAAI,CAAC,EAAE,CAAC,CAC9E,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,GAAG,CACN,gGAAgG,EAChG,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,MAAM,IAAI,GAAG,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,QAAQ,IAAI,EAAE,CAAC,CAAC,CAC5G,CAAC;QACJ,CAAC;IACH,CAAC;IAED,QAAQ,CAAC,MAAc,EAAE,YAAmC,MAAM;QAChE,IAAI,GAAG,GAAG,EAAE,CAAC;QACb,IAAI,MAAgB,CAAC;QACrB,IAAI,SAAS,KAAK,KAAK,EAAE,CAAC;YAAC,GAAG,GAAG,yCAAyC,CAAC;YAAC,MAAM,GAAG,CAAC,MAAM,CAAC,CAAC;QAAC,CAAC;aAC3F,IAAI,SAAS,KAAK,IAAI,EAAE,CAAC;YAAC,GAAG,GAAG,yCAAyC,CAAC;YAAC,MAAM,GAAG,CAAC,MAAM,CAAC,CAAC;QAAC,CAAC;aAC/F,CAAC;YAAC,GAAG,GAAG,0DAA0D,CAAC;YAAC,MAAM,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAAC,CAAC;QAErG,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAA0B,GAAG,EAAE,MAAM,CAAC,CAAC;QAC9D,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC;IAC5C,CAAC;IAED,mBAAmB,CAAC,MAAc,EAAE,KAAK,GAAG,CAAC;QAC3C,MAAM,OAAO,GAAG,IAAI,GAAG,EAAU,CAAC;QAClC,MAAM,MAAM,GAAgB,EAAE,CAAC;QAE/B,MAAM,QAAQ,GAAG,CAAC,EAAU,EAAE,YAAoB,EAAE,EAAE;YACpD,IAAI,YAAY,KAAK,CAAC,IAAI,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC;gBAAE,OAAO;YAClD,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YAEhB,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CACtB,iEAAiE,EACjE,CAAC,EAAE,CAAC,CACL,CAAC;YAEF,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBACzB,MAAM,QAAQ,GAAG,IAAI,CAAC,WAAW,CAAW,CAAC;gBAC7C,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;gBACpC,IAAI,IAAI,EAAE,CAAC;oBACT,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;oBAClB,QAAQ,CAAC,QAAQ,EAAE,YAAY,GAAG,CAAC,CAAC,CAAC;gBACvC,CAAC;YACH,CAAC;QACH,CAAC,CAAC;QAEF,QAAQ,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;QACxB,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,8EAA8E;IAE9E,QAAQ,CAAC,IAAqB;QAC5B,OAAO,UAAU,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;IAChC,CAAC;IAED,MAAM,CAAC,SAAiB,EAAE,IAAoB;QAC5C,OAAO,QAAQ,CAAC,IAAI,EAAE,SAAS,EAAE,IAAI,CAAC,CAAC;IACzC,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,SAAiB,EAAE,IAAoB;QACxD,MAAM,QAAQ,GAAG,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAC1C,IAAI,CAAC,QAAQ,CAAC,OAAO,IAAI,iBAAiB,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC;YAC3D,OAAO,QAAQ,CAAC,IAAI,EAAE,SAAS,EAAE,IAAI,CAAC,CAAC;QACzC,CAAC;QAED,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC;QAChE,MAAM,aAAa,GAAG,YAAY,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QAC/C,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC/B,OAAO,QAAQ,CAAC,IAAI,EAAE,SAAS,EAAE,IAAI,CAAC,CAAC;QACzC,CAAC;QAED,MAAM,UAAU,GAAG,IAAI,CAAC,wBAAwB,CAAC,CAAC,KAAK,EAAE,KAAK,GAAG,EAAE,EAAE,EAAE;YACrE,IAAI,KAAK,KAAK,SAAS;gBAAE,OAAO,aAAa,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;YAC9D,OAAO,IAAI,CAAC,kBAAkB,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;QACH,OAAO,QAAQ,CAAC,UAAU,EAAE,SAAS,EAAE,IAAI,CAAC,CAAC;IAC/C,CAAC;IAED,mBAAmB,CAAC,EAAU;QAC5B,MAAM,UAAU,GAAG,EAAE,CAAC,UAAU,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QACxF,MAAM,QAAQ,GAAG,eAAe,UAAU,EAAE,CAAC;QAC7C,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QACtC,IAAI,MAAM,EAAE,IAAI,KAAK,aAAa;YAAE,OAAO,MAAwB,CAAC;QAEpE,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,IAAI,CAAC,QAAQ,CACvB,0GAA0G,EAC1G,CAAC,UAAU,CAAC,CACb,CAAC;YACF,IAAI,GAAG;gBAAE,OAAO,IAAI,CAAC,SAAS,CAAC,GAAG,CAAmB,CAAC;QACxD,CAAC;QAAC,MAAM,CAAC;YACP,kDAAkD;QACpD,CAAC;QAED,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,UAAU,CAAC,aAAa,EAAE,KAAM,CAAC,EAAE,CAAC;YAC1D,IAAI,IAAI,CAAC,QAAQ,EAAE,aAAa,KAAK,UAAU;gBAAE,OAAO,IAAsB,CAAC;QACjF,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,WAAW;QACT,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CACrB,uDAAuD,CACxD,CAAC;QACF,MAAM,OAAO,GAAgB,EAAE,CAAC;QAChC,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;YACvB,MAAM,OAAO,GAAG,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;YACnD,IAAI,CAAC,OAAO,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC;gBAAE,SAAS;YAC/C,OAAO,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC;QAC3C,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,aAAa;QACX,MAAM,aAAa,GAAG,IAAI,CAAC,QAAQ,CAAoB,qCAAqC,CAAC,CAAC;QAC9F,MAAM,aAAa,GAAG,IAAI,CAAC,QAAQ,CAAoB,qCAAqC,CAAC,CAAC;QAC9F,MAAM,UAAU,GAAG,aAAa,EAAE,KAAK,IAAI,CAAC,CAAC;QAC7C,MAAM,UAAU,GAAG,aAAa,EAAE,KAAK,IAAI,CAAC,CAAC;QAE7C,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAkC,yDAAyD,CAAC,CAAC;QACxH,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAkC,yDAAyD,CAAC,CAAC;QAExH,MAAM,WAAW,GAAG,gBAAgB,EAAE,CAAC;QACvC,MAAM,WAAW,GAAG,gBAAgB,EAAE,CAAC;QACvC,KAAK,MAAM,CAAC,IAAI,QAAQ,EAAE,CAAC;YACzB,MAAM,CAAC,GAAG,CAAC,CAAC,IAAgB,CAAC;YAC7B,IAAI,CAAC,IAAI,WAAW;gBAAE,WAAW,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC;QACjD,CAAC;QACD,KAAK,MAAM,CAAC,IAAI,QAAQ,EAAE,CAAC;YACzB,MAAM,CAAC,GAAG,CAAC,CAAC,IAAgB,CAAC;YAC7B,IAAI,CAAC,IAAI,WAAW;gBAAE,WAAW,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC;QACjD,CAAC;QAED,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,WAAW,EAAE,WAAW,EAAE,CAAC;IAC9D,CAAC;IAED,YAAY;QACV,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAA0B,qBAAqB,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC;QACvG,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAA0B,qBAAqB,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC;QACvG,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC;IAC1B,CAAC;IAED,cAAc,CAAC,IAAgD;QAC7D,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC;QAC3C,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC;IAC7C,CAAC;IAED,KAAK;QACH,IAAI,CAAC,IAAI,EAAE,CAAC;QACZ,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC;IAClB,CAAC;IAED,8EAA8E;IAEtE,SAAS,CAAC,GAA4B;QAC5C,MAAM,MAAM,GAAG,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC;QACrD,OAAO;YACL,EAAE,EAAE,GAAG,CAAC,IAAI,CAAW;YACvB,IAAI,EAAE,GAAG,CAAC,MAAM,CAAa;YAC7B,KAAK,EAAE,GAAG,CAAC,OAAO,CAAW;YAC7B,WAAW,EAAE,GAAG,CAAC,aAAa,CAAuB;YACrD,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,SAAS;YAC/C,QAAQ,EAAE,IAAI,CAAC,KAAK,CAAE,GAAG,CAAC,UAAU,CAAY,IAAI,IAAI,CAAC;YACzD,SAAS,EAAE,GAAG,CAAC,YAAY,CAAW;YACtC,SAAS,EAAE,GAAG,CAAC,YAAY,CAAW;SACvC,CAAC;IACJ,CAAC;IAEO,SAAS,CAAC,GAA4B;QAC5C,OAAO;YACL,EAAE,EAAE,GAAG,CAAC,IAAI,CAAW;YACvB,QAAQ,EAAE,GAAG,CAAC,WAAW,CAAW;YACpC,QAAQ,EAAE,GAAG,CAAC,WAAW,CAAW;YACpC,IAAI,EAAE,GAAG,CAAC,MAAM,CAAa;YAC7B,MAAM,EAAE,GAAG,CAAC,QAAQ,CAAW;YAC/B,QAAQ,EAAE,IAAI,CAAC,KAAK,CAAE,GAAG,CAAC,UAAU,CAAY,IAAI,IAAI,CAAC;SAC1D,CAAC;IACJ,CAAC;IAEO,eAAe,CAAC,MAAyB;QAC/C,MAAM,WAAW,GAAG,IAAI,YAAY,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAC3E,OAAO,IAAI,UAAU,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;IAC5C,CAAC;IAEO,iBAAiB,CAAC,KAAc;QACtC,MAAM,KAAK,GAAG,YAAY,CAAC,KAAK,CAAC,CAAC;QAClC,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,UAAU,KAAK,CAAC,IAAI,KAAK,CAAC,UAAU,GAAG,YAAY,CAAC,iBAAiB,KAAK,CAAC,EAAE,CAAC;YAChG,OAAO,IAAI,CAAC;QACd,CAAC;QACD,MAAM,IAAI,GAAG,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACpC,OAAO,IAAI,YAAY,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,UAAU,GAAG,YAAY,CAAC,iBAAiB,CAAC,CAAC;IAC1G,CAAC;IAEO,wBAAwB,CAC9B,MAAyD;QAEzD,MAAM,IAAI,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,CAAmB,CAAC;QACnD,IAAI,CAAC,WAAW,GAAG,MAAM,CAAC;QAC1B,OAAO,IAAI,CAAC;IACd,CAAC;IAEO,iBAAiB;QACvB,IAAI,IAAI,CAAC,mBAAmB,EAAE,CAAC;YAC7B,OAAO,IAAI,CAAC,mBAAmB,CAAC;QAClC,CAAC;QAED,MAAM,WAAW,GAAG,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAC9C,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,kBAAkB,CAAC,CAAC;QAC9D,MAAM,SAAS,GACb,EAAE,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC,CAAC,CAAE,EAAE,CAAC,YAAY,CAAC,UAAU,CAAa,CAAC,CAAC,CAAC,EAAE,CAAC;QAC5E,MAAM,MAAM,GAAG,mBAAmB,CAAC,SAAS,CAAC,CAAC;QAC9C,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;QAClC,MAAM,SAAS,GACb,OAAO,MAAM,CAAC,SAAS,KAAK,QAAQ,IAAI,MAAM,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC;YACxE,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,IAAI,EAAE;YACzB,CAAC,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,SAAS,KAAK,EAAE,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAClE,IAAI,CAAC,mBAAmB,GAAG;YACzB,OAAO,EAAE,MAAM,CAAC,OAAO,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC;YAC3C,KAAK;YACL,SAAS;YACT,IAAI,EAAE,MAAM,CAAC,IAAI;SAClB,CAAC;QACF,OAAO,IAAI,CAAC,mBAAmB,CAAC;IAClC,CAAC;IAEO,kBAAkB;QACxB,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACxC,IAAI,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,KAAK,OAAO,EAAE,CAAC;YACrC,OAAO,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QAC7B,CAAC;QACD,OAAO,OAAO,CAAC,GAAG,EAAE,CAAC;IACvB,CAAC;IAEO,KAAK,CAAC,gBAAgB,CAAC,QAAgC;QAC7D,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;YACzB,OAAO,IAAI,CAAC,eAAe,CAAC;QAC9B,CAAC;QACD,IAAI,CAAC,eAAe,GAAG,YAAY,CAAC;YAClC,KAAK,EAAE,QAAQ,CAAC,KAAK;YACrB,SAAS,EAAE,QAAQ,CAAC,SAAS;SAC9B,CAAC,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;YACjB,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,CAAC;YAChC,OAAO,IAAI,CAAC;QACd,CAAC,CAAC,CAAC;QACH,OAAO,IAAI,CAAC,eAAe,CAAC;IAC9B,CAAC;IAEO,mBAAmB,CAAC,KAAc;QACxC,IAAI,IAAI,CAAC,sBAAsB;YAAE,OAAO;QACxC,IAAI,CAAC,sBAAsB,GAAG,IAAI,CAAC;QAEnC,MAAM,MAAM,GACV,KAAK,YAAY,0BAA0B;YACzC,CAAC,CAAC,KAAK,CAAC,OAAO;YACf,CAAC,CAAC,KAAK,YAAY,KAAK;gBACtB,CAAC,CAAC,KAAK,CAAC,OAAO;gBACf,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QACtB,OAAO,CAAC,IAAI,CAAC,yDAAyD,MAAM,EAAE,CAAC,CAAC;IAClF,CAAC;CACF;AAED,SAAS,YAAY,CAAC,KAAc;IAClC,IAAI,CAAC,KAAK;QAAE,OAAO,IAAI,CAAC;IACxB,IAAI,KAAK,YAAY,UAAU;QAAE,OAAO,KAAK,CAAC;IAC9C,IAAI,KAAK,YAAY,WAAW;QAAE,OAAO,IAAI,UAAU,CAAC,KAAK,CAAC,CAAC;IAC/D,IAAI,WAAW,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC;QAC9B,OAAO,IAAI,UAAU,CAAC,KAAK,CAAC,MAAM,EAAE,KAAK,CAAC,UAAU,EAAE,KAAK,CAAC,UAAU,CAAC,CAAC;IAC1E,CAAC;IACD,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QACzB,MAAM,KAAK,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;QAClD,IAAI,KAAK,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;YAAE,OAAO,IAAI,CAAC;QAChE,OAAO,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAChC,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC"}
|
|
@@ -0,0 +1,8 @@
|
|
|
1
|
+
import type { Embedder } from './embeddings.js';
|
|
2
|
+
import type { KnowledgeGraph, SearchResult } from './knowledge-graph.js';
|
|
3
|
+
export interface HybridOptions {
|
|
4
|
+
readonly k?: number;
|
|
5
|
+
readonly rrfK?: number;
|
|
6
|
+
}
|
|
7
|
+
export declare function hybridSearch(graph: KnowledgeGraph, embedder: Embedder | null, query: string, opts?: HybridOptions): Promise<SearchResult[]>;
|
|
8
|
+
//# sourceMappingURL=hybrid.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"hybrid.d.ts","sourceRoot":"","sources":["../../src/graphrag/hybrid.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,iBAAiB,CAAC;AAChD,OAAO,KAAK,EAAwB,cAAc,EAAE,YAAY,EAAE,MAAM,sBAAsB,CAAC;AAQ/F,MAAM,WAAW,aAAa;IAC5B,QAAQ,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC;IACpB,QAAQ,CAAC,IAAI,CAAC,EAAE,MAAM,CAAC;CACxB;AA8LD,wBAAsB,YAAY,CAChC,KAAK,EAAE,cAAc,EACrB,QAAQ,EAAE,QAAQ,GAAG,IAAI,EACzB,KAAK,EAAE,MAAM,EACb,IAAI,GAAE,aAAkB,GACvB,OAAO,CAAC,YAAY,EAAE,CAAC,CA0CzB"}
|