@modusensus/dsh-mneme 0.8.2 → 0.8.4
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 +718 -633
- package/bin/cli.mjs +0 -0
- package/bin/dsh-mneme-mcp.mjs +502 -0
- package/cordis.patch.yml +21 -21
- package/lib/api-standalone.js +495 -316
- package/lib/api.js +1382 -1316
- package/lib/bootstrap.js +211 -0
- package/lib/client.js +4475 -4362
- package/lib/commands.js +95 -95
- package/lib/config.js +533 -500
- package/lib/dream/clustering.js +118 -118
- package/lib/dream/decisions.js +658 -658
- package/lib/dream/narratives.js +44 -0
- package/lib/dream/sleep.js +700 -700
- package/lib/dream.js +1319 -1216
- package/lib/embedding.js +177 -177
- package/lib/entities/extractor.js +249 -249
- package/lib/heat.js +142 -136
- package/lib/hot-memory.js +46 -46
- package/lib/index.js +515 -513
- package/lib/inject.js +299 -253
- package/lib/lang.js +202 -179
- package/lib/local-embedder.js +320 -308
- package/lib/mirror.js +198 -198
- package/lib/quality-filter.js +126 -126
- package/lib/recall-stats.js +103 -0
- package/lib/reranker.js +269 -260
- 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 +235 -235
- package/lib/runtime/loader.js +249 -249
- package/lib/runtime/model-fetch.js +361 -0
- package/lib/runtime/provision.js +237 -237
- package/lib/runtime/tarball.js +130 -130
- package/lib/runtime/verify.js +191 -191
- package/lib/scope.js +118 -118
- package/lib/search/adaptive.js +22 -22
- package/lib/search/bm25.js +96 -96
- package/lib/service.js +2171 -2071
- package/lib/settings.js +430 -419
- package/lib/store.js +2448 -2367
- package/lib/summarize.js +575 -449
- package/lib/tools.js +574 -574
- package/lib/vector-index.js +110 -110
- package/lib/version-check.js +122 -122
- package/package.json +109 -108
- package/runtime-manifest.json +596 -596
- package/scripts/benchmark-embed.js +201 -201
- package/scripts/benchmark-recall.js +199 -199
- package/scripts/benchmark-rerank.js +166 -166
- package/scripts/build-runtime-manifest.mjs +217 -217
- 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 +61 -61
- package/src/api-standalone.js +495 -316
- package/src/api.js +1382 -1316
- package/src/bootstrap.js +211 -0
- package/src/commands.js +95 -95
- package/src/config.js +533 -500
- package/src/dream/clustering.js +118 -118
- package/src/dream/decisions.js +658 -658
- package/src/dream/narratives.js +44 -0
- package/src/dream/sleep.js +700 -700
- package/src/dream.js +1319 -1216
- package/src/embedding.js +177 -177
- package/src/entities/extractor.js +249 -249
- package/src/heat.js +142 -136
- package/src/hot-memory.js +46 -46
- package/src/index.js +515 -513
- package/src/inject.js +299 -253
- package/src/lang.js +202 -179
- package/src/local-embedder.js +320 -308
- package/src/mirror.js +198 -198
- package/src/quality-filter.js +126 -126
- package/src/recall-stats.js +103 -0
- package/src/reranker.js +269 -260
- 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 +235 -235
- package/src/runtime/loader.js +249 -249
- package/src/runtime/model-fetch.js +361 -0
- package/src/runtime/provision.js +237 -237
- package/src/runtime/tarball.js +130 -130
- package/src/runtime/verify.js +191 -191
- package/src/scope.js +118 -118
- package/src/search/adaptive.js +22 -22
- package/src/search/bm25.js +96 -96
- package/src/service.js +2171 -2071
- package/src/settings.js +430 -419
- package/src/store.js +2448 -2367
- package/src/summarize.js +575 -449
- package/src/tools.js +574 -574
- package/src/vector-index.js +110 -110
- package/src/version-check.js +122 -122
- package/test/api-routes-gaps.test.js +218 -218
- package/test/api.test.js +1578 -1481
- package/test/audit.test.js +448 -448
- package/test/benchmark.test.js +94 -94
- package/test/bootstrap.test.js +220 -0
- package/test/client.test.js +799 -736
- 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/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.test.js +1257 -1254
- package/test/embedding-ready.test.js +196 -196
- package/test/entities.test.js +547 -547
- package/test/entity-recall.test.js +179 -0
- package/test/epistemic.test.js +298 -298
- package/test/fnew-0112.test.js +311 -311
- package/test/fnew-03.test.js +422 -422
- package/test/graph-api.test.js +175 -175
- package/test/heat.test.js +175 -148
- 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/hot-memory.test.js +145 -145
- package/test/inject-content-cap.test.js +89 -0
- package/test/inject-first-round-bm25.test.js +89 -89
- package/test/inject-heat.test.js +61 -0
- package/test/inject-rotation.test.js +161 -0
- package/test/inject.test.js +249 -249
- package/test/lang.test.js +87 -87
- package/test/lib-smoke.test.js +96 -96
- package/test/llm-audit.test.js +309 -309
- package/test/llm-message-source.test.js +177 -177
- package/test/local-embedder.test.js +249 -229
- package/test/mcp-stdio.test.js +385 -0
- 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.test.js +249 -249
- package/test/narrative-bars.test.js +162 -0
- package/test/peer-blockers.test.js +193 -193
- package/test/policy-epoch.test.js +259 -259
- package/test/quality-filter.test.js +186 -186
- package/test/reasoning-effort.test.js +746 -746
- package/test/recall-evals.test.js +235 -235
- package/test/recall-layer.test.js +326 -326
- package/test/recall-stats.test.js +117 -0
- 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 -0
- 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 +237 -237
- package/test/runtime-manifest.test.js +158 -158
- package/test/runtime-model-fetch.test.js +348 -0
- package/test/runtime-provision.test.js +101 -101
- package/test/runtime-tarball.test.js +81 -81
- package/test/runtime-verify.test.js +214 -214
- package/test/scope-candidates.test.js +239 -239
- package/test/scope-explicit.test.js +452 -452
- package/test/scope-retrieval.test.js +197 -197
- package/test/scope-storage.test.js +255 -255
- package/test/scope-strict.test.js +231 -231
- package/test/search-fusion.test.js +90 -90
- package/test/semantic.test.js +124 -124
- 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 -125
- package/test/sleep-scheduler-rearm.test.js +89 -89
- package/test/sleep.test.js +373 -373
- package/test/standalone-api.test.js +544 -359
- package/test/store-vector-cache.test.js +98 -98
- package/test/store.test.js +383 -383
- package/test/stress.test.js +209 -209
- package/test/summarize.test.js +884 -575
- package/test/tools-memory-get-schema.test.js +65 -65
- package/test/tools.test.js +341 -341
- 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/lib/scope.js
CHANGED
|
@@ -1,118 +1,118 @@
|
|
|
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
|
-
let session = null;
|
|
38
|
-
try {
|
|
39
|
-
session = exec?.agent?.session ?? null;
|
|
40
|
-
} catch {
|
|
41
|
-
session = null;
|
|
42
|
-
}
|
|
43
|
-
|
|
44
|
-
// 持久化 header(issue #17 的身份契约):session.header 是创建时冻结的
|
|
45
|
-
// SessionHeader(含 agentPreset/cwd)。⚠️ 不能先读 requestHeader()——那返回
|
|
46
|
-
// EpochHeader(请求级模型路由头,首个请求后即为真值),会把 session.header
|
|
47
|
-
// 遮蔽掉,agentPreset/cwd 双双取空(本地 web 复验 2026-09-13 实测踩中)。
|
|
48
|
-
// 仅当宿主没有 .header 属性时才降级尝试 requestHeader().config(旧形状)。
|
|
49
|
-
let header = null;
|
|
50
|
-
try {
|
|
51
|
-
header = session?.header ?? session?.requestHeader?.()?.config ?? null;
|
|
52
|
-
} catch {
|
|
53
|
-
header = null;
|
|
54
|
-
}
|
|
55
|
-
|
|
56
|
-
let agentScope = null;
|
|
57
|
-
const preset = header?.agentPreset;
|
|
58
|
-
if (typeof preset === "string" && preset.trim()) agentScope = preset.trim();
|
|
59
|
-
|
|
60
|
-
// workspace:registry 反查优先(成员资格 = id 命中 + header canonical cwd
|
|
61
|
-
// 校验过,path 已 realpath 归一),失败回退 header.cwd,再失败留 NULL。
|
|
62
|
-
let workspaceScope = null;
|
|
63
|
-
try {
|
|
64
|
-
const list = ctx?.workspaceRegistry?.list?.();
|
|
65
|
-
const sessionId = session?.id;
|
|
66
|
-
if (Array.isArray(list) && sessionId != null) {
|
|
67
|
-
const hit = list.find(
|
|
68
|
-
(ws) => Array.isArray(ws?.sessionIds) && ws.sessionIds.includes(sessionId)
|
|
69
|
-
);
|
|
70
|
-
if (typeof hit?.path === "string" && hit.path.trim()) workspaceScope = hit.path.trim();
|
|
71
|
-
}
|
|
72
|
-
} catch (e) {
|
|
73
|
-
warnOnce(
|
|
74
|
-
`[dsh-mneme] workspaceRegistry lookup failed, falling back to session header cwd: ${String(e)}`
|
|
75
|
-
);
|
|
76
|
-
}
|
|
77
|
-
if (!workspaceScope) {
|
|
78
|
-
const cwd = header?.cwd;
|
|
79
|
-
if (typeof cwd === "string" && cwd.trim()) workspaceScope = cwd.trim();
|
|
80
|
-
}
|
|
81
|
-
|
|
82
|
-
return { agent_scope: agentScope, workspace_scope: workspaceScope };
|
|
83
|
-
};
|
|
84
|
-
}
|
|
85
|
-
|
|
86
|
-
/**
|
|
87
|
-
* v0.8.1 底座(issue #170):显式 scope 参数(memory_save / memory_update /
|
|
88
|
-
* 面板编辑)的归一化。约定:
|
|
89
|
-
* - 省略(undefined,由调用方判断)= 该维不声明,回落载体自动标注;
|
|
90
|
-
* - "global"(大小写不敏感)/ "*" / 空白 = 显式声明全局 → 存储 NULL
|
|
91
|
-
* (行上的 agent_scope_source='explicit' 承担「显式全局」与「从未标注」
|
|
92
|
-
* 的区分,NULL 本身继续表示全局可见);
|
|
93
|
-
* - 其余非空 trim 字符串 = 收窄到该标签。
|
|
94
|
-
* 脏输入(非字符串)落 NULL(等同声明全局)——与 resolveWriteScope 的「解析
|
|
95
|
-
* 绝不阻塞写入」同一哲学:显式声明是用户意图,形状不对时宁可放宽不报错。
|
|
96
|
-
*
|
|
97
|
-
* @param {unknown} raw 工具/HTTP 参数原值
|
|
98
|
-
* @returns {string|null} 归一化后的标签值(null=显式全局)
|
|
99
|
-
*/
|
|
100
|
-
export function normalizeExplicitScope(raw) {
|
|
101
|
-
if (typeof raw !== "string") return null;
|
|
102
|
-
const s = raw.trim();
|
|
103
|
-
if (!s || s === "*" || s.toLowerCase() === "global") return null;
|
|
104
|
-
return s;
|
|
105
|
-
}
|
|
106
|
-
|
|
107
|
-
/**
|
|
108
|
-
* scope 标注的比较键(去重键/检索加权/跨 scope 配对共用口径):非空 trim 字符串
|
|
109
|
-
* 原样返回,其余(NULL/undefined/空串/非字符串)统一为 null(「未标注」)。
|
|
110
|
-
* v0.8.1(issue #170 第 2 步)从 service.js 上提为共享定义——sleep 的跨 scope
|
|
111
|
-
* 候选配对必须与去重键用同一把钥匙,NULL 与已标注才不会错配。
|
|
112
|
-
*
|
|
113
|
-
* @param {unknown} v scope 标注原值
|
|
114
|
-
* @returns {string|null} 比较键(null=未标注)
|
|
115
|
-
*/
|
|
116
|
-
export function scopeKeyOf(v) {
|
|
117
|
-
return typeof v === "string" && v.trim() ? v.trim() : null;
|
|
118
|
-
}
|
|
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
|
+
let session = null;
|
|
38
|
+
try {
|
|
39
|
+
session = exec?.agent?.session ?? null;
|
|
40
|
+
} catch {
|
|
41
|
+
session = null;
|
|
42
|
+
}
|
|
43
|
+
|
|
44
|
+
// 持久化 header(issue #17 的身份契约):session.header 是创建时冻结的
|
|
45
|
+
// SessionHeader(含 agentPreset/cwd)。⚠️ 不能先读 requestHeader()——那返回
|
|
46
|
+
// EpochHeader(请求级模型路由头,首个请求后即为真值),会把 session.header
|
|
47
|
+
// 遮蔽掉,agentPreset/cwd 双双取空(本地 web 复验 2026-09-13 实测踩中)。
|
|
48
|
+
// 仅当宿主没有 .header 属性时才降级尝试 requestHeader().config(旧形状)。
|
|
49
|
+
let header = null;
|
|
50
|
+
try {
|
|
51
|
+
header = session?.header ?? session?.requestHeader?.()?.config ?? null;
|
|
52
|
+
} catch {
|
|
53
|
+
header = null;
|
|
54
|
+
}
|
|
55
|
+
|
|
56
|
+
let agentScope = null;
|
|
57
|
+
const preset = header?.agentPreset;
|
|
58
|
+
if (typeof preset === "string" && preset.trim()) agentScope = preset.trim();
|
|
59
|
+
|
|
60
|
+
// workspace:registry 反查优先(成员资格 = id 命中 + header canonical cwd
|
|
61
|
+
// 校验过,path 已 realpath 归一),失败回退 header.cwd,再失败留 NULL。
|
|
62
|
+
let workspaceScope = null;
|
|
63
|
+
try {
|
|
64
|
+
const list = ctx?.workspaceRegistry?.list?.();
|
|
65
|
+
const sessionId = session?.id;
|
|
66
|
+
if (Array.isArray(list) && sessionId != null) {
|
|
67
|
+
const hit = list.find(
|
|
68
|
+
(ws) => Array.isArray(ws?.sessionIds) && ws.sessionIds.includes(sessionId)
|
|
69
|
+
);
|
|
70
|
+
if (typeof hit?.path === "string" && hit.path.trim()) workspaceScope = hit.path.trim();
|
|
71
|
+
}
|
|
72
|
+
} catch (e) {
|
|
73
|
+
warnOnce(
|
|
74
|
+
`[dsh-mneme] workspaceRegistry lookup failed, falling back to session header cwd: ${String(e)}`
|
|
75
|
+
);
|
|
76
|
+
}
|
|
77
|
+
if (!workspaceScope) {
|
|
78
|
+
const cwd = header?.cwd;
|
|
79
|
+
if (typeof cwd === "string" && cwd.trim()) workspaceScope = cwd.trim();
|
|
80
|
+
}
|
|
81
|
+
|
|
82
|
+
return { agent_scope: agentScope, workspace_scope: workspaceScope };
|
|
83
|
+
};
|
|
84
|
+
}
|
|
85
|
+
|
|
86
|
+
/**
|
|
87
|
+
* v0.8.1 底座(issue #170):显式 scope 参数(memory_save / memory_update /
|
|
88
|
+
* 面板编辑)的归一化。约定:
|
|
89
|
+
* - 省略(undefined,由调用方判断)= 该维不声明,回落载体自动标注;
|
|
90
|
+
* - "global"(大小写不敏感)/ "*" / 空白 = 显式声明全局 → 存储 NULL
|
|
91
|
+
* (行上的 agent_scope_source='explicit' 承担「显式全局」与「从未标注」
|
|
92
|
+
* 的区分,NULL 本身继续表示全局可见);
|
|
93
|
+
* - 其余非空 trim 字符串 = 收窄到该标签。
|
|
94
|
+
* 脏输入(非字符串)落 NULL(等同声明全局)——与 resolveWriteScope 的「解析
|
|
95
|
+
* 绝不阻塞写入」同一哲学:显式声明是用户意图,形状不对时宁可放宽不报错。
|
|
96
|
+
*
|
|
97
|
+
* @param {unknown} raw 工具/HTTP 参数原值
|
|
98
|
+
* @returns {string|null} 归一化后的标签值(null=显式全局)
|
|
99
|
+
*/
|
|
100
|
+
export function normalizeExplicitScope(raw) {
|
|
101
|
+
if (typeof raw !== "string") return null;
|
|
102
|
+
const s = raw.trim();
|
|
103
|
+
if (!s || s === "*" || s.toLowerCase() === "global") return null;
|
|
104
|
+
return s;
|
|
105
|
+
}
|
|
106
|
+
|
|
107
|
+
/**
|
|
108
|
+
* scope 标注的比较键(去重键/检索加权/跨 scope 配对共用口径):非空 trim 字符串
|
|
109
|
+
* 原样返回,其余(NULL/undefined/空串/非字符串)统一为 null(「未标注」)。
|
|
110
|
+
* v0.8.1(issue #170 第 2 步)从 service.js 上提为共享定义——sleep 的跨 scope
|
|
111
|
+
* 候选配对必须与去重键用同一把钥匙,NULL 与已标注才不会错配。
|
|
112
|
+
*
|
|
113
|
+
* @param {unknown} v scope 标注原值
|
|
114
|
+
* @returns {string|null} 比较键(null=未标注)
|
|
115
|
+
*/
|
|
116
|
+
export function scopeKeyOf(v) {
|
|
117
|
+
return typeof v === "string" && v.trim() ? v.trim() : null;
|
|
118
|
+
}
|
package/lib/search/adaptive.js
CHANGED
|
@@ -1,22 +1,22 @@
|
|
|
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
|
-
}
|
|
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
|
+
}
|
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
|
+
}
|