clementine-agent 1.0.32 → 1.0.33
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/dist/memory/embeddings.d.ts +7 -0
- package/dist/memory/embeddings.js +14 -0
- package/dist/memory/store.d.ts +6 -0
- package/dist/memory/store.js +112 -33
- package/dist/tools/shared.d.ts +5 -0
- package/package.json +1 -1
|
@@ -35,4 +35,11 @@ export declare function deserializeEmbedding(buf: Buffer): Float32Array;
|
|
|
35
35
|
* Check if the embedding system is ready (vocabulary loaded with sufficient words).
|
|
36
36
|
*/
|
|
37
37
|
export declare function isReady(): boolean;
|
|
38
|
+
/**
|
|
39
|
+
* Stable hash of the current vocabulary's word→dimension mapping. When this
|
|
40
|
+
* changes, previously-stored embedding vectors become silently incorrect
|
|
41
|
+
* because dimension N now represents a different word. Callers (MemoryStore
|
|
42
|
+
* backfill) use this hash to detect staleness and invalidate stored vectors.
|
|
43
|
+
*/
|
|
44
|
+
export declare function getVocabHash(): string;
|
|
38
45
|
//# sourceMappingURL=embeddings.d.ts.map
|
|
@@ -9,6 +9,7 @@
|
|
|
9
9
|
* Query-time: embed the query, compute cosine similarity against stored vectors.
|
|
10
10
|
*/
|
|
11
11
|
import { existsSync, mkdirSync, readFileSync, writeFileSync } from 'node:fs';
|
|
12
|
+
import { createHash } from 'node:crypto';
|
|
12
13
|
import path from 'node:path';
|
|
13
14
|
import pino from 'pino';
|
|
14
15
|
import { BASE_DIR } from '../config.js';
|
|
@@ -163,6 +164,19 @@ export function isReady() {
|
|
|
163
164
|
loadVocab();
|
|
164
165
|
return vocabWords.length >= 50; // need at least 50 vocab words
|
|
165
166
|
}
|
|
167
|
+
/**
|
|
168
|
+
* Stable hash of the current vocabulary's word→dimension mapping. When this
|
|
169
|
+
* changes, previously-stored embedding vectors become silently incorrect
|
|
170
|
+
* because dimension N now represents a different word. Callers (MemoryStore
|
|
171
|
+
* backfill) use this hash to detect staleness and invalidate stored vectors.
|
|
172
|
+
*/
|
|
173
|
+
export function getVocabHash() {
|
|
174
|
+
loadVocab();
|
|
175
|
+
if (vocabWords.length === 0)
|
|
176
|
+
return '';
|
|
177
|
+
// Order-sensitive: dimension assignment depends on insertion order.
|
|
178
|
+
return createHash('sha1').update(vocabWords.join('|')).digest('hex').slice(0, 16);
|
|
179
|
+
}
|
|
166
180
|
const STOP_WORDS = new Set([
|
|
167
181
|
'the', 'be', 'to', 'of', 'and', 'in', 'that', 'have', 'it', 'for',
|
|
168
182
|
'not', 'on', 'with', 'he', 'as', 'you', 'do', 'at', 'this', 'but',
|
package/dist/memory/store.d.ts
CHANGED
|
@@ -174,10 +174,15 @@ export declare class MemoryStore {
|
|
|
174
174
|
salienceThreshold?: number;
|
|
175
175
|
accessLogRetentionDays?: number;
|
|
176
176
|
transcriptRetentionDays?: number;
|
|
177
|
+
behavioralRetentionDays?: number;
|
|
177
178
|
}): {
|
|
178
179
|
episodicPruned: number;
|
|
179
180
|
accessLogPruned: number;
|
|
180
181
|
transcriptsPruned: number;
|
|
182
|
+
skillUsagePruned: number;
|
|
183
|
+
feedbackPruned: number;
|
|
184
|
+
reflectionsPruned: number;
|
|
185
|
+
usageLogPruned: number;
|
|
181
186
|
};
|
|
182
187
|
/**
|
|
183
188
|
* Get chunks within a date range, ordered chronologically.
|
|
@@ -533,6 +538,7 @@ export declare class MemoryStore {
|
|
|
533
538
|
buildEmbeddings(): {
|
|
534
539
|
vocabSize: number;
|
|
535
540
|
backfilled: number;
|
|
541
|
+
invalidated: number;
|
|
536
542
|
};
|
|
537
543
|
/**
|
|
538
544
|
* Delete all chunks, wikilinks, file hash, and access log for a given file.
|
package/dist/memory/store.js
CHANGED
|
@@ -10,9 +10,10 @@
|
|
|
10
10
|
* (single-user, one MCP subprocess handles all writes).
|
|
11
11
|
*/
|
|
12
12
|
import { createHash } from 'node:crypto';
|
|
13
|
-
import { appendFileSync, existsSync, mkdirSync, readFileSync, readdirSync, statSync } from 'node:fs';
|
|
13
|
+
import { appendFileSync, existsSync, mkdirSync, readFileSync, readdirSync, statSync, writeFileSync } from 'node:fs';
|
|
14
14
|
import path from 'node:path';
|
|
15
15
|
import Database from 'better-sqlite3';
|
|
16
|
+
import { BASE_DIR } from '../config.js';
|
|
16
17
|
import * as embeddingsModule from './embeddings.js';
|
|
17
18
|
import { chunkFile } from './chunker.js';
|
|
18
19
|
import { mmrRerank } from './mmr.js';
|
|
@@ -184,6 +185,24 @@ export class MemoryStore {
|
|
|
184
185
|
catch {
|
|
185
186
|
// Index already exists
|
|
186
187
|
}
|
|
188
|
+
// Hot-path indices: every chat turn sorts/filters chunks by updated_at
|
|
189
|
+
// (recency) and by (agent_slug, updated_at) for agent-scoped recent
|
|
190
|
+
// context. Without these the queries do full table scans.
|
|
191
|
+
try {
|
|
192
|
+
this.conn.exec('CREATE INDEX idx_chunks_updated_at ON chunks(updated_at DESC)');
|
|
193
|
+
}
|
|
194
|
+
catch { /* already exists */ }
|
|
195
|
+
try {
|
|
196
|
+
this.conn.exec('CREATE INDEX idx_chunks_agent_updated ON chunks(agent_slug, updated_at DESC)');
|
|
197
|
+
}
|
|
198
|
+
catch { /* already exists */ }
|
|
199
|
+
// Embedding filter — searchByEmbedding's base predicate is
|
|
200
|
+
// `embedding IS NOT NULL`; a partial index turns that into an
|
|
201
|
+
// index-only scan for the candidate set.
|
|
202
|
+
try {
|
|
203
|
+
this.conn.exec('CREATE INDEX idx_chunks_has_embedding ON chunks(id) WHERE embedding IS NOT NULL');
|
|
204
|
+
}
|
|
205
|
+
catch { /* already exists */ }
|
|
187
206
|
// Access log table for salience tracking
|
|
188
207
|
this.conn.exec(`
|
|
189
208
|
CREATE TABLE IF NOT EXISTS access_log (
|
|
@@ -581,32 +600,35 @@ export class MemoryStore {
|
|
|
581
600
|
stats.filesDeleted++;
|
|
582
601
|
}
|
|
583
602
|
}
|
|
584
|
-
// Process changed/new files
|
|
585
|
-
|
|
603
|
+
// Process changed/new files inside a single transaction so a 1000-file
|
|
604
|
+
// sync produces one WAL commit instead of 1000+. Prepared statements are
|
|
605
|
+
// hoisted out of the loop — better-sqlite3 caches by SQL text anyway, but
|
|
606
|
+
// the explicit handle avoids re-parsing and makes the intent clear.
|
|
607
|
+
const insertStmt = this.conn.prepare(`INSERT INTO chunks
|
|
608
|
+
(source_file, section, content, chunk_type, frontmatter_json, content_hash, category, topic)
|
|
609
|
+
VALUES (?, ?, ?, ?, ?, ?, ?, ?)`);
|
|
610
|
+
const upsertHashStmt = this.conn.prepare(`INSERT OR REPLACE INTO file_hashes (rel_path, content_hash, last_synced)
|
|
611
|
+
VALUES (?, ?, datetime('now'))`);
|
|
612
|
+
const processFile = (filePath) => {
|
|
586
613
|
const rel = path.relative(this.vaultDir, filePath);
|
|
587
614
|
const chunks = chunkFile(filePath, this.vaultDir);
|
|
588
615
|
if (chunks.length === 0)
|
|
589
|
-
|
|
590
|
-
// Delete old chunks for this file
|
|
616
|
+
return;
|
|
591
617
|
this.deleteFileChunks(rel);
|
|
592
|
-
// Insert new chunks
|
|
593
|
-
const insertStmt = this.conn.prepare(`INSERT INTO chunks
|
|
594
|
-
(source_file, section, content, chunk_type, frontmatter_json, content_hash, category, topic)
|
|
595
|
-
VALUES (?, ?, ?, ?, ?, ?, ?, ?)`);
|
|
596
618
|
for (const chunk of chunks) {
|
|
597
619
|
insertStmt.run(chunk.sourceFile, chunk.section, chunk.content, chunk.chunkType, chunk.frontmatterJson, chunk.contentHash, chunk.category ?? null, chunk.topic ?? null);
|
|
598
620
|
}
|
|
599
|
-
// Parse and index wikilinks
|
|
600
621
|
this.indexWikilinks(rel, filePath);
|
|
601
|
-
// Update file hash
|
|
602
622
|
const bytes = readFileSync(filePath);
|
|
603
623
|
const fileHash = createHash('sha256').update(bytes).digest('hex').slice(0, 16);
|
|
604
|
-
|
|
605
|
-
.prepare(`INSERT OR REPLACE INTO file_hashes (rel_path, content_hash, last_synced)
|
|
606
|
-
VALUES (?, ?, datetime('now'))`)
|
|
607
|
-
.run(rel, fileHash);
|
|
624
|
+
upsertHashStmt.run(rel, fileHash);
|
|
608
625
|
stats.filesUpdated++;
|
|
609
|
-
}
|
|
626
|
+
};
|
|
627
|
+
const processAll = this.conn.transaction((files) => {
|
|
628
|
+
for (const f of files)
|
|
629
|
+
processFile(f);
|
|
630
|
+
});
|
|
631
|
+
processAll(filesToUpdate);
|
|
610
632
|
// Count total chunks
|
|
611
633
|
const countRow = this.conn
|
|
612
634
|
.prepare('SELECT COUNT(*) as cnt FROM chunks')
|
|
@@ -838,17 +860,20 @@ export class MemoryStore {
|
|
|
838
860
|
* Scans chunks that have stored embeddings and returns top matches.
|
|
839
861
|
*/
|
|
840
862
|
searchByEmbedding(queryVec, limit, agentSlug, strict = false) {
|
|
841
|
-
|
|
842
|
-
|
|
843
|
-
|
|
844
|
-
|
|
845
|
-
|
|
863
|
+
// Push agent-isolation into SQL so we don't deserialize embeddings for
|
|
864
|
+
// rows we'd immediately reject. Soft isolation (non-strict) still loads
|
|
865
|
+
// all embeddings because the boost is applied post-scoring, but at
|
|
866
|
+
// least strict mode no longer scans foreign-agent chunks.
|
|
867
|
+
let sql = 'SELECT id, source_file, section, content, chunk_type, embedding, salience, agent_slug, updated_at, category, topic FROM chunks WHERE embedding IS NOT NULL';
|
|
868
|
+
const params = [];
|
|
869
|
+
if (strict && agentSlug) {
|
|
870
|
+
sql += ' AND (agent_slug IS NULL OR agent_slug = ?)';
|
|
871
|
+
params.push(agentSlug);
|
|
872
|
+
}
|
|
873
|
+
const rows = this.conn.prepare(sql).all(...params);
|
|
846
874
|
const scored = [];
|
|
847
875
|
for (const row of rows) {
|
|
848
876
|
try {
|
|
849
|
-
// Hard isolation: skip chunks from other agents (allow own + global)
|
|
850
|
-
if (strict && agentSlug && row.agent_slug !== null && row.agent_slug !== agentSlug)
|
|
851
|
-
continue;
|
|
852
877
|
const vec = embeddingsModule.deserializeEmbedding(row.embedding);
|
|
853
878
|
const sim = embeddingsModule.cosineSimilarity(queryVec, vec);
|
|
854
879
|
if (sim < 0.15)
|
|
@@ -1148,6 +1173,10 @@ export class MemoryStore {
|
|
|
1148
1173
|
const threshold = opts.salienceThreshold ?? 0.01;
|
|
1149
1174
|
const accessRetention = opts.accessLogRetentionDays ?? 60;
|
|
1150
1175
|
const transcriptRetention = opts.transcriptRetentionDays ?? 90;
|
|
1176
|
+
// Behavioral telemetry kept longer than transcripts so the feedback loop
|
|
1177
|
+
// (getFeedbackStats, getBehavioralPatterns, getSkillsToSuppress) has a
|
|
1178
|
+
// wide enough window to aggregate meaningful signal.
|
|
1179
|
+
const behavioralRetention = opts.behavioralRetentionDays ?? 180;
|
|
1151
1180
|
// Prune stale episodic chunks (not vault-sourced content)
|
|
1152
1181
|
const episodicResult = this.conn
|
|
1153
1182
|
.prepare(`DELETE FROM chunks
|
|
@@ -1167,10 +1196,30 @@ export class MemoryStore {
|
|
|
1167
1196
|
.prepare(`DELETE FROM transcripts
|
|
1168
1197
|
WHERE created_at < datetime('now', ?)`)
|
|
1169
1198
|
.run(`-${transcriptRetention} days`);
|
|
1199
|
+
// Behavioral telemetry pruning — these tables were previously unbounded.
|
|
1200
|
+
// Each is append-only, so a rolling window is safe; aggregate stats
|
|
1201
|
+
// consume the window directly rather than historical totals.
|
|
1202
|
+
const skillUsageResult = this.conn
|
|
1203
|
+
.prepare(`DELETE FROM skill_usage WHERE retrieved_at < datetime('now', ?)`)
|
|
1204
|
+
.run(`-${behavioralRetention} days`);
|
|
1205
|
+
const feedbackResult = this.conn
|
|
1206
|
+
.prepare(`DELETE FROM feedback WHERE created_at < datetime('now', ?)`)
|
|
1207
|
+
.run(`-${behavioralRetention} days`);
|
|
1208
|
+
const reflectionsResult = this.conn
|
|
1209
|
+
.prepare(`DELETE FROM session_reflections WHERE created_at < datetime('now', ?)`)
|
|
1210
|
+
.run(`-${behavioralRetention} days`);
|
|
1211
|
+
// Usage log is denser (per-exchange) — keep a shorter window.
|
|
1212
|
+
const usageResult = this.conn
|
|
1213
|
+
.prepare(`DELETE FROM usage_log WHERE created_at < datetime('now', ?)`)
|
|
1214
|
+
.run(`-${Math.min(behavioralRetention, 90)} days`);
|
|
1170
1215
|
return {
|
|
1171
1216
|
episodicPruned: episodicResult.changes,
|
|
1172
1217
|
accessLogPruned: accessResult.changes,
|
|
1173
1218
|
transcriptsPruned: transcriptResult.changes,
|
|
1219
|
+
skillUsagePruned: skillUsageResult.changes,
|
|
1220
|
+
feedbackPruned: feedbackResult.changes,
|
|
1221
|
+
reflectionsPruned: reflectionsResult.changes,
|
|
1222
|
+
usageLogPruned: usageResult.changes,
|
|
1174
1223
|
};
|
|
1175
1224
|
}
|
|
1176
1225
|
// ── Timeline Query ─────────────────────────────────────────────
|
|
@@ -2045,25 +2094,55 @@ export class MemoryStore {
|
|
|
2045
2094
|
.prepare('SELECT id, content FROM chunks')
|
|
2046
2095
|
.all();
|
|
2047
2096
|
if (rows.length === 0)
|
|
2048
|
-
return { vocabSize: 0, backfilled: 0 };
|
|
2097
|
+
return { vocabSize: 0, backfilled: 0, invalidated: 0 };
|
|
2098
|
+
// Capture prior vocab hash BEFORE rebuild. If buildVocab produces a
|
|
2099
|
+
// different word→dimension mapping, previously-stored embedding vectors
|
|
2100
|
+
// become silently wrong (dimension N now represents a different word).
|
|
2101
|
+
const hashFile = path.join(BASE_DIR, '.embedding-vocab.hash');
|
|
2102
|
+
let priorHash = '';
|
|
2103
|
+
try {
|
|
2104
|
+
if (existsSync(hashFile))
|
|
2105
|
+
priorHash = readFileSync(hashFile, 'utf-8').trim();
|
|
2106
|
+
}
|
|
2107
|
+
catch { /* first run */ }
|
|
2049
2108
|
// Build vocabulary from entire corpus (including consolidated summaries)
|
|
2050
2109
|
embeddingsModule.buildVocab(rows.map((r) => r.content));
|
|
2051
2110
|
if (!embeddingsModule.isReady())
|
|
2052
|
-
return { vocabSize: 0, backfilled: 0 };
|
|
2111
|
+
return { vocabSize: 0, backfilled: 0, invalidated: 0 };
|
|
2112
|
+
// If the vocab shifted, invalidate every stored vector so they re-embed
|
|
2113
|
+
// against the new word→dim mapping. Without this, old vectors silently
|
|
2114
|
+
// mismatch query vectors and cosine similarity returns nonsense.
|
|
2115
|
+
const newHash = embeddingsModule.getVocabHash();
|
|
2116
|
+
let invalidated = 0;
|
|
2117
|
+
if (priorHash && priorHash !== newHash) {
|
|
2118
|
+
const res = this.conn.prepare('UPDATE chunks SET embedding = NULL WHERE embedding IS NOT NULL').run();
|
|
2119
|
+
invalidated = res.changes;
|
|
2120
|
+
// Count is returned in the result object — callers (maintenance cycle)
|
|
2121
|
+
// log it there. No local logger in this file to avoid the import.
|
|
2122
|
+
}
|
|
2123
|
+
try {
|
|
2124
|
+
writeFileSync(hashFile, newHash);
|
|
2125
|
+
}
|
|
2126
|
+
catch { /* non-fatal */ }
|
|
2053
2127
|
// Backfill embeddings for all chunks that don't have one
|
|
2054
2128
|
const missing = this.conn
|
|
2055
2129
|
.prepare('SELECT id, content FROM chunks WHERE embedding IS NULL')
|
|
2056
2130
|
.all();
|
|
2057
2131
|
const updateStmt = this.conn.prepare('UPDATE chunks SET embedding = ? WHERE id = ?');
|
|
2058
2132
|
let backfilled = 0;
|
|
2059
|
-
|
|
2060
|
-
|
|
2061
|
-
|
|
2062
|
-
|
|
2063
|
-
|
|
2133
|
+
// Wrap backfill in a transaction — potentially thousands of UPDATEs
|
|
2134
|
+
// per vocab shift, and a single WAL commit is dramatically faster.
|
|
2135
|
+
const backfillAll = this.conn.transaction((items) => {
|
|
2136
|
+
for (const row of items) {
|
|
2137
|
+
const vec = embeddingsModule.embed(row.content);
|
|
2138
|
+
if (vec) {
|
|
2139
|
+
updateStmt.run(embeddingsModule.serializeEmbedding(vec), row.id);
|
|
2140
|
+
backfilled++;
|
|
2141
|
+
}
|
|
2064
2142
|
}
|
|
2065
|
-
}
|
|
2066
|
-
|
|
2143
|
+
});
|
|
2144
|
+
backfillAll(missing);
|
|
2145
|
+
return { vocabSize: rows.length, backfilled, invalidated };
|
|
2067
2146
|
}
|
|
2068
2147
|
// ── Helpers ───────────────────────────────────────────────────────
|
|
2069
2148
|
/**
|
package/dist/tools/shared.d.ts
CHANGED
|
@@ -86,10 +86,15 @@ export type MemoryStoreType = {
|
|
|
86
86
|
salienceThreshold?: number;
|
|
87
87
|
accessLogRetentionDays?: number;
|
|
88
88
|
transcriptRetentionDays?: number;
|
|
89
|
+
behavioralRetentionDays?: number;
|
|
89
90
|
}): {
|
|
90
91
|
episodicPruned: number;
|
|
91
92
|
accessLogPruned: number;
|
|
92
93
|
transcriptsPruned: number;
|
|
94
|
+
skillUsagePruned: number;
|
|
95
|
+
feedbackPruned: number;
|
|
96
|
+
reflectionsPruned: number;
|
|
97
|
+
usageLogPruned: number;
|
|
93
98
|
};
|
|
94
99
|
checkDuplicate(content: string, sourceFile?: string): {
|
|
95
100
|
isDuplicate: boolean;
|