@dudousxd/nestjs-agent-store-drizzle 0.14.0 → 0.16.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';
@@ -153,6 +153,25 @@ declare const agentThread: drizzle_orm_sqlite_core.SQLiteTableWithColumns<{
         }, {}, {
             length: number | undefined;
         }>;
+        model: drizzle_orm_sqlite_core.SQLiteColumn<{
+            name: "model";
+            tableName: "agent_thread";
+            dataType: "string";
+            columnType: "SQLiteText";
+            data: string;
+            driverParam: string;
+            notNull: false;
+            hasDefault: false;
+            isPrimaryKey: false;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: [string, ...string[]];
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}, {
+            length: number | undefined;
+        }>;
         createdAt: drizzle_orm_sqlite_core.SQLiteColumn<{
             name: "created_at";
             tableName: "agent_thread";
@@ -481,6 +500,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";
@@ -2090,6 +2128,25 @@ declare const agentSchema: {
             }, {}, {
                 length: number | undefined;
             }>;
+            model: drizzle_orm_sqlite_core.SQLiteColumn<{
+                name: "model";
+                tableName: "agent_thread";
+                dataType: "string";
+                columnType: "SQLiteText";
+                data: string;
+                driverParam: string;
+                notNull: false;
+                hasDefault: false;
+                isPrimaryKey: false;
+                isAutoincrement: false;
+                hasRuntimeDefault: false;
+                enumValues: [string, ...string[]];
+                baseColumn: never;
+                identity: undefined;
+                generated: undefined;
+            }, {}, {
+                length: number | undefined;
+            }>;
             createdAt: drizzle_orm_sqlite_core.SQLiteColumn<{
                 name: "created_at";
                 tableName: "agent_thread";
@@ -2413,6 +2470,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";
@@ -3913,6 +3989,8 @@ declare class DrizzleAgentStore implements AgentStore, ThreadTurnReader {
      * 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}. */
+    modelForThread(threadId: string): Promise<string | null>;
     defaultAgentForThread(threadId: string): Promise<string | null>;
     /**
      * Persist the start of a run (turn). Replay-safe: called under a durable localStep.
@@ -3934,6 +4012,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';
@@ -153,6 +153,25 @@ declare const agentThread: drizzle_orm_sqlite_core.SQLiteTableWithColumns<{
         }, {}, {
             length: number | undefined;
         }>;
+        model: drizzle_orm_sqlite_core.SQLiteColumn<{
+            name: "model";
+            tableName: "agent_thread";
+            dataType: "string";
+            columnType: "SQLiteText";
+            data: string;
+            driverParam: string;
+            notNull: false;
+            hasDefault: false;
+            isPrimaryKey: false;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: [string, ...string[]];
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}, {
+            length: number | undefined;
+        }>;
         createdAt: drizzle_orm_sqlite_core.SQLiteColumn<{
             name: "created_at";
             tableName: "agent_thread";
@@ -481,6 +500,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";
@@ -2090,6 +2128,25 @@ declare const agentSchema: {
             }, {}, {
                 length: number | undefined;
             }>;
+            model: drizzle_orm_sqlite_core.SQLiteColumn<{
+                name: "model";
+                tableName: "agent_thread";
+                dataType: "string";
+                columnType: "SQLiteText";
+                data: string;
+                driverParam: string;
+                notNull: false;
+                hasDefault: false;
+                isPrimaryKey: false;
+                isAutoincrement: false;
+                hasRuntimeDefault: false;
+                enumValues: [string, ...string[]];
+                baseColumn: never;
+                identity: undefined;
+                generated: undefined;
+            }, {}, {
+                length: number | undefined;
+            }>;
             createdAt: drizzle_orm_sqlite_core.SQLiteColumn<{
                 name: "created_at";
                 tableName: "agent_thread";
@@ -2413,6 +2470,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";
@@ -3913,6 +3989,8 @@ declare class DrizzleAgentStore implements AgentStore, ThreadTurnReader {
      * 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}. */
+    modelForThread(threadId: string): Promise<string | null>;
     defaultAgentForThread(threadId: string): Promise<string | null>;
     /**
      * Persist the start of a run (turn). Replay-safe: called under a durable localStep.
@@ -3934,6 +4012,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

@@ -14,6 +14,8 @@ var agentThread = sqliteTable("agent_thread", {
   activeStreamId: text("active_stream_id"),
   /** The agent a new turn on this thread defaults to when the caller names none. */
   defaultAgent: text("default_agent"),
+  /** The model every turn on this thread runs on unless the send names one; `null` → default. */
+  model: text("model"),
   createdAt: integer("created_at", {
     mode: "timestamp_ms"
   }).notNull(),
@@ -57,6 +59,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()
@@ -247,6 +253,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
@@ -268,6 +275,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 +415,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",
@@ -472,6 +490,7 @@ var DrizzleAgentStore = class {
       transient: input.transient ?? false,
       activeStreamId: null,
       defaultAgent: null,
+      model: null,
       createdAt: now,
       updatedAt: now,
       deletedAt: null
@@ -577,6 +596,7 @@ var DrizzleAgentStore = class {
       transient: false,
       activeStreamId: null,
       defaultAgent: source.defaultAgent,
+      model: source.model,
       createdAt: now,
       updatedAt: now,
       deletedAt: null
@@ -725,6 +745,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;
     }
@@ -736,6 +759,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(and(eq(agentThread.id, threadId), isNull(agentThread.deletedAt)));
+    return thread?.model ?? null;
+  }
   async defaultAgentForThread(threadId) {
     const [thread] = await this.db.select({
       defaultAgent: agentThread.defaultAgent
@@ -810,6 +840,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 +859,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);
@@ -954,6 +996,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)
       } : {}
@@ -994,6 +1039,9 @@ var DrizzleAgentStore = class {
       } : {},
       ...message.ui != null ? {
         ui: message.ui
+      } : {},
+      ...message.feedback != null ? {
+        feedback: message.feedback
       } : {}
     };
   }