@claude-flow/cli 3.10.46 → 3.12.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/.claude/agents/analysis/analyze-code-quality.md +178 -178
- package/.claude/agents/analysis/code-analyzer.md +209 -209
- package/.claude/agents/analysis/code-review/analyze-code-quality.md +178 -178
- package/.claude/agents/architecture/arch-system-design.md +156 -156
- package/.claude/agents/architecture/system-design/arch-system-design.md +154 -154
- package/.claude/agents/browser/browser-agent.yaml +182 -182
- package/.claude/agents/consensus/byzantine-coordinator.md +62 -62
- package/.claude/agents/consensus/crdt-synchronizer.md +996 -996
- package/.claude/agents/consensus/gossip-coordinator.md +62 -62
- package/.claude/agents/consensus/performance-benchmarker.md +850 -850
- package/.claude/agents/consensus/quorum-manager.md +822 -822
- package/.claude/agents/consensus/raft-manager.md +62 -62
- package/.claude/agents/consensus/security-manager.md +621 -621
- package/.claude/agents/core/planner.md +374 -374
- package/.claude/agents/custom/test-long-runner.md +44 -44
- package/.claude/agents/data/data-ml-model.md +444 -444
- package/.claude/agents/data/ml/data-ml-model.md +192 -192
- package/.claude/agents/development/backend/dev-backend-api.md +141 -141
- package/.claude/agents/development/dev-backend-api.md +344 -344
- package/.claude/agents/devops/ci-cd/ops-cicd-github.md +163 -163
- package/.claude/agents/devops/ops-cicd-github.md +164 -164
- package/.claude/agents/documentation/api-docs/docs-api-openapi.md +173 -173
- package/.claude/agents/documentation/docs-api-openapi.md +354 -354
- package/.claude/agents/flow-nexus/app-store.md +87 -87
- package/.claude/agents/flow-nexus/authentication.md +68 -68
- package/.claude/agents/flow-nexus/challenges.md +80 -80
- package/.claude/agents/flow-nexus/neural-network.md +87 -87
- package/.claude/agents/flow-nexus/payments.md +82 -82
- package/.claude/agents/flow-nexus/sandbox.md +75 -75
- package/.claude/agents/flow-nexus/swarm.md +75 -75
- package/.claude/agents/flow-nexus/user-tools.md +95 -95
- package/.claude/agents/flow-nexus/workflow.md +83 -83
- package/.claude/agents/github/code-review-swarm.md +377 -377
- package/.claude/agents/github/github-modes.md +172 -172
- package/.claude/agents/github/issue-tracker.md +575 -575
- package/.claude/agents/github/multi-repo-swarm.md +552 -552
- package/.claude/agents/github/pr-manager.md +437 -437
- package/.claude/agents/github/project-board-sync.md +508 -508
- package/.claude/agents/github/release-manager.md +604 -604
- package/.claude/agents/github/release-swarm.md +582 -582
- package/.claude/agents/github/repo-architect.md +397 -397
- package/.claude/agents/github/swarm-issue.md +572 -572
- package/.claude/agents/github/swarm-pr.md +427 -427
- package/.claude/agents/github/sync-coordinator.md +451 -451
- package/.claude/agents/github/workflow-automation.md +902 -902
- package/.claude/agents/goal/agent.md +815 -815
- package/.claude/agents/optimization/benchmark-suite.md +664 -664
- package/.claude/agents/optimization/load-balancer.md +430 -430
- package/.claude/agents/optimization/performance-monitor.md +671 -671
- package/.claude/agents/optimization/resource-allocator.md +673 -673
- package/.claude/agents/optimization/topology-optimizer.md +807 -807
- package/.claude/agents/payments/agentic-payments.md +126 -126
- package/.claude/agents/sona/sona-learning-optimizer.md +74 -74
- package/.claude/agents/sparc/architecture.md +698 -698
- package/.claude/agents/sparc/pseudocode.md +519 -519
- package/.claude/agents/sparc/refinement.md +801 -801
- package/.claude/agents/sparc/specification.md +477 -477
- package/.claude/agents/specialized/mobile/spec-mobile-react-native.md +224 -224
- package/.claude/agents/specialized/spec-mobile-react-native.md +226 -226
- package/.claude/agents/sublinear/consensus-coordinator.md +337 -337
- package/.claude/agents/sublinear/matrix-optimizer.md +184 -184
- package/.claude/agents/sublinear/pagerank-analyzer.md +298 -298
- package/.claude/agents/sublinear/performance-optimizer.md +367 -367
- package/.claude/agents/sublinear/trading-predictor.md +245 -245
- package/.claude/agents/swarm/adaptive-coordinator.md +1126 -1126
- package/.claude/agents/swarm/hierarchical-coordinator.md +709 -709
- package/.claude/agents/swarm/mesh-coordinator.md +962 -962
- package/.claude/agents/templates/automation-smart-agent.md +204 -204
- package/.claude/agents/templates/base-template-generator.md +289 -289
- package/.claude/agents/templates/coordinator-swarm-init.md +89 -89
- package/.claude/agents/templates/github-pr-manager.md +176 -176
- package/.claude/agents/templates/implementer-sparc-coder.md +258 -258
- package/.claude/agents/templates/memory-coordinator.md +186 -186
- package/.claude/agents/templates/orchestrator-task.md +138 -138
- package/.claude/agents/templates/performance-analyzer.md +198 -198
- package/.claude/agents/templates/sparc-coordinator.md +513 -513
- package/.claude/agents/testing/production-validator.md +394 -394
- package/.claude/agents/testing/tdd-london-swarm.md +243 -243
- package/.claude/agents/v3/aidefence-guardian.md +282 -282
- package/.claude/agents/v3/claims-authorizer.md +208 -208
- package/.claude/agents/v3/collective-intelligence-coordinator.md +993 -993
- package/.claude/agents/v3/ddd-domain-expert.md +220 -220
- package/.claude/agents/v3/injection-analyst.md +236 -236
- package/.claude/agents/v3/performance-engineer.md +1233 -1233
- package/.claude/agents/v3/pii-detector.md +151 -151
- package/.claude/agents/v3/reasoningbank-learner.md +213 -213
- package/.claude/agents/v3/security-architect-aidefence.md +410 -410
- package/.claude/agents/v3/security-architect.md +867 -867
- package/.claude/agents/v3/swarm-memory-manager.md +157 -157
- package/.claude/agents/v3/v3-integration-architect.md +205 -205
- package/.claude/commands/agents/README.md +50 -50
- package/.claude/commands/agents/agent-capabilities.md +140 -140
- package/.claude/commands/agents/agent-coordination.md +28 -28
- package/.claude/commands/agents/agent-spawning.md +28 -28
- package/.claude/commands/agents/agent-types.md +216 -216
- package/.claude/commands/agents/health.md +139 -139
- package/.claude/commands/agents/list.md +100 -100
- package/.claude/commands/agents/logs.md +130 -130
- package/.claude/commands/agents/metrics.md +122 -122
- package/.claude/commands/agents/pool.md +127 -127
- package/.claude/commands/agents/spawn.md +140 -140
- package/.claude/commands/agents/status.md +115 -115
- package/.claude/commands/agents/stop.md +102 -102
- package/.claude/commands/analysis/COMMAND_COMPLIANCE_REPORT.md +53 -53
- package/.claude/commands/analysis/README.md +9 -9
- package/.claude/commands/analysis/bottleneck-detect.md +162 -162
- package/.claude/commands/analysis/performance-bottlenecks.md +58 -58
- package/.claude/commands/analysis/performance-report.md +25 -25
- package/.claude/commands/analysis/token-efficiency.md +44 -44
- package/.claude/commands/analysis/token-usage.md +25 -25
- package/.claude/commands/automation/README.md +9 -9
- package/.claude/commands/automation/auto-agent.md +122 -122
- package/.claude/commands/automation/self-healing.md +105 -105
- package/.claude/commands/automation/session-memory.md +89 -89
- package/.claude/commands/automation/smart-agents.md +72 -72
- package/.claude/commands/automation/smart-spawn.md +25 -25
- package/.claude/commands/automation/workflow-select.md +25 -25
- package/.claude/commands/claude-flow-help.md +103 -103
- package/.claude/commands/claude-flow-memory.md +107 -107
- package/.claude/commands/claude-flow-swarm.md +205 -205
- package/.claude/commands/coordination/README.md +9 -9
- package/.claude/commands/coordination/agent-spawn.md +25 -25
- package/.claude/commands/coordination/init.md +44 -44
- package/.claude/commands/coordination/orchestrate.md +43 -43
- package/.claude/commands/coordination/spawn.md +45 -45
- package/.claude/commands/coordination/swarm-init.md +85 -85
- package/.claude/commands/coordination/task-orchestrate.md +25 -25
- package/.claude/commands/github/README.md +11 -11
- package/.claude/commands/github/code-review-swarm.md +513 -513
- package/.claude/commands/github/code-review.md +25 -25
- package/.claude/commands/github/github-modes.md +146 -146
- package/.claude/commands/github/github-swarm.md +121 -121
- package/.claude/commands/github/issue-tracker.md +291 -291
- package/.claude/commands/github/issue-triage.md +25 -25
- package/.claude/commands/github/multi-repo-swarm.md +518 -518
- package/.claude/commands/github/pr-enhance.md +26 -26
- package/.claude/commands/github/pr-manager.md +169 -169
- package/.claude/commands/github/project-board-sync.md +470 -470
- package/.claude/commands/github/release-manager.md +339 -339
- package/.claude/commands/github/release-swarm.md +543 -543
- package/.claude/commands/github/repo-analyze.md +25 -25
- package/.claude/commands/github/repo-architect.md +366 -366
- package/.claude/commands/github/swarm-issue.md +484 -484
- package/.claude/commands/github/swarm-pr.md +287 -287
- package/.claude/commands/github/sync-coordinator.md +302 -302
- package/.claude/commands/github/workflow-automation.md +441 -441
- package/.claude/commands/hive-mind/README.md +17 -17
- package/.claude/commands/hive-mind/hive-mind-consensus.md +8 -8
- package/.claude/commands/hive-mind/hive-mind-init.md +18 -18
- package/.claude/commands/hive-mind/hive-mind-memory.md +8 -8
- package/.claude/commands/hive-mind/hive-mind-metrics.md +8 -8
- package/.claude/commands/hive-mind/hive-mind-resume.md +8 -8
- package/.claude/commands/hive-mind/hive-mind-sessions.md +8 -8
- package/.claude/commands/hive-mind/hive-mind-spawn.md +21 -21
- package/.claude/commands/hive-mind/hive-mind-status.md +8 -8
- package/.claude/commands/hive-mind/hive-mind-stop.md +8 -8
- package/.claude/commands/hive-mind/hive-mind-wizard.md +8 -8
- package/.claude/commands/hive-mind/hive-mind.md +27 -27
- package/.claude/commands/hooks/README.md +11 -11
- package/.claude/commands/hooks/overview.md +57 -57
- package/.claude/commands/hooks/post-edit.md +117 -117
- package/.claude/commands/hooks/post-task.md +112 -112
- package/.claude/commands/hooks/pre-edit.md +113 -113
- package/.claude/commands/hooks/pre-task.md +111 -111
- package/.claude/commands/hooks/session-end.md +118 -118
- package/.claude/commands/hooks/setup.md +102 -102
- package/.claude/commands/memory/README.md +9 -9
- package/.claude/commands/memory/memory-persist.md +25 -25
- package/.claude/commands/memory/memory-search.md +25 -25
- package/.claude/commands/memory/memory-usage.md +25 -25
- package/.claude/commands/memory/neural.md +47 -47
- package/.claude/commands/monitoring/README.md +9 -9
- package/.claude/commands/monitoring/agent-metrics.md +25 -25
- package/.claude/commands/monitoring/agents.md +44 -44
- package/.claude/commands/monitoring/real-time-view.md +25 -25
- package/.claude/commands/monitoring/status.md +46 -46
- package/.claude/commands/monitoring/swarm-monitor.md +25 -25
- package/.claude/commands/optimization/README.md +9 -9
- package/.claude/commands/optimization/auto-topology.md +61 -61
- package/.claude/commands/optimization/cache-manage.md +25 -25
- package/.claude/commands/optimization/parallel-execute.md +25 -25
- package/.claude/commands/optimization/parallel-execution.md +49 -49
- package/.claude/commands/optimization/topology-optimize.md +25 -25
- package/.claude/commands/pair/README.md +260 -260
- package/.claude/commands/pair/commands.md +545 -545
- package/.claude/commands/pair/config.md +509 -509
- package/.claude/commands/pair/examples.md +511 -511
- package/.claude/commands/pair/modes.md +347 -347
- package/.claude/commands/pair/session.md +406 -406
- package/.claude/commands/pair/start.md +208 -208
- package/.claude/commands/sparc/analyzer.md +51 -51
- package/.claude/commands/sparc/architect.md +53 -53
- package/.claude/commands/sparc/ask.md +97 -97
- package/.claude/commands/sparc/batch-executor.md +54 -54
- package/.claude/commands/sparc/code.md +89 -89
- package/.claude/commands/sparc/coder.md +54 -54
- package/.claude/commands/sparc/debug.md +83 -83
- package/.claude/commands/sparc/debugger.md +54 -54
- package/.claude/commands/sparc/designer.md +53 -53
- package/.claude/commands/sparc/devops.md +109 -109
- package/.claude/commands/sparc/docs-writer.md +80 -80
- package/.claude/commands/sparc/documenter.md +54 -54
- package/.claude/commands/sparc/innovator.md +54 -54
- package/.claude/commands/sparc/integration.md +83 -83
- package/.claude/commands/sparc/mcp.md +117 -117
- package/.claude/commands/sparc/memory-manager.md +54 -54
- package/.claude/commands/sparc/optimizer.md +54 -54
- package/.claude/commands/sparc/orchestrator.md +131 -131
- package/.claude/commands/sparc/post-deployment-monitoring-mode.md +83 -83
- package/.claude/commands/sparc/refinement-optimization-mode.md +83 -83
- package/.claude/commands/sparc/researcher.md +54 -54
- package/.claude/commands/sparc/reviewer.md +54 -54
- package/.claude/commands/sparc/security-review.md +80 -80
- package/.claude/commands/sparc/sparc-modes.md +174 -174
- package/.claude/commands/sparc/sparc.md +111 -111
- package/.claude/commands/sparc/spec-pseudocode.md +80 -80
- package/.claude/commands/sparc/supabase-admin.md +348 -348
- package/.claude/commands/sparc/swarm-coordinator.md +54 -54
- package/.claude/commands/sparc/tdd.md +54 -54
- package/.claude/commands/sparc/tester.md +54 -54
- package/.claude/commands/sparc/tutorial.md +79 -79
- package/.claude/commands/sparc/workflow-manager.md +54 -54
- package/.claude/commands/sparc.md +166 -166
- package/.claude/commands/stream-chain/pipeline.md +120 -120
- package/.claude/commands/stream-chain/run.md +69 -69
- package/.claude/commands/swarm/README.md +15 -15
- package/.claude/commands/swarm/analysis.md +95 -95
- package/.claude/commands/swarm/development.md +96 -96
- package/.claude/commands/swarm/examples.md +168 -168
- package/.claude/commands/swarm/maintenance.md +102 -102
- package/.claude/commands/swarm/optimization.md +117 -117
- package/.claude/commands/swarm/research.md +136 -136
- package/.claude/commands/swarm/swarm-analysis.md +8 -8
- package/.claude/commands/swarm/swarm-background.md +8 -8
- package/.claude/commands/swarm/swarm-init.md +19 -19
- package/.claude/commands/swarm/swarm-modes.md +8 -8
- package/.claude/commands/swarm/swarm-monitor.md +8 -8
- package/.claude/commands/swarm/swarm-spawn.md +19 -19
- package/.claude/commands/swarm/swarm-status.md +8 -8
- package/.claude/commands/swarm/swarm-strategies.md +8 -8
- package/.claude/commands/swarm/swarm.md +87 -87
- package/.claude/commands/swarm/testing.md +131 -131
- package/.claude/commands/training/README.md +9 -9
- package/.claude/commands/training/model-update.md +25 -25
- package/.claude/commands/training/neural-patterns.md +107 -107
- package/.claude/commands/training/neural-train.md +75 -75
- package/.claude/commands/training/pattern-learn.md +25 -25
- package/.claude/commands/training/specialization.md +62 -62
- package/.claude/commands/truth/start.md +142 -142
- package/.claude/commands/verify/check.md +49 -49
- package/.claude/commands/verify/start.md +127 -127
- package/.claude/commands/workflows/README.md +9 -9
- package/.claude/commands/workflows/development.md +77 -77
- package/.claude/commands/workflows/research.md +62 -62
- package/.claude/commands/workflows/workflow-create.md +25 -25
- package/.claude/commands/workflows/workflow-execute.md +25 -25
- package/.claude/commands/workflows/workflow-export.md +25 -25
- package/.claude/helpers/README.md +96 -96
- package/.claude/helpers/adr-compliance.sh +186 -186
- package/.claude/helpers/auto-commit.sh +178 -178
- package/.claude/helpers/auto-memory-hook.mjs +368 -368
- package/.claude/helpers/checkpoint-manager.sh +251 -251
- package/.claude/helpers/daemon-manager.sh +252 -252
- package/.claude/helpers/ddd-tracker.sh +144 -144
- package/.claude/helpers/github-safe.js +156 -156
- package/.claude/helpers/github-setup.sh +45 -45
- package/.claude/helpers/guidance-hook.sh +13 -13
- package/.claude/helpers/guidance-hooks.sh +102 -102
- package/.claude/helpers/health-monitor.sh +108 -108
- package/.claude/helpers/hook-handler.cjs +286 -286
- package/.claude/helpers/intelligence.cjs +1031 -1031
- package/.claude/helpers/learning-hooks.sh +329 -329
- package/.claude/helpers/learning-optimizer.sh +127 -127
- package/.claude/helpers/learning-service.mjs +1144 -1144
- package/.claude/helpers/memory.js +83 -83
- package/.claude/helpers/metrics-db.mjs +488 -488
- package/.claude/helpers/pattern-consolidator.sh +86 -86
- package/.claude/helpers/perf-worker.sh +160 -160
- package/.claude/helpers/post-commit +16 -16
- package/.claude/helpers/pre-commit +26 -26
- package/.claude/helpers/quick-start.sh +19 -19
- package/.claude/helpers/router.js +105 -105
- package/.claude/helpers/security-scanner.sh +127 -127
- package/.claude/helpers/session.js +157 -157
- package/.claude/helpers/setup-mcp.sh +18 -18
- package/.claude/helpers/standard-checkpoint-hooks.sh +189 -189
- package/.claude/helpers/statusline-hook.sh +21 -21
- package/.claude/helpers/statusline.cjs +604 -604
- package/.claude/helpers/statusline.js +352 -352
- package/.claude/helpers/swarm-comms.sh +353 -353
- package/.claude/helpers/swarm-hooks.sh +761 -761
- package/.claude/helpers/swarm-monitor.sh +210 -210
- package/.claude/helpers/sync-v3-metrics.sh +245 -245
- package/.claude/helpers/update-v3-progress.sh +165 -165
- package/.claude/helpers/v3-quick-status.sh +57 -57
- package/.claude/helpers/v3.sh +110 -110
- package/.claude/helpers/validate-v3-config.sh +215 -215
- package/.claude/helpers/worker-manager.sh +170 -170
- package/.claude/settings.json +182 -182
- package/.claude/skills/agentdb-advanced/SKILL.md +550 -550
- package/.claude/skills/agentdb-learning/SKILL.md +545 -545
- package/.claude/skills/agentdb-memory-patterns/SKILL.md +339 -339
- package/.claude/skills/agentdb-optimization/SKILL.md +509 -509
- package/.claude/skills/agentdb-vector-search/SKILL.md +339 -339
- package/.claude/skills/browser/SKILL.md +204 -204
- package/.claude/skills/dual-mode/README.md +71 -71
- package/.claude/skills/dual-mode/dual-collect.md +103 -103
- package/.claude/skills/dual-mode/dual-coordinate.md +85 -85
- package/.claude/skills/dual-mode/dual-spawn.md +81 -81
- package/.claude/skills/flow-nexus-neural/SKILL.md +727 -727
- package/.claude/skills/flow-nexus-platform/SKILL.md +1154 -1154
- package/.claude/skills/flow-nexus-swarm/SKILL.md +604 -604
- package/.claude/skills/github-code-review/SKILL.md +1125 -1125
- package/.claude/skills/github-multi-repo/SKILL.md +862 -862
- package/.claude/skills/github-project-management/SKILL.md +1262 -1262
- package/.claude/skills/github-release-management/SKILL.md +1064 -1064
- package/.claude/skills/github-workflow-automation/SKILL.md +1047 -1047
- package/.claude/skills/hooks-automation/SKILL.md +1201 -1201
- package/.claude/skills/pair-programming/SKILL.md +1202 -1202
- package/.claude/skills/reasoningbank-agentdb/SKILL.md +446 -446
- package/.claude/skills/reasoningbank-intelligence/SKILL.md +201 -201
- package/.claude/skills/skill-builder/SKILL.md +910 -910
- package/.claude/skills/sparc-methodology/SKILL.md +1106 -1106
- package/.claude/skills/stream-chain/SKILL.md +560 -560
- package/.claude/skills/swarm-advanced/SKILL.md +970 -970
- package/.claude/skills/swarm-orchestration/SKILL.md +179 -179
- package/.claude/skills/v3-cli-modernization/SKILL.md +871 -871
- package/.claude/skills/v3-core-implementation/SKILL.md +796 -796
- package/.claude/skills/v3-ddd-architecture/SKILL.md +441 -441
- package/.claude/skills/v3-integration-deep/SKILL.md +240 -240
- package/.claude/skills/v3-mcp-optimization/SKILL.md +776 -776
- package/.claude/skills/v3-memory-unification/SKILL.md +173 -173
- package/.claude/skills/v3-performance-optimization/SKILL.md +389 -389
- package/.claude/skills/v3-security-overhaul/SKILL.md +81 -81
- package/.claude/skills/v3-swarm-coordination/SKILL.md +339 -339
- package/.claude/skills/verification-quality/SKILL.md +691 -691
- package/README.md +412 -412
- package/bin/cli.js +314 -314
- package/bin/mcp-server.js +224 -224
- package/bin/preinstall.cjs +2 -2
- package/dist/src/benchmarks/gaia-critic.js +24 -24
- package/dist/src/commands/completions.js +409 -409
- package/dist/src/commands/daemon.js +44 -44
- package/dist/src/commands/doctor.d.ts.map +1 -1
- package/dist/src/commands/doctor.js +175 -1
- package/dist/src/commands/doctor.js.map +1 -1
- package/dist/src/commands/eject.d.ts +33 -0
- package/dist/src/commands/eject.d.ts.map +1 -0
- package/dist/src/commands/eject.js +195 -0
- package/dist/src/commands/eject.js.map +1 -0
- package/dist/src/commands/embeddings.js +26 -26
- package/dist/src/commands/hive-mind.js +97 -97
- package/dist/src/commands/hooks.d.ts.map +1 -1
- package/dist/src/commands/hooks.js +11 -10
- package/dist/src/commands/hooks.js.map +1 -1
- package/dist/src/commands/index.d.ts.map +1 -1
- package/dist/src/commands/index.js +4 -0
- package/dist/src/commands/index.js.map +1 -1
- package/dist/src/commands/init.d.ts.map +1 -1
- package/dist/src/commands/init.js +3 -0
- package/dist/src/commands/init.js.map +1 -1
- package/dist/src/commands/metaharness.d.ts +39 -0
- package/dist/src/commands/metaharness.d.ts.map +1 -0
- package/dist/src/commands/metaharness.js +194 -0
- package/dist/src/commands/metaharness.js.map +1 -0
- package/dist/src/commands/neural.d.ts.map +1 -1
- package/dist/src/commands/neural.js +2351 -1
- package/dist/src/commands/neural.js.map +1 -1
- package/dist/src/commands/ruvector/backup.js +23 -23
- package/dist/src/commands/ruvector/benchmark.js +31 -31
- package/dist/src/commands/ruvector/import.js +14 -14
- package/dist/src/commands/ruvector/init.js +115 -115
- package/dist/src/commands/ruvector/migrate.js +99 -99
- package/dist/src/commands/ruvector/optimize.js +51 -51
- package/dist/src/commands/ruvector/setup.js +624 -624
- package/dist/src/commands/ruvector/status.js +38 -38
- package/dist/src/init/claudemd-generator.js +231 -231
- package/dist/src/init/executor.js +453 -453
- package/dist/src/init/helpers-generator.js +727 -727
- package/dist/src/init/statusline-generator.js +717 -717
- package/dist/src/mcp-client.d.ts.map +1 -1
- package/dist/src/mcp-client.js +4 -0
- package/dist/src/mcp-client.js.map +1 -1
- package/dist/src/mcp-tools/agent-execute-core.d.ts +23 -1
- package/dist/src/mcp-tools/agent-execute-core.d.ts.map +1 -1
- package/dist/src/mcp-tools/agent-execute-core.js +136 -9
- package/dist/src/mcp-tools/agent-execute-core.js.map +1 -1
- package/dist/src/mcp-tools/agent-tools.d.ts.map +1 -1
- package/dist/src/mcp-tools/agent-tools.js +48 -4
- package/dist/src/mcp-tools/agent-tools.js.map +1 -1
- package/dist/src/mcp-tools/agentdb-tools.js +15 -15
- package/dist/src/mcp-tools/hooks-tools.d.ts.map +1 -1
- package/dist/src/mcp-tools/hooks-tools.js +196 -0
- package/dist/src/mcp-tools/hooks-tools.js.map +1 -1
- package/dist/src/mcp-tools/index.d.ts +1 -0
- package/dist/src/mcp-tools/index.d.ts.map +1 -1
- package/dist/src/mcp-tools/index.js +2 -0
- package/dist/src/mcp-tools/index.js.map +1 -1
- package/dist/src/mcp-tools/metaharness-tools.d.ts +31 -0
- package/dist/src/mcp-tools/metaharness-tools.d.ts.map +1 -0
- package/dist/src/mcp-tools/metaharness-tools.js +372 -0
- package/dist/src/mcp-tools/metaharness-tools.js.map +1 -0
- package/dist/src/memory/graph-edge-writer.js +18 -18
- package/dist/src/memory/memory-bridge.js +84 -84
- package/dist/src/memory/memory-initializer.js +393 -393
- package/dist/src/memory/rabitq-index.js +5 -5
- package/dist/src/plugins/store/discovery.d.ts.map +1 -1
- package/dist/src/plugins/store/discovery.js +3 -0
- package/dist/src/plugins/store/discovery.js.map +1 -1
- package/dist/src/plugins/store/types.d.ts +6 -1
- package/dist/src/plugins/store/types.d.ts.map +1 -1
- package/dist/src/runtime/headless.js +28 -28
- package/dist/src/ruvector/enhanced-model-router.d.ts +5 -0
- package/dist/src/ruvector/enhanced-model-router.d.ts.map +1 -1
- package/dist/src/ruvector/enhanced-model-router.js +18 -1
- package/dist/src/ruvector/enhanced-model-router.js.map +1 -1
- package/dist/src/ruvector/model-prices.d.ts +50 -0
- package/dist/src/ruvector/model-prices.d.ts.map +1 -0
- package/dist/src/ruvector/model-prices.js +72 -0
- package/dist/src/ruvector/model-prices.js.map +1 -0
- package/dist/src/ruvector/model-router.d.ts +147 -8
- package/dist/src/ruvector/model-router.d.ts.map +1 -1
- package/dist/src/ruvector/model-router.js +455 -18
- package/dist/src/ruvector/model-router.js.map +1 -1
- package/dist/src/ruvector/neural-router.d.ts +172 -39
- package/dist/src/ruvector/neural-router.d.ts.map +1 -1
- package/dist/src/ruvector/neural-router.js +780 -91
- package/dist/src/ruvector/neural-router.js.map +1 -1
- package/dist/src/ruvector/router-calibrator.d.ts +65 -0
- package/dist/src/ruvector/router-calibrator.d.ts.map +1 -0
- package/dist/src/ruvector/router-calibrator.js +120 -0
- package/dist/src/ruvector/router-calibrator.js.map +1 -0
- package/dist/src/ruvector/router-parallel-recorder.d.ts +127 -0
- package/dist/src/ruvector/router-parallel-recorder.d.ts.map +1 -0
- package/dist/src/ruvector/router-parallel-recorder.js +183 -0
- package/dist/src/ruvector/router-parallel-recorder.js.map +1 -0
- package/dist/src/ruvector/router-trajectory.d.ts +137 -45
- package/dist/src/ruvector/router-trajectory.d.ts.map +1 -1
- package/dist/src/ruvector/router-trajectory.js +226 -63
- package/dist/src/ruvector/router-trajectory.js.map +1 -1
- package/dist/src/ruvector/task-embedder.d.ts +56 -0
- package/dist/src/ruvector/task-embedder.d.ts.map +1 -0
- package/dist/src/ruvector/task-embedder.js +237 -0
- package/dist/src/ruvector/task-embedder.js.map +1 -0
- package/dist/src/services/headless-worker-executor.js +84 -84
- package/dist/src/transfer/deploy-seraphine.js +23 -23
- package/dist/tsconfig.tsbuildinfo +1 -1
- package/package.json +138 -134
- package/scripts/beir-bootstrap-significance.mjs +162 -162
- package/scripts/bench-rvagent.mjs +312 -312
- package/scripts/benchmark-codemods.mjs +127 -127
- package/scripts/benchmark-cross-repo.mjs +206 -206
- package/scripts/benchmark-pretrained-retrieval.mjs +305 -305
- package/scripts/benchmark-self-learning.mjs +241 -241
- package/scripts/benchmark-trajectory-mrr.mjs +191 -191
- package/scripts/deploy-ipfs-node.sh +153 -153
- package/scripts/grid-search-retrieval.mjs +216 -216
- package/scripts/postinstall.cjs +153 -153
- package/scripts/pretrain-from-github.mjs +358 -358
- package/scripts/publish-registry.ts +345 -345
- package/scripts/publish.sh +57 -57
- package/scripts/run-beir-bge.mjs +324 -324
- package/scripts/run-beir-hybrid.mjs +261 -261
- package/scripts/run-beir-lucene-bm25.mjs +154 -154
- package/scripts/run-beir-nfcorpus.mjs +231 -231
- package/scripts/run-beir-rrf-ablation.mjs +221 -221
- package/scripts/setup-ipfs-registry.md +366 -366
- package/dist/src/ruvector/flash-attention.d.ts +0 -195
- package/dist/src/ruvector/flash-attention.d.ts.map +0 -1
- package/dist/src/ruvector/flash-attention.js +0 -643
- package/dist/src/ruvector/flash-attention.js.map +0 -1
- package/dist/src/ruvector/moe-router.d.ts +0 -206
- package/dist/src/ruvector/moe-router.d.ts.map +0 -1
- package/dist/src/ruvector/moe-router.js +0 -626
- package/dist/src/ruvector/moe-router.js.map +0 -1
- package/dist/src/services/event-stream.d.ts +0 -25
- package/dist/src/services/event-stream.js +0 -27
- package/dist/src/services/loop-worker-runner.d.ts +0 -16
- package/dist/src/services/loop-worker-runner.js +0 -34
- package/dist/src/services/runtime-capabilities.d.ts +0 -22
- package/dist/src/services/runtime-capabilities.js +0 -45
|
@@ -1,261 +1,261 @@
|
|
|
1
|
-
#!/usr/bin/env node
|
|
2
|
-
// run-beir-hybrid.mjs — BM25 + dense (BGE) + RRF fusion + optional
|
|
3
|
-
// cross-encoder rerank, evaluated on BEIR (ADR-087).
|
|
4
|
-
//
|
|
5
|
-
// Pipeline:
|
|
6
|
-
// 1. BM25 over corpus (multi-field BM25 from hybrid-retrieval.ts)
|
|
7
|
-
// 2. Dense cosine over BGE-base embeddings (cached from run-beir-bge.mjs)
|
|
8
|
-
// 3. RRF fusion: score = sum over systems of 1/(k + rank) (k=60)
|
|
9
|
-
// 4. Optional cross-encoder rerank of fused top-100 → top-10
|
|
10
|
-
//
|
|
11
|
-
// Reuses the .f32 doc-embedding cache from run-beir-bge.mjs — must run
|
|
12
|
-
// that first to populate the cache.
|
|
13
|
-
//
|
|
14
|
-
// Usage:
|
|
15
|
-
// cd /tmp/beir-nfcorpus
|
|
16
|
-
// node /path/to/scripts/run-beir-hybrid.mjs # RRF only
|
|
17
|
-
// RERANK=1 node /path/to/scripts/run-beir-hybrid.mjs # + cross-encoder rerank
|
|
18
|
-
// RRF_K=60 node /path/to/scripts/run-beir-hybrid.mjs # tune RRF k
|
|
19
|
-
//
|
|
20
|
-
// ADR-087.
|
|
21
|
-
|
|
22
|
-
import { readFileSync, writeFileSync, existsSync, mkdirSync } from 'node:fs';
|
|
23
|
-
import { fileURLToPath } from 'node:url';
|
|
24
|
-
import { dirname, join, resolve } from 'node:path';
|
|
25
|
-
import { performance } from 'node:perf_hooks';
|
|
26
|
-
|
|
27
|
-
const SCRIPT_DIR = dirname(fileURLToPath(import.meta.url));
|
|
28
|
-
const CLI_ROOT = resolve(SCRIPT_DIR, '..');
|
|
29
|
-
const RUFLO_ROOT = resolve(SCRIPT_DIR, '../../../..');
|
|
30
|
-
const RUNS_DIR = join(RUFLO_ROOT, 'docs', 'benchmarks', 'runs');
|
|
31
|
-
|
|
32
|
-
const DATA_DIR = process.env.BEIR_DATA_DIR || '/tmp/beir-nfcorpus/nfcorpus';
|
|
33
|
-
const BGE_MODEL = process.env.BGE_MODEL || 'Xenova/bge-base-en-v1.5';
|
|
34
|
-
const CACHE_DIR = join(process.env.CACHE_BASE_DIR || dirname(DATA_DIR), 'bge-cache');
|
|
35
|
-
const RRF_K = Number(process.env.RRF_K) || 60;
|
|
36
|
-
const RERANK = process.env.RERANK === '1';
|
|
37
|
-
const RERANK_TOP_K = Number(process.env.RERANK_TOP_K) || 100;
|
|
38
|
-
const MAX_QUERIES = Number(process.env.MAX_QUERIES) || 0;
|
|
39
|
-
|
|
40
|
-
// Same baselines table as run-beir-bge.mjs.
|
|
41
|
-
const BASELINES_BY_DATASET = {
|
|
42
|
-
nfcorpus: { 'BM25 (Lucene)': 0.325, 'DocT5query': 0.328, 'TAS-B': 0.319, 'GenQ': 0.319, 'ColBERT': 0.305, 'Contriever': 0.328, 'GTR-XL': 0.343, 'SPLADE++': 0.347, 'BGE-large-v1.5 (pub)': 0.380, 'SBERT msmarco': 0.272 },
|
|
43
|
-
scifact: { 'BM25 (Lucene)': 0.679, 'DocT5query': 0.675, 'TAS-B': 0.643, 'GenQ': 0.644, 'ColBERT': 0.671, 'Contriever': 0.677, 'GTR-XL': 0.662, 'SPLADE++': 0.704, 'BGE-large-v1.5 (pub)': 0.722, 'SBERT msmarco': 0.555 },
|
|
44
|
-
arguana: { 'BM25 (Lucene)': 0.397, 'DocT5query': 0.349, 'TAS-B': 0.429, 'GenQ': 0.493, 'ColBERT': 0.233, 'Contriever': 0.379, 'GTR-XL': 0.439, 'SPLADE++': 0.521, 'BGE-large-v1.5 (pub)': 0.636, 'SBERT msmarco': 0.371 },
|
|
45
|
-
scidocs: { 'BM25 (Lucene)': 0.158, 'DocT5query': 0.162, 'TAS-B': 0.149, 'GenQ': 0.143, 'ColBERT': 0.145, 'Contriever': 0.165, 'GTR-XL': 0.174, 'SPLADE++': 0.159, 'BGE-large-v1.5 (pub)': 0.225, 'SBERT msmarco': 0.122 },
|
|
46
|
-
};
|
|
47
|
-
function detectDataset(path) {
|
|
48
|
-
const p = path.toLowerCase();
|
|
49
|
-
for (const ds of Object.keys(BASELINES_BY_DATASET)) if (p.includes(ds)) return ds;
|
|
50
|
-
return 'nfcorpus';
|
|
51
|
-
}
|
|
52
|
-
|
|
53
|
-
function dcg(rels, k) { let s = 0; for (let i = 0; i < Math.min(rels.length, k); i++) s += (Math.pow(2, rels[i]) - 1) / Math.log2(i + 2); return s; }
|
|
54
|
-
function ndcg(retrieved, qrels, k) { const rels = retrieved.slice(0, k).map((id) => qrels.get(id) ?? 0); const ideal = [...qrels.values()].sort((a, b) => b - a).slice(0, k); const idcg = dcg(ideal, k); return idcg > 0 ? dcg(rels, k) / idcg : 0; }
|
|
55
|
-
function mrr(retrieved, qrels, k) { for (let i = 0; i < Math.min(retrieved.length, k); i++) if ((qrels.get(retrieved[i]) ?? 0) > 0) return 1 / (i + 1); return 0; }
|
|
56
|
-
function recall(retrieved, qrels, k) { const tot = [...qrels.values()].filter((v) => v > 0).length; if (tot === 0) return 0; let h = 0; for (let i = 0; i < Math.min(retrieved.length, k); i++) if ((qrels.get(retrieved[i]) ?? 0) > 0) h++; return h / tot; }
|
|
57
|
-
function cosine(a, b) { let s = 0; for (let i = 0; i < a.length; i++) s += a[i] * b[i]; return s; }
|
|
58
|
-
|
|
59
|
-
function loadJsonl(path) { return readFileSync(path, 'utf-8').split('\n').filter(Boolean).map((l) => JSON.parse(l)); }
|
|
60
|
-
function loadQrels(path) { const q = new Map(); const lines = readFileSync(path, 'utf-8').split('\n'); for (let i = 1; i < lines.length; i++) { if (!lines[i].trim()) continue; const [qid, did, score] = lines[i].split('\t'); if (!q.has(qid)) q.set(qid, new Map()); q.get(qid).set(did, Number(score)); } return q; }
|
|
61
|
-
|
|
62
|
-
function cachePath(suffix) { const safe = BGE_MODEL.replace(/\//g, '_'); return join(CACHE_DIR, `${safe}.${suffix}`); }
|
|
63
|
-
function loadEmbeddings(dim) {
|
|
64
|
-
if (!existsSync(cachePath('ids')) || !existsSync(cachePath('f32'))) return null;
|
|
65
|
-
const ids = readFileSync(cachePath('ids'), 'utf-8').split('\n').filter(Boolean);
|
|
66
|
-
const raw = readFileSync(cachePath('f32'));
|
|
67
|
-
const buf = new Float32Array(raw.buffer, raw.byteOffset, raw.length / 4);
|
|
68
|
-
const embeds = [];
|
|
69
|
-
for (let i = 0; i < ids.length; i++) embeds.push(buf.slice(i * dim, (i + 1) * dim));
|
|
70
|
-
return { ids, embeds };
|
|
71
|
-
}
|
|
72
|
-
|
|
73
|
-
async function main() {
|
|
74
|
-
const dataset = detectDataset(DATA_DIR);
|
|
75
|
-
const BASELINES_NDCG10 = BASELINES_BY_DATASET[dataset];
|
|
76
|
-
console.log(`# BEIR ${dataset} — hybrid RRF${RERANK ? ' + cross-encoder rerank' : ''} (ADR-087)`);
|
|
77
|
-
console.log(`Data: ${DATA_DIR}`);
|
|
78
|
-
console.log(`Dense: ${BGE_MODEL}`);
|
|
79
|
-
console.log(`RRF k: ${RRF_K}${RERANK ? `, rerank top-${RERANK_TOP_K}` : ''}`);
|
|
80
|
-
|
|
81
|
-
const corpus = loadJsonl(join(DATA_DIR, 'corpus.jsonl'));
|
|
82
|
-
const queries = loadJsonl(join(DATA_DIR, 'queries.jsonl'));
|
|
83
|
-
const qrels = loadQrels(join(DATA_DIR, 'qrels/test.tsv'));
|
|
84
|
-
console.log(`Corpus: ${corpus.length} docs · Test qrels: ${qrels.size}`);
|
|
85
|
-
|
|
86
|
-
// BGE embeddings — must be pre-cached from run-beir-bge.mjs.
|
|
87
|
-
const bge = await import(join(CLI_ROOT, 'dist/src/memory/bge-embedder.js'));
|
|
88
|
-
const emb = await bge.getBgeEmbedder(BGE_MODEL);
|
|
89
|
-
if (!emb) { console.error('BGE failed:', bge.getBgeStatus()); process.exit(1); }
|
|
90
|
-
const dim = emb.dim();
|
|
91
|
-
const cached = loadEmbeddings(dim);
|
|
92
|
-
if (!cached) { console.error(`No cached BGE embeddings at ${cachePath('f32')}. Run run-beir-bge.mjs first.`); process.exit(2); }
|
|
93
|
-
const docIds = cached.ids;
|
|
94
|
-
const docEmbeds = cached.embeds;
|
|
95
|
-
console.log(`Loaded ${docIds.length} cached BGE embeddings`);
|
|
96
|
-
|
|
97
|
-
// BM25 setup — USE_LUCENE_BM25=1 uses Porter+Lucene-stopword BM25 (matches
|
|
98
|
-
// published BEIR baselines); default is hybrid-retrieval's multi-field BM25.
|
|
99
|
-
const USE_LUCENE_BM25 = process.env.USE_LUCENE_BM25 === '1';
|
|
100
|
-
let bm25Score, tokenizeFn, titleDocs, textDocs, titleStats, textStats, luceneDocs, luceneStats;
|
|
101
|
-
if (USE_LUCENE_BM25) {
|
|
102
|
-
const { luceneTokenize, buildLuceneCorpusStats, luceneBM25 } = await import(join(CLI_ROOT, 'dist/src/memory/lucene-bm25.js'));
|
|
103
|
-
tokenizeFn = luceneTokenize;
|
|
104
|
-
luceneDocs = corpus.map((d) => luceneTokenize(`${d.title || ''} ${d.text || ''}`));
|
|
105
|
-
luceneStats = buildLuceneCorpusStats(luceneDocs);
|
|
106
|
-
bm25Score = (qTokens, idx) => luceneBM25(qTokens, luceneDocs[idx], luceneStats);
|
|
107
|
-
console.log(`BM25: Lucene-style (Porter stem + Lucene stopwords + length norm) — ADR-088`);
|
|
108
|
-
} else {
|
|
109
|
-
const { tokenize, buildCorpusStats, multiFieldBM25 } = await import(join(CLI_ROOT, 'dist/src/memory/hybrid-retrieval.js'));
|
|
110
|
-
tokenizeFn = tokenize;
|
|
111
|
-
titleDocs = corpus.map((d) => tokenize(d.title || ''));
|
|
112
|
-
textDocs = corpus.map((d) => tokenize(d.text || ''));
|
|
113
|
-
titleStats = buildCorpusStats(titleDocs);
|
|
114
|
-
textStats = buildCorpusStats(textDocs);
|
|
115
|
-
const SUBJECT_WEIGHT = Number(process.env.SUBJECT_WEIGHT ?? 1.0);
|
|
116
|
-
const BODY_WEIGHT = Number(process.env.BODY_WEIGHT ?? 1.0);
|
|
117
|
-
bm25Score = (qTokens, idx) => multiFieldBM25(qTokens, titleDocs[idx], textDocs[idx], titleStats, textStats, SUBJECT_WEIGHT, BODY_WEIGHT);
|
|
118
|
-
console.log(`BM25: multi-field (title sw=${SUBJECT_WEIGHT}, text bw=${BODY_WEIGHT})`);
|
|
119
|
-
}
|
|
120
|
-
|
|
121
|
-
// Optional cross-encoder reranker.
|
|
122
|
-
let crossEncoder = null;
|
|
123
|
-
if (RERANK) {
|
|
124
|
-
const ce = await import(join(CLI_ROOT, 'dist/src/memory/cross-encoder-rerank.js'));
|
|
125
|
-
const ceFn = await ce.getCrossEncoder('Xenova/ms-marco-MiniLM-L-6-v2');
|
|
126
|
-
if (!ceFn) { console.error('Cross-encoder failed to load:', ce.getCrossEncoderStatus()); process.exit(3); }
|
|
127
|
-
crossEncoder = ceFn;
|
|
128
|
-
console.log('Cross-encoder loaded: Xenova/ms-marco-MiniLM-L-6-v2');
|
|
129
|
-
}
|
|
130
|
-
|
|
131
|
-
const queriesById = new Map(queries.map((q) => [q._id, q.text]));
|
|
132
|
-
const evalQueryIds = MAX_QUERIES ? [...qrels.keys()].slice(0, MAX_QUERIES) : [...qrels.keys()];
|
|
133
|
-
|
|
134
|
-
// Build doc id → corpus index lookup (BM25 returns indices, dense returns ids).
|
|
135
|
-
const idToIdx = new Map(docIds.map((id, i) => [id, i]));
|
|
136
|
-
// Need a reverse map too: corpus index → doc id (for BM25 over corpus order).
|
|
137
|
-
const corpusIdToIdx = new Map(corpus.map((d, i) => [d._id, i]));
|
|
138
|
-
const corpusIdxToId = corpus.map((d) => d._id);
|
|
139
|
-
|
|
140
|
-
console.log(`\nRunning ${evalQueryIds.length} queries...`);
|
|
141
|
-
let nSum = 0, mSum = 0, r10Sum = 0, r100Sum = 0, n = 0;
|
|
142
|
-
const perQuery = [];
|
|
143
|
-
const tQ = performance.now();
|
|
144
|
-
|
|
145
|
-
for (const qid of evalQueryIds) {
|
|
146
|
-
const qtext = queriesById.get(qid);
|
|
147
|
-
if (!qtext) continue;
|
|
148
|
-
|
|
149
|
-
// §1 — dense BGE retrieval (ADR-090: opt-in query prefix when BGE_QUERY_PREFIX=1)
|
|
150
|
-
const qEmb = (process.env.BGE_QUERY_PREFIX === '1' && emb.embedQuery)
|
|
151
|
-
? await emb.embedQuery(qtext)
|
|
152
|
-
: await emb.embed(qtext);
|
|
153
|
-
const denseScored = new Array(docEmbeds.length);
|
|
154
|
-
for (let i = 0; i < docEmbeds.length; i++) denseScored[i] = { id: docIds[i], score: cosine(qEmb, docEmbeds[i]) };
|
|
155
|
-
denseScored.sort((a, b) => b.score - a.score);
|
|
156
|
-
|
|
157
|
-
// §2 — BM25 retrieval (Lucene or multi-field depending on USE_LUCENE_BM25)
|
|
158
|
-
const qTokens = tokenizeFn(qtext);
|
|
159
|
-
const bm25Scored = new Array(corpus.length);
|
|
160
|
-
for (let i = 0; i < corpus.length; i++) {
|
|
161
|
-
bm25Scored[i] = { id: corpusIdxToId[i], score: bm25Score(qTokens, i) };
|
|
162
|
-
}
|
|
163
|
-
bm25Scored.sort((a, b) => b.score - a.score);
|
|
164
|
-
|
|
165
|
-
// §3 — RRF fusion: score = sum over systems of 1/(k + rank).
|
|
166
|
-
// Only need top ~200 from each for stable RRF; cap at top-500 per system.
|
|
167
|
-
const TOP_PER_SYSTEM = 500;
|
|
168
|
-
const rrfScores = new Map();
|
|
169
|
-
for (let r = 0; r < Math.min(TOP_PER_SYSTEM, denseScored.length); r++) {
|
|
170
|
-
const id = denseScored[r].id;
|
|
171
|
-
rrfScores.set(id, (rrfScores.get(id) || 0) + 1 / (RRF_K + r + 1));
|
|
172
|
-
}
|
|
173
|
-
for (let r = 0; r < Math.min(TOP_PER_SYSTEM, bm25Scored.length); r++) {
|
|
174
|
-
const id = bm25Scored[r].id;
|
|
175
|
-
// Skip if this doc has 0 BM25 score (no token overlap) — true random docs would otherwise leak in
|
|
176
|
-
if (bm25Scored[r].score === 0) break;
|
|
177
|
-
rrfScores.set(id, (rrfScores.get(id) || 0) + 1 / (RRF_K + r + 1));
|
|
178
|
-
}
|
|
179
|
-
const fused = [...rrfScores.entries()]
|
|
180
|
-
.map(([id, score]) => ({ id, score }))
|
|
181
|
-
.sort((a, b) => b.score - a.score)
|
|
182
|
-
.slice(0, RERANK ? RERANK_TOP_K : 100);
|
|
183
|
-
|
|
184
|
-
let final = fused;
|
|
185
|
-
|
|
186
|
-
// §4 — optional cross-encoder rerank.
|
|
187
|
-
if (crossEncoder && fused.length > 0) {
|
|
188
|
-
const docsForRerank = fused.map(({ id }) => {
|
|
189
|
-
const ci = corpusIdToIdx.get(id);
|
|
190
|
-
const d = corpus[ci];
|
|
191
|
-
return `${d.title || ''} ${d.text || ''}`.slice(0, 4096);
|
|
192
|
-
});
|
|
193
|
-
const ceScores = await crossEncoder.scoreBatch(qtext, docsForRerank);
|
|
194
|
-
final = fused
|
|
195
|
-
.map((f, i) => ({ id: f.id, score: ceScores[i], rrf: f.score }))
|
|
196
|
-
.sort((a, b) => b.score - a.score);
|
|
197
|
-
}
|
|
198
|
-
|
|
199
|
-
const top100 = final.slice(0, 100).map((s) => s.id);
|
|
200
|
-
const qmap = qrels.get(qid);
|
|
201
|
-
const qNdcg = ndcg(top100, qmap, 10);
|
|
202
|
-
const qMrr = mrr(top100, qmap, 10);
|
|
203
|
-
const qR10 = recall(top100, qmap, 10);
|
|
204
|
-
const qR100 = recall(top100, qmap, 100);
|
|
205
|
-
nSum += qNdcg; mSum += qMrr; r10Sum += qR10; r100Sum += qR100; n++;
|
|
206
|
-
perQuery.push({ qid, ndcg10: qNdcg, mrr10: qMrr, recall10: qR10, recall100: qR100 });
|
|
207
|
-
if (n % 50 === 0) {
|
|
208
|
-
const elapsed = (performance.now() - tQ) / 1000;
|
|
209
|
-
console.log(` ${n}/${evalQueryIds.length} in ${elapsed.toFixed(0)}s running nDCG@10=${(nSum / n).toFixed(4)}`);
|
|
210
|
-
}
|
|
211
|
-
}
|
|
212
|
-
const queryMs = performance.now() - tQ;
|
|
213
|
-
|
|
214
|
-
const ndcg10 = nSum / n;
|
|
215
|
-
const mrr10 = mSum / n;
|
|
216
|
-
const recall10 = r10Sum / n;
|
|
217
|
-
const recall100 = r100Sum / n;
|
|
218
|
-
|
|
219
|
-
console.log(`\n=== Results (N=${n}, ${dataset}, BM25+BGE-base RRF k=${RRF_K}${RERANK ? ' + CE rerank' : ''}) ===`);
|
|
220
|
-
console.log(` nDCG@10: ${ndcg10.toFixed(4)}`);
|
|
221
|
-
console.log(` MRR@10: ${mrr10.toFixed(4)}`);
|
|
222
|
-
console.log(` Recall@10: ${recall10.toFixed(4)}`);
|
|
223
|
-
console.log(` Recall@100: ${recall100.toFixed(4)}`);
|
|
224
|
-
console.log(` Avg query latency: ${(queryMs / n).toFixed(0)}ms`);
|
|
225
|
-
|
|
226
|
-
const ourLabel = `ruflo + BM25+BGE-base RRF${RERANK ? '+CE' : ''}`;
|
|
227
|
-
const ranking = [
|
|
228
|
-
...Object.entries(BASELINES_NDCG10).map(([name, score]) => ({ name, score, ours: false })),
|
|
229
|
-
{ name: ourLabel, score: ndcg10, ours: true },
|
|
230
|
-
].sort((a, b) => b.score - a.score);
|
|
231
|
-
console.log(`\n=== vs ${dataset} listed baselines ===`);
|
|
232
|
-
for (const r of ranking) console.log(` ${r.score.toFixed(3)} ${r.name}${r.ours ? ' ← us' : ''}`);
|
|
233
|
-
const ourRank = ranking.findIndex((r) => r.ours) + 1;
|
|
234
|
-
console.log(`\n Our rank: ${ourRank} / ${ranking.length}`);
|
|
235
|
-
|
|
236
|
-
const summary = {
|
|
237
|
-
runAt: new Date().toISOString(),
|
|
238
|
-
benchmark: `beir-${dataset}-hybrid${RERANK ? '-rerank' : ''}`,
|
|
239
|
-
dataset,
|
|
240
|
-
pipeline: `BM25+BGE RRF${RERANK ? '+CE-rerank' : ''}`,
|
|
241
|
-
bgeModel: BGE_MODEL,
|
|
242
|
-
rrfK: RRF_K,
|
|
243
|
-
rerank: RERANK,
|
|
244
|
-
queries: n,
|
|
245
|
-
corpusSize: corpus.length,
|
|
246
|
-
metrics: { ndcg10, mrr10, recall10, recall100, avgQueryLatencyMs: queryMs / n },
|
|
247
|
-
perQuery,
|
|
248
|
-
baselines: BASELINES_NDCG10,
|
|
249
|
-
ourRank,
|
|
250
|
-
leaderboardLength: ranking.length,
|
|
251
|
-
};
|
|
252
|
-
mkdirSync(RUNS_DIR, { recursive: true });
|
|
253
|
-
const stamp = summary.runAt.replace(/[:.]/g, '-');
|
|
254
|
-
const suffix = RERANK ? 'hybrid-rerank' : 'hybrid-rrf';
|
|
255
|
-
writeFileSync(join(RUNS_DIR, `beir-${dataset}-${suffix}-${stamp}.json`), JSON.stringify(summary, null, 2));
|
|
256
|
-
writeFileSync(join(RUNS_DIR, `beir-${dataset}-${suffix}-latest.json`), JSON.stringify(summary, null, 2));
|
|
257
|
-
console.log(`\nWrote ${join(RUNS_DIR, `beir-${dataset}-${suffix}-${stamp}.json`)}`);
|
|
258
|
-
process.exit(0);
|
|
259
|
-
}
|
|
260
|
-
|
|
261
|
-
main().catch((err) => { console.error(err); process.exit(1); });
|
|
1
|
+
#!/usr/bin/env node
|
|
2
|
+
// run-beir-hybrid.mjs — BM25 + dense (BGE) + RRF fusion + optional
|
|
3
|
+
// cross-encoder rerank, evaluated on BEIR (ADR-087).
|
|
4
|
+
//
|
|
5
|
+
// Pipeline:
|
|
6
|
+
// 1. BM25 over corpus (multi-field BM25 from hybrid-retrieval.ts)
|
|
7
|
+
// 2. Dense cosine over BGE-base embeddings (cached from run-beir-bge.mjs)
|
|
8
|
+
// 3. RRF fusion: score = sum over systems of 1/(k + rank) (k=60)
|
|
9
|
+
// 4. Optional cross-encoder rerank of fused top-100 → top-10
|
|
10
|
+
//
|
|
11
|
+
// Reuses the .f32 doc-embedding cache from run-beir-bge.mjs — must run
|
|
12
|
+
// that first to populate the cache.
|
|
13
|
+
//
|
|
14
|
+
// Usage:
|
|
15
|
+
// cd /tmp/beir-nfcorpus
|
|
16
|
+
// node /path/to/scripts/run-beir-hybrid.mjs # RRF only
|
|
17
|
+
// RERANK=1 node /path/to/scripts/run-beir-hybrid.mjs # + cross-encoder rerank
|
|
18
|
+
// RRF_K=60 node /path/to/scripts/run-beir-hybrid.mjs # tune RRF k
|
|
19
|
+
//
|
|
20
|
+
// ADR-087.
|
|
21
|
+
|
|
22
|
+
import { readFileSync, writeFileSync, existsSync, mkdirSync } from 'node:fs';
|
|
23
|
+
import { fileURLToPath } from 'node:url';
|
|
24
|
+
import { dirname, join, resolve } from 'node:path';
|
|
25
|
+
import { performance } from 'node:perf_hooks';
|
|
26
|
+
|
|
27
|
+
const SCRIPT_DIR = dirname(fileURLToPath(import.meta.url));
|
|
28
|
+
const CLI_ROOT = resolve(SCRIPT_DIR, '..');
|
|
29
|
+
const RUFLO_ROOT = resolve(SCRIPT_DIR, '../../../..');
|
|
30
|
+
const RUNS_DIR = join(RUFLO_ROOT, 'docs', 'benchmarks', 'runs');
|
|
31
|
+
|
|
32
|
+
const DATA_DIR = process.env.BEIR_DATA_DIR || '/tmp/beir-nfcorpus/nfcorpus';
|
|
33
|
+
const BGE_MODEL = process.env.BGE_MODEL || 'Xenova/bge-base-en-v1.5';
|
|
34
|
+
const CACHE_DIR = join(process.env.CACHE_BASE_DIR || dirname(DATA_DIR), 'bge-cache');
|
|
35
|
+
const RRF_K = Number(process.env.RRF_K) || 60;
|
|
36
|
+
const RERANK = process.env.RERANK === '1';
|
|
37
|
+
const RERANK_TOP_K = Number(process.env.RERANK_TOP_K) || 100;
|
|
38
|
+
const MAX_QUERIES = Number(process.env.MAX_QUERIES) || 0;
|
|
39
|
+
|
|
40
|
+
// Same baselines table as run-beir-bge.mjs.
|
|
41
|
+
const BASELINES_BY_DATASET = {
|
|
42
|
+
nfcorpus: { 'BM25 (Lucene)': 0.325, 'DocT5query': 0.328, 'TAS-B': 0.319, 'GenQ': 0.319, 'ColBERT': 0.305, 'Contriever': 0.328, 'GTR-XL': 0.343, 'SPLADE++': 0.347, 'BGE-large-v1.5 (pub)': 0.380, 'SBERT msmarco': 0.272 },
|
|
43
|
+
scifact: { 'BM25 (Lucene)': 0.679, 'DocT5query': 0.675, 'TAS-B': 0.643, 'GenQ': 0.644, 'ColBERT': 0.671, 'Contriever': 0.677, 'GTR-XL': 0.662, 'SPLADE++': 0.704, 'BGE-large-v1.5 (pub)': 0.722, 'SBERT msmarco': 0.555 },
|
|
44
|
+
arguana: { 'BM25 (Lucene)': 0.397, 'DocT5query': 0.349, 'TAS-B': 0.429, 'GenQ': 0.493, 'ColBERT': 0.233, 'Contriever': 0.379, 'GTR-XL': 0.439, 'SPLADE++': 0.521, 'BGE-large-v1.5 (pub)': 0.636, 'SBERT msmarco': 0.371 },
|
|
45
|
+
scidocs: { 'BM25 (Lucene)': 0.158, 'DocT5query': 0.162, 'TAS-B': 0.149, 'GenQ': 0.143, 'ColBERT': 0.145, 'Contriever': 0.165, 'GTR-XL': 0.174, 'SPLADE++': 0.159, 'BGE-large-v1.5 (pub)': 0.225, 'SBERT msmarco': 0.122 },
|
|
46
|
+
};
|
|
47
|
+
function detectDataset(path) {
|
|
48
|
+
const p = path.toLowerCase();
|
|
49
|
+
for (const ds of Object.keys(BASELINES_BY_DATASET)) if (p.includes(ds)) return ds;
|
|
50
|
+
return 'nfcorpus';
|
|
51
|
+
}
|
|
52
|
+
|
|
53
|
+
function dcg(rels, k) { let s = 0; for (let i = 0; i < Math.min(rels.length, k); i++) s += (Math.pow(2, rels[i]) - 1) / Math.log2(i + 2); return s; }
|
|
54
|
+
function ndcg(retrieved, qrels, k) { const rels = retrieved.slice(0, k).map((id) => qrels.get(id) ?? 0); const ideal = [...qrels.values()].sort((a, b) => b - a).slice(0, k); const idcg = dcg(ideal, k); return idcg > 0 ? dcg(rels, k) / idcg : 0; }
|
|
55
|
+
function mrr(retrieved, qrels, k) { for (let i = 0; i < Math.min(retrieved.length, k); i++) if ((qrels.get(retrieved[i]) ?? 0) > 0) return 1 / (i + 1); return 0; }
|
|
56
|
+
function recall(retrieved, qrels, k) { const tot = [...qrels.values()].filter((v) => v > 0).length; if (tot === 0) return 0; let h = 0; for (let i = 0; i < Math.min(retrieved.length, k); i++) if ((qrels.get(retrieved[i]) ?? 0) > 0) h++; return h / tot; }
|
|
57
|
+
function cosine(a, b) { let s = 0; for (let i = 0; i < a.length; i++) s += a[i] * b[i]; return s; }
|
|
58
|
+
|
|
59
|
+
function loadJsonl(path) { return readFileSync(path, 'utf-8').split('\n').filter(Boolean).map((l) => JSON.parse(l)); }
|
|
60
|
+
function loadQrels(path) { const q = new Map(); const lines = readFileSync(path, 'utf-8').split('\n'); for (let i = 1; i < lines.length; i++) { if (!lines[i].trim()) continue; const [qid, did, score] = lines[i].split('\t'); if (!q.has(qid)) q.set(qid, new Map()); q.get(qid).set(did, Number(score)); } return q; }
|
|
61
|
+
|
|
62
|
+
function cachePath(suffix) { const safe = BGE_MODEL.replace(/\//g, '_'); return join(CACHE_DIR, `${safe}.${suffix}`); }
|
|
63
|
+
function loadEmbeddings(dim) {
|
|
64
|
+
if (!existsSync(cachePath('ids')) || !existsSync(cachePath('f32'))) return null;
|
|
65
|
+
const ids = readFileSync(cachePath('ids'), 'utf-8').split('\n').filter(Boolean);
|
|
66
|
+
const raw = readFileSync(cachePath('f32'));
|
|
67
|
+
const buf = new Float32Array(raw.buffer, raw.byteOffset, raw.length / 4);
|
|
68
|
+
const embeds = [];
|
|
69
|
+
for (let i = 0; i < ids.length; i++) embeds.push(buf.slice(i * dim, (i + 1) * dim));
|
|
70
|
+
return { ids, embeds };
|
|
71
|
+
}
|
|
72
|
+
|
|
73
|
+
async function main() {
|
|
74
|
+
const dataset = detectDataset(DATA_DIR);
|
|
75
|
+
const BASELINES_NDCG10 = BASELINES_BY_DATASET[dataset];
|
|
76
|
+
console.log(`# BEIR ${dataset} — hybrid RRF${RERANK ? ' + cross-encoder rerank' : ''} (ADR-087)`);
|
|
77
|
+
console.log(`Data: ${DATA_DIR}`);
|
|
78
|
+
console.log(`Dense: ${BGE_MODEL}`);
|
|
79
|
+
console.log(`RRF k: ${RRF_K}${RERANK ? `, rerank top-${RERANK_TOP_K}` : ''}`);
|
|
80
|
+
|
|
81
|
+
const corpus = loadJsonl(join(DATA_DIR, 'corpus.jsonl'));
|
|
82
|
+
const queries = loadJsonl(join(DATA_DIR, 'queries.jsonl'));
|
|
83
|
+
const qrels = loadQrels(join(DATA_DIR, 'qrels/test.tsv'));
|
|
84
|
+
console.log(`Corpus: ${corpus.length} docs · Test qrels: ${qrels.size}`);
|
|
85
|
+
|
|
86
|
+
// BGE embeddings — must be pre-cached from run-beir-bge.mjs.
|
|
87
|
+
const bge = await import(join(CLI_ROOT, 'dist/src/memory/bge-embedder.js'));
|
|
88
|
+
const emb = await bge.getBgeEmbedder(BGE_MODEL);
|
|
89
|
+
if (!emb) { console.error('BGE failed:', bge.getBgeStatus()); process.exit(1); }
|
|
90
|
+
const dim = emb.dim();
|
|
91
|
+
const cached = loadEmbeddings(dim);
|
|
92
|
+
if (!cached) { console.error(`No cached BGE embeddings at ${cachePath('f32')}. Run run-beir-bge.mjs first.`); process.exit(2); }
|
|
93
|
+
const docIds = cached.ids;
|
|
94
|
+
const docEmbeds = cached.embeds;
|
|
95
|
+
console.log(`Loaded ${docIds.length} cached BGE embeddings`);
|
|
96
|
+
|
|
97
|
+
// BM25 setup — USE_LUCENE_BM25=1 uses Porter+Lucene-stopword BM25 (matches
|
|
98
|
+
// published BEIR baselines); default is hybrid-retrieval's multi-field BM25.
|
|
99
|
+
const USE_LUCENE_BM25 = process.env.USE_LUCENE_BM25 === '1';
|
|
100
|
+
let bm25Score, tokenizeFn, titleDocs, textDocs, titleStats, textStats, luceneDocs, luceneStats;
|
|
101
|
+
if (USE_LUCENE_BM25) {
|
|
102
|
+
const { luceneTokenize, buildLuceneCorpusStats, luceneBM25 } = await import(join(CLI_ROOT, 'dist/src/memory/lucene-bm25.js'));
|
|
103
|
+
tokenizeFn = luceneTokenize;
|
|
104
|
+
luceneDocs = corpus.map((d) => luceneTokenize(`${d.title || ''} ${d.text || ''}`));
|
|
105
|
+
luceneStats = buildLuceneCorpusStats(luceneDocs);
|
|
106
|
+
bm25Score = (qTokens, idx) => luceneBM25(qTokens, luceneDocs[idx], luceneStats);
|
|
107
|
+
console.log(`BM25: Lucene-style (Porter stem + Lucene stopwords + length norm) — ADR-088`);
|
|
108
|
+
} else {
|
|
109
|
+
const { tokenize, buildCorpusStats, multiFieldBM25 } = await import(join(CLI_ROOT, 'dist/src/memory/hybrid-retrieval.js'));
|
|
110
|
+
tokenizeFn = tokenize;
|
|
111
|
+
titleDocs = corpus.map((d) => tokenize(d.title || ''));
|
|
112
|
+
textDocs = corpus.map((d) => tokenize(d.text || ''));
|
|
113
|
+
titleStats = buildCorpusStats(titleDocs);
|
|
114
|
+
textStats = buildCorpusStats(textDocs);
|
|
115
|
+
const SUBJECT_WEIGHT = Number(process.env.SUBJECT_WEIGHT ?? 1.0);
|
|
116
|
+
const BODY_WEIGHT = Number(process.env.BODY_WEIGHT ?? 1.0);
|
|
117
|
+
bm25Score = (qTokens, idx) => multiFieldBM25(qTokens, titleDocs[idx], textDocs[idx], titleStats, textStats, SUBJECT_WEIGHT, BODY_WEIGHT);
|
|
118
|
+
console.log(`BM25: multi-field (title sw=${SUBJECT_WEIGHT}, text bw=${BODY_WEIGHT})`);
|
|
119
|
+
}
|
|
120
|
+
|
|
121
|
+
// Optional cross-encoder reranker.
|
|
122
|
+
let crossEncoder = null;
|
|
123
|
+
if (RERANK) {
|
|
124
|
+
const ce = await import(join(CLI_ROOT, 'dist/src/memory/cross-encoder-rerank.js'));
|
|
125
|
+
const ceFn = await ce.getCrossEncoder('Xenova/ms-marco-MiniLM-L-6-v2');
|
|
126
|
+
if (!ceFn) { console.error('Cross-encoder failed to load:', ce.getCrossEncoderStatus()); process.exit(3); }
|
|
127
|
+
crossEncoder = ceFn;
|
|
128
|
+
console.log('Cross-encoder loaded: Xenova/ms-marco-MiniLM-L-6-v2');
|
|
129
|
+
}
|
|
130
|
+
|
|
131
|
+
const queriesById = new Map(queries.map((q) => [q._id, q.text]));
|
|
132
|
+
const evalQueryIds = MAX_QUERIES ? [...qrels.keys()].slice(0, MAX_QUERIES) : [...qrels.keys()];
|
|
133
|
+
|
|
134
|
+
// Build doc id → corpus index lookup (BM25 returns indices, dense returns ids).
|
|
135
|
+
const idToIdx = new Map(docIds.map((id, i) => [id, i]));
|
|
136
|
+
// Need a reverse map too: corpus index → doc id (for BM25 over corpus order).
|
|
137
|
+
const corpusIdToIdx = new Map(corpus.map((d, i) => [d._id, i]));
|
|
138
|
+
const corpusIdxToId = corpus.map((d) => d._id);
|
|
139
|
+
|
|
140
|
+
console.log(`\nRunning ${evalQueryIds.length} queries...`);
|
|
141
|
+
let nSum = 0, mSum = 0, r10Sum = 0, r100Sum = 0, n = 0;
|
|
142
|
+
const perQuery = [];
|
|
143
|
+
const tQ = performance.now();
|
|
144
|
+
|
|
145
|
+
for (const qid of evalQueryIds) {
|
|
146
|
+
const qtext = queriesById.get(qid);
|
|
147
|
+
if (!qtext) continue;
|
|
148
|
+
|
|
149
|
+
// §1 — dense BGE retrieval (ADR-090: opt-in query prefix when BGE_QUERY_PREFIX=1)
|
|
150
|
+
const qEmb = (process.env.BGE_QUERY_PREFIX === '1' && emb.embedQuery)
|
|
151
|
+
? await emb.embedQuery(qtext)
|
|
152
|
+
: await emb.embed(qtext);
|
|
153
|
+
const denseScored = new Array(docEmbeds.length);
|
|
154
|
+
for (let i = 0; i < docEmbeds.length; i++) denseScored[i] = { id: docIds[i], score: cosine(qEmb, docEmbeds[i]) };
|
|
155
|
+
denseScored.sort((a, b) => b.score - a.score);
|
|
156
|
+
|
|
157
|
+
// §2 — BM25 retrieval (Lucene or multi-field depending on USE_LUCENE_BM25)
|
|
158
|
+
const qTokens = tokenizeFn(qtext);
|
|
159
|
+
const bm25Scored = new Array(corpus.length);
|
|
160
|
+
for (let i = 0; i < corpus.length; i++) {
|
|
161
|
+
bm25Scored[i] = { id: corpusIdxToId[i], score: bm25Score(qTokens, i) };
|
|
162
|
+
}
|
|
163
|
+
bm25Scored.sort((a, b) => b.score - a.score);
|
|
164
|
+
|
|
165
|
+
// §3 — RRF fusion: score = sum over systems of 1/(k + rank).
|
|
166
|
+
// Only need top ~200 from each for stable RRF; cap at top-500 per system.
|
|
167
|
+
const TOP_PER_SYSTEM = 500;
|
|
168
|
+
const rrfScores = new Map();
|
|
169
|
+
for (let r = 0; r < Math.min(TOP_PER_SYSTEM, denseScored.length); r++) {
|
|
170
|
+
const id = denseScored[r].id;
|
|
171
|
+
rrfScores.set(id, (rrfScores.get(id) || 0) + 1 / (RRF_K + r + 1));
|
|
172
|
+
}
|
|
173
|
+
for (let r = 0; r < Math.min(TOP_PER_SYSTEM, bm25Scored.length); r++) {
|
|
174
|
+
const id = bm25Scored[r].id;
|
|
175
|
+
// Skip if this doc has 0 BM25 score (no token overlap) — true random docs would otherwise leak in
|
|
176
|
+
if (bm25Scored[r].score === 0) break;
|
|
177
|
+
rrfScores.set(id, (rrfScores.get(id) || 0) + 1 / (RRF_K + r + 1));
|
|
178
|
+
}
|
|
179
|
+
const fused = [...rrfScores.entries()]
|
|
180
|
+
.map(([id, score]) => ({ id, score }))
|
|
181
|
+
.sort((a, b) => b.score - a.score)
|
|
182
|
+
.slice(0, RERANK ? RERANK_TOP_K : 100);
|
|
183
|
+
|
|
184
|
+
let final = fused;
|
|
185
|
+
|
|
186
|
+
// §4 — optional cross-encoder rerank.
|
|
187
|
+
if (crossEncoder && fused.length > 0) {
|
|
188
|
+
const docsForRerank = fused.map(({ id }) => {
|
|
189
|
+
const ci = corpusIdToIdx.get(id);
|
|
190
|
+
const d = corpus[ci];
|
|
191
|
+
return `${d.title || ''} ${d.text || ''}`.slice(0, 4096);
|
|
192
|
+
});
|
|
193
|
+
const ceScores = await crossEncoder.scoreBatch(qtext, docsForRerank);
|
|
194
|
+
final = fused
|
|
195
|
+
.map((f, i) => ({ id: f.id, score: ceScores[i], rrf: f.score }))
|
|
196
|
+
.sort((a, b) => b.score - a.score);
|
|
197
|
+
}
|
|
198
|
+
|
|
199
|
+
const top100 = final.slice(0, 100).map((s) => s.id);
|
|
200
|
+
const qmap = qrels.get(qid);
|
|
201
|
+
const qNdcg = ndcg(top100, qmap, 10);
|
|
202
|
+
const qMrr = mrr(top100, qmap, 10);
|
|
203
|
+
const qR10 = recall(top100, qmap, 10);
|
|
204
|
+
const qR100 = recall(top100, qmap, 100);
|
|
205
|
+
nSum += qNdcg; mSum += qMrr; r10Sum += qR10; r100Sum += qR100; n++;
|
|
206
|
+
perQuery.push({ qid, ndcg10: qNdcg, mrr10: qMrr, recall10: qR10, recall100: qR100 });
|
|
207
|
+
if (n % 50 === 0) {
|
|
208
|
+
const elapsed = (performance.now() - tQ) / 1000;
|
|
209
|
+
console.log(` ${n}/${evalQueryIds.length} in ${elapsed.toFixed(0)}s running nDCG@10=${(nSum / n).toFixed(4)}`);
|
|
210
|
+
}
|
|
211
|
+
}
|
|
212
|
+
const queryMs = performance.now() - tQ;
|
|
213
|
+
|
|
214
|
+
const ndcg10 = nSum / n;
|
|
215
|
+
const mrr10 = mSum / n;
|
|
216
|
+
const recall10 = r10Sum / n;
|
|
217
|
+
const recall100 = r100Sum / n;
|
|
218
|
+
|
|
219
|
+
console.log(`\n=== Results (N=${n}, ${dataset}, BM25+BGE-base RRF k=${RRF_K}${RERANK ? ' + CE rerank' : ''}) ===`);
|
|
220
|
+
console.log(` nDCG@10: ${ndcg10.toFixed(4)}`);
|
|
221
|
+
console.log(` MRR@10: ${mrr10.toFixed(4)}`);
|
|
222
|
+
console.log(` Recall@10: ${recall10.toFixed(4)}`);
|
|
223
|
+
console.log(` Recall@100: ${recall100.toFixed(4)}`);
|
|
224
|
+
console.log(` Avg query latency: ${(queryMs / n).toFixed(0)}ms`);
|
|
225
|
+
|
|
226
|
+
const ourLabel = `ruflo + BM25+BGE-base RRF${RERANK ? '+CE' : ''}`;
|
|
227
|
+
const ranking = [
|
|
228
|
+
...Object.entries(BASELINES_NDCG10).map(([name, score]) => ({ name, score, ours: false })),
|
|
229
|
+
{ name: ourLabel, score: ndcg10, ours: true },
|
|
230
|
+
].sort((a, b) => b.score - a.score);
|
|
231
|
+
console.log(`\n=== vs ${dataset} listed baselines ===`);
|
|
232
|
+
for (const r of ranking) console.log(` ${r.score.toFixed(3)} ${r.name}${r.ours ? ' ← us' : ''}`);
|
|
233
|
+
const ourRank = ranking.findIndex((r) => r.ours) + 1;
|
|
234
|
+
console.log(`\n Our rank: ${ourRank} / ${ranking.length}`);
|
|
235
|
+
|
|
236
|
+
const summary = {
|
|
237
|
+
runAt: new Date().toISOString(),
|
|
238
|
+
benchmark: `beir-${dataset}-hybrid${RERANK ? '-rerank' : ''}`,
|
|
239
|
+
dataset,
|
|
240
|
+
pipeline: `BM25+BGE RRF${RERANK ? '+CE-rerank' : ''}`,
|
|
241
|
+
bgeModel: BGE_MODEL,
|
|
242
|
+
rrfK: RRF_K,
|
|
243
|
+
rerank: RERANK,
|
|
244
|
+
queries: n,
|
|
245
|
+
corpusSize: corpus.length,
|
|
246
|
+
metrics: { ndcg10, mrr10, recall10, recall100, avgQueryLatencyMs: queryMs / n },
|
|
247
|
+
perQuery,
|
|
248
|
+
baselines: BASELINES_NDCG10,
|
|
249
|
+
ourRank,
|
|
250
|
+
leaderboardLength: ranking.length,
|
|
251
|
+
};
|
|
252
|
+
mkdirSync(RUNS_DIR, { recursive: true });
|
|
253
|
+
const stamp = summary.runAt.replace(/[:.]/g, '-');
|
|
254
|
+
const suffix = RERANK ? 'hybrid-rerank' : 'hybrid-rrf';
|
|
255
|
+
writeFileSync(join(RUNS_DIR, `beir-${dataset}-${suffix}-${stamp}.json`), JSON.stringify(summary, null, 2));
|
|
256
|
+
writeFileSync(join(RUNS_DIR, `beir-${dataset}-${suffix}-latest.json`), JSON.stringify(summary, null, 2));
|
|
257
|
+
console.log(`\nWrote ${join(RUNS_DIR, `beir-${dataset}-${suffix}-${stamp}.json`)}`);
|
|
258
|
+
process.exit(0);
|
|
259
|
+
}
|
|
260
|
+
|
|
261
|
+
main().catch((err) => { console.error(err); process.exit(1); });
|