@economic/agents 2.3.22 → 2.3.23

package/dist/index.d.mts

@@ -239,6 +239,8 @@ type Chat = {
   summary?: string;
   created_at: number;
   updated_at: number;
+  positive_feedback_count: number;
+  negative_feedback_count: number;
 };
 declare const DELETE_CHAT_CALLBACK: "deleteChatCallback";
 //#endregion
@@ -252,7 +254,9 @@ declare abstract class Assistant<State = unknown> extends Agent$1<Cloudflare.Env
   createChat(): Promise<string>;
   deleteChat(id: string): Promise<void>;
   getChats(): Promise<Chat[]>;
+  getChatAgentClassName(): Promise<string>;
   recordChatTurn(durableObjectName: string, messages: UIMessage[]): Promise<void>;
+  recordChatFeedback(durableObjectName: string, messageId: string, rating: number): Promise<void>;
   private [DELETE_CHAT_CALLBACK];
   private scheduleChatForAutoDeletion;
   /**

package/dist/index.mjs

@@ -388,12 +388,33 @@ function ensureChatsTableExists(sql) {
         summary             TEXT,
         created_at          INTEGER NOT NULL,
         updated_at          INTEGER NOT NULL,
+        positive_feedback_count INTEGER NOT NULL DEFAULT 0,
+        negative_feedback_count INTEGER NOT NULL DEFAULT 0,
         PRIMARY KEY (durable_object_name)
       )`;
+		ensureChatsFeedbackColumnsExist(sql);
+		ensureChatMessageFeedbackTableExists(sql);
 	} catch (error) {
 		console.error("[Agent] Failed to create chats table", error);
 	}
 }
+function ensureChatsFeedbackColumnsExist(sql) {
+	try {
+		sql`ALTER TABLE chats ADD COLUMN positive_feedback_count INTEGER NOT NULL DEFAULT 0`;
+	} catch {}
+	try {
+		sql`ALTER TABLE chats ADD COLUMN negative_feedback_count INTEGER NOT NULL DEFAULT 0`;
+	} catch {}
+}
+function ensureChatMessageFeedbackTableExists(sql) {
+	sql`CREATE TABLE IF NOT EXISTS chat_message_feedback (
+      chat_id    TEXT NOT NULL,
+      message_id TEXT NOT NULL,
+      rating     INTEGER NOT NULL,
+      updated_at INTEGER NOT NULL,
+      PRIMARY KEY (chat_id, message_id)
+    )`;
+}
 function registerChat(sql, durableObjectName, dateTime) {
 	sql`INSERT INTO chats (durable_object_name, created_at, updated_at)
     VALUES (${durableObjectName}, ${dateTime}, ${dateTime})`;
@@ -415,6 +436,7 @@ function registerChatWithMetadata(sql, durableObjectName, metadata) {
 }
 function deleteChat(sql, durableObjectName) {
 	sql`DELETE FROM chats WHERE durable_object_name = ${durableObjectName}`;
+	sql`DELETE FROM chat_message_feedback WHERE chat_id = ${durableObjectName}`;
 }
 function getChat(sql, durableObjectName) {
 	return sql`SELECT * FROM chats WHERE durable_object_name = ${durableObjectName}`[0] ?? null;
@@ -422,6 +444,23 @@ function getChat(sql, durableObjectName) {
 async function getChats(sql) {
 	return sql`SELECT * FROM chats ORDER BY updated_at DESC`;
 }
+function recordChatFeedback(sql, durableObjectName, messageId, rating, dateTime = Date.now()) {
+	sql`INSERT INTO chat_message_feedback (chat_id, message_id, rating, updated_at)
+    VALUES (${durableObjectName}, ${messageId}, ${rating}, ${dateTime})
+    ON CONFLICT (chat_id, message_id) DO UPDATE SET
+      rating = excluded.rating,
+      updated_at = excluded.updated_at`;
+	const positiveRows = sql`SELECT COUNT(*) AS count
+    FROM chat_message_feedback
+    WHERE chat_id = ${durableObjectName} AND rating > 0`;
+	const negativeRows = sql`SELECT COUNT(*) AS count
+    FROM chat_message_feedback
+    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)}
+    WHERE durable_object_name = ${durableObjectName}`;
+}
 function getDeleteChatScheduleIds(schedules) {
 	return schedules.filter((schedule) => schedule.callback === DELETE_CHAT_CALLBACK).map((schedule) => schedule.id);
 }
@@ -717,6 +756,11 @@ var Assistant = class extends Agent$1 {
 				callable(),
 				2,
 				"getChats"
+			],
+			[
+				callable(),
+				2,
+				"getChatAgentClassName"
 			]
 		], 0, void 0, Agent$1).e;
 	}
@@ -774,10 +818,16 @@ var Assistant = class extends Agent$1 {
 	async getChats() {
 		return getChats(this.sql.bind(this));
 	}
+	async getChatAgentClassName() {
+		return this.agent.name;
+	}
 	async recordChatTurn(durableObjectName, messages) {
 		summariseChatWithAI(this.sql.bind(this), durableObjectName, messages, this.fastModel);
 		this.scheduleChatForAutoDeletion(durableObjectName);
 	}
+	async recordChatFeedback(durableObjectName, messageId, rating) {
+		recordChatFeedback(this.sql.bind(this), durableObjectName, messageId, rating);
+	}
 	async [DELETE_CHAT_CALLBACK](durableObjectName) {
 		await this.deleteChat(durableObjectName);
 	}
@@ -1092,11 +1142,21 @@ var ChatAgent = class extends Agent {
 	*/
 	async submitMessageFeedback(messageId, rating, comment) {
 		const result = submitMessageFeedback(this.sql.bind(this), messageId, rating, comment);
+		const parent = await this.getParentAgent();
 		recordFeedbackAnalytics(this.env.AGENTS_ANALYTICS, {
 			agentName: this.getTelemetryAgentName(),
 			durableObjectName: this.name,
 			actorId: this.getTelemetryActorId()
 		}, messageId, rating);
+		if (parent?.recordChatFeedback) try {
+			await parent.recordChatFeedback(this.name, messageId, rating);
+		} catch (error) {
+			console.error("[ChatAgent] Failed to update parent chat feedback summary", {
+				durableObjectName: this.name,
+				messageId,
+				error
+			});
+		}
 		return result;
 	}
 	/**
@@ -1120,11 +1180,23 @@ var ChatAgent = class extends Agent {
 	*/
 	async importLegacyMessages(messages, feedback = []) {
 		let parentId = null;
+		const parent = await this.getParentAgent();
 		for (const message of messages) {
 			await this.appendMessageToHistory(message, parentId);
 			parentId = message.id;
 		}
-		for (const item of feedback) submitMessageFeedback(this.sql.bind(this), item.messageId, item.rating, item.comment);
+		for (const item of feedback) {
+			submitMessageFeedback(this.sql.bind(this), item.messageId, item.rating, item.comment);
+			try {
+				await parent?.recordChatFeedback?.(this.name, item.messageId, item.rating);
+			} catch (error) {
+				console.error("[ChatAgent] Failed to import parent chat feedback summary", {
+					durableObjectName: this.name,
+					messageId: item.messageId,
+					error
+				});
+			}
+		}
 	}
 };
 //#endregion

package/package.json

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