agents 0.16.0 → 0.16.2

package/dist/experimental/memory/session/index.js

@@ -740,6 +740,7 @@ var AgentSessionProvider = class {
 	*/
 	constructor(agent, sessionId) {
 		this.initialized = false;
+		this.activeLeafId = void 0;
 		this.agent = agent;
 		this.sessionId = sessionId ?? "";
 	}
@@ -880,6 +881,7 @@ var AgentSessionProvider = class {
       VALUES (${message.id}, ${this.sessionId}, ${parent}, ${message.role}, ${json})
     `;
 		this.indexFTS(message);
+		this.activeLeafId = message.id;
 	}
 	updateMessage(message) {
 		this.ensureTable();
@@ -895,6 +897,7 @@ var AgentSessionProvider = class {
 			this.agent.sql`DELETE FROM assistant_messages WHERE id = ${id} AND session_id = ${this.sessionId}`;
 			this.deleteFTS(id);
 		}
+		if (typeof this.activeLeafId === "string" && messageIds.includes(this.activeLeafId)) this.activeLeafId = void 0;
 	}
 	clearMessages() {
 		this.ensureTable();
@@ -904,6 +907,7 @@ var AgentSessionProvider = class {
       SELECT rowid FROM assistant_fts WHERE session_id = ${this.sessionId}
     `;
 		for (const row of ftsRows) this.agent.sql`DELETE FROM assistant_fts WHERE rowid = ${row.rowid}`;
+		this.activeLeafId = void 0;
 	}
 	addCompaction(summary, fromMessageId, toMessageId) {
 		this.ensureTable();
@@ -951,12 +955,26 @@ var AgentSessionProvider = class {
 		}
 	}
 	latestLeafRow() {
-		return this.agent.sql`
+		if (this.activeLeafId !== void 0) {
+			if (this.agent.sql`
+        SELECT m.id FROM assistant_messages m
+        WHERE m.id = ${this.activeLeafId} AND m.session_id = ${this.sessionId}
+          AND NOT EXISTS (
+            SELECT 1 FROM assistant_messages c
+            WHERE c.parent_id = ${this.activeLeafId}
+              AND c.session_id = ${this.sessionId}
+          )
+      `.length > 0) return { id: this.activeLeafId };
+			this.activeLeafId = void 0;
+		}
+		const rows = this.agent.sql`
       SELECT m.id FROM assistant_messages m
       LEFT JOIN assistant_messages c ON c.parent_id = m.id AND c.session_id = ${this.sessionId}
       WHERE c.id IS NULL AND m.session_id = ${this.sessionId}
       ORDER BY m.created_at DESC, m.rowid DESC LIMIT 1
-    `[0] ?? null;
+    `;
+		this.activeLeafId = rows[0]?.id;
+		return rows[0] ?? null;
 	}
 	leafRowById(leafId) {
 		return this.agent.sql`