@modusensus/dsh-mneme 0.8.13 → 0.8.15
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/LICENSE +21 -21
- package/README.md +655 -626
- package/assets/icon.svg +3 -3
- package/bin/cli.mjs +689 -689
- package/bin/dsh-mneme-mcp.mjs +510 -510
- package/bin/dsh-mneme-serve.mjs +151 -0
- package/cordis.patch.yml +21 -21
- package/docs/CONFIGURATION.md +263 -0
- package/docs/CONTINUITY.md +124 -0
- package/docs/DAEMON.md +46 -0
- package/docs/ENTITIES.md +245 -0
- package/docs/LOCAL_MODEL.md +222 -0
- package/docs/SEMANTIC.md +256 -0
- package/docs/SLEEP.md +164 -0
- package/docs/STORAGE.md +89 -0
- package/lib/api-standalone.js +673 -572
- package/lib/api.js +1437 -1433
- package/lib/bootstrap.js +211 -211
- package/lib/client.js +5444 -5295
- package/lib/commands.js +95 -95
- package/lib/config.js +880 -836
- package/lib/content-hash.js +54 -54
- package/lib/continuity.js +220 -220
- package/lib/document-index.js +94 -94
- package/lib/document.js +303 -303
- package/lib/dream/clustering.js +118 -118
- package/lib/dream/decisions.js +674 -674
- package/lib/dream/narratives.js +44 -44
- package/lib/dream/sleep.js +819 -819
- package/lib/dream.js +1501 -1501
- package/lib/embedding.js +177 -177
- package/lib/entities/extractor.js +252 -252
- package/lib/graph/anchoring.js +179 -179
- package/lib/guide.js +74 -74
- package/lib/heat.js +144 -144
- package/lib/hot-memory.js +46 -46
- package/lib/index.js +535 -671
- package/lib/inject.js +464 -450
- package/lib/lang.js +248 -229
- package/lib/local-embedder.js +358 -358
- package/lib/maintenance.js +142 -142
- package/lib/mirror.js +330 -330
- package/lib/organize.js +450 -450
- package/lib/peak-hours.js +117 -117
- package/lib/placement.js +13 -0
- package/lib/pre-inject-gate.js +340 -0
- package/lib/quality-filter.js +126 -126
- package/lib/recall-stats.js +175 -175
- package/lib/reranker.js +269 -269
- package/lib/runtime/adopt.js +176 -176
- package/lib/runtime/closure.js +172 -172
- package/lib/runtime/download.js +247 -247
- package/lib/runtime/layout.js +265 -265
- package/lib/runtime/loader.js +252 -252
- package/lib/runtime/model-fetch.js +361 -361
- package/lib/runtime/provision.js +237 -237
- package/lib/runtime/tarball.js +130 -130
- package/lib/runtime/verify.js +223 -204
- package/lib/scope.js +146 -146
- package/lib/search/adaptive.js +60 -60
- package/lib/search/bm25.js +96 -96
- package/lib/semantic.js +229 -0
- package/lib/sensitive-scan.js +209 -209
- package/lib/serve.js +215 -0
- package/lib/service.js +2663 -2596
- package/lib/settings.js +517 -509
- package/lib/store.js +3007 -3007
- package/lib/summarize.js +891 -856
- package/lib/tools.js +743 -737
- package/lib/vector-index.js +110 -110
- package/lib/version-check.js +122 -122
- package/lib/write-admission.js +402 -402
- package/locale/en.json +6 -6
- package/locale/zh.json +6 -6
- package/package.json +8 -4
- package/runtime-manifest.json +526 -526
- package/scripts/benchmark-embed.js +201 -201
- package/scripts/benchmark-recall.js +249 -249
- package/scripts/benchmark-rerank.js +166 -166
- package/scripts/build-runtime-manifest.mjs +224 -224
- package/scripts/changelog-prep.mjs +36 -0
- package/scripts/check-sync.js +42 -42
- package/scripts/e2e-dsh.js +220 -220
- package/scripts/mneme-runtime.mjs +277 -277
- package/scripts/stress-dsh.js +255 -255
- package/scripts/sync-lib.js +47 -47
- package/scripts/sync-test-badge.mjs +69 -69
- package/scripts/test-count-sync.mjs +44 -44
- package/src/api-standalone.js +673 -572
- package/src/api.js +1437 -1433
- package/src/bootstrap.js +211 -211
- package/src/commands.js +95 -95
- package/src/config.js +880 -836
- package/src/content-hash.js +54 -54
- package/src/continuity.js +220 -220
- package/src/document-index.js +94 -94
- package/src/document.js +303 -303
- package/src/dream/clustering.js +118 -118
- package/src/dream/decisions.js +674 -674
- package/src/dream/narratives.js +44 -44
- package/src/dream/sleep.js +819 -819
- package/src/dream.js +1501 -1501
- package/src/embedding.js +177 -177
- package/src/entities/extractor.js +252 -252
- package/src/graph/anchoring.js +179 -179
- package/src/guide.js +74 -74
- package/src/heat.js +144 -144
- package/src/hot-memory.js +46 -46
- package/src/index.js +535 -671
- package/src/inject.js +464 -450
- package/src/lang.js +248 -229
- package/src/local-embedder.js +358 -358
- package/src/maintenance.js +142 -142
- package/src/mirror.js +330 -330
- package/src/organize.js +450 -450
- package/src/peak-hours.js +117 -117
- package/src/placement.js +13 -0
- package/src/pre-inject-gate.js +340 -0
- package/src/quality-filter.js +126 -126
- package/src/recall-stats.js +175 -175
- package/src/reranker.js +269 -269
- package/src/runtime/adopt.js +176 -176
- package/src/runtime/closure.js +172 -172
- package/src/runtime/download.js +247 -247
- package/src/runtime/layout.js +265 -265
- package/src/runtime/loader.js +252 -252
- package/src/runtime/model-fetch.js +361 -361
- package/src/runtime/provision.js +237 -237
- package/src/runtime/tarball.js +130 -130
- package/src/runtime/verify.js +223 -204
- package/src/scope.js +146 -146
- package/src/search/adaptive.js +60 -60
- package/src/search/bm25.js +96 -96
- package/src/semantic.js +229 -0
- package/src/sensitive-scan.js +209 -209
- package/src/serve.js +215 -0
- package/src/service.js +2663 -2596
- package/src/settings.js +517 -509
- package/src/store.js +3007 -3007
- package/src/summarize.js +891 -856
- package/src/tools.js +743 -737
- package/src/vector-index.js +110 -110
- package/src/version-check.js +122 -122
- package/src/write-admission.js +402 -402
- package/test/adaptive-inject-budget.test.js +85 -85
- package/test/api-routes-gaps.test.js +218 -218
- package/test/api.test.js +1702 -1688
- package/test/audit.test.js +448 -448
- package/test/baseline-persist.test.js +248 -248
- package/test/benchmark.test.js +94 -94
- package/test/bootstrap.test.js +220 -220
- package/test/changelog-prep.test.js +60 -0
- package/test/client.test.js +1230 -1169
- package/test/clustering.test.js +100 -100
- package/test/commands.test.js +106 -106
- package/test/config.test.js +66 -66
- package/test/conflict-freeze.test.js +300 -300
- package/test/content-hash.test.js +204 -204
- package/test/continuity.test.js +333 -333
- package/test/document-dir.test.js +233 -233
- package/test/document.test.js +671 -671
- package/test/dream-archive-guardrail.test.js +116 -116
- package/test/dream-candidate-hybrid.test.js +281 -281
- package/test/dream-coverage-degrade.test.js +86 -86
- package/test/dream-failure-backoff.test.js +173 -173
- package/test/dream-peak-hours.test.js +194 -194
- package/test/dream.test.js +1385 -1383
- package/test/embedding-ready.test.js +196 -196
- package/test/entities.test.js +627 -547
- package/test/entity-recall.test.js +179 -179
- package/test/epistemic.test.js +325 -325
- package/test/fnew-0112.test.js +311 -311
- package/test/fnew-03.test.js +422 -422
- package/test/graph-anchoring.test.js +184 -184
- package/test/graph-api.test.js +175 -175
- package/test/graph-inject-budget.test.js +103 -103
- package/test/graph-passive-confirm.test.js +91 -91
- package/test/graph-weight-evolution.test.js +104 -104
- package/test/heat.test.js +175 -175
- package/test/helpers/dream-mock.js +82 -82
- package/test/helpers/peer-worker.mjs +25 -25
- package/test/helpers/runtime-source.js +60 -60
- package/test/helpers/tar-builder.js +80 -80
- package/test/helpers/write-admission-samples.js +308 -308
- package/test/hot-memory.test.js +145 -145
- package/test/inject-content-cap.test.js +91 -91
- package/test/inject-first-round-bm25.test.js +89 -89
- package/test/inject-heat.test.js +77 -61
- package/test/inject-parent-gate.test.js +106 -106
- package/test/inject-pools.test.js +194 -194
- package/test/inject-recall.test.js +105 -105
- package/test/inject-rotation.test.js +161 -161
- package/test/inject.test.js +420 -420
- package/test/lang.test.js +87 -87
- package/test/lastrunat-persist.test.js +101 -101
- package/test/lib-smoke.test.js +96 -96
- package/test/llm-audit.test.js +598 -598
- package/test/llm-message-source.test.js +207 -207
- package/test/local-embedder.test.js +292 -292
- package/test/maintenance.test.js +391 -391
- package/test/mcp-stdio.test.js +385 -385
- package/test/mirror-dirty.test.js +424 -424
- package/test/mirror-edit-digest.test.js +187 -187
- package/test/mirror-generation.test.js +499 -499
- package/test/mirror-types.test.js +208 -208
- package/test/mirror.test.js +249 -249
- package/test/narrative-bars.test.js +162 -162
- package/test/organize.test.js +418 -418
- package/test/peer-blockers.test.js +193 -193
- package/test/placement.test.js +36 -0
- package/test/policy-epoch.test.js +259 -259
- package/test/pre-inject-gate.test.js +434 -0
- package/test/quality-filter.test.js +186 -186
- package/test/reasoning-effort.test.js +752 -752
- package/test/recall-anti-update.test.js +82 -82
- package/test/recall-evals.test.js +253 -253
- package/test/recall-layer.test.js +326 -326
- package/test/recall-stats.test.js +189 -189
- package/test/receipt-chain.test.js +451 -451
- package/test/reflection.test.js +226 -226
- package/test/reindex-backfill.test.js +62 -62
- package/test/reranker-single-logit.test.js +73 -73
- package/test/reranker.test.js +197 -197
- package/test/resident-status.test.js +83 -83
- package/test/runtime-adopt.test.js +206 -206
- package/test/runtime-cli.test.js +89 -89
- package/test/runtime-closure.test.js +193 -193
- package/test/runtime-download.test.js +302 -302
- package/test/runtime-layout.test.js +163 -163
- package/test/runtime-loader.test.js +255 -255
- package/test/runtime-manifest.test.js +159 -159
- package/test/runtime-model-fetch.test.js +348 -348
- package/test/runtime-provision.test.js +101 -101
- package/test/runtime-tarball.test.js +81 -81
- package/test/runtime-verify.test.js +319 -279
- package/test/scope-candidates.test.js +239 -239
- package/test/scope-explicit.test.js +452 -452
- package/test/scope-inject-soft.test.js +106 -106
- package/test/scope-retrieval.test.js +197 -197
- package/test/scope-storage.test.js +255 -255
- package/test/scope-strict.test.js +367 -343
- package/test/search-fusion.test.js +90 -90
- package/test/semantic.test.js +124 -124
- package/test/sensitive-scan.test.js +207 -207
- package/test/serve-bin.test.js +121 -0
- package/test/serve.test.js +196 -0
- package/test/service-search.test.js +174 -174
- package/test/service.test.js +405 -405
- package/test/settings.test.js +261 -261
- package/test/skip-invalid-restore.test.js +262 -262
- package/test/sleep-action-set.test.js +422 -422
- package/test/sleep-heat.test.js +126 -126
- package/test/sleep-scheduler-rearm.test.js +89 -89
- package/test/sleep.test.js +527 -527
- package/test/standalone-api.test.js +751 -544
- package/test/store-vector-cache.test.js +98 -98
- package/test/store.test.js +396 -396
- package/test/stress.test.js +209 -209
- package/test/summarize-occurred-at.test.js +127 -0
- package/test/summarize-reasoning-effort.test.js +189 -189
- package/test/summarize.test.js +1465 -1465
- package/test/supersede-state-tracking.test.js +394 -0
- package/test/test-count-sync.test.js +60 -60
- package/test/tools-dto-schema-matrix.test.js +325 -325
- package/test/tools-memory-get-schema.test.js +65 -65
- package/test/tools.test.js +428 -374
- package/test/updated-at-semantics.test.js +113 -113
- package/test/vector-index.test.js +221 -221
- package/test/version-check.test.js +152 -152
- package/test/webServer-optional.test.js +82 -82
- package/test/write-admission.test.js +668 -668
package/lib/scope.js
CHANGED
|
@@ -1,146 +1,146 @@
|
|
|
1
|
-
// v0.8.0 A1(issue #17):写入端 scope 解析。
|
|
2
|
-
//
|
|
3
|
-
// 身份模型(issue #17 的核心约定):session 只是「载体」,记忆挂在持久化身份
|
|
4
|
-
// 标签上——agent_scope = session header 的 agentPreset(跨会话稳定的 agent 模板
|
|
5
|
-
// 身份),workspace_scope = 会话工作目录(registry 反查的 canonical path 优先,
|
|
6
|
-
// header.cwd 兜底)。session_id 永不作为作用域键。
|
|
7
|
-
//
|
|
8
|
-
// 硬约束:解析失败绝不阻塞写入——任何一步取不到就落 NULL(读侧 NULL = 未标注
|
|
9
|
-
// = 全局可见),任何异常都被吞掉(registry 故障 warn 一次后降级)。
|
|
10
|
-
//
|
|
11
|
-
// v0.8.1(issue #170 复核项 2):scopeEnabled 的闸门从解析器挪到「写标注」处
|
|
12
|
-
// (memory_save 的 auto 分支)。解析器恒返回会话身份(真取不到才是 null)——
|
|
13
|
-
// 读取侧(strictVisibility / memory_get / inject)与显式声明都依赖它:flag 关
|
|
14
|
-
// 只关「自动标注」,不能连读取身份一起关,否则 strictScope 硬墙被静默拆掉
|
|
15
|
-
// (fail-open)。
|
|
16
|
-
|
|
17
|
-
/**
|
|
18
|
-
* @param {object} deps
|
|
19
|
-
* @param {object} [deps.ctx] 宿主上下文(可能带 workspaceRegistry;旧版 dsh 没有)
|
|
20
|
-
* @param {object} [deps.config] 已解析配置(读 scopeEnabled)
|
|
21
|
-
* @param {object} [deps.logger]
|
|
22
|
-
* @returns {(exec: object) => {agent_scope: string|null, workspace_scope: string|null}|null}
|
|
23
|
-
*/
|
|
24
|
-
export function createScopeResolver({ ctx, config, logger } = {}) {
|
|
25
|
-
let warned = false;
|
|
26
|
-
const warnOnce = (msg) => {
|
|
27
|
-
if (warned) return;
|
|
28
|
-
warned = true;
|
|
29
|
-
try {
|
|
30
|
-
logger?.warn?.(msg);
|
|
31
|
-
} catch {
|
|
32
|
-
// logger 自身故障同样不能影响写入路径
|
|
33
|
-
}
|
|
34
|
-
};
|
|
35
|
-
|
|
36
|
-
return function resolveWriteScope(exec) {
|
|
37
|
-
const session = sessionOf(exec);
|
|
38
|
-
|
|
39
|
-
// 持久化 header(issue #17 的身份契约):session.header 是创建时冻结的
|
|
40
|
-
// SessionHeader(含 agentPreset/cwd)。⚠️ 不能先读 requestHeader()——那返回
|
|
41
|
-
// EpochHeader(请求级模型路由头,首个请求后即为真值),会把 session.header
|
|
42
|
-
// 遮蔽掉,agentPreset/cwd 双双取空(本地 web 复验 2026-09-13 实测踩中)。
|
|
43
|
-
// 仅当宿主没有 .header 属性时才降级尝试 requestHeader().config(旧形状)。
|
|
44
|
-
let header = null;
|
|
45
|
-
try {
|
|
46
|
-
header = session?.header ?? session?.requestHeader?.()?.config ?? null;
|
|
47
|
-
} catch {
|
|
48
|
-
header = null;
|
|
49
|
-
}
|
|
50
|
-
|
|
51
|
-
let agentScope = null;
|
|
52
|
-
const preset = header?.agentPreset;
|
|
53
|
-
if (typeof preset === "string" && preset.trim()) agentScope = preset.trim();
|
|
54
|
-
|
|
55
|
-
// workspace:registry 反查优先(成员资格 = id 命中 + header canonical cwd
|
|
56
|
-
// 校验过,path 已 realpath 归一),失败回退 header.cwd,再失败留 NULL。
|
|
57
|
-
let workspaceScope = null;
|
|
58
|
-
try {
|
|
59
|
-
const list = ctx?.workspaceRegistry?.list?.();
|
|
60
|
-
const sessionId = session?.id;
|
|
61
|
-
if (Array.isArray(list) && sessionId != null) {
|
|
62
|
-
const hit = list.find(
|
|
63
|
-
(ws) => Array.isArray(ws?.sessionIds) && ws.sessionIds.includes(sessionId)
|
|
64
|
-
);
|
|
65
|
-
if (typeof hit?.path === "string" && hit.path.trim()) workspaceScope = hit.path.trim();
|
|
66
|
-
}
|
|
67
|
-
} catch (e) {
|
|
68
|
-
warnOnce(
|
|
69
|
-
`[dsh-mneme] workspaceRegistry lookup failed, falling back to session header cwd: ${String(e)}`
|
|
70
|
-
);
|
|
71
|
-
}
|
|
72
|
-
if (!workspaceScope) {
|
|
73
|
-
const cwd = header?.cwd;
|
|
74
|
-
if (typeof cwd === "string" && cwd.trim()) workspaceScope = cwd.trim();
|
|
75
|
-
}
|
|
76
|
-
|
|
77
|
-
return { agent_scope: agentScope, workspace_scope: workspaceScope };
|
|
78
|
-
};
|
|
79
|
-
}
|
|
80
|
-
|
|
81
|
-
/**
|
|
82
|
-
* 工具 exec 上的会话对象。宿主 ctx 是 Proxy(cordis 4),裸读未 inject 的属性会
|
|
83
|
-
* 抛错——所以取值恒包在 try 里,取不到返回 null。scope 解析与写入准入的会话键
|
|
84
|
-
* 共用这一处取值口径。
|
|
85
|
-
* @param {object} exec
|
|
86
|
-
* @returns {object|null}
|
|
87
|
-
*/
|
|
88
|
-
export function sessionOf(exec) {
|
|
89
|
-
try {
|
|
90
|
-
return exec?.agent?.session ?? null;
|
|
91
|
-
} catch {
|
|
92
|
-
return null;
|
|
93
|
-
}
|
|
94
|
-
}
|
|
95
|
-
|
|
96
|
-
/**
|
|
97
|
-
* 写入准入的会话键(#254 G1 会话写入预算的计数单位)。session_id 在这里只作
|
|
98
|
-
* 「同一段对话」的标识,不作为作用域键(见文件头:记忆挂在持久化身份标签上);
|
|
99
|
-
* 取不到会话身份(系统写入)返回 null = 不进预算。
|
|
100
|
-
* @param {object} exec
|
|
101
|
-
* @returns {string|null}
|
|
102
|
-
*/
|
|
103
|
-
export function sessionKeyOf(exec) {
|
|
104
|
-
// 整个取值(含 .id)都在 try 里:宿主 session 可能是会抛错的 accessor,而这个
|
|
105
|
-
// 函数在 memory_save 的保存之前被调用——抛出去就是一次工具调用直接失败。
|
|
106
|
-
try {
|
|
107
|
-
const id = sessionOf(exec)?.id;
|
|
108
|
-
return id == null ? null : String(id);
|
|
109
|
-
} catch {
|
|
110
|
-
return null;
|
|
111
|
-
}
|
|
112
|
-
}
|
|
113
|
-
|
|
114
|
-
/**
|
|
115
|
-
* v0.8.1 底座(issue #170):显式 scope 参数(memory_save / memory_update /
|
|
116
|
-
* 面板编辑)的归一化。约定:
|
|
117
|
-
* - 省略(undefined,由调用方判断)= 该维不声明,回落载体自动标注;
|
|
118
|
-
* - "global"(大小写不敏感)/ "*" / 空白 = 显式声明全局 → 存储 NULL
|
|
119
|
-
* (行上的 agent_scope_source='explicit' 承担「显式全局」与「从未标注」
|
|
120
|
-
* 的区分,NULL 本身继续表示全局可见);
|
|
121
|
-
* - 其余非空 trim 字符串 = 收窄到该标签。
|
|
122
|
-
* 脏输入(非字符串)落 NULL(等同声明全局)——与 resolveWriteScope 的「解析
|
|
123
|
-
* 绝不阻塞写入」同一哲学:显式声明是用户意图,形状不对时宁可放宽不报错。
|
|
124
|
-
*
|
|
125
|
-
* @param {unknown} raw 工具/HTTP 参数原值
|
|
126
|
-
* @returns {string|null} 归一化后的标签值(null=显式全局)
|
|
127
|
-
*/
|
|
128
|
-
export function normalizeExplicitScope(raw) {
|
|
129
|
-
if (typeof raw !== "string") return null;
|
|
130
|
-
const s = raw.trim();
|
|
131
|
-
if (!s || s === "*" || s.toLowerCase() === "global") return null;
|
|
132
|
-
return s;
|
|
133
|
-
}
|
|
134
|
-
|
|
135
|
-
/**
|
|
136
|
-
* scope 标注的比较键(去重键/检索加权/跨 scope 配对共用口径):非空 trim 字符串
|
|
137
|
-
* 原样返回,其余(NULL/undefined/空串/非字符串)统一为 null(「未标注」)。
|
|
138
|
-
* v0.8.1(issue #170 第 2 步)从 service.js 上提为共享定义——sleep 的跨 scope
|
|
139
|
-
* 候选配对必须与去重键用同一把钥匙,NULL 与已标注才不会错配。
|
|
140
|
-
*
|
|
141
|
-
* @param {unknown} v scope 标注原值
|
|
142
|
-
* @returns {string|null} 比较键(null=未标注)
|
|
143
|
-
*/
|
|
144
|
-
export function scopeKeyOf(v) {
|
|
145
|
-
return typeof v === "string" && v.trim() ? v.trim() : null;
|
|
146
|
-
}
|
|
1
|
+
// v0.8.0 A1(issue #17):写入端 scope 解析。
|
|
2
|
+
//
|
|
3
|
+
// 身份模型(issue #17 的核心约定):session 只是「载体」,记忆挂在持久化身份
|
|
4
|
+
// 标签上——agent_scope = session header 的 agentPreset(跨会话稳定的 agent 模板
|
|
5
|
+
// 身份),workspace_scope = 会话工作目录(registry 反查的 canonical path 优先,
|
|
6
|
+
// header.cwd 兜底)。session_id 永不作为作用域键。
|
|
7
|
+
//
|
|
8
|
+
// 硬约束:解析失败绝不阻塞写入——任何一步取不到就落 NULL(读侧 NULL = 未标注
|
|
9
|
+
// = 全局可见),任何异常都被吞掉(registry 故障 warn 一次后降级)。
|
|
10
|
+
//
|
|
11
|
+
// v0.8.1(issue #170 复核项 2):scopeEnabled 的闸门从解析器挪到「写标注」处
|
|
12
|
+
// (memory_save 的 auto 分支)。解析器恒返回会话身份(真取不到才是 null)——
|
|
13
|
+
// 读取侧(strictVisibility / memory_get / inject)与显式声明都依赖它:flag 关
|
|
14
|
+
// 只关「自动标注」,不能连读取身份一起关,否则 strictScope 硬墙被静默拆掉
|
|
15
|
+
// (fail-open)。
|
|
16
|
+
|
|
17
|
+
/**
|
|
18
|
+
* @param {object} deps
|
|
19
|
+
* @param {object} [deps.ctx] 宿主上下文(可能带 workspaceRegistry;旧版 dsh 没有)
|
|
20
|
+
* @param {object} [deps.config] 已解析配置(读 scopeEnabled)
|
|
21
|
+
* @param {object} [deps.logger]
|
|
22
|
+
* @returns {(exec: object) => {agent_scope: string|null, workspace_scope: string|null}|null}
|
|
23
|
+
*/
|
|
24
|
+
export function createScopeResolver({ ctx, config, logger } = {}) {
|
|
25
|
+
let warned = false;
|
|
26
|
+
const warnOnce = (msg) => {
|
|
27
|
+
if (warned) return;
|
|
28
|
+
warned = true;
|
|
29
|
+
try {
|
|
30
|
+
logger?.warn?.(msg);
|
|
31
|
+
} catch {
|
|
32
|
+
// logger 自身故障同样不能影响写入路径
|
|
33
|
+
}
|
|
34
|
+
};
|
|
35
|
+
|
|
36
|
+
return function resolveWriteScope(exec) {
|
|
37
|
+
const session = sessionOf(exec);
|
|
38
|
+
|
|
39
|
+
// 持久化 header(issue #17 的身份契约):session.header 是创建时冻结的
|
|
40
|
+
// SessionHeader(含 agentPreset/cwd)。⚠️ 不能先读 requestHeader()——那返回
|
|
41
|
+
// EpochHeader(请求级模型路由头,首个请求后即为真值),会把 session.header
|
|
42
|
+
// 遮蔽掉,agentPreset/cwd 双双取空(本地 web 复验 2026-09-13 实测踩中)。
|
|
43
|
+
// 仅当宿主没有 .header 属性时才降级尝试 requestHeader().config(旧形状)。
|
|
44
|
+
let header = null;
|
|
45
|
+
try {
|
|
46
|
+
header = session?.header ?? session?.requestHeader?.()?.config ?? null;
|
|
47
|
+
} catch {
|
|
48
|
+
header = null;
|
|
49
|
+
}
|
|
50
|
+
|
|
51
|
+
let agentScope = null;
|
|
52
|
+
const preset = header?.agentPreset;
|
|
53
|
+
if (typeof preset === "string" && preset.trim()) agentScope = preset.trim();
|
|
54
|
+
|
|
55
|
+
// workspace:registry 反查优先(成员资格 = id 命中 + header canonical cwd
|
|
56
|
+
// 校验过,path 已 realpath 归一),失败回退 header.cwd,再失败留 NULL。
|
|
57
|
+
let workspaceScope = null;
|
|
58
|
+
try {
|
|
59
|
+
const list = ctx?.workspaceRegistry?.list?.();
|
|
60
|
+
const sessionId = session?.id;
|
|
61
|
+
if (Array.isArray(list) && sessionId != null) {
|
|
62
|
+
const hit = list.find(
|
|
63
|
+
(ws) => Array.isArray(ws?.sessionIds) && ws.sessionIds.includes(sessionId)
|
|
64
|
+
);
|
|
65
|
+
if (typeof hit?.path === "string" && hit.path.trim()) workspaceScope = hit.path.trim();
|
|
66
|
+
}
|
|
67
|
+
} catch (e) {
|
|
68
|
+
warnOnce(
|
|
69
|
+
`[dsh-mneme] workspaceRegistry lookup failed, falling back to session header cwd: ${String(e)}`
|
|
70
|
+
);
|
|
71
|
+
}
|
|
72
|
+
if (!workspaceScope) {
|
|
73
|
+
const cwd = header?.cwd;
|
|
74
|
+
if (typeof cwd === "string" && cwd.trim()) workspaceScope = cwd.trim();
|
|
75
|
+
}
|
|
76
|
+
|
|
77
|
+
return { agent_scope: agentScope, workspace_scope: workspaceScope };
|
|
78
|
+
};
|
|
79
|
+
}
|
|
80
|
+
|
|
81
|
+
/**
|
|
82
|
+
* 工具 exec 上的会话对象。宿主 ctx 是 Proxy(cordis 4),裸读未 inject 的属性会
|
|
83
|
+
* 抛错——所以取值恒包在 try 里,取不到返回 null。scope 解析与写入准入的会话键
|
|
84
|
+
* 共用这一处取值口径。
|
|
85
|
+
* @param {object} exec
|
|
86
|
+
* @returns {object|null}
|
|
87
|
+
*/
|
|
88
|
+
export function sessionOf(exec) {
|
|
89
|
+
try {
|
|
90
|
+
return exec?.agent?.session ?? null;
|
|
91
|
+
} catch {
|
|
92
|
+
return null;
|
|
93
|
+
}
|
|
94
|
+
}
|
|
95
|
+
|
|
96
|
+
/**
|
|
97
|
+
* 写入准入的会话键(#254 G1 会话写入预算的计数单位)。session_id 在这里只作
|
|
98
|
+
* 「同一段对话」的标识,不作为作用域键(见文件头:记忆挂在持久化身份标签上);
|
|
99
|
+
* 取不到会话身份(系统写入)返回 null = 不进预算。
|
|
100
|
+
* @param {object} exec
|
|
101
|
+
* @returns {string|null}
|
|
102
|
+
*/
|
|
103
|
+
export function sessionKeyOf(exec) {
|
|
104
|
+
// 整个取值(含 .id)都在 try 里:宿主 session 可能是会抛错的 accessor,而这个
|
|
105
|
+
// 函数在 memory_save 的保存之前被调用——抛出去就是一次工具调用直接失败。
|
|
106
|
+
try {
|
|
107
|
+
const id = sessionOf(exec)?.id;
|
|
108
|
+
return id == null ? null : String(id);
|
|
109
|
+
} catch {
|
|
110
|
+
return null;
|
|
111
|
+
}
|
|
112
|
+
}
|
|
113
|
+
|
|
114
|
+
/**
|
|
115
|
+
* v0.8.1 底座(issue #170):显式 scope 参数(memory_save / memory_update /
|
|
116
|
+
* 面板编辑)的归一化。约定:
|
|
117
|
+
* - 省略(undefined,由调用方判断)= 该维不声明,回落载体自动标注;
|
|
118
|
+
* - "global"(大小写不敏感)/ "*" / 空白 = 显式声明全局 → 存储 NULL
|
|
119
|
+
* (行上的 agent_scope_source='explicit' 承担「显式全局」与「从未标注」
|
|
120
|
+
* 的区分,NULL 本身继续表示全局可见);
|
|
121
|
+
* - 其余非空 trim 字符串 = 收窄到该标签。
|
|
122
|
+
* 脏输入(非字符串)落 NULL(等同声明全局)——与 resolveWriteScope 的「解析
|
|
123
|
+
* 绝不阻塞写入」同一哲学:显式声明是用户意图,形状不对时宁可放宽不报错。
|
|
124
|
+
*
|
|
125
|
+
* @param {unknown} raw 工具/HTTP 参数原值
|
|
126
|
+
* @returns {string|null} 归一化后的标签值(null=显式全局)
|
|
127
|
+
*/
|
|
128
|
+
export function normalizeExplicitScope(raw) {
|
|
129
|
+
if (typeof raw !== "string") return null;
|
|
130
|
+
const s = raw.trim();
|
|
131
|
+
if (!s || s === "*" || s.toLowerCase() === "global") return null;
|
|
132
|
+
return s;
|
|
133
|
+
}
|
|
134
|
+
|
|
135
|
+
/**
|
|
136
|
+
* scope 标注的比较键(去重键/检索加权/跨 scope 配对共用口径):非空 trim 字符串
|
|
137
|
+
* 原样返回,其余(NULL/undefined/空串/非字符串)统一为 null(「未标注」)。
|
|
138
|
+
* v0.8.1(issue #170 第 2 步)从 service.js 上提为共享定义——sleep 的跨 scope
|
|
139
|
+
* 候选配对必须与去重键用同一把钥匙,NULL 与已标注才不会错配。
|
|
140
|
+
*
|
|
141
|
+
* @param {unknown} v scope 标注原值
|
|
142
|
+
* @returns {string|null} 比较键(null=未标注)
|
|
143
|
+
*/
|
|
144
|
+
export function scopeKeyOf(v) {
|
|
145
|
+
return typeof v === "string" && v.trim() ? v.trim() : null;
|
|
146
|
+
}
|
package/lib/search/adaptive.js
CHANGED
|
@@ -1,60 +1,60 @@
|
|
|
1
|
-
// Adaptive vector threshold (v0.5.0 召回率优化 1.2): replaces the fixed
|
|
2
|
-
// vectorSearchThreshold=0.65 with a query-aware cutoff.
|
|
3
|
-
// entity:/attr: prefixes → 0.5 (entity recall is name-driven; loosen)
|
|
4
|
-
// very short queries → 0.7 (<5 chars match almost anything; tighten)
|
|
5
|
-
// very long queries → 0.6 (semantically specific; loosen a little)
|
|
6
|
-
// head-gap rule → when the top-1 vs top-5 candidate gap exceeds
|
|
7
|
-
// 0.3 the head is decisive — loosen to 0.5 so
|
|
8
|
-
// the tail still reaches the reranker
|
|
9
|
-
// otherwise → 0.65 (the legacy default)
|
|
10
|
-
// Pure and total: same inputs, same cutoff, no store access.
|
|
11
|
-
export function adaptiveThreshold(query, candidates = []) {
|
|
12
|
-
const q = String(query ?? "");
|
|
13
|
-
if (q.startsWith("entity:") || q.startsWith("attr:")) return 0.5;
|
|
14
|
-
if (q.length > 0 && q.length < 5) return 0.7;
|
|
15
|
-
if (q.length > 50) return 0.6;
|
|
16
|
-
const scores = (Array.isArray(candidates) ? candidates : [])
|
|
17
|
-
.map((c) => (typeof c?._score === "number" ? c._score : typeof c?.score === "number" ? c.score : 0))
|
|
18
|
-
.filter((s) => s > 0)
|
|
19
|
-
.sort((a, b) => b - a);
|
|
20
|
-
if (scores.length >= 5 && scores[0] - scores[4] > 0.3) return 0.5;
|
|
21
|
-
return 0.65;
|
|
22
|
-
}
|
|
23
|
-
|
|
24
|
-
// Issue #239(第 5 项):注入条数的查询自适应(去 always-on 化的一半)。
|
|
25
|
-
// 依据:不确定性强的话题多召回、确定性的少召回(Oblivion 的不确定性驱动读路径);
|
|
26
|
-
// 但**注入相关却带偏生成**的内容可能比不注入更糟(2607.24010 与 2604.02280 两处
|
|
27
|
-
// 独立出处),所以这里只做**单向收缩**:模糊 → 维持用户配置的上限,确定 → 收缩,
|
|
28
|
-
// 绝不越过上限。判据只看查询本身——不做额外检索(2607.24010:触发侧成本别忽略,
|
|
29
|
-
// 先探针检索等于白付一次 fuseRecall)。
|
|
30
|
-
const RECALL_CUES = [
|
|
31
|
-
// 中文回指 / 时间线索:出现即说明用户要的是「记忆里的东西」,当场算不出来
|
|
32
|
-
"上次", "之前", "以前", "那天", "那次", "刚才", "刚刚", "后来", "很早",
|
|
33
|
-
"还记得", "记不记得", "我们讨论", "我们聊", "我说过", "你提过", "提到过",
|
|
34
|
-
"最近", "上次说", "之前说",
|
|
35
|
-
// 英文同类线索
|
|
36
|
-
"last time", "previously", "earlier", "remember", "we discussed",
|
|
37
|
-
"i mentioned", "you said", "that thing", "remind me"
|
|
38
|
-
];
|
|
39
|
-
|
|
40
|
-
/**
|
|
41
|
-
* 这段查询是否需要「宽召回」。命中回指/时间线索,或短到几乎匹配任何东西
|
|
42
|
-
* (与 adaptiveThreshold 对 <5 字符的判断同源)。纯函数、零成本。
|
|
43
|
-
* 拿不到查询时不猜——返回 true 维持现状。
|
|
44
|
-
*/
|
|
45
|
-
export function needsBroadRecall(query) {
|
|
46
|
-
const q = String(query ?? "").trim().toLowerCase();
|
|
47
|
-
if (q === "") return true;
|
|
48
|
-
if (q.length < 5) return true;
|
|
49
|
-
return RECALL_CUES.some((cue) => q.includes(cue));
|
|
50
|
-
}
|
|
51
|
-
|
|
52
|
-
/**
|
|
53
|
-
* 注入条数的自适应预算:模糊 → 维持配置上限;确定 → 收缩到一半(下限 1)。
|
|
54
|
-
* `maxItems <= 1` 时原样返回(含 0 = 不注入,flag 打开也不该把 0 变成 1)。
|
|
55
|
-
*/
|
|
56
|
-
export function adaptiveInjectBudget(query, maxItems) {
|
|
57
|
-
const cap = Number.isFinite(maxItems) && maxItems > 0 ? Math.floor(maxItems) : maxItems;
|
|
58
|
-
if (!Number.isFinite(cap) || cap <= 1) return cap;
|
|
59
|
-
return needsBroadRecall(query) ? cap : Math.max(1, Math.ceil(cap / 2));
|
|
60
|
-
}
|
|
1
|
+
// Adaptive vector threshold (v0.5.0 召回率优化 1.2): replaces the fixed
|
|
2
|
+
// vectorSearchThreshold=0.65 with a query-aware cutoff.
|
|
3
|
+
// entity:/attr: prefixes → 0.5 (entity recall is name-driven; loosen)
|
|
4
|
+
// very short queries → 0.7 (<5 chars match almost anything; tighten)
|
|
5
|
+
// very long queries → 0.6 (semantically specific; loosen a little)
|
|
6
|
+
// head-gap rule → when the top-1 vs top-5 candidate gap exceeds
|
|
7
|
+
// 0.3 the head is decisive — loosen to 0.5 so
|
|
8
|
+
// the tail still reaches the reranker
|
|
9
|
+
// otherwise → 0.65 (the legacy default)
|
|
10
|
+
// Pure and total: same inputs, same cutoff, no store access.
|
|
11
|
+
export function adaptiveThreshold(query, candidates = []) {
|
|
12
|
+
const q = String(query ?? "");
|
|
13
|
+
if (q.startsWith("entity:") || q.startsWith("attr:")) return 0.5;
|
|
14
|
+
if (q.length > 0 && q.length < 5) return 0.7;
|
|
15
|
+
if (q.length > 50) return 0.6;
|
|
16
|
+
const scores = (Array.isArray(candidates) ? candidates : [])
|
|
17
|
+
.map((c) => (typeof c?._score === "number" ? c._score : typeof c?.score === "number" ? c.score : 0))
|
|
18
|
+
.filter((s) => s > 0)
|
|
19
|
+
.sort((a, b) => b - a);
|
|
20
|
+
if (scores.length >= 5 && scores[0] - scores[4] > 0.3) return 0.5;
|
|
21
|
+
return 0.65;
|
|
22
|
+
}
|
|
23
|
+
|
|
24
|
+
// Issue #239(第 5 项):注入条数的查询自适应(去 always-on 化的一半)。
|
|
25
|
+
// 依据:不确定性强的话题多召回、确定性的少召回(Oblivion 的不确定性驱动读路径);
|
|
26
|
+
// 但**注入相关却带偏生成**的内容可能比不注入更糟(2607.24010 与 2604.02280 两处
|
|
27
|
+
// 独立出处),所以这里只做**单向收缩**:模糊 → 维持用户配置的上限,确定 → 收缩,
|
|
28
|
+
// 绝不越过上限。判据只看查询本身——不做额外检索(2607.24010:触发侧成本别忽略,
|
|
29
|
+
// 先探针检索等于白付一次 fuseRecall)。
|
|
30
|
+
const RECALL_CUES = [
|
|
31
|
+
// 中文回指 / 时间线索:出现即说明用户要的是「记忆里的东西」,当场算不出来
|
|
32
|
+
"上次", "之前", "以前", "那天", "那次", "刚才", "刚刚", "后来", "很早",
|
|
33
|
+
"还记得", "记不记得", "我们讨论", "我们聊", "我说过", "你提过", "提到过",
|
|
34
|
+
"最近", "上次说", "之前说",
|
|
35
|
+
// 英文同类线索
|
|
36
|
+
"last time", "previously", "earlier", "remember", "we discussed",
|
|
37
|
+
"i mentioned", "you said", "that thing", "remind me"
|
|
38
|
+
];
|
|
39
|
+
|
|
40
|
+
/**
|
|
41
|
+
* 这段查询是否需要「宽召回」。命中回指/时间线索,或短到几乎匹配任何东西
|
|
42
|
+
* (与 adaptiveThreshold 对 <5 字符的判断同源)。纯函数、零成本。
|
|
43
|
+
* 拿不到查询时不猜——返回 true 维持现状。
|
|
44
|
+
*/
|
|
45
|
+
export function needsBroadRecall(query) {
|
|
46
|
+
const q = String(query ?? "").trim().toLowerCase();
|
|
47
|
+
if (q === "") return true;
|
|
48
|
+
if (q.length < 5) return true;
|
|
49
|
+
return RECALL_CUES.some((cue) => q.includes(cue));
|
|
50
|
+
}
|
|
51
|
+
|
|
52
|
+
/**
|
|
53
|
+
* 注入条数的自适应预算:模糊 → 维持配置上限;确定 → 收缩到一半(下限 1)。
|
|
54
|
+
* `maxItems <= 1` 时原样返回(含 0 = 不注入,flag 打开也不该把 0 变成 1)。
|
|
55
|
+
*/
|
|
56
|
+
export function adaptiveInjectBudget(query, maxItems) {
|
|
57
|
+
const cap = Number.isFinite(maxItems) && maxItems > 0 ? Math.floor(maxItems) : maxItems;
|
|
58
|
+
if (!Number.isFinite(cap) || cap <= 1) return cap;
|
|
59
|
+
return needsBroadRecall(query) ? cap : Math.max(1, Math.ceil(cap / 2));
|
|
60
|
+
}
|
package/lib/search/bm25.js
CHANGED
|
@@ -1,96 +1,96 @@
|
|
|
1
|
-
// BM25 sparse retrieval (v0.5.0 召回率优化 1.1): the third recall path beside
|
|
2
|
-
// vector search and the LIKE keyword scan. The LIKE path only matches full
|
|
3
|
-
// substrings, so a multi-term query ("rust 异步 tokio") misses rows whose
|
|
4
|
-
// terms are scattered. BM25 scores per-token overlap with IDF weighting,
|
|
5
|
-
// which is exactly the gap: identifiers, code fragments and mixed CJK/ASCII
|
|
6
|
-
// queries recall rows the substring scan cannot see.
|
|
7
|
-
|
|
8
|
-
// Tokenizer: ASCII words keep their shape (identifiers like "dsh-mneme" or
|
|
9
|
-
// "ZFS_4421" survive as whole tokens); CJK runs become sliding bigrams
|
|
10
|
-
// (unigram only for single characters), the standard workaround for BM25's
|
|
11
|
-
// whitespace tokenization on Chinese.
|
|
12
|
-
export function tokenize(text) {
|
|
13
|
-
const raw = String(text ?? "").toLowerCase();
|
|
14
|
-
const tokens = [];
|
|
15
|
-
const ascii = raw.match(/[a-z0-9_]+/g) ?? [];
|
|
16
|
-
tokens.push(...ascii);
|
|
17
|
-
const cjkRuns = raw.match(/[\u4e00-\u9fff]+/g) ?? [];
|
|
18
|
-
for (const run of cjkRuns) {
|
|
19
|
-
if (run.length === 1) { tokens.push(run); continue; }
|
|
20
|
-
for (let i = 0; i < run.length - 1; i++) tokens.push(run.slice(i, i + 2));
|
|
21
|
-
}
|
|
22
|
-
return tokens;
|
|
23
|
-
}
|
|
24
|
-
|
|
25
|
-
const K1 = 1.5; // term-frequency saturation
|
|
26
|
-
const B = 0.75; // length normalization
|
|
27
|
-
|
|
28
|
-
/**
|
|
29
|
-
* Build a BM25 index over documents: [{id, title, content}].
|
|
30
|
-
* Returns { score, search }:
|
|
31
|
-
* score(query, doc) — per-spec ad-hoc scoring (re-tokenizes the doc)
|
|
32
|
-
* search(query, {limit}) — precomputed-tf ranking, scores normalized to
|
|
33
|
-
* [0,1] by the max so BM25 hits can weight-blend with vector/keyword
|
|
34
|
-
* scores on one scale. Rows the query does not touch at all are dropped.
|
|
35
|
-
*/
|
|
36
|
-
export function createBM25Index(documents) {
|
|
37
|
-
const docs = Array.isArray(documents) ? documents.filter(Boolean) : [];
|
|
38
|
-
const N = docs.length;
|
|
39
|
-
const df = new Map();
|
|
40
|
-
const prepared = docs.map((doc) => {
|
|
41
|
-
const tokens = tokenize(`${doc.title ?? ""} ${doc.content ?? ""}`);
|
|
42
|
-
const tf = new Map();
|
|
43
|
-
for (const t of tokens) tf.set(t, (tf.get(t) ?? 0) + 1);
|
|
44
|
-
for (const t of tf.keys()) df.set(t, (df.get(t) ?? 0) + 1);
|
|
45
|
-
return { doc, tf, len: tokens.length };
|
|
46
|
-
});
|
|
47
|
-
const avgLen = N ? prepared.reduce((s, p) => s + p.len, 0) / N : 0 || 1;
|
|
48
|
-
|
|
49
|
-
const idf = (t) => {
|
|
50
|
-
const n = df.get(t) ?? 0;
|
|
51
|
-
return Math.log((N - n + 0.5) / (n + 0.5) + 1);
|
|
52
|
-
};
|
|
53
|
-
|
|
54
|
-
function scorePrepared(queryTokens, p) {
|
|
55
|
-
let score = 0;
|
|
56
|
-
for (const t of queryTokens) {
|
|
57
|
-
const f = p.tf.get(t);
|
|
58
|
-
if (!f) continue;
|
|
59
|
-
const norm = p.len ? K1 * (1 - B + B * (p.len / avgLen)) : K1;
|
|
60
|
-
score += idf(t) * ((f * (K1 + 1)) / (f + norm));
|
|
61
|
-
}
|
|
62
|
-
return score;
|
|
63
|
-
}
|
|
64
|
-
|
|
65
|
-
return {
|
|
66
|
-
score(query, doc) {
|
|
67
|
-
const tokens = tokenize(`${doc?.title ?? ""} ${doc?.content ?? ""}`);
|
|
68
|
-
const tf = new Map();
|
|
69
|
-
for (const t of tokens) tf.set(t, (tf.get(t) ?? 0) + 1);
|
|
70
|
-
const len = tokens.length;
|
|
71
|
-
// Ad-hoc scoring can't see corpus df; fall back to tf-only saturation
|
|
72
|
-
// (df is approximated as 1 so idf ≈ log(N - 0.5 + 1) is constant).
|
|
73
|
-
let score = 0;
|
|
74
|
-
for (const t of tokenize(query)) {
|
|
75
|
-
const f = tf.get(t);
|
|
76
|
-
if (!f) continue;
|
|
77
|
-
const norm = len ? K1 * (1 - B + B * (len / avgLen)) : K1;
|
|
78
|
-
score += idf(t) * ((f * (K1 + 1)) / (f + norm));
|
|
79
|
-
}
|
|
80
|
-
return score;
|
|
81
|
-
},
|
|
82
|
-
search(query, { limit = 20 } = {}) {
|
|
83
|
-
const qTokens = tokenize(query);
|
|
84
|
-
if (!qTokens.length || !N) return [];
|
|
85
|
-
const scored = [];
|
|
86
|
-
for (const p of prepared) {
|
|
87
|
-
const s = scorePrepared(qTokens, p);
|
|
88
|
-
if (s > 0) scored.push({ row: p.doc, raw: s });
|
|
89
|
-
}
|
|
90
|
-
scored.sort((a, b) => b.raw - a.raw);
|
|
91
|
-
const top = scored.slice(0, limit);
|
|
92
|
-
const max = top[0]?.raw || 1;
|
|
93
|
-
return top.map(({ row, raw }) => ({ ...row, score: max ? raw / max : 0 }));
|
|
94
|
-
}
|
|
95
|
-
};
|
|
96
|
-
}
|
|
1
|
+
// BM25 sparse retrieval (v0.5.0 召回率优化 1.1): the third recall path beside
|
|
2
|
+
// vector search and the LIKE keyword scan. The LIKE path only matches full
|
|
3
|
+
// substrings, so a multi-term query ("rust 异步 tokio") misses rows whose
|
|
4
|
+
// terms are scattered. BM25 scores per-token overlap with IDF weighting,
|
|
5
|
+
// which is exactly the gap: identifiers, code fragments and mixed CJK/ASCII
|
|
6
|
+
// queries recall rows the substring scan cannot see.
|
|
7
|
+
|
|
8
|
+
// Tokenizer: ASCII words keep their shape (identifiers like "dsh-mneme" or
|
|
9
|
+
// "ZFS_4421" survive as whole tokens); CJK runs become sliding bigrams
|
|
10
|
+
// (unigram only for single characters), the standard workaround for BM25's
|
|
11
|
+
// whitespace tokenization on Chinese.
|
|
12
|
+
export function tokenize(text) {
|
|
13
|
+
const raw = String(text ?? "").toLowerCase();
|
|
14
|
+
const tokens = [];
|
|
15
|
+
const ascii = raw.match(/[a-z0-9_]+/g) ?? [];
|
|
16
|
+
tokens.push(...ascii);
|
|
17
|
+
const cjkRuns = raw.match(/[\u4e00-\u9fff]+/g) ?? [];
|
|
18
|
+
for (const run of cjkRuns) {
|
|
19
|
+
if (run.length === 1) { tokens.push(run); continue; }
|
|
20
|
+
for (let i = 0; i < run.length - 1; i++) tokens.push(run.slice(i, i + 2));
|
|
21
|
+
}
|
|
22
|
+
return tokens;
|
|
23
|
+
}
|
|
24
|
+
|
|
25
|
+
const K1 = 1.5; // term-frequency saturation
|
|
26
|
+
const B = 0.75; // length normalization
|
|
27
|
+
|
|
28
|
+
/**
|
|
29
|
+
* Build a BM25 index over documents: [{id, title, content}].
|
|
30
|
+
* Returns { score, search }:
|
|
31
|
+
* score(query, doc) — per-spec ad-hoc scoring (re-tokenizes the doc)
|
|
32
|
+
* search(query, {limit}) — precomputed-tf ranking, scores normalized to
|
|
33
|
+
* [0,1] by the max so BM25 hits can weight-blend with vector/keyword
|
|
34
|
+
* scores on one scale. Rows the query does not touch at all are dropped.
|
|
35
|
+
*/
|
|
36
|
+
export function createBM25Index(documents) {
|
|
37
|
+
const docs = Array.isArray(documents) ? documents.filter(Boolean) : [];
|
|
38
|
+
const N = docs.length;
|
|
39
|
+
const df = new Map();
|
|
40
|
+
const prepared = docs.map((doc) => {
|
|
41
|
+
const tokens = tokenize(`${doc.title ?? ""} ${doc.content ?? ""}`);
|
|
42
|
+
const tf = new Map();
|
|
43
|
+
for (const t of tokens) tf.set(t, (tf.get(t) ?? 0) + 1);
|
|
44
|
+
for (const t of tf.keys()) df.set(t, (df.get(t) ?? 0) + 1);
|
|
45
|
+
return { doc, tf, len: tokens.length };
|
|
46
|
+
});
|
|
47
|
+
const avgLen = N ? prepared.reduce((s, p) => s + p.len, 0) / N : 0 || 1;
|
|
48
|
+
|
|
49
|
+
const idf = (t) => {
|
|
50
|
+
const n = df.get(t) ?? 0;
|
|
51
|
+
return Math.log((N - n + 0.5) / (n + 0.5) + 1);
|
|
52
|
+
};
|
|
53
|
+
|
|
54
|
+
function scorePrepared(queryTokens, p) {
|
|
55
|
+
let score = 0;
|
|
56
|
+
for (const t of queryTokens) {
|
|
57
|
+
const f = p.tf.get(t);
|
|
58
|
+
if (!f) continue;
|
|
59
|
+
const norm = p.len ? K1 * (1 - B + B * (p.len / avgLen)) : K1;
|
|
60
|
+
score += idf(t) * ((f * (K1 + 1)) / (f + norm));
|
|
61
|
+
}
|
|
62
|
+
return score;
|
|
63
|
+
}
|
|
64
|
+
|
|
65
|
+
return {
|
|
66
|
+
score(query, doc) {
|
|
67
|
+
const tokens = tokenize(`${doc?.title ?? ""} ${doc?.content ?? ""}`);
|
|
68
|
+
const tf = new Map();
|
|
69
|
+
for (const t of tokens) tf.set(t, (tf.get(t) ?? 0) + 1);
|
|
70
|
+
const len = tokens.length;
|
|
71
|
+
// Ad-hoc scoring can't see corpus df; fall back to tf-only saturation
|
|
72
|
+
// (df is approximated as 1 so idf ≈ log(N - 0.5 + 1) is constant).
|
|
73
|
+
let score = 0;
|
|
74
|
+
for (const t of tokenize(query)) {
|
|
75
|
+
const f = tf.get(t);
|
|
76
|
+
if (!f) continue;
|
|
77
|
+
const norm = len ? K1 * (1 - B + B * (len / avgLen)) : K1;
|
|
78
|
+
score += idf(t) * ((f * (K1 + 1)) / (f + norm));
|
|
79
|
+
}
|
|
80
|
+
return score;
|
|
81
|
+
},
|
|
82
|
+
search(query, { limit = 20 } = {}) {
|
|
83
|
+
const qTokens = tokenize(query);
|
|
84
|
+
if (!qTokens.length || !N) return [];
|
|
85
|
+
const scored = [];
|
|
86
|
+
for (const p of prepared) {
|
|
87
|
+
const s = scorePrepared(qTokens, p);
|
|
88
|
+
if (s > 0) scored.push({ row: p.doc, raw: s });
|
|
89
|
+
}
|
|
90
|
+
scored.sort((a, b) => b.raw - a.raw);
|
|
91
|
+
const top = scored.slice(0, limit);
|
|
92
|
+
const max = top[0]?.raw || 1;
|
|
93
|
+
return top.map(({ row, raw }) => ({ ...row, score: max ? raw / max : 0 }));
|
|
94
|
+
}
|
|
95
|
+
};
|
|
96
|
+
}
|