@sanqianx/project-knowledge 5.0.19 → 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.
- package/.agents/plugins/marketplace.json +3 -3
- package/.claude-plugin/marketplace.json +1 -1
- package/CHANGELOG.md +5 -0
- package/LICENSE +201 -201
- package/NOTICE +5 -5
- package/README.md +349 -349
- package/_modules/knowledge-engine/README.md +4 -0
- package/_modules/knowledge-engine/src/analysis-failures.js +39 -0
- package/_modules/knowledge-engine/src/analysis-worker.js +6 -2
- package/_modules/knowledge-engine/src/coordinator.js +7 -0
- package/_modules/knowledge-engine/src/headless-analyzer.js +2 -1
- package/_modules/knowledge-engine/src/server.js +1 -1
- package/_site/_test/agent-capture-mcp-ui-test.js +67 -67
- package/_site/_test/ai-profile-test.js +75 -75
- package/_site/_test/ai-vendor-presets-test.js +91 -91
- package/_site/_test/automatic-context-test.js +70 -70
- package/_site/_test/bin-cli-test.js +230 -230
- package/_site/_test/bridge-adapter-test.js +95 -95
- package/_site/_test/bridge-consumer-service-test.js +310 -310
- package/_site/_test/candidate-bin-content-test.js +35 -35
- package/_site/_test/candidate-product-assets-test.js +13 -13
- package/_site/_test/candidate-release-contract-test.js +14 -14
- package/_site/_test/capture-hook-dedup-test.js +158 -158
- package/_site/_test/claude-executable-discovery-test.js +28 -28
- package/_site/_test/claude-session-lifecycle-sweep-test.js +252 -252
- package/_site/_test/cli-instance-management-test.js +62 -62
- package/_site/_test/client-capture-restart-e2e-test.js +189 -189
- package/_site/_test/codex-conversation-projection-test.js +60 -60
- package/_site/_test/codex-marketplace-scoped-path-test.js +186 -186
- package/_site/_test/codex-notify-capture-e2e-test.js +101 -101
- package/_site/_test/codex-notify-fanout-test.js +134 -134
- package/_site/_test/codex-notify-upgrade-test.js +253 -253
- package/_site/_test/commit-boundary-freeze-test.js +147 -147
- package/_site/_test/commit-conversation-binding-test.js +4 -4
- package/_site/_test/configuration-generation-test.js +22 -22
- package/_site/_test/context-pack-test.js +200 -200
- package/_site/_test/conversation-api-test.js +10 -10
- package/_site/_test/conversation-origin-test.js +15 -15
- package/_site/_test/cross-repo-isolation-e2e-test.js +187 -187
- package/_site/_test/embedding-config-test.js +32 -32
- package/_site/_test/fixtures/fake-browser-exit.js +10 -10
- package/_site/_test/fixtures/fake-browser-hang.js +10 -10
- package/_site/_test/fixtures/integration-busy-worker.cjs +6 -6
- package/_site/_test/fixtures/make-git-repos.js +92 -92
- package/_site/_test/fixtures/retrieval-gold-queries.json +32 -32
- package/_site/_test/fresh-hook-reimport-test.js +53 -53
- package/_site/_test/fresh-product-data-test.js +56 -56
- package/_site/_test/git-path-compatibility-test.js +80 -80
- package/_site/_test/git-validation-test.js +81 -81
- package/_site/_test/hook-worktree-test.js +42 -42
- package/_site/_test/index-builder-test.js +41 -41
- package/_site/_test/index-error-display-contract-test.js +10 -10
- package/_site/_test/integration-adapters-test.js +67 -67
- package/_site/_test/integration-capture-setup-test.js +118 -118
- package/_site/_test/integration-runtime-binding-a2-test.js +566 -566
- package/_site/_test/integration-runtime-binding-a3-test.js +227 -227
- package/_site/_test/integration-runtime-binding-test.js +473 -473
- package/_site/_test/integration-status-scan-test.js +29 -29
- package/_site/_test/kb-framework-test.js +57 -57
- package/_site/_test/kbpath-follow-test.js +63 -63
- package/_site/_test/knowledge-db-test.js +144 -144
- package/_site/_test/knowledge-engine-client-test.js +109 -109
- package/_site/_test/knowledge-maintenance-test.js +40 -40
- package/_site/_test/knowledge-panel-ui-contract-test.js +112 -0
- package/_site/_test/knowledge-scopes-test.js +37 -37
- package/_site/_test/knowledge-storage-location-test.js +28 -28
- package/_site/_test/launcher-readiness-test.js +80 -80
- package/_site/_test/legacy-claude-session-alias-test.js +106 -106
- package/_site/_test/legacy-routes-removed-test.js +47 -47
- package/_site/_test/legacy-workbench-exclusion-test.js +92 -92
- package/_site/_test/live-gateway-test.js +48 -48
- package/_site/_test/local-marketplace-test.js +16 -16
- package/_site/_test/log-redaction-test.js +58 -58
- package/_site/_test/markdown-maintenance-test.js +11 -11
- package/_site/_test/model-context-windows-test.js +89 -89
- package/_site/_test/module-agent-start-proxy-test.js +65 -65
- package/_site/_test/module-artifact-integrity-test.js +27 -27
- package/_site/_test/module-boundary-test.js +30 -30
- package/_site/_test/module-bridge-eventbridge-test.js +202 -202
- package/_site/_test/module-dynamic-port-test.js +118 -118
- package/_site/_test/module-index-proxy-test.js +33 -33
- package/_site/_test/module-model-configuration-test.js +44 -44
- package/_site/_test/module-process-lifecycle-test.js +35 -35
- package/_site/_test/module-stream-proxy-test.js +34 -34
- package/_site/_test/module-terminal-restart-test.js +63 -63
- package/_site/_test/module-wiring-unbind-race-test.js +59 -59
- package/_site/_test/native-context-integrations-test.js +86 -86
- package/_site/_test/orphaned-claim-recovery-test.js +80 -80
- package/_site/_test/package-boundary-test.js +43 -42
- package/_site/_test/package-startup-test.js +56 -56
- package/_site/_test/packaged-ui-smoke-test.js +88 -88
- package/_site/_test/pr-consumer-contract-test.js +55 -55
- package/_site/_test/product-brand-contract-test.js +25 -25
- package/_site/_test/product-diagnostics-ui-test.js +93 -93
- package/_site/_test/product-import-ui-test.js +104 -104
- package/_site/_test/product-progress-status-test.js +12 -1
- package/_site/_test/project-remove-running-guard-test.js +79 -79
- package/_site/_test/project-store-test.js +90 -90
- package/_site/_test/registry-events-test.js +38 -38
- package/_site/_test/release-version-sync-test.js +115 -115
- package/_site/_test/requirement-binding-test.js +113 -113
- package/_site/_test/runtime-endpoint-test.js +64 -64
- package/_site/_test/scanner-test.js +83 -83
- package/_site/_test/server-security-test.js +96 -96
- package/_site/_test/sessions-stream-test.js +93 -93
- package/_site/_test/shared-scope-root-binding-test.js +67 -67
- package/_site/_test/simple-import-test.js +98 -98
- package/_site/_test/startup-analysis-disabled-test.js +6 -6
- package/_site/_test/team-knowledge-store-sparse-checkout-test.js +196 -196
- package/_site/_test/team-knowledge-store-test.js +132 -132
- package/_site/_test/test-server-cleanup-test.js +35 -35
- package/_site/_test/test-server-module-isolation-test.js +37 -37
- package/_site/_test/tracking-start-test.js +35 -35
- package/_site/_test/workbench-context-receipts-test.js +22 -22
- package/_site/_test/workspace-ui-contract-test.js +31 -31
- package/_site/_test/zcode-capture-integration-test.js +108 -108
- package/_site/lib/ai-vendor-presets.js +83 -83
- package/_site/lib/ai-workspace.js +3 -3
- package/_site/lib/atomic-file.js +204 -204
- package/_site/lib/bridge-adapter.js +260 -260
- package/_site/lib/capture-hook-config.js +108 -108
- package/_site/lib/claude-cli-runner.js +89 -89
- package/_site/lib/cli-instances.js +43 -43
- package/_site/lib/codex-context-status.js +17 -17
- package/_site/lib/codex-notify-config.js +193 -193
- package/_site/lib/commit-conversation-binder.js +12 -12
- package/_site/lib/commit-reconciler.js +38 -38
- package/_site/lib/context-pack-builder.js +299 -299
- package/_site/lib/conversation-exclusions.js +204 -204
- package/_site/lib/conversation-query-service.js +17 -17
- package/_site/lib/data-dir.js +13 -13
- package/_site/lib/embedding-service.js +85 -85
- package/_site/lib/engine-connection.js +93 -93
- package/_site/lib/engine-knowledge-client.js +72 -72
- package/_site/lib/folder-picker-output.js +3 -3
- package/_site/lib/git-path.js +14 -14
- package/_site/lib/github-team-store.js +3 -3
- package/_site/lib/index-builder.js +157 -157
- package/_site/lib/integration-status-scan.js +99 -99
- package/_site/lib/integration-status-worker.js +14 -14
- package/_site/lib/kb-framework.js +15 -15
- package/_site/lib/knowledge-db.js +363 -363
- package/_site/lib/knowledge-schema.js +121 -121
- package/_site/lib/live-gateway.js +78 -78
- package/_site/lib/llm-client.js +12 -12
- package/_site/lib/model-context-windows.js +82 -82
- package/_site/lib/module-bridge.js +874 -874
- package/_site/lib/module-port-resolver.js +57 -57
- package/_site/lib/post-commit-automation.js +54 -54
- package/_site/lib/product-data.js +35 -35
- package/_site/lib/project-lifecycle-service.js +14 -14
- package/_site/lib/project-registry-store.js +129 -129
- package/_site/lib/prompt-registry.js +148 -148
- package/_site/lib/registry-events.js +16 -16
- package/_site/lib/requirement-adapters.js +10 -10
- package/_site/lib/requirement-binder.js +112 -112
- package/_site/lib/runtime-control.js +89 -89
- package/_site/lib/runtime-endpoint.js +5 -5
- package/_site/lib/zcode-context-integration.js +60 -60
- package/_site/scripts/codex-notify-fanout.cjs +95 -95
- package/_site/scripts/install-module-candidates.js +69 -69
- package/_site/scripts/pack-module-candidates.js +89 -89
- package/_site/scripts/sync-product-content.js +28 -28
- package/_site/scripts/sync-release-version.js +164 -164
- package/_site/scripts/vendor-modules.js +49 -49
- package/_site/scripts/verify-module-runtime.js +50 -50
- package/_site/scripts/verify-product-runtime.js +36 -36
- package/_site/scripts/verify-retrieval-gold.js +83 -83
- package/_site/server.js +15 -15
- package/_site/start.bat +26 -26
- package/_site/stop.bat +11 -11
- package/bin/project-knowledge-integrations.js +254 -254
- package/bin/project-knowledge-kb.js +67 -67
- package/bin/project-knowledge-mcp.js +8 -8
- package/bin/project-knowledge.js +170 -170
- package/claude-prompts.json +21 -21
- package/docs/README.zh-CN.md +205 -205
- package/docs/assets/architecture.svg +181 -181
- package/docs/assets/logo.svg +49 -49
- package/docs/fresh-v5-start.md +42 -42
- package/docs/leader-glm-workbench-playbook.md +122 -122
- package/docs/project-registry-schema.md +11 -11
- package/docs/release-5.0.16-acceptance.md +176 -176
- package/docs/release-5.0.17-acceptance.md +117 -117
- package/docs/release-5.0.19-acceptance.md +39 -25
- package/docs/team-knowledge-mode-a-plan.md +73 -73
- package/module-runtime-manifest.json +53 -33
- package/package.json +3 -2
- package/plugins/project-knowledge/.claude-plugin/plugin.json +1 -1
- package/plugins/project-knowledge/.codex-plugin/plugin.json +1 -1
- package/plugins/project-knowledge/.mcp.json +1 -1
- package/plugins/project-knowledge/opencode/project-knowledge.md +3 -3
- package/plugins/project-knowledge/skills/project-knowledge/SKILL.md +28 -28
- package/templates/change.md +30 -30
- package/templates/project-readme.md +18 -18
- package/ui/index.html +1 -1
- package/ui/knowledge-panel.html +118 -0
- package/ui/product.css +8 -0
- package/ui/product.js +146 -4
- package/vendor-manifest.json +55 -54
|
@@ -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
|
+
};
|