@dudousxd/nestjs-agent-store-drizzle 0.19.0 → 0.20.0

package/dist/index.cjs

@@ -904,6 +904,9 @@ var DrizzleAgentStore = class {
     if (patch.attachments !== void 0) {
       updates.attachments = patch.attachments === null || patch.attachments.length === 0 ? null : patch.attachments;
     }
+    if (patch.interrupt !== void 0) {
+      updates.interrupt = patch.interrupt;
+    }
     await this.db.update(agentQueuedMessage).set(updates).where((0, import_drizzle_orm2.eq)(agentQueuedMessage.id, id));
     return this.getQueuedMessage(id);
   }
@@ -1148,6 +1151,43 @@ var DrizzleAgentStore = class {
     }
     await this.db.update(agentToolCall).set(updates).where((0, import_drizzle_orm2.eq)(agentToolCall.id, input.toolCallId));
   }
+  async toolCallOutcomes(toolCallIds) {
+    if (toolCallIds.length === 0) {
+      return [];
+    }
+    const rows = await this.db.select({
+      id: agentToolCall.id,
+      status: agentToolCall.status,
+      output: agentToolCall.output,
+      error: agentToolCall.error
+    }).from(agentToolCall).where((0, import_drizzle_orm2.inArray)(agentToolCall.id, [
+      ...toolCallIds
+    ]));
+    return rows.map((row) => ({
+      id: row.id,
+      status: row.status,
+      ...row.output !== void 0 && row.output !== null ? {
+        output: row.output
+      } : {},
+      ...typeof row.error === "string" ? {
+        error: row.error
+      } : {}
+    }));
+  }
+  async failUnsettledToolCalls(runId, error) {
+    const pending = (0, import_drizzle_orm2.and)((0, import_drizzle_orm2.eq)(agentToolCall.runId, runId), (0, import_drizzle_orm2.eq)(agentToolCall.status, "pending_approval"));
+    const rows = await this.db.select({
+      id: agentToolCall.id
+    }).from(agentToolCall).where(pending);
+    if (rows.length === 0) {
+      return 0;
+    }
+    await this.db.update(agentToolCall).set({
+      status: "failed",
+      error
+    }).where(pending);
+    return rows.length;
+  }
   /**
   * Of `mediaIds`, the ones a surviving message in one of this actor's threads still carries.
   *