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

package/dist/index.d.cts

@@ -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";
@@ -2109,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";
@@ -3951,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.

package/dist/index.d.ts

@@ -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";
@@ -2109,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";
@@ -3951,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.

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(),
@@ -251,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
@@ -417,6 +420,11 @@ var additiveColumns = [
     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",
@@ -482,6 +490,7 @@ var DrizzleAgentStore = class {
       transient: input.transient ?? false,
       activeStreamId: null,
       defaultAgent: null,
+      model: null,
       createdAt: now,
       updatedAt: now,
       deletedAt: null
@@ -587,6 +596,7 @@ var DrizzleAgentStore = class {
       transient: false,
       activeStreamId: null,
       defaultAgent: source.defaultAgent,
+      model: source.model,
       createdAt: now,
       updatedAt: now,
       deletedAt: null
@@ -735,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;
     }
@@ -746,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
@@ -976,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)
       } : {}