@roarpeng/graphflow 0.6.5
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/AGENTS.md +58 -0
- package/CHANGELOG.md +205 -0
- package/CLAUDE.md +39 -0
- package/LICENSE +17 -0
- package/README.md +434 -0
- package/dist/agents/brainstormer.d.ts +5 -0
- package/dist/agents/brainstormer.d.ts.map +1 -0
- package/dist/agents/brainstormer.js +61 -0
- package/dist/agents/brainstormer.js.map +1 -0
- package/dist/agents/planner.d.ts +14 -0
- package/dist/agents/planner.d.ts.map +1 -0
- package/dist/agents/planner.js +166 -0
- package/dist/agents/planner.js.map +1 -0
- package/dist/agents/validator.d.ts +6 -0
- package/dist/agents/validator.d.ts.map +1 -0
- package/dist/agents/validator.js +133 -0
- package/dist/agents/validator.js.map +1 -0
- package/dist/agents/worker.d.ts +18 -0
- package/dist/agents/worker.d.ts.map +1 -0
- package/dist/agents/worker.js +34 -0
- package/dist/agents/worker.js.map +1 -0
- package/dist/config/defaults.d.ts +16 -0
- package/dist/config/defaults.d.ts.map +1 -0
- package/dist/config/defaults.js +85 -0
- package/dist/config/defaults.js.map +1 -0
- package/dist/config/embedding-factory.d.ts +4 -0
- package/dist/config/embedding-factory.d.ts.map +1 -0
- package/dist/config/embedding-factory.js +37 -0
- package/dist/config/embedding-factory.js.map +1 -0
- package/dist/config/loader.d.ts +4 -0
- package/dist/config/loader.d.ts.map +1 -0
- package/dist/config/loader.js +194 -0
- package/dist/config/loader.js.map +1 -0
- package/dist/config/merge.d.ts +6 -0
- package/dist/config/merge.d.ts.map +1 -0
- package/dist/config/merge.js +130 -0
- package/dist/config/merge.js.map +1 -0
- package/dist/config/paths.d.ts +6 -0
- package/dist/config/paths.d.ts.map +1 -0
- package/dist/config/paths.js +32 -0
- package/dist/config/paths.js.map +1 -0
- package/dist/config/resolve.d.ts +4 -0
- package/dist/config/resolve.d.ts.map +1 -0
- package/dist/config/resolve.js +45 -0
- package/dist/config/resolve.js.map +1 -0
- package/dist/config/scaffold.d.ts +11 -0
- package/dist/config/scaffold.d.ts.map +1 -0
- package/dist/config/scaffold.js +42 -0
- package/dist/config/scaffold.js.map +1 -0
- package/dist/config/schema.d.ts +102 -0
- package/dist/config/schema.d.ts.map +1 -0
- package/dist/config/schema.js +3 -0
- package/dist/config/schema.js.map +1 -0
- package/dist/config/secrets.d.ts +9 -0
- package/dist/config/secrets.d.ts.map +1 -0
- package/dist/config/secrets.js +55 -0
- package/dist/config/secrets.js.map +1 -0
- package/dist/core/dag-engine.d.ts +14 -0
- package/dist/core/dag-engine.d.ts.map +1 -0
- package/dist/core/dag-engine.js +97 -0
- package/dist/core/dag-engine.js.map +1 -0
- package/dist/core/errors.d.ts +15 -0
- package/dist/core/errors.d.ts.map +1 -0
- package/dist/core/errors.js +29 -0
- package/dist/core/errors.js.map +1 -0
- package/dist/core/orchestrator.d.ts +31 -0
- package/dist/core/orchestrator.d.ts.map +1 -0
- package/dist/core/orchestrator.js +321 -0
- package/dist/core/orchestrator.js.map +1 -0
- package/dist/core/state-machine.d.ts +14 -0
- package/dist/core/state-machine.d.ts.map +1 -0
- package/dist/core/state-machine.js +77 -0
- package/dist/core/state-machine.js.map +1 -0
- package/dist/core/triage.d.ts +6 -0
- package/dist/core/triage.d.ts.map +1 -0
- package/dist/core/triage.js +58 -0
- package/dist/core/triage.js.map +1 -0
- package/dist/core/types.d.ts +61 -0
- package/dist/core/types.d.ts.map +1 -0
- package/dist/core/types.js +3 -0
- package/dist/core/types.js.map +1 -0
- package/dist/graph/client-factory.d.ts +22 -0
- package/dist/graph/client-factory.d.ts.map +1 -0
- package/dist/graph/client-factory.js +60 -0
- package/dist/graph/client-factory.js.map +1 -0
- package/dist/graph/context-slicer.d.ts +47 -0
- package/dist/graph/context-slicer.d.ts.map +1 -0
- package/dist/graph/context-slicer.js +246 -0
- package/dist/graph/context-slicer.js.map +1 -0
- package/dist/graph/file-indexer.d.ts +15 -0
- package/dist/graph/file-indexer.d.ts.map +1 -0
- package/dist/graph/file-indexer.js +343 -0
- package/dist/graph/file-indexer.js.map +1 -0
- package/dist/graph/graph-indexer.d.ts +7 -0
- package/dist/graph/graph-indexer.d.ts.map +1 -0
- package/dist/graph/graph-indexer.js +32 -0
- package/dist/graph/graph-indexer.js.map +1 -0
- package/dist/graph/graphify-client.d.ts +30 -0
- package/dist/graph/graphify-client.d.ts.map +1 -0
- package/dist/graph/graphify-client.js +186 -0
- package/dist/graph/graphify-client.js.map +1 -0
- package/dist/graph/graphify-file-client.d.ts +26 -0
- package/dist/graph/graphify-file-client.d.ts.map +1 -0
- package/dist/graph/graphify-file-client.js +140 -0
- package/dist/graph/graphify-file-client.js.map +1 -0
- package/dist/graph/graphify-mcp-client.d.ts +18 -0
- package/dist/graph/graphify-mcp-client.d.ts.map +1 -0
- package/dist/graph/graphify-mcp-client.js +73 -0
- package/dist/graph/graphify-mcp-client.js.map +1 -0
- package/dist/graph/language-indexers/c-cpp.d.ts +3 -0
- package/dist/graph/language-indexers/c-cpp.d.ts.map +1 -0
- package/dist/graph/language-indexers/c-cpp.js +121 -0
- package/dist/graph/language-indexers/c-cpp.js.map +1 -0
- package/dist/graph/language-indexers/go.d.ts +3 -0
- package/dist/graph/language-indexers/go.d.ts.map +1 -0
- package/dist/graph/language-indexers/go.js +97 -0
- package/dist/graph/language-indexers/go.js.map +1 -0
- package/dist/graph/language-indexers/index.d.ts +29 -0
- package/dist/graph/language-indexers/index.d.ts.map +1 -0
- package/dist/graph/language-indexers/index.js +27 -0
- package/dist/graph/language-indexers/index.js.map +1 -0
- package/dist/graph/language-indexers/python.d.ts +3 -0
- package/dist/graph/language-indexers/python.d.ts.map +1 -0
- package/dist/graph/language-indexers/python.js +91 -0
- package/dist/graph/language-indexers/python.js.map +1 -0
- package/dist/graph/language-indexers/rust.d.ts +3 -0
- package/dist/graph/language-indexers/rust.d.ts.map +1 -0
- package/dist/graph/language-indexers/rust.js +99 -0
- package/dist/graph/language-indexers/rust.js.map +1 -0
- package/dist/graph/language-indexers/tree-sitter-loader.d.ts +18 -0
- package/dist/graph/language-indexers/tree-sitter-loader.d.ts.map +1 -0
- package/dist/graph/language-indexers/tree-sitter-loader.js +34 -0
- package/dist/graph/language-indexers/tree-sitter-loader.js.map +1 -0
- package/dist/graph/language-indexers/typescript.d.ts +3 -0
- package/dist/graph/language-indexers/typescript.d.ts.map +1 -0
- package/dist/graph/language-indexers/typescript.js +194 -0
- package/dist/graph/language-indexers/typescript.js.map +1 -0
- package/dist/graph/semantic-enricher.d.ts +17 -0
- package/dist/graph/semantic-enricher.d.ts.map +1 -0
- package/dist/graph/semantic-enricher.js +97 -0
- package/dist/graph/semantic-enricher.js.map +1 -0
- package/dist/graph/sqlite-client.d.ts +24 -0
- package/dist/graph/sqlite-client.d.ts.map +1 -0
- package/dist/graph/sqlite-client.js +215 -0
- package/dist/graph/sqlite-client.js.map +1 -0
- package/dist/hooks/post-run-sync.d.ts +7 -0
- package/dist/hooks/post-run-sync.d.ts.map +1 -0
- package/dist/hooks/post-run-sync.js +26 -0
- package/dist/hooks/post-run-sync.js.map +1 -0
- package/dist/index.d.ts +10 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +42 -0
- package/dist/index.js.map +1 -0
- package/dist/integrations/agent-mcp-installer.d.ts +50 -0
- package/dist/integrations/agent-mcp-installer.d.ts.map +1 -0
- package/dist/integrations/agent-mcp-installer.js +414 -0
- package/dist/integrations/agent-mcp-installer.js.map +1 -0
- package/dist/learning/canary-gate.d.ts +6 -0
- package/dist/learning/canary-gate.d.ts.map +1 -0
- package/dist/learning/canary-gate.js +13 -0
- package/dist/learning/canary-gate.js.map +1 -0
- package/dist/learning/embeddings.d.ts +19 -0
- package/dist/learning/embeddings.d.ts.map +1 -0
- package/dist/learning/embeddings.js +160 -0
- package/dist/learning/embeddings.js.map +1 -0
- package/dist/learning/episodic-memory.d.ts +36 -0
- package/dist/learning/episodic-memory.d.ts.map +1 -0
- package/dist/learning/episodic-memory.js +228 -0
- package/dist/learning/episodic-memory.js.map +1 -0
- package/dist/learning/exporter.d.ts +10 -0
- package/dist/learning/exporter.d.ts.map +1 -0
- package/dist/learning/exporter.js +25 -0
- package/dist/learning/exporter.js.map +1 -0
- package/dist/learning/feedback-collector.d.ts +12 -0
- package/dist/learning/feedback-collector.d.ts.map +1 -0
- package/dist/learning/feedback-collector.js +14 -0
- package/dist/learning/feedback-collector.js.map +1 -0
- package/dist/learning/learning-events.d.ts +4 -0
- package/dist/learning/learning-events.d.ts.map +1 -0
- package/dist/learning/learning-events.js +35 -0
- package/dist/learning/learning-events.js.map +1 -0
- package/dist/learning/local-embedding.d.ts +3 -0
- package/dist/learning/local-embedding.d.ts.map +1 -0
- package/dist/learning/local-embedding.js +21 -0
- package/dist/learning/local-embedding.js.map +1 -0
- package/dist/learning/nightly-trainer.d.ts +14 -0
- package/dist/learning/nightly-trainer.d.ts.map +1 -0
- package/dist/learning/nightly-trainer.js +56 -0
- package/dist/learning/nightly-trainer.js.map +1 -0
- package/dist/learning/reflector.d.ts +19 -0
- package/dist/learning/reflector.d.ts.map +1 -0
- package/dist/learning/reflector.js +130 -0
- package/dist/learning/reflector.js.map +1 -0
- package/dist/learning/sample-builder.d.ts +7 -0
- package/dist/learning/sample-builder.d.ts.map +1 -0
- package/dist/learning/sample-builder.js +10 -0
- package/dist/learning/sample-builder.js.map +1 -0
- package/dist/learning/skill-flywheel.d.ts +52 -0
- package/dist/learning/skill-flywheel.d.ts.map +1 -0
- package/dist/learning/skill-flywheel.js +533 -0
- package/dist/learning/skill-flywheel.js.map +1 -0
- package/dist/learning/vector-store.d.ts +18 -0
- package/dist/learning/vector-store.d.ts.map +1 -0
- package/dist/learning/vector-store.js +73 -0
- package/dist/learning/vector-store.js.map +1 -0
- package/dist/routing/model-router.d.ts +13 -0
- package/dist/routing/model-router.d.ts.map +1 -0
- package/dist/routing/model-router.js +113 -0
- package/dist/routing/model-router.js.map +1 -0
- package/dist/routing/provider-adapters/anthropic.d.ts +3 -0
- package/dist/routing/provider-adapters/anthropic.d.ts.map +1 -0
- package/dist/routing/provider-adapters/anthropic.js +57 -0
- package/dist/routing/provider-adapters/anthropic.js.map +1 -0
- package/dist/routing/provider-adapters/bailian.d.ts +3 -0
- package/dist/routing/provider-adapters/bailian.d.ts.map +1 -0
- package/dist/routing/provider-adapters/bailian.js +56 -0
- package/dist/routing/provider-adapters/bailian.js.map +1 -0
- package/dist/routing/provider-adapters/doubao.d.ts +3 -0
- package/dist/routing/provider-adapters/doubao.d.ts.map +1 -0
- package/dist/routing/provider-adapters/doubao.js +56 -0
- package/dist/routing/provider-adapters/doubao.js.map +1 -0
- package/dist/routing/provider-adapters/openai.d.ts +6 -0
- package/dist/routing/provider-adapters/openai.d.ts.map +1 -0
- package/dist/routing/provider-adapters/openai.js +80 -0
- package/dist/routing/provider-adapters/openai.js.map +1 -0
- package/dist/routing/provider-adapters/openbmb.d.ts +16 -0
- package/dist/routing/provider-adapters/openbmb.d.ts.map +1 -0
- package/dist/routing/provider-adapters/openbmb.js +368 -0
- package/dist/routing/provider-adapters/openbmb.js.map +1 -0
- package/dist/routing/provider-executor.d.ts +21 -0
- package/dist/routing/provider-executor.d.ts.map +1 -0
- package/dist/routing/provider-executor.js +260 -0
- package/dist/routing/provider-executor.js.map +1 -0
- package/dist/routing/provider-health.d.ts +9 -0
- package/dist/routing/provider-health.d.ts.map +1 -0
- package/dist/routing/provider-health.js +66 -0
- package/dist/routing/provider-health.js.map +1 -0
- package/dist/surfaces/cli/index.d.ts +3 -0
- package/dist/surfaces/cli/index.d.ts.map +1 -0
- package/dist/surfaces/cli/index.js +200 -0
- package/dist/surfaces/cli/index.js.map +1 -0
- package/dist/surfaces/cli/init.d.ts +2 -0
- package/dist/surfaces/cli/init.d.ts.map +1 -0
- package/dist/surfaces/cli/init.js +121 -0
- package/dist/surfaces/cli/init.js.map +1 -0
- package/dist/surfaces/cli/output.d.ts +16 -0
- package/dist/surfaces/cli/output.d.ts.map +1 -0
- package/dist/surfaces/cli/output.js +74 -0
- package/dist/surfaces/cli/output.js.map +1 -0
- package/dist/surfaces/cli/runtime.d.ts +218 -0
- package/dist/surfaces/cli/runtime.d.ts.map +1 -0
- package/dist/surfaces/cli/runtime.js +1130 -0
- package/dist/surfaces/cli/runtime.js.map +1 -0
- package/dist/surfaces/mcp/server.d.ts +60 -0
- package/dist/surfaces/mcp/server.d.ts.map +1 -0
- package/dist/surfaces/mcp/server.js +425 -0
- package/dist/surfaces/mcp/server.js.map +1 -0
- package/dist/surfaces/vscode/extension.d.ts +17 -0
- package/dist/surfaces/vscode/extension.d.ts.map +1 -0
- package/dist/surfaces/vscode/extension.js +35 -0
- package/dist/surfaces/vscode/extension.js.map +1 -0
- package/dist/utils/file-lock.d.ts +9 -0
- package/dist/utils/file-lock.d.ts.map +1 -0
- package/dist/utils/file-lock.js +90 -0
- package/dist/utils/file-lock.js.map +1 -0
- package/dist/utils/logger.d.ts +3 -0
- package/dist/utils/logger.d.ts.map +1 -0
- package/dist/utils/logger.js +36 -0
- package/dist/utils/logger.js.map +1 -0
- package/package.json +89 -0
- package/scripts/safe-postinstall.cjs +14 -0
|
@@ -0,0 +1,61 @@
|
|
|
1
|
+
export type TaskStatus = "PENDING" | "RUNNING" | "VALIDATING" | "COMPLETED" | "FAILED" | "HUMAN_REVIEW_REQUIRED";
|
|
2
|
+
export type AgentRole = "planner" | "worker" | "validator" | "enricher" | "evolver";
|
|
3
|
+
export interface ValidationResult {
|
|
4
|
+
passed: boolean;
|
|
5
|
+
feedback: string;
|
|
6
|
+
matchedCriteria: string[];
|
|
7
|
+
missingCriteria: string[];
|
|
8
|
+
riskTags: string[];
|
|
9
|
+
}
|
|
10
|
+
export interface RouteDecision {
|
|
11
|
+
role: AgentRole;
|
|
12
|
+
provider: string;
|
|
13
|
+
model: string;
|
|
14
|
+
tier: "smart" | "economy";
|
|
15
|
+
fallbackApplied: boolean;
|
|
16
|
+
}
|
|
17
|
+
export interface TaskRunResult {
|
|
18
|
+
status: TaskStatus;
|
|
19
|
+
attempts: number;
|
|
20
|
+
feedback: string;
|
|
21
|
+
routeDecisions?: RouteDecision[];
|
|
22
|
+
executionRounds?: string[][];
|
|
23
|
+
validationSummary?: {
|
|
24
|
+
matched: number;
|
|
25
|
+
missing: number;
|
|
26
|
+
riskTags: string[];
|
|
27
|
+
};
|
|
28
|
+
replanRounds?: number;
|
|
29
|
+
brainstormIdeas?: string[];
|
|
30
|
+
promptContextLines?: number;
|
|
31
|
+
episodeId?: string;
|
|
32
|
+
similarEpisodes?: Array<{
|
|
33
|
+
id: string;
|
|
34
|
+
task: string;
|
|
35
|
+
score: number;
|
|
36
|
+
}>;
|
|
37
|
+
}
|
|
38
|
+
export interface TaskNode {
|
|
39
|
+
id: string;
|
|
40
|
+
description: string;
|
|
41
|
+
dependencies: string[];
|
|
42
|
+
status: TaskStatus;
|
|
43
|
+
contextQuery: string;
|
|
44
|
+
retryCount: number;
|
|
45
|
+
}
|
|
46
|
+
export interface OrchestrationInput {
|
|
47
|
+
task: string;
|
|
48
|
+
maxRetries?: number;
|
|
49
|
+
}
|
|
50
|
+
export interface GraphNode {
|
|
51
|
+
id: string;
|
|
52
|
+
type: "File" | "Symbol" | "Module" | "TaskRun" | "Decision" | "Skill";
|
|
53
|
+
content: string;
|
|
54
|
+
metadata?: Record<string, unknown>;
|
|
55
|
+
}
|
|
56
|
+
export interface GraphEdge {
|
|
57
|
+
from: string;
|
|
58
|
+
to: string;
|
|
59
|
+
relation: "defines" | "references" | "imports" | "depends_on" | "changes" | "validates" | "co_occurs" | "prerequisite" | "improves" | "conflicts_with";
|
|
60
|
+
}
|
|
61
|
+
//# sourceMappingURL=types.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/core/types.ts"],"names":[],"mappings":"AAAA,MAAM,MAAM,UAAU,GAClB,SAAS,GACT,SAAS,GACT,YAAY,GACZ,WAAW,GACX,QAAQ,GACR,uBAAuB,CAAC;AAE5B,MAAM,MAAM,SAAS,GAAG,SAAS,GAAG,QAAQ,GAAG,WAAW,GAAG,UAAU,GAAG,SAAS,CAAC;AAEpF,MAAM,WAAW,gBAAgB;IAC/B,MAAM,EAAE,OAAO,CAAC;IAChB,QAAQ,EAAE,MAAM,CAAC;IACjB,eAAe,EAAE,MAAM,EAAE,CAAC;IAC1B,eAAe,EAAE,MAAM,EAAE,CAAC;IAC1B,QAAQ,EAAE,MAAM,EAAE,CAAC;CACpB;AAED,MAAM,WAAW,aAAa;IAC5B,IAAI,EAAE,SAAS,CAAC;IAChB,QAAQ,EAAE,MAAM,CAAC;IACjB,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,OAAO,GAAG,SAAS,CAAC;IAC1B,eAAe,EAAE,OAAO,CAAC;CAC1B;AAED,MAAM,WAAW,aAAa;IAC5B,MAAM,EAAE,UAAU,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC;IACjB,cAAc,CAAC,EAAE,aAAa,EAAE,CAAC;IACjC,eAAe,CAAC,EAAE,MAAM,EAAE,EAAE,CAAC;IAC7B,iBAAiB,CAAC,EAAE;QAClB,OAAO,EAAE,MAAM,CAAC;QAChB,OAAO,EAAE,MAAM,CAAC;QAChB,QAAQ,EAAE,MAAM,EAAE,CAAC;KACpB,CAAC;IACF,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,eAAe,CAAC,EAAE,MAAM,EAAE,CAAC;IAC3B,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,eAAe,CAAC,EAAE,KAAK,CAAC;QAAE,EAAE,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;CACtE;AAED,MAAM,WAAW,QAAQ;IACvB,EAAE,EAAE,MAAM,CAAC;IACX,WAAW,EAAE,MAAM,CAAC;IACpB,YAAY,EAAE,MAAM,EAAE,CAAC;IACvB,MAAM,EAAE,UAAU,CAAC;IACnB,YAAY,EAAE,MAAM,CAAC;IACrB,UAAU,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,kBAAkB;IACjC,IAAI,EAAE,MAAM,CAAC;IACb,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB;AAED,MAAM,WAAW,SAAS;IACxB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,GAAG,QAAQ,GAAG,QAAQ,GAAG,SAAS,GAAG,UAAU,GAAG,OAAO,CAAC;IACtE,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACpC;AAED,MAAM,WAAW,SAAS;IACxB,IAAI,EAAE,MAAM,CAAC;IACb,EAAE,EAAE,MAAM,CAAC;IACX,QAAQ,EACJ,SAAS,GACT,YAAY,GACZ,SAAS,GACT,YAAY,GACZ,SAAS,GACT,WAAW,GACX,WAAW,GACX,cAAc,GACd,UAAU,GACV,gBAAgB,CAAC;CACtB"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"types.js","sourceRoot":"","sources":["../../src/core/types.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,22 @@
|
|
|
1
|
+
import type { GraphFlowConfig } from "../config/schema";
|
|
2
|
+
import type { GraphEdge, GraphNode } from "../core/types";
|
|
3
|
+
export interface GraphStoreSnapshot {
|
|
4
|
+
nodes: GraphNode[];
|
|
5
|
+
edges: GraphEdge[];
|
|
6
|
+
}
|
|
7
|
+
export interface GraphClient {
|
|
8
|
+
upsertNodes(nodes: GraphNode[]): Promise<void>;
|
|
9
|
+
upsertEdges(edges: GraphEdge[]): Promise<void>;
|
|
10
|
+
queryByKeyword(query: string): Promise<GraphNode[]>;
|
|
11
|
+
readSnapshot?(): GraphStoreSnapshot;
|
|
12
|
+
getNodesByIds?(ids: string[]): Promise<GraphNode[]>;
|
|
13
|
+
getNeighbors?(nodeIds: string[], relations?: GraphEdge["relation"][], direction?: "out" | "in" | "both"): Promise<{
|
|
14
|
+
node: GraphNode;
|
|
15
|
+
via: GraphEdge["relation"];
|
|
16
|
+
}[]>;
|
|
17
|
+
deleteNode?(id: string): Promise<void>;
|
|
18
|
+
deleteEdge?(from: string, to: string, relation: GraphEdge["relation"]): Promise<void>;
|
|
19
|
+
vacuum?(): Promise<void> | void;
|
|
20
|
+
}
|
|
21
|
+
export declare function createGraphClient(config: GraphFlowConfig): GraphClient;
|
|
22
|
+
//# sourceMappingURL=client-factory.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"client-factory.d.ts","sourceRoot":"","sources":["../../src/graph/client-factory.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AACxD,OAAO,KAAK,EAAE,SAAS,EAAE,SAAS,EAAE,MAAM,eAAe,CAAC;AAQ1D,MAAM,WAAW,kBAAkB;IACjC,KAAK,EAAE,SAAS,EAAE,CAAC;IACnB,KAAK,EAAE,SAAS,EAAE,CAAC;CACpB;AAED,MAAM,WAAW,WAAW;IAC1B,WAAW,CAAC,KAAK,EAAE,SAAS,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAC/C,WAAW,CAAC,KAAK,EAAE,SAAS,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAC/C,cAAc,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC,CAAC;IACpD,YAAY,CAAC,IAAI,kBAAkB,CAAC;IACpC,aAAa,CAAC,CAAC,GAAG,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC,CAAC;IACpD,YAAY,CAAC,CACX,OAAO,EAAE,MAAM,EAAE,EACjB,SAAS,CAAC,EAAE,SAAS,CAAC,UAAU,CAAC,EAAE,EACnC,SAAS,CAAC,EAAE,KAAK,GAAG,IAAI,GAAG,MAAM,GAChC,OAAO,CAAC;QAAE,IAAI,EAAE,SAAS,CAAC;QAAC,GAAG,EAAE,SAAS,CAAC,UAAU,CAAC,CAAA;KAAE,EAAE,CAAC,CAAC;IAC9D,UAAU,CAAC,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACvC,UAAU,CAAC,CAAC,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,QAAQ,EAAE,SAAS,CAAC,UAAU,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACtF,MAAM,CAAC,IAAI,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;CACjC;AA0CD,wBAAgB,iBAAiB,CAAC,MAAM,EAAE,eAAe,GAAG,WAAW,CAwBtE"}
|
|
@@ -0,0 +1,60 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.createGraphClient = createGraphClient;
|
|
4
|
+
const paths_1 = require("../config/paths");
|
|
5
|
+
const logger_1 = require("../utils/logger");
|
|
6
|
+
const graphify_client_1 = require("./graphify-client");
|
|
7
|
+
const graphify_file_client_1 = require("./graphify-file-client");
|
|
8
|
+
const graphify_mcp_client_1 = require("./graphify-mcp-client");
|
|
9
|
+
const sqlite_client_1 = require("./sqlite-client");
|
|
10
|
+
class InMemoryGraphClientAdapter {
|
|
11
|
+
client;
|
|
12
|
+
constructor(client) {
|
|
13
|
+
this.client = client;
|
|
14
|
+
}
|
|
15
|
+
async upsertNodes(nodes) {
|
|
16
|
+
this.client.upsertNodes(nodes);
|
|
17
|
+
}
|
|
18
|
+
async upsertEdges(edges) {
|
|
19
|
+
this.client.upsertEdges(edges);
|
|
20
|
+
}
|
|
21
|
+
async queryByKeyword(query) {
|
|
22
|
+
return this.client.queryByKeyword(query);
|
|
23
|
+
}
|
|
24
|
+
async getNodesByIds(ids) {
|
|
25
|
+
return this.client.getNodesByIds(ids);
|
|
26
|
+
}
|
|
27
|
+
async getNeighbors(nodeIds, relations, direction) {
|
|
28
|
+
return this.client.getNeighbors(nodeIds, relations, direction);
|
|
29
|
+
}
|
|
30
|
+
readSnapshot() {
|
|
31
|
+
return this.client.readSnapshot();
|
|
32
|
+
}
|
|
33
|
+
async deleteNode(id) {
|
|
34
|
+
return this.client.deleteNode(id);
|
|
35
|
+
}
|
|
36
|
+
async deleteEdge(from, to, relation) {
|
|
37
|
+
return this.client.deleteEdge(from, to, relation);
|
|
38
|
+
}
|
|
39
|
+
}
|
|
40
|
+
function createGraphClient(config) {
|
|
41
|
+
if (config.graphPolicy.transport === "mcp-http") {
|
|
42
|
+
return new graphify_mcp_client_1.GraphifyMcpClient(config.graphPolicy.mcpEndpoint, config.graphPolicy.mcpApiKey);
|
|
43
|
+
}
|
|
44
|
+
if (config.graphPolicy.transport === "file") {
|
|
45
|
+
return new graphify_file_client_1.GraphifyFileClient((0, paths_1.resolveGraphStorePath)(config));
|
|
46
|
+
}
|
|
47
|
+
if (config.graphPolicy.transport === "sqlite") {
|
|
48
|
+
try {
|
|
49
|
+
return new sqlite_client_1.GraphifySqliteClient((0, paths_1.resolveGraphStorePath)(config));
|
|
50
|
+
}
|
|
51
|
+
catch (err) {
|
|
52
|
+
const msg = err instanceof Error ? err.message : String(err);
|
|
53
|
+
const fallbackPath = (0, paths_1.resolveGraphStorePath)(config).replace(/\.sqlite$/i, ".json");
|
|
54
|
+
logger_1.logger.warn({ err, fallbackPath }, `[graphflow] sqlite transport unavailable, falling back to file. Reason: ${msg}`);
|
|
55
|
+
return new graphify_file_client_1.GraphifyFileClient(fallbackPath);
|
|
56
|
+
}
|
|
57
|
+
}
|
|
58
|
+
return new InMemoryGraphClientAdapter(new graphify_client_1.GraphifyClient());
|
|
59
|
+
}
|
|
60
|
+
//# sourceMappingURL=client-factory.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"client-factory.js","sourceRoot":"","sources":["../../src/graph/client-factory.ts"],"names":[],"mappings":";;AAsEA,8CAwBC;AA5FD,2CAAwD;AACxD,4CAAyC;AACzC,uDAAmD;AACnD,iEAA4D;AAC5D,+DAA0D;AAC1D,mDAAuD;AAuBvD,MAAM,0BAA0B;IACD;IAA7B,YAA6B,MAAsB;QAAtB,WAAM,GAAN,MAAM,CAAgB;IAAG,CAAC;IAEvD,KAAK,CAAC,WAAW,CAAC,KAAkB;QAClC,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;IACjC,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,KAAkB;QAClC,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;IACjC,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,KAAa;QAChC,OAAO,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC;IAC3C,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,GAAa;QAC/B,OAAO,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC;IACxC,CAAC;IAED,KAAK,CAAC,YAAY,CAChB,OAAiB,EACjB,SAAmC,EACnC,SAAiC;QAEjC,OAAO,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,OAAO,EAAE,SAAS,EAAE,SAAS,CAAC,CAAC;IACjE,CAAC;IAED,YAAY;QACV,OAAO,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,CAAC;IACpC,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,EAAU;QACzB,OAAO,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;IACpC,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,IAAY,EAAE,EAAU,EAAE,QAA+B;QACxE,OAAO,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,EAAE,EAAE,EAAE,QAAQ,CAAC,CAAC;IACpD,CAAC;CACF;AAED,SAAgB,iBAAiB,CAAC,MAAuB;IACvD,IAAI,MAAM,CAAC,WAAW,CAAC,SAAS,KAAK,UAAU,EAAE,CAAC;QAChD,OAAO,IAAI,uCAAiB,CAAC,MAAM,CAAC,WAAW,CAAC,WAAY,EAAE,MAAM,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC;IAC9F,CAAC;IAED,IAAI,MAAM,CAAC,WAAW,CAAC,SAAS,KAAK,MAAM,EAAE,CAAC;QAC5C,OAAO,IAAI,yCAAkB,CAAC,IAAA,6BAAqB,EAAC,MAAM,CAAC,CAAC,CAAC;IAC/D,CAAC;IAED,IAAI,MAAM,CAAC,WAAW,CAAC,SAAS,KAAK,QAAQ,EAAE,CAAC;QAC9C,IAAI,CAAC;YACH,OAAO,IAAI,oCAAoB,CAAC,IAAA,6BAAqB,EAAC,MAAM,CAAC,CAAC,CAAC;QACjE,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,GAAG,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YAC7D,MAAM,YAAY,GAAG,IAAA,6BAAqB,EAAC,MAAM,CAAC,CAAC,OAAO,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;YAClF,eAAM,CAAC,IAAI,CACT,EAAE,GAAG,EAAE,YAAY,EAAE,EACrB,2EAA2E,GAAG,EAAE,CACjF,CAAC;YACF,OAAO,IAAI,yCAAkB,CAAC,YAAY,CAAC,CAAC;QAC9C,CAAC;IACH,CAAC;IAED,OAAO,IAAI,0BAA0B,CAAC,IAAI,gCAAc,EAAE,CAAC,CAAC;AAC9D,CAAC"}
|
|
@@ -0,0 +1,47 @@
|
|
|
1
|
+
import type { GraphEdge, GraphNode } from "../core/types";
|
|
2
|
+
import { type EmbeddingProvider } from "../learning/embeddings";
|
|
3
|
+
import type { GraphClient } from "./client-factory";
|
|
4
|
+
export interface ContextSlice {
|
|
5
|
+
items: string[];
|
|
6
|
+
tokenEstimate: number;
|
|
7
|
+
}
|
|
8
|
+
export type ContextLayer = "L1" | "L2" | "L3";
|
|
9
|
+
export interface ContextAnchorItem {
|
|
10
|
+
id: string;
|
|
11
|
+
type: GraphNode["type"];
|
|
12
|
+
layer: ContextLayer;
|
|
13
|
+
}
|
|
14
|
+
export interface LayeredContextPackage {
|
|
15
|
+
summaryChannel: string[];
|
|
16
|
+
anchorChannel: ContextAnchorItem[];
|
|
17
|
+
tokenEstimate: number;
|
|
18
|
+
truncated: boolean;
|
|
19
|
+
}
|
|
20
|
+
export interface LayeredPackageOptions {
|
|
21
|
+
layerQuota?: {
|
|
22
|
+
l1: number;
|
|
23
|
+
l2: number;
|
|
24
|
+
l3: number;
|
|
25
|
+
};
|
|
26
|
+
enableEdgeExpansion?: boolean;
|
|
27
|
+
enableVectorRecall?: boolean;
|
|
28
|
+
embeddingProvider?: EmbeddingProvider;
|
|
29
|
+
vectorTopK?: number;
|
|
30
|
+
vectorMinSimilarity?: number;
|
|
31
|
+
}
|
|
32
|
+
export interface SubgraphExpansionOptions {
|
|
33
|
+
hops?: number;
|
|
34
|
+
maxNodes?: number;
|
|
35
|
+
relations?: GraphEdge["relation"][];
|
|
36
|
+
}
|
|
37
|
+
export declare function expandSubgraph(client: GraphClient, seedIds: string[], options?: SubgraphExpansionOptions): Promise<GraphNode[]>;
|
|
38
|
+
export declare function buildContextSlice(client: GraphClient, query: string, maxTokens: number): Promise<ContextSlice>;
|
|
39
|
+
export declare function vectorRecall(nodes: GraphNode[], queryEmbedding: number[], topK: number, minSimilarity: number): GraphNode[];
|
|
40
|
+
export declare function buildLayeredContextPackage(client: GraphClient, query: string, maxTokens: number, options?: LayeredPackageOptions): Promise<LayeredContextPackage>;
|
|
41
|
+
export interface ContextRefillManager {
|
|
42
|
+
initialPackage(query: string): Promise<LayeredContextPackage>;
|
|
43
|
+
refill(evidenceHints: string[]): Promise<string[]>;
|
|
44
|
+
}
|
|
45
|
+
export declare function createContextRefillManager(client: GraphClient, maxTokens: number, options?: LayeredPackageOptions): ContextRefillManager;
|
|
46
|
+
export declare function summarizeNodes(nodes: GraphNode[]): string[];
|
|
47
|
+
//# sourceMappingURL=context-slicer.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"context-slicer.d.ts","sourceRoot":"","sources":["../../src/graph/context-slicer.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,SAAS,EAAE,SAAS,EAAE,MAAM,eAAe,CAAC;AAC1D,OAAO,EAIL,KAAK,iBAAiB,EACvB,MAAM,wBAAwB,CAAC;AAChC,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAgCpD,MAAM,WAAW,YAAY;IAC3B,KAAK,EAAE,MAAM,EAAE,CAAC;IAChB,aAAa,EAAE,MAAM,CAAC;CACvB;AAED,MAAM,MAAM,YAAY,GAAG,IAAI,GAAG,IAAI,GAAG,IAAI,CAAC;AAE9C,MAAM,WAAW,iBAAiB;IAChC,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,SAAS,CAAC,MAAM,CAAC,CAAC;IACxB,KAAK,EAAE,YAAY,CAAC;CACrB;AAED,MAAM,WAAW,qBAAqB;IACpC,cAAc,EAAE,MAAM,EAAE,CAAC;IACzB,aAAa,EAAE,iBAAiB,EAAE,CAAC;IACnC,aAAa,EAAE,MAAM,CAAC;IACtB,SAAS,EAAE,OAAO,CAAC;CACpB;AAED,MAAM,WAAW,qBAAqB;IACpC,UAAU,CAAC,EAAE;QACX,EAAE,EAAE,MAAM,CAAC;QACX,EAAE,EAAE,MAAM,CAAC;QACX,EAAE,EAAE,MAAM,CAAC;KACZ,CAAC;IACF,mBAAmB,CAAC,EAAE,OAAO,CAAC;IAC9B,kBAAkB,CAAC,EAAE,OAAO,CAAC;IAC7B,iBAAiB,CAAC,EAAE,iBAAiB,CAAC;IACtC,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,mBAAmB,CAAC,EAAE,MAAM,CAAC;CAC9B;AAED,MAAM,WAAW,wBAAwB;IACvC,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,SAAS,CAAC,EAAE,SAAS,CAAC,UAAU,CAAC,EAAE,CAAC;CACrC;AASD,wBAAsB,cAAc,CAClC,MAAM,EAAE,WAAW,EACnB,OAAO,EAAE,MAAM,EAAE,EACjB,OAAO,CAAC,EAAE,wBAAwB,GACjC,OAAO,CAAC,SAAS,EAAE,CAAC,CAyBtB;AAED,wBAAsB,iBAAiB,CACrC,MAAM,EAAE,WAAW,EACnB,KAAK,EAAE,MAAM,EACb,SAAS,EAAE,MAAM,GAChB,OAAO,CAAC,YAAY,CAAC,CAGvB;AAED,wBAAgB,YAAY,CAC1B,KAAK,EAAE,SAAS,EAAE,EAClB,cAAc,EAAE,MAAM,EAAE,EACxB,IAAI,EAAE,MAAM,EACZ,aAAa,EAAE,MAAM,GACpB,SAAS,EAAE,CAWb;AAED,wBAAsB,0BAA0B,CAC9C,MAAM,EAAE,WAAW,EACnB,KAAK,EAAE,MAAM,EACb,SAAS,EAAE,MAAM,EACjB,OAAO,CAAC,EAAE,qBAAqB,GAC9B,OAAO,CAAC,qBAAqB,CAAC,CA6EhC;AAED,MAAM,WAAW,oBAAoB;IACnC,cAAc,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,qBAAqB,CAAC,CAAC;IAC9D,MAAM,CAAC,aAAa,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;CACpD;AAED,wBAAgB,0BAA0B,CACxC,MAAM,EAAE,WAAW,EACnB,SAAS,EAAE,MAAM,EACjB,OAAO,CAAC,EAAE,qBAAqB,GAC9B,oBAAoB,CAsCtB;AAgBD,wBAAgB,cAAc,CAAC,KAAK,EAAE,SAAS,EAAE,GAAG,MAAM,EAAE,CAE3D"}
|
|
@@ -0,0 +1,246 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.expandSubgraph = expandSubgraph;
|
|
4
|
+
exports.buildContextSlice = buildContextSlice;
|
|
5
|
+
exports.vectorRecall = vectorRecall;
|
|
6
|
+
exports.buildLayeredContextPackage = buildLayeredContextPackage;
|
|
7
|
+
exports.createContextRefillManager = createContextRefillManager;
|
|
8
|
+
exports.summarizeNodes = summarizeNodes;
|
|
9
|
+
const logger_1 = require("../utils/logger");
|
|
10
|
+
const embeddings_1 = require("../learning/embeddings");
|
|
11
|
+
let encoderFn = null;
|
|
12
|
+
let encoderLoaded = false;
|
|
13
|
+
function getEncoder() {
|
|
14
|
+
if (encoderLoaded)
|
|
15
|
+
return encoderFn;
|
|
16
|
+
encoderLoaded = true;
|
|
17
|
+
try {
|
|
18
|
+
const mod = require("gpt-tokenizer/encoding/o200k_base");
|
|
19
|
+
if (typeof mod.encode === "function") {
|
|
20
|
+
encoderFn = mod.encode.bind(mod);
|
|
21
|
+
return encoderFn;
|
|
22
|
+
}
|
|
23
|
+
}
|
|
24
|
+
catch (error) {
|
|
25
|
+
logger_1.logger.error({ error }, "Caught error");
|
|
26
|
+
// fall through
|
|
27
|
+
}
|
|
28
|
+
try {
|
|
29
|
+
const mod = require("gpt-tokenizer");
|
|
30
|
+
if (typeof mod.encode === "function") {
|
|
31
|
+
encoderFn = mod.encode.bind(mod);
|
|
32
|
+
return encoderFn;
|
|
33
|
+
}
|
|
34
|
+
}
|
|
35
|
+
catch (error) {
|
|
36
|
+
logger_1.logger.error({ error }, "Caught error");
|
|
37
|
+
// fall through
|
|
38
|
+
}
|
|
39
|
+
encoderFn = null;
|
|
40
|
+
return null;
|
|
41
|
+
}
|
|
42
|
+
const DEFAULT_EXPANSION_RELATIONS = [
|
|
43
|
+
"references",
|
|
44
|
+
"imports",
|
|
45
|
+
"depends_on",
|
|
46
|
+
"prerequisite",
|
|
47
|
+
];
|
|
48
|
+
async function expandSubgraph(client, seedIds, options) {
|
|
49
|
+
if (typeof client.getNeighbors !== "function")
|
|
50
|
+
return [];
|
|
51
|
+
const hops = options?.hops ?? 1;
|
|
52
|
+
const maxNodes = options?.maxNodes ?? 20;
|
|
53
|
+
const relations = options?.relations ?? DEFAULT_EXPANSION_RELATIONS;
|
|
54
|
+
const excluded = new Set(seedIds);
|
|
55
|
+
const collected = new Map();
|
|
56
|
+
let frontier = [...seedIds];
|
|
57
|
+
for (let depth = 0; depth < hops; depth += 1) {
|
|
58
|
+
if (frontier.length === 0)
|
|
59
|
+
break;
|
|
60
|
+
const neighbors = await client.getNeighbors(frontier, relations, "both");
|
|
61
|
+
const nextFrontier = [];
|
|
62
|
+
for (const { node } of neighbors) {
|
|
63
|
+
if (excluded.has(node.id) || collected.has(node.id))
|
|
64
|
+
continue;
|
|
65
|
+
collected.set(node.id, node);
|
|
66
|
+
nextFrontier.push(node.id);
|
|
67
|
+
if (collected.size >= maxNodes)
|
|
68
|
+
break;
|
|
69
|
+
}
|
|
70
|
+
if (collected.size >= maxNodes)
|
|
71
|
+
break;
|
|
72
|
+
frontier = nextFrontier;
|
|
73
|
+
}
|
|
74
|
+
return Array.from(collected.values());
|
|
75
|
+
}
|
|
76
|
+
async function buildContextSlice(client, query, maxTokens) {
|
|
77
|
+
const pkg = await buildLayeredContextPackage(client, query, maxTokens);
|
|
78
|
+
return { items: pkg.summaryChannel, tokenEstimate: pkg.tokenEstimate };
|
|
79
|
+
}
|
|
80
|
+
function vectorRecall(nodes, queryEmbedding, topK, minSimilarity) {
|
|
81
|
+
if (!queryEmbedding || queryEmbedding.length === 0)
|
|
82
|
+
return [];
|
|
83
|
+
const scored = [];
|
|
84
|
+
for (const node of nodes) {
|
|
85
|
+
const emb = (0, embeddings_1.extractEmbedding)(node);
|
|
86
|
+
if (!emb)
|
|
87
|
+
continue;
|
|
88
|
+
const sim = (0, embeddings_1.cosineSimilarity)(queryEmbedding, emb);
|
|
89
|
+
if (sim >= minSimilarity)
|
|
90
|
+
scored.push({ node, sim });
|
|
91
|
+
}
|
|
92
|
+
scored.sort((a, b) => b.sim - a.sim);
|
|
93
|
+
return scored.slice(0, topK).map((s) => s.node);
|
|
94
|
+
}
|
|
95
|
+
async function buildLayeredContextPackage(client, query, maxTokens, options) {
|
|
96
|
+
const keywordHits = await client.queryByKeyword(query);
|
|
97
|
+
let hits = keywordHits;
|
|
98
|
+
if (options?.enableVectorRecall === true && options.embeddingProvider) {
|
|
99
|
+
try {
|
|
100
|
+
const queryEmbedding = await options.embeddingProvider.embed(query);
|
|
101
|
+
const topK = options.vectorTopK ?? 8;
|
|
102
|
+
const minSim = options.vectorMinSimilarity ?? 0.05;
|
|
103
|
+
const vectorHits = vectorRecall(keywordHits, queryEmbedding, topK, minSim);
|
|
104
|
+
hits = (0, embeddings_1.reciprocalRankFusion)([keywordHits, vectorHits]);
|
|
105
|
+
}
|
|
106
|
+
catch (error) {
|
|
107
|
+
logger_1.logger.error({ error }, "Caught error");
|
|
108
|
+
hits = keywordHits;
|
|
109
|
+
}
|
|
110
|
+
}
|
|
111
|
+
const summaryChannel = [];
|
|
112
|
+
const anchorChannel = [];
|
|
113
|
+
let tokens = 0;
|
|
114
|
+
let truncated = false;
|
|
115
|
+
const quota = {
|
|
116
|
+
l1: options?.layerQuota?.l1 ?? Number.POSITIVE_INFINITY,
|
|
117
|
+
l2: options?.layerQuota?.l2 ?? Number.POSITIVE_INFINITY,
|
|
118
|
+
l3: options?.layerQuota?.l3 ?? Number.POSITIVE_INFINITY,
|
|
119
|
+
};
|
|
120
|
+
const used = { l1: 0, l2: 0, l3: 0 };
|
|
121
|
+
const added = new Set();
|
|
122
|
+
for (const hit of hits) {
|
|
123
|
+
const layer = classifyLayer(hit);
|
|
124
|
+
if (!canUseLayer(layer, quota, used)) {
|
|
125
|
+
continue;
|
|
126
|
+
}
|
|
127
|
+
const summary = `${hit.type}: ${hit.content}`;
|
|
128
|
+
const estimate = estimateTokens(summary);
|
|
129
|
+
if (tokens + estimate > maxTokens) {
|
|
130
|
+
truncated = true;
|
|
131
|
+
break;
|
|
132
|
+
}
|
|
133
|
+
summaryChannel.push(summary);
|
|
134
|
+
anchorChannel.push({ id: hit.id, type: hit.type, layer });
|
|
135
|
+
added.add(hit.id);
|
|
136
|
+
tokens += estimate;
|
|
137
|
+
markLayerUsed(layer, used);
|
|
138
|
+
}
|
|
139
|
+
const enableExpansion = options?.enableEdgeExpansion !== false;
|
|
140
|
+
if (enableExpansion && !truncated && typeof client.getNeighbors === "function") {
|
|
141
|
+
const seedIds = anchorChannel.slice(0, 5).map((a) => a.id);
|
|
142
|
+
if (seedIds.length > 0) {
|
|
143
|
+
const expanded = await expandSubgraph(client, seedIds, { hops: 1 });
|
|
144
|
+
for (const node of expanded) {
|
|
145
|
+
if (added.has(node.id))
|
|
146
|
+
continue;
|
|
147
|
+
const layer = classifyLayer(node);
|
|
148
|
+
if (!canUseLayer(layer, quota, used))
|
|
149
|
+
continue;
|
|
150
|
+
const summary = `${node.type}: ${node.content}`;
|
|
151
|
+
const estimate = estimateTokens(summary);
|
|
152
|
+
if (tokens + estimate > maxTokens) {
|
|
153
|
+
truncated = true;
|
|
154
|
+
break;
|
|
155
|
+
}
|
|
156
|
+
summaryChannel.push(summary);
|
|
157
|
+
anchorChannel.push({ id: node.id, type: node.type, layer });
|
|
158
|
+
added.add(node.id);
|
|
159
|
+
tokens += estimate;
|
|
160
|
+
markLayerUsed(layer, used);
|
|
161
|
+
}
|
|
162
|
+
}
|
|
163
|
+
}
|
|
164
|
+
return { summaryChannel, anchorChannel, tokenEstimate: tokens, truncated };
|
|
165
|
+
}
|
|
166
|
+
function createContextRefillManager(client, maxTokens, options) {
|
|
167
|
+
const seenAnchors = new Set();
|
|
168
|
+
return {
|
|
169
|
+
async initialPackage(query) {
|
|
170
|
+
const pkg = await buildLayeredContextPackage(client, query, maxTokens, options);
|
|
171
|
+
for (const anchor of pkg.anchorChannel) {
|
|
172
|
+
seenAnchors.add(anchor.id);
|
|
173
|
+
}
|
|
174
|
+
return pkg;
|
|
175
|
+
},
|
|
176
|
+
async refill(evidenceHints) {
|
|
177
|
+
const items = [];
|
|
178
|
+
let tokens = 0;
|
|
179
|
+
for (const hint of evidenceHints) {
|
|
180
|
+
const hits = await client.queryByKeyword(hint);
|
|
181
|
+
for (const hit of hits) {
|
|
182
|
+
if (seenAnchors.has(hit.id)) {
|
|
183
|
+
continue;
|
|
184
|
+
}
|
|
185
|
+
const line = `${hit.id} | ${hit.type}: ${hit.content}`;
|
|
186
|
+
const estimate = estimateTokens(line);
|
|
187
|
+
if (tokens + estimate > maxTokens) {
|
|
188
|
+
return items;
|
|
189
|
+
}
|
|
190
|
+
seenAnchors.add(hit.id);
|
|
191
|
+
tokens += estimate;
|
|
192
|
+
items.push(line);
|
|
193
|
+
break;
|
|
194
|
+
}
|
|
195
|
+
}
|
|
196
|
+
return items;
|
|
197
|
+
},
|
|
198
|
+
};
|
|
199
|
+
}
|
|
200
|
+
function estimateTokens(text) {
|
|
201
|
+
const enc = getEncoder();
|
|
202
|
+
if (enc) {
|
|
203
|
+
try {
|
|
204
|
+
const n = enc(text).length;
|
|
205
|
+
return Math.max(1, n);
|
|
206
|
+
}
|
|
207
|
+
catch (error) {
|
|
208
|
+
logger_1.logger.error({ error }, "Caught error");
|
|
209
|
+
// fall back below
|
|
210
|
+
}
|
|
211
|
+
}
|
|
212
|
+
return Math.max(1, Math.ceil(text.length / 4));
|
|
213
|
+
}
|
|
214
|
+
function summarizeNodes(nodes) {
|
|
215
|
+
return nodes.map((node) => `${node.type}(${node.id})`);
|
|
216
|
+
}
|
|
217
|
+
function classifyLayer(node) {
|
|
218
|
+
if (node.type === "File" || node.type === "Symbol") {
|
|
219
|
+
return "L1";
|
|
220
|
+
}
|
|
221
|
+
if (node.type === "Module") {
|
|
222
|
+
return "L2";
|
|
223
|
+
}
|
|
224
|
+
return "L3";
|
|
225
|
+
}
|
|
226
|
+
function canUseLayer(layer, quota, used) {
|
|
227
|
+
if (layer === "L1") {
|
|
228
|
+
return used.l1 < quota.l1;
|
|
229
|
+
}
|
|
230
|
+
if (layer === "L2") {
|
|
231
|
+
return used.l2 < quota.l2;
|
|
232
|
+
}
|
|
233
|
+
return used.l3 < quota.l3;
|
|
234
|
+
}
|
|
235
|
+
function markLayerUsed(layer, used) {
|
|
236
|
+
if (layer === "L1") {
|
|
237
|
+
used.l1 += 1;
|
|
238
|
+
return;
|
|
239
|
+
}
|
|
240
|
+
if (layer === "L2") {
|
|
241
|
+
used.l2 += 1;
|
|
242
|
+
return;
|
|
243
|
+
}
|
|
244
|
+
used.l3 += 1;
|
|
245
|
+
}
|
|
246
|
+
//# sourceMappingURL=context-slicer.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"context-slicer.js","sourceRoot":"","sources":["../../src/graph/context-slicer.ts"],"names":[],"mappings":";;AAsFA,wCA6BC;AAED,8CAOC;AAED,oCAgBC;AAED,gEAkFC;AAOD,gEA0CC;AAgBD,wCAEC;AArSD,4CAAyC;AAEzC,uDAKgC;AAGhC,IAAI,SAAS,GAAwC,IAAI,CAAC;AAC1D,IAAI,aAAa,GAAG,KAAK,CAAC;AAE1B,SAAS,UAAU;IACjB,IAAI,aAAa;QAAE,OAAO,SAAS,CAAC;IACpC,aAAa,GAAG,IAAI,CAAC;IACrB,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,OAAO,CAAC,mCAAmC,CAAwC,CAAC;QAChG,IAAI,OAAO,GAAG,CAAC,MAAM,KAAK,UAAU,EAAE,CAAC;YACrC,SAAS,GAAG,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACjC,OAAO,SAAS,CAAC;QACnB,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,eAAM,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,EAAE,cAAc,CAAC,CAAC;QACxC,eAAe;IACjB,CAAC;IACD,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,OAAO,CAAC,eAAe,CAAwC,CAAC;QAC5E,IAAI,OAAO,GAAG,CAAC,MAAM,KAAK,UAAU,EAAE,CAAC;YACrC,SAAS,GAAG,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACjC,OAAO,SAAS,CAAC;QACnB,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,eAAM,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,EAAE,cAAc,CAAC,CAAC;QACxC,eAAe;IACjB,CAAC;IACD,SAAS,GAAG,IAAI,CAAC;IACjB,OAAO,IAAI,CAAC;AACd,CAAC;AAyCD,MAAM,2BAA2B,GAA4B;IAC3D,YAAY;IACZ,SAAS;IACT,YAAY;IACZ,cAAc;CACf,CAAC;AAEK,KAAK,UAAU,cAAc,CAClC,MAAmB,EACnB,OAAiB,EACjB,OAAkC;IAElC,IAAI,OAAO,MAAM,CAAC,YAAY,KAAK,UAAU;QAAE,OAAO,EAAE,CAAC;IACzD,MAAM,IAAI,GAAG,OAAO,EAAE,IAAI,IAAI,CAAC,CAAC;IAChC,MAAM,QAAQ,GAAG,OAAO,EAAE,QAAQ,IAAI,EAAE,CAAC;IACzC,MAAM,SAAS,GAAG,OAAO,EAAE,SAAS,IAAI,2BAA2B,CAAC;IAEpE,MAAM,QAAQ,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,CAAC;IAClC,MAAM,SAAS,GAAG,IAAI,GAAG,EAAqB,CAAC;IAC/C,IAAI,QAAQ,GAAG,CAAC,GAAG,OAAO,CAAC,CAAC;IAE5B,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,IAAI,EAAE,KAAK,IAAI,CAAC,EAAE,CAAC;QAC7C,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC;YAAE,MAAM;QACjC,MAAM,SAAS,GAAG,MAAM,MAAM,CAAC,YAAY,CAAC,QAAQ,EAAE,SAAS,EAAE,MAAM,CAAC,CAAC;QACzE,MAAM,YAAY,GAAa,EAAE,CAAC;QAClC,KAAK,MAAM,EAAE,IAAI,EAAE,IAAI,SAAS,EAAE,CAAC;YACjC,IAAI,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;gBAAE,SAAS;YAC9D,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC;YAC7B,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAC3B,IAAI,SAAS,CAAC,IAAI,IAAI,QAAQ;gBAAE,MAAM;QACxC,CAAC;QACD,IAAI,SAAS,CAAC,IAAI,IAAI,QAAQ;YAAE,MAAM;QACtC,QAAQ,GAAG,YAAY,CAAC;IAC1B,CAAC;IAED,OAAO,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,CAAC,CAAC;AACxC,CAAC;AAEM,KAAK,UAAU,iBAAiB,CACrC,MAAmB,EACnB,KAAa,EACb,SAAiB;IAEjB,MAAM,GAAG,GAAG,MAAM,0BAA0B,CAAC,MAAM,EAAE,KAAK,EAAE,SAAS,CAAC,CAAC;IACvE,OAAO,EAAE,KAAK,EAAE,GAAG,CAAC,cAAc,EAAE,aAAa,EAAE,GAAG,CAAC,aAAa,EAAE,CAAC;AACzE,CAAC;AAED,SAAgB,YAAY,CAC1B,KAAkB,EAClB,cAAwB,EACxB,IAAY,EACZ,aAAqB;IAErB,IAAI,CAAC,cAAc,IAAI,cAAc,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,EAAE,CAAC;IAC9D,MAAM,MAAM,GAAuC,EAAE,CAAC;IACtD,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,MAAM,GAAG,GAAG,IAAA,6BAAgB,EAAC,IAAI,CAAC,CAAC;QACnC,IAAI,CAAC,GAAG;YAAE,SAAS;QACnB,MAAM,GAAG,GAAG,IAAA,6BAAgB,EAAC,cAAc,EAAE,GAAG,CAAC,CAAC;QAClD,IAAI,GAAG,IAAI,aAAa;YAAE,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC,CAAC;IACvD,CAAC;IACD,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC;IACrC,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;AAClD,CAAC;AAEM,KAAK,UAAU,0BAA0B,CAC9C,MAAmB,EACnB,KAAa,EACb,SAAiB,EACjB,OAA+B;IAE/B,MAAM,WAAW,GAAG,MAAM,MAAM,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC;IACvD,IAAI,IAAI,GAAgB,WAAW,CAAC;IAEpC,IAAI,OAAO,EAAE,kBAAkB,KAAK,IAAI,IAAI,OAAO,CAAC,iBAAiB,EAAE,CAAC;QACtE,IAAI,CAAC;YACH,MAAM,cAAc,GAAG,MAAM,OAAO,CAAC,iBAAiB,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YACpE,MAAM,IAAI,GAAG,OAAO,CAAC,UAAU,IAAI,CAAC,CAAC;YACrC,MAAM,MAAM,GAAG,OAAO,CAAC,mBAAmB,IAAI,IAAI,CAAC;YACnD,MAAM,UAAU,GAAG,YAAY,CAAC,WAAW,EAAE,cAAc,EAAE,IAAI,EAAE,MAAM,CAAC,CAAC;YAC3E,IAAI,GAAG,IAAA,iCAAoB,EAAC,CAAC,WAAW,EAAE,UAAU,CAAC,CAAC,CAAC;QACzD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACjB,eAAM,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,EAAE,cAAc,CAAC,CAAC;YACtC,IAAI,GAAG,WAAW,CAAC;QACrB,CAAC;IACH,CAAC;IAED,MAAM,cAAc,GAAa,EAAE,CAAC;IACpC,MAAM,aAAa,GAAwB,EAAE,CAAC;IAC9C,IAAI,MAAM,GAAG,CAAC,CAAC;IACf,IAAI,SAAS,GAAG,KAAK,CAAC;IAEtB,MAAM,KAAK,GAAG;QACZ,EAAE,EAAE,OAAO,EAAE,UAAU,EAAE,EAAE,IAAI,MAAM,CAAC,iBAAiB;QACvD,EAAE,EAAE,OAAO,EAAE,UAAU,EAAE,EAAE,IAAI,MAAM,CAAC,iBAAiB;QACvD,EAAE,EAAE,OAAO,EAAE,UAAU,EAAE,EAAE,IAAI,MAAM,CAAC,iBAAiB;KACxD,CAAC;IACF,MAAM,IAAI,GAAG,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC;IACrC,MAAM,KAAK,GAAG,IAAI,GAAG,EAAU,CAAC;IAEhC,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;QACvB,MAAM,KAAK,GAAG,aAAa,CAAC,GAAG,CAAC,CAAC;QACjC,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,KAAK,EAAE,IAAI,CAAC,EAAE,CAAC;YACrC,SAAS;QACX,CAAC;QAED,MAAM,OAAO,GAAG,GAAG,GAAG,CAAC,IAAI,KAAK,GAAG,CAAC,OAAO,EAAE,CAAC;QAC9C,MAAM,QAAQ,GAAG,cAAc,CAAC,OAAO,CAAC,CAAC;QACzC,IAAI,MAAM,GAAG,QAAQ,GAAG,SAAS,EAAE,CAAC;YAClC,SAAS,GAAG,IAAI,CAAC;YACjB,MAAM;QACR,CAAC;QAED,cAAc,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC7B,aAAa,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,IAAI,EAAE,GAAG,CAAC,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;QAC1D,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAClB,MAAM,IAAI,QAAQ,CAAC;QACnB,aAAa,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;IAC7B,CAAC;IAED,MAAM,eAAe,GAAG,OAAO,EAAE,mBAAmB,KAAK,KAAK,CAAC;IAC/D,IAAI,eAAe,IAAI,CAAC,SAAS,IAAI,OAAO,MAAM,CAAC,YAAY,KAAK,UAAU,EAAE,CAAC;QAC/E,MAAM,OAAO,GAAG,aAAa,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QAC3D,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACvB,MAAM,QAAQ,GAAG,MAAM,cAAc,CAAC,MAAM,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC;YACpE,KAAK,MAAM,IAAI,IAAI,QAAQ,EAAE,CAAC;gBAC5B,IAAI,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;oBAAE,SAAS;gBACjC,MAAM,KAAK,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC;gBAClC,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,KAAK,EAAE,IAAI,CAAC;oBAAE,SAAS;gBAE/C,MAAM,OAAO,GAAG,GAAG,IAAI,CAAC,IAAI,KAAK,IAAI,CAAC,OAAO,EAAE,CAAC;gBAChD,MAAM,QAAQ,GAAG,cAAc,CAAC,OAAO,CAAC,CAAC;gBACzC,IAAI,MAAM,GAAG,QAAQ,GAAG,SAAS,EAAE,CAAC;oBAClC,SAAS,GAAG,IAAI,CAAC;oBACjB,MAAM;gBACR,CAAC;gBAED,cAAc,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBAC7B,aAAa,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,IAAI,CAAC,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;gBAC5D,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBACnB,MAAM,IAAI,QAAQ,CAAC;gBACnB,aAAa,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;YAC7B,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,EAAE,cAAc,EAAE,aAAa,EAAE,aAAa,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC;AAC7E,CAAC;AAOD,SAAgB,0BAA0B,CACxC,MAAmB,EACnB,SAAiB,EACjB,OAA+B;IAE/B,MAAM,WAAW,GAAG,IAAI,GAAG,EAAU,CAAC;IAEtC,OAAO;QACL,KAAK,CAAC,cAAc,CAAC,KAAa;YAChC,MAAM,GAAG,GAAG,MAAM,0BAA0B,CAAC,MAAM,EAAE,KAAK,EAAE,SAAS,EAAE,OAAO,CAAC,CAAC;YAChF,KAAK,MAAM,MAAM,IAAI,GAAG,CAAC,aAAa,EAAE,CAAC;gBACvC,WAAW,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;YAC7B,CAAC;YACD,OAAO,GAAG,CAAC;QACb,CAAC;QACD,KAAK,CAAC,MAAM,CAAC,aAAuB;YAClC,MAAM,KAAK,GAAa,EAAE,CAAC;YAC3B,IAAI,MAAM,GAAG,CAAC,CAAC;YAEf,KAAK,MAAM,IAAI,IAAI,aAAa,EAAE,CAAC;gBACjC,MAAM,IAAI,GAAG,MAAM,MAAM,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC;gBAC/C,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;oBACvB,IAAI,WAAW,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC;wBAC5B,SAAS;oBACX,CAAC;oBAED,MAAM,IAAI,GAAG,GAAG,GAAG,CAAC,EAAE,MAAM,GAAG,CAAC,IAAI,KAAK,GAAG,CAAC,OAAO,EAAE,CAAC;oBACvD,MAAM,QAAQ,GAAG,cAAc,CAAC,IAAI,CAAC,CAAC;oBACtC,IAAI,MAAM,GAAG,QAAQ,GAAG,SAAS,EAAE,CAAC;wBAClC,OAAO,KAAK,CAAC;oBACf,CAAC;oBAED,WAAW,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;oBACxB,MAAM,IAAI,QAAQ,CAAC;oBACnB,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;oBACjB,MAAM;gBACR,CAAC;YACH,CAAC;YAED,OAAO,KAAK,CAAC;QACf,CAAC;KACF,CAAC;AACJ,CAAC;AAED,SAAS,cAAc,CAAC,IAAY;IAClC,MAAM,GAAG,GAAG,UAAU,EAAE,CAAC;IACzB,IAAI,GAAG,EAAE,CAAC;QACR,IAAI,CAAC;YACH,MAAM,CAAC,GAAG,GAAG,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC;YAC3B,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QACxB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACjB,eAAM,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,EAAE,cAAc,CAAC,CAAC;YACtC,kBAAkB;QACpB,CAAC;IACH,CAAC;IACD,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC;AACjD,CAAC;AAED,SAAgB,cAAc,CAAC,KAAkB;IAC/C,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,GAAG,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,EAAE,GAAG,CAAC,CAAC;AACzD,CAAC;AAED,SAAS,aAAa,CAAC,IAAe;IACpC,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,IAAI,IAAI,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;QACnD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,IAAI,IAAI,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;QAC3B,OAAO,IAAI,CAAC;IACd,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,WAAW,CAClB,KAAmB,EACnB,KAA6C,EAC7C,IAA4C;IAE5C,IAAI,KAAK,KAAK,IAAI,EAAE,CAAC;QACnB,OAAO,IAAI,CAAC,EAAE,GAAG,KAAK,CAAC,EAAE,CAAC;IAC5B,CAAC;IAED,IAAI,KAAK,KAAK,IAAI,EAAE,CAAC;QACnB,OAAO,IAAI,CAAC,EAAE,GAAG,KAAK,CAAC,EAAE,CAAC;IAC5B,CAAC;IAED,OAAO,IAAI,CAAC,EAAE,GAAG,KAAK,CAAC,EAAE,CAAC;AAC5B,CAAC;AAED,SAAS,aAAa,CAAC,KAAmB,EAAE,IAA4C;IACtF,IAAI,KAAK,KAAK,IAAI,EAAE,CAAC;QACnB,IAAI,CAAC,EAAE,IAAI,CAAC,CAAC;QACb,OAAO;IACT,CAAC;IAED,IAAI,KAAK,KAAK,IAAI,EAAE,CAAC;QACnB,IAAI,CAAC,EAAE,IAAI,CAAC,CAAC;QACb,OAAO;IACT,CAAC;IAED,IAAI,CAAC,EAAE,IAAI,CAAC,CAAC;AACf,CAAC"}
|
|
@@ -0,0 +1,15 @@
|
|
|
1
|
+
import type { GraphClient } from "./client-factory";
|
|
2
|
+
export interface FileIndexerOptions {
|
|
3
|
+
includeExtensions?: string[];
|
|
4
|
+
maxFileSizeBytes?: number;
|
|
5
|
+
/** When true, ignore index cache and re-process every file. */
|
|
6
|
+
forceReindex?: boolean;
|
|
7
|
+
}
|
|
8
|
+
/** Remove graph store, index cache, and vector DB for a full rebuild. */
|
|
9
|
+
export declare function clearGraphIndexArtifacts(rootDir: string, graphStorePath: string): void;
|
|
10
|
+
export declare function indexWorkspaceFiles(client: GraphClient, rootDir: string, options?: FileIndexerOptions): Promise<{
|
|
11
|
+
indexedFiles: number;
|
|
12
|
+
indexedSymbols: number;
|
|
13
|
+
indexedReferences: number;
|
|
14
|
+
}>;
|
|
15
|
+
//# sourceMappingURL=file-indexer.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"file-indexer.d.ts","sourceRoot":"","sources":["../../src/graph/file-indexer.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAOpD,MAAM,WAAW,kBAAkB;IACjC,iBAAiB,CAAC,EAAE,MAAM,EAAE,CAAC;IAC7B,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,+DAA+D;IAC/D,YAAY,CAAC,EAAE,OAAO,CAAC;CACxB;AAoDD,yEAAyE;AACzE,wBAAgB,wBAAwB,CAAC,OAAO,EAAE,MAAM,EAAE,cAAc,EAAE,MAAM,GAAG,IAAI,CAMtF;AAED,wBAAsB,mBAAmB,CACvC,MAAM,EAAE,WAAW,EACnB,OAAO,EAAE,MAAM,EACf,OAAO,CAAC,EAAE,kBAAkB,GAC3B,OAAO,CAAC;IAAE,YAAY,EAAE,MAAM,CAAC;IAAC,cAAc,EAAE,MAAM,CAAC;IAAC,iBAAiB,EAAE,MAAM,CAAA;CAAE,CAAC,CAmPtF"}
|