@dudousxd/nestjs-agent-store-drizzle 0.13.0 → 0.15.0

package/README.md

@@ -119,6 +119,7 @@ ALTER TABLE agent_run ADD COLUMN parent_run_id TEXT;
 ALTER TABLE agent_message ADD COLUMN reasoning TEXT;
 ALTER TABLE agent_message ADD COLUMN reasoning_ms INTEGER;
 ALTER TABLE agent_message ADD COLUMN ui TEXT;
+ALTER TABLE agent_message ADD COLUMN feedback TEXT;
 ALTER TABLE agent_tool_call ADD COLUMN approver TEXT;
 ALTER TABLE agent_tool_call ADD COLUMN expires_at INTEGER;
 ALTER TABLE agent_tool_call ADD COLUMN remember INTEGER;

package/dist/index.cjs

@@ -97,6 +97,10 @@ var agentMessage = (0, import_sqlite_core.sqliteTable)("agent_message", {
   ui: (0, import_sqlite_core.text)("ui", {
     mode: "json"
   }).$type(),
+  /** The thread owner's thumbs-up/down (+ comment); `null` when unrated. Not copied on fork. */
+  feedback: (0, import_sqlite_core.text)("feedback", {
+    mode: "json"
+  }).$type(),
   createdAt: (0, import_sqlite_core.integer)("created_at", {
     mode: "timestamp_ms"
   }).notNull()
@@ -308,6 +312,7 @@ var statements = [
     reasoning TEXT,
     reasoning_ms INTEGER,
     ui TEXT,
+    feedback TEXT,
     created_at INTEGER NOT NULL
   )`,
   `CREATE INDEX IF NOT EXISTS agent_message_thread_created_idx
@@ -447,6 +452,11 @@ var additiveColumns = [
     column: "ui",
     ddl: "ALTER TABLE agent_message ADD COLUMN ui TEXT"
   },
+  {
+    table: "agent_message",
+    column: "feedback",
+    ddl: "ALTER TABLE agent_message ADD COLUMN feedback TEXT"
+  },
   {
     table: "agent_tool_call",
     column: "approver",
@@ -850,6 +860,7 @@ var DrizzleAgentStore = class {
       reasoning: input.reasoning ?? null,
       reasoningMs: input.reasoningMs ?? null,
       ui: input.ui ?? null,
+      feedback: null,
       createdAt: now
     };
     await this.db.insert(agentMessage).values(message);
@@ -863,6 +874,22 @@ var DrizzleAgentStore = class {
       toolResults: results
     }).where((0, import_drizzle_orm2.eq)(agentMessage.id, messageId));
   }
+  async setMessageUi(messageId, ui) {
+    await this.db.update(agentMessage).set({
+      ui: ui.length > 0 ? ui : null
+    }).where((0, import_drizzle_orm2.eq)(agentMessage.id, messageId));
+  }
+  async threadOfMessage(messageId) {
+    const [row] = await this.db.select({
+      threadId: agentMessage.threadId
+    }).from(agentMessage).where((0, import_drizzle_orm2.eq)(agentMessage.id, messageId)).limit(1);
+    return row?.threadId ?? null;
+  }
+  async setMessageFeedback(messageId, feedback) {
+    await this.db.update(agentMessage).set({
+      feedback
+    }).where((0, import_drizzle_orm2.eq)(agentMessage.id, messageId));
+  }
   async truncateFrom(threadId, messageId) {
     const messages = await this.db.select().from(agentMessage).where((0, import_drizzle_orm2.eq)(agentMessage.threadId, threadId)).orderBy((0, import_drizzle_orm2.asc)(agentMessage.createdAt), (0, import_drizzle_orm2.asc)(agentMessage.id));
     const cutoff = messages.findIndex((message) => message.id === messageId);
@@ -1029,6 +1056,9 @@ var DrizzleAgentStore = class {
       } : {},
       ...message.ui != null ? {
         ui: message.ui
+      } : {},
+      ...message.feedback != null ? {
+        feedback: message.feedback
       } : {}
     };
   }