@economic/agents 2.10.0 → 2.10.2

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).
@@ -1487,11 +1492,12 @@ function registerChat(sql, durableObjectName, dateTime) {
 function renameChat(sql, durableObjectName, title, dateTime = Date.now()) {
 	const trimmed = title.trim();
 	if (!trimmed) throw new Error("Chat title cannot be empty");
-	sql`UPDATE chats
+	if (!sql`UPDATE chats
     SET title = ${trimmed},
         title_set_by_user = 1,
         updated_at = ${dateTime}
-    WHERE durable_object_name = ${durableObjectName}`;
+    WHERE durable_object_name = ${durableObjectName}
+    RETURNING durable_object_name`.length) throw new Error(`Chat not found: ${durableObjectName}`);
 }
 function deleteChat(sql, durableObjectName) {
 	sql`DELETE FROM chats WHERE durable_object_name = ${durableObjectName}`;
@@ -1504,6 +1510,7 @@ async function getChats(sql) {
 	return sql`SELECT * FROM chats ORDER BY updated_at DESC`;
 }
 function recordChatFeedback(sql, durableObjectName, messageId, rating, dateTime = Date.now()) {
+	if (rating !== 1 && rating !== -1 && rating !== 0) throw new RangeError("Message feedback rating must be 1, -1, or 0");
 	if (!getChat(sql, durableObjectName)) throw new Error(`Chat not found: ${durableObjectName}`);
 	if (rating === 0) sql`DELETE FROM chat_message_feedback
       WHERE chat_id = ${durableObjectName} AND message_id = ${messageId}`;
@@ -1526,15 +1533,14 @@ function recordChatFeedback(sql, durableObjectName, messageId, rating, dateTime
 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.
@@ -1836,9 +1842,16 @@ var Assistant = class extends Agent$1 {
 		registerChat(this.sql.bind(this), durableObjectName, Date.now());
 	}
 	async recordChatTurn(durableObjectName, messages) {
-		summariseChatWithAI(this.sql.bind(this), durableObjectName, messages, this.fastModel);
-		this.scheduleChatForAutoDeletion(durableObjectName).catch((error) => {
-			console.error("[Assistant] Failed to reschedule chat deletion", {
+		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", {
+					durableObjectName,
+					error
+				});
+			})]);
+		}).catch((error) => {
+			console.error("[Assistant] Failed to record chat metadata", {
 				durableObjectName,
 				error
 			});
@@ -1939,6 +1952,7 @@ function ensureFeedbackTableExists(sql) {
 * @param now - The date and time to use for the created_at and updated_at columns.
 */
 function submitMessageFeedback(sql, messageId, rating, comment, now = /* @__PURE__ */ new Date()) {
+	if (rating !== 1 && rating !== -1 && rating !== 0) throw new RangeError("Message feedback rating must be 1, -1, or 0");
 	try {
 		if (rating === 0) {
 			sql`DELETE FROM assistant_messages_feedback WHERE message_id = ${messageId}`;

package/package.json

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