@economic/agents 2.1.2 → 2.1.3

package/dist/v2.mjs

@@ -473,12 +473,11 @@ function ensureRatingsTableExists(sql) {
 	try {
 		sql`CREATE TABLE IF NOT EXISTS assistant_messages_ratings (
         message_id          TEXT NOT NULL,
-        durable_object_name TEXT NOT NULL,
         rating              INTEGER,
         comment             TEXT,
-        created_at          TEXT NOT NULL,
-        updated_at          TEXT NOT NULL,
-        PRIMARY KEY (message_id, durable_object_name)
+        created_at          INTEGER NOT NULL,
+        updated_at          INTEGER NOT NULL,
+        PRIMARY KEY (message_id)
       )`;
 	} catch (error) {
 		console.error("[Agent] Failed to create ratings table", error);
@@ -488,15 +487,14 @@ function ensureRatingsTableExists(sql) {
 * Rates a message.
 * @param sql - The SQL function to use to execute the query.
 * @param messageId - The ID of the message to rate.
-* @param durable_object_name - The name of the Durable Object to rate the message for.
 * @param rating - The rating to give the message.
 * @param now - The date and time to use for the created_at and updated_at columns.
 */
-function rateMessage(sql, messageId, durable_object_name, rating, comment, now = /* @__PURE__ */ new Date()) {
+function rateMessage(sql, messageId, rating, comment, now = /* @__PURE__ */ new Date()) {
 	try {
-		sql`INSERT INTO assistant_messages_ratings (message_id, durable_object_name, rating, comment, created_at, updated_at)
-        VALUES (${messageId}, ${durable_object_name}, ${rating}, ${comment ?? null}, ${now.toISOString()}, ${now.toISOString()})
-        ON CONFLICT (message_id, durable_object_name) DO UPDATE SET
+		sql`INSERT INTO assistant_messages_ratings (message_id, rating, comment, created_at, updated_at)
+        VALUES (${messageId}, ${rating}, ${comment ?? null}, ${now.getTime()}, ${now.getTime()})
+        ON CONFLICT (message_id) DO UPDATE SET
           rating = excluded.rating,
           comment = excluded.comment,
         updated_at = excluded.updated_at`;
@@ -507,12 +505,11 @@ function rateMessage(sql, messageId, durable_object_name, rating, comment, now =
 /**
 * Gets the ratings for a message.
 * @param sql - The SQL function to use to execute the query.
-* @param durable_object_name - The name of the Durable Object to get the ratings for.
 * @returns A record of message IDs and their ratings.
 */
-function getMessageRatings(sql, durable_object_name) {
+function getMessageRatings(sql) {
 	try {
-		const ratings = sql`SELECT message_id, rating, comment FROM assistant_messages_ratings WHERE durable_object_name = ${durable_object_name}`;
+		const ratings = sql`SELECT message_id, rating, comment FROM assistant_messages_ratings`;
 		return Object.fromEntries(ratings.map((row) => [row.message_id, {
 			rating: row.rating,
 			comment: row.comment
@@ -546,14 +543,14 @@ var ChatAgent = class extends Agent {
 	* @returns The message id and the rating.
 	*/
 	@callable({ description: "Rate a message by its id" }) async rateMessage(messageId, rating, comment) {
-		return rateMessage(this.sql.bind(this), messageId, this.name, rating, comment);
+		return rateMessage(this.sql.bind(this), messageId, rating, comment);
 	}
 	/**
 	* Returns all message ratings for the current chat.
 	* @returns All message ratings for the current chat.
 	*/
 	@callable({ description: "Returns all message ratings for the current chat" }) async getMessageRatings() {
-		return getMessageRatings(this.sql.bind(this), this.name);
+		return getMessageRatings(this.sql.bind(this));
 	}
 };
 //#endregion

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@economic/agents",
-  "version": "2.1.2",
+  "version": "2.1.3",
   "description": "A starter for creating a TypeScript package.",
   "license": "MIT",
   "bin": {