@economic/agents 2.1.2 → 2.1.4

package/dist/v2.d.mts

@@ -2,7 +2,6 @@ import { t as JwtAuthConfig } from "./index-DzOC3mNl.mjs";
 import { JSONValue, LanguageModel, Tool as Tool$1, UIMessage } from "ai";
 import { Agent as Agent$1, Connection, ConnectionContext, SubAgentClass } from "agents";
 import { ChatResponseResult, Session, StepConfig, Think, ToolCallContext, ToolCallDecision, TurnConfig, TurnContext } from "@cloudflare/think";
-
 //#region src/server/v2/types.d.ts
 /**
  * The context object available throughout an agent's lifetime — passed via
@@ -118,6 +117,15 @@ declare abstract class Agent<RequestContext extends Record<string, unknown> = Re
   protected getUserContext?(jwtToken: string): Promise<UserContext>;
 }
 //#endregion
+//#region src/server/v2/features/messages.d.ts
+type MessageFeedback = {
+  message_id: string;
+  rating: number;
+  comment?: string;
+  created_at: number;
+  updated_at: number;
+};
+//#endregion
 //#region src/server/v2/agents/ChatAgent.d.ts
 declare abstract class ChatAgent<RequestContext extends Record<string, unknown> = Record<string, unknown>, UserContext extends Record<string, unknown> = Record<string, unknown>> extends Agent<RequestContext, UserContext> {
   initialState: AgentConnectionState;
@@ -125,17 +133,17 @@ declare abstract class ChatAgent<RequestContext extends Record<string, unknown>
   configureSession(session: Session): Session;
   onChatResponse(_result: ChatResponseResult): Promise<void>;
   /**
-   * Rate a message by its id.
-   * @param messageId - The id of the message to rate.
+   * Submit feedback for a message by its id.
+   * @param messageId - The id of the message to give feedback on.
    * @param rating - The rating to give the message. 1 = thumbs up, -1 = thumbs down.
    * @returns The message id and the rating.
    */
-  rateMessage(messageId: string, rating: number, comment?: string): Promise<void>;
+  submitMessageFeedback(messageId: string, rating: number, comment?: string): Promise<void>;
   /**
-   * Returns all message ratings for the current chat.
-   * @returns All message ratings for the current chat.
+   * Returns all message feedback for the current chat.
+   * @returns All message feedback for the current chat.
    */
-  getMessageRatings(): Promise<any>;
+  getMessageFeedback(): Promise<Record<string, MessageFeedback>>;
 }
 //#endregion
 //#region src/server/v2/features/chats.d.ts

package/dist/v2.mjs

@@ -466,57 +466,51 @@ const createCompactFn = (model) => createCompactFunction({ summarize: (prompt) =
 	prompt
 }).then((r) => r.text) });
 /**
-* Ensures that the ratings table exists.
+* Ensures that the feedback table exists.
 * @param sql - The SQL function to use to execute the query.
 */
-function ensureRatingsTableExists(sql) {
+function ensureFeedbackTableExists(sql) {
 	try {
-		sql`CREATE TABLE IF NOT EXISTS assistant_messages_ratings (
+		sql`CREATE TABLE IF NOT EXISTS assistant_messages_feedback (
         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);
+		console.error("[Agent] Failed to create feedback table", error);
 	}
 }
 /**
-* Rates a message.
+* Submits feedback for 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 messageId - The ID of the message to give feedback on.
 * @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 submitMessageFeedback(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_feedback (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`;
 	} catch (error) {
-		console.error("[Agent] Failed to rate message", error);
+		console.error("[Agent] Failed to submit message feedback", error);
 	}
 }
 /**
-* Gets the ratings for a message.
+* Gets the feedback for all messages.
 * @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.
+* @returns A dictionary of message feedback keyed by message id.
 */
-function getMessageRatings(sql, durable_object_name) {
+function getMessageFeedback(sql) {
 	try {
-		const ratings = sql`SELECT message_id, rating, comment FROM assistant_messages_ratings WHERE durable_object_name = ${durable_object_name}`;
-		return Object.fromEntries(ratings.map((row) => [row.message_id, {
-			rating: row.rating,
-			comment: row.comment
-		}]));
+		const feedback = sql`SELECT message_id, rating, comment, created_at, updated_at FROM assistant_messages_feedback`;
+		return Object.fromEntries(feedback.map((row) => [row.message_id, row]));
 	} catch {
 		return {};
 	}
@@ -530,7 +524,7 @@ var ChatAgent = class extends Agent {
 	};
 	async onStart() {
 		await super.onStart();
-		ensureRatingsTableExists(this.sql.bind(this));
+		ensureFeedbackTableExists(this.sql.bind(this));
 	}
 	configureSession(session) {
 		return super.configureSession(session).onCompaction(createCompactFn(this.getModel())).compactAfter(COMPACTION_TOKEN_THRESHOLD);
@@ -540,20 +534,20 @@ var ChatAgent = class extends Agent {
 		if (parent?.recordChatTurn) await parent.recordChatTurn(this.name, this.messages);
 	}
 	/**
-	* Rate a message by its id.
-	* @param messageId - The id of the message to rate.
+	* Submit feedback for a message by its id.
+	* @param messageId - The id of the message to give feedback on.
 	* @param rating - The rating to give the message. 1 = thumbs up, -1 = thumbs down.
 	* @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);
+	@callable({ description: "Submit feedback for a message by its id" }) async submitMessageFeedback(messageId, rating, comment) {
+		return submitMessageFeedback(this.sql.bind(this), messageId, rating, comment);
 	}
 	/**
-	* Returns all message ratings for the current chat.
-	* @returns All message ratings for the current chat.
+	* Returns all message feedback for the current chat.
+	* @returns All message feedback for the current chat.
 	*/
-	@callable({ description: "Returns all message ratings for the current chat" }) async getMessageRatings() {
-		return getMessageRatings(this.sql.bind(this), this.name);
+	@callable({ description: "Returns all message feedback for the current chat" }) async getMessageFeedback() {
+		return getMessageFeedback(this.sql.bind(this));
 	}
 };
 //#endregion

package/package.json

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