@claude-flow/cli 3.21.1 → 3.22.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.
Files changed (568) hide show
  1. package/.claude/helpers/helpers.manifest.json +12 -0
  2. package/.claude/helpers/hook-handler.cjs +26 -5
  3. package/.claude/helpers/intelligence.cjs +4 -1
  4. package/dist/src/commands/daemon.js +23 -2
  5. package/dist/src/commands/memory-distill.d.ts +27 -0
  6. package/dist/src/commands/memory-distill.js +374 -0
  7. package/dist/src/commands/memory.js +4 -2
  8. package/dist/src/index.d.ts +1 -1
  9. package/dist/src/index.js +22 -1
  10. package/dist/src/init/executor.js +20 -0
  11. package/dist/src/init/helper-refresh.d.ts +19 -0
  12. package/dist/src/init/helper-refresh.js +175 -0
  13. package/dist/src/init/helper-signing.d.ts +30 -0
  14. package/dist/src/init/helper-signing.js +60 -0
  15. package/dist/src/init/helpers-generator.d.ts +16 -0
  16. package/dist/src/init/helpers-generator.js +63 -6
  17. package/dist/src/init/statusline-generator.js +18 -7
  18. package/dist/src/mcp-client.js +13 -0
  19. package/dist/src/mcp-tools/agenticow-speculate-tools.js +9 -2
  20. package/dist/src/mcp-tools/browser-intent-tools.d.ts +162 -0
  21. package/dist/src/mcp-tools/browser-intent-tools.js +548 -0
  22. package/dist/src/mcp-tools/browser-tools.d.ts +9 -1
  23. package/dist/src/mcp-tools/browser-tools.js +4 -1
  24. package/dist/src/memory/memory-bridge.js +67 -47
  25. package/dist/src/memory/memory-initializer.d.ts +71 -0
  26. package/dist/src/memory/memory-initializer.js +330 -2
  27. package/dist/src/services/distill-tuning.d.ts +111 -0
  28. package/dist/src/services/distill-tuning.js +510 -0
  29. package/dist/src/services/memory-distillation.d.ts +41 -0
  30. package/dist/src/services/memory-distillation.js +277 -0
  31. package/dist/src/services/worker-daemon.d.ts +22 -1
  32. package/dist/src/services/worker-daemon.js +117 -6
  33. package/package.json +5 -2
  34. package/dist/src/agenticow/speculative-exploration.d.ts.map +0 -1
  35. package/dist/src/agenticow/speculative-exploration.js.map +0 -1
  36. package/dist/src/appliance/gguf-engine.d.ts.map +0 -1
  37. package/dist/src/appliance/gguf-engine.js.map +0 -1
  38. package/dist/src/appliance/ruvllm-bridge.d.ts.map +0 -1
  39. package/dist/src/appliance/ruvllm-bridge.js.map +0 -1
  40. package/dist/src/appliance/rvfa-builder.d.ts.map +0 -1
  41. package/dist/src/appliance/rvfa-builder.js.map +0 -1
  42. package/dist/src/appliance/rvfa-distribution.d.ts.map +0 -1
  43. package/dist/src/appliance/rvfa-distribution.js.map +0 -1
  44. package/dist/src/appliance/rvfa-format.d.ts.map +0 -1
  45. package/dist/src/appliance/rvfa-format.js.map +0 -1
  46. package/dist/src/appliance/rvfa-runner.d.ts.map +0 -1
  47. package/dist/src/appliance/rvfa-runner.js.map +0 -1
  48. package/dist/src/appliance/rvfa-signing.d.ts.map +0 -1
  49. package/dist/src/appliance/rvfa-signing.js.map +0 -1
  50. package/dist/src/autopilot-state.d.ts.map +0 -1
  51. package/dist/src/autopilot-state.js.map +0 -1
  52. package/dist/src/benchmarks/gaia-agent-planning.smoke.d.ts.map +0 -1
  53. package/dist/src/benchmarks/gaia-agent-planning.smoke.js.map +0 -1
  54. package/dist/src/benchmarks/gaia-agent.d.ts.map +0 -1
  55. package/dist/src/benchmarks/gaia-agent.js.map +0 -1
  56. package/dist/src/benchmarks/gaia-causal-memory.d.ts.map +0 -1
  57. package/dist/src/benchmarks/gaia-causal-memory.js.map +0 -1
  58. package/dist/src/benchmarks/gaia-causal-memory.smoke.d.ts.map +0 -1
  59. package/dist/src/benchmarks/gaia-causal-memory.smoke.js.map +0 -1
  60. package/dist/src/benchmarks/gaia-convergence.d.ts.map +0 -1
  61. package/dist/src/benchmarks/gaia-convergence.js.map +0 -1
  62. package/dist/src/benchmarks/gaia-convergence.smoke.d.ts.map +0 -1
  63. package/dist/src/benchmarks/gaia-convergence.smoke.js.map +0 -1
  64. package/dist/src/benchmarks/gaia-critic.d.ts.map +0 -1
  65. package/dist/src/benchmarks/gaia-critic.js.map +0 -1
  66. package/dist/src/benchmarks/gaia-critic.smoke.d.ts.map +0 -1
  67. package/dist/src/benchmarks/gaia-critic.smoke.js.map +0 -1
  68. package/dist/src/benchmarks/gaia-decomposer.d.ts.map +0 -1
  69. package/dist/src/benchmarks/gaia-decomposer.js.map +0 -1
  70. package/dist/src/benchmarks/gaia-decomposer.smoke.d.ts.map +0 -1
  71. package/dist/src/benchmarks/gaia-decomposer.smoke.js.map +0 -1
  72. package/dist/src/benchmarks/gaia-e2e-smoke.d.ts.map +0 -1
  73. package/dist/src/benchmarks/gaia-e2e-smoke.js.map +0 -1
  74. package/dist/src/benchmarks/gaia-extract.smoke.d.ts.map +0 -1
  75. package/dist/src/benchmarks/gaia-extract.smoke.js.map +0 -1
  76. package/dist/src/benchmarks/gaia-hardness/features.d.ts.map +0 -1
  77. package/dist/src/benchmarks/gaia-hardness/features.js.map +0 -1
  78. package/dist/src/benchmarks/gaia-hardness/predictor.d.ts.map +0 -1
  79. package/dist/src/benchmarks/gaia-hardness/predictor.js.map +0 -1
  80. package/dist/src/benchmarks/gaia-hardness/predictor.smoke.d.ts.map +0 -1
  81. package/dist/src/benchmarks/gaia-hardness/predictor.smoke.js.map +0 -1
  82. package/dist/src/benchmarks/gaia-hardness/train-data-loader.d.ts.map +0 -1
  83. package/dist/src/benchmarks/gaia-hardness/train-data-loader.js.map +0 -1
  84. package/dist/src/benchmarks/gaia-judge.d.ts.map +0 -1
  85. package/dist/src/benchmarks/gaia-judge.js.map +0 -1
  86. package/dist/src/benchmarks/gaia-loader.d.ts.map +0 -1
  87. package/dist/src/benchmarks/gaia-loader.js.map +0 -1
  88. package/dist/src/benchmarks/gaia-tools/file_read.d.ts.map +0 -1
  89. package/dist/src/benchmarks/gaia-tools/file_read.js.map +0 -1
  90. package/dist/src/benchmarks/gaia-tools/grounded_query.d.ts.map +0 -1
  91. package/dist/src/benchmarks/gaia-tools/grounded_query.js.map +0 -1
  92. package/dist/src/benchmarks/gaia-tools/index.d.ts.map +0 -1
  93. package/dist/src/benchmarks/gaia-tools/index.js.map +0 -1
  94. package/dist/src/benchmarks/gaia-tools/types.d.ts.map +0 -1
  95. package/dist/src/benchmarks/gaia-tools/types.js.map +0 -1
  96. package/dist/src/benchmarks/gaia-tools/web_search.d.ts.map +0 -1
  97. package/dist/src/benchmarks/gaia-tools/web_search.js.map +0 -1
  98. package/dist/src/benchmarks/gaia-voting.d.ts.map +0 -1
  99. package/dist/src/benchmarks/gaia-voting.js.map +0 -1
  100. package/dist/src/benchmarks/gaia-voting.smoke.d.ts.map +0 -1
  101. package/dist/src/benchmarks/gaia-voting.smoke.js.map +0 -1
  102. package/dist/src/benchmarks/pretrain/index.d.ts.map +0 -1
  103. package/dist/src/benchmarks/pretrain/index.js.map +0 -1
  104. package/dist/src/business-pods/bbs-budget-tracker.d.ts.map +0 -1
  105. package/dist/src/business-pods/bbs-budget-tracker.js.map +0 -1
  106. package/dist/src/business-pods/domain-affinity-policy.d.ts.map +0 -1
  107. package/dist/src/business-pods/domain-affinity-policy.js.map +0 -1
  108. package/dist/src/business-pods/pod-schema.d.ts.map +0 -1
  109. package/dist/src/business-pods/pod-schema.js.map +0 -1
  110. package/dist/src/commands/agent-wasm.d.ts.map +0 -1
  111. package/dist/src/commands/agent-wasm.js.map +0 -1
  112. package/dist/src/commands/agent.d.ts.map +0 -1
  113. package/dist/src/commands/agent.js.map +0 -1
  114. package/dist/src/commands/analyze.d.ts.map +0 -1
  115. package/dist/src/commands/analyze.js.map +0 -1
  116. package/dist/src/commands/appliance-advanced.d.ts.map +0 -1
  117. package/dist/src/commands/appliance-advanced.js.map +0 -1
  118. package/dist/src/commands/appliance.d.ts.map +0 -1
  119. package/dist/src/commands/appliance.js.map +0 -1
  120. package/dist/src/commands/autopilot.d.ts.map +0 -1
  121. package/dist/src/commands/autopilot.js.map +0 -1
  122. package/dist/src/commands/benchmark.d.ts.map +0 -1
  123. package/dist/src/commands/benchmark.js.map +0 -1
  124. package/dist/src/commands/claims.d.ts.map +0 -1
  125. package/dist/src/commands/claims.js.map +0 -1
  126. package/dist/src/commands/cleanup.d.ts.map +0 -1
  127. package/dist/src/commands/cleanup.js.map +0 -1
  128. package/dist/src/commands/completions.d.ts.map +0 -1
  129. package/dist/src/commands/completions.js.map +0 -1
  130. package/dist/src/commands/config.d.ts.map +0 -1
  131. package/dist/src/commands/config.js.map +0 -1
  132. package/dist/src/commands/daemon.d.ts.map +0 -1
  133. package/dist/src/commands/daemon.js.map +0 -1
  134. package/dist/src/commands/deployment.d.ts.map +0 -1
  135. package/dist/src/commands/deployment.js.map +0 -1
  136. package/dist/src/commands/doctor.d.ts.map +0 -1
  137. package/dist/src/commands/doctor.js.map +0 -1
  138. package/dist/src/commands/eject.d.ts.map +0 -1
  139. package/dist/src/commands/eject.js.map +0 -1
  140. package/dist/src/commands/embeddings.d.ts.map +0 -1
  141. package/dist/src/commands/embeddings.js.map +0 -1
  142. package/dist/src/commands/gaia-bench.d.ts.map +0 -1
  143. package/dist/src/commands/gaia-bench.js.map +0 -1
  144. package/dist/src/commands/guidance.d.ts.map +0 -1
  145. package/dist/src/commands/guidance.js.map +0 -1
  146. package/dist/src/commands/hive-mind.d.ts.map +0 -1
  147. package/dist/src/commands/hive-mind.js.map +0 -1
  148. package/dist/src/commands/hooks.d.ts.map +0 -1
  149. package/dist/src/commands/hooks.js.map +0 -1
  150. package/dist/src/commands/index.d.ts.map +0 -1
  151. package/dist/src/commands/index.js.map +0 -1
  152. package/dist/src/commands/init.d.ts.map +0 -1
  153. package/dist/src/commands/init.js.map +0 -1
  154. package/dist/src/commands/issues.d.ts.map +0 -1
  155. package/dist/src/commands/issues.js.map +0 -1
  156. package/dist/src/commands/mcp.d.ts.map +0 -1
  157. package/dist/src/commands/mcp.js.map +0 -1
  158. package/dist/src/commands/memory.d.ts.map +0 -1
  159. package/dist/src/commands/memory.js.map +0 -1
  160. package/dist/src/commands/metaharness.d.ts.map +0 -1
  161. package/dist/src/commands/metaharness.js.map +0 -1
  162. package/dist/src/commands/migrate.d.ts.map +0 -1
  163. package/dist/src/commands/migrate.js.map +0 -1
  164. package/dist/src/commands/neural.d.ts.map +0 -1
  165. package/dist/src/commands/neural.js.map +0 -1
  166. package/dist/src/commands/performance.d.ts.map +0 -1
  167. package/dist/src/commands/performance.js.map +0 -1
  168. package/dist/src/commands/plugins.d.ts.map +0 -1
  169. package/dist/src/commands/plugins.js.map +0 -1
  170. package/dist/src/commands/process.d.ts.map +0 -1
  171. package/dist/src/commands/process.js.map +0 -1
  172. package/dist/src/commands/progress.d.ts.map +0 -1
  173. package/dist/src/commands/progress.js.map +0 -1
  174. package/dist/src/commands/providers.d.ts.map +0 -1
  175. package/dist/src/commands/providers.js.map +0 -1
  176. package/dist/src/commands/route.d.ts.map +0 -1
  177. package/dist/src/commands/route.js.map +0 -1
  178. package/dist/src/commands/ruvector/backup.d.ts.map +0 -1
  179. package/dist/src/commands/ruvector/backup.js.map +0 -1
  180. package/dist/src/commands/ruvector/benchmark.d.ts.map +0 -1
  181. package/dist/src/commands/ruvector/benchmark.js.map +0 -1
  182. package/dist/src/commands/ruvector/import.d.ts.map +0 -1
  183. package/dist/src/commands/ruvector/import.js.map +0 -1
  184. package/dist/src/commands/ruvector/index.d.ts.map +0 -1
  185. package/dist/src/commands/ruvector/index.js.map +0 -1
  186. package/dist/src/commands/ruvector/init.d.ts.map +0 -1
  187. package/dist/src/commands/ruvector/init.js.map +0 -1
  188. package/dist/src/commands/ruvector/migrate.d.ts.map +0 -1
  189. package/dist/src/commands/ruvector/migrate.js.map +0 -1
  190. package/dist/src/commands/ruvector/optimize.d.ts.map +0 -1
  191. package/dist/src/commands/ruvector/optimize.js.map +0 -1
  192. package/dist/src/commands/ruvector/pg-utils.d.ts.map +0 -1
  193. package/dist/src/commands/ruvector/pg-utils.js.map +0 -1
  194. package/dist/src/commands/ruvector/setup.d.ts.map +0 -1
  195. package/dist/src/commands/ruvector/setup.js.map +0 -1
  196. package/dist/src/commands/ruvector/status.d.ts.map +0 -1
  197. package/dist/src/commands/ruvector/status.js.map +0 -1
  198. package/dist/src/commands/security.d.ts.map +0 -1
  199. package/dist/src/commands/security.js.map +0 -1
  200. package/dist/src/commands/session.d.ts.map +0 -1
  201. package/dist/src/commands/session.js.map +0 -1
  202. package/dist/src/commands/start.d.ts.map +0 -1
  203. package/dist/src/commands/start.js.map +0 -1
  204. package/dist/src/commands/status.d.ts.map +0 -1
  205. package/dist/src/commands/status.js.map +0 -1
  206. package/dist/src/commands/swarm.d.ts.map +0 -1
  207. package/dist/src/commands/swarm.js.map +0 -1
  208. package/dist/src/commands/task.d.ts.map +0 -1
  209. package/dist/src/commands/task.js.map +0 -1
  210. package/dist/src/commands/transfer-store.d.ts.map +0 -1
  211. package/dist/src/commands/transfer-store.js.map +0 -1
  212. package/dist/src/commands/update.d.ts.map +0 -1
  213. package/dist/src/commands/update.js.map +0 -1
  214. package/dist/src/commands/verify.d.ts.map +0 -1
  215. package/dist/src/commands/verify.js.map +0 -1
  216. package/dist/src/commands/workflow.d.ts.map +0 -1
  217. package/dist/src/commands/workflow.js.map +0 -1
  218. package/dist/src/config-adapter.d.ts.map +0 -1
  219. package/dist/src/config-adapter.js.map +0 -1
  220. package/dist/src/encryption/vault.d.ts.map +0 -1
  221. package/dist/src/encryption/vault.js.map +0 -1
  222. package/dist/src/fs-secure.d.ts.map +0 -1
  223. package/dist/src/fs-secure.js.map +0 -1
  224. package/dist/src/index.d.ts.map +0 -1
  225. package/dist/src/index.js.map +0 -1
  226. package/dist/src/infrastructure/in-memory-repositories.d.ts.map +0 -1
  227. package/dist/src/infrastructure/in-memory-repositories.js.map +0 -1
  228. package/dist/src/init/claudemd-generator.d.ts.map +0 -1
  229. package/dist/src/init/claudemd-generator.js.map +0 -1
  230. package/dist/src/init/executor.d.ts.map +0 -1
  231. package/dist/src/init/executor.js.map +0 -1
  232. package/dist/src/init/helpers-generator.d.ts.map +0 -1
  233. package/dist/src/init/helpers-generator.js.map +0 -1
  234. package/dist/src/init/index.d.ts.map +0 -1
  235. package/dist/src/init/index.js.map +0 -1
  236. package/dist/src/init/mcp-generator.d.ts.map +0 -1
  237. package/dist/src/init/mcp-generator.js.map +0 -1
  238. package/dist/src/init/memory-package-resolver.d.ts.map +0 -1
  239. package/dist/src/init/memory-package-resolver.js.map +0 -1
  240. package/dist/src/init/settings-generator.d.ts.map +0 -1
  241. package/dist/src/init/settings-generator.js.map +0 -1
  242. package/dist/src/init/statusline-generator.d.ts.map +0 -1
  243. package/dist/src/init/statusline-generator.js.map +0 -1
  244. package/dist/src/init/types.d.ts.map +0 -1
  245. package/dist/src/init/types.js.map +0 -1
  246. package/dist/src/log-filters.d.ts.map +0 -1
  247. package/dist/src/log-filters.js.map +0 -1
  248. package/dist/src/mcp-client.d.ts.map +0 -1
  249. package/dist/src/mcp-client.js.map +0 -1
  250. package/dist/src/mcp-server.d.ts.map +0 -1
  251. package/dist/src/mcp-server.js.map +0 -1
  252. package/dist/src/mcp-tools/agent-execute-core.d.ts.map +0 -1
  253. package/dist/src/mcp-tools/agent-execute-core.js.map +0 -1
  254. package/dist/src/mcp-tools/agent-tools.d.ts.map +0 -1
  255. package/dist/src/mcp-tools/agent-tools.js.map +0 -1
  256. package/dist/src/mcp-tools/agentbbs-tools.d.ts.map +0 -1
  257. package/dist/src/mcp-tools/agentbbs-tools.js.map +0 -1
  258. package/dist/src/mcp-tools/agentdb-tools.d.ts.map +0 -1
  259. package/dist/src/mcp-tools/agentdb-tools.js.map +0 -1
  260. package/dist/src/mcp-tools/agenticow-loader.d.ts.map +0 -1
  261. package/dist/src/mcp-tools/agenticow-loader.js.map +0 -1
  262. package/dist/src/mcp-tools/agenticow-speculate-tools.d.ts.map +0 -1
  263. package/dist/src/mcp-tools/agenticow-speculate-tools.js.map +0 -1
  264. package/dist/src/mcp-tools/agenticow-tools.d.ts.map +0 -1
  265. package/dist/src/mcp-tools/agenticow-tools.js.map +0 -1
  266. package/dist/src/mcp-tools/analyze-tools.d.ts.map +0 -1
  267. package/dist/src/mcp-tools/analyze-tools.js.map +0 -1
  268. package/dist/src/mcp-tools/auto-install.d.ts.map +0 -1
  269. package/dist/src/mcp-tools/auto-install.js.map +0 -1
  270. package/dist/src/mcp-tools/autopilot-tools.d.ts.map +0 -1
  271. package/dist/src/mcp-tools/autopilot-tools.js.map +0 -1
  272. package/dist/src/mcp-tools/browser-session-tools.d.ts.map +0 -1
  273. package/dist/src/mcp-tools/browser-session-tools.js.map +0 -1
  274. package/dist/src/mcp-tools/browser-tools.d.ts.map +0 -1
  275. package/dist/src/mcp-tools/browser-tools.js.map +0 -1
  276. package/dist/src/mcp-tools/business-pod-tools.d.ts.map +0 -1
  277. package/dist/src/mcp-tools/business-pod-tools.js.map +0 -1
  278. package/dist/src/mcp-tools/claims-tools.d.ts.map +0 -1
  279. package/dist/src/mcp-tools/claims-tools.js.map +0 -1
  280. package/dist/src/mcp-tools/config-tools.d.ts.map +0 -1
  281. package/dist/src/mcp-tools/config-tools.js.map +0 -1
  282. package/dist/src/mcp-tools/coordination-tools.d.ts.map +0 -1
  283. package/dist/src/mcp-tools/coordination-tools.js.map +0 -1
  284. package/dist/src/mcp-tools/daa-tools.d.ts.map +0 -1
  285. package/dist/src/mcp-tools/daa-tools.js.map +0 -1
  286. package/dist/src/mcp-tools/embeddings-tools.d.ts.map +0 -1
  287. package/dist/src/mcp-tools/embeddings-tools.js.map +0 -1
  288. package/dist/src/mcp-tools/github-tools.d.ts.map +0 -1
  289. package/dist/src/mcp-tools/github-tools.js.map +0 -1
  290. package/dist/src/mcp-tools/guidance-tools.d.ts.map +0 -1
  291. package/dist/src/mcp-tools/guidance-tools.js.map +0 -1
  292. package/dist/src/mcp-tools/hive-mind-tools.d.ts.map +0 -1
  293. package/dist/src/mcp-tools/hive-mind-tools.js.map +0 -1
  294. package/dist/src/mcp-tools/hooks-tools.d.ts.map +0 -1
  295. package/dist/src/mcp-tools/hooks-tools.js.map +0 -1
  296. package/dist/src/mcp-tools/http-fetch-tools.d.ts.map +0 -1
  297. package/dist/src/mcp-tools/http-fetch-tools.js.map +0 -1
  298. package/dist/src/mcp-tools/index.d.ts.map +0 -1
  299. package/dist/src/mcp-tools/index.js.map +0 -1
  300. package/dist/src/mcp-tools/managed-agent-tools.d.ts.map +0 -1
  301. package/dist/src/mcp-tools/managed-agent-tools.js.map +0 -1
  302. package/dist/src/mcp-tools/memory-tools.d.ts.map +0 -1
  303. package/dist/src/mcp-tools/memory-tools.js.map +0 -1
  304. package/dist/src/mcp-tools/metaharness-tools.d.ts.map +0 -1
  305. package/dist/src/mcp-tools/metaharness-tools.js.map +0 -1
  306. package/dist/src/mcp-tools/neural-tools.d.ts.map +0 -1
  307. package/dist/src/mcp-tools/neural-tools.js.map +0 -1
  308. package/dist/src/mcp-tools/performance-tools.d.ts.map +0 -1
  309. package/dist/src/mcp-tools/performance-tools.js.map +0 -1
  310. package/dist/src/mcp-tools/progress-tools.d.ts.map +0 -1
  311. package/dist/src/mcp-tools/progress-tools.js.map +0 -1
  312. package/dist/src/mcp-tools/request-tracker.d.ts.map +0 -1
  313. package/dist/src/mcp-tools/request-tracker.js.map +0 -1
  314. package/dist/src/mcp-tools/ruvllm-tools.d.ts.map +0 -1
  315. package/dist/src/mcp-tools/ruvllm-tools.js.map +0 -1
  316. package/dist/src/mcp-tools/security-tools.d.ts.map +0 -1
  317. package/dist/src/mcp-tools/security-tools.js.map +0 -1
  318. package/dist/src/mcp-tools/session-tools.d.ts.map +0 -1
  319. package/dist/src/mcp-tools/session-tools.js.map +0 -1
  320. package/dist/src/mcp-tools/swarm-tools.d.ts.map +0 -1
  321. package/dist/src/mcp-tools/swarm-tools.js.map +0 -1
  322. package/dist/src/mcp-tools/system-tools.d.ts.map +0 -1
  323. package/dist/src/mcp-tools/system-tools.js.map +0 -1
  324. package/dist/src/mcp-tools/task-tools.d.ts.map +0 -1
  325. package/dist/src/mcp-tools/task-tools.js.map +0 -1
  326. package/dist/src/mcp-tools/terminal-tools.d.ts.map +0 -1
  327. package/dist/src/mcp-tools/terminal-tools.js.map +0 -1
  328. package/dist/src/mcp-tools/testgen-tools.d.ts.map +0 -1
  329. package/dist/src/mcp-tools/testgen-tools.js.map +0 -1
  330. package/dist/src/mcp-tools/tool-loop-guardrail.d.ts.map +0 -1
  331. package/dist/src/mcp-tools/tool-loop-guardrail.js.map +0 -1
  332. package/dist/src/mcp-tools/transfer-tools.d.ts.map +0 -1
  333. package/dist/src/mcp-tools/transfer-tools.js.map +0 -1
  334. package/dist/src/mcp-tools/types.d.ts.map +0 -1
  335. package/dist/src/mcp-tools/types.js.map +0 -1
  336. package/dist/src/mcp-tools/validate-input.d.ts.map +0 -1
  337. package/dist/src/mcp-tools/validate-input.js.map +0 -1
  338. package/dist/src/mcp-tools/wasm-agent-tools.d.ts.map +0 -1
  339. package/dist/src/mcp-tools/wasm-agent-tools.js.map +0 -1
  340. package/dist/src/mcp-tools/workflow-tools.d.ts.map +0 -1
  341. package/dist/src/mcp-tools/workflow-tools.js.map +0 -1
  342. package/dist/src/memory/bge-embedder.d.ts.map +0 -1
  343. package/dist/src/memory/bge-embedder.js.map +0 -1
  344. package/dist/src/memory/cross-encoder-rerank.d.ts.map +0 -1
  345. package/dist/src/memory/cross-encoder-rerank.js.map +0 -1
  346. package/dist/src/memory/embedding-quantization.d.ts.map +0 -1
  347. package/dist/src/memory/embedding-quantization.js.map +0 -1
  348. package/dist/src/memory/ewc-consolidation.d.ts.map +0 -1
  349. package/dist/src/memory/ewc-consolidation.js.map +0 -1
  350. package/dist/src/memory/graph-edge-writer.d.ts.map +0 -1
  351. package/dist/src/memory/graph-edge-writer.js.map +0 -1
  352. package/dist/src/memory/hybrid-retrieval.d.ts.map +0 -1
  353. package/dist/src/memory/hybrid-retrieval.js.map +0 -1
  354. package/dist/src/memory/intelligence.d.ts.map +0 -1
  355. package/dist/src/memory/intelligence.js.map +0 -1
  356. package/dist/src/memory/lucene-bm25.d.ts.map +0 -1
  357. package/dist/src/memory/lucene-bm25.js.map +0 -1
  358. package/dist/src/memory/memory-bridge.d.ts.map +0 -1
  359. package/dist/src/memory/memory-bridge.js.map +0 -1
  360. package/dist/src/memory/memory-initializer.d.ts.map +0 -1
  361. package/dist/src/memory/memory-initializer.js.map +0 -1
  362. package/dist/src/memory/neural-package-bridge.d.ts.map +0 -1
  363. package/dist/src/memory/neural-package-bridge.js.map +0 -1
  364. package/dist/src/memory/rabitq-index.d.ts.map +0 -1
  365. package/dist/src/memory/rabitq-index.js.map +0 -1
  366. package/dist/src/memory/sona-optimizer.d.ts.map +0 -1
  367. package/dist/src/memory/sona-optimizer.js.map +0 -1
  368. package/dist/src/memory/structured-distill.d.ts.map +0 -1
  369. package/dist/src/memory/structured-distill.js.map +0 -1
  370. package/dist/src/output.d.ts.map +0 -1
  371. package/dist/src/output.js.map +0 -1
  372. package/dist/src/parser.d.ts.map +0 -1
  373. package/dist/src/parser.js.map +0 -1
  374. package/dist/src/plugins/manager.d.ts.map +0 -1
  375. package/dist/src/plugins/manager.js.map +0 -1
  376. package/dist/src/plugins/store/discovery.d.ts.map +0 -1
  377. package/dist/src/plugins/store/discovery.js.map +0 -1
  378. package/dist/src/plugins/store/index.d.ts.map +0 -1
  379. package/dist/src/plugins/store/index.js.map +0 -1
  380. package/dist/src/plugins/store/search.d.ts.map +0 -1
  381. package/dist/src/plugins/store/search.js.map +0 -1
  382. package/dist/src/plugins/store/types.d.ts.map +0 -1
  383. package/dist/src/plugins/store/types.js.map +0 -1
  384. package/dist/src/plugins/tests/demo-plugin-store.d.ts.map +0 -1
  385. package/dist/src/plugins/tests/demo-plugin-store.js.map +0 -1
  386. package/dist/src/plugins/tests/standalone-test.d.ts.map +0 -1
  387. package/dist/src/plugins/tests/standalone-test.js.map +0 -1
  388. package/dist/src/plugins/tests/test-plugin-store.d.ts.map +0 -1
  389. package/dist/src/plugins/tests/test-plugin-store.js.map +0 -1
  390. package/dist/src/production/circuit-breaker.d.ts.map +0 -1
  391. package/dist/src/production/circuit-breaker.js.map +0 -1
  392. package/dist/src/production/error-handler.d.ts.map +0 -1
  393. package/dist/src/production/error-handler.js.map +0 -1
  394. package/dist/src/production/index.d.ts.map +0 -1
  395. package/dist/src/production/index.js.map +0 -1
  396. package/dist/src/production/monitoring.d.ts.map +0 -1
  397. package/dist/src/production/monitoring.js.map +0 -1
  398. package/dist/src/production/rate-limiter.d.ts.map +0 -1
  399. package/dist/src/production/rate-limiter.js.map +0 -1
  400. package/dist/src/production/retry.d.ts.map +0 -1
  401. package/dist/src/production/retry.js.map +0 -1
  402. package/dist/src/prompt.d.ts.map +0 -1
  403. package/dist/src/prompt.js.map +0 -1
  404. package/dist/src/runtime/headless.d.ts.map +0 -1
  405. package/dist/src/runtime/headless.js.map +0 -1
  406. package/dist/src/runtime/parent-death-watchdog.d.ts.map +0 -1
  407. package/dist/src/runtime/parent-death-watchdog.js.map +0 -1
  408. package/dist/src/ruvector/agent-wasm.d.ts.map +0 -1
  409. package/dist/src/ruvector/agent-wasm.js.map +0 -1
  410. package/dist/src/ruvector/ast-analyzer.d.ts.map +0 -1
  411. package/dist/src/ruvector/ast-analyzer.js.map +0 -1
  412. package/dist/src/ruvector/codemods/engine.d.ts.map +0 -1
  413. package/dist/src/ruvector/codemods/engine.js.map +0 -1
  414. package/dist/src/ruvector/codemods/scope-analysis.d.ts.map +0 -1
  415. package/dist/src/ruvector/codemods/scope-analysis.js.map +0 -1
  416. package/dist/src/ruvector/coverage-router.d.ts.map +0 -1
  417. package/dist/src/ruvector/coverage-router.js.map +0 -1
  418. package/dist/src/ruvector/coverage-tools.d.ts.map +0 -1
  419. package/dist/src/ruvector/coverage-tools.js.map +0 -1
  420. package/dist/src/ruvector/diff-classifier.d.ts.map +0 -1
  421. package/dist/src/ruvector/diff-classifier.js.map +0 -1
  422. package/dist/src/ruvector/diskann-backend.d.ts.map +0 -1
  423. package/dist/src/ruvector/diskann-backend.js.map +0 -1
  424. package/dist/src/ruvector/enhanced-model-router.d.ts.map +0 -1
  425. package/dist/src/ruvector/enhanced-model-router.js.map +0 -1
  426. package/dist/src/ruvector/graph-analyzer.d.ts.map +0 -1
  427. package/dist/src/ruvector/graph-analyzer.js.map +0 -1
  428. package/dist/src/ruvector/graph-backend.d.ts.map +0 -1
  429. package/dist/src/ruvector/graph-backend.js.map +0 -1
  430. package/dist/src/ruvector/index.d.ts.map +0 -1
  431. package/dist/src/ruvector/index.js.map +0 -1
  432. package/dist/src/ruvector/lora-adapter.d.ts.map +0 -1
  433. package/dist/src/ruvector/lora-adapter.js.map +0 -1
  434. package/dist/src/ruvector/model-prices.d.ts.map +0 -1
  435. package/dist/src/ruvector/model-prices.js.map +0 -1
  436. package/dist/src/ruvector/model-router.d.ts.map +0 -1
  437. package/dist/src/ruvector/model-router.js.map +0 -1
  438. package/dist/src/ruvector/neural-router.d.ts.map +0 -1
  439. package/dist/src/ruvector/neural-router.js.map +0 -1
  440. package/dist/src/ruvector/output-verifier.d.ts.map +0 -1
  441. package/dist/src/ruvector/output-verifier.js.map +0 -1
  442. package/dist/src/ruvector/q-learning-router.d.ts.map +0 -1
  443. package/dist/src/ruvector/q-learning-router.js.map +0 -1
  444. package/dist/src/ruvector/router-calibrator.d.ts.map +0 -1
  445. package/dist/src/ruvector/router-calibrator.js.map +0 -1
  446. package/dist/src/ruvector/router-parallel-recorder.d.ts.map +0 -1
  447. package/dist/src/ruvector/router-parallel-recorder.js.map +0 -1
  448. package/dist/src/ruvector/router-trajectory.d.ts.map +0 -1
  449. package/dist/src/ruvector/router-trajectory.js.map +0 -1
  450. package/dist/src/ruvector/run-transcript-recorder.d.ts.map +0 -1
  451. package/dist/src/ruvector/run-transcript-recorder.js.map +0 -1
  452. package/dist/src/ruvector/ruvllm-wasm.d.ts.map +0 -1
  453. package/dist/src/ruvector/ruvllm-wasm.js.map +0 -1
  454. package/dist/src/ruvector/semantic-router.d.ts.map +0 -1
  455. package/dist/src/ruvector/semantic-router.js.map +0 -1
  456. package/dist/src/ruvector/task-embedder.d.ts.map +0 -1
  457. package/dist/src/ruvector/task-embedder.js.map +0 -1
  458. package/dist/src/ruvector/trajectory-tree.d.ts.map +0 -1
  459. package/dist/src/ruvector/trajectory-tree.js.map +0 -1
  460. package/dist/src/ruvector/vector-db.d.ts.map +0 -1
  461. package/dist/src/ruvector/vector-db.js.map +0 -1
  462. package/dist/src/services/agentic-flow-bridge.d.ts.map +0 -1
  463. package/dist/src/services/agentic-flow-bridge.js.map +0 -1
  464. package/dist/src/services/checkpoint-gate.d.ts.map +0 -1
  465. package/dist/src/services/checkpoint-gate.js.map +0 -1
  466. package/dist/src/services/claim-service.d.ts.map +0 -1
  467. package/dist/src/services/claim-service.js.map +0 -1
  468. package/dist/src/services/config-file-manager.d.ts.map +0 -1
  469. package/dist/src/services/config-file-manager.js.map +0 -1
  470. package/dist/src/services/container-worker-pool.d.ts.map +0 -1
  471. package/dist/src/services/container-worker-pool.js.map +0 -1
  472. package/dist/src/services/distill-oracle.d.ts.map +0 -1
  473. package/dist/src/services/distill-oracle.js.map +0 -1
  474. package/dist/src/services/fable-harness.d.ts.map +0 -1
  475. package/dist/src/services/fable-harness.js.map +0 -1
  476. package/dist/src/services/headless-worker-executor.d.ts.map +0 -1
  477. package/dist/src/services/headless-worker-executor.js.map +0 -1
  478. package/dist/src/services/index.d.ts.map +0 -1
  479. package/dist/src/services/index.js.map +0 -1
  480. package/dist/src/services/native-training.d.ts.map +0 -1
  481. package/dist/src/services/native-training.js.map +0 -1
  482. package/dist/src/services/registry-api.d.ts.map +0 -1
  483. package/dist/src/services/registry-api.js.map +0 -1
  484. package/dist/src/services/ruvector-training.d.ts.map +0 -1
  485. package/dist/src/services/ruvector-training.js.map +0 -1
  486. package/dist/src/services/swarm-memory-branches.d.ts.map +0 -1
  487. package/dist/src/services/swarm-memory-branches.js.map +0 -1
  488. package/dist/src/services/weight-eft.d.ts.map +0 -1
  489. package/dist/src/services/weight-eft.js.map +0 -1
  490. package/dist/src/services/worker-daemon.d.ts.map +0 -1
  491. package/dist/src/services/worker-daemon.js.map +0 -1
  492. package/dist/src/services/worker-queue.d.ts.map +0 -1
  493. package/dist/src/services/worker-queue.js.map +0 -1
  494. package/dist/src/suggest.d.ts.map +0 -1
  495. package/dist/src/suggest.js.map +0 -1
  496. package/dist/src/transfer/anonymization/index.d.ts.map +0 -1
  497. package/dist/src/transfer/anonymization/index.js.map +0 -1
  498. package/dist/src/transfer/deploy-seraphine.d.ts.map +0 -1
  499. package/dist/src/transfer/deploy-seraphine.js.map +0 -1
  500. package/dist/src/transfer/export.d.ts.map +0 -1
  501. package/dist/src/transfer/export.js.map +0 -1
  502. package/dist/src/transfer/index.d.ts.map +0 -1
  503. package/dist/src/transfer/index.js.map +0 -1
  504. package/dist/src/transfer/ipfs/client.d.ts.map +0 -1
  505. package/dist/src/transfer/ipfs/client.js.map +0 -1
  506. package/dist/src/transfer/ipfs/upload.d.ts.map +0 -1
  507. package/dist/src/transfer/ipfs/upload.js.map +0 -1
  508. package/dist/src/transfer/models/seraphine.d.ts.map +0 -1
  509. package/dist/src/transfer/models/seraphine.js.map +0 -1
  510. package/dist/src/transfer/serialization/cfp.d.ts.map +0 -1
  511. package/dist/src/transfer/serialization/cfp.js.map +0 -1
  512. package/dist/src/transfer/storage/gcs.d.ts.map +0 -1
  513. package/dist/src/transfer/storage/gcs.js.map +0 -1
  514. package/dist/src/transfer/storage/index.d.ts.map +0 -1
  515. package/dist/src/transfer/storage/index.js.map +0 -1
  516. package/dist/src/transfer/store/discovery.d.ts.map +0 -1
  517. package/dist/src/transfer/store/discovery.js.map +0 -1
  518. package/dist/src/transfer/store/download.d.ts.map +0 -1
  519. package/dist/src/transfer/store/download.js.map +0 -1
  520. package/dist/src/transfer/store/index.d.ts.map +0 -1
  521. package/dist/src/transfer/store/index.js.map +0 -1
  522. package/dist/src/transfer/store/publish.d.ts.map +0 -1
  523. package/dist/src/transfer/store/publish.js.map +0 -1
  524. package/dist/src/transfer/store/registry.d.ts.map +0 -1
  525. package/dist/src/transfer/store/registry.js.map +0 -1
  526. package/dist/src/transfer/store/search.d.ts.map +0 -1
  527. package/dist/src/transfer/store/search.js.map +0 -1
  528. package/dist/src/transfer/store/tests/standalone-test.d.ts.map +0 -1
  529. package/dist/src/transfer/store/tests/standalone-test.js.map +0 -1
  530. package/dist/src/transfer/store/types.d.ts.map +0 -1
  531. package/dist/src/transfer/store/types.js.map +0 -1
  532. package/dist/src/transfer/test-seraphine.d.ts.map +0 -1
  533. package/dist/src/transfer/test-seraphine.js.map +0 -1
  534. package/dist/src/transfer/tests/test-store.d.ts.map +0 -1
  535. package/dist/src/transfer/tests/test-store.js.map +0 -1
  536. package/dist/src/transfer/types.d.ts.map +0 -1
  537. package/dist/src/transfer/types.js.map +0 -1
  538. package/dist/src/types.d.ts.map +0 -1
  539. package/dist/src/types.js.map +0 -1
  540. package/dist/src/update/checker.d.ts.map +0 -1
  541. package/dist/src/update/checker.js.map +0 -1
  542. package/dist/src/update/executor.d.ts.map +0 -1
  543. package/dist/src/update/executor.js.map +0 -1
  544. package/dist/src/update/index.d.ts.map +0 -1
  545. package/dist/src/update/index.js.map +0 -1
  546. package/dist/src/update/rate-limiter.d.ts.map +0 -1
  547. package/dist/src/update/rate-limiter.js.map +0 -1
  548. package/dist/src/update/validator.d.ts.map +0 -1
  549. package/dist/src/update/validator.js.map +0 -1
  550. package/dist/tsconfig.tsbuildinfo +0 -1
  551. package/scripts/beir-bootstrap-significance.mjs +0 -162
  552. package/scripts/bench-rvagent.mjs +0 -312
  553. package/scripts/benchmark-codemods.mjs +0 -127
  554. package/scripts/benchmark-cross-repo.mjs +0 -206
  555. package/scripts/benchmark-pretrained-retrieval.mjs +0 -305
  556. package/scripts/benchmark-self-learning.mjs +0 -241
  557. package/scripts/benchmark-trajectory-mrr.mjs +0 -191
  558. package/scripts/deploy-ipfs-node.sh +0 -153
  559. package/scripts/grid-search-retrieval.mjs +0 -216
  560. package/scripts/pretrain-from-github.mjs +0 -358
  561. package/scripts/publish-registry.ts +0 -345
  562. package/scripts/publish.sh +0 -57
  563. package/scripts/run-beir-bge.mjs +0 -324
  564. package/scripts/run-beir-hybrid.mjs +0 -284
  565. package/scripts/run-beir-lucene-bm25.mjs +0 -154
  566. package/scripts/run-beir-nfcorpus.mjs +0 -231
  567. package/scripts/run-beir-rrf-ablation.mjs +0 -221
  568. package/scripts/setup-ipfs-registry.md +0 -366
@@ -1,324 +0,0 @@
1
- #!/usr/bin/env node
2
- // run-beir-bge.mjs — NFCorpus retrieval using real BGE-base-en-v1.5
3
- // embeddings (bypasses neural-tools' broken agentic-flow path that hash-fell-
4
- // back on darwin-arm64 without sharp/libvips).
5
- //
6
- // Stores doc embeddings in a single .f32 binary file (3633 * 768 * 4 = ~11MB)
7
- // for fast bench iteration without re-embedding.
8
- //
9
- // Usage:
10
- // cd /tmp/beir-nfcorpus
11
- // node /path/to/scripts/run-beir-bge.mjs # full ingest + bench
12
- // SKIP_INGEST=1 node /path/to/scripts/run-beir-bge.mjs # reuse cached embeds
13
- // BGE_MODEL=Xenova/bge-small-en-v1.5 node ... # faster, lower-quality
14
- // BGE_MODEL=Xenova/bge-large-en-v1.5 node ... # slower, higher-quality
15
- //
16
- // ADR-086.
17
-
18
- import { readFileSync, writeFileSync, existsSync, mkdirSync } from 'node:fs';
19
- import { fileURLToPath } from 'node:url';
20
- import { dirname, join, resolve } from 'node:path';
21
- import { performance } from 'node:perf_hooks';
22
-
23
- const SCRIPT_DIR = dirname(fileURLToPath(import.meta.url));
24
- const CLI_ROOT = resolve(SCRIPT_DIR, '..');
25
- const RUFLO_ROOT = resolve(SCRIPT_DIR, '../../../..');
26
- const RUNS_DIR = join(RUFLO_ROOT, 'docs', 'benchmarks', 'runs');
27
-
28
- const DATA_DIR = process.env.BEIR_DATA_DIR || '/tmp/beir-nfcorpus/nfcorpus';
29
- // BUG-FIX (ADR-087): CACHE_DIR was hardcoded to /tmp/beir-nfcorpus/bge-cache,
30
- // which made the SciFact run silently overwrite the NFCorpus cache. Now
31
- // derived from DATA_DIR — each dataset gets its own cache directory.
32
- const CACHE_DIR = join(dirname(DATA_DIR), 'bge-cache');
33
- const BGE_MODEL = process.env.BGE_MODEL || 'Xenova/bge-base-en-v1.5';
34
- const SKIP_INGEST = process.env.SKIP_INGEST === '1';
35
- const MAX_QUERIES = Number(process.env.MAX_QUERIES) || 0;
36
-
37
- // Published BEIR baselines per dataset (nDCG@10).
38
- // Sources: Thakur et al. 2021 (BEIR paper), BGE paper (BAAI 2024),
39
- // SPLADE++ paper, papers-with-code BEIR leaderboards.
40
- const BASELINES_BY_DATASET = {
41
- nfcorpus: {
42
- 'BM25 (Lucene)': 0.325,
43
- 'DocT5query': 0.328,
44
- 'TAS-B': 0.319,
45
- 'GenQ': 0.319,
46
- 'ColBERT': 0.305,
47
- 'Contriever': 0.328,
48
- 'GTR-XL': 0.343,
49
- 'SPLADE++': 0.347,
50
- 'BGE-large-v1.5 (pub)': 0.380,
51
- 'SBERT msmarco': 0.272,
52
- },
53
- scifact: {
54
- 'BM25 (Lucene)': 0.679,
55
- 'DocT5query': 0.675,
56
- 'TAS-B': 0.643,
57
- 'GenQ': 0.644,
58
- 'ColBERT': 0.671,
59
- 'Contriever': 0.677,
60
- 'GTR-XL': 0.662,
61
- 'SPLADE++': 0.704,
62
- 'BGE-large-v1.5 (pub)': 0.722,
63
- 'SBERT msmarco': 0.555,
64
- },
65
- arguana: {
66
- 'BM25 (Lucene)': 0.397,
67
- 'DocT5query': 0.349,
68
- 'TAS-B': 0.429,
69
- 'GenQ': 0.493,
70
- 'ColBERT': 0.233,
71
- 'Contriever': 0.379,
72
- 'GTR-XL': 0.439,
73
- 'SPLADE++': 0.521,
74
- 'BGE-large-v1.5 (pub)': 0.636,
75
- 'SBERT msmarco': 0.371,
76
- },
77
- scidocs: {
78
- 'BM25 (Lucene)': 0.158,
79
- 'DocT5query': 0.162,
80
- 'TAS-B': 0.149,
81
- 'GenQ': 0.143,
82
- 'ColBERT': 0.145,
83
- 'Contriever': 0.165,
84
- 'GTR-XL': 0.174,
85
- 'SPLADE++': 0.159,
86
- 'BGE-large-v1.5 (pub)': 0.225,
87
- 'SBERT msmarco': 0.122,
88
- },
89
- };
90
-
91
- // Auto-detect dataset from DATA_DIR path; default to nfcorpus baselines.
92
- function detectDataset(path) {
93
- const p = path.toLowerCase();
94
- for (const ds of Object.keys(BASELINES_BY_DATASET)) {
95
- if (p.includes(ds)) return ds;
96
- }
97
- return 'nfcorpus';
98
- }
99
-
100
- // ---------------------------------------------------------------------------
101
- // nDCG (graded)
102
- // ---------------------------------------------------------------------------
103
-
104
- function dcg(rels, k) {
105
- let s = 0;
106
- for (let i = 0; i < Math.min(rels.length, k); i++) {
107
- s += (Math.pow(2, rels[i]) - 1) / Math.log2(i + 2);
108
- }
109
- return s;
110
- }
111
- function ndcg(retrieved, qrels, k) {
112
- const rels = retrieved.slice(0, k).map((id) => qrels.get(id) ?? 0);
113
- const ideal = [...qrels.values()].sort((a, b) => b - a).slice(0, k);
114
- const idcg = dcg(ideal, k);
115
- return idcg > 0 ? dcg(rels, k) / idcg : 0;
116
- }
117
- function mrr(retrieved, qrels, k) {
118
- for (let i = 0; i < Math.min(retrieved.length, k); i++) {
119
- if ((qrels.get(retrieved[i]) ?? 0) > 0) return 1 / (i + 1);
120
- }
121
- return 0;
122
- }
123
- function recall(retrieved, qrels, k) {
124
- const tot = [...qrels.values()].filter((v) => v > 0).length;
125
- if (tot === 0) return 0;
126
- let h = 0;
127
- for (let i = 0; i < Math.min(retrieved.length, k); i++) {
128
- if ((qrels.get(retrieved[i]) ?? 0) > 0) h++;
129
- }
130
- return h / tot;
131
- }
132
-
133
- function cosine(a, b) {
134
- let s = 0;
135
- for (let i = 0; i < a.length; i++) s += a[i] * b[i];
136
- return s; // already L2-normalised in BGE
137
- }
138
-
139
- // ---------------------------------------------------------------------------
140
- // Loaders
141
- // ---------------------------------------------------------------------------
142
-
143
- function loadJsonl(path) {
144
- return readFileSync(path, 'utf-8').split('\n').filter(Boolean).map((l) => JSON.parse(l));
145
- }
146
- function loadQrels(path) {
147
- const qrels = new Map();
148
- const lines = readFileSync(path, 'utf-8').split('\n');
149
- for (let i = 1; i < lines.length; i++) {
150
- if (!lines[i].trim()) continue;
151
- const [qid, did, score] = lines[i].split('\t');
152
- if (!qrels.has(qid)) qrels.set(qid, new Map());
153
- qrels.get(qid).set(did, Number(score));
154
- }
155
- return qrels;
156
- }
157
-
158
- // ---------------------------------------------------------------------------
159
- // Embedding cache I/O — single .f32 file + .ids file
160
- // ---------------------------------------------------------------------------
161
-
162
- function cachePath(suffix) {
163
- const safe = BGE_MODEL.replace(/\//g, '_');
164
- return join(CACHE_DIR, `${safe}.${suffix}`);
165
- }
166
-
167
- function saveEmbeddings(ids, embeddings, dim) {
168
- mkdirSync(CACHE_DIR, { recursive: true });
169
- // ids: one per line
170
- writeFileSync(cachePath('ids'), ids.join('\n') + '\n');
171
- // embeddings: concatenated Float32Array
172
- const buf = new Float32Array(ids.length * dim);
173
- for (let i = 0; i < ids.length; i++) buf.set(embeddings[i], i * dim);
174
- writeFileSync(cachePath('f32'), Buffer.from(buf.buffer));
175
- }
176
-
177
- function loadEmbeddings(dim) {
178
- if (!existsSync(cachePath('ids')) || !existsSync(cachePath('f32'))) return null;
179
- const ids = readFileSync(cachePath('ids'), 'utf-8').split('\n').filter(Boolean);
180
- const raw = readFileSync(cachePath('f32'));
181
- const buf = new Float32Array(raw.buffer, raw.byteOffset, raw.length / 4);
182
- const embeds = [];
183
- for (let i = 0; i < ids.length; i++) embeds.push(buf.slice(i * dim, (i + 1) * dim));
184
- return { ids, embeds };
185
- }
186
-
187
- // ---------------------------------------------------------------------------
188
- // Main
189
- // ---------------------------------------------------------------------------
190
-
191
- async function main() {
192
- const dataset = detectDataset(DATA_DIR);
193
- const BASELINES_NDCG10 = BASELINES_BY_DATASET[dataset];
194
- console.log(`# BEIR ${dataset} — BGE embedder (ADR-085/086)`);
195
- console.log(`Model: ${BGE_MODEL}`);
196
- console.log(`Data: ${DATA_DIR}`);
197
-
198
- const corpus = loadJsonl(join(DATA_DIR, 'corpus.jsonl'));
199
- const queries = loadJsonl(join(DATA_DIR, 'queries.jsonl'));
200
- const qrels = loadQrels(join(DATA_DIR, 'qrels/test.tsv'));
201
- console.log(`Corpus: ${corpus.length} docs · Queries: ${queries.length} · Test qrels: ${qrels.size}`);
202
-
203
- const bge = await import(join(CLI_ROOT, 'dist/src/memory/bge-embedder.js'));
204
- const emb = await bge.getBgeEmbedder(BGE_MODEL);
205
- if (!emb) {
206
- console.error('BGE failed to load:', bge.getBgeStatus());
207
- process.exit(1);
208
- }
209
- const dim = emb.dim();
210
- console.log(`BGE loaded (dim=${dim})`);
211
-
212
- // Ingest or load cached doc embeddings.
213
- let docIds, docEmbeds;
214
- const cached = SKIP_INGEST ? loadEmbeddings(dim) : null;
215
- if (cached) {
216
- docIds = cached.ids;
217
- docEmbeds = cached.embeds;
218
- console.log(`Loaded ${docIds.length} cached doc embeddings from ${cachePath('f32')}`);
219
- } else {
220
- console.log(`\nEmbedding ${corpus.length} docs (this is the slow step, ~20 min on M-series CPU)...`);
221
- docIds = corpus.map((d) => d._id);
222
- docEmbeds = new Array(corpus.length);
223
- const tIngest = performance.now();
224
- for (let i = 0; i < corpus.length; i++) {
225
- const d = corpus[i];
226
- const text = `${d.title || ''}\n${d.text || ''}`.slice(0, 4096);
227
- docEmbeds[i] = await emb.embed(text);
228
- if ((i + 1) % 100 === 0) {
229
- const elapsed = (performance.now() - tIngest) / 1000;
230
- const rate = (i + 1) / elapsed;
231
- const eta = (corpus.length - i - 1) / rate;
232
- console.log(` ${i + 1}/${corpus.length} ${rate.toFixed(1)} docs/s ETA ${eta.toFixed(0)}s`);
233
- }
234
- }
235
- console.log(`Embedded ${corpus.length} docs in ${((performance.now() - tIngest) / 1000).toFixed(0)}s`);
236
- saveEmbeddings(docIds, docEmbeds, dim);
237
- console.log(`Cached to ${cachePath('f32')}`);
238
- }
239
-
240
- // Run retrieval per query.
241
- const queriesById = new Map(queries.map((q) => [q._id, q.text]));
242
- const evalQueryIds = MAX_QUERIES
243
- ? [...qrels.keys()].slice(0, MAX_QUERIES)
244
- : [...qrels.keys()];
245
-
246
- // ADR-090: BGE_QUERY_PREFIX=1 enables BAAI's recommended query prefix.
247
- const USE_QUERY_PREFIX = process.env.BGE_QUERY_PREFIX === '1';
248
- console.log(`\nRunning ${evalQueryIds.length} queries${USE_QUERY_PREFIX ? ' (with BGE query prefix, ADR-090)' : ''}...`);
249
- let nSum = 0, mSum = 0, r10Sum = 0, r100Sum = 0, n = 0;
250
- // ADR-086 — save per-query metrics for paired bootstrap significance testing.
251
- const perQuery = [];
252
- const tQ = performance.now();
253
- for (const qid of evalQueryIds) {
254
- const qtext = queriesById.get(qid);
255
- if (!qtext) continue;
256
- // ADR-090: opt-in BGE query prefix per BAAI's docs (+0.009 nDCG@10 on
257
- // NFCorpus dense-alone). Falls back to plain embed() if not enabled.
258
- const qEmb = USE_QUERY_PREFIX && emb.embedQuery ? await emb.embedQuery(qtext) : await emb.embed(qtext);
259
- const scores = new Array(docEmbeds.length);
260
- for (let i = 0; i < docEmbeds.length; i++) scores[i] = { id: docIds[i], score: cosine(qEmb, docEmbeds[i]) };
261
- scores.sort((a, b) => b.score - a.score);
262
- const top100 = scores.slice(0, 100).map((s) => s.id);
263
-
264
- const qmap = qrels.get(qid);
265
- const qNdcg = ndcg(top100, qmap, 10);
266
- const qMrr = mrr(top100, qmap, 10);
267
- const qR10 = recall(top100, qmap, 10);
268
- const qR100 = recall(top100, qmap, 100);
269
- nSum += qNdcg; mSum += qMrr; r10Sum += qR10; r100Sum += qR100; n++;
270
- perQuery.push({ qid, ndcg10: qNdcg, mrr10: qMrr, recall10: qR10, recall100: qR100 });
271
- if (n % 50 === 0) {
272
- const elapsed = (performance.now() - tQ) / 1000;
273
- console.log(` ${n}/${evalQueryIds.length} in ${elapsed.toFixed(0)}s running nDCG@10=${(nSum / n).toFixed(4)}`);
274
- }
275
- }
276
- const queryMs = performance.now() - tQ;
277
-
278
- const ndcg10 = nSum / n;
279
- const mrr10 = mSum / n;
280
- const recall10 = r10Sum / n;
281
- const recall100 = r100Sum / n;
282
-
283
- console.log(`\n=== Results (N=${n}, BGE-base-en-v1.5) ===`);
284
- console.log(` nDCG@10: ${ndcg10.toFixed(4)}`);
285
- console.log(` MRR@10: ${mrr10.toFixed(4)}`);
286
- console.log(` Recall@10: ${recall10.toFixed(4)}`);
287
- console.log(` Recall@100: ${recall100.toFixed(4)}`);
288
- console.log(` Avg query latency: ${(queryMs / n).toFixed(0)}ms`);
289
-
290
- console.log(`\n=== vs published NFCorpus baselines (nDCG@10) ===`);
291
- const ourLabel = `ruflo + ${BGE_MODEL.replace('Xenova/', '')}`;
292
- const ranking = [
293
- ...Object.entries(BASELINES_NDCG10).map(([name, score]) => ({ name, score, ours: false })),
294
- { name: ourLabel, score: ndcg10, ours: true },
295
- ].sort((a, b) => b.score - a.score);
296
- for (const r of ranking) {
297
- console.log(` ${r.score.toFixed(3)} ${r.name}${r.ours ? ' ← us' : ''}`);
298
- }
299
- const ourRank = ranking.findIndex((r) => r.ours) + 1;
300
- console.log(`\n Our rank: ${ourRank} / ${ranking.length}`);
301
-
302
- const summary = {
303
- runAt: new Date().toISOString(),
304
- benchmark: `beir-${dataset}-bge`,
305
- dataset,
306
- model: BGE_MODEL,
307
- queries: n,
308
- corpusSize: corpus.length,
309
- metrics: { ndcg10, mrr10, recall10, recall100, avgQueryLatencyMs: queryMs / n },
310
- perQuery, // ADR-086: per-query metrics for paired bootstrap significance testing
311
- baselines: BASELINES_NDCG10,
312
- ourRank,
313
- leaderboardLength: ranking.length,
314
- };
315
- mkdirSync(RUNS_DIR, { recursive: true });
316
- const stamp = summary.runAt.replace(/[:.]/g, '-');
317
- const safe = BGE_MODEL.replace(/\//g, '_');
318
- writeFileSync(join(RUNS_DIR, `beir-${dataset}-bge-${safe}-${stamp}.json`), JSON.stringify(summary, null, 2));
319
- writeFileSync(join(RUNS_DIR, `beir-${dataset}-bge-latest.json`), JSON.stringify(summary, null, 2));
320
- console.log(`\nWrote ${join(RUNS_DIR, `beir-${dataset}-bge-${safe}-${stamp}.json`)}`);
321
- process.exit(0);
322
- }
323
-
324
- main().catch((err) => { console.error(err); process.exit(1); });
@@ -1,284 +0,0 @@
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_DEFAULT = 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
- // Iter 3: dataset-specific RRF weights for symmetric + dense-favored regimes (arguana: dense 1.6x stronger than BM25).
48
- // Iter 4: nfcorpus medical IR — downweight BM25 (0.7) to favor dense semantics over lexical noise.
49
- // Iter 26: arguana — align with validated nfcorpus/scifact recipe (1.0 dense, 0.2 BM25).
50
- const DATASET_RRF_WEIGHTS = {
51
- arguana: { dense: 1.0, bm25: 0.2 }, // iter 26: match nfcorpus/scifact recipe (aggressive dense)
52
- nfcorpus: { dense: 1.0, bm25: 0.0 }, // iter 14: pure dense fusion (0.2→0.0) RRF with single system + minMax norm preserved
53
- scifact: { dense: 1.0, bm25: 0.05 }, // darwin iter2: small bm25 tie-breaker on top of pure dense (0.0→0.05)
54
- };
55
- function detectDataset(path) {
56
- const p = path.toLowerCase();
57
- for (const ds of Object.keys(BASELINES_BY_DATASET)) if (p.includes(ds)) return ds;
58
- return 'nfcorpus';
59
- }
60
-
61
- 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; }
62
- 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; }
63
- 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; }
64
- 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; }
65
- function cosine(a, b) { let s = 0; for (let i = 0; i < a.length; i++) s += a[i] * b[i]; return s; }
66
- function adaptiveRrfK(corpusSize) { return corpusSize < 20000 ? 40 : 60; } // tighter weighting for small corpora
67
- function adaptiveTopK(corpusSize) { return corpusSize > 150000 ? 2000 : corpusSize > 50000 ? 1000 : 500; } // pool more candidates for large corpora (iter 2)
68
- function minMaxNorm(scores) { const [min, max] = [Math.min(...scores), Math.max(...scores)]; return scores.map((s) => max === min ? 0.5 : (s - min) / (max - min)); }
69
-
70
- function loadJsonl(path) { return readFileSync(path, 'utf-8').split('\n').filter(Boolean).map((l) => JSON.parse(l)); }
71
- 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; }
72
-
73
- function cachePath(suffix) { const safe = BGE_MODEL.replace(/\//g, '_'); return join(CACHE_DIR, `${safe}.${suffix}`); }
74
- function loadEmbeddings(dim) {
75
- if (!existsSync(cachePath('ids')) || !existsSync(cachePath('f32'))) return null;
76
- const ids = readFileSync(cachePath('ids'), 'utf-8').split('\n').filter(Boolean);
77
- const raw = readFileSync(cachePath('f32'));
78
- const buf = new Float32Array(raw.buffer, raw.byteOffset, raw.length / 4);
79
- const embeds = [];
80
- for (let i = 0; i < ids.length; i++) embeds.push(buf.slice(i * dim, (i + 1) * dim));
81
- return { ids, embeds };
82
- }
83
-
84
- async function main() {
85
- const dataset = detectDataset(DATA_DIR);
86
- const BASELINES_NDCG10 = BASELINES_BY_DATASET[dataset];
87
- const corpus = loadJsonl(join(DATA_DIR, 'corpus.jsonl'));
88
- const RRF_K = adaptiveRrfK(corpus.length); // adaptive k based on corpus size (iter 1: normalize scores + tighter k for small corpora)
89
-
90
- const weights = DATASET_RRF_WEIGHTS[dataset] || { dense: 1.0, bm25: 1.0 };
91
- console.log(`# BEIR ${dataset} — hybrid RRF${RERANK ? ' + cross-encoder rerank' : ''} (ADR-087 + iter1 + iter2 + iter3)`);
92
- console.log(`Data: ${DATA_DIR}`);
93
- console.log(`Dense: ${BGE_MODEL}`);
94
- console.log(`RRF k: ${RRF_K} (adaptive for corpus size ${corpus.length})${RERANK ? `, rerank top-${RERANK_TOP_K}` : ''}`);
95
- console.log(`Normalization: min-max before RRF fusion`);
96
- console.log(`Candidate pool: top-${adaptiveTopK(corpus.length)} per system (iter 2: scaled for large corpora)`);
97
- console.log(`RRF weights: dense=${weights.dense} bm25=${weights.bm25} (iter 3: dataset-specific for ${dataset})`);
98
- const queries = loadJsonl(join(DATA_DIR, 'queries.jsonl'));
99
- const qrels = loadQrels(join(DATA_DIR, 'qrels/test.tsv'));
100
- console.log(`Corpus: ${corpus.length} docs · Test qrels: ${qrels.size}`);
101
-
102
- // BGE embeddings — must be pre-cached from run-beir-bge.mjs.
103
- const bge = await import(join(CLI_ROOT, 'dist/src/memory/bge-embedder.js'));
104
- const emb = await bge.getBgeEmbedder(BGE_MODEL);
105
- if (!emb) { console.error('BGE failed:', bge.getBgeStatus()); process.exit(1); }
106
- const dim = emb.dim();
107
- const cached = loadEmbeddings(dim);
108
- if (!cached) { console.error(`No cached BGE embeddings at ${cachePath('f32')}. Run run-beir-bge.mjs first.`); process.exit(2); }
109
- const docIds = cached.ids;
110
- const docEmbeds = cached.embeds;
111
- console.log(`Loaded ${docIds.length} cached BGE embeddings`);
112
-
113
- // BM25 setup — USE_LUCENE_BM25=1 uses Porter+Lucene-stopword BM25 (matches
114
- // published BEIR baselines); default is hybrid-retrieval's multi-field BM25.
115
- const USE_LUCENE_BM25 = process.env.USE_LUCENE_BM25 === '1';
116
- let bm25Score, tokenizeFn, titleDocs, textDocs, titleStats, textStats, luceneDocs, luceneStats;
117
- if (USE_LUCENE_BM25) {
118
- const { luceneTokenize, buildLuceneCorpusStats, luceneBM25 } = await import(join(CLI_ROOT, 'dist/src/memory/lucene-bm25.js'));
119
- tokenizeFn = luceneTokenize;
120
- luceneDocs = corpus.map((d) => luceneTokenize(`${d.title || ''} ${d.text || ''}`));
121
- luceneStats = buildLuceneCorpusStats(luceneDocs);
122
- bm25Score = (qTokens, idx) => luceneBM25(qTokens, luceneDocs[idx], luceneStats);
123
- console.log(`BM25: Lucene-style (Porter stem + Lucene stopwords + length norm) — ADR-088`);
124
- } else {
125
- const { tokenize, buildCorpusStats, multiFieldBM25 } = await import(join(CLI_ROOT, 'dist/src/memory/hybrid-retrieval.js'));
126
- tokenizeFn = tokenize;
127
- titleDocs = corpus.map((d) => tokenize(d.title || ''));
128
- textDocs = corpus.map((d) => tokenize(d.text || ''));
129
- titleStats = buildCorpusStats(titleDocs);
130
- textStats = buildCorpusStats(textDocs);
131
- const SUBJECT_WEIGHT = Number(process.env.SUBJECT_WEIGHT ?? 1.0);
132
- const BODY_WEIGHT = Number(process.env.BODY_WEIGHT ?? 1.0);
133
- bm25Score = (qTokens, idx) => multiFieldBM25(qTokens, titleDocs[idx], textDocs[idx], titleStats, textStats, SUBJECT_WEIGHT, BODY_WEIGHT);
134
- console.log(`BM25: multi-field (title sw=${SUBJECT_WEIGHT}, text bw=${BODY_WEIGHT})`);
135
- }
136
-
137
- // Optional cross-encoder reranker.
138
- let crossEncoder = null;
139
- if (RERANK) {
140
- const ce = await import(join(CLI_ROOT, 'dist/src/memory/cross-encoder-rerank.js'));
141
- const ceFn = await ce.getCrossEncoder('Xenova/ms-marco-MiniLM-L-6-v2');
142
- if (!ceFn) { console.error('Cross-encoder failed to load:', ce.getCrossEncoderStatus()); process.exit(3); }
143
- crossEncoder = ceFn;
144
- console.log('Cross-encoder loaded: Xenova/ms-marco-MiniLM-L-6-v2');
145
- }
146
-
147
- const queriesById = new Map(queries.map((q) => [q._id, q.text]));
148
- const evalQueryIds = MAX_QUERIES ? [...qrels.keys()].slice(0, MAX_QUERIES) : [...qrels.keys()];
149
-
150
- // Build doc id → corpus index lookup (BM25 returns indices, dense returns ids).
151
- const idToIdx = new Map(docIds.map((id, i) => [id, i]));
152
- // Need a reverse map too: corpus index → doc id (for BM25 over corpus order).
153
- const corpusIdToIdx = new Map(corpus.map((d, i) => [d._id, i]));
154
- const corpusIdxToId = corpus.map((d) => d._id);
155
-
156
- console.log(`\nRunning ${evalQueryIds.length} queries...`);
157
- let nSum = 0, mSum = 0, r10Sum = 0, r100Sum = 0, n = 0;
158
- const perQuery = [];
159
- const tQ = performance.now();
160
-
161
- for (const qid of evalQueryIds) {
162
- const qtext = queriesById.get(qid);
163
- if (!qtext) continue;
164
-
165
- // §1 — dense BGE retrieval (ADR-090: opt-in query prefix when BGE_QUERY_PREFIX=1)
166
- const qEmb = (process.env.BGE_QUERY_PREFIX === '1' && emb.embedQuery)
167
- ? await emb.embedQuery(qtext)
168
- : await emb.embed(qtext);
169
- const denseScored = new Array(docEmbeds.length);
170
- for (let i = 0; i < docEmbeds.length; i++) denseScored[i] = { id: docIds[i], score: cosine(qEmb, docEmbeds[i]) };
171
- denseScored.sort((a, b) => b.score - a.score);
172
-
173
- // §2 — BM25 retrieval (Lucene or multi-field depending on USE_LUCENE_BM25)
174
- const qTokens = tokenizeFn(qtext);
175
- const bm25Scored = new Array(corpus.length);
176
- for (let i = 0; i < corpus.length; i++) {
177
- bm25Scored[i] = { id: corpusIdxToId[i], score: bm25Score(qTokens, i) };
178
- }
179
- bm25Scored.sort((a, b) => b.score - a.score);
180
-
181
- // §3 — RRF fusion: normalize scores first, then score = sum over systems of 1/(k + rank).
182
- // Min-max normalize each system's scores to [0,1] for fair fusion (iter 1 optimization).
183
- // For large corpora, pull more candidates before fusion (iter 2 optimization).
184
- const TOP_PER_SYSTEM = adaptiveTopK(corpus.length);
185
- const denseTopK = denseScored.slice(0, Math.min(TOP_PER_SYSTEM, denseScored.length));
186
- const bm25TopK = bm25Scored.slice(0, Math.min(TOP_PER_SYSTEM, bm25Scored.length)).filter((s) => s.score > 0);
187
- const denseScoresNorm = minMaxNorm(denseTopK.map((s) => s.score));
188
- const bm25ScoresNorm = minMaxNorm(bm25TopK.map((s) => s.score));
189
- denseTopK.forEach((d, i) => { d.scoreNorm = denseScoresNorm[i]; });
190
- bm25TopK.forEach((d, i) => { d.scoreNorm = bm25ScoresNorm[i]; });
191
-
192
- const rrfScores = new Map();
193
- const weights = DATASET_RRF_WEIGHTS[dataset] || { dense: 1.0, bm25: 1.0 }; // iter 3: dataset-specific weights
194
- for (let r = 0; r < denseTopK.length; r++) {
195
- const id = denseTopK[r].id;
196
- rrfScores.set(id, (rrfScores.get(id) || 0) + weights.dense / (RRF_K + r + 1));
197
- }
198
- for (let r = 0; r < bm25TopK.length; r++) {
199
- const id = bm25TopK[r].id;
200
- rrfScores.set(id, (rrfScores.get(id) || 0) + weights.bm25 / (RRF_K + r + 1));
201
- }
202
- const fused = [...rrfScores.entries()]
203
- .map(([id, score]) => ({ id, score }))
204
- .sort((a, b) => b.score - a.score)
205
- .slice(0, RERANK ? RERANK_TOP_K : 100);
206
-
207
- let final = fused;
208
-
209
- // §4 — optional cross-encoder rerank.
210
- if (crossEncoder && fused.length > 0) {
211
- const docsForRerank = fused.map(({ id }) => {
212
- const ci = corpusIdToIdx.get(id);
213
- const d = corpus[ci];
214
- return `${d.title || ''} ${d.text || ''}`.slice(0, 4096);
215
- });
216
- const ceScores = await crossEncoder.scoreBatch(qtext, docsForRerank);
217
- final = fused
218
- .map((f, i) => ({ id: f.id, score: ceScores[i], rrf: f.score }))
219
- .sort((a, b) => b.score - a.score);
220
- }
221
-
222
- const top100 = final.slice(0, 100).map((s) => s.id);
223
- const qmap = qrels.get(qid);
224
- const qNdcg = ndcg(top100, qmap, 10);
225
- const qMrr = mrr(top100, qmap, 10);
226
- const qR10 = recall(top100, qmap, 10);
227
- const qR100 = recall(top100, qmap, 100);
228
- nSum += qNdcg; mSum += qMrr; r10Sum += qR10; r100Sum += qR100; n++;
229
- perQuery.push({ qid, ndcg10: qNdcg, mrr10: qMrr, recall10: qR10, recall100: qR100 });
230
- if (n % 50 === 0) {
231
- const elapsed = (performance.now() - tQ) / 1000;
232
- console.log(` ${n}/${evalQueryIds.length} in ${elapsed.toFixed(0)}s running nDCG@10=${(nSum / n).toFixed(4)}`);
233
- }
234
- }
235
- const queryMs = performance.now() - tQ;
236
-
237
- const ndcg10 = nSum / n;
238
- const mrr10 = mSum / n;
239
- const recall10 = r10Sum / n;
240
- const recall100 = r100Sum / n;
241
-
242
- console.log(`\n=== Results (N=${n}, ${dataset}, BM25+BGE-base RRF k=${RRF_K}${RERANK ? ' + CE rerank' : ''}) ===`);
243
- console.log(` nDCG@10: ${ndcg10.toFixed(4)}`);
244
- console.log(` MRR@10: ${mrr10.toFixed(4)}`);
245
- console.log(` Recall@10: ${recall10.toFixed(4)}`);
246
- console.log(` Recall@100: ${recall100.toFixed(4)}`);
247
- console.log(` Avg query latency: ${(queryMs / n).toFixed(0)}ms`);
248
-
249
- const ourLabel = `ruflo + BM25+BGE-base RRF${RERANK ? '+CE' : ''}`;
250
- const ranking = [
251
- ...Object.entries(BASELINES_NDCG10).map(([name, score]) => ({ name, score, ours: false })),
252
- { name: ourLabel, score: ndcg10, ours: true },
253
- ].sort((a, b) => b.score - a.score);
254
- console.log(`\n=== vs ${dataset} listed baselines ===`);
255
- for (const r of ranking) console.log(` ${r.score.toFixed(3)} ${r.name}${r.ours ? ' ← us' : ''}`);
256
- const ourRank = ranking.findIndex((r) => r.ours) + 1;
257
- console.log(`\n Our rank: ${ourRank} / ${ranking.length}`);
258
-
259
- const summary = {
260
- runAt: new Date().toISOString(),
261
- benchmark: `beir-${dataset}-hybrid${RERANK ? '-rerank' : ''}`,
262
- dataset,
263
- pipeline: `BM25+BGE RRF${RERANK ? '+CE-rerank' : ''}`,
264
- bgeModel: BGE_MODEL,
265
- rrfK: RRF_K,
266
- rerank: RERANK,
267
- queries: n,
268
- corpusSize: corpus.length,
269
- metrics: { ndcg10, mrr10, recall10, recall100, avgQueryLatencyMs: queryMs / n },
270
- perQuery,
271
- baselines: BASELINES_NDCG10,
272
- ourRank,
273
- leaderboardLength: ranking.length,
274
- };
275
- mkdirSync(RUNS_DIR, { recursive: true });
276
- const stamp = summary.runAt.replace(/[:.]/g, '-');
277
- const suffix = RERANK ? 'hybrid-rerank' : 'hybrid-rrf';
278
- writeFileSync(join(RUNS_DIR, `beir-${dataset}-${suffix}-${stamp}.json`), JSON.stringify(summary, null, 2));
279
- writeFileSync(join(RUNS_DIR, `beir-${dataset}-${suffix}-latest.json`), JSON.stringify(summary, null, 2));
280
- console.log(`\nWrote ${join(RUNS_DIR, `beir-${dataset}-${suffix}-${stamp}.json`)}`);
281
- process.exit(0);
282
- }
283
-
284
- main().catch((err) => { console.error(err); process.exit(1); });