@economic/agents 2.10.1 → 2.10.3

package/dist/index.d.mts

@@ -383,6 +383,8 @@ declare abstract class Assistant<State = unknown> extends Agent$1<Cloudflare.Env
    * Best-effort and non-blocking.
    */
   private touchInstance;
+  private syncChatToGlobalList;
+  private deleteChatFromGlobalList;
   observability?: Observability;
 }
 //#endregion

package/dist/index.mjs

@@ -1480,6 +1480,11 @@ function registerChat(sql, durableObjectName, dateTime) {
     VALUES (${durableObjectName}, ${dateTime}, ${dateTime})
     ON CONFLICT (durable_object_name) DO NOTHING`;
 }
+function touchChat(sql, durableObjectName, dateTime = Date.now()) {
+	sql`UPDATE chats
+    SET updated_at = ${dateTime}
+    WHERE durable_object_name = ${durableObjectName}`;
+}
 /**
 * Sets a chat's title on the user's behalf. Marks the title as user-set so
 * the AI summariser stops overwriting it (it keeps updating the summary).
@@ -1522,21 +1527,21 @@ function recordChatFeedback(sql, durableObjectName, messageId, rating, dateTime
     WHERE chat_id = ${durableObjectName} AND rating < 0`;
 	sql`UPDATE chats
     SET positive_feedback_count = ${Number(positiveRows[0]?.count ?? 0)},
-        negative_feedback_count = ${Number(negativeRows[0]?.count ?? 0)}
+        negative_feedback_count = ${Number(negativeRows[0]?.count ?? 0)},
+        updated_at = ${dateTime}
     WHERE durable_object_name = ${durableObjectName}`;
 }
 function getDeleteChatScheduleIds(schedules, durableObjectName) {
 	return schedules.filter((schedule) => schedule.callback === "deleteChatCallback" && schedule.payload === durableObjectName).map((schedule) => schedule.id);
 }
-/**
-* Number of recent messages passed to `generateSummary` for rolling
-* summarization. Keeping this bounded prevents the prompt growing
-* unboundedly regardless of chat length.
-*/
-const CHAT_RECENT_MESSAGES_COUNT = 20;
+const CHAT_SUMMARY_TURN_INTERVAL = 5;
+function isChatSummaryDue(messages) {
+	const completedTurns = messages.filter((message) => message.role === "user").length;
+	return completedTurns > 0 && completedTurns % CHAT_SUMMARY_TURN_INTERVAL === 0;
+}
 async function summariseChatWithAI(sql, durableObjectName, messages, model) {
 	const chat = getChat(sql, durableObjectName);
-	if (!(!chat || !chat.title || messages.length % CHAT_RECENT_MESSAGES_COUNT === 0)) return;
+	if (!(!chat || !chat.title || isChatSummaryDue(messages))) return;
 	let systemPrompt = `
     You are a helpful assistant that summarises chats.
     You will be given a list of messages and you need to generate a title and summary for the chat.
@@ -1587,6 +1592,24 @@ function getChatRetentionMs(days) {
 	return Math.floor(days * 24 * 60 * 60 * 1e3);
 }
 //#endregion
+//#region src/server/features/global-chats.ts
+async function upsertGlobalChat(db, chat) {
+	await db.prepare(`INSERT INTO agent_chats (
+         agent_name, assistant_name, chat_id, title, summary, created_at, updated_at,
+         positive_feedback_count, negative_feedback_count
+       ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)
+       ON CONFLICT (agent_name, assistant_name, chat_id) DO UPDATE SET
+         title = excluded.title,
+         summary = excluded.summary,
+         updated_at = excluded.updated_at,
+         positive_feedback_count = excluded.positive_feedback_count,
+         negative_feedback_count = excluded.negative_feedback_count`).bind(chat.agentName, chat.assistantName, chat.chatId, chat.title, chat.summary, chat.createdAt, chat.updatedAt, chat.positiveFeedbackCount, chat.negativeFeedbackCount).run();
+}
+async function deleteGlobalChat(db, agentName, assistantName, chatId) {
+	await db.prepare(`DELETE FROM agent_chats
+       WHERE agent_name = ? AND assistant_name = ? AND chat_id = ?`).bind(agentName, assistantName, chatId).run();
+}
+//#endregion
 //#region src/server/agents/Assistant.ts
 let _initProto$1;
 function _applyDecs$1(e, t, n, r, o, i) {
@@ -1824,9 +1847,11 @@ var Assistant = class extends Agent$1 {
 	async deleteChat(id) {
 		await this.deleteSubAgent(this.agent, id);
 		deleteChat(this.sql.bind(this), id);
+		await this.deleteChatFromGlobalList(id);
 	}
 	async renameChat(id, title) {
 		renameChat(this.sql.bind(this), id, title);
+		await this.syncChatToGlobalList(id);
 	}
 	async getChats() {
 		return getChats(this.sql.bind(this));
@@ -1836,8 +1861,10 @@ var Assistant = class extends Agent$1 {
 	}
 	async registerChat(durableObjectName) {
 		registerChat(this.sql.bind(this), durableObjectName, Date.now());
+		await this.syncChatToGlobalList(durableObjectName);
 	}
 	async recordChatTurn(durableObjectName, messages) {
+		touchChat(this.sql.bind(this), durableObjectName);
 		this.keepAliveWhile(async () => {
 			await Promise.all([summariseChatWithAI(this.sql.bind(this), durableObjectName, messages, this.fastModel), this.scheduleChatForAutoDeletion(durableObjectName).catch((error) => {
 				console.error("[Assistant] Failed to reschedule chat deletion", {
@@ -1845,6 +1872,7 @@ var Assistant = class extends Agent$1 {
 					error
 				});
 			})]);
+			await this.syncChatToGlobalList(durableObjectName);
 		}).catch((error) => {
 			console.error("[Assistant] Failed to record chat metadata", {
 				durableObjectName,
@@ -1854,6 +1882,7 @@ var Assistant = class extends Agent$1 {
 	}
 	async recordChatFeedback(durableObjectName, messageId, rating) {
 		recordChatFeedback(this.sql.bind(this), durableObjectName, messageId, rating);
+		await this.syncChatToGlobalList(durableObjectName);
 	}
 	async [DELETE_CHAT_CALLBACK](durableObjectName) {
 		await this.deleteChat(durableObjectName);
@@ -1909,6 +1938,38 @@ var Assistant = class extends Agent$1 {
 			console.error("[Assistant] Failed to update agent_registry activity", error);
 		}
 	}
+	async syncChatToGlobalList(durableObjectName) {
+		const chat = getChat(this.sql.bind(this), durableObjectName);
+		if (!chat) return;
+		try {
+			await upsertGlobalChat(this.env.AGENTS_DB, {
+				agentName: this.constructor.name,
+				assistantName: this.name,
+				chatId: chat.durable_object_name,
+				title: chat.title ?? null,
+				summary: chat.summary ?? null,
+				createdAt: chat.created_at,
+				updatedAt: chat.updated_at,
+				positiveFeedbackCount: chat.positive_feedback_count,
+				negativeFeedbackCount: chat.negative_feedback_count
+			});
+		} catch (error) {
+			console.error("[Assistant] Failed to sync chat to agent_chats", {
+				durableObjectName,
+				error
+			});
+		}
+	}
+	async deleteChatFromGlobalList(durableObjectName) {
+		try {
+			await deleteGlobalChat(this.env.AGENTS_DB, this.constructor.name, this.name, durableObjectName);
+		} catch (error) {
+			console.error("[Assistant] Failed to delete chat from agent_chats", {
+				durableObjectName,
+				error
+			});
+		}
+	}
 	observability = createWorkerObservability({ getIdentity: () => ({
 		agentName: this.constructor.name,
 		agentClass: this.constructor.name

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@economic/agents",
-  "version": "2.10.1",
+  "version": "2.10.3",
   "description": "A starter for creating a TypeScript package.",
   "license": "MIT",
   "files": [