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

package/README.md

@@ -150,7 +150,7 @@ reader's `IN (…)` and `truncateFrom`'s delete) filter on it.
 ## Attachment housekeeping
 
 `referencedMediaIds(actorRef, mediaIds)` answers which of a set of media ids a message that still
-exists carries, for one actor — the inverse of the host's own staged-media inventory, and the half a
+exists — or one still waiting in a thread's queue (`agent_queued_message`) — carries, for one actor — the inverse of the host's own staged-media inventory, and the half a
 sweep cannot work out for itself. It needs **no schema change**: it reads the `attachments` JSON
 column messages have carried since attachments shipped, so an existing database answers correctly
 the moment you upgrade, with nothing to backfill.

package/dist/index.cjs

@@ -62,6 +62,8 @@ var agentThread = (0, import_sqlite_core.sqliteTable)("agent_thread", {
   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"),
+  /** The persona a send on this thread runs under when it names none; `null` → the agent's default. */
+  persona: (0, import_sqlite_core.text)("persona"),
   /** Why the thread's message queue stopped draining; `null` → it drains. */
   queuePause: (0, import_sqlite_core.text)("queue_pause", {
     mode: "json"
@@ -101,6 +103,8 @@ var agentMessage = (0, import_sqlite_core.sqliteTable)("agent_message", {
     mode: "json"
   }).$type(),
   agentName: (0, import_sqlite_core.text)("agent_name"),
+  /** The persona the turn that wrote this message ran under; `null` → none. */
+  persona: (0, import_sqlite_core.text)("persona"),
   runId: (0, import_sqlite_core.text)("run_id"),
   /** The step's streamed thinking; `null` when the model produced none. */
   reasoning: (0, import_sqlite_core.text)("reasoning"),
@@ -173,6 +177,8 @@ var agentQueuedMessage = (0, import_sqlite_core.sqliteTable)("agent_queued_messa
     mode: "json"
   }).$type(),
   agentName: (0, import_sqlite_core.text)("agent_name"),
+  /** The persona the send resolved — what the message starts under. */
+  persona: (0, import_sqlite_core.text)("persona"),
   model: (0, import_sqlite_core.text)("model"),
   pageContext: (0, import_sqlite_core.text)("page_context", {
     mode: "json"
@@ -360,6 +366,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,
@@ -378,6 +385,7 @@ var statements = [
     follow_ups TEXT,
     usage TEXT,
     agent_name TEXT,
+    persona TEXT,
     run_id TEXT,
     reasoning TEXT,
     reasoning_ms INTEGER,
@@ -394,6 +402,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,
@@ -589,6 +598,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) {
@@ -636,6 +660,7 @@ var DrizzleAgentStore = class {
       activeStreamId: null,
       defaultAgent: null,
       model: null,
+      persona: input.persona ?? null,
       queuePause: null,
       createdAt: now,
       updatedAt: now,
@@ -743,6 +768,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,
@@ -762,6 +788,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.
@@ -922,6 +949,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,
@@ -1017,6 +1045,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;
     }
@@ -1035,6 +1066,13 @@ var DrizzleAgentStore = class {
     }).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;
   }
+  /** The thread's pinned persona, projected like {@link defaultAgentForThread}. */
+  async personaForThread(threadId) {
+    const [thread] = await this.db.select({
+      persona: agentThread.persona
+    }).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?.persona ?? null;
+  }
   async defaultAgentForThread(threadId) {
     const [thread] = await this.db.select({
       defaultAgent: agentThread.defaultAgent
@@ -1105,6 +1143,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,
@@ -1235,7 +1274,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
@@ -1254,9 +1294,15 @@ var DrizzleAgentStore = class {
     const rows = await this.db.select({
       attachments: agentMessage.attachments
     }).from(agentMessage).innerJoin(agentThread, (0, import_drizzle_orm2.eq)(agentMessage.threadId, agentThread.id)).where((0, import_drizzle_orm2.and)((0, import_drizzle_orm2.eq)(agentThread.actorRef, actorRef), (0, import_drizzle_orm2.isNotNull)(agentMessage.attachments)));
+    const queued = await this.db.select({
+      attachments: agentQueuedMessage.attachments
+    }).from(agentQueuedMessage).innerJoin(agentThread, (0, import_drizzle_orm2.eq)(agentQueuedMessage.threadId, agentThread.id)).where((0, import_drizzle_orm2.and)((0, import_drizzle_orm2.eq)(agentThread.actorRef, actorRef), (0, import_drizzle_orm2.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);
@@ -1308,6 +1354,7 @@ var DrizzleAgentStore = class {
       ...thread.model != null ? {
         model: thread.model
       } : {},
+      persona: thread.persona ?? null,
       ...lastContent !== void 0 ? {
         lastMessagePreview: lastContent.slice(0, 120)
       } : {}
@@ -1337,6 +1384,9 @@ var DrizzleAgentStore = class {
       ...message.agentName != null ? {
         agentName: message.agentName
       } : {},
+      ...message.persona != null ? {
+        persona: message.persona
+      } : {},
       ...message.runId != null ? {
         runId: message.runId
       } : {},
@@ -1367,6 +1417,9 @@ function toQueuedMessage(row) {
     ...row.agentName !== null ? {
       agentName: row.agentName
     } : {},
+    ...row.persona !== null ? {
+      persona: row.persona
+    } : {},
     ...row.model !== null ? {
       model: row.model
     } : {},