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

package/dist/index.d.cts

@@ -1,6 +1,6 @@
 import * as drizzle_orm_sqlite_core from 'drizzle-orm/sqlite-core';
 import { BaseSQLiteDatabase } from 'drizzle-orm/sqlite-core';
-import { MessageRole, ToolCallRequest, ToolResult, MessageAttachment, MessageUsage, AgentUiComponent, ToolKind, ToolCallStatus, UsagePurpose, AgentStore, ThreadTurnReader, CreateThreadInput, ThreadSummary, ThreadDetail, ThreadTurnQuery, ThreadTurnPage, ToolCallApprovalState, UpdateThreadInput, RecordRunStartInput, AppendMessageInput, StoredMessage, RecordToolCallInput, UpdateToolCallInput, RecordUsageInput, AgentGovernanceQueries, AgentPricingStore, GovernanceRange, ModelSpendRow, ActorSpendRow, ThreadSpendRow, UsageTrendPoint, ToolCallActivityRow, ThreadActivityRow, RunMetrics, RunAgentBreakdownRow, RunErrorBreakdownRow, RunTrendPoint, RecentRunRow, PendingApprovalRow, ToolStatRow, GovernancePageQuery, ToolCallWhere, GovernancePage, ThreadWhere, RunWhere, ApprovalWhere, GovernanceRunDetail, GovernanceThreadDetailQuery, GovernanceThreadDetail, MemoryProvider, ListMemoriesInput, MemoryRecord, ForgetMemoryInput, StoreMemoryInput, ModelPriceInput, CurrentModelPrice } from '@dudousxd/nestjs-agent-core';
+import { MessageRole, ToolCallRequest, ToolResult, MessageAttachment, MessageUsage, AgentUiComponent, MessageFeedback, ToolKind, ToolCallStatus, UsagePurpose, AgentStore, ThreadTurnReader, CreateThreadInput, ThreadSummary, ThreadDetail, ThreadTurnQuery, ThreadTurnPage, ToolCallApprovalState, UpdateThreadInput, RecordRunStartInput, AppendMessageInput, StoredMessage, RecordToolCallInput, UpdateToolCallInput, RecordUsageInput, AgentGovernanceQueries, AgentPricingStore, GovernanceRange, ModelSpendRow, ActorSpendRow, ThreadSpendRow, UsageTrendPoint, ToolCallActivityRow, ThreadActivityRow, RunMetrics, RunAgentBreakdownRow, RunErrorBreakdownRow, RunTrendPoint, RecentRunRow, PendingApprovalRow, ToolStatRow, GovernancePageQuery, ToolCallWhere, GovernancePage, ThreadWhere, RunWhere, ApprovalWhere, GovernanceRunDetail, GovernanceThreadDetailQuery, GovernanceThreadDetail, MemoryProvider, ListMemoriesInput, MemoryRecord, ForgetMemoryInput, StoreMemoryInput, ModelPriceInput, CurrentModelPrice } from '@dudousxd/nestjs-agent-core';
 export { ThreadTurnPage, ThreadTurnQuery } from '@dudousxd/nestjs-agent-core';
 import { TablesRelationalConfig, SQL } from 'drizzle-orm';
 import { OnModuleInit, OnModuleDestroy, Type, DynamicModule } from '@nestjs/common';
@@ -481,6 +481,25 @@ declare const agentMessage: drizzle_orm_sqlite_core.SQLiteTableWithColumns<{
         }, {}, {
             $type: AgentUiComponent[];
         }>;
+        feedback: drizzle_orm_sqlite_core.SQLiteColumn<{
+            name: "feedback";
+            tableName: "agent_message";
+            dataType: "json";
+            columnType: "SQLiteTextJson";
+            data: MessageFeedback;
+            driverParam: string;
+            notNull: false;
+            hasDefault: false;
+            isPrimaryKey: false;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: undefined;
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}, {
+            $type: MessageFeedback;
+        }>;
         createdAt: drizzle_orm_sqlite_core.SQLiteColumn<{
             name: "created_at";
             tableName: "agent_message";
@@ -2413,6 +2432,25 @@ declare const agentSchema: {
             }, {}, {
                 $type: AgentUiComponent[];
             }>;
+            feedback: drizzle_orm_sqlite_core.SQLiteColumn<{
+                name: "feedback";
+                tableName: "agent_message";
+                dataType: "json";
+                columnType: "SQLiteTextJson";
+                data: MessageFeedback;
+                driverParam: string;
+                notNull: false;
+                hasDefault: false;
+                isPrimaryKey: false;
+                isAutoincrement: false;
+                hasRuntimeDefault: false;
+                enumValues: undefined;
+                baseColumn: never;
+                identity: undefined;
+                generated: undefined;
+            }, {}, {
+                $type: MessageFeedback;
+            }>;
             createdAt: drizzle_orm_sqlite_core.SQLiteColumn<{
                 name: "created_at";
                 tableName: "agent_message";
@@ -3934,6 +3972,8 @@ declare class DrizzleAgentStore implements AgentStore, ThreadTurnReader {
     appendMessage(input: AppendMessageInput): Promise<StoredMessage>;
     setMessageToolResults(messageId: string, results: ToolResult[]): Promise<void>;
     setMessageUi(messageId: string, ui: AgentUiComponent[]): Promise<void>;
+    threadOfMessage(messageId: string): Promise<string | null>;
+    setMessageFeedback(messageId: string, feedback: MessageFeedback | null): Promise<void>;
     truncateFrom(threadId: string, messageId: string): Promise<void>;
     recordToolCall(input: RecordToolCallInput): Promise<void>;
     updateToolCall(input: UpdateToolCallInput): Promise<void>;

package/dist/index.d.ts

@@ -1,6 +1,6 @@
 import * as drizzle_orm_sqlite_core from 'drizzle-orm/sqlite-core';
 import { BaseSQLiteDatabase } from 'drizzle-orm/sqlite-core';
-import { MessageRole, ToolCallRequest, ToolResult, MessageAttachment, MessageUsage, AgentUiComponent, ToolKind, ToolCallStatus, UsagePurpose, AgentStore, ThreadTurnReader, CreateThreadInput, ThreadSummary, ThreadDetail, ThreadTurnQuery, ThreadTurnPage, ToolCallApprovalState, UpdateThreadInput, RecordRunStartInput, AppendMessageInput, StoredMessage, RecordToolCallInput, UpdateToolCallInput, RecordUsageInput, AgentGovernanceQueries, AgentPricingStore, GovernanceRange, ModelSpendRow, ActorSpendRow, ThreadSpendRow, UsageTrendPoint, ToolCallActivityRow, ThreadActivityRow, RunMetrics, RunAgentBreakdownRow, RunErrorBreakdownRow, RunTrendPoint, RecentRunRow, PendingApprovalRow, ToolStatRow, GovernancePageQuery, ToolCallWhere, GovernancePage, ThreadWhere, RunWhere, ApprovalWhere, GovernanceRunDetail, GovernanceThreadDetailQuery, GovernanceThreadDetail, MemoryProvider, ListMemoriesInput, MemoryRecord, ForgetMemoryInput, StoreMemoryInput, ModelPriceInput, CurrentModelPrice } from '@dudousxd/nestjs-agent-core';
+import { MessageRole, ToolCallRequest, ToolResult, MessageAttachment, MessageUsage, AgentUiComponent, MessageFeedback, ToolKind, ToolCallStatus, UsagePurpose, AgentStore, ThreadTurnReader, CreateThreadInput, ThreadSummary, ThreadDetail, ThreadTurnQuery, ThreadTurnPage, ToolCallApprovalState, UpdateThreadInput, RecordRunStartInput, AppendMessageInput, StoredMessage, RecordToolCallInput, UpdateToolCallInput, RecordUsageInput, AgentGovernanceQueries, AgentPricingStore, GovernanceRange, ModelSpendRow, ActorSpendRow, ThreadSpendRow, UsageTrendPoint, ToolCallActivityRow, ThreadActivityRow, RunMetrics, RunAgentBreakdownRow, RunErrorBreakdownRow, RunTrendPoint, RecentRunRow, PendingApprovalRow, ToolStatRow, GovernancePageQuery, ToolCallWhere, GovernancePage, ThreadWhere, RunWhere, ApprovalWhere, GovernanceRunDetail, GovernanceThreadDetailQuery, GovernanceThreadDetail, MemoryProvider, ListMemoriesInput, MemoryRecord, ForgetMemoryInput, StoreMemoryInput, ModelPriceInput, CurrentModelPrice } from '@dudousxd/nestjs-agent-core';
 export { ThreadTurnPage, ThreadTurnQuery } from '@dudousxd/nestjs-agent-core';
 import { TablesRelationalConfig, SQL } from 'drizzle-orm';
 import { OnModuleInit, OnModuleDestroy, Type, DynamicModule } from '@nestjs/common';
@@ -481,6 +481,25 @@ declare const agentMessage: drizzle_orm_sqlite_core.SQLiteTableWithColumns<{
         }, {}, {
             $type: AgentUiComponent[];
         }>;
+        feedback: drizzle_orm_sqlite_core.SQLiteColumn<{
+            name: "feedback";
+            tableName: "agent_message";
+            dataType: "json";
+            columnType: "SQLiteTextJson";
+            data: MessageFeedback;
+            driverParam: string;
+            notNull: false;
+            hasDefault: false;
+            isPrimaryKey: false;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: undefined;
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}, {
+            $type: MessageFeedback;
+        }>;
         createdAt: drizzle_orm_sqlite_core.SQLiteColumn<{
             name: "created_at";
             tableName: "agent_message";
@@ -2413,6 +2432,25 @@ declare const agentSchema: {
             }, {}, {
                 $type: AgentUiComponent[];
             }>;
+            feedback: drizzle_orm_sqlite_core.SQLiteColumn<{
+                name: "feedback";
+                tableName: "agent_message";
+                dataType: "json";
+                columnType: "SQLiteTextJson";
+                data: MessageFeedback;
+                driverParam: string;
+                notNull: false;
+                hasDefault: false;
+                isPrimaryKey: false;
+                isAutoincrement: false;
+                hasRuntimeDefault: false;
+                enumValues: undefined;
+                baseColumn: never;
+                identity: undefined;
+                generated: undefined;
+            }, {}, {
+                $type: MessageFeedback;
+            }>;
             createdAt: drizzle_orm_sqlite_core.SQLiteColumn<{
                 name: "created_at";
                 tableName: "agent_message";
@@ -3934,6 +3972,8 @@ declare class DrizzleAgentStore implements AgentStore, ThreadTurnReader {
     appendMessage(input: AppendMessageInput): Promise<StoredMessage>;
     setMessageToolResults(messageId: string, results: ToolResult[]): Promise<void>;
     setMessageUi(messageId: string, ui: AgentUiComponent[]): Promise<void>;
+    threadOfMessage(messageId: string): Promise<string | null>;
+    setMessageFeedback(messageId: string, feedback: MessageFeedback | null): Promise<void>;
     truncateFrom(threadId: string, messageId: string): Promise<void>;
     recordToolCall(input: RecordToolCallInput): Promise<void>;
     updateToolCall(input: UpdateToolCallInput): Promise<void>;

package/dist/index.js

@@ -57,6 +57,10 @@ var agentMessage = sqliteTable("agent_message", {
   ui: text("ui", {
     mode: "json"
   }).$type(),
+  /** The thread owner's thumbs-up/down (+ comment); `null` when unrated. Not copied on fork. */
+  feedback: text("feedback", {
+    mode: "json"
+  }).$type(),
   createdAt: integer("created_at", {
     mode: "timestamp_ms"
   }).notNull()
@@ -268,6 +272,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
@@ -407,6 +412,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",
@@ -810,6 +820,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);
@@ -828,6 +839,17 @@ var DrizzleAgentStore = class {
       ui: ui.length > 0 ? ui : null
     }).where(eq(agentMessage.id, messageId));
   }
+  async threadOfMessage(messageId) {
+    const [row] = await this.db.select({
+      threadId: agentMessage.threadId
+    }).from(agentMessage).where(eq(agentMessage.id, messageId)).limit(1);
+    return row?.threadId ?? null;
+  }
+  async setMessageFeedback(messageId, feedback) {
+    await this.db.update(agentMessage).set({
+      feedback
+    }).where(eq(agentMessage.id, messageId));
+  }
   async truncateFrom(threadId, messageId) {
     const messages = await this.db.select().from(agentMessage).where(eq(agentMessage.threadId, threadId)).orderBy(asc(agentMessage.createdAt), asc(agentMessage.id));
     const cutoff = messages.findIndex((message) => message.id === messageId);
@@ -994,6 +1016,9 @@ var DrizzleAgentStore = class {
       } : {},
       ...message.ui != null ? {
         ui: message.ui
+      } : {},
+      ...message.feedback != null ? {
+        feedback: message.feedback
       } : {}
     };
   }