audrey 0.9.0 → 0.14.0

package/src/recall.js

@@ -150,7 +150,8 @@ function matchesDateFilters(createdAt, filters) {
   return true;
 }
 
-function knnEpisodic(db, queryBuffer, candidateK, now, minConfidence, includeProvenance, confidenceConfig, filters = {}) {
+function knnEpisodic(db, queryBuffer, candidateK, now, minConfidence, includeProvenance, confidenceConfig, filters = {}, includePrivate = false) {
+  const privateClause = includePrivate ? '' : 'AND e."private" = 0';
   const rows = db.prepare(`
     SELECT e.*, (1.0 - v.distance) AS similarity
     FROM vec_episodes v
@@ -158,6 +159,7 @@ function knnEpisodic(db, queryBuffer, candidateK, now, minConfidence, includePro
     WHERE v.embedding MATCH ?
       AND k = ?
       AND e.superseded_by IS NULL
+      ${privateClause}
   `).all(queryBuffer, candidateK);
 
   const results = [];
@@ -258,6 +260,7 @@ export async function* recallStream(db, embeddingProvider, query, options = {})
     sources,
     after,
     before,
+    includePrivate = false,
   } = options;
 
   const queryVector = await embeddingProvider.embed(query);
@@ -271,7 +274,7 @@ export async function* recallStream(db, embeddingProvider, query, options = {})
   const allResults = [];
 
   if (searchTypes.includes('episodic')) {
-    const episodic = knnEpisodic(db, queryBuffer, candidateK, now, minConfidence, includeProvenance, confidenceConfig, filters);
+    const episodic = knnEpisodic(db, queryBuffer, candidateK, now, minConfidence, includeProvenance, confidenceConfig, filters, includePrivate);
     allResults.push(...episodic);
   }
 
@@ -281,13 +284,11 @@ export async function* recallStream(db, embeddingProvider, query, options = {})
     allResults.push(...semResults);
 
     if (semIds.length > 0) {
-      const updateStmt = db.prepare(
-        'UPDATE semantics SET retrieval_count = retrieval_count + 1, last_reinforced_at = ? WHERE id = ?'
-      );
       const nowISO = now.toISOString();
-      for (const id of semIds) {
-        updateStmt.run(nowISO, id);
-      }
+      const placeholders = semIds.map(() => '?').join(',');
+      db.prepare(
+        `UPDATE semantics SET retrieval_count = retrieval_count + 1, last_reinforced_at = ? WHERE id IN (${placeholders})`
+      ).run(nowISO, ...semIds);
     }
   }
 
@@ -297,13 +298,11 @@ export async function* recallStream(db, embeddingProvider, query, options = {})
     allResults.push(...procResults);
 
     if (procIds.length > 0) {
-      const updateStmt = db.prepare(
-        'UPDATE procedures SET retrieval_count = retrieval_count + 1, last_reinforced_at = ? WHERE id = ?'
-      );
       const nowISO = now.toISOString();
-      for (const id of procIds) {
-        updateStmt.run(nowISO, id);
-      }
+      const placeholders = procIds.map(() => '?').join(',');
+      db.prepare(
+        `UPDATE procedures SET retrieval_count = retrieval_count + 1, last_reinforced_at = ? WHERE id IN (${placeholders})`
+      ).run(nowISO, ...procIds);
     }
   }