@economic/agents 1.5.0 → 1.6.1

package/dist/index.d.mts

@@ -134,6 +134,12 @@ declare abstract class Agent<Env extends Cloudflare.Env = Cloudflare.Env> extend
   }): Promise<LLMParams>;
 }
 //#endregion
+//#region src/server/agent-chat/features/conversations/rating.d.ts
+interface MessageRating {
+  rating: number;
+  comment?: string;
+}
+//#endregion
 //#region src/server/agent-chat/ChatAgent.d.ts
 /**
  * Chat agent for Cloudflare Agents SDK: lazy skill loading, audit logging,
@@ -227,8 +233,8 @@ declare abstract class ChatAgent<Env extends Cloudflare.Env = Cloudflare.Env> ex
     _deleteStaleRows?: boolean;
   }): Promise<void>;
   protected onChatResponse(result: ChatResponseResult): Promise<void>;
-  rateMessage(messageId: string, rating: number): Promise<void>;
-  getMessageRatings(): Promise<Record<string, number>>;
+  rateMessage(messageId: string, rating: number, comment?: string): Promise<void>;
+  getMessageRatings(): Promise<Record<string, MessageRating>>;
   getConversations(): Promise<Record<string, unknown>[]>;
   deleteConversation(id: string): Promise<boolean>;
   destroy(): Promise<void>;

package/dist/index.mjs

@@ -799,17 +799,21 @@ function filterEphemeralMessages(messages) {
 }
 //#endregion
 //#region src/server/agent-chat/features/conversations/rating.ts
-async function rateMessage(db, durableObjectName, messageId, rating) {
+async function rateMessage(db, durableObjectName, messageId, rating, comment) {
 	const now = (/* @__PURE__ */ new Date()).toISOString();
-	await db.prepare(`INSERT INTO message_ratings (message_id, durable_object_name, rating, created_at, updated_at)
-      VALUES (?, ?, ?, ?, ?)
+	await db.prepare(`INSERT INTO message_ratings (message_id, durable_object_name, rating, comment, created_at, updated_at)
+      VALUES (?, ?, ?, ?, ?, ?)
       ON CONFLICT (message_id, durable_object_name) DO UPDATE SET
         rating = excluded.rating,
-        updated_at = excluded.updated_at`).bind(messageId, durableObjectName, rating, now, now).run();
+        comment = excluded.comment,
+        updated_at = excluded.updated_at`).bind(messageId, durableObjectName, rating, comment ?? null, now, now).run();
 }
 async function getMessageRatings(db, durableObjectName) {
-	const result = await db.prepare(`SELECT message_id, rating FROM message_ratings WHERE durable_object_name = ?`).bind(durableObjectName).all();
-	return Object.fromEntries(result.results.map((row) => [row.message_id, row.rating]));
+	const result = await db.prepare(`SELECT message_id, rating, comment FROM message_ratings WHERE durable_object_name = ?`).bind(durableObjectName).all();
+	return Object.fromEntries(result.results.map((row) => [row.message_id, {
+		rating: row.rating,
+		...row.comment != null ? { comment: row.comment } : {}
+	}]));
 }
 //#endregion
 //#region src/server/agent-chat/ChatAgent.ts
@@ -944,8 +948,8 @@ var ChatAgent = class extends AIChatAgent {
 		recordConversationSummary(this.env.AGENT_DB, this.name, this.messages, this.getFastModel());
 		this.scheduleConversationForAutoDeletion();
 	}
-	@callable({ description: "Rate a message by its id" }) async rateMessage(messageId, rating) {
-		await rateMessage(this.env.AGENT_DB, this.name, messageId, rating);
+	@callable({ description: "Rate a message by its id" }) async rateMessage(messageId, rating, comment) {
+		await rateMessage(this.env.AGENT_DB, this.name, messageId, rating, comment);
 	}
 	@callable({ description: "Returns all message ratings for the current conversation" }) async getMessageRatings() {
 		return getMessageRatings(this.env.AGENT_DB, this.name);
@@ -1032,7 +1036,7 @@ var ChatAgentHarness = class extends ChatAgent {
 			system: this.getSystemPrompt(ctx),
 			skills: this.getSkills(ctx),
 			tools: this.getTools(ctx)
-		})).toUIMessageStreamResponse();
+		})).toUIMessageStreamResponse({ sendSources: true });
 	}
 };
 //#endregion

package/package.json

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

package/schema/chat/0002_add_rating_comment.sql

@@ -0,0 +1 @@
+ALTER TABLE message_ratings ADD COLUMN comment TEXT;