causantic 0.2.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/LICENSE +19 -0
- package/README.md +332 -0
- package/config.schema.json +254 -0
- package/dist/cli/commands/archive.d.ts +4 -0
- package/dist/cli/commands/archive.d.ts.map +1 -0
- package/dist/cli/commands/archive.js +76 -0
- package/dist/cli/commands/archive.js.map +1 -0
- package/dist/cli/commands/benchmark-collection.d.ts +8 -0
- package/dist/cli/commands/benchmark-collection.d.ts.map +1 -0
- package/dist/cli/commands/benchmark-collection.js +146 -0
- package/dist/cli/commands/benchmark-collection.js.map +1 -0
- package/dist/cli/commands/config.d.ts +3 -0
- package/dist/cli/commands/config.d.ts.map +1 -0
- package/dist/cli/commands/config.js +69 -0
- package/dist/cli/commands/config.js.map +1 -0
- package/dist/cli/commands/dashboard.d.ts +3 -0
- package/dist/cli/commands/dashboard.d.ts.map +1 -0
- package/dist/cli/commands/dashboard.js +12 -0
- package/dist/cli/commands/dashboard.js.map +1 -0
- package/dist/cli/commands/encryption.d.ts +3 -0
- package/dist/cli/commands/encryption.d.ts.map +1 -0
- package/dist/cli/commands/encryption.js +207 -0
- package/dist/cli/commands/encryption.js.map +1 -0
- package/dist/cli/commands/hook.d.ts +3 -0
- package/dist/cli/commands/hook.d.ts.map +1 -0
- package/dist/cli/commands/hook.js +35 -0
- package/dist/cli/commands/hook.js.map +1 -0
- package/dist/cli/commands/ingest.d.ts +4 -0
- package/dist/cli/commands/ingest.d.ts.map +1 -0
- package/dist/cli/commands/ingest.js +31 -0
- package/dist/cli/commands/ingest.js.map +1 -0
- package/dist/cli/commands/init.d.ts +3 -0
- package/dist/cli/commands/init.d.ts.map +1 -0
- package/dist/cli/commands/init.js +680 -0
- package/dist/cli/commands/init.js.map +1 -0
- package/dist/cli/commands/maintenance.d.ts +3 -0
- package/dist/cli/commands/maintenance.d.ts.map +1 -0
- package/dist/cli/commands/maintenance.js +67 -0
- package/dist/cli/commands/maintenance.js.map +1 -0
- package/dist/cli/commands/search.d.ts +3 -0
- package/dist/cli/commands/search.d.ts.map +1 -0
- package/dist/cli/commands/search.js +16 -0
- package/dist/cli/commands/search.js.map +1 -0
- package/dist/cli/commands/serve.d.ts +3 -0
- package/dist/cli/commands/serve.d.ts.map +1 -0
- package/dist/cli/commands/serve.js +19 -0
- package/dist/cli/commands/serve.js.map +1 -0
- package/dist/cli/commands/stats.d.ts +4 -0
- package/dist/cli/commands/stats.d.ts.map +1 -0
- package/dist/cli/commands/stats.js +49 -0
- package/dist/cli/commands/stats.js.map +1 -0
- package/dist/cli/commands/uninstall.d.ts +3 -0
- package/dist/cli/commands/uninstall.d.ts.map +1 -0
- package/dist/cli/commands/uninstall.js +10 -0
- package/dist/cli/commands/uninstall.js.map +1 -0
- package/dist/cli/index.d.ts +8 -0
- package/dist/cli/index.d.ts.map +1 -0
- package/dist/cli/index.js +86 -0
- package/dist/cli/index.js.map +1 -0
- package/dist/cli/skill-templates.d.ts +19 -0
- package/dist/cli/skill-templates.d.ts.map +1 -0
- package/dist/cli/skill-templates.js +1128 -0
- package/dist/cli/skill-templates.js.map +1 -0
- package/dist/cli/types.d.ts +10 -0
- package/dist/cli/types.d.ts.map +1 -0
- package/dist/cli/types.js +5 -0
- package/dist/cli/types.js.map +1 -0
- package/dist/cli/uninstall.d.ts +66 -0
- package/dist/cli/uninstall.d.ts.map +1 -0
- package/dist/cli/uninstall.js +490 -0
- package/dist/cli/uninstall.js.map +1 -0
- package/dist/cli/utils.d.ts +26 -0
- package/dist/cli/utils.d.ts.map +1 -0
- package/dist/cli/utils.js +105 -0
- package/dist/cli/utils.js.map +1 -0
- package/dist/clusters/cluster-manager.d.ts +96 -0
- package/dist/clusters/cluster-manager.d.ts.map +1 -0
- package/dist/clusters/cluster-manager.js +321 -0
- package/dist/clusters/cluster-manager.js.map +1 -0
- package/dist/clusters/cluster-refresh.d.ts +56 -0
- package/dist/clusters/cluster-refresh.d.ts.map +1 -0
- package/dist/clusters/cluster-refresh.js +187 -0
- package/dist/clusters/cluster-refresh.js.map +1 -0
- package/dist/clusters/hdbscan/cluster-extraction.d.ts +26 -0
- package/dist/clusters/hdbscan/cluster-extraction.d.ts.map +1 -0
- package/dist/clusters/hdbscan/cluster-extraction.js +232 -0
- package/dist/clusters/hdbscan/cluster-extraction.js.map +1 -0
- package/dist/clusters/hdbscan/core-distance-worker.d.ts +6 -0
- package/dist/clusters/hdbscan/core-distance-worker.d.ts.map +1 -0
- package/dist/clusters/hdbscan/core-distance-worker.js +54 -0
- package/dist/clusters/hdbscan/core-distance-worker.js.map +1 -0
- package/dist/clusters/hdbscan/core-distance.d.ts +26 -0
- package/dist/clusters/hdbscan/core-distance.d.ts.map +1 -0
- package/dist/clusters/hdbscan/core-distance.js +94 -0
- package/dist/clusters/hdbscan/core-distance.js.map +1 -0
- package/dist/clusters/hdbscan/distance-cache.d.ts +46 -0
- package/dist/clusters/hdbscan/distance-cache.d.ts.map +1 -0
- package/dist/clusters/hdbscan/distance-cache.js +84 -0
- package/dist/clusters/hdbscan/distance-cache.js.map +1 -0
- package/dist/clusters/hdbscan/hierarchy.d.ts +34 -0
- package/dist/clusters/hdbscan/hierarchy.d.ts.map +1 -0
- package/dist/clusters/hdbscan/hierarchy.js +442 -0
- package/dist/clusters/hdbscan/hierarchy.js.map +1 -0
- package/dist/clusters/hdbscan/incremental.d.ts +30 -0
- package/dist/clusters/hdbscan/incremental.d.ts.map +1 -0
- package/dist/clusters/hdbscan/incremental.js +130 -0
- package/dist/clusters/hdbscan/incremental.js.map +1 -0
- package/dist/clusters/hdbscan/kd-tree.d.ts +51 -0
- package/dist/clusters/hdbscan/kd-tree.d.ts.map +1 -0
- package/dist/clusters/hdbscan/kd-tree.js +130 -0
- package/dist/clusters/hdbscan/kd-tree.js.map +1 -0
- package/dist/clusters/hdbscan/min-heap.d.ts +62 -0
- package/dist/clusters/hdbscan/min-heap.d.ts.map +1 -0
- package/dist/clusters/hdbscan/min-heap.js +139 -0
- package/dist/clusters/hdbscan/min-heap.js.map +1 -0
- package/dist/clusters/hdbscan/mst.d.ts +20 -0
- package/dist/clusters/hdbscan/mst.d.ts.map +1 -0
- package/dist/clusters/hdbscan/mst.js +77 -0
- package/dist/clusters/hdbscan/mst.js.map +1 -0
- package/dist/clusters/hdbscan/probabilities.d.ts +37 -0
- package/dist/clusters/hdbscan/probabilities.d.ts.map +1 -0
- package/dist/clusters/hdbscan/probabilities.js +149 -0
- package/dist/clusters/hdbscan/probabilities.js.map +1 -0
- package/dist/clusters/hdbscan/types.d.ts +123 -0
- package/dist/clusters/hdbscan/types.d.ts.map +1 -0
- package/dist/clusters/hdbscan/types.js +5 -0
- package/dist/clusters/hdbscan/types.js.map +1 -0
- package/dist/clusters/hdbscan/union-find.d.ts +51 -0
- package/dist/clusters/hdbscan/union-find.d.ts.map +1 -0
- package/dist/clusters/hdbscan/union-find.js +115 -0
- package/dist/clusters/hdbscan/union-find.js.map +1 -0
- package/dist/clusters/hdbscan.d.ts +71 -0
- package/dist/clusters/hdbscan.d.ts.map +1 -0
- package/dist/clusters/hdbscan.js +283 -0
- package/dist/clusters/hdbscan.js.map +1 -0
- package/dist/clusters/index.d.ts +8 -0
- package/dist/clusters/index.d.ts.map +1 -0
- package/dist/clusters/index.js +8 -0
- package/dist/clusters/index.js.map +1 -0
- package/dist/config/index.d.ts +6 -0
- package/dist/config/index.d.ts.map +1 -0
- package/dist/config/index.js +6 -0
- package/dist/config/index.js.map +1 -0
- package/dist/config/loader.d.ts +105 -0
- package/dist/config/loader.d.ts.map +1 -0
- package/dist/config/loader.js +339 -0
- package/dist/config/loader.js.map +1 -0
- package/dist/config/memory-config.d.ts +109 -0
- package/dist/config/memory-config.d.ts.map +1 -0
- package/dist/config/memory-config.js +182 -0
- package/dist/config/memory-config.js.map +1 -0
- package/dist/core/benchmark-types.d.ts +62 -0
- package/dist/core/benchmark-types.d.ts.map +1 -0
- package/dist/core/benchmark-types.js +11 -0
- package/dist/core/benchmark-types.js.map +1 -0
- package/dist/core/decay-types.d.ts +111 -0
- package/dist/core/decay-types.d.ts.map +1 -0
- package/dist/core/decay-types.js +30 -0
- package/dist/core/decay-types.js.map +1 -0
- package/dist/core/index.d.ts +11 -0
- package/dist/core/index.d.ts.map +1 -0
- package/dist/core/index.js +9 -0
- package/dist/core/index.js.map +1 -0
- package/dist/core/lexical-features.d.ts +34 -0
- package/dist/core/lexical-features.d.ts.map +1 -0
- package/dist/core/lexical-features.js +155 -0
- package/dist/core/lexical-features.js.map +1 -0
- package/dist/dashboard/client/assets/index-CMUKy4f9.css +1 -0
- package/dist/dashboard/client/assets/index-esv8TpCN.js +168 -0
- package/dist/dashboard/client/index.html +14 -0
- package/dist/dashboard/middleware/async-handler.d.ts +11 -0
- package/dist/dashboard/middleware/async-handler.d.ts.map +1 -0
- package/dist/dashboard/middleware/async-handler.js +12 -0
- package/dist/dashboard/middleware/async-handler.js.map +1 -0
- package/dist/dashboard/middleware/error-handler.d.ts +9 -0
- package/dist/dashboard/middleware/error-handler.d.ts.map +1 -0
- package/dist/dashboard/middleware/error-handler.js +14 -0
- package/dist/dashboard/middleware/error-handler.js.map +1 -0
- package/dist/dashboard/routes/benchmark-collection.d.ts +11 -0
- package/dist/dashboard/routes/benchmark-collection.d.ts.map +1 -0
- package/dist/dashboard/routes/benchmark-collection.js +49 -0
- package/dist/dashboard/routes/benchmark-collection.js.map +1 -0
- package/dist/dashboard/routes/chunks.d.ts +3 -0
- package/dist/dashboard/routes/chunks.d.ts.map +1 -0
- package/dist/dashboard/routes/chunks.js +25 -0
- package/dist/dashboard/routes/chunks.js.map +1 -0
- package/dist/dashboard/routes/clusters.d.ts +3 -0
- package/dist/dashboard/routes/clusters.d.ts.map +1 -0
- package/dist/dashboard/routes/clusters.js +28 -0
- package/dist/dashboard/routes/clusters.js.map +1 -0
- package/dist/dashboard/routes/edges.d.ts +3 -0
- package/dist/dashboard/routes/edges.d.ts.map +1 -0
- package/dist/dashboard/routes/edges.js +32 -0
- package/dist/dashboard/routes/edges.js.map +1 -0
- package/dist/dashboard/routes/graph.d.ts +3 -0
- package/dist/dashboard/routes/graph.d.ts.map +1 -0
- package/dist/dashboard/routes/graph.js +158 -0
- package/dist/dashboard/routes/graph.js.map +1 -0
- package/dist/dashboard/routes/projects.d.ts +3 -0
- package/dist/dashboard/routes/projects.d.ts.map +1 -0
- package/dist/dashboard/routes/projects.js +9 -0
- package/dist/dashboard/routes/projects.js.map +1 -0
- package/dist/dashboard/routes/search.d.ts +3 -0
- package/dist/dashboard/routes/search.d.ts.map +1 -0
- package/dist/dashboard/routes/search.js +105 -0
- package/dist/dashboard/routes/search.js.map +1 -0
- package/dist/dashboard/routes/sessions.d.ts +3 -0
- package/dist/dashboard/routes/sessions.d.ts.map +1 -0
- package/dist/dashboard/routes/sessions.js +16 -0
- package/dist/dashboard/routes/sessions.js.map +1 -0
- package/dist/dashboard/routes/stats.d.ts +3 -0
- package/dist/dashboard/routes/stats.d.ts.map +1 -0
- package/dist/dashboard/routes/stats.js +38 -0
- package/dist/dashboard/routes/stats.js.map +1 -0
- package/dist/dashboard/server.d.ts +3 -0
- package/dist/dashboard/server.d.ts.map +1 -0
- package/dist/dashboard/server.js +85 -0
- package/dist/dashboard/server.js.map +1 -0
- package/dist/eval/annotation-schema.d.ts +41 -0
- package/dist/eval/annotation-schema.d.ts.map +1 -0
- package/dist/eval/annotation-schema.js +171 -0
- package/dist/eval/annotation-schema.js.map +1 -0
- package/dist/eval/benchmark-runner.d.ts +15 -0
- package/dist/eval/benchmark-runner.d.ts.map +1 -0
- package/dist/eval/benchmark-runner.js +93 -0
- package/dist/eval/benchmark-runner.js.map +1 -0
- package/dist/eval/cluster-evaluator.d.ts +39 -0
- package/dist/eval/cluster-evaluator.d.ts.map +1 -0
- package/dist/eval/cluster-evaluator.js +51 -0
- package/dist/eval/cluster-evaluator.js.map +1 -0
- package/dist/eval/code-nl-alignment.d.ts +25 -0
- package/dist/eval/code-nl-alignment.d.ts.map +1 -0
- package/dist/eval/code-nl-alignment.js +49 -0
- package/dist/eval/code-nl-alignment.js.map +1 -0
- package/dist/eval/collection-benchmark/graph-value.d.ts +16 -0
- package/dist/eval/collection-benchmark/graph-value.d.ts.map +1 -0
- package/dist/eval/collection-benchmark/graph-value.js +162 -0
- package/dist/eval/collection-benchmark/graph-value.js.map +1 -0
- package/dist/eval/collection-benchmark/health.d.ts +12 -0
- package/dist/eval/collection-benchmark/health.d.ts.map +1 -0
- package/dist/eval/collection-benchmark/health.js +246 -0
- package/dist/eval/collection-benchmark/health.js.map +1 -0
- package/dist/eval/collection-benchmark/history.d.ts +27 -0
- package/dist/eval/collection-benchmark/history.d.ts.map +1 -0
- package/dist/eval/collection-benchmark/history.js +134 -0
- package/dist/eval/collection-benchmark/history.js.map +1 -0
- package/dist/eval/collection-benchmark/latency.d.ts +16 -0
- package/dist/eval/collection-benchmark/latency.d.ts.map +1 -0
- package/dist/eval/collection-benchmark/latency.js +115 -0
- package/dist/eval/collection-benchmark/latency.js.map +1 -0
- package/dist/eval/collection-benchmark/reporter.d.ts +18 -0
- package/dist/eval/collection-benchmark/reporter.d.ts.map +1 -0
- package/dist/eval/collection-benchmark/reporter.js +244 -0
- package/dist/eval/collection-benchmark/reporter.js.map +1 -0
- package/dist/eval/collection-benchmark/retrieval.d.ts +15 -0
- package/dist/eval/collection-benchmark/retrieval.d.ts.map +1 -0
- package/dist/eval/collection-benchmark/retrieval.js +203 -0
- package/dist/eval/collection-benchmark/retrieval.js.map +1 -0
- package/dist/eval/collection-benchmark/runner.d.ts +20 -0
- package/dist/eval/collection-benchmark/runner.d.ts.map +1 -0
- package/dist/eval/collection-benchmark/runner.js +189 -0
- package/dist/eval/collection-benchmark/runner.js.map +1 -0
- package/dist/eval/collection-benchmark/sampler.d.ts +18 -0
- package/dist/eval/collection-benchmark/sampler.d.ts.map +1 -0
- package/dist/eval/collection-benchmark/sampler.js +186 -0
- package/dist/eval/collection-benchmark/sampler.js.map +1 -0
- package/dist/eval/collection-benchmark/tuning.d.ts +12 -0
- package/dist/eval/collection-benchmark/tuning.d.ts.map +1 -0
- package/dist/eval/collection-benchmark/tuning.js +222 -0
- package/dist/eval/collection-benchmark/tuning.js.map +1 -0
- package/dist/eval/collection-benchmark/types.d.ts +183 -0
- package/dist/eval/collection-benchmark/types.d.ts.map +1 -0
- package/dist/eval/collection-benchmark/types.js +7 -0
- package/dist/eval/collection-benchmark/types.js.map +1 -0
- package/dist/eval/context-window-test.d.ts +34 -0
- package/dist/eval/context-window-test.d.ts.map +1 -0
- package/dist/eval/context-window-test.js +59 -0
- package/dist/eval/context-window-test.js.map +1 -0
- package/dist/eval/corpus-builder.d.ts +33 -0
- package/dist/eval/corpus-builder.d.ts.map +1 -0
- package/dist/eval/corpus-builder.js +88 -0
- package/dist/eval/corpus-builder.js.map +1 -0
- package/dist/eval/experiments/boilerplate-filter.d.ts +26 -0
- package/dist/eval/experiments/boilerplate-filter.d.ts.map +1 -0
- package/dist/eval/experiments/boilerplate-filter.js +95 -0
- package/dist/eval/experiments/boilerplate-filter.js.map +1 -0
- package/dist/eval/experiments/cluster-threshold/index.d.ts +6 -0
- package/dist/eval/experiments/cluster-threshold/index.d.ts.map +1 -0
- package/dist/eval/experiments/cluster-threshold/index.js +5 -0
- package/dist/eval/experiments/cluster-threshold/index.js.map +1 -0
- package/dist/eval/experiments/cluster-threshold/run-threshold-sweep.d.ts +58 -0
- package/dist/eval/experiments/cluster-threshold/run-threshold-sweep.d.ts.map +1 -0
- package/dist/eval/experiments/cluster-threshold/run-threshold-sweep.js +246 -0
- package/dist/eval/experiments/cluster-threshold/run-threshold-sweep.js.map +1 -0
- package/dist/eval/experiments/code-focused-mode.d.ts +20 -0
- package/dist/eval/experiments/code-focused-mode.d.ts.map +1 -0
- package/dist/eval/experiments/code-focused-mode.js +66 -0
- package/dist/eval/experiments/code-focused-mode.js.map +1 -0
- package/dist/eval/experiments/edge-decay/decay-curves.d.ts +59 -0
- package/dist/eval/experiments/edge-decay/decay-curves.d.ts.map +1 -0
- package/dist/eval/experiments/edge-decay/decay-curves.js +124 -0
- package/dist/eval/experiments/edge-decay/decay-curves.js.map +1 -0
- package/dist/eval/experiments/edge-decay/index.d.ts +13 -0
- package/dist/eval/experiments/edge-decay/index.d.ts.map +1 -0
- package/dist/eval/experiments/edge-decay/index.js +16 -0
- package/dist/eval/experiments/edge-decay/index.js.map +1 -0
- package/dist/eval/experiments/edge-decay/presets.d.ts +52 -0
- package/dist/eval/experiments/edge-decay/presets.d.ts.map +1 -0
- package/dist/eval/experiments/edge-decay/presets.js +234 -0
- package/dist/eval/experiments/edge-decay/presets.js.map +1 -0
- package/dist/eval/experiments/edge-decay/reference-extractor.d.ts +34 -0
- package/dist/eval/experiments/edge-decay/reference-extractor.d.ts.map +1 -0
- package/dist/eval/experiments/edge-decay/reference-extractor.js +353 -0
- package/dist/eval/experiments/edge-decay/reference-extractor.js.map +1 -0
- package/dist/eval/experiments/edge-decay/reference-types.d.ts +131 -0
- package/dist/eval/experiments/edge-decay/reference-types.d.ts.map +1 -0
- package/dist/eval/experiments/edge-decay/reference-types.js +5 -0
- package/dist/eval/experiments/edge-decay/reference-types.js.map +1 -0
- package/dist/eval/experiments/edge-decay/retrieval-ranking.d.ts +64 -0
- package/dist/eval/experiments/edge-decay/retrieval-ranking.d.ts.map +1 -0
- package/dist/eval/experiments/edge-decay/retrieval-ranking.js +565 -0
- package/dist/eval/experiments/edge-decay/retrieval-ranking.js.map +1 -0
- package/dist/eval/experiments/edge-decay/run-experiments.d.ts +25 -0
- package/dist/eval/experiments/edge-decay/run-experiments.d.ts.map +1 -0
- package/dist/eval/experiments/edge-decay/run-experiments.js +229 -0
- package/dist/eval/experiments/edge-decay/run-experiments.js.map +1 -0
- package/dist/eval/experiments/edge-decay/simulate.d.ts +42 -0
- package/dist/eval/experiments/edge-decay/simulate.d.ts.map +1 -0
- package/dist/eval/experiments/edge-decay/simulate.js +232 -0
- package/dist/eval/experiments/edge-decay/simulate.js.map +1 -0
- package/dist/eval/experiments/edge-decay/types.d.ts +10 -0
- package/dist/eval/experiments/edge-decay/types.d.ts.map +1 -0
- package/dist/eval/experiments/edge-decay/types.js +9 -0
- package/dist/eval/experiments/edge-decay/types.js.map +1 -0
- package/dist/eval/experiments/hdbscan-sweep.d.ts +19 -0
- package/dist/eval/experiments/hdbscan-sweep.d.ts.map +1 -0
- package/dist/eval/experiments/hdbscan-sweep.js +71 -0
- package/dist/eval/experiments/hdbscan-sweep.js.map +1 -0
- package/dist/eval/experiments/single-model-run.d.ts +54 -0
- package/dist/eval/experiments/single-model-run.d.ts.map +1 -0
- package/dist/eval/experiments/single-model-run.js +120 -0
- package/dist/eval/experiments/single-model-run.js.map +1 -0
- package/dist/eval/experiments/thinking-ablation.d.ts +21 -0
- package/dist/eval/experiments/thinking-ablation.d.ts.map +1 -0
- package/dist/eval/experiments/thinking-ablation.js +68 -0
- package/dist/eval/experiments/thinking-ablation.js.map +1 -0
- package/dist/eval/experiments/topic-continuity/embedding-classifier.d.ts +55 -0
- package/dist/eval/experiments/topic-continuity/embedding-classifier.d.ts.map +1 -0
- package/dist/eval/experiments/topic-continuity/embedding-classifier.js +126 -0
- package/dist/eval/experiments/topic-continuity/embedding-classifier.js.map +1 -0
- package/dist/eval/experiments/topic-continuity/hybrid-classifier.d.ts +68 -0
- package/dist/eval/experiments/topic-continuity/hybrid-classifier.d.ts.map +1 -0
- package/dist/eval/experiments/topic-continuity/hybrid-classifier.js +175 -0
- package/dist/eval/experiments/topic-continuity/hybrid-classifier.js.map +1 -0
- package/dist/eval/experiments/topic-continuity/index.d.ts +13 -0
- package/dist/eval/experiments/topic-continuity/index.d.ts.map +1 -0
- package/dist/eval/experiments/topic-continuity/index.js +17 -0
- package/dist/eval/experiments/topic-continuity/index.js.map +1 -0
- package/dist/eval/experiments/topic-continuity/labeler.d.ts +42 -0
- package/dist/eval/experiments/topic-continuity/labeler.d.ts.map +1 -0
- package/dist/eval/experiments/topic-continuity/labeler.js +253 -0
- package/dist/eval/experiments/topic-continuity/labeler.js.map +1 -0
- package/dist/eval/experiments/topic-continuity/lexical-features.d.ts +26 -0
- package/dist/eval/experiments/topic-continuity/lexical-features.d.ts.map +1 -0
- package/dist/eval/experiments/topic-continuity/lexical-features.js +58 -0
- package/dist/eval/experiments/topic-continuity/lexical-features.js.map +1 -0
- package/dist/eval/experiments/topic-continuity/run-experiment.d.ts +40 -0
- package/dist/eval/experiments/topic-continuity/run-experiment.d.ts.map +1 -0
- package/dist/eval/experiments/topic-continuity/run-experiment.js +379 -0
- package/dist/eval/experiments/topic-continuity/run-experiment.js.map +1 -0
- package/dist/eval/experiments/topic-continuity/types.d.ts +146 -0
- package/dist/eval/experiments/topic-continuity/types.d.ts.map +1 -0
- package/dist/eval/experiments/topic-continuity/types.js +9 -0
- package/dist/eval/experiments/topic-continuity/types.js.map +1 -0
- package/dist/eval/experiments/truncation.d.ts +21 -0
- package/dist/eval/experiments/truncation.d.ts.map +1 -0
- package/dist/eval/experiments/truncation.js +59 -0
- package/dist/eval/experiments/truncation.js.map +1 -0
- package/dist/eval/experiments/types.d.ts +37 -0
- package/dist/eval/experiments/types.d.ts.map +1 -0
- package/dist/eval/experiments/types.js +13 -0
- package/dist/eval/experiments/types.js.map +1 -0
- package/dist/eval/experiments/vector-decay-shapes/curve-shapes-experiment.d.ts +10 -0
- package/dist/eval/experiments/vector-decay-shapes/curve-shapes-experiment.d.ts.map +1 -0
- package/dist/eval/experiments/vector-decay-shapes/curve-shapes-experiment.js +229 -0
- package/dist/eval/experiments/vector-decay-shapes/curve-shapes-experiment.js.map +1 -0
- package/dist/eval/experiments/vector-decay-shapes/graph-range-experiment.d.ts +12 -0
- package/dist/eval/experiments/vector-decay-shapes/graph-range-experiment.d.ts.map +1 -0
- package/dist/eval/experiments/vector-decay-shapes/graph-range-experiment.js +316 -0
- package/dist/eval/experiments/vector-decay-shapes/graph-range-experiment.js.map +1 -0
- package/dist/eval/experiments/vector-decay-shapes/hop-decay.d.ts +19 -0
- package/dist/eval/experiments/vector-decay-shapes/hop-decay.d.ts.map +1 -0
- package/dist/eval/experiments/vector-decay-shapes/hop-decay.js +127 -0
- package/dist/eval/experiments/vector-decay-shapes/hop-decay.js.map +1 -0
- package/dist/eval/experiments/vector-decay-shapes/index.d.ts +11 -0
- package/dist/eval/experiments/vector-decay-shapes/index.d.ts.map +1 -0
- package/dist/eval/experiments/vector-decay-shapes/index.js +11 -0
- package/dist/eval/experiments/vector-decay-shapes/index.js.map +1 -0
- package/dist/eval/experiments/vector-decay-shapes/path-traversal-experiment.d.ts +14 -0
- package/dist/eval/experiments/vector-decay-shapes/path-traversal-experiment.d.ts.map +1 -0
- package/dist/eval/experiments/vector-decay-shapes/path-traversal-experiment.js +274 -0
- package/dist/eval/experiments/vector-decay-shapes/path-traversal-experiment.js.map +1 -0
- package/dist/eval/experiments/vector-decay-shapes/presets.d.ts +36 -0
- package/dist/eval/experiments/vector-decay-shapes/presets.d.ts.map +1 -0
- package/dist/eval/experiments/vector-decay-shapes/presets.js +157 -0
- package/dist/eval/experiments/vector-decay-shapes/presets.js.map +1 -0
- package/dist/eval/experiments/vector-decay-shapes/run-experiment.d.ts +20 -0
- package/dist/eval/experiments/vector-decay-shapes/run-experiment.d.ts.map +1 -0
- package/dist/eval/experiments/vector-decay-shapes/run-experiment.js +504 -0
- package/dist/eval/experiments/vector-decay-shapes/run-experiment.js.map +1 -0
- package/dist/eval/experiments/vector-decay-shapes/types.d.ts +111 -0
- package/dist/eval/experiments/vector-decay-shapes/types.d.ts.map +1 -0
- package/dist/eval/experiments/vector-decay-shapes/types.js +15 -0
- package/dist/eval/experiments/vector-decay-shapes/types.js.map +1 -0
- package/dist/eval/experiments/vector-decay-sweep/index.d.ts +7 -0
- package/dist/eval/experiments/vector-decay-sweep/index.d.ts.map +1 -0
- package/dist/eval/experiments/vector-decay-sweep/index.js +7 -0
- package/dist/eval/experiments/vector-decay-sweep/index.js.map +1 -0
- package/dist/eval/experiments/vector-decay-sweep/run-sweep.d.ts +30 -0
- package/dist/eval/experiments/vector-decay-sweep/run-sweep.d.ts.map +1 -0
- package/dist/eval/experiments/vector-decay-sweep/run-sweep.js +378 -0
- package/dist/eval/experiments/vector-decay-sweep/run-sweep.js.map +1 -0
- package/dist/eval/experiments/vector-decay-sweep/types.d.ts +109 -0
- package/dist/eval/experiments/vector-decay-sweep/types.d.ts.map +1 -0
- package/dist/eval/experiments/vector-decay-sweep/types.js +16 -0
- package/dist/eval/experiments/vector-decay-sweep/types.js.map +1 -0
- package/dist/eval/metrics.d.ts +45 -0
- package/dist/eval/metrics.d.ts.map +1 -0
- package/dist/eval/metrics.js +134 -0
- package/dist/eval/metrics.js.map +1 -0
- package/dist/hooks/claudemd-generator.d.ts +62 -0
- package/dist/hooks/claudemd-generator.d.ts.map +1 -0
- package/dist/hooks/claudemd-generator.js +220 -0
- package/dist/hooks/claudemd-generator.js.map +1 -0
- package/dist/hooks/hook-utils.d.ts +84 -0
- package/dist/hooks/hook-utils.d.ts.map +1 -0
- package/dist/hooks/hook-utils.js +187 -0
- package/dist/hooks/hook-utils.js.map +1 -0
- package/dist/hooks/index.d.ts +12 -0
- package/dist/hooks/index.d.ts.map +1 -0
- package/dist/hooks/index.js +12 -0
- package/dist/hooks/index.js.map +1 -0
- package/dist/hooks/pre-compact.d.ts +58 -0
- package/dist/hooks/pre-compact.d.ts.map +1 -0
- package/dist/hooks/pre-compact.js +127 -0
- package/dist/hooks/pre-compact.js.map +1 -0
- package/dist/hooks/session-start.d.ts +62 -0
- package/dist/hooks/session-start.d.ts.map +1 -0
- package/dist/hooks/session-start.js +249 -0
- package/dist/hooks/session-start.js.map +1 -0
- package/dist/index.d.ts +22 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +29 -0
- package/dist/index.js.map +1 -0
- package/dist/ingest/batch-ingest.d.ts +94 -0
- package/dist/ingest/batch-ingest.d.ts.map +1 -0
- package/dist/ingest/batch-ingest.js +167 -0
- package/dist/ingest/batch-ingest.js.map +1 -0
- package/dist/ingest/brief-debrief-detector.d.ts +74 -0
- package/dist/ingest/brief-debrief-detector.d.ts.map +1 -0
- package/dist/ingest/brief-debrief-detector.js +253 -0
- package/dist/ingest/brief-debrief-detector.js.map +1 -0
- package/dist/ingest/cross-session-linker.d.ts +39 -0
- package/dist/ingest/cross-session-linker.d.ts.map +1 -0
- package/dist/ingest/cross-session-linker.js +103 -0
- package/dist/ingest/cross-session-linker.js.map +1 -0
- package/dist/ingest/edge-creator.d.ts +62 -0
- package/dist/ingest/edge-creator.d.ts.map +1 -0
- package/dist/ingest/edge-creator.js +186 -0
- package/dist/ingest/edge-creator.js.map +1 -0
- package/dist/ingest/edge-detector.d.ts +37 -0
- package/dist/ingest/edge-detector.d.ts.map +1 -0
- package/dist/ingest/edge-detector.js +234 -0
- package/dist/ingest/edge-detector.js.map +1 -0
- package/dist/ingest/index.d.ts +14 -0
- package/dist/ingest/index.d.ts.map +1 -0
- package/dist/ingest/index.js +14 -0
- package/dist/ingest/index.js.map +1 -0
- package/dist/ingest/ingest-session.d.ts +83 -0
- package/dist/ingest/ingest-session.d.ts.map +1 -0
- package/dist/ingest/ingest-session.js +415 -0
- package/dist/ingest/ingest-session.js.map +1 -0
- package/dist/maintenance/index.d.ts +5 -0
- package/dist/maintenance/index.d.ts.map +1 -0
- package/dist/maintenance/index.js +5 -0
- package/dist/maintenance/index.js.map +1 -0
- package/dist/maintenance/scheduler.d.ts +79 -0
- package/dist/maintenance/scheduler.d.ts.map +1 -0
- package/dist/maintenance/scheduler.js +362 -0
- package/dist/maintenance/scheduler.js.map +1 -0
- package/dist/maintenance/tasks/cleanup-vectors.d.ts +10 -0
- package/dist/maintenance/tasks/cleanup-vectors.d.ts.map +1 -0
- package/dist/maintenance/tasks/cleanup-vectors.js +23 -0
- package/dist/maintenance/tasks/cleanup-vectors.js.map +1 -0
- package/dist/maintenance/tasks/index.d.ts +9 -0
- package/dist/maintenance/tasks/index.d.ts.map +1 -0
- package/dist/maintenance/tasks/index.js +9 -0
- package/dist/maintenance/tasks/index.js.map +1 -0
- package/dist/maintenance/tasks/prune-graph.d.ts +12 -0
- package/dist/maintenance/tasks/prune-graph.d.ts.map +1 -0
- package/dist/maintenance/tasks/prune-graph.js +23 -0
- package/dist/maintenance/tasks/prune-graph.js.map +1 -0
- package/dist/maintenance/tasks/scan-projects.d.ts +12 -0
- package/dist/maintenance/tasks/scan-projects.d.ts.map +1 -0
- package/dist/maintenance/tasks/scan-projects.js +31 -0
- package/dist/maintenance/tasks/scan-projects.js.map +1 -0
- package/dist/maintenance/tasks/update-clusters.d.ts +12 -0
- package/dist/maintenance/tasks/update-clusters.d.ts.map +1 -0
- package/dist/maintenance/tasks/update-clusters.js +41 -0
- package/dist/maintenance/tasks/update-clusters.js.map +1 -0
- package/dist/maintenance/tasks/vacuum.d.ts +10 -0
- package/dist/maintenance/tasks/vacuum.d.ts.map +1 -0
- package/dist/maintenance/tasks/vacuum.js +23 -0
- package/dist/maintenance/tasks/vacuum.js.map +1 -0
- package/dist/mcp/index.d.ts +7 -0
- package/dist/mcp/index.d.ts.map +1 -0
- package/dist/mcp/index.js +8 -0
- package/dist/mcp/index.js.map +1 -0
- package/dist/mcp/server.d.ts +90 -0
- package/dist/mcp/server.d.ts.map +1 -0
- package/dist/mcp/server.js +395 -0
- package/dist/mcp/server.js.map +1 -0
- package/dist/mcp/tools.d.ts +52 -0
- package/dist/mcp/tools.d.ts.map +1 -0
- package/dist/mcp/tools.js +292 -0
- package/dist/mcp/tools.js.map +1 -0
- package/dist/models/device-detector.d.ts +32 -0
- package/dist/models/device-detector.d.ts.map +1 -0
- package/dist/models/device-detector.js +123 -0
- package/dist/models/device-detector.js.map +1 -0
- package/dist/models/embedder.d.ts +60 -0
- package/dist/models/embedder.d.ts.map +1 -0
- package/dist/models/embedder.js +171 -0
- package/dist/models/embedder.js.map +1 -0
- package/dist/models/model-registry.d.ts +25 -0
- package/dist/models/model-registry.d.ts.map +1 -0
- package/dist/models/model-registry.js +56 -0
- package/dist/models/model-registry.js.map +1 -0
- package/dist/parser/chunker.d.ts +74 -0
- package/dist/parser/chunker.d.ts.map +1 -0
- package/dist/parser/chunker.js +330 -0
- package/dist/parser/chunker.js.map +1 -0
- package/dist/parser/session-reader.d.ts +75 -0
- package/dist/parser/session-reader.d.ts.map +1 -0
- package/dist/parser/session-reader.js +198 -0
- package/dist/parser/session-reader.js.map +1 -0
- package/dist/parser/turn-assembler.d.ts +18 -0
- package/dist/parser/turn-assembler.d.ts.map +1 -0
- package/dist/parser/turn-assembler.js +150 -0
- package/dist/parser/turn-assembler.js.map +1 -0
- package/dist/parser/types.d.ts +124 -0
- package/dist/parser/types.d.ts.map +1 -0
- package/dist/parser/types.js +8 -0
- package/dist/parser/types.js.map +1 -0
- package/dist/report/reporter.d.ts +18 -0
- package/dist/report/reporter.d.ts.map +1 -0
- package/dist/report/reporter.js +121 -0
- package/dist/report/reporter.js.map +1 -0
- package/dist/retrieval/cluster-expander.d.ts +30 -0
- package/dist/retrieval/cluster-expander.d.ts.map +1 -0
- package/dist/retrieval/cluster-expander.js +87 -0
- package/dist/retrieval/cluster-expander.js.map +1 -0
- package/dist/retrieval/context-assembler.d.ts +89 -0
- package/dist/retrieval/context-assembler.d.ts.map +1 -0
- package/dist/retrieval/context-assembler.js +313 -0
- package/dist/retrieval/context-assembler.js.map +1 -0
- package/dist/retrieval/index.d.ts +12 -0
- package/dist/retrieval/index.d.ts.map +1 -0
- package/dist/retrieval/index.js +12 -0
- package/dist/retrieval/index.js.map +1 -0
- package/dist/retrieval/rrf.d.ts +25 -0
- package/dist/retrieval/rrf.d.ts.map +1 -0
- package/dist/retrieval/rrf.js +56 -0
- package/dist/retrieval/rrf.js.map +1 -0
- package/dist/retrieval/session-reconstructor.d.ts +79 -0
- package/dist/retrieval/session-reconstructor.d.ts.map +1 -0
- package/dist/retrieval/session-reconstructor.js +173 -0
- package/dist/retrieval/session-reconstructor.js.map +1 -0
- package/dist/retrieval/traverser.d.ts +66 -0
- package/dist/retrieval/traverser.d.ts.map +1 -0
- package/dist/retrieval/traverser.js +160 -0
- package/dist/retrieval/traverser.js.map +1 -0
- package/dist/storage/archive.d.ts +81 -0
- package/dist/storage/archive.d.ts.map +1 -0
- package/dist/storage/archive.js +204 -0
- package/dist/storage/archive.js.map +1 -0
- package/dist/storage/audit-log.d.ts +29 -0
- package/dist/storage/audit-log.d.ts.map +1 -0
- package/dist/storage/audit-log.js +69 -0
- package/dist/storage/audit-log.js.map +1 -0
- package/dist/storage/checkpoint-store.d.ts +38 -0
- package/dist/storage/checkpoint-store.d.ts.map +1 -0
- package/dist/storage/checkpoint-store.js +76 -0
- package/dist/storage/checkpoint-store.js.map +1 -0
- package/dist/storage/chunk-store.d.ts +111 -0
- package/dist/storage/chunk-store.d.ts.map +1 -0
- package/dist/storage/chunk-store.js +298 -0
- package/dist/storage/chunk-store.js.map +1 -0
- package/dist/storage/clock-store.d.ts +80 -0
- package/dist/storage/clock-store.d.ts.map +1 -0
- package/dist/storage/clock-store.js +155 -0
- package/dist/storage/clock-store.js.map +1 -0
- package/dist/storage/cluster-store.d.ts +63 -0
- package/dist/storage/cluster-store.d.ts.map +1 -0
- package/dist/storage/cluster-store.js +219 -0
- package/dist/storage/cluster-store.js.map +1 -0
- package/dist/storage/db.d.ts +57 -0
- package/dist/storage/db.d.ts.map +1 -0
- package/dist/storage/db.js +249 -0
- package/dist/storage/db.js.map +1 -0
- package/dist/storage/decay.d.ts +125 -0
- package/dist/storage/decay.d.ts.map +1 -0
- package/dist/storage/decay.js +276 -0
- package/dist/storage/decay.js.map +1 -0
- package/dist/storage/edge-store.d.ts +75 -0
- package/dist/storage/edge-store.d.ts.map +1 -0
- package/dist/storage/edge-store.js +259 -0
- package/dist/storage/edge-store.js.map +1 -0
- package/dist/storage/embedding-cache.d.ts +51 -0
- package/dist/storage/embedding-cache.d.ts.map +1 -0
- package/dist/storage/embedding-cache.js +155 -0
- package/dist/storage/embedding-cache.js.map +1 -0
- package/dist/storage/encryption.d.ts +55 -0
- package/dist/storage/encryption.d.ts.map +1 -0
- package/dist/storage/encryption.js +130 -0
- package/dist/storage/encryption.js.map +1 -0
- package/dist/storage/index.d.ts +15 -0
- package/dist/storage/index.d.ts.map +1 -0
- package/dist/storage/index.js +20 -0
- package/dist/storage/index.js.map +1 -0
- package/dist/storage/keyword-store.d.ts +25 -0
- package/dist/storage/keyword-store.d.ts.map +1 -0
- package/dist/storage/keyword-store.js +104 -0
- package/dist/storage/keyword-store.js.map +1 -0
- package/dist/storage/migrations.d.ts +12 -0
- package/dist/storage/migrations.d.ts.map +1 -0
- package/dist/storage/migrations.js +345 -0
- package/dist/storage/migrations.js.map +1 -0
- package/dist/storage/pruner.d.ts +111 -0
- package/dist/storage/pruner.d.ts.map +1 -0
- package/dist/storage/pruner.js +280 -0
- package/dist/storage/pruner.js.map +1 -0
- package/dist/storage/schema-loader.d.ts +18 -0
- package/dist/storage/schema-loader.d.ts.map +1 -0
- package/dist/storage/schema-loader.js +62 -0
- package/dist/storage/schema-loader.js.map +1 -0
- package/dist/storage/schema.sql +133 -0
- package/dist/storage/types.d.ts +278 -0
- package/dist/storage/types.d.ts.map +1 -0
- package/dist/storage/types.js +14 -0
- package/dist/storage/types.js.map +1 -0
- package/dist/storage/vector-store.d.ts +213 -0
- package/dist/storage/vector-store.d.ts.map +1 -0
- package/dist/storage/vector-store.js +440 -0
- package/dist/storage/vector-store.js.map +1 -0
- package/dist/temporal/clock-compactor.d.ts +65 -0
- package/dist/temporal/clock-compactor.d.ts.map +1 -0
- package/dist/temporal/clock-compactor.js +157 -0
- package/dist/temporal/clock-compactor.js.map +1 -0
- package/dist/temporal/index.d.ts +6 -0
- package/dist/temporal/index.d.ts.map +1 -0
- package/dist/temporal/index.js +6 -0
- package/dist/temporal/index.js.map +1 -0
- package/dist/temporal/vector-clock.d.ts +143 -0
- package/dist/temporal/vector-clock.d.ts.map +1 -0
- package/dist/temporal/vector-clock.js +231 -0
- package/dist/temporal/vector-clock.js.map +1 -0
- package/dist/utils/angular-distance.d.ts +31 -0
- package/dist/utils/angular-distance.d.ts.map +1 -0
- package/dist/utils/angular-distance.js +61 -0
- package/dist/utils/angular-distance.js.map +1 -0
- package/dist/utils/array-utils.d.ts +28 -0
- package/dist/utils/array-utils.d.ts.map +1 -0
- package/dist/utils/array-utils.js +67 -0
- package/dist/utils/array-utils.js.map +1 -0
- package/dist/utils/embedding-utils.d.ts +25 -0
- package/dist/utils/embedding-utils.d.ts.map +1 -0
- package/dist/utils/embedding-utils.js +30 -0
- package/dist/utils/embedding-utils.js.map +1 -0
- package/dist/utils/errors.d.ts +143 -0
- package/dist/utils/errors.d.ts.map +1 -0
- package/dist/utils/errors.js +220 -0
- package/dist/utils/errors.js.map +1 -0
- package/dist/utils/keychain.d.ts +43 -0
- package/dist/utils/keychain.d.ts.map +1 -0
- package/dist/utils/keychain.js +82 -0
- package/dist/utils/keychain.js.map +1 -0
- package/dist/utils/logger.d.ts +50 -0
- package/dist/utils/logger.d.ts.map +1 -0
- package/dist/utils/logger.js +104 -0
- package/dist/utils/logger.js.map +1 -0
- package/dist/utils/secret-store.d.ts +110 -0
- package/dist/utils/secret-store.d.ts.map +1 -0
- package/dist/utils/secret-store.js +340 -0
- package/dist/utils/secret-store.js.map +1 -0
- package/dist/utils/secure-buffer.d.ts +66 -0
- package/dist/utils/secure-buffer.d.ts.map +1 -0
- package/dist/utils/secure-buffer.js +104 -0
- package/dist/utils/secure-buffer.js.map +1 -0
- package/dist/utils/token-counter.d.ts +13 -0
- package/dist/utils/token-counter.d.ts.map +1 -0
- package/dist/utils/token-counter.js +18 -0
- package/dist/utils/token-counter.js.map +1 -0
- package/package.json +110 -0
- package/src/dashboard/client/package-lock.json +3045 -0
- package/src/dashboard/client/package.json +30 -0
|
@@ -0,0 +1,203 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Retrieval quality benchmarks.
|
|
3
|
+
*
|
|
4
|
+
* Measures whether retrieval returns the right chunks using
|
|
5
|
+
* self-supervised ground truth from structural relationships.
|
|
6
|
+
*/
|
|
7
|
+
import { getChunkById } from '../../storage/chunk-store.js';
|
|
8
|
+
import { assembleContext } from '../../retrieval/context-assembler.js';
|
|
9
|
+
import { approximateTokens } from '../../utils/token-counter.js';
|
|
10
|
+
/**
|
|
11
|
+
* Run retrieval quality benchmarks.
|
|
12
|
+
*/
|
|
13
|
+
export async function runRetrievalBenchmarks(sample, topK, onProgress) {
|
|
14
|
+
const skipped = [];
|
|
15
|
+
// Adjacent recall
|
|
16
|
+
let adjacentRecallAt5 = 0;
|
|
17
|
+
let adjacentRecallAt10 = 0;
|
|
18
|
+
let mrrSum = 0;
|
|
19
|
+
let adjacentCount = 0;
|
|
20
|
+
if (sample.thresholds.canRunAdjacentRecall && sample.adjacentPairs.length > 0) {
|
|
21
|
+
let processed = 0;
|
|
22
|
+
for (const pair of sample.adjacentPairs) {
|
|
23
|
+
const queryChunk = getChunkById(pair.queryChunkId);
|
|
24
|
+
if (!queryChunk)
|
|
25
|
+
continue;
|
|
26
|
+
onProgress?.(`[${++processed}/${sample.adjacentPairs.length}] Adjacent chunk recall...`);
|
|
27
|
+
const response = await assembleContext({
|
|
28
|
+
query: queryChunk.content.slice(0, 500),
|
|
29
|
+
mode: 'recall',
|
|
30
|
+
projectFilter: queryChunk.sessionSlug,
|
|
31
|
+
maxTokens: 10000,
|
|
32
|
+
vectorSearchLimit: topK * 2,
|
|
33
|
+
});
|
|
34
|
+
const resultIds = response.chunks.map(c => c.id);
|
|
35
|
+
const foundInTop5 = resultIds.slice(0, 5).includes(pair.adjacentChunkId);
|
|
36
|
+
const foundInTop10 = resultIds.slice(0, topK).includes(pair.adjacentChunkId);
|
|
37
|
+
if (foundInTop5)
|
|
38
|
+
adjacentRecallAt5++;
|
|
39
|
+
if (foundInTop10)
|
|
40
|
+
adjacentRecallAt10++;
|
|
41
|
+
// MRR
|
|
42
|
+
const rank = resultIds.indexOf(pair.adjacentChunkId);
|
|
43
|
+
if (rank >= 0)
|
|
44
|
+
mrrSum += 1 / (rank + 1);
|
|
45
|
+
adjacentCount++;
|
|
46
|
+
}
|
|
47
|
+
}
|
|
48
|
+
else {
|
|
49
|
+
const reason = sample.thresholds.reasons.get('adjacentRecall') ?? 'insufficient data';
|
|
50
|
+
skipped.push({
|
|
51
|
+
name: 'Adjacent Chunk Recall',
|
|
52
|
+
reason: `Skipped: ${reason}`,
|
|
53
|
+
threshold: '>=2 sessions with >=3 chunks each',
|
|
54
|
+
current: reason,
|
|
55
|
+
});
|
|
56
|
+
}
|
|
57
|
+
// Cross-session bridging
|
|
58
|
+
let bridgingRecallAt10 = 0;
|
|
59
|
+
let bridgingCount = 0;
|
|
60
|
+
let randomRecallAt10 = 0;
|
|
61
|
+
let randomCount = 0;
|
|
62
|
+
if (sample.thresholds.canRunCrossSessionBridging && sample.crossSessionPairs.length > 0) {
|
|
63
|
+
let processed = 0;
|
|
64
|
+
for (const pair of sample.crossSessionPairs) {
|
|
65
|
+
const queryChunk = getChunkById(pair.chunkIdA);
|
|
66
|
+
if (!queryChunk)
|
|
67
|
+
continue;
|
|
68
|
+
onProgress?.(`[${++processed}/${sample.crossSessionPairs.length}] Cross-session bridging...`);
|
|
69
|
+
const response = await assembleContext({
|
|
70
|
+
query: queryChunk.content.slice(0, 500),
|
|
71
|
+
mode: 'recall',
|
|
72
|
+
projectFilter: queryChunk.sessionSlug,
|
|
73
|
+
maxTokens: 10000,
|
|
74
|
+
vectorSearchLimit: topK * 2,
|
|
75
|
+
});
|
|
76
|
+
const resultIds = response.chunks.map(c => c.id);
|
|
77
|
+
if (resultIds.slice(0, topK).includes(pair.chunkIdB)) {
|
|
78
|
+
bridgingRecallAt10++;
|
|
79
|
+
}
|
|
80
|
+
bridgingCount++;
|
|
81
|
+
}
|
|
82
|
+
// Random baseline: use cross-project pairs as negative control
|
|
83
|
+
if (sample.crossProjectPairs.length > 0) {
|
|
84
|
+
const randomSample = sample.crossProjectPairs.slice(0, Math.min(20, sample.crossProjectPairs.length));
|
|
85
|
+
for (const pair of randomSample) {
|
|
86
|
+
const queryChunk = getChunkById(pair.chunkIdA);
|
|
87
|
+
if (!queryChunk)
|
|
88
|
+
continue;
|
|
89
|
+
const response = await assembleContext({
|
|
90
|
+
query: queryChunk.content.slice(0, 500),
|
|
91
|
+
mode: 'recall',
|
|
92
|
+
projectFilter: queryChunk.sessionSlug,
|
|
93
|
+
maxTokens: 10000,
|
|
94
|
+
vectorSearchLimit: topK * 2,
|
|
95
|
+
});
|
|
96
|
+
const resultIds = response.chunks.map(c => c.id);
|
|
97
|
+
if (resultIds.slice(0, topK).includes(pair.chunkIdB)) {
|
|
98
|
+
randomRecallAt10++;
|
|
99
|
+
}
|
|
100
|
+
randomCount++;
|
|
101
|
+
}
|
|
102
|
+
}
|
|
103
|
+
}
|
|
104
|
+
else {
|
|
105
|
+
const reason = sample.thresholds.reasons.get('crossSessionBridging') ?? 'insufficient data';
|
|
106
|
+
skipped.push({
|
|
107
|
+
name: 'Cross-Session Bridging',
|
|
108
|
+
reason: `Skipped: ${reason}`,
|
|
109
|
+
threshold: '>=3 sessions in same project with cross-session edges',
|
|
110
|
+
current: reason,
|
|
111
|
+
});
|
|
112
|
+
}
|
|
113
|
+
// Precision@K
|
|
114
|
+
let precisionAt5 = 0;
|
|
115
|
+
let precisionAt10 = 0;
|
|
116
|
+
let precisionCount = 0;
|
|
117
|
+
if (sample.thresholds.canRunPrecisionAtK && sample.crossProjectPairs.length > 0) {
|
|
118
|
+
// Use chunk IDs from the first project in cross-project pairs
|
|
119
|
+
const projectASamples = new Map();
|
|
120
|
+
for (const pair of sample.crossProjectPairs) {
|
|
121
|
+
const list = projectASamples.get(pair.projectA) ?? [];
|
|
122
|
+
list.push(pair.chunkIdA);
|
|
123
|
+
projectASamples.set(pair.projectA, list);
|
|
124
|
+
}
|
|
125
|
+
let processed = 0;
|
|
126
|
+
for (const [project, chunkIds] of projectASamples) {
|
|
127
|
+
const uniqueIds = [...new Set(chunkIds)].slice(0, 10);
|
|
128
|
+
for (const chunkId of uniqueIds) {
|
|
129
|
+
const queryChunk = getChunkById(chunkId);
|
|
130
|
+
if (!queryChunk)
|
|
131
|
+
continue;
|
|
132
|
+
onProgress?.(`[${++processed}/${Math.min(20, [...projectASamples.values()].flat().length)}] Precision@K...`);
|
|
133
|
+
const response = await assembleContext({
|
|
134
|
+
query: queryChunk.content.slice(0, 500),
|
|
135
|
+
mode: 'recall',
|
|
136
|
+
projectFilter: project,
|
|
137
|
+
maxTokens: 10000,
|
|
138
|
+
vectorSearchLimit: topK * 2,
|
|
139
|
+
});
|
|
140
|
+
const top5FromProject = response.chunks.slice(0, 5).filter(c => c.sessionSlug === project).length;
|
|
141
|
+
const top10FromProject = response.chunks.slice(0, topK).filter(c => c.sessionSlug === project).length;
|
|
142
|
+
precisionAt5 += top5FromProject / Math.min(5, response.chunks.length || 1);
|
|
143
|
+
precisionAt10 += top10FromProject / Math.min(topK, response.chunks.length || 1);
|
|
144
|
+
precisionCount++;
|
|
145
|
+
}
|
|
146
|
+
}
|
|
147
|
+
}
|
|
148
|
+
else {
|
|
149
|
+
const reason = sample.thresholds.reasons.get('precisionAtK') ?? 'insufficient data';
|
|
150
|
+
skipped.push({
|
|
151
|
+
name: 'Precision@K',
|
|
152
|
+
reason: `Skipped: ${reason}`,
|
|
153
|
+
threshold: '>=2 projects with >=10 chunks each',
|
|
154
|
+
current: reason,
|
|
155
|
+
});
|
|
156
|
+
}
|
|
157
|
+
// Token efficiency
|
|
158
|
+
let tokenEfficiencySum = 0;
|
|
159
|
+
let usefulTokenSum = 0;
|
|
160
|
+
let tokenEfficiencyCount = 0;
|
|
161
|
+
// Reuse adjacent recall results for token efficiency
|
|
162
|
+
if (adjacentCount > 0) {
|
|
163
|
+
for (const pair of sample.adjacentPairs.slice(0, 20)) {
|
|
164
|
+
const queryChunk = getChunkById(pair.queryChunkId);
|
|
165
|
+
if (!queryChunk)
|
|
166
|
+
continue;
|
|
167
|
+
const response = await assembleContext({
|
|
168
|
+
query: queryChunk.content.slice(0, 500),
|
|
169
|
+
mode: 'recall',
|
|
170
|
+
projectFilter: queryChunk.sessionSlug,
|
|
171
|
+
maxTokens: 10000,
|
|
172
|
+
vectorSearchLimit: topK * 2,
|
|
173
|
+
});
|
|
174
|
+
if (response.chunks.length === 0)
|
|
175
|
+
continue;
|
|
176
|
+
// Relevant = same session or same project
|
|
177
|
+
const relevantChunks = response.chunks.filter(c => c.sessionSlug === queryChunk.sessionSlug);
|
|
178
|
+
const totalTokens = response.tokenCount || 1;
|
|
179
|
+
const relevantTokens = relevantChunks.reduce((sum, c) => {
|
|
180
|
+
const chunk = getChunkById(c.id);
|
|
181
|
+
return sum + (chunk?.approxTokens ?? approximateTokens(c.preview));
|
|
182
|
+
}, 0);
|
|
183
|
+
tokenEfficiencySum += relevantTokens / totalTokens;
|
|
184
|
+
usefulTokenSum += relevantTokens;
|
|
185
|
+
tokenEfficiencyCount++;
|
|
186
|
+
}
|
|
187
|
+
}
|
|
188
|
+
const result = {
|
|
189
|
+
adjacentRecallAt5: adjacentCount > 0 ? adjacentRecallAt5 / adjacentCount : 0,
|
|
190
|
+
adjacentRecallAt10: adjacentCount > 0 ? adjacentRecallAt10 / adjacentCount : 0,
|
|
191
|
+
mrr: adjacentCount > 0 ? mrrSum / adjacentCount : 0,
|
|
192
|
+
bridgingRecallAt10: bridgingCount > 0 ? bridgingRecallAt10 / bridgingCount : 0,
|
|
193
|
+
bridgingVsRandom: randomCount > 0 && randomRecallAt10 > 0
|
|
194
|
+
? (bridgingRecallAt10 / bridgingCount) / (randomRecallAt10 / randomCount)
|
|
195
|
+
: bridgingCount > 0 ? bridgingRecallAt10 / bridgingCount : 0,
|
|
196
|
+
precisionAt5: precisionCount > 0 ? precisionAt5 / precisionCount : 0,
|
|
197
|
+
precisionAt10: precisionCount > 0 ? precisionAt10 / precisionCount : 0,
|
|
198
|
+
tokenEfficiency: tokenEfficiencyCount > 0 ? tokenEfficiencySum / tokenEfficiencyCount : 0,
|
|
199
|
+
meanUsefulTokensPerQuery: tokenEfficiencyCount > 0 ? usefulTokenSum / tokenEfficiencyCount : 0,
|
|
200
|
+
};
|
|
201
|
+
return { result, skipped };
|
|
202
|
+
}
|
|
203
|
+
//# sourceMappingURL=retrieval.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"retrieval.js","sourceRoot":"","sources":["../../../src/eval/collection-benchmark/retrieval.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,YAAY,EAAE,MAAM,8BAA8B,CAAC;AAC5D,OAAO,EAAE,eAAe,EAAyB,MAAM,sCAAsC,CAAC;AAC9F,OAAO,EAAE,iBAAiB,EAAE,MAAM,8BAA8B,CAAC;AAOjE;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,sBAAsB,CAC1C,MAAuB,EACvB,IAAY,EACZ,UAAkC;IAElC,MAAM,OAAO,GAAuB,EAAE,CAAC;IAEvC,kBAAkB;IAClB,IAAI,iBAAiB,GAAG,CAAC,CAAC;IAC1B,IAAI,kBAAkB,GAAG,CAAC,CAAC;IAC3B,IAAI,MAAM,GAAG,CAAC,CAAC;IACf,IAAI,aAAa,GAAG,CAAC,CAAC;IAEtB,IAAI,MAAM,CAAC,UAAU,CAAC,oBAAoB,IAAI,MAAM,CAAC,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC9E,IAAI,SAAS,GAAG,CAAC,CAAC;QAClB,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,aAAa,EAAE,CAAC;YACxC,MAAM,UAAU,GAAG,YAAY,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YACnD,IAAI,CAAC,UAAU;gBAAE,SAAS;YAE1B,UAAU,EAAE,CAAC,IAAI,EAAE,SAAS,IAAI,MAAM,CAAC,aAAa,CAAC,MAAM,4BAA4B,CAAC,CAAC;YAEzF,MAAM,QAAQ,GAAG,MAAM,eAAe,CAAC;gBACrC,KAAK,EAAE,UAAU,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC;gBACvC,IAAI,EAAE,QAAQ;gBACd,aAAa,EAAE,UAAU,CAAC,WAAW;gBACrC,SAAS,EAAE,KAAK;gBAChB,iBAAiB,EAAE,IAAI,GAAG,CAAC;aAC5B,CAAC,CAAC;YAEH,MAAM,SAAS,GAAG,QAAQ,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;YACjD,MAAM,WAAW,GAAG,SAAS,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;YACzE,MAAM,YAAY,GAAG,SAAS,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;YAE7E,IAAI,WAAW;gBAAE,iBAAiB,EAAE,CAAC;YACrC,IAAI,YAAY;gBAAE,kBAAkB,EAAE,CAAC;YAEvC,MAAM;YACN,MAAM,IAAI,GAAG,SAAS,CAAC,OAAO,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;YACrD,IAAI,IAAI,IAAI,CAAC;gBAAE,MAAM,IAAI,CAAC,GAAG,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC;YAExC,aAAa,EAAE,CAAC;QAClB,CAAC;IACH,CAAC;SAAM,CAAC;QACN,MAAM,MAAM,GAAG,MAAM,CAAC,UAAU,CAAC,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC,IAAI,mBAAmB,CAAC;QACtF,OAAO,CAAC,IAAI,CAAC;YACX,IAAI,EAAE,uBAAuB;YAC7B,MAAM,EAAE,YAAY,MAAM,EAAE;YAC5B,SAAS,EAAE,mCAAmC;YAC9C,OAAO,EAAE,MAAM;SAChB,CAAC,CAAC;IACL,CAAC;IAED,yBAAyB;IACzB,IAAI,kBAAkB,GAAG,CAAC,CAAC;IAC3B,IAAI,aAAa,GAAG,CAAC,CAAC;IACtB,IAAI,gBAAgB,GAAG,CAAC,CAAC;IACzB,IAAI,WAAW,GAAG,CAAC,CAAC;IAEpB,IAAI,MAAM,CAAC,UAAU,CAAC,0BAA0B,IAAI,MAAM,CAAC,iBAAiB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACxF,IAAI,SAAS,GAAG,CAAC,CAAC;QAClB,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,iBAAiB,EAAE,CAAC;YAC5C,MAAM,UAAU,GAAG,YAAY,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAC/C,IAAI,CAAC,UAAU;gBAAE,SAAS;YAE1B,UAAU,EAAE,CAAC,IAAI,EAAE,SAAS,IAAI,MAAM,CAAC,iBAAiB,CAAC,MAAM,6BAA6B,CAAC,CAAC;YAE9F,MAAM,QAAQ,GAAG,MAAM,eAAe,CAAC;gBACrC,KAAK,EAAE,UAAU,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC;gBACvC,IAAI,EAAE,QAAQ;gBACd,aAAa,EAAE,UAAU,CAAC,WAAW;gBACrC,SAAS,EAAE,KAAK;gBAChB,iBAAiB,EAAE,IAAI,GAAG,CAAC;aAC5B,CAAC,CAAC;YAEH,MAAM,SAAS,GAAG,QAAQ,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;YACjD,IAAI,SAAS,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC;gBACrD,kBAAkB,EAAE,CAAC;YACvB,CAAC;YACD,aAAa,EAAE,CAAC;QAClB,CAAC;QAED,+DAA+D;QAC/D,IAAI,MAAM,CAAC,iBAAiB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACxC,MAAM,YAAY,GAAG,MAAM,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,MAAM,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC,CAAC;YACtG,KAAK,MAAM,IAAI,IAAI,YAAY,EAAE,CAAC;gBAChC,MAAM,UAAU,GAAG,YAAY,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;gBAC/C,IAAI,CAAC,UAAU;oBAAE,SAAS;gBAE1B,MAAM,QAAQ,GAAG,MAAM,eAAe,CAAC;oBACrC,KAAK,EAAE,UAAU,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC;oBACvC,IAAI,EAAE,QAAQ;oBACd,aAAa,EAAE,UAAU,CAAC,WAAW;oBACrC,SAAS,EAAE,KAAK;oBAChB,iBAAiB,EAAE,IAAI,GAAG,CAAC;iBAC5B,CAAC,CAAC;gBAEH,MAAM,SAAS,GAAG,QAAQ,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;gBACjD,IAAI,SAAS,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC;oBACrD,gBAAgB,EAAE,CAAC;gBACrB,CAAC;gBACD,WAAW,EAAE,CAAC;YAChB,CAAC;QACH,CAAC;IACH,CAAC;SAAM,CAAC;QACN,MAAM,MAAM,GAAG,MAAM,CAAC,UAAU,CAAC,OAAO,CAAC,GAAG,CAAC,sBAAsB,CAAC,IAAI,mBAAmB,CAAC;QAC5F,OAAO,CAAC,IAAI,CAAC;YACX,IAAI,EAAE,wBAAwB;YAC9B,MAAM,EAAE,YAAY,MAAM,EAAE;YAC5B,SAAS,EAAE,uDAAuD;YAClE,OAAO,EAAE,MAAM;SAChB,CAAC,CAAC;IACL,CAAC;IAED,cAAc;IACd,IAAI,YAAY,GAAG,CAAC,CAAC;IACrB,IAAI,aAAa,GAAG,CAAC,CAAC;IACtB,IAAI,cAAc,GAAG,CAAC,CAAC;IAEvB,IAAI,MAAM,CAAC,UAAU,CAAC,kBAAkB,IAAI,MAAM,CAAC,iBAAiB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAChF,8DAA8D;QAC9D,MAAM,eAAe,GAAG,IAAI,GAAG,EAAoB,CAAC;QACpD,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,iBAAiB,EAAE,CAAC;YAC5C,MAAM,IAAI,GAAG,eAAe,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;YACtD,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACzB,eAAe,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QAC3C,CAAC;QAED,IAAI,SAAS,GAAG,CAAC,CAAC;QAClB,KAAK,MAAM,CAAC,OAAO,EAAE,QAAQ,CAAC,IAAI,eAAe,EAAE,CAAC;YAClD,MAAM,SAAS,GAAG,CAAC,GAAG,IAAI,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACtD,KAAK,MAAM,OAAO,IAAI,SAAS,EAAE,CAAC;gBAChC,MAAM,UAAU,GAAG,YAAY,CAAC,OAAO,CAAC,CAAC;gBACzC,IAAI,CAAC,UAAU;oBAAE,SAAS;gBAE1B,UAAU,EAAE,CAAC,IAAI,EAAE,SAAS,IAAI,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC,GAAG,eAAe,CAAC,MAAM,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC,MAAM,CAAC,kBAAkB,CAAC,CAAC;gBAE7G,MAAM,QAAQ,GAAG,MAAM,eAAe,CAAC;oBACrC,KAAK,EAAE,UAAU,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC;oBACvC,IAAI,EAAE,QAAQ;oBACd,aAAa,EAAE,OAAO;oBACtB,SAAS,EAAE,KAAK;oBAChB,iBAAiB,EAAE,IAAI,GAAG,CAAC;iBAC5B,CAAC,CAAC;gBAEH,MAAM,eAAe,GAAG,QAAQ,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,WAAW,KAAK,OAAO,CAAC,CAAC,MAAM,CAAC;gBAClG,MAAM,gBAAgB,GAAG,QAAQ,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,WAAW,KAAK,OAAO,CAAC,CAAC,MAAM,CAAC;gBAEtG,YAAY,IAAI,eAAe,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,QAAQ,CAAC,MAAM,CAAC,MAAM,IAAI,CAAC,CAAC,CAAC;gBAC3E,aAAa,IAAI,gBAAgB,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,QAAQ,CAAC,MAAM,CAAC,MAAM,IAAI,CAAC,CAAC,CAAC;gBAChF,cAAc,EAAE,CAAC;YACnB,CAAC;QACH,CAAC;IACH,CAAC;SAAM,CAAC;QACN,MAAM,MAAM,GAAG,MAAM,CAAC,UAAU,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,IAAI,mBAAmB,CAAC;QACpF,OAAO,CAAC,IAAI,CAAC;YACX,IAAI,EAAE,aAAa;YACnB,MAAM,EAAE,YAAY,MAAM,EAAE;YAC5B,SAAS,EAAE,oCAAoC;YAC/C,OAAO,EAAE,MAAM;SAChB,CAAC,CAAC;IACL,CAAC;IAED,mBAAmB;IACnB,IAAI,kBAAkB,GAAG,CAAC,CAAC;IAC3B,IAAI,cAAc,GAAG,CAAC,CAAC;IACvB,IAAI,oBAAoB,GAAG,CAAC,CAAC;IAE7B,qDAAqD;IACrD,IAAI,aAAa,GAAG,CAAC,EAAE,CAAC;QACtB,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC;YACrD,MAAM,UAAU,GAAG,YAAY,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YACnD,IAAI,CAAC,UAAU;gBAAE,SAAS;YAE1B,MAAM,QAAQ,GAAG,MAAM,eAAe,CAAC;gBACrC,KAAK,EAAE,UAAU,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC;gBACvC,IAAI,EAAE,QAAQ;gBACd,aAAa,EAAE,UAAU,CAAC,WAAW;gBACrC,SAAS,EAAE,KAAK;gBAChB,iBAAiB,EAAE,IAAI,GAAG,CAAC;aAC5B,CAAC,CAAC;YAEH,IAAI,QAAQ,CAAC,MAAM,CAAC,MAAM,KAAK,CAAC;gBAAE,SAAS;YAE3C,0CAA0C;YAC1C,MAAM,cAAc,GAAG,QAAQ,CAAC,MAAM,CAAC,MAAM,CAC3C,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,WAAW,KAAK,UAAU,CAAC,WAAW,CAC9C,CAAC;YAEF,MAAM,WAAW,GAAG,QAAQ,CAAC,UAAU,IAAI,CAAC,CAAC;YAC7C,MAAM,cAAc,GAAG,cAAc,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE;gBACtD,MAAM,KAAK,GAAG,YAAY,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;gBACjC,OAAO,GAAG,GAAG,CAAC,KAAK,EAAE,YAAY,IAAI,iBAAiB,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;YACrE,CAAC,EAAE,CAAC,CAAC,CAAC;YAEN,kBAAkB,IAAI,cAAc,GAAG,WAAW,CAAC;YACnD,cAAc,IAAI,cAAc,CAAC;YACjC,oBAAoB,EAAE,CAAC;QACzB,CAAC;IACH,CAAC;IAED,MAAM,MAAM,GAAoB;QAC9B,iBAAiB,EAAE,aAAa,GAAG,CAAC,CAAC,CAAC,CAAC,iBAAiB,GAAG,aAAa,CAAC,CAAC,CAAC,CAAC;QAC5E,kBAAkB,EAAE,aAAa,GAAG,CAAC,CAAC,CAAC,CAAC,kBAAkB,GAAG,aAAa,CAAC,CAAC,CAAC,CAAC;QAC9E,GAAG,EAAE,aAAa,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,GAAG,aAAa,CAAC,CAAC,CAAC,CAAC;QACnD,kBAAkB,EAAE,aAAa,GAAG,CAAC,CAAC,CAAC,CAAC,kBAAkB,GAAG,aAAa,CAAC,CAAC,CAAC,CAAC;QAC9E,gBAAgB,EAAE,WAAW,GAAG,CAAC,IAAI,gBAAgB,GAAG,CAAC;YACvD,CAAC,CAAC,CAAC,kBAAkB,GAAG,aAAa,CAAC,GAAG,CAAC,gBAAgB,GAAG,WAAW,CAAC;YACzE,CAAC,CAAC,aAAa,GAAG,CAAC,CAAC,CAAC,CAAC,kBAAkB,GAAG,aAAa,CAAC,CAAC,CAAC,CAAC;QAC9D,YAAY,EAAE,cAAc,GAAG,CAAC,CAAC,CAAC,CAAC,YAAY,GAAG,cAAc,CAAC,CAAC,CAAC,CAAC;QACpE,aAAa,EAAE,cAAc,GAAG,CAAC,CAAC,CAAC,CAAC,aAAa,GAAG,cAAc,CAAC,CAAC,CAAC,CAAC;QACtE,eAAe,EAAE,oBAAoB,GAAG,CAAC,CAAC,CAAC,CAAC,kBAAkB,GAAG,oBAAoB,CAAC,CAAC,CAAC,CAAC;QACzF,wBAAwB,EAAE,oBAAoB,GAAG,CAAC,CAAC,CAAC,CAAC,cAAc,GAAG,oBAAoB,CAAC,CAAC,CAAC,CAAC;KAC/F,CAAC;IAEF,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC;AAC7B,CAAC"}
|
|
@@ -0,0 +1,20 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Benchmark orchestrator.
|
|
3
|
+
*
|
|
4
|
+
* Runs all benchmark categories based on profile, generates samples,
|
|
5
|
+
* computes composite scores, and assembles the final result.
|
|
6
|
+
*/
|
|
7
|
+
import type { CollectionBenchmarkOptions, CollectionBenchmarkResult, HealthResult, RetrievalResult, GraphValueResult, LatencyResult } from './types.js';
|
|
8
|
+
/**
|
|
9
|
+
* Compute overall score (0-100) from sub-scores.
|
|
10
|
+
*/
|
|
11
|
+
export declare function computeOverallScore(health: HealthResult, retrieval?: RetrievalResult, graph?: GraphValueResult, latency?: LatencyResult): number;
|
|
12
|
+
/**
|
|
13
|
+
* Generate human-readable highlights from results.
|
|
14
|
+
*/
|
|
15
|
+
export declare function generateHighlights(health: HealthResult, retrieval?: RetrievalResult, graph?: GraphValueResult, latency?: LatencyResult): string[];
|
|
16
|
+
/**
|
|
17
|
+
* Run the collection benchmark suite.
|
|
18
|
+
*/
|
|
19
|
+
export declare function runCollectionBenchmark(options?: CollectionBenchmarkOptions): Promise<CollectionBenchmarkResult>;
|
|
20
|
+
//# sourceMappingURL=runner.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"runner.d.ts","sourceRoot":"","sources":["../../../src/eval/collection-benchmark/runner.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAWH,OAAO,KAAK,EAGV,0BAA0B,EAC1B,yBAAyB,EAEzB,YAAY,EACZ,eAAe,EACf,gBAAgB,EAChB,aAAa,EAGd,MAAM,YAAY,CAAC;AAkBpB;;GAEG;AACH,wBAAgB,mBAAmB,CACjC,MAAM,EAAE,YAAY,EACpB,SAAS,CAAC,EAAE,eAAe,EAC3B,KAAK,CAAC,EAAE,gBAAgB,EACxB,OAAO,CAAC,EAAE,aAAa,GACtB,MAAM,CA+CR;AAED;;GAEG;AACH,wBAAgB,kBAAkB,CAChC,MAAM,EAAE,YAAY,EACpB,SAAS,CAAC,EAAE,eAAe,EAC3B,KAAK,CAAC,EAAE,gBAAgB,EACxB,OAAO,CAAC,EAAE,aAAa,GACtB,MAAM,EAAE,CA+CV;AAED;;GAEG;AACH,wBAAsB,sBAAsB,CAC1C,OAAO,GAAE,0BAA+B,GACvC,OAAO,CAAC,yBAAyB,CAAC,CAiGpC"}
|
|
@@ -0,0 +1,189 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Benchmark orchestrator.
|
|
3
|
+
*
|
|
4
|
+
* Runs all benchmark categories based on profile, generates samples,
|
|
5
|
+
* computes composite scores, and assembles the final result.
|
|
6
|
+
*/
|
|
7
|
+
import { generateSamples } from './sampler.js';
|
|
8
|
+
import { runHealthBenchmarks } from './health.js';
|
|
9
|
+
import { runRetrievalBenchmarks } from './retrieval.js';
|
|
10
|
+
import { runGraphValueBenchmarks } from './graph-value.js';
|
|
11
|
+
import { runLatencyBenchmarks } from './latency.js';
|
|
12
|
+
import { generateTuningRecommendations } from './tuning.js';
|
|
13
|
+
import { storeBenchmarkRun, getLatestBenchmarkRun, computeTrend } from './history.js';
|
|
14
|
+
import { loadConfig } from '../../config/loader.js';
|
|
15
|
+
import { getDistinctProjects } from '../../storage/chunk-store.js';
|
|
16
|
+
/**
|
|
17
|
+
* Resolve profile to category list.
|
|
18
|
+
*/
|
|
19
|
+
function resolveCategories(profile, explicit) {
|
|
20
|
+
if (explicit && explicit.length > 0)
|
|
21
|
+
return explicit;
|
|
22
|
+
switch (profile) {
|
|
23
|
+
case 'quick': return ['health'];
|
|
24
|
+
case 'standard': return ['health', 'retrieval'];
|
|
25
|
+
case 'full': return ['health', 'retrieval', 'graph', 'latency'];
|
|
26
|
+
}
|
|
27
|
+
}
|
|
28
|
+
/**
|
|
29
|
+
* Compute overall score (0-100) from sub-scores.
|
|
30
|
+
*/
|
|
31
|
+
export function computeOverallScore(health, retrieval, graph, latency) {
|
|
32
|
+
const scores = [];
|
|
33
|
+
// Health score (0-100)
|
|
34
|
+
const healthScore = Math.min(100, (Math.min(1, health.edgeToChunkRatio / 3) * 30 +
|
|
35
|
+
health.clusterCoverage * 40 +
|
|
36
|
+
(1 - health.orphanChunkPercentage) * 30));
|
|
37
|
+
scores.push({ score: healthScore, weight: 20 });
|
|
38
|
+
// Retrieval score (0-100)
|
|
39
|
+
if (retrieval) {
|
|
40
|
+
const retrievalScore = Math.min(100, retrieval.adjacentRecallAt10 * 30 +
|
|
41
|
+
retrieval.bridgingRecallAt10 * 20 +
|
|
42
|
+
retrieval.precisionAt10 * 25 +
|
|
43
|
+
retrieval.tokenEfficiency * 25);
|
|
44
|
+
scores.push({ score: retrievalScore, weight: 35 });
|
|
45
|
+
}
|
|
46
|
+
// Graph value score (0-100)
|
|
47
|
+
if (graph) {
|
|
48
|
+
const augScore = Math.min(1, (graph.sourceAttribution.augmentationRatio - 1) / 2);
|
|
49
|
+
const liftScore = Math.min(1, graph.lift);
|
|
50
|
+
const graphScore = Math.min(100, augScore * 50 + liftScore * 50);
|
|
51
|
+
scores.push({ score: graphScore, weight: 30 });
|
|
52
|
+
}
|
|
53
|
+
// Latency score (0-100, based on p95 thresholds)
|
|
54
|
+
if (latency) {
|
|
55
|
+
const recallLatencyScore = latency.recall.p95 <= 50 ? 100
|
|
56
|
+
: latency.recall.p95 <= 100 ? 80
|
|
57
|
+
: latency.recall.p95 <= 200 ? 60
|
|
58
|
+
: latency.recall.p95 <= 500 ? 40
|
|
59
|
+
: 20;
|
|
60
|
+
scores.push({ score: recallLatencyScore, weight: 15 });
|
|
61
|
+
}
|
|
62
|
+
// Weighted average with renormalization
|
|
63
|
+
const totalWeight = scores.reduce((sum, s) => sum + s.weight, 0);
|
|
64
|
+
if (totalWeight === 0)
|
|
65
|
+
return 0;
|
|
66
|
+
return Math.round(scores.reduce((sum, s) => sum + s.score * s.weight, 0) / totalWeight);
|
|
67
|
+
}
|
|
68
|
+
/**
|
|
69
|
+
* Generate human-readable highlights from results.
|
|
70
|
+
*/
|
|
71
|
+
export function generateHighlights(health, retrieval, graph, latency) {
|
|
72
|
+
const highlights = [];
|
|
73
|
+
// Health highlights
|
|
74
|
+
highlights.push(`${(health.clusterCoverage * 100).toFixed(0)}% of chunks organized into topic clusters`);
|
|
75
|
+
if (health.orphanChunkPercentage < 0.05) {
|
|
76
|
+
highlights.push(`Only ${(health.orphanChunkPercentage * 100).toFixed(1)}% orphan chunks — knowledge graph is well-connected`);
|
|
77
|
+
}
|
|
78
|
+
if (health.temporalSpan) {
|
|
79
|
+
const days = Math.round((new Date(health.temporalSpan.latest).getTime() - new Date(health.temporalSpan.earliest).getTime()) /
|
|
80
|
+
(1000 * 60 * 60 * 24));
|
|
81
|
+
highlights.push(`Collection spans ${days} days across ${health.projectCount} project(s)`);
|
|
82
|
+
}
|
|
83
|
+
// Retrieval highlights
|
|
84
|
+
if (retrieval) {
|
|
85
|
+
if (retrieval.adjacentRecallAt10 > 0) {
|
|
86
|
+
highlights.push(`Adjacent chunk recall@10: ${(retrieval.adjacentRecallAt10 * 100).toFixed(0)}%`);
|
|
87
|
+
}
|
|
88
|
+
if (retrieval.bridgingRecallAt10 > 0) {
|
|
89
|
+
highlights.push(`Cross-session bridging recall: ${(retrieval.bridgingRecallAt10 * 100).toFixed(0)}%`);
|
|
90
|
+
}
|
|
91
|
+
if (retrieval.tokenEfficiency > 0) {
|
|
92
|
+
highlights.push(`Token efficiency: ${(retrieval.tokenEfficiency * 100).toFixed(0)}% of returned context is relevant`);
|
|
93
|
+
}
|
|
94
|
+
}
|
|
95
|
+
// Graph highlights
|
|
96
|
+
if (graph) {
|
|
97
|
+
if (graph.sourceAttribution.augmentationRatio > 1.1) {
|
|
98
|
+
highlights.push(`Graph traversal adds ${graph.sourceAttribution.augmentationRatio.toFixed(1)}x more results vs vector-only`);
|
|
99
|
+
}
|
|
100
|
+
if (graph.lift > 0.1) {
|
|
101
|
+
highlights.push(`Recall lift from graph: +${(graph.lift * 100).toFixed(0)}% over vector-only`);
|
|
102
|
+
}
|
|
103
|
+
}
|
|
104
|
+
// Latency highlights
|
|
105
|
+
if (latency) {
|
|
106
|
+
highlights.push(`p95 recall latency: ${latency.recall.p95.toFixed(0)}ms`);
|
|
107
|
+
}
|
|
108
|
+
return highlights;
|
|
109
|
+
}
|
|
110
|
+
/**
|
|
111
|
+
* Run the collection benchmark suite.
|
|
112
|
+
*/
|
|
113
|
+
export async function runCollectionBenchmark(options = {}) {
|
|
114
|
+
const { profile = 'standard', sampleSize = 50, seed, projectFilter, topK = 10, includeTuning = true, onProgress, } = options;
|
|
115
|
+
const categories = resolveCategories(profile, options.categories);
|
|
116
|
+
const skipped = [];
|
|
117
|
+
let retrieval;
|
|
118
|
+
let graphValue;
|
|
119
|
+
let latency;
|
|
120
|
+
// 1. Always run health
|
|
121
|
+
onProgress?.('[1/4] Collection health...');
|
|
122
|
+
const includeClusterQuality = categories.includes('retrieval') || categories.includes('graph');
|
|
123
|
+
const health = await runHealthBenchmarks(includeClusterQuality);
|
|
124
|
+
onProgress?.('[1/4] Collection health... done');
|
|
125
|
+
// 2. Generate samples if needed
|
|
126
|
+
let sample;
|
|
127
|
+
if (categories.some(c => c !== 'health')) {
|
|
128
|
+
onProgress?.(`[2/4] Sampling queries (${sampleSize} samples${seed !== null && seed !== undefined ? `, seed=${seed}` : ''})...`);
|
|
129
|
+
sample = generateSamples({ sampleSize, seed, projectFilter });
|
|
130
|
+
onProgress?.('[2/4] Sampling... done');
|
|
131
|
+
// 3. Warm-up
|
|
132
|
+
onProgress?.('[3/4] Warm-up queries...');
|
|
133
|
+
// Warm-up is handled within each benchmark module
|
|
134
|
+
onProgress?.('[3/4] Warm-up... done');
|
|
135
|
+
}
|
|
136
|
+
// 4. Run requested categories
|
|
137
|
+
if (categories.includes('retrieval') && sample) {
|
|
138
|
+
onProgress?.('[4/4] Retrieval quality benchmarks');
|
|
139
|
+
const retrievalResult = await runRetrievalBenchmarks(sample, topK, onProgress);
|
|
140
|
+
retrieval = retrievalResult.result;
|
|
141
|
+
skipped.push(...retrievalResult.skipped);
|
|
142
|
+
}
|
|
143
|
+
if (categories.includes('graph') && sample) {
|
|
144
|
+
onProgress?.('Graph value benchmarks');
|
|
145
|
+
const graphResult = await runGraphValueBenchmarks(sample, topK, onProgress);
|
|
146
|
+
graphValue = graphResult.result;
|
|
147
|
+
skipped.push(...graphResult.skipped);
|
|
148
|
+
}
|
|
149
|
+
if (categories.includes('latency') && sample) {
|
|
150
|
+
onProgress?.('Latency benchmarks');
|
|
151
|
+
const effectiveProject = projectFilter ?? getDistinctProjects()[0]?.slug;
|
|
152
|
+
latency = await runLatencyBenchmarks(sample, effectiveProject, onProgress);
|
|
153
|
+
}
|
|
154
|
+
// Compute composite score
|
|
155
|
+
const overallScore = computeOverallScore(health, retrieval, graphValue, latency);
|
|
156
|
+
// Generate highlights
|
|
157
|
+
const highlights = generateHighlights(health, retrieval, graphValue, latency);
|
|
158
|
+
// Build result
|
|
159
|
+
const result = {
|
|
160
|
+
timestamp: new Date().toISOString(),
|
|
161
|
+
profile,
|
|
162
|
+
collectionStats: health,
|
|
163
|
+
retrieval,
|
|
164
|
+
graphValue,
|
|
165
|
+
latency,
|
|
166
|
+
overallScore,
|
|
167
|
+
highlights,
|
|
168
|
+
skipped,
|
|
169
|
+
};
|
|
170
|
+
// Tuning recommendations
|
|
171
|
+
if (includeTuning) {
|
|
172
|
+
result.tuning = generateTuningRecommendations(result);
|
|
173
|
+
}
|
|
174
|
+
// Historical trending
|
|
175
|
+
const previousRun = getLatestBenchmarkRun();
|
|
176
|
+
if (previousRun) {
|
|
177
|
+
result.trend = computeTrend(previousRun, result);
|
|
178
|
+
}
|
|
179
|
+
// Store this run
|
|
180
|
+
try {
|
|
181
|
+
const configSnapshot = loadConfig();
|
|
182
|
+
storeBenchmarkRun(result, configSnapshot);
|
|
183
|
+
}
|
|
184
|
+
catch {
|
|
185
|
+
// Non-critical: don't fail benchmark if history storage fails
|
|
186
|
+
}
|
|
187
|
+
return result;
|
|
188
|
+
}
|
|
189
|
+
//# sourceMappingURL=runner.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"runner.js","sourceRoot":"","sources":["../../../src/eval/collection-benchmark/runner.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAC/C,OAAO,EAAE,mBAAmB,EAAE,MAAM,aAAa,CAAC;AAClD,OAAO,EAAE,sBAAsB,EAAE,MAAM,gBAAgB,CAAC;AACxD,OAAO,EAAE,uBAAuB,EAAE,MAAM,kBAAkB,CAAC;AAC3D,OAAO,EAAE,oBAAoB,EAAE,MAAM,cAAc,CAAC;AACpD,OAAO,EAAE,6BAA6B,EAAE,MAAM,aAAa,CAAC;AAC5D,OAAO,EAAE,iBAAiB,EAAE,qBAAqB,EAAE,YAAY,EAAE,MAAM,cAAc,CAAC;AACtF,OAAO,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AACpD,OAAO,EAAE,mBAAmB,EAAE,MAAM,8BAA8B,CAAC;AAenE;;GAEG;AACH,SAAS,iBAAiB,CACxB,OAAyB,EACzB,QAA8B;IAE9B,IAAI,QAAQ,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC;QAAE,OAAO,QAAQ,CAAC;IAErD,QAAQ,OAAO,EAAE,CAAC;QAChB,KAAK,OAAO,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QAChC,KAAK,UAAU,CAAC,CAAC,OAAO,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;QAChD,KAAK,MAAM,CAAC,CAAC,OAAO,CAAC,QAAQ,EAAE,WAAW,EAAE,OAAO,EAAE,SAAS,CAAC,CAAC;IAClE,CAAC;AACH,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,mBAAmB,CACjC,MAAoB,EACpB,SAA2B,EAC3B,KAAwB,EACxB,OAAuB;IAEvB,MAAM,MAAM,GAA6C,EAAE,CAAC;IAE5D,uBAAuB;IACvB,MAAM,WAAW,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,CAChC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,MAAM,CAAC,gBAAgB,GAAG,CAAC,CAAC,GAAG,EAAE;QAC7C,MAAM,CAAC,eAAe,GAAG,EAAE;QAC3B,CAAC,CAAC,GAAG,MAAM,CAAC,qBAAqB,CAAC,GAAG,EAAE,CACxC,CAAC,CAAC;IACH,MAAM,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,WAAW,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC;IAEhD,0BAA0B;IAC1B,IAAI,SAAS,EAAE,CAAC;QACd,MAAM,cAAc,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,EACjC,SAAS,CAAC,kBAAkB,GAAG,EAAE;YACjC,SAAS,CAAC,kBAAkB,GAAG,EAAE;YACjC,SAAS,CAAC,aAAa,GAAG,EAAE;YAC5B,SAAS,CAAC,eAAe,GAAG,EAAE,CAC/B,CAAC;QACF,MAAM,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,cAAc,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC;IACrD,CAAC;IAED,4BAA4B;IAC5B,IAAI,KAAK,EAAE,CAAC;QACV,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,iBAAiB,CAAC,iBAAiB,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;QAClF,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;QAC1C,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,QAAQ,GAAG,EAAE,GAAG,SAAS,GAAG,EAAE,CAAC,CAAC;QACjE,MAAM,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,UAAU,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC;IACjD,CAAC;IAED,iDAAiD;IACjD,IAAI,OAAO,EAAE,CAAC;QACZ,MAAM,kBAAkB,GAAG,OAAO,CAAC,MAAM,CAAC,GAAG,IAAI,EAAE,CAAC,CAAC,CAAC,GAAG;YACvD,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,GAAG,IAAI,GAAG,CAAC,CAAC,CAAC,EAAE;gBAChC,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,GAAG,IAAI,GAAG,CAAC,CAAC,CAAC,EAAE;oBAChC,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,GAAG,IAAI,GAAG,CAAC,CAAC,CAAC,EAAE;wBAChC,CAAC,CAAC,EAAE,CAAC;QACP,MAAM,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,kBAAkB,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC;IACzD,CAAC;IAED,wCAAwC;IACxC,MAAM,WAAW,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;IACjE,IAAI,WAAW,KAAK,CAAC;QAAE,OAAO,CAAC,CAAC;IAEhC,OAAO,IAAI,CAAC,KAAK,CACf,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,WAAW,CACrE,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,kBAAkB,CAChC,MAAoB,EACpB,SAA2B,EAC3B,KAAwB,EACxB,OAAuB;IAEvB,MAAM,UAAU,GAAa,EAAE,CAAC;IAEhC,oBAAoB;IACpB,UAAU,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,eAAe,GAAG,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,2CAA2C,CAAC,CAAC;IAEzG,IAAI,MAAM,CAAC,qBAAqB,GAAG,IAAI,EAAE,CAAC;QACxC,UAAU,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,qBAAqB,GAAG,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,qDAAqD,CAAC,CAAC;IAChI,CAAC;IAED,IAAI,MAAM,CAAC,YAAY,EAAE,CAAC;QACxB,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CACrB,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,GAAG,IAAI,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,OAAO,EAAE,CAAC;YACnG,CAAC,IAAI,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,CACtB,CAAC;QACF,UAAU,CAAC,IAAI,CAAC,oBAAoB,IAAI,gBAAgB,MAAM,CAAC,YAAY,aAAa,CAAC,CAAC;IAC5F,CAAC;IAED,uBAAuB;IACvB,IAAI,SAAS,EAAE,CAAC;QACd,IAAI,SAAS,CAAC,kBAAkB,GAAG,CAAC,EAAE,CAAC;YACrC,UAAU,CAAC,IAAI,CAAC,6BAA6B,CAAC,SAAS,CAAC,kBAAkB,GAAG,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;QACnG,CAAC;QACD,IAAI,SAAS,CAAC,kBAAkB,GAAG,CAAC,EAAE,CAAC;YACrC,UAAU,CAAC,IAAI,CAAC,kCAAkC,CAAC,SAAS,CAAC,kBAAkB,GAAG,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;QACxG,CAAC;QACD,IAAI,SAAS,CAAC,eAAe,GAAG,CAAC,EAAE,CAAC;YAClC,UAAU,CAAC,IAAI,CAAC,qBAAqB,CAAC,SAAS,CAAC,eAAe,GAAG,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,mCAAmC,CAAC,CAAC;QACxH,CAAC;IACH,CAAC;IAED,mBAAmB;IACnB,IAAI,KAAK,EAAE,CAAC;QACV,IAAI,KAAK,CAAC,iBAAiB,CAAC,iBAAiB,GAAG,GAAG,EAAE,CAAC;YACpD,UAAU,CAAC,IAAI,CAAC,wBAAwB,KAAK,CAAC,iBAAiB,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC,CAAC,+BAA+B,CAAC,CAAC;QAC/H,CAAC;QACD,IAAI,KAAK,CAAC,IAAI,GAAG,GAAG,EAAE,CAAC;YACrB,UAAU,CAAC,IAAI,CAAC,4BAA4B,CAAC,KAAK,CAAC,IAAI,GAAG,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,oBAAoB,CAAC,CAAC;QACjG,CAAC;IACH,CAAC;IAED,qBAAqB;IACrB,IAAI,OAAO,EAAE,CAAC;QACZ,UAAU,CAAC,IAAI,CAAC,uBAAuB,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;IAC5E,CAAC;IAED,OAAO,UAAU,CAAC;AACpB,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,sBAAsB,CAC1C,UAAsC,EAAE;IAExC,MAAM,EACJ,OAAO,GAAG,UAAU,EACpB,UAAU,GAAG,EAAE,EACf,IAAI,EACJ,aAAa,EACb,IAAI,GAAG,EAAE,EACT,aAAa,GAAG,IAAI,EACpB,UAAU,GACX,GAAG,OAAO,CAAC;IAEZ,MAAM,UAAU,GAAG,iBAAiB,CAAC,OAAO,EAAE,OAAO,CAAC,UAAU,CAAC,CAAC;IAElE,MAAM,OAAO,GAAuB,EAAE,CAAC;IACvC,IAAI,SAAsC,CAAC;IAC3C,IAAI,UAAwC,CAAC;IAC7C,IAAI,OAAkC,CAAC;IAEvC,uBAAuB;IACvB,UAAU,EAAE,CAAC,4BAA4B,CAAC,CAAC;IAC3C,MAAM,qBAAqB,GAAG,UAAU,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,UAAU,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;IAC/F,MAAM,MAAM,GAAG,MAAM,mBAAmB,CAAC,qBAAqB,CAAC,CAAC;IAChE,UAAU,EAAE,CAAC,iCAAiC,CAAC,CAAC;IAEhD,gCAAgC;IAChC,IAAI,MAAM,CAAC;IACX,IAAI,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,KAAK,QAAQ,CAAC,EAAE,CAAC;QACzC,UAAU,EAAE,CAAC,2BAA2B,UAAU,WAAW,IAAI,KAAK,IAAI,IAAI,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,UAAU,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC;QAChI,MAAM,GAAG,eAAe,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC,CAAC;QAC9D,UAAU,EAAE,CAAC,wBAAwB,CAAC,CAAC;QAEvC,aAAa;QACb,UAAU,EAAE,CAAC,0BAA0B,CAAC,CAAC;QACzC,kDAAkD;QAClD,UAAU,EAAE,CAAC,uBAAuB,CAAC,CAAC;IACxC,CAAC;IAED,8BAA8B;IAC9B,IAAI,UAAU,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,MAAM,EAAE,CAAC;QAC/C,UAAU,EAAE,CAAC,oCAAoC,CAAC,CAAC;QACnD,MAAM,eAAe,GAAG,MAAM,sBAAsB,CAAC,MAAM,EAAE,IAAI,EAAE,UAAU,CAAC,CAAC;QAC/E,SAAS,GAAG,eAAe,CAAC,MAAM,CAAC;QACnC,OAAO,CAAC,IAAI,CAAC,GAAG,eAAe,CAAC,OAAO,CAAC,CAAC;IAC3C,CAAC;IAED,IAAI,UAAU,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,MAAM,EAAE,CAAC;QAC3C,UAAU,EAAE,CAAC,wBAAwB,CAAC,CAAC;QACvC,MAAM,WAAW,GAAG,MAAM,uBAAuB,CAAC,MAAM,EAAE,IAAI,EAAE,UAAU,CAAC,CAAC;QAC5E,UAAU,GAAG,WAAW,CAAC,MAAM,CAAC;QAChC,OAAO,CAAC,IAAI,CAAC,GAAG,WAAW,CAAC,OAAO,CAAC,CAAC;IACvC,CAAC;IAED,IAAI,UAAU,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,MAAM,EAAE,CAAC;QAC7C,UAAU,EAAE,CAAC,oBAAoB,CAAC,CAAC;QACnC,MAAM,gBAAgB,GAAG,aAAa,IAAI,mBAAmB,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC;QACzE,OAAO,GAAG,MAAM,oBAAoB,CAAC,MAAM,EAAE,gBAAgB,EAAE,UAAU,CAAC,CAAC;IAC7E,CAAC;IAED,0BAA0B;IAC1B,MAAM,YAAY,GAAG,mBAAmB,CAAC,MAAM,EAAE,SAAS,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC;IAEjF,sBAAsB;IACtB,MAAM,UAAU,GAAG,kBAAkB,CAAC,MAAM,EAAE,SAAS,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC;IAE9E,eAAe;IACf,MAAM,MAAM,GAA8B;QACxC,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;QACnC,OAAO;QACP,eAAe,EAAE,MAAM;QACvB,SAAS;QACT,UAAU;QACV,OAAO;QACP,YAAY;QACZ,UAAU;QACV,OAAO;KACR,CAAC;IAEF,yBAAyB;IACzB,IAAI,aAAa,EAAE,CAAC;QAClB,MAAM,CAAC,MAAM,GAAG,6BAA6B,CAAC,MAAM,CAAC,CAAC;IACxD,CAAC;IAED,sBAAsB;IACtB,MAAM,WAAW,GAAG,qBAAqB,EAAE,CAAC;IAC5C,IAAI,WAAW,EAAE,CAAC;QAChB,MAAM,CAAC,KAAK,GAAG,YAAY,CAAC,WAAW,EAAE,MAAM,CAAC,CAAC;IACnD,CAAC;IAED,iBAAiB;IACjB,IAAI,CAAC;QACH,MAAM,cAAc,GAAG,UAAU,EAAE,CAAC;QACpC,iBAAiB,CAAC,MAAM,EAAE,cAAoD,CAAC,CAAC;IAClF,CAAC;IAAC,MAAM,CAAC;QACP,8DAA8D;IAChE,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC"}
|
|
@@ -0,0 +1,18 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Query sampling with seeded PRNG for reproducible benchmarks.
|
|
3
|
+
*
|
|
4
|
+
* Generates benchmark samples from the user's collection, including
|
|
5
|
+
* adjacent pairs (ground truth for recall), cross-session pairs, and
|
|
6
|
+
* cross-project negative controls.
|
|
7
|
+
*/
|
|
8
|
+
import type { StoredChunk, StoredEdge } from '../../storage/types.js';
|
|
9
|
+
import type { BenchmarkSample, SamplerOptions, SamplerThresholds } from './types.js';
|
|
10
|
+
/**
|
|
11
|
+
* Check minimum thresholds for each benchmark type.
|
|
12
|
+
*/
|
|
13
|
+
export declare function checkThresholds(chunks: StoredChunk[], edges: StoredEdge[]): SamplerThresholds;
|
|
14
|
+
/**
|
|
15
|
+
* Generate benchmark samples from the collection.
|
|
16
|
+
*/
|
|
17
|
+
export declare function generateSamples(options: SamplerOptions): BenchmarkSample;
|
|
18
|
+
//# sourceMappingURL=sampler.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"sampler.d.ts","sourceRoot":"","sources":["../../../src/eval/collection-benchmark/sampler.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAIH,OAAO,KAAK,EAAE,WAAW,EAAE,UAAU,EAAiB,MAAM,wBAAwB,CAAC;AACrF,OAAO,KAAK,EACV,eAAe,EACf,cAAc,EACd,iBAAiB,EAIlB,MAAM,YAAY,CAAC;AAgBpB;;GAEG;AACH,wBAAgB,eAAe,CAC7B,MAAM,EAAE,WAAW,EAAE,EACrB,KAAK,EAAE,UAAU,EAAE,GAClB,iBAAiB,CAuDnB;AAED;;GAEG;AACH,wBAAgB,eAAe,CAAC,OAAO,EAAE,cAAc,GAAG,eAAe,CA4HxE"}
|
|
@@ -0,0 +1,186 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Query sampling with seeded PRNG for reproducible benchmarks.
|
|
3
|
+
*
|
|
4
|
+
* Generates benchmark samples from the user's collection, including
|
|
5
|
+
* adjacent pairs (ground truth for recall), cross-session pairs, and
|
|
6
|
+
* cross-project negative controls.
|
|
7
|
+
*/
|
|
8
|
+
import { getAllChunks } from '../../storage/chunk-store.js';
|
|
9
|
+
import { getAllEdges } from '../../storage/edge-store.js';
|
|
10
|
+
/**
|
|
11
|
+
* Deterministic seeded shuffle (same algorithm as annotation-schema.ts).
|
|
12
|
+
*/
|
|
13
|
+
function seededShuffle(arr, seed) {
|
|
14
|
+
const result = [...arr];
|
|
15
|
+
let s = seed;
|
|
16
|
+
for (let i = result.length - 1; i > 0; i--) {
|
|
17
|
+
s = (s * 1664525 + 1013904223) & 0x7fffffff;
|
|
18
|
+
const j = s % (i + 1);
|
|
19
|
+
[result[i], result[j]] = [result[j], result[i]];
|
|
20
|
+
}
|
|
21
|
+
return result;
|
|
22
|
+
}
|
|
23
|
+
/**
|
|
24
|
+
* Check minimum thresholds for each benchmark type.
|
|
25
|
+
*/
|
|
26
|
+
export function checkThresholds(chunks, edges) {
|
|
27
|
+
const reasons = new Map();
|
|
28
|
+
// Group chunks by session
|
|
29
|
+
const sessionChunks = new Map();
|
|
30
|
+
for (const chunk of chunks) {
|
|
31
|
+
const list = sessionChunks.get(chunk.sessionId) ?? [];
|
|
32
|
+
list.push(chunk);
|
|
33
|
+
sessionChunks.set(chunk.sessionId, list);
|
|
34
|
+
}
|
|
35
|
+
// Adjacent recall: need >= 2 sessions with >= 3 chunks each
|
|
36
|
+
const sessionsWithEnoughChunks = [...sessionChunks.values()].filter(c => c.length >= 3);
|
|
37
|
+
const canRunAdjacentRecall = sessionsWithEnoughChunks.length >= 2;
|
|
38
|
+
if (!canRunAdjacentRecall) {
|
|
39
|
+
reasons.set('adjacentRecall', `need >=2 sessions with >=3 chunks each, you have ${sessionsWithEnoughChunks.length}`);
|
|
40
|
+
}
|
|
41
|
+
// Cross-session bridging: need >= 3 sessions in same project with cross-session edges
|
|
42
|
+
const projectSessions = new Map();
|
|
43
|
+
for (const chunk of chunks) {
|
|
44
|
+
const sessions = projectSessions.get(chunk.sessionSlug) ?? new Set();
|
|
45
|
+
sessions.add(chunk.sessionId);
|
|
46
|
+
projectSessions.set(chunk.sessionSlug, sessions);
|
|
47
|
+
}
|
|
48
|
+
const crossSessionEdges = edges.filter(e => e.referenceType === 'file-path' || e.referenceType === 'code-entity');
|
|
49
|
+
const projectsWithCrossSession = [...projectSessions.entries()].filter(([, sessions]) => sessions.size >= 3);
|
|
50
|
+
const canRunCrossSessionBridging = projectsWithCrossSession.length > 0 && crossSessionEdges.length > 0;
|
|
51
|
+
if (!canRunCrossSessionBridging) {
|
|
52
|
+
const sessCount = Math.max(...[...projectSessions.values()].map(s => s.size), 0);
|
|
53
|
+
reasons.set('crossSessionBridging', `need >=3 sessions in same project with cross-session edges, best project has ${sessCount} sessions`);
|
|
54
|
+
}
|
|
55
|
+
// Precision@K: need >= 2 projects with >= 10 chunks each
|
|
56
|
+
const projectChunkCounts = new Map();
|
|
57
|
+
for (const chunk of chunks) {
|
|
58
|
+
projectChunkCounts.set(chunk.sessionSlug, (projectChunkCounts.get(chunk.sessionSlug) ?? 0) + 1);
|
|
59
|
+
}
|
|
60
|
+
const projectsWithEnoughChunks = [...projectChunkCounts.entries()].filter(([, count]) => count >= 10);
|
|
61
|
+
const canRunPrecisionAtK = projectsWithEnoughChunks.length >= 2;
|
|
62
|
+
if (!canRunPrecisionAtK) {
|
|
63
|
+
reasons.set('precisionAtK', `need >=2 projects with >=10 chunks each, you have ${projectsWithEnoughChunks.length}`);
|
|
64
|
+
}
|
|
65
|
+
return {
|
|
66
|
+
canRunAdjacentRecall,
|
|
67
|
+
canRunCrossSessionBridging,
|
|
68
|
+
canRunPrecisionAtK,
|
|
69
|
+
reasons,
|
|
70
|
+
};
|
|
71
|
+
}
|
|
72
|
+
/**
|
|
73
|
+
* Generate benchmark samples from the collection.
|
|
74
|
+
*/
|
|
75
|
+
export function generateSamples(options) {
|
|
76
|
+
const { sampleSize, seed = 42, projectFilter } = options;
|
|
77
|
+
let chunks = getAllChunks();
|
|
78
|
+
let edges = getAllEdges();
|
|
79
|
+
// Apply project filter if specified
|
|
80
|
+
if (projectFilter) {
|
|
81
|
+
chunks = chunks.filter(c => c.sessionSlug === projectFilter);
|
|
82
|
+
const chunkIds = new Set(chunks.map(c => c.id));
|
|
83
|
+
edges = edges.filter(e => chunkIds.has(e.sourceChunkId) || chunkIds.has(e.targetChunkId));
|
|
84
|
+
}
|
|
85
|
+
const thresholds = checkThresholds(chunks, edges);
|
|
86
|
+
// Group chunks by session, sorted by start_time
|
|
87
|
+
const sessionChunks = new Map();
|
|
88
|
+
for (const chunk of chunks) {
|
|
89
|
+
const list = sessionChunks.get(chunk.sessionId) ?? [];
|
|
90
|
+
list.push(chunk);
|
|
91
|
+
sessionChunks.set(chunk.sessionId, list);
|
|
92
|
+
}
|
|
93
|
+
for (const list of sessionChunks.values()) {
|
|
94
|
+
list.sort((a, b) => a.startTime.localeCompare(b.startTime));
|
|
95
|
+
}
|
|
96
|
+
// Sample query chunk IDs
|
|
97
|
+
const shuffledChunks = seededShuffle(chunks, seed);
|
|
98
|
+
const queryChunkIds = shuffledChunks.slice(0, sampleSize).map(c => c.id);
|
|
99
|
+
// Generate adjacent pairs
|
|
100
|
+
const adjacentPairs = [];
|
|
101
|
+
if (thresholds.canRunAdjacentRecall) {
|
|
102
|
+
let pairSeed = seed + 100;
|
|
103
|
+
for (const [sessionId, sessionList] of sessionChunks) {
|
|
104
|
+
if (sessionList.length < 3)
|
|
105
|
+
continue;
|
|
106
|
+
// Exclude first and last chunks, sample from middle
|
|
107
|
+
const middleIndices = Array.from({ length: sessionList.length - 2 }, (_, i) => i + 1);
|
|
108
|
+
const shuffled = seededShuffle(middleIndices, pairSeed++);
|
|
109
|
+
const limit = Math.min(shuffled.length, Math.ceil(sampleSize / sessionChunks.size) + 1);
|
|
110
|
+
for (let k = 0; k < limit && adjacentPairs.length < sampleSize; k++) {
|
|
111
|
+
const idx = shuffled[k];
|
|
112
|
+
const chunk = sessionList[idx];
|
|
113
|
+
// Pick one adjacent (previous or next)
|
|
114
|
+
const adjacentIdx = (pairSeed++ % 2 === 0) ? idx - 1 : idx + 1;
|
|
115
|
+
if (adjacentIdx >= 0 && adjacentIdx < sessionList.length) {
|
|
116
|
+
adjacentPairs.push({
|
|
117
|
+
queryChunkId: chunk.id,
|
|
118
|
+
adjacentChunkId: sessionList[adjacentIdx].id,
|
|
119
|
+
sessionId,
|
|
120
|
+
});
|
|
121
|
+
}
|
|
122
|
+
}
|
|
123
|
+
}
|
|
124
|
+
}
|
|
125
|
+
// Generate cross-session pairs from edges
|
|
126
|
+
const crossSessionPairs = [];
|
|
127
|
+
if (thresholds.canRunCrossSessionBridging) {
|
|
128
|
+
const chunkSessionMap = new Map();
|
|
129
|
+
for (const chunk of chunks) {
|
|
130
|
+
chunkSessionMap.set(chunk.id, chunk.sessionId);
|
|
131
|
+
}
|
|
132
|
+
const bridgingEdges = edges.filter(e => {
|
|
133
|
+
const srcSession = chunkSessionMap.get(e.sourceChunkId);
|
|
134
|
+
const tgtSession = chunkSessionMap.get(e.targetChunkId);
|
|
135
|
+
return srcSession && tgtSession && srcSession !== tgtSession &&
|
|
136
|
+
(e.referenceType === 'file-path' || e.referenceType === 'code-entity');
|
|
137
|
+
});
|
|
138
|
+
const shuffledEdges = seededShuffle(bridgingEdges, seed + 200);
|
|
139
|
+
for (const edge of shuffledEdges.slice(0, sampleSize)) {
|
|
140
|
+
crossSessionPairs.push({
|
|
141
|
+
chunkIdA: edge.sourceChunkId,
|
|
142
|
+
chunkIdB: edge.targetChunkId,
|
|
143
|
+
edgeType: edge.referenceType,
|
|
144
|
+
});
|
|
145
|
+
}
|
|
146
|
+
}
|
|
147
|
+
// Generate cross-project pairs (negative controls)
|
|
148
|
+
const crossProjectPairs = [];
|
|
149
|
+
if (thresholds.canRunPrecisionAtK) {
|
|
150
|
+
const projectChunks = new Map();
|
|
151
|
+
for (const chunk of chunks) {
|
|
152
|
+
const list = projectChunks.get(chunk.sessionSlug) ?? [];
|
|
153
|
+
list.push(chunk);
|
|
154
|
+
projectChunks.set(chunk.sessionSlug, list);
|
|
155
|
+
}
|
|
156
|
+
const projectKeys = [...projectChunks.keys()].filter(k => (projectChunks.get(k)?.length ?? 0) >= 10);
|
|
157
|
+
if (projectKeys.length >= 2) {
|
|
158
|
+
const shuffledKeys = seededShuffle(projectKeys, seed + 300);
|
|
159
|
+
let count = 0;
|
|
160
|
+
for (let pi = 0; pi < shuffledKeys.length - 1 && count < sampleSize; pi++) {
|
|
161
|
+
const chunksA = seededShuffle(projectChunks.get(shuffledKeys[pi]), seed + 301);
|
|
162
|
+
for (let pj = pi + 1; pj < shuffledKeys.length && count < sampleSize; pj++) {
|
|
163
|
+
const chunksB = seededShuffle(projectChunks.get(shuffledKeys[pj]), seed + 302);
|
|
164
|
+
const limit = Math.min(chunksA.length, chunksB.length, sampleSize - count);
|
|
165
|
+
for (let i = 0; i < limit; i++) {
|
|
166
|
+
crossProjectPairs.push({
|
|
167
|
+
chunkIdA: chunksA[i].id,
|
|
168
|
+
projectA: shuffledKeys[pi],
|
|
169
|
+
chunkIdB: chunksB[i].id,
|
|
170
|
+
projectB: shuffledKeys[pj],
|
|
171
|
+
});
|
|
172
|
+
count++;
|
|
173
|
+
}
|
|
174
|
+
}
|
|
175
|
+
}
|
|
176
|
+
}
|
|
177
|
+
}
|
|
178
|
+
return {
|
|
179
|
+
queryChunkIds,
|
|
180
|
+
adjacentPairs,
|
|
181
|
+
crossSessionPairs,
|
|
182
|
+
crossProjectPairs,
|
|
183
|
+
thresholds,
|
|
184
|
+
};
|
|
185
|
+
}
|
|
186
|
+
//# sourceMappingURL=sampler.js.map
|