@sanqianx/project-knowledge 5.0.18 → 5.0.20-rc.1

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 (209) hide show
  1. package/.agents/plugins/marketplace.json +3 -3
  2. package/.claude-plugin/marketplace.json +1 -1
  3. package/CHANGELOG.md +14 -0
  4. package/LICENSE +201 -201
  5. package/NOTICE +5 -5
  6. package/README.md +349 -349
  7. package/_modules/knowledge-engine/README.md +4 -0
  8. package/_modules/knowledge-engine/src/analysis-contract.js +4 -3
  9. package/_modules/knowledge-engine/src/analysis-failures.js +39 -0
  10. package/_modules/knowledge-engine/src/analysis-worker.js +6 -2
  11. package/_modules/knowledge-engine/src/coordinator.js +8 -0
  12. package/_modules/knowledge-engine/src/headless-analyzer.js +2 -1
  13. package/_modules/knowledge-engine/src/memories.js +10 -5
  14. package/_modules/knowledge-engine/src/memory-eligibility.js +74 -0
  15. package/_modules/knowledge-engine/src/retrieval-ranking.js +19 -1
  16. package/_modules/knowledge-engine/src/runtime/commit-conversation-binder.js +1 -0
  17. package/_modules/knowledge-engine/src/runtime/commit-prompt.js +12 -9
  18. package/_modules/knowledge-engine/src/runtime/conversation-store.js +2 -0
  19. package/_modules/knowledge-engine/src/server.js +1 -1
  20. package/_site/_test/agent-capture-mcp-ui-test.js +67 -67
  21. package/_site/_test/ai-profile-test.js +75 -75
  22. package/_site/_test/ai-vendor-presets-test.js +91 -91
  23. package/_site/_test/automatic-context-test.js +70 -70
  24. package/_site/_test/bin-cli-test.js +230 -230
  25. package/_site/_test/bridge-adapter-test.js +95 -95
  26. package/_site/_test/bridge-consumer-service-test.js +310 -310
  27. package/_site/_test/candidate-bin-content-test.js +35 -0
  28. package/_site/_test/candidate-product-assets-test.js +13 -0
  29. package/_site/_test/candidate-release-contract-test.js +14 -14
  30. package/_site/_test/capture-hook-dedup-test.js +158 -158
  31. package/_site/_test/claude-executable-discovery-test.js +28 -28
  32. package/_site/_test/claude-session-lifecycle-sweep-test.js +252 -252
  33. package/_site/_test/cli-instance-management-test.js +62 -62
  34. package/_site/_test/client-capture-restart-e2e-test.js +189 -189
  35. package/_site/_test/codex-conversation-projection-test.js +60 -60
  36. package/_site/_test/codex-marketplace-scoped-path-test.js +186 -186
  37. package/_site/_test/codex-notify-capture-e2e-test.js +101 -101
  38. package/_site/_test/codex-notify-fanout-test.js +134 -134
  39. package/_site/_test/codex-notify-upgrade-test.js +253 -253
  40. package/_site/_test/commit-boundary-freeze-test.js +147 -147
  41. package/_site/_test/commit-conversation-binding-test.js +4 -4
  42. package/_site/_test/configuration-generation-test.js +22 -22
  43. package/_site/_test/context-pack-test.js +200 -200
  44. package/_site/_test/conversation-api-test.js +10 -10
  45. package/_site/_test/conversation-origin-test.js +15 -0
  46. package/_site/_test/cross-repo-isolation-e2e-test.js +187 -187
  47. package/_site/_test/embedding-config-test.js +32 -32
  48. package/_site/_test/fixtures/fake-browser-exit.js +10 -10
  49. package/_site/_test/fixtures/fake-browser-hang.js +10 -10
  50. package/_site/_test/fixtures/integration-busy-worker.cjs +6 -6
  51. package/_site/_test/fixtures/make-git-repos.js +92 -92
  52. package/_site/_test/fixtures/retrieval-gold-queries.json +32 -0
  53. package/_site/_test/fresh-hook-reimport-test.js +53 -53
  54. package/_site/_test/fresh-product-data-test.js +56 -56
  55. package/_site/_test/git-path-compatibility-test.js +80 -80
  56. package/_site/_test/git-validation-test.js +81 -81
  57. package/_site/_test/hook-worktree-test.js +42 -42
  58. package/_site/_test/index-builder-test.js +41 -41
  59. package/_site/_test/index-error-display-contract-test.js +10 -10
  60. package/_site/_test/integration-adapters-test.js +67 -67
  61. package/_site/_test/integration-capture-setup-test.js +118 -118
  62. package/_site/_test/integration-runtime-binding-a2-test.js +566 -566
  63. package/_site/_test/integration-runtime-binding-a3-test.js +227 -227
  64. package/_site/_test/integration-runtime-binding-test.js +473 -473
  65. package/_site/_test/integration-status-scan-test.js +29 -29
  66. package/_site/_test/kb-framework-test.js +57 -57
  67. package/_site/_test/kbpath-follow-test.js +63 -63
  68. package/_site/_test/knowledge-db-test.js +144 -144
  69. package/_site/_test/knowledge-engine-client-test.js +109 -109
  70. package/_site/_test/knowledge-maintenance-test.js +40 -40
  71. package/_site/_test/knowledge-panel-ui-contract-test.js +112 -0
  72. package/_site/_test/knowledge-scopes-test.js +37 -37
  73. package/_site/_test/knowledge-storage-location-test.js +28 -28
  74. package/_site/_test/launcher-readiness-test.js +80 -80
  75. package/_site/_test/legacy-claude-session-alias-test.js +106 -106
  76. package/_site/_test/legacy-routes-removed-test.js +47 -47
  77. package/_site/_test/legacy-workbench-exclusion-test.js +92 -92
  78. package/_site/_test/live-gateway-test.js +48 -48
  79. package/_site/_test/local-marketplace-test.js +16 -16
  80. package/_site/_test/log-redaction-test.js +58 -58
  81. package/_site/_test/markdown-maintenance-test.js +11 -11
  82. package/_site/_test/model-context-windows-test.js +89 -89
  83. package/_site/_test/module-agent-start-proxy-test.js +65 -65
  84. package/_site/_test/module-artifact-integrity-test.js +27 -27
  85. package/_site/_test/module-boundary-test.js +30 -30
  86. package/_site/_test/module-bridge-eventbridge-test.js +202 -202
  87. package/_site/_test/module-dynamic-port-test.js +118 -118
  88. package/_site/_test/module-index-proxy-test.js +33 -0
  89. package/_site/_test/module-model-configuration-test.js +44 -44
  90. package/_site/_test/module-process-lifecycle-test.js +35 -35
  91. package/_site/_test/module-stream-proxy-test.js +34 -34
  92. package/_site/_test/module-terminal-restart-test.js +63 -0
  93. package/_site/_test/module-wiring-unbind-race-test.js +59 -59
  94. package/_site/_test/native-context-integrations-test.js +86 -86
  95. package/_site/_test/orphaned-claim-recovery-test.js +80 -80
  96. package/_site/_test/package-boundary-test.js +43 -42
  97. package/_site/_test/package-startup-test.js +56 -56
  98. package/_site/_test/packaged-ui-smoke-test.js +88 -88
  99. package/_site/_test/pr-consumer-contract-test.js +55 -55
  100. package/_site/_test/product-brand-contract-test.js +25 -25
  101. package/_site/_test/product-diagnostics-ui-test.js +93 -93
  102. package/_site/_test/product-import-ui-test.js +104 -104
  103. package/_site/_test/product-progress-status-test.js +12 -1
  104. package/_site/_test/project-remove-running-guard-test.js +79 -79
  105. package/_site/_test/project-store-test.js +90 -90
  106. package/_site/_test/registry-events-test.js +38 -0
  107. package/_site/_test/release-version-sync-test.js +115 -115
  108. package/_site/_test/requirement-binding-test.js +113 -113
  109. package/_site/_test/runtime-endpoint-test.js +64 -64
  110. package/_site/_test/scanner-test.js +83 -83
  111. package/_site/_test/server-security-test.js +96 -96
  112. package/_site/_test/sessions-stream-test.js +93 -93
  113. package/_site/_test/shared-scope-root-binding-test.js +67 -67
  114. package/_site/_test/simple-import-test.js +98 -98
  115. package/_site/_test/startup-analysis-disabled-test.js +6 -6
  116. package/_site/_test/team-knowledge-store-sparse-checkout-test.js +196 -196
  117. package/_site/_test/team-knowledge-store-test.js +132 -132
  118. package/_site/_test/test-server-cleanup-test.js +35 -35
  119. package/_site/_test/test-server-module-isolation-test.js +37 -37
  120. package/_site/_test/tracking-start-test.js +35 -35
  121. package/_site/_test/workbench-context-receipts-test.js +22 -0
  122. package/_site/_test/workspace-ui-contract-test.js +31 -31
  123. package/_site/_test/zcode-capture-integration-test.js +108 -108
  124. package/_site/lib/ai-vendor-presets.js +83 -83
  125. package/_site/lib/ai-workspace.js +3 -3
  126. package/_site/lib/atomic-file.js +204 -204
  127. package/_site/lib/bridge-adapter.js +260 -260
  128. package/_site/lib/capture-hook-config.js +108 -108
  129. package/_site/lib/claude-cli-runner.js +89 -89
  130. package/_site/lib/cli-instances.js +43 -43
  131. package/_site/lib/codex-context-status.js +17 -17
  132. package/_site/lib/codex-notify-config.js +193 -193
  133. package/_site/lib/commit-conversation-binder.js +13 -12
  134. package/_site/lib/commit-reconciler.js +38 -38
  135. package/_site/lib/context-pack-builder.js +299 -299
  136. package/_site/lib/conversation-exclusions.js +204 -204
  137. package/_site/lib/conversation-query-service.js +17 -17
  138. package/_site/lib/conversation-store.js +2 -0
  139. package/_site/lib/data-dir.js +13 -13
  140. package/_site/lib/embedding-service.js +85 -85
  141. package/_site/lib/engine-connection.js +93 -90
  142. package/_site/lib/engine-knowledge-client.js +72 -72
  143. package/_site/lib/folder-picker-output.js +3 -3
  144. package/_site/lib/git-path.js +14 -14
  145. package/_site/lib/github-team-store.js +3 -3
  146. package/_site/lib/index-builder.js +157 -157
  147. package/_site/lib/integration-status-scan.js +99 -99
  148. package/_site/lib/integration-status-worker.js +14 -14
  149. package/_site/lib/kb-framework.js +15 -15
  150. package/_site/lib/knowledge-db.js +363 -363
  151. package/_site/lib/knowledge-schema.js +121 -121
  152. package/_site/lib/live-gateway.js +78 -76
  153. package/_site/lib/llm-client.js +12 -12
  154. package/_site/lib/model-context-windows.js +82 -82
  155. package/_site/lib/module-bridge.js +874 -865
  156. package/_site/lib/module-port-resolver.js +57 -57
  157. package/_site/lib/post-commit-automation.js +54 -54
  158. package/_site/lib/product-data.js +35 -35
  159. package/_site/lib/project-lifecycle-service.js +14 -14
  160. package/_site/lib/project-registry-store.js +129 -129
  161. package/_site/lib/prompt-registry.js +148 -148
  162. package/_site/lib/registry-events.js +16 -0
  163. package/_site/lib/requirement-adapters.js +10 -10
  164. package/_site/lib/requirement-binder.js +112 -112
  165. package/_site/lib/runtime-control.js +89 -89
  166. package/_site/lib/runtime-endpoint.js +5 -5
  167. package/_site/lib/server-app.js +10 -1
  168. package/_site/lib/zcode-context-integration.js +60 -60
  169. package/_site/scripts/codex-notify-fanout.cjs +95 -95
  170. package/_site/scripts/install-module-candidates.js +69 -69
  171. package/_site/scripts/pack-module-candidates.js +89 -72
  172. package/_site/scripts/sync-product-content.js +28 -28
  173. package/_site/scripts/sync-release-version.js +164 -164
  174. package/_site/scripts/vendor-modules.js +49 -49
  175. package/_site/scripts/verify-module-runtime.js +50 -50
  176. package/_site/scripts/verify-product-runtime.js +36 -36
  177. package/_site/scripts/verify-retrieval-gold.js +83 -0
  178. package/_site/server.js +15 -15
  179. package/_site/start.bat +26 -26
  180. package/_site/stop.bat +11 -11
  181. package/bin/project-knowledge-integrations.js +254 -254
  182. package/bin/project-knowledge-kb.js +67 -67
  183. package/bin/project-knowledge-mcp.js +8 -8
  184. package/bin/project-knowledge.js +170 -170
  185. package/claude-prompts.json +21 -21
  186. package/docs/README.zh-CN.md +205 -205
  187. package/docs/assets/architecture.svg +181 -181
  188. package/docs/assets/logo.svg +49 -49
  189. package/docs/fresh-v5-start.md +42 -42
  190. package/docs/leader-glm-workbench-playbook.md +122 -119
  191. package/docs/project-registry-schema.md +11 -11
  192. package/docs/release-5.0.16-acceptance.md +176 -176
  193. package/docs/release-5.0.17-acceptance.md +117 -117
  194. package/docs/release-5.0.19-acceptance.md +39 -0
  195. package/docs/team-knowledge-mode-a-plan.md +73 -73
  196. package/module-runtime-manifest.json +126 -104
  197. package/package.json +110 -108
  198. package/plugins/project-knowledge/.claude-plugin/plugin.json +1 -1
  199. package/plugins/project-knowledge/.codex-plugin/plugin.json +1 -1
  200. package/plugins/project-knowledge/.mcp.json +1 -1
  201. package/plugins/project-knowledge/opencode/project-knowledge.md +3 -3
  202. package/plugins/project-knowledge/skills/project-knowledge/SKILL.md +28 -28
  203. package/templates/change.md +30 -30
  204. package/templates/project-readme.md +18 -18
  205. package/ui/index.html +1 -1
  206. package/ui/knowledge-panel.html +118 -0
  207. package/ui/product.css +8 -0
  208. package/ui/product.js +175 -8
  209. package/vendor-manifest.json +55 -53
@@ -1,363 +1,363 @@
1
- const fs = require('fs');
2
- const path = require('path');
3
- const lancedb = require('@lancedb/lancedb');
4
- const {
5
- KNOWLEDGE_TABLE,
6
- EMBEDDING_DIMENSIONS,
7
- knowledgeChunkSchema,
8
- normalizeKnowledgeChunk,
9
- decodeKnowledgeChunk,
10
- } = require('./knowledge-schema');
11
-
12
- const FTS_SCHEMA_VERSION = 2;
13
- const DEFAULT_OPTIMIZE_OPERATIONS = 20;
14
- const DEFAULT_OPTIMIZE_ROWS = 100000;
15
-
16
- function readJson(filePath, fallback) {
17
- try {
18
- const value = JSON.parse(fs.readFileSync(filePath, 'utf8'));
19
- return value && typeof value === 'object' ? value : fallback;
20
- } catch {
21
- return fallback;
22
- }
23
- }
24
-
25
- function writeJsonAtomic(filePath, value) {
26
- fs.mkdirSync(path.dirname(filePath), { recursive: true });
27
- const temp = `${filePath}.${process.pid}.tmp`;
28
- fs.writeFileSync(temp, `${JSON.stringify(value, null, 2)}\n`, 'utf8');
29
- fs.renameSync(temp, filePath);
30
- }
31
-
32
- function sqlString(value) {
33
- return `'${String(value).replace(/'/g, "''")}'`;
34
- }
35
-
36
- function inPredicate(column, values) {
37
- const clean = Array.from(new Set((values || []).map(value => String(value).trim()).filter(Boolean)));
38
- if (!clean.length) throw new Error(`${column} filter must not be empty`);
39
- return `${column} IN (${clean.map(sqlString).join(', ')})`;
40
- }
41
-
42
- class KnowledgeDatabase {
43
- constructor(options = {}) {
44
- if (!options.dbPath) throw new Error('dbPath is required');
45
- this.dbPath = path.resolve(options.dbPath);
46
- this.tableName = options.tableName || KNOWLEDGE_TABLE;
47
- this.dimensions = options.dimensions || EMBEDDING_DIMENSIONS;
48
- this.maintenancePath = options.maintenancePath || `${this.dbPath}.maintenance.json`;
49
- this.optimizeOperationsThreshold = Number(options.optimizeOperationsThreshold || DEFAULT_OPTIMIZE_OPERATIONS);
50
- this.optimizeRowsThreshold = Number(options.optimizeRowsThreshold || DEFAULT_OPTIMIZE_ROWS);
51
- this.connection = null;
52
- this.table = null;
53
- }
54
-
55
- maintenanceState() {
56
- return {
57
- schema: 'project-knowledge/lancedb-maintenance/v1',
58
- ftsSchemaVersion: 0,
59
- ftsUpgradeRequired: false,
60
- pendingOperations: 0,
61
- pendingRows: 0,
62
- lastOptimizedAt: null,
63
- ...readJson(this.maintenancePath, {}),
64
- };
65
- }
66
-
67
- saveMaintenanceState(patch = {}) {
68
- const state = { ...this.maintenanceState(), ...patch };
69
- writeJsonAtomic(this.maintenancePath, state);
70
- return state;
71
- }
72
-
73
- noteMutations(input = {}) {
74
- const operations = Math.max(0, Number(input.operations || 0));
75
- const rows = Math.max(0, Number(input.rows || 0));
76
- if (!operations && !rows) return this.maintenanceState();
77
- const current = this.maintenanceState();
78
- return this.saveMaintenanceState({
79
- pendingOperations: current.pendingOperations + operations,
80
- pendingRows: current.pendingRows + rows,
81
- });
82
- }
83
-
84
- async open() {
85
- if (this.table) return this;
86
- fs.mkdirSync(this.dbPath, { recursive: true });
87
- this.connection = await lancedb.connect(this.dbPath);
88
- const names = await this.connection.tableNames();
89
- this.table = names.includes(this.tableName)
90
- ? await this.connection.openTable(this.tableName)
91
- : await this.connection.createEmptyTable(this.tableName, knowledgeChunkSchema(this.dimensions));
92
- return this;
93
- }
94
-
95
- async close() {
96
- this.table = null;
97
- if (this.connection) this.connection.close();
98
- this.connection = null;
99
- }
100
-
101
- async count(spaceIds = null) {
102
- await this.open();
103
- return this.table.countRows(spaceIds ? inPredicate('space_id', spaceIds) : undefined);
104
- }
105
-
106
- async rowsForEntry(spaceId, entryId) {
107
- await this.open();
108
- const rows = await this.table.query()
109
- .where(`space_id = ${sqlString(spaceId)} AND entry_id = ${sqlString(entryId)}`)
110
- .toArray();
111
- return rows.map(decodeKnowledgeChunk);
112
- }
113
-
114
- async entryIds(spaceId) {
115
- await this.open();
116
- const rows = await this.table.query()
117
- .where(`space_id = ${sqlString(spaceId)}`)
118
- .select(['entry_id'])
119
- .toArray();
120
- return Array.from(new Set(rows.map(row => String(row.entry_id)))).sort();
121
- }
122
-
123
- async allRows() {
124
- await this.open();
125
- return this.table.query().toArray();
126
- }
127
-
128
- async addRows(rows) {
129
- await this.open();
130
- if (!Array.isArray(rows) || !rows.length) return { ok: true, added: 0 };
131
- if (await this.table.countRows()) throw new Error('bulk destination table must be empty');
132
- await this.table.add(rows);
133
- return { ok: true, added: rows.length };
134
- }
135
-
136
- async versions() {
137
- await this.open();
138
- return this.table.listVersions();
139
- }
140
-
141
- async indexDetails() {
142
- await this.open();
143
- const indices = await this.table.listIndices();
144
- const out = [];
145
- for (const index of indices) {
146
- out.push({ ...index, statistics: await this.table.indexStats(index.name) });
147
- }
148
- return out;
149
- }
150
-
151
- async getEntry(entryId, options = {}) {
152
- await this.open();
153
- const predicates = [inPredicate('space_id', options.spaceIds || []), `entry_id = ${sqlString(entryId)}`];
154
- if (options.spaceId) predicates.push(`space_id = ${sqlString(options.spaceId)}`);
155
- const rows = await this.table.query().where(predicates.join(' AND ')).toArray();
156
- return rows.map(decodeKnowledgeChunk).sort((a, b) => a.chunk_order - b.chunk_order);
157
- }
158
-
159
- async history(options = {}) {
160
- await this.open();
161
- const limit = Math.max(1, Math.min(Number(options.limit || 20), 100));
162
- const rows = await this.table.query()
163
- .where(`${inPredicate('space_id', options.spaceIds || [])} AND entry_type = 'change'`)
164
- .limit(limit * 4)
165
- .toArray();
166
- const entries = new Map();
167
- for (const row of rows.map(decodeKnowledgeChunk)) {
168
- const current = entries.get(row.record_id) || row;
169
- if (String(row.updated_at) > String(current.updated_at)) entries.set(row.record_id, row);
170
- else if (!entries.has(row.record_id)) entries.set(row.record_id, row);
171
- }
172
- return Array.from(entries.values())
173
- .sort((a, b) => String(b.updated_at).localeCompare(String(a.updated_at)))
174
- .slice(0, limit);
175
- }
176
-
177
- async replaceEntry(spaceId, entryId, chunks) {
178
- await this.open();
179
- if (!Array.isArray(chunks) || chunks.length === 0) {
180
- const result = await this.deleteEntry(spaceId, entryId);
181
- return { ok: true, action: 'deleted', deleted: result.deleted || 0, upserted: 0 };
182
- }
183
- const now = new Date().toISOString();
184
- const existing = await this.rowsForEntry(spaceId, entryId);
185
- const existingById = new Map(existing.map(row => [row.record_id, row]));
186
- const normalized = chunks.map(chunk => normalizeKnowledgeChunk({ ...chunk, spaceId, entryId }, {
187
- dimensions: this.dimensions,
188
- now,
189
- })).map(row => ({
190
- ...row,
191
- created_at: existingById.get(row.record_id)?.created_at || row.created_at,
192
- }));
193
-
194
- const unchanged = existing.length === normalized.length && normalized.every(row => {
195
- const old = existingById.get(row.record_id);
196
- return old && old.content_hash === row.content_hash && old.source_commit === row.source_commit;
197
- });
198
- if (unchanged) return { ok: true, action: 'unchanged', upserted: 0, deleted: 0, total: normalized.length };
199
-
200
- await this.table.mergeInsert('record_id')
201
- .whenMatchedUpdateAll()
202
- .whenNotMatchedInsertAll()
203
- .execute(normalized);
204
-
205
- const keep = new Set(normalized.map(row => row.record_id));
206
- const stale = existing.filter(row => !keep.has(row.record_id));
207
- if (stale.length) {
208
- await this.table.delete(`record_id IN (${stale.map(row => sqlString(row.record_id)).join(', ')})`);
209
- }
210
- return { ok: true, action: existing.length ? 'updated' : 'created', upserted: normalized.length, deleted: stale.length, total: normalized.length };
211
- }
212
-
213
- async deleteEntry(spaceId, entryId) {
214
- await this.open();
215
- const before = await this.table.countRows(`space_id = ${sqlString(spaceId)} AND entry_id = ${sqlString(entryId)}`);
216
- if (before) await this.table.delete(`space_id = ${sqlString(spaceId)} AND entry_id = ${sqlString(entryId)}`);
217
- return { ok: true, deleted: before };
218
- }
219
-
220
- async deleteSpace(spaceId) {
221
- await this.open();
222
- const before = await this.table.countRows(`space_id = ${sqlString(spaceId)}`);
223
- if (before) await this.table.delete(`space_id = ${sqlString(spaceId)}`);
224
- return { ok: true, deleted: before };
225
- }
226
-
227
- async vectorSearch(vector, options = {}) {
228
- await this.open();
229
- const spaceIds = options.spaceIds || [];
230
- const limit = Math.max(1, Math.min(Number(options.limit || 10), 100));
231
- const query = this.table.query()
232
- .nearestTo(Array.from(vector || [], Number))
233
- .column('vector')
234
- .distanceType('cosine')
235
- .where(inPredicate('space_id', spaceIds))
236
- .limit(limit);
237
- const rows = await query.toArray();
238
- return rows.map(decodeKnowledgeChunk);
239
- }
240
-
241
- async ensureSearchIndexes() {
242
- await this.open();
243
- if (await this.table.countRows() === 0) return { ok: true, created: [], skipped: 'empty-table' };
244
- const existing = new Set((await this.table.listIndices()).map(index => index.name));
245
- let state = this.maintenanceState();
246
- const created = [];
247
- // A v4.0.0 FTS index can be very large. Rebuilding it in place would keep
248
- // both generations alive until Lance's old-version retention expires and
249
- // could exhaust the user's disk. Mark it for the atomic rebuild workflow
250
- // instead. New databases create the compact v2 index immediately.
251
- if (existing.has('search_text_idx') && !existing.has('chunk_text_idx') && state.ftsSchemaVersion === 0) {
252
- state = this.saveMaintenanceState({ ftsSchemaVersion: 1, ftsUpgradeRequired: true });
253
- }
254
- const definitions = [
255
- ['space_id', 'space_id_idx', lancedb.Index.bitmap()],
256
- ['entry_id', 'entry_id_idx', lancedb.Index.btree()],
257
- ];
258
- if (!existing.has('search_text_idx') || existing.has('chunk_text_idx')) {
259
- definitions.push(['chunk_text', 'chunk_text_idx', lancedb.Index.fts({
260
- baseTokenizer: 'ngram',
261
- ngramMinLength: 2,
262
- // Chinese bigrams preserve exact keyword recall without creating the
263
- // duplicate 2/3/4-gram postings used by v4.0.0.
264
- ngramMaxLength: 2,
265
- withPosition: false,
266
- stem: false,
267
- removeStopWords: false,
268
- })]);
269
- }
270
- for (const [column, name, config] of definitions) {
271
- if (existing.has(name)) continue;
272
- await this.table.createIndex(column, { name, config, replace: false, waitTimeoutSeconds: 60 });
273
- created.push(name);
274
- }
275
- if (created.includes('chunk_text_idx') || existing.has('chunk_text_idx')) {
276
- state = this.saveMaintenanceState({ ftsSchemaVersion: FTS_SCHEMA_VERSION, ftsUpgradeRequired: false });
277
- }
278
- return {
279
- ok: true,
280
- created,
281
- ftsSchemaVersion: state.ftsSchemaVersion,
282
- ftsUpgradeRequired: state.ftsUpgradeRequired === true,
283
- };
284
- }
285
-
286
- async optimize(options = {}) {
287
- await this.open();
288
- return this.table.optimize(options);
289
- }
290
-
291
- async maybeOptimize(options = {}) {
292
- const state = this.maintenanceState();
293
- if (state.ftsUpgradeRequired === true && options.allowLegacyFts !== true) {
294
- return {
295
- ok: true,
296
- optimized: false,
297
- due: false,
298
- blockedBy: 'fts-upgrade-required',
299
- state,
300
- };
301
- }
302
- const due = options.force === true
303
- || state.pendingOperations >= this.optimizeOperationsThreshold
304
- || state.pendingRows >= this.optimizeRowsThreshold;
305
- if (!due || (!state.pendingOperations && !state.pendingRows)) {
306
- return { ok: true, optimized: false, due: false, state };
307
- }
308
- const stats = await this.optimize();
309
- const next = this.saveMaintenanceState({
310
- pendingOperations: 0,
311
- pendingRows: 0,
312
- lastOptimizedAt: new Date().toISOString(),
313
- });
314
- return { ok: true, optimized: true, due: true, stats, state: next };
315
- }
316
-
317
- async fullTextSearch(text, options = {}) {
318
- await this.open();
319
- const queryText = String(text || '').trim();
320
- if (!queryText) throw new Error('search text is required');
321
- const limit = Math.max(1, Math.min(Number(options.limit || 10), 100));
322
- const ftsColumn = this.maintenanceState().ftsSchemaVersion >= FTS_SCHEMA_VERSION
323
- ? 'chunk_text'
324
- : 'search_text';
325
- const rows = await this.table.query()
326
- .fullTextSearch(queryText, { columns: ftsColumn })
327
- .where(inPredicate('space_id', options.spaceIds || []))
328
- .limit(limit)
329
- .toArray();
330
- return rows.map(decodeKnowledgeChunk);
331
- }
332
-
333
- async hybridSearch(input = {}) {
334
- const limit = Math.max(1, Math.min(Number(input.limit || 10), 50));
335
- const candidates = Math.max(limit, Math.min(Number(input.candidates || limit * 4), 200));
336
- const [semantic, keyword] = await Promise.all([
337
- this.vectorSearch(input.vector, { spaceIds: input.spaceIds, limit: candidates }),
338
- this.fullTextSearch(input.text, { spaceIds: input.spaceIds, limit: candidates }),
339
- ]);
340
- const scores = new Map();
341
- const add = (rows, channel, weight) => rows.forEach((row, rank) => {
342
- const current = scores.get(row.record_id) || { row, score: 0, channels: [] };
343
- current.score += weight / (60 + rank + 1);
344
- current.channels.push(channel);
345
- scores.set(row.record_id, current);
346
- });
347
- add(semantic, 'semantic', Number(input.semanticWeight || 0.65));
348
- add(keyword, 'keyword', Number(input.keywordWeight || 0.35));
349
- return Array.from(scores.values())
350
- .sort((a, b) => b.score - a.score || a.row.record_id.localeCompare(b.row.record_id))
351
- .slice(0, limit)
352
- .map(item => ({ ...item.row, relevance_score: item.score, match_channels: item.channels }));
353
- }
354
- }
355
-
356
- module.exports = {
357
- KnowledgeDatabase,
358
- sqlString,
359
- inPredicate,
360
- FTS_SCHEMA_VERSION,
361
- DEFAULT_OPTIMIZE_OPERATIONS,
362
- DEFAULT_OPTIMIZE_ROWS,
363
- };
1
+ const fs = require('fs');
2
+ const path = require('path');
3
+ const lancedb = require('@lancedb/lancedb');
4
+ const {
5
+ KNOWLEDGE_TABLE,
6
+ EMBEDDING_DIMENSIONS,
7
+ knowledgeChunkSchema,
8
+ normalizeKnowledgeChunk,
9
+ decodeKnowledgeChunk,
10
+ } = require('./knowledge-schema');
11
+
12
+ const FTS_SCHEMA_VERSION = 2;
13
+ const DEFAULT_OPTIMIZE_OPERATIONS = 20;
14
+ const DEFAULT_OPTIMIZE_ROWS = 100000;
15
+
16
+ function readJson(filePath, fallback) {
17
+ try {
18
+ const value = JSON.parse(fs.readFileSync(filePath, 'utf8'));
19
+ return value && typeof value === 'object' ? value : fallback;
20
+ } catch {
21
+ return fallback;
22
+ }
23
+ }
24
+
25
+ function writeJsonAtomic(filePath, value) {
26
+ fs.mkdirSync(path.dirname(filePath), { recursive: true });
27
+ const temp = `${filePath}.${process.pid}.tmp`;
28
+ fs.writeFileSync(temp, `${JSON.stringify(value, null, 2)}\n`, 'utf8');
29
+ fs.renameSync(temp, filePath);
30
+ }
31
+
32
+ function sqlString(value) {
33
+ return `'${String(value).replace(/'/g, "''")}'`;
34
+ }
35
+
36
+ function inPredicate(column, values) {
37
+ const clean = Array.from(new Set((values || []).map(value => String(value).trim()).filter(Boolean)));
38
+ if (!clean.length) throw new Error(`${column} filter must not be empty`);
39
+ return `${column} IN (${clean.map(sqlString).join(', ')})`;
40
+ }
41
+
42
+ class KnowledgeDatabase {
43
+ constructor(options = {}) {
44
+ if (!options.dbPath) throw new Error('dbPath is required');
45
+ this.dbPath = path.resolve(options.dbPath);
46
+ this.tableName = options.tableName || KNOWLEDGE_TABLE;
47
+ this.dimensions = options.dimensions || EMBEDDING_DIMENSIONS;
48
+ this.maintenancePath = options.maintenancePath || `${this.dbPath}.maintenance.json`;
49
+ this.optimizeOperationsThreshold = Number(options.optimizeOperationsThreshold || DEFAULT_OPTIMIZE_OPERATIONS);
50
+ this.optimizeRowsThreshold = Number(options.optimizeRowsThreshold || DEFAULT_OPTIMIZE_ROWS);
51
+ this.connection = null;
52
+ this.table = null;
53
+ }
54
+
55
+ maintenanceState() {
56
+ return {
57
+ schema: 'project-knowledge/lancedb-maintenance/v1',
58
+ ftsSchemaVersion: 0,
59
+ ftsUpgradeRequired: false,
60
+ pendingOperations: 0,
61
+ pendingRows: 0,
62
+ lastOptimizedAt: null,
63
+ ...readJson(this.maintenancePath, {}),
64
+ };
65
+ }
66
+
67
+ saveMaintenanceState(patch = {}) {
68
+ const state = { ...this.maintenanceState(), ...patch };
69
+ writeJsonAtomic(this.maintenancePath, state);
70
+ return state;
71
+ }
72
+
73
+ noteMutations(input = {}) {
74
+ const operations = Math.max(0, Number(input.operations || 0));
75
+ const rows = Math.max(0, Number(input.rows || 0));
76
+ if (!operations && !rows) return this.maintenanceState();
77
+ const current = this.maintenanceState();
78
+ return this.saveMaintenanceState({
79
+ pendingOperations: current.pendingOperations + operations,
80
+ pendingRows: current.pendingRows + rows,
81
+ });
82
+ }
83
+
84
+ async open() {
85
+ if (this.table) return this;
86
+ fs.mkdirSync(this.dbPath, { recursive: true });
87
+ this.connection = await lancedb.connect(this.dbPath);
88
+ const names = await this.connection.tableNames();
89
+ this.table = names.includes(this.tableName)
90
+ ? await this.connection.openTable(this.tableName)
91
+ : await this.connection.createEmptyTable(this.tableName, knowledgeChunkSchema(this.dimensions));
92
+ return this;
93
+ }
94
+
95
+ async close() {
96
+ this.table = null;
97
+ if (this.connection) this.connection.close();
98
+ this.connection = null;
99
+ }
100
+
101
+ async count(spaceIds = null) {
102
+ await this.open();
103
+ return this.table.countRows(spaceIds ? inPredicate('space_id', spaceIds) : undefined);
104
+ }
105
+
106
+ async rowsForEntry(spaceId, entryId) {
107
+ await this.open();
108
+ const rows = await this.table.query()
109
+ .where(`space_id = ${sqlString(spaceId)} AND entry_id = ${sqlString(entryId)}`)
110
+ .toArray();
111
+ return rows.map(decodeKnowledgeChunk);
112
+ }
113
+
114
+ async entryIds(spaceId) {
115
+ await this.open();
116
+ const rows = await this.table.query()
117
+ .where(`space_id = ${sqlString(spaceId)}`)
118
+ .select(['entry_id'])
119
+ .toArray();
120
+ return Array.from(new Set(rows.map(row => String(row.entry_id)))).sort();
121
+ }
122
+
123
+ async allRows() {
124
+ await this.open();
125
+ return this.table.query().toArray();
126
+ }
127
+
128
+ async addRows(rows) {
129
+ await this.open();
130
+ if (!Array.isArray(rows) || !rows.length) return { ok: true, added: 0 };
131
+ if (await this.table.countRows()) throw new Error('bulk destination table must be empty');
132
+ await this.table.add(rows);
133
+ return { ok: true, added: rows.length };
134
+ }
135
+
136
+ async versions() {
137
+ await this.open();
138
+ return this.table.listVersions();
139
+ }
140
+
141
+ async indexDetails() {
142
+ await this.open();
143
+ const indices = await this.table.listIndices();
144
+ const out = [];
145
+ for (const index of indices) {
146
+ out.push({ ...index, statistics: await this.table.indexStats(index.name) });
147
+ }
148
+ return out;
149
+ }
150
+
151
+ async getEntry(entryId, options = {}) {
152
+ await this.open();
153
+ const predicates = [inPredicate('space_id', options.spaceIds || []), `entry_id = ${sqlString(entryId)}`];
154
+ if (options.spaceId) predicates.push(`space_id = ${sqlString(options.spaceId)}`);
155
+ const rows = await this.table.query().where(predicates.join(' AND ')).toArray();
156
+ return rows.map(decodeKnowledgeChunk).sort((a, b) => a.chunk_order - b.chunk_order);
157
+ }
158
+
159
+ async history(options = {}) {
160
+ await this.open();
161
+ const limit = Math.max(1, Math.min(Number(options.limit || 20), 100));
162
+ const rows = await this.table.query()
163
+ .where(`${inPredicate('space_id', options.spaceIds || [])} AND entry_type = 'change'`)
164
+ .limit(limit * 4)
165
+ .toArray();
166
+ const entries = new Map();
167
+ for (const row of rows.map(decodeKnowledgeChunk)) {
168
+ const current = entries.get(row.record_id) || row;
169
+ if (String(row.updated_at) > String(current.updated_at)) entries.set(row.record_id, row);
170
+ else if (!entries.has(row.record_id)) entries.set(row.record_id, row);
171
+ }
172
+ return Array.from(entries.values())
173
+ .sort((a, b) => String(b.updated_at).localeCompare(String(a.updated_at)))
174
+ .slice(0, limit);
175
+ }
176
+
177
+ async replaceEntry(spaceId, entryId, chunks) {
178
+ await this.open();
179
+ if (!Array.isArray(chunks) || chunks.length === 0) {
180
+ const result = await this.deleteEntry(spaceId, entryId);
181
+ return { ok: true, action: 'deleted', deleted: result.deleted || 0, upserted: 0 };
182
+ }
183
+ const now = new Date().toISOString();
184
+ const existing = await this.rowsForEntry(spaceId, entryId);
185
+ const existingById = new Map(existing.map(row => [row.record_id, row]));
186
+ const normalized = chunks.map(chunk => normalizeKnowledgeChunk({ ...chunk, spaceId, entryId }, {
187
+ dimensions: this.dimensions,
188
+ now,
189
+ })).map(row => ({
190
+ ...row,
191
+ created_at: existingById.get(row.record_id)?.created_at || row.created_at,
192
+ }));
193
+
194
+ const unchanged = existing.length === normalized.length && normalized.every(row => {
195
+ const old = existingById.get(row.record_id);
196
+ return old && old.content_hash === row.content_hash && old.source_commit === row.source_commit;
197
+ });
198
+ if (unchanged) return { ok: true, action: 'unchanged', upserted: 0, deleted: 0, total: normalized.length };
199
+
200
+ await this.table.mergeInsert('record_id')
201
+ .whenMatchedUpdateAll()
202
+ .whenNotMatchedInsertAll()
203
+ .execute(normalized);
204
+
205
+ const keep = new Set(normalized.map(row => row.record_id));
206
+ const stale = existing.filter(row => !keep.has(row.record_id));
207
+ if (stale.length) {
208
+ await this.table.delete(`record_id IN (${stale.map(row => sqlString(row.record_id)).join(', ')})`);
209
+ }
210
+ return { ok: true, action: existing.length ? 'updated' : 'created', upserted: normalized.length, deleted: stale.length, total: normalized.length };
211
+ }
212
+
213
+ async deleteEntry(spaceId, entryId) {
214
+ await this.open();
215
+ const before = await this.table.countRows(`space_id = ${sqlString(spaceId)} AND entry_id = ${sqlString(entryId)}`);
216
+ if (before) await this.table.delete(`space_id = ${sqlString(spaceId)} AND entry_id = ${sqlString(entryId)}`);
217
+ return { ok: true, deleted: before };
218
+ }
219
+
220
+ async deleteSpace(spaceId) {
221
+ await this.open();
222
+ const before = await this.table.countRows(`space_id = ${sqlString(spaceId)}`);
223
+ if (before) await this.table.delete(`space_id = ${sqlString(spaceId)}`);
224
+ return { ok: true, deleted: before };
225
+ }
226
+
227
+ async vectorSearch(vector, options = {}) {
228
+ await this.open();
229
+ const spaceIds = options.spaceIds || [];
230
+ const limit = Math.max(1, Math.min(Number(options.limit || 10), 100));
231
+ const query = this.table.query()
232
+ .nearestTo(Array.from(vector || [], Number))
233
+ .column('vector')
234
+ .distanceType('cosine')
235
+ .where(inPredicate('space_id', spaceIds))
236
+ .limit(limit);
237
+ const rows = await query.toArray();
238
+ return rows.map(decodeKnowledgeChunk);
239
+ }
240
+
241
+ async ensureSearchIndexes() {
242
+ await this.open();
243
+ if (await this.table.countRows() === 0) return { ok: true, created: [], skipped: 'empty-table' };
244
+ const existing = new Set((await this.table.listIndices()).map(index => index.name));
245
+ let state = this.maintenanceState();
246
+ const created = [];
247
+ // A v4.0.0 FTS index can be very large. Rebuilding it in place would keep
248
+ // both generations alive until Lance's old-version retention expires and
249
+ // could exhaust the user's disk. Mark it for the atomic rebuild workflow
250
+ // instead. New databases create the compact v2 index immediately.
251
+ if (existing.has('search_text_idx') && !existing.has('chunk_text_idx') && state.ftsSchemaVersion === 0) {
252
+ state = this.saveMaintenanceState({ ftsSchemaVersion: 1, ftsUpgradeRequired: true });
253
+ }
254
+ const definitions = [
255
+ ['space_id', 'space_id_idx', lancedb.Index.bitmap()],
256
+ ['entry_id', 'entry_id_idx', lancedb.Index.btree()],
257
+ ];
258
+ if (!existing.has('search_text_idx') || existing.has('chunk_text_idx')) {
259
+ definitions.push(['chunk_text', 'chunk_text_idx', lancedb.Index.fts({
260
+ baseTokenizer: 'ngram',
261
+ ngramMinLength: 2,
262
+ // Chinese bigrams preserve exact keyword recall without creating the
263
+ // duplicate 2/3/4-gram postings used by v4.0.0.
264
+ ngramMaxLength: 2,
265
+ withPosition: false,
266
+ stem: false,
267
+ removeStopWords: false,
268
+ })]);
269
+ }
270
+ for (const [column, name, config] of definitions) {
271
+ if (existing.has(name)) continue;
272
+ await this.table.createIndex(column, { name, config, replace: false, waitTimeoutSeconds: 60 });
273
+ created.push(name);
274
+ }
275
+ if (created.includes('chunk_text_idx') || existing.has('chunk_text_idx')) {
276
+ state = this.saveMaintenanceState({ ftsSchemaVersion: FTS_SCHEMA_VERSION, ftsUpgradeRequired: false });
277
+ }
278
+ return {
279
+ ok: true,
280
+ created,
281
+ ftsSchemaVersion: state.ftsSchemaVersion,
282
+ ftsUpgradeRequired: state.ftsUpgradeRequired === true,
283
+ };
284
+ }
285
+
286
+ async optimize(options = {}) {
287
+ await this.open();
288
+ return this.table.optimize(options);
289
+ }
290
+
291
+ async maybeOptimize(options = {}) {
292
+ const state = this.maintenanceState();
293
+ if (state.ftsUpgradeRequired === true && options.allowLegacyFts !== true) {
294
+ return {
295
+ ok: true,
296
+ optimized: false,
297
+ due: false,
298
+ blockedBy: 'fts-upgrade-required',
299
+ state,
300
+ };
301
+ }
302
+ const due = options.force === true
303
+ || state.pendingOperations >= this.optimizeOperationsThreshold
304
+ || state.pendingRows >= this.optimizeRowsThreshold;
305
+ if (!due || (!state.pendingOperations && !state.pendingRows)) {
306
+ return { ok: true, optimized: false, due: false, state };
307
+ }
308
+ const stats = await this.optimize();
309
+ const next = this.saveMaintenanceState({
310
+ pendingOperations: 0,
311
+ pendingRows: 0,
312
+ lastOptimizedAt: new Date().toISOString(),
313
+ });
314
+ return { ok: true, optimized: true, due: true, stats, state: next };
315
+ }
316
+
317
+ async fullTextSearch(text, options = {}) {
318
+ await this.open();
319
+ const queryText = String(text || '').trim();
320
+ if (!queryText) throw new Error('search text is required');
321
+ const limit = Math.max(1, Math.min(Number(options.limit || 10), 100));
322
+ const ftsColumn = this.maintenanceState().ftsSchemaVersion >= FTS_SCHEMA_VERSION
323
+ ? 'chunk_text'
324
+ : 'search_text';
325
+ const rows = await this.table.query()
326
+ .fullTextSearch(queryText, { columns: ftsColumn })
327
+ .where(inPredicate('space_id', options.spaceIds || []))
328
+ .limit(limit)
329
+ .toArray();
330
+ return rows.map(decodeKnowledgeChunk);
331
+ }
332
+
333
+ async hybridSearch(input = {}) {
334
+ const limit = Math.max(1, Math.min(Number(input.limit || 10), 50));
335
+ const candidates = Math.max(limit, Math.min(Number(input.candidates || limit * 4), 200));
336
+ const [semantic, keyword] = await Promise.all([
337
+ this.vectorSearch(input.vector, { spaceIds: input.spaceIds, limit: candidates }),
338
+ this.fullTextSearch(input.text, { spaceIds: input.spaceIds, limit: candidates }),
339
+ ]);
340
+ const scores = new Map();
341
+ const add = (rows, channel, weight) => rows.forEach((row, rank) => {
342
+ const current = scores.get(row.record_id) || { row, score: 0, channels: [] };
343
+ current.score += weight / (60 + rank + 1);
344
+ current.channels.push(channel);
345
+ scores.set(row.record_id, current);
346
+ });
347
+ add(semantic, 'semantic', Number(input.semanticWeight || 0.65));
348
+ add(keyword, 'keyword', Number(input.keywordWeight || 0.35));
349
+ return Array.from(scores.values())
350
+ .sort((a, b) => b.score - a.score || a.row.record_id.localeCompare(b.row.record_id))
351
+ .slice(0, limit)
352
+ .map(item => ({ ...item.row, relevance_score: item.score, match_channels: item.channels }));
353
+ }
354
+ }
355
+
356
+ module.exports = {
357
+ KnowledgeDatabase,
358
+ sqlString,
359
+ inPredicate,
360
+ FTS_SCHEMA_VERSION,
361
+ DEFAULT_OPTIMIZE_OPERATIONS,
362
+ DEFAULT_OPTIMIZE_ROWS,
363
+ };