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

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

@@ -1235,7 +1235,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 +1255,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);