@sanqianx/project-knowledge 4.6.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 +20 -0
- package/.claude-plugin/marketplace.json +17 -0
- package/CHANGELOG.md +780 -0
- package/LICENSE +201 -0
- package/NOTICE +5 -0
- package/README.md +266 -0
- package/_site/README.md +30 -0
- package/_site/_test/ai-profile-resolver-test.js +137 -0
- package/_site/_test/ai-profile-test.js +75 -0
- package/_site/_test/ai-vendor-presets-test.js +91 -0
- package/_site/_test/automation-queue-test.js +14 -0
- package/_site/_test/automation-ui-test.js +40 -0
- package/_site/_test/background-task-registry-test.js +43 -0
- package/_site/_test/baseline-schema-test.js +91 -0
- package/_site/_test/bin-cli-test.js +218 -0
- package/_site/_test/bridge-adapter-test.js +95 -0
- package/_site/_test/bridge-consumer-service-test.js +310 -0
- package/_site/_test/cdp-browser-failure-test.js +100 -0
- package/_site/_test/chat-claudecodeui-match-test.js +92 -0
- package/_site/_test/claude-executable-discovery-test.js +90 -0
- package/_site/_test/claude-session-lifecycle-sweep-test.js +252 -0
- package/_site/_test/claude-workbench-test.js +170 -0
- package/_site/_test/client-capture-restart-e2e-test.js +189 -0
- package/_site/_test/codex-conversation-projection-test.js +59 -0
- package/_site/_test/codex-notify-fanout-test.js +45 -0
- package/_site/_test/commit-boundary-freeze-test.js +147 -0
- package/_site/_test/commit-conversation-binding-test.js +99 -0
- package/_site/_test/commit-evidence-test.js +110 -0
- package/_site/_test/context-pack-test.js +198 -0
- package/_site/_test/conversation-api-test.js +119 -0
- package/_site/_test/conversation-store-test.js +136 -0
- package/_site/_test/cross-repo-isolation-e2e-test.js +187 -0
- package/_site/_test/data-dir-migration-test.js +77 -0
- package/_site/_test/desktop-browser-compat-test.js +28 -0
- package/_site/_test/desktop-hook-runtime-regression-test.js +204 -0
- package/_site/_test/embedding-config-test.js +32 -0
- package/_site/_test/explicit-commit-processor-test.js +37 -0
- package/_site/_test/fixtures/fake-browser-exit.js +10 -0
- package/_site/_test/fixtures/fake-browser-hang.js +10 -0
- package/_site/_test/fixtures/make-git-repos.js +89 -0
- package/_site/_test/folder-picker-output-test.js +19 -0
- package/_site/_test/full-integration-e2e-test.js +165 -0
- package/_site/_test/git-validation-test.js +81 -0
- package/_site/_test/hook-runtime-endpoint-test.js +88 -0
- package/_site/_test/hook-status-repair-api-test.js +152 -0
- package/_site/_test/hook-trigger-test.js +85 -0
- package/_site/_test/hook-worktree-test.js +42 -0
- package/_site/_test/import-preflight-api-test.js +233 -0
- package/_site/_test/import-ui-flow-test.js +171 -0
- package/_site/_test/index-builder-test.js +41 -0
- package/_site/_test/index-writer-concurrency-test.js +135 -0
- package/_site/_test/integration-adapters-test.js +243 -0
- package/_site/_test/integration-capture-setup-test.js +108 -0
- package/_site/_test/integration-surface-coverage-test.js +165 -0
- package/_site/_test/kb-framework-test.js +57 -0
- package/_site/_test/kbpath-follow-test.js +63 -0
- package/_site/_test/knowledge-db-test.js +144 -0
- package/_site/_test/knowledge-language-control-test.js +170 -0
- package/_site/_test/knowledge-maintenance-test.js +40 -0
- package/_site/_test/knowledge-migration-test.js +75 -0
- package/_site/_test/knowledge-promotion-recovery-test.js +260 -0
- package/_site/_test/knowledge-query-test.js +54 -0
- package/_site/_test/knowledge-retrieval-service-test.js +77 -0
- package/_site/_test/knowledge-scopes-test.js +37 -0
- package/_site/_test/knowledge-storage-location-test.js +28 -0
- package/_site/_test/knowledge-storage-startup-test.js +52 -0
- package/_site/_test/knowledge-store-logs-supervision-test.js +89 -0
- package/_site/_test/legacy-claude-session-alias-test.js +106 -0
- package/_site/_test/legacy-forward-compat-test.js +257 -0
- package/_site/_test/legacy-project-upgrade-e2e-test.js +363 -0
- package/_site/_test/legacy-routes-removed-test.js +47 -0
- package/_site/_test/legacy-workbench-exclusion-test.js +92 -0
- package/_site/_test/log-redaction-test.js +58 -0
- package/_site/_test/logging-api-test.js +89 -0
- package/_site/_test/logging-sse-no-gap-test.js +88 -0
- package/_site/_test/logging-ui-test.js +104 -0
- package/_site/_test/markdown-delta-overlay-test.js +85 -0
- package/_site/_test/markdown-maintenance-api-test.js +75 -0
- package/_site/_test/markdown-maintenance-test.js +11 -0
- package/_site/_test/mcp-server-test.js +149 -0
- package/_site/_test/model-context-windows-test.js +89 -0
- package/_site/_test/module-bridge-eventbridge-test.js +193 -0
- package/_site/_test/non-release-ci-test.js +34 -0
- package/_site/_test/offline-boundary-isolation-test.js +32 -0
- package/_site/_test/orphaned-claim-recovery-test.js +80 -0
- package/_site/_test/p0-data-migration-characterization-test.js +37 -0
- package/_site/_test/p0-e2e-gate-test.js +347 -0
- package/_site/_test/package-boundary-test.js +42 -0
- package/_site/_test/package-startup-test.js +56 -0
- package/_site/_test/packaged-ui-smoke-test.js +79 -0
- package/_site/_test/path-consistency-test.js +151 -0
- package/_site/_test/pending-sweep-test.js +9 -0
- package/_site/_test/post-commit-automation-test.js +87 -0
- package/_site/_test/pr-consumer-contract-test.js +55 -0
- package/_site/_test/project-control-panel-task14-test.js +63 -0
- package/_site/_test/project-delete-recovery-test.js +64 -0
- package/_site/_test/project-goal-editor-test.js +145 -0
- package/_site/_test/project-layout-v2-migration-test.js +127 -0
- package/_site/_test/project-lifecycle-transaction-test.js +106 -0
- package/_site/_test/project-remove-running-guard-test.js +79 -0
- package/_site/_test/project-store-test.js +90 -0
- package/_site/_test/prompt-settings-test.js +115 -0
- package/_site/_test/protected-architecture-gate-test.js +123 -0
- package/_site/_test/refactor-characterization-test.js +36 -0
- package/_site/_test/release-version-sync-test.js +98 -0
- package/_site/_test/requirement-binding-test.js +113 -0
- package/_site/_test/requirement-recorder-test.js +173 -0
- package/_site/_test/run-all-tests.js +156 -0
- package/_site/_test/runtime-endpoint-test.js +64 -0
- package/_site/_test/scanner-test.js +83 -0
- package/_site/_test/server-runtime-migration-safety-test.js +27 -0
- package/_site/_test/server-security-test.js +96 -0
- package/_site/_test/sessions-stream-test.js +92 -0
- package/_site/_test/shared-contracts-test.js +58 -0
- package/_site/_test/simple-import-test.js +97 -0
- package/_site/_test/startup-analysis-disabled-test.js +16 -0
- package/_site/_test/storage-foundation-test.js +78 -0
- package/_site/_test/structured-logger-test.js +101 -0
- package/_site/_test/task15-20-ui-flow-test.js +148 -0
- package/_site/_test/team-knowledge-store-sparse-checkout-test.js +196 -0
- package/_site/_test/team-knowledge-store-test.js +131 -0
- package/_site/_test/tracking-start-test.js +143 -0
- package/_site/_test/ui-i18n-toggle-test.js +114 -0
- package/_site/_test/ui-smoke-test.js +73 -0
- package/_site/_test/v4122-upgrade-data-contract-test.js +56 -0
- package/_site/_test/workbench-permission-test.js +118 -0
- package/_site/_test/workspace-ui-contract-test.js +43 -0
- package/_site/lib/ai-profile-resolver.js +78 -0
- package/_site/lib/ai-vendor-presets.js +83 -0
- package/_site/lib/ai-workspace.js +104 -0
- package/_site/lib/atomic-file.js +204 -0
- package/_site/lib/automation-config.js +140 -0
- package/_site/lib/bridge-adapter.js +256 -0
- package/_site/lib/bridge-consumer-service.js +402 -0
- package/_site/lib/claude-cli-runner.js +1599 -0
- package/_site/lib/commit-conversation-binder.js +246 -0
- package/_site/lib/commit-processing-ledger.js +27 -0
- package/_site/lib/commit-prompt.js +202 -0
- package/_site/lib/commit-reconciler.js +444 -0
- package/_site/lib/context-pack-builder.js +299 -0
- package/_site/lib/contracts.js +279 -0
- package/_site/lib/conversation-exclusions.js +204 -0
- package/_site/lib/conversation-query-service.js +161 -0
- package/_site/lib/conversation-store.js +385 -0
- package/_site/lib/data-dir.js +81 -0
- package/_site/lib/data-state-classifier.js +40 -0
- package/_site/lib/embedding-service.js +85 -0
- package/_site/lib/evidence-bundle.js +228 -0
- package/_site/lib/folder-picker-output.js +24 -0
- package/_site/lib/git-runner.js +62 -0
- package/_site/lib/github-team-store.js +1175 -0
- package/_site/lib/hook-manager.js +202 -0
- package/_site/lib/index-builder.js +157 -0
- package/_site/lib/index-service.js +173 -0
- package/_site/lib/integration-installer.js +907 -0
- package/_site/lib/kb-framework.js +191 -0
- package/_site/lib/kb-validator.js +124 -0
- package/_site/lib/knowledge-db.js +363 -0
- package/_site/lib/knowledge-promotion.js +435 -0
- package/_site/lib/knowledge-retrieval-service.js +401 -0
- package/_site/lib/knowledge-schema.js +121 -0
- package/_site/lib/knowledge-tool-runtime.js +359 -0
- package/_site/lib/legacy-data-manifest.js +26 -0
- package/_site/lib/llm-client.js +173 -0
- package/_site/lib/markdown-knowledge-indexer.js +230 -0
- package/_site/lib/migration-service.js +442 -0
- package/_site/lib/model-context-windows.js +82 -0
- package/_site/lib/module-bridge.js +496 -0
- package/_site/lib/post-commit-automation.js +141 -0
- package/_site/lib/project-lifecycle-service.js +511 -0
- package/_site/lib/project-registry-store.js +129 -0
- package/_site/lib/project-store.js +262 -0
- package/_site/lib/prompt-registry.js +148 -0
- package/_site/lib/requirement-adapters.js +10 -0
- package/_site/lib/requirement-binder.js +112 -0
- package/_site/lib/requirement-recorder.js +278 -0
- package/_site/lib/runtime-endpoint.js +160 -0
- package/_site/lib/scanner.js +298 -0
- package/_site/lib/server-app.js +1520 -0
- package/_site/lib/settings-store.js +113 -0
- package/_site/lib/storage-layout.js +162 -0
- package/_site/lib/structured-logger.js +574 -0
- package/_site/scripts/codex-notify-fanout.cjs +51 -0
- package/_site/scripts/folder-picker.ps1 +156 -0
- package/_site/scripts/hook-trigger.js +156 -0
- package/_site/scripts/sync-release-version.js +158 -0
- package/_site/server.js +12 -0
- package/_site/start.bat +26 -0
- package/_site/stop.bat +11 -0
- package/bin/project-knowledge-integrations.js +241 -0
- package/bin/project-knowledge-kb.js +65 -0
- package/bin/project-knowledge-mcp.js +195 -0
- package/bin/project-knowledge.js +439 -0
- package/claude-prompts.json +21 -0
- package/docs/README.zh-CN.md +175 -0
- package/docs/assets/architecture.svg +182 -0
- package/docs/assets/dashboard.png +0 -0
- package/docs/assets/draft-review.png +0 -0
- package/docs/assets/logo.svg +50 -0
- package/docs/assets/runs-drafts.png +0 -0
- package/docs/assets/settings.png +0 -0
- package/docs/project-registry-schema.md +33 -0
- package/docs/team-knowledge-mode-a-plan.md +73 -0
- package/docs/testing-strategy.md +33 -0
- package/package.json +100 -0
- package/plugins/project-knowledge/.claude-plugin/plugin.json +19 -0
- package/plugins/project-knowledge/.codex-plugin/plugin.json +38 -0
- package/plugins/project-knowledge/.mcp.json +14 -0
- package/plugins/project-knowledge/opencode/project-knowledge.md +3 -0
- package/plugins/project-knowledge/skills/project-knowledge/SKILL.md +28 -0
- package/plugins/project-knowledge/skills/project-knowledge/agents/openai.yaml +11 -0
- package/templates/change.md +30 -0
- package/templates/project-readme.md +52 -0
- package/ui/app.css +58 -0
- package/ui/app.js +607 -0
- package/ui/favicon.svg +38 -0
- package/ui/i18n.js +146 -0
- package/ui/index.html +162 -0
|
@@ -0,0 +1,191 @@
|
|
|
1
|
+
const fs = require('fs');
|
|
2
|
+
const path = require('path');
|
|
3
|
+
const { migrateAIWorkspace, ensureProjectAIPath } = require('./ai-workspace');
|
|
4
|
+
const { regenerateIndexes } = require('./index-builder');
|
|
5
|
+
|
|
6
|
+
const PROJECT_SCHEMA_VERSION = 'minimal';
|
|
7
|
+
const KB_FRAMEWORK_SCHEMA = 'minimal-kb/v1';
|
|
8
|
+
const TOP_LEVEL = ['README.md', 'GOAL.md', 'ARCHITECTURE.md', 'modules', 'changes'];
|
|
9
|
+
|
|
10
|
+
function todayIso() {
|
|
11
|
+
return new Date().toISOString().slice(0, 10);
|
|
12
|
+
}
|
|
13
|
+
|
|
14
|
+
function readTemplate(name) {
|
|
15
|
+
const file = path.resolve(__dirname, '..', '..', 'templates', name);
|
|
16
|
+
if (!fs.existsSync(file)) return null;
|
|
17
|
+
return fs.readFileSync(file, 'utf-8');
|
|
18
|
+
}
|
|
19
|
+
|
|
20
|
+
function renderTemplate(name, vars, fallback) {
|
|
21
|
+
const tpl = readTemplate(name) || fallback;
|
|
22
|
+
return String(tpl || '').replace(/__([A-Z_][A-Z0-9_]*)__/g, (_, key) => (
|
|
23
|
+
key in vars ? String(vars[key]) : `__${key}__`
|
|
24
|
+
));
|
|
25
|
+
}
|
|
26
|
+
|
|
27
|
+
function frontmatter(fields) {
|
|
28
|
+
const lines = ['---'];
|
|
29
|
+
for (const [key, value] of Object.entries(fields)) {
|
|
30
|
+
if (value == null || value === '') continue;
|
|
31
|
+
if (Array.isArray(value)) lines.push(`${key}: [${value.join(', ')}]`);
|
|
32
|
+
else lines.push(`${key}: ${String(value).replace(/\r?\n/g, ' ')}`);
|
|
33
|
+
}
|
|
34
|
+
lines.push('---', '');
|
|
35
|
+
return lines.join('\n');
|
|
36
|
+
}
|
|
37
|
+
|
|
38
|
+
function ensureFile(file, content, created, base) {
|
|
39
|
+
if (fs.existsSync(file)) return;
|
|
40
|
+
fs.mkdirSync(path.dirname(file), { recursive: true });
|
|
41
|
+
fs.writeFileSync(file, content, 'utf-8');
|
|
42
|
+
if (created) created.push(path.relative(base, file).replace(/\\/g, '/'));
|
|
43
|
+
}
|
|
44
|
+
|
|
45
|
+
function initProjectDirs(slug, kbPath) {
|
|
46
|
+
const base = path.resolve(kbPath);
|
|
47
|
+
// Import only declares the external project directory. Real Markdown and
|
|
48
|
+
// collection directories are created lazily by validated promotion; an
|
|
49
|
+
// empty/TODO skeleton would turn guesses into durable project knowledge.
|
|
50
|
+
fs.mkdirSync(base, { recursive: true });
|
|
51
|
+
return {
|
|
52
|
+
created: [],
|
|
53
|
+
basePath: base,
|
|
54
|
+
kbSchemaVersion: PROJECT_SCHEMA_VERSION,
|
|
55
|
+
frameworkSchema: KB_FRAMEWORK_SCHEMA,
|
|
56
|
+
topLevel: TOP_LEVEL.slice(),
|
|
57
|
+
lazy: true,
|
|
58
|
+
};
|
|
59
|
+
}
|
|
60
|
+
|
|
61
|
+
function readIfExists(file) {
|
|
62
|
+
return fs.existsSync(file) ? fs.readFileSync(file, 'utf-8') : '';
|
|
63
|
+
}
|
|
64
|
+
|
|
65
|
+
function visibleTopLevel(kbPath) {
|
|
66
|
+
if (!fs.existsSync(kbPath)) return [];
|
|
67
|
+
return fs.readdirSync(kbPath).filter(name => !name.startsWith('.')).sort();
|
|
68
|
+
}
|
|
69
|
+
|
|
70
|
+
function isCurrentKb(kbPath) {
|
|
71
|
+
return fs.existsSync(path.join(kbPath, 'GOAL.md')) &&
|
|
72
|
+
fs.existsSync(path.join(kbPath, 'ARCHITECTURE.md')) &&
|
|
73
|
+
fs.existsSync(path.join(kbPath, 'modules')) &&
|
|
74
|
+
fs.existsSync(path.join(kbPath, 'changes'));
|
|
75
|
+
}
|
|
76
|
+
|
|
77
|
+
function consolidateLegacyCommits({ slug, kbPath }) {
|
|
78
|
+
const commitsDir = path.join(kbPath, 'commits');
|
|
79
|
+
const aiRoot = ensureProjectAIPath(slug);
|
|
80
|
+
const backupDir = path.join(aiRoot, 'legacy-commits');
|
|
81
|
+
const result = { found: 0, created: [], backupDir };
|
|
82
|
+
if (!fs.existsSync(commitsDir)) return result;
|
|
83
|
+
const files = fs.readdirSync(commitsDir).filter(f => f.endsWith('.md') && f !== '00-index.md').sort();
|
|
84
|
+
result.found = files.length;
|
|
85
|
+
if (!files.length) return result;
|
|
86
|
+
fs.mkdirSync(backupDir, { recursive: true });
|
|
87
|
+
for (const file of files) fs.copyFileSync(path.join(commitsDir, file), path.join(backupDir, file));
|
|
88
|
+
const bucketCount = Math.min(10, Math.max(1, Math.ceil(files.length / 4)));
|
|
89
|
+
const bucketSize = Math.ceil(files.length / bucketCount);
|
|
90
|
+
const changesDir = path.join(kbPath, 'changes');
|
|
91
|
+
fs.mkdirSync(changesDir, { recursive: true });
|
|
92
|
+
for (let i = 0; i < bucketCount; i++) {
|
|
93
|
+
const bucket = files.slice(i * bucketSize, (i + 1) * bucketSize);
|
|
94
|
+
if (!bucket.length) continue;
|
|
95
|
+
const rel = `legacy-change-${String(i + 1).padStart(2, '0')}.md`;
|
|
96
|
+
const title = `Legacy Change ${i + 1}`;
|
|
97
|
+
const body = [
|
|
98
|
+
frontmatter({
|
|
99
|
+
schema: KB_FRAMEWORK_SCHEMA,
|
|
100
|
+
title,
|
|
101
|
+
tags: ['legacy-migration'],
|
|
102
|
+
updatedAt: todayIso(),
|
|
103
|
+
aggregatedFrom: `[${bucket.map(f => `commits/${f}`).join(', ')}]`,
|
|
104
|
+
}),
|
|
105
|
+
`# ${title}`,
|
|
106
|
+
'',
|
|
107
|
+
'## Development Intent',
|
|
108
|
+
'',
|
|
109
|
+
'This change was generated by consolidating older per-commit records. No raw prompts are preserved.',
|
|
110
|
+
'',
|
|
111
|
+
'## Implementation Result',
|
|
112
|
+
'',
|
|
113
|
+
'Legacy commit notes were grouped into this durable change summary.',
|
|
114
|
+
'',
|
|
115
|
+
'## Evidence',
|
|
116
|
+
'',
|
|
117
|
+
...bucket.map(file => `- \`commits/${file}\``),
|
|
118
|
+
'',
|
|
119
|
+
].join('\n');
|
|
120
|
+
fs.writeFileSync(path.join(changesDir, rel), body, 'utf-8');
|
|
121
|
+
result.created.push(`changes/${rel}`);
|
|
122
|
+
}
|
|
123
|
+
return result;
|
|
124
|
+
}
|
|
125
|
+
|
|
126
|
+
function removeLegacyItems(kbPath) {
|
|
127
|
+
const removed = [];
|
|
128
|
+
const names = [
|
|
129
|
+
'_ai', 'kb-manifest.json', 'project-goal.md', 'project-analysis.md', 'framework.md',
|
|
130
|
+
'architecture', 'commits', 'features', 'operations', 'quality', 'requirements', 'references',
|
|
131
|
+
];
|
|
132
|
+
for (const name of names) {
|
|
133
|
+
const target = path.join(kbPath, name);
|
|
134
|
+
if (fs.existsSync(target)) {
|
|
135
|
+
fs.rmSync(target, { recursive: true, force: true });
|
|
136
|
+
removed.push(name);
|
|
137
|
+
}
|
|
138
|
+
}
|
|
139
|
+
return removed;
|
|
140
|
+
}
|
|
141
|
+
|
|
142
|
+
function migrateToFramework({ slug, kbPath, preserveLegacyAI = true }) {
|
|
143
|
+
const base = path.resolve(kbPath);
|
|
144
|
+
fs.mkdirSync(base, { recursive: true });
|
|
145
|
+
const aiMigration = migrateAIWorkspace({ slug, kbPath: base, preserveOriginal: preserveLegacyAI });
|
|
146
|
+
|
|
147
|
+
const oldGoal = readIfExists(path.join(base, 'project-goal.md'));
|
|
148
|
+
const oldFramework = readIfExists(path.join(base, 'framework.md'));
|
|
149
|
+
const oldArch = readIfExists(path.join(base, 'architecture', 'overview.md'));
|
|
150
|
+
const oldAnalysis = readIfExists(path.join(base, 'project-analysis.md'));
|
|
151
|
+
|
|
152
|
+
const init = initProjectDirs(slug, base);
|
|
153
|
+
if (oldGoal && !oldGoal.includes('TODO: confirm the project goal')) {
|
|
154
|
+
fs.writeFileSync(path.join(base, 'GOAL.md'), oldGoal, 'utf-8');
|
|
155
|
+
}
|
|
156
|
+
const archParts = [
|
|
157
|
+
oldFramework && `## Migrated Framework\n\n${oldFramework}`,
|
|
158
|
+
oldArch && `## Migrated Architecture Overview\n\n${oldArch}`,
|
|
159
|
+
oldAnalysis && `## Migrated Project Analysis\n\n${oldAnalysis}`,
|
|
160
|
+
].filter(Boolean);
|
|
161
|
+
if (archParts.length) {
|
|
162
|
+
fs.writeFileSync(path.join(base, 'ARCHITECTURE.md'), `${frontmatter({ schema: KB_FRAMEWORK_SCHEMA, title: `${slug} Architecture`, updatedAt: todayIso() })}# ${slug} Architecture\n\n${archParts.join('\n\n')}\n`, 'utf-8');
|
|
163
|
+
}
|
|
164
|
+
const commits = consolidateLegacyCommits({ slug, kbPath: base });
|
|
165
|
+
const removed = removeLegacyItems(base);
|
|
166
|
+
regenerateIndexes(base);
|
|
167
|
+
return {
|
|
168
|
+
ok: true,
|
|
169
|
+
slug,
|
|
170
|
+
kbPath: base,
|
|
171
|
+
kbSchemaVersion: PROJECT_SCHEMA_VERSION,
|
|
172
|
+
frameworkSchema: KB_FRAMEWORK_SCHEMA,
|
|
173
|
+
init,
|
|
174
|
+
aiMigration,
|
|
175
|
+
legacyCommits: commits,
|
|
176
|
+
removed,
|
|
177
|
+
topLevel: visibleTopLevel(base),
|
|
178
|
+
};
|
|
179
|
+
}
|
|
180
|
+
|
|
181
|
+
module.exports = {
|
|
182
|
+
PROJECT_SCHEMA_VERSION,
|
|
183
|
+
KB_FRAMEWORK_SCHEMA,
|
|
184
|
+
TOP_LEVEL,
|
|
185
|
+
initProjectDirs,
|
|
186
|
+
migrateToFramework,
|
|
187
|
+
consolidateLegacyCommits,
|
|
188
|
+
isCurrentKb,
|
|
189
|
+
visibleTopLevel,
|
|
190
|
+
frontmatter,
|
|
191
|
+
};
|
|
@@ -0,0 +1,124 @@
|
|
|
1
|
+
// KB Validator
|
|
2
|
+
//
|
|
3
|
+
// The current knowledge base framework is intentionally small:
|
|
4
|
+
//
|
|
5
|
+
// README.md
|
|
6
|
+
// GOAL.md
|
|
7
|
+
// ARCHITECTURE.md
|
|
8
|
+
// modules/
|
|
9
|
+
// 00-index.md
|
|
10
|
+
// changes/
|
|
11
|
+
// 00-index.md
|
|
12
|
+
//
|
|
13
|
+
// Everything inside this layout is trusted project memory. AI run state,
|
|
14
|
+
// context packs, backups, and Claude workbench sessions live under the
|
|
15
|
+
// application data directory, outside the project KB.
|
|
16
|
+
|
|
17
|
+
const fs = require('fs');
|
|
18
|
+
const path = require('path');
|
|
19
|
+
const { KB_FRAMEWORK_SCHEMA, PROJECT_SCHEMA_VERSION, TOP_LEVEL } = require('./kb-framework');
|
|
20
|
+
|
|
21
|
+
function walkTrustedFiles(root, dir = root, prefix = '') {
|
|
22
|
+
const out = [];
|
|
23
|
+
let entries = [];
|
|
24
|
+
try { entries = fs.readdirSync(dir, { withFileTypes: true }); } catch { return out; }
|
|
25
|
+
for (const entry of entries.sort((a, b) => a.name.localeCompare(b.name))) {
|
|
26
|
+
if (entry.name.startsWith('.')) continue;
|
|
27
|
+
if (entry.name === '_ai' || entry.name === '_meta' || entry.name === 'node_modules') continue;
|
|
28
|
+
const rel = (prefix ? `${prefix}/` : '') + entry.name;
|
|
29
|
+
const full = path.join(dir, entry.name);
|
|
30
|
+
if (entry.isDirectory()) out.push(...walkTrustedFiles(root, full, rel));
|
|
31
|
+
else if (entry.isFile() && entry.name.endsWith('.md')) {
|
|
32
|
+
out.push({ path: rel, content: fs.readFileSync(full, 'utf-8'), size: fs.statSync(full).size });
|
|
33
|
+
}
|
|
34
|
+
}
|
|
35
|
+
return out;
|
|
36
|
+
}
|
|
37
|
+
|
|
38
|
+
function validateKb(kbPath) {
|
|
39
|
+
const info = {
|
|
40
|
+
kbPath,
|
|
41
|
+
schemaVersion: PROJECT_SCHEMA_VERSION,
|
|
42
|
+
frameworkSchema: KB_FRAMEWORK_SCHEMA,
|
|
43
|
+
goalStatus: 'unknown',
|
|
44
|
+
architectureStatus: 'unknown',
|
|
45
|
+
trustedKnowledgeEntries: 0,
|
|
46
|
+
};
|
|
47
|
+
|
|
48
|
+
if (!kbPath || !fs.existsSync(kbPath)) {
|
|
49
|
+
return { ok: false, status: 400, error: `kbPath does not exist: ${kbPath}` };
|
|
50
|
+
}
|
|
51
|
+
if (!fs.statSync(kbPath).isDirectory()) {
|
|
52
|
+
return { ok: false, status: 400, error: `kbPath is not a directory: ${kbPath}` };
|
|
53
|
+
}
|
|
54
|
+
|
|
55
|
+
const errors = [];
|
|
56
|
+
const warnings = [];
|
|
57
|
+
const required = [
|
|
58
|
+
['README.md', 'file'],
|
|
59
|
+
['GOAL.md', 'file'],
|
|
60
|
+
['ARCHITECTURE.md', 'file'],
|
|
61
|
+
['modules', 'dir'],
|
|
62
|
+
['changes', 'dir'],
|
|
63
|
+
['modules/00-index.md', 'file'],
|
|
64
|
+
['changes/00-index.md', 'file'],
|
|
65
|
+
];
|
|
66
|
+
for (const [rel, kind] of required) {
|
|
67
|
+
const abs = path.join(kbPath, rel);
|
|
68
|
+
if (!fs.existsSync(abs)) {
|
|
69
|
+
errors.push(`${rel} missing`);
|
|
70
|
+
continue;
|
|
71
|
+
}
|
|
72
|
+
const stat = fs.statSync(abs);
|
|
73
|
+
if (kind === 'file' && !stat.isFile()) errors.push(`${rel} must be a file`);
|
|
74
|
+
if (kind === 'dir' && !stat.isDirectory()) errors.push(`${rel} must be a directory`);
|
|
75
|
+
}
|
|
76
|
+
|
|
77
|
+
const top = fs.readdirSync(kbPath).filter(name => !name.startsWith('.')).sort();
|
|
78
|
+
const unexpected = top.filter(name => !TOP_LEVEL.includes(name));
|
|
79
|
+
if (unexpected.length) errors.push(`unexpected top-level KB items: ${unexpected.join(', ')}`);
|
|
80
|
+
if (fs.existsSync(path.join(kbPath, '_ai'))) errors.push('_ai must not live inside a project KB');
|
|
81
|
+
if (fs.existsSync(path.join(kbPath, 'kb-manifest.json'))) errors.push('kb-manifest.json is not part of the current KB framework');
|
|
82
|
+
|
|
83
|
+
info.goalStatus = fs.existsSync(path.join(kbPath, 'GOAL.md')) ? 'present' : 'missing';
|
|
84
|
+
info.architectureStatus = fs.existsSync(path.join(kbPath, 'ARCHITECTURE.md')) ? 'present' : 'missing';
|
|
85
|
+
info.trustedKnowledgeEntries = walkTrustedFiles(kbPath).length;
|
|
86
|
+
|
|
87
|
+
return {
|
|
88
|
+
ok: errors.length === 0,
|
|
89
|
+
status: errors.length ? 422 : 200,
|
|
90
|
+
info,
|
|
91
|
+
errors,
|
|
92
|
+
warnings,
|
|
93
|
+
};
|
|
94
|
+
}
|
|
95
|
+
|
|
96
|
+
function buildPrContextPack(kbPath) {
|
|
97
|
+
const validation = validateKb(kbPath);
|
|
98
|
+
if (!validation.ok) return { ok: false, status: validation.status || 422, error: 'kb invalid', validation };
|
|
99
|
+
const files = walkTrustedFiles(kbPath);
|
|
100
|
+
const readFile = rel => files.find(item => item.path === rel) || null;
|
|
101
|
+
return {
|
|
102
|
+
ok: true,
|
|
103
|
+
pack: {
|
|
104
|
+
schema: 'pr-context-pack/v1',
|
|
105
|
+
frameworkSchema: KB_FRAMEWORK_SCHEMA,
|
|
106
|
+
generatedAt: new Date().toISOString(),
|
|
107
|
+
project: path.basename(kbPath),
|
|
108
|
+
goal: readFile('GOAL.md'),
|
|
109
|
+
architecture: readFile('ARCHITECTURE.md'),
|
|
110
|
+
indexes: {
|
|
111
|
+
modules: readFile('modules/00-index.md'),
|
|
112
|
+
changes: readFile('changes/00-index.md'),
|
|
113
|
+
},
|
|
114
|
+
trustedKnowledge: files,
|
|
115
|
+
},
|
|
116
|
+
};
|
|
117
|
+
}
|
|
118
|
+
|
|
119
|
+
module.exports = {
|
|
120
|
+
KB_FRAMEWORK_SCHEMA,
|
|
121
|
+
validateKb,
|
|
122
|
+
buildPrContextPack,
|
|
123
|
+
walkTrustedFiles,
|
|
124
|
+
};
|
|
@@ -0,0 +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
|
+
};
|