@dudousxd/nestjs-agent-store-drizzle 0.22.0 → 0.23.0

package/dist/index.d.cts

@@ -172,6 +172,25 @@ declare const agentThread: drizzle_orm_sqlite_core.SQLiteTableWithColumns<{
         }, {}, {
             length: number | undefined;
         }>;
+        persona: drizzle_orm_sqlite_core.SQLiteColumn<{
+            name: "persona";
+            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;
+        }>;
         queuePause: drizzle_orm_sqlite_core.SQLiteColumn<{
             name: "queue_pause";
             tableName: "agent_thread";
@@ -445,6 +464,25 @@ declare const agentMessage: drizzle_orm_sqlite_core.SQLiteTableWithColumns<{
         }, {}, {
             length: number | undefined;
         }>;
+        persona: drizzle_orm_sqlite_core.SQLiteColumn<{
+            name: "persona";
+            tableName: "agent_message";
+            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;
+        }>;
         runId: drizzle_orm_sqlite_core.SQLiteColumn<{
             name: "run_id";
             tableName: "agent_message";
@@ -1007,6 +1045,25 @@ declare const agentQueuedMessage: drizzle_orm_sqlite_core.SQLiteTableWithColumns
         }, {}, {
             length: number | undefined;
         }>;
+        persona: drizzle_orm_sqlite_core.SQLiteColumn<{
+            name: "persona";
+            tableName: "agent_queued_message";
+            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;
+        }>;
         model: drizzle_orm_sqlite_core.SQLiteColumn<{
             name: "model";
             tableName: "agent_queued_message";
@@ -2606,6 +2663,25 @@ declare const agentSchema: {
             }, {}, {
                 length: number | undefined;
             }>;
+            persona: drizzle_orm_sqlite_core.SQLiteColumn<{
+                name: "persona";
+                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;
+            }>;
             queuePause: drizzle_orm_sqlite_core.SQLiteColumn<{
                 name: "queue_pause";
                 tableName: "agent_thread";
@@ -2874,6 +2950,25 @@ declare const agentSchema: {
             }, {}, {
                 length: number | undefined;
             }>;
+            persona: drizzle_orm_sqlite_core.SQLiteColumn<{
+                name: "persona";
+                tableName: "agent_message";
+                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;
+            }>;
             runId: drizzle_orm_sqlite_core.SQLiteColumn<{
                 name: "run_id";
                 tableName: "agent_message";
@@ -3105,6 +3200,25 @@ declare const agentSchema: {
             }, {}, {
                 length: number | undefined;
             }>;
+            persona: drizzle_orm_sqlite_core.SQLiteColumn<{
+                name: "persona";
+                tableName: "agent_queued_message";
+                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;
+            }>;
             model: drizzle_orm_sqlite_core.SQLiteColumn<{
                 name: "model";
                 tableName: "agent_queued_message";
@@ -4816,6 +4930,8 @@ declare class DrizzleAgentStore implements AgentStore, ThreadTurnReader, ChatQue
      */
     /** The thread's pinned model, projected like {@link defaultAgentForThread}. */
     modelForThread(threadId: string): Promise<string | null>;
+    /** The thread's pinned persona, projected like {@link defaultAgentForThread}. */
+    personaForThread(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.
@@ -4845,7 +4961,8 @@ declare class DrizzleAgentStore implements AgentStore, ThreadTurnReader, ChatQue
     toolCallOutcomes(toolCallIds: readonly string[]): Promise<ToolCallOutcome[]>;
     failUnsettledToolCalls(runId: string, error: string): Promise<number>;
     /**
-     * Of `mediaIds`, the ones a surviving message in one of this actor's threads still carries.
+     * Of `mediaIds`, the ones a surviving message — or a message waiting in the queue — in one of
+     * this actor's threads still carries.
      *
      * Reads the `attachments` JSON back out and matches in memory rather than pushing the match into
      * SQL: the column holds an array of objects, and every dialect this adapter's sibling targets

package/dist/index.d.ts

@@ -172,6 +172,25 @@ declare const agentThread: drizzle_orm_sqlite_core.SQLiteTableWithColumns<{
         }, {}, {
             length: number | undefined;
         }>;
+        persona: drizzle_orm_sqlite_core.SQLiteColumn<{
+            name: "persona";
+            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;
+        }>;
         queuePause: drizzle_orm_sqlite_core.SQLiteColumn<{
             name: "queue_pause";
             tableName: "agent_thread";
@@ -445,6 +464,25 @@ declare const agentMessage: drizzle_orm_sqlite_core.SQLiteTableWithColumns<{
         }, {}, {
             length: number | undefined;
         }>;
+        persona: drizzle_orm_sqlite_core.SQLiteColumn<{
+            name: "persona";
+            tableName: "agent_message";
+            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;
+        }>;
         runId: drizzle_orm_sqlite_core.SQLiteColumn<{
             name: "run_id";
             tableName: "agent_message";
@@ -1007,6 +1045,25 @@ declare const agentQueuedMessage: drizzle_orm_sqlite_core.SQLiteTableWithColumns
         }, {}, {
             length: number | undefined;
         }>;
+        persona: drizzle_orm_sqlite_core.SQLiteColumn<{
+            name: "persona";
+            tableName: "agent_queued_message";
+            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;
+        }>;
         model: drizzle_orm_sqlite_core.SQLiteColumn<{
             name: "model";
             tableName: "agent_queued_message";
@@ -2606,6 +2663,25 @@ declare const agentSchema: {
             }, {}, {
                 length: number | undefined;
             }>;
+            persona: drizzle_orm_sqlite_core.SQLiteColumn<{
+                name: "persona";
+                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;
+            }>;
             queuePause: drizzle_orm_sqlite_core.SQLiteColumn<{
                 name: "queue_pause";
                 tableName: "agent_thread";
@@ -2874,6 +2950,25 @@ declare const agentSchema: {
             }, {}, {
                 length: number | undefined;
             }>;
+            persona: drizzle_orm_sqlite_core.SQLiteColumn<{
+                name: "persona";
+                tableName: "agent_message";
+                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;
+            }>;
             runId: drizzle_orm_sqlite_core.SQLiteColumn<{
                 name: "run_id";
                 tableName: "agent_message";
@@ -3105,6 +3200,25 @@ declare const agentSchema: {
             }, {}, {
                 length: number | undefined;
             }>;
+            persona: drizzle_orm_sqlite_core.SQLiteColumn<{
+                name: "persona";
+                tableName: "agent_queued_message";
+                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;
+            }>;
             model: drizzle_orm_sqlite_core.SQLiteColumn<{
                 name: "model";
                 tableName: "agent_queued_message";
@@ -4816,6 +4930,8 @@ declare class DrizzleAgentStore implements AgentStore, ThreadTurnReader, ChatQue
      */
     /** The thread's pinned model, projected like {@link defaultAgentForThread}. */
     modelForThread(threadId: string): Promise<string | null>;
+    /** The thread's pinned persona, projected like {@link defaultAgentForThread}. */
+    personaForThread(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.
@@ -4845,7 +4961,8 @@ declare class DrizzleAgentStore implements AgentStore, ThreadTurnReader, ChatQue
     toolCallOutcomes(toolCallIds: readonly string[]): Promise<ToolCallOutcome[]>;
     failUnsettledToolCalls(runId: string, error: string): Promise<number>;
     /**
-     * Of `mediaIds`, the ones a surviving message in one of this actor's threads still carries.
+     * Of `mediaIds`, the ones a surviving message — or a message waiting in the queue — in one of
+     * this actor's threads still carries.
      *
      * Reads the `attachments` JSON back out and matches in memory rather than pushing the match into
      * SQL: the column holds an array of objects, and every dialect this adapter's sibling targets

package/dist/index.js

@@ -16,6 +16,8 @@ var agentThread = sqliteTable("agent_thread", {
   defaultAgent: text("default_agent"),
   /** The model every turn on this thread runs on unless the send names one; `null` → default. */
   model: text("model"),
+  /** The persona a send on this thread runs under when it names none; `null` → the agent's default. */
+  persona: text("persona"),
   /** Why the thread's message queue stopped draining; `null` → it drains. */
   queuePause: text("queue_pause", {
     mode: "json"
@@ -55,6 +57,8 @@ var agentMessage = sqliteTable("agent_message", {
     mode: "json"
   }).$type(),
   agentName: text("agent_name"),
+  /** The persona the turn that wrote this message ran under; `null` → none. */
+  persona: text("persona"),
   runId: text("run_id"),
   /** The step's streamed thinking; `null` when the model produced none. */
   reasoning: text("reasoning"),
@@ -127,6 +131,8 @@ var agentQueuedMessage = sqliteTable("agent_queued_message", {
     mode: "json"
   }).$type(),
   agentName: text("agent_name"),
+  /** The persona the send resolved — what the message starts under. */
+  persona: text("persona"),
   model: text("model"),
   pageContext: text("page_context", {
     mode: "json"
@@ -314,6 +320,7 @@ var statements = [
     active_stream_id TEXT,
     default_agent TEXT,
     model TEXT,
+    persona TEXT,
     queue_pause TEXT,
     created_at INTEGER NOT NULL,
     updated_at INTEGER NOT NULL,
@@ -332,6 +339,7 @@ var statements = [
     follow_ups TEXT,
     usage TEXT,
     agent_name TEXT,
+    persona TEXT,
     run_id TEXT,
     reasoning TEXT,
     reasoning_ms INTEGER,
@@ -348,6 +356,7 @@ var statements = [
     content TEXT NOT NULL,
     attachments TEXT,
     agent_name TEXT,
+    persona TEXT,
     model TEXT,
     page_context TEXT,
     interrupt INTEGER NOT NULL DEFAULT 0,
@@ -543,6 +552,21 @@ var additiveColumns = [
     table: "agent_tool_call",
     column: "decided_via",
     ddl: "ALTER TABLE agent_tool_call ADD COLUMN decided_via TEXT"
+  },
+  {
+    table: "agent_thread",
+    column: "persona",
+    ddl: "ALTER TABLE agent_thread ADD COLUMN persona TEXT"
+  },
+  {
+    table: "agent_message",
+    column: "persona",
+    ddl: "ALTER TABLE agent_message ADD COLUMN persona TEXT"
+  },
+  {
+    table: "agent_queued_message",
+    column: "persona",
+    ddl: "ALTER TABLE agent_queued_message ADD COLUMN persona TEXT"
   }
 ];
 async function ensureAgentSchema(db) {
@@ -590,6 +614,7 @@ var DrizzleAgentStore = class {
       activeStreamId: null,
       defaultAgent: null,
       model: null,
+      persona: input.persona ?? null,
       queuePause: null,
       createdAt: now,
       updatedAt: now,
@@ -697,6 +722,7 @@ var DrizzleAgentStore = class {
       activeStreamId: null,
       defaultAgent: source.defaultAgent,
       model: source.model,
+      persona: source.persona,
       // A fork is a new conversation: nothing is waiting on it, and nothing is paused.
       queuePause: null,
       createdAt: now,
@@ -716,6 +742,7 @@ var DrizzleAgentStore = class {
         followUps: message.followUps,
         usage: message.usage,
         agentName: message.agentName,
+        persona: message.persona,
         // The copy is the same message, so it keeps the run that wrote it. A reader asking which
         // turn produced this text gets the truthful answer; the run's own thread is still the
         // original, so a run-scoped read never picks the fork's rows up.
@@ -876,6 +903,7 @@ var DrizzleAgentStore = class {
       content: input.content,
       attachments: input.attachments !== void 0 && input.attachments.length > 0 ? input.attachments : null,
       agentName: input.agentName ?? null,
+      persona: input.persona ?? null,
       model: input.model ?? null,
       pageContext: input.pageContext ?? null,
       interrupt: input.interrupt === true,
@@ -971,6 +999,9 @@ var DrizzleAgentStore = class {
     if (patch.model !== void 0) {
       updates.model = patch.model;
     }
+    if (patch.persona !== void 0) {
+      updates.persona = patch.persona;
+    }
     if (Object.keys(updates).length === 0) {
       return;
     }
@@ -989,6 +1020,13 @@ var DrizzleAgentStore = class {
     }).from(agentThread).where(and(eq(agentThread.id, threadId), isNull(agentThread.deletedAt)));
     return thread?.model ?? null;
   }
+  /** The thread's pinned persona, projected like {@link defaultAgentForThread}. */
+  async personaForThread(threadId) {
+    const [thread] = await this.db.select({
+      persona: agentThread.persona
+    }).from(agentThread).where(and(eq(agentThread.id, threadId), isNull(agentThread.deletedAt)));
+    return thread?.persona ?? null;
+  }
   async defaultAgentForThread(threadId) {
     const [thread] = await this.db.select({
       defaultAgent: agentThread.defaultAgent
@@ -1059,6 +1097,7 @@ var DrizzleAgentStore = class {
       followUps: input.followUps ?? null,
       usage: input.usage ?? null,
       agentName: input.agentName ?? null,
+      persona: input.persona ?? null,
       runId: input.runId ?? null,
       reasoning: input.reasoning ?? null,
       reasoningMs: input.reasoningMs ?? null,
@@ -1189,7 +1228,8 @@ var DrizzleAgentStore = class {
     return rows.length;
   }
   /**
-  * Of `mediaIds`, the ones a surviving message in one of this actor's threads still carries.
+  * Of `mediaIds`, the ones a surviving message — or a message waiting in the queue — in one of
+  * this actor's threads still carries.
   *
   * Reads the `attachments` JSON back out and matches in memory rather than pushing the match into
   * SQL: the column holds an array of objects, and every dialect this adapter's sibling targets
@@ -1208,9 +1248,15 @@ var DrizzleAgentStore = class {
     const rows = await this.db.select({
       attachments: agentMessage.attachments
     }).from(agentMessage).innerJoin(agentThread, eq(agentMessage.threadId, agentThread.id)).where(and(eq(agentThread.actorRef, actorRef), isNotNull(agentMessage.attachments)));
+    const queued = await this.db.select({
+      attachments: agentQueuedMessage.attachments
+    }).from(agentQueuedMessage).innerJoin(agentThread, eq(agentQueuedMessage.threadId, agentThread.id)).where(and(eq(agentThread.actorRef, actorRef), isNotNull(agentQueuedMessage.attachments)));
     const wanted = new Set(mediaIds);
     const found = /* @__PURE__ */ new Set();
-    for (const row of rows) {
+    for (const row of [
+      ...rows,
+      ...queued
+    ]) {
       for (const attachment of row.attachments ?? []) {
         if (wanted.has(attachment.mediaId)) {
           found.add(attachment.mediaId);
@@ -1262,6 +1308,7 @@ var DrizzleAgentStore = class {
       ...thread.model != null ? {
         model: thread.model
       } : {},
+      persona: thread.persona ?? null,
       ...lastContent !== void 0 ? {
         lastMessagePreview: lastContent.slice(0, 120)
       } : {}
@@ -1291,6 +1338,9 @@ var DrizzleAgentStore = class {
       ...message.agentName != null ? {
         agentName: message.agentName
       } : {},
+      ...message.persona != null ? {
+        persona: message.persona
+      } : {},
       ...message.runId != null ? {
         runId: message.runId
       } : {},
@@ -1321,6 +1371,9 @@ function toQueuedMessage(row) {
     ...row.agentName !== null ? {
       agentName: row.agentName
     } : {},
+    ...row.persona !== null ? {
+      persona: row.persona
+    } : {},
     ...row.model !== null ? {
       model: row.model
     } : {},