@dudousxd/nestjs-agent-store-drizzle 0.14.0 → 0.16.0

package/README.md

@@ -119,6 +119,8 @@ 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_thread ADD COLUMN model 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

@@ -54,6 +54,8 @@ var agentThread = (0, import_sqlite_core.sqliteTable)("agent_thread", {
   activeStreamId: (0, import_sqlite_core.text)("active_stream_id"),
   /** The agent a new turn on this thread defaults to when the caller names none. */
   defaultAgent: (0, import_sqlite_core.text)("default_agent"),
+  /** The model every turn on this thread runs on unless the send names one; `null` → default. */
+  model: (0, import_sqlite_core.text)("model"),
   createdAt: (0, import_sqlite_core.integer)("created_at", {
     mode: "timestamp_ms"
   }).notNull(),
@@ -97,6 +99,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()
@@ -287,6 +293,7 @@ var statements = [
     transient INTEGER NOT NULL DEFAULT 0,
     active_stream_id TEXT,
     default_agent TEXT,
+    model TEXT,
     created_at INTEGER NOT NULL,
     updated_at INTEGER NOT NULL,
     deleted_at INTEGER
@@ -308,6 +315,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 +455,16 @@ 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_thread",
+    column: "model",
+    ddl: "ALTER TABLE agent_thread ADD COLUMN model TEXT"
+  },
   {
     table: "agent_tool_call",
     column: "approver",
@@ -512,6 +530,7 @@ var DrizzleAgentStore = class {
       transient: input.transient ?? false,
       activeStreamId: null,
       defaultAgent: null,
+      model: null,
       createdAt: now,
       updatedAt: now,
       deletedAt: null
@@ -617,6 +636,7 @@ var DrizzleAgentStore = class {
       transient: false,
       activeStreamId: null,
       defaultAgent: source.defaultAgent,
+      model: source.model,
       createdAt: now,
       updatedAt: now,
       deletedAt: null
@@ -765,6 +785,9 @@ var DrizzleAgentStore = class {
     if (patch.defaultAgent !== void 0) {
       updates.defaultAgent = patch.defaultAgent;
     }
+    if (patch.model !== void 0) {
+      updates.model = patch.model;
+    }
     if (Object.keys(updates).length === 0) {
       return;
     }
@@ -776,6 +799,13 @@ var DrizzleAgentStore = class {
   * agent answers the next turn, and {@link getThread} would materialize the whole transcript to
   * hand it over. `null` when the thread is unknown, soft-deleted, or has no default set.
   */
+  /** The thread's pinned model, projected like {@link defaultAgentForThread}. */
+  async modelForThread(threadId) {
+    const [thread] = await this.db.select({
+      model: agentThread.model
+    }).from(agentThread).where((0, import_drizzle_orm2.and)((0, import_drizzle_orm2.eq)(agentThread.id, threadId), (0, import_drizzle_orm2.isNull)(agentThread.deletedAt)));
+    return thread?.model ?? null;
+  }
   async defaultAgentForThread(threadId) {
     const [thread] = await this.db.select({
       defaultAgent: agentThread.defaultAgent
@@ -850,6 +880,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);
@@ -868,6 +899,17 @@ var DrizzleAgentStore = class {
       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);
@@ -994,6 +1036,9 @@ var DrizzleAgentStore = class {
       createdAt: thread.createdAt.toISOString(),
       updatedAt: thread.updatedAt.toISOString(),
       defaultAgent: thread.defaultAgent,
+      ...thread.model != null ? {
+        model: thread.model
+      } : {},
       ...lastContent !== void 0 ? {
         lastMessagePreview: lastContent.slice(0, 120)
       } : {}
@@ -1034,6 +1079,9 @@ var DrizzleAgentStore = class {
       } : {},
       ...message.ui != null ? {
         ui: message.ui
+      } : {},
+      ...message.feedback != null ? {
+        feedback: message.feedback
       } : {}
     };
   }