@signetai/connector-base 0.206.0 → 0.206.2

package/dist/index.js

@@ -11311,6 +11311,7 @@ var DAEMON_DERIVED_MEMORY_SOURCE_TYPES = [
   "dreaming"
 ];
 var MEMORIES_FTS_TOKENIZER = "unicode61";
+var FTS_STATE_TABLE = "memories_fts_state";
 function normalizeSql(sql) {
   return sql.replace(/\s+/g, " ").trim().toLowerCase();
 }
@@ -11323,18 +11324,67 @@ function createMemoriesFts(db) {
 			tokenize='${MEMORIES_FTS_TOKENIZER}'
 		);
 	`);
+  ensureMemoriesFtsState(db);
+  createMemoriesFtsTriggers(db);
+}
+function ensureMemoriesFtsState(db) {
   db.exec(`
-		CREATE TRIGGER IF NOT EXISTS memories_ai AFTER INSERT ON memories BEGIN
-			INSERT INTO memories_fts(rowid, content) VALUES (new.rowid, new.content);
-		END;
+		CREATE TABLE IF NOT EXISTS ${FTS_STATE_TABLE} (
+			id INTEGER PRIMARY KEY CHECK (id = 1),
+			memory_count INTEGER NOT NULL,
+			indexed_count INTEGER NOT NULL,
+			updated_at TEXT NOT NULL
+		);
+		INSERT OR IGNORE INTO ${FTS_STATE_TABLE} (id, memory_count, indexed_count, updated_at)
+		VALUES (
+			1,
+			-1,
+			0,
+			datetime('now')
+		);
 	`);
+}
+function createMemoriesFtsTriggers(db) {
   db.exec(`
-		CREATE TRIGGER IF NOT EXISTS memories_ad AFTER DELETE ON memories BEGIN
+		DROP TRIGGER IF EXISTS memories_ai;
+		DROP TRIGGER IF EXISTS memories_ad;
+		DROP TRIGGER IF EXISTS memories_au;
+
+		CREATE TRIGGER memories_ai AFTER INSERT ON memories BEGIN
+			INSERT INTO memories_fts(rowid, content) VALUES (new.rowid, new.content);
+			UPDATE ${FTS_STATE_TABLE}
+			 SET memory_count = CASE WHEN memory_count < 0 THEN -1 ELSE memory_count + 1 END,
+			     indexed_count = CASE WHEN indexed_count < 0 THEN 0 ELSE indexed_count + 1 END,
+			     updated_at = datetime('now')
+			 WHERE id = 1;
+		END;
+
+		CREATE TRIGGER memories_ad AFTER DELETE ON memories BEGIN
+			UPDATE ${FTS_STATE_TABLE}
+			 SET memory_count = CASE WHEN memory_count < 0 THEN -1 ELSE MAX(0, memory_count - 1) END,
+			     indexed_count = CASE
+					WHEN indexed_count < 0 THEN 0
+					ELSE MAX(0, indexed_count - CASE WHEN EXISTS(
+						SELECT 1 FROM memories_fts_docsize WHERE id = old.rowid
+					) THEN 1 ELSE 0 END)
+				END,
+			     updated_at = datetime('now')
+			 WHERE id = 1;
 			INSERT INTO memories_fts(memories_fts, rowid, content) VALUES('delete', old.rowid, old.content);
 		END;
-	`);
-  db.exec(`
-		CREATE TRIGGER IF NOT EXISTS memories_au AFTER UPDATE OF content ON memories BEGIN
+
+		CREATE TRIGGER memories_au AFTER UPDATE OF content ON memories BEGIN
+			UPDATE ${FTS_STATE_TABLE}
+			 SET indexed_count = CASE
+					WHEN indexed_count < 0 THEN CASE WHEN EXISTS(
+						SELECT 1 FROM memories_fts_docsize WHERE id = old.rowid
+					) THEN 0 ELSE 1 END
+					ELSE indexed_count + CASE WHEN EXISTS(
+						SELECT 1 FROM memories_fts_docsize WHERE id = old.rowid
+					) THEN 0 ELSE 1 END
+				END,
+				 updated_at = datetime('now')
+			 WHERE id = 1;
 			INSERT INTO memories_fts(memories_fts, rowid, content) VALUES('delete', old.rowid, old.content);
 			INSERT INTO memories_fts(rowid, content) VALUES (new.rowid, new.content);
 		END;
@@ -11345,11 +11395,23 @@ function recreateMemoriesFtsSchema(db) {
   db.exec("DROP TRIGGER IF EXISTS memories_ad");
   db.exec("DROP TRIGGER IF EXISTS memories_au");
   db.exec("DROP TABLE IF EXISTS memories_fts");
+  db.exec(`DROP TABLE IF EXISTS ${FTS_STATE_TABLE}`);
   createMemoriesFts(db);
 }
 function recreateMemoriesFts(db) {
   recreateMemoriesFtsSchema(db);
   db.exec("INSERT INTO memories_fts(rowid, content) SELECT rowid, content FROM memories");
+  refreshMemoriesFtsState(db);
+}
+function refreshMemoriesFtsState(db) {
+  ensureMemoriesFtsState(db);
+  db.exec(`
+		UPDATE ${FTS_STATE_TABLE}
+		 SET memory_count = (SELECT COUNT(*) FROM memories),
+		     indexed_count = (SELECT COUNT(*) FROM memories_fts_docsize),
+		     updated_at = datetime('now')
+		 WHERE id = 1;
+	`);
 }
 function readMemoriesFtsSql(db) {
   const row = db.prepare("SELECT sql FROM sqlite_master WHERE name = 'memories_fts' AND type = 'table'").get();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@signetai/connector-base",
-  "version": "0.206.0",
+  "version": "0.206.2",
   "description": "Base class for Signet harness connectors",
   "type": "module",
   "main": "./dist/index.js",
@@ -26,7 +26,7 @@
     "typecheck": "tsc --noEmit"
   },
   "dependencies": {
-    "@signetai/core": "0.206.0",
+    "@signetai/core": "0.206.2",
     "json5": "^2.2.3",
     "jsonc-parser": "3.3.1"
   },