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

package/dist/index.d.cts

@@ -4845,7 +4845,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

@@ -4845,7 +4845,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

@@ -1189,7 +1189,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 +1209,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);