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.
@@ -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',
@@ -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.
@@ -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
- for (const filePath of filesToUpdate) {
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
- continue;
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
- this.conn
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
- const rows = this.conn
842
- .prepare(`SELECT id, source_file, section, content, chunk_type, embedding, salience, agent_slug, updated_at, category, topic
843
- FROM chunks
844
- WHERE embedding IS NOT NULL`)
845
- .all();
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
- for (const row of missing) {
2060
- const vec = embeddingsModule.embed(row.content);
2061
- if (vec) {
2062
- updateStmt.run(embeddingsModule.serializeEmbedding(vec), row.id);
2063
- backfilled++;
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
- return { vocabSize: rows.length, backfilled };
2143
+ });
2144
+ backfillAll(missing);
2145
+ return { vocabSize: rows.length, backfilled, invalidated };
2067
2146
  }
2068
2147
  // ── Helpers ───────────────────────────────────────────────────────
2069
2148
  /**
@@ -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;
package/package.json CHANGED
@@ -1,6 +1,6 @@
1
1
  {
2
2
  "name": "clementine-agent",
3
- "version": "1.0.32",
3
+ "version": "1.0.33",
4
4
  "description": "Clementine — Personal AI Assistant (TypeScript)",
5
5
  "type": "module",
6
6
  "main": "dist/index.js",