@dudousxd/nestjs-agent-store-drizzle 0.28.2 → 0.28.4

package/dist/index.cjs

@@ -1398,6 +1398,11 @@ var DrizzleActionProposalStore = class {
         })
       };
       if ((this.dialect === "mysql" ? mysqlAffectedRows(updated) : updated.length) !== 1) throw new Error("Replacement lost its locked proposal fence");
+      const settled = this.toolCallSettlement(db, old.proposal, next);
+      if (settled) yield {
+        kind: "write",
+        query: settled
+      };
     }
     return {
       status: "created",
@@ -1650,6 +1655,38 @@ var DrizzleActionProposalStore = class {
       messageId
     };
   }
+  /**
+  * The update that settles the proposal's `proposed` tool-call record after a transition from
+  * `previous` to `next` (see `toolCallUpdateForTransition`), or `undefined` when its status stays —
+  * so the dashboard and run detail show `executed` / `failed` / `rejected` / `expired` instead of
+  * `proposed` forever.
+  */
+  toolCallSettlement(db, previous, next) {
+    const update = (0, import_nestjs_agent_core3.toolCallUpdateForTransition)(previous, next);
+    if (update === null) return void 0;
+    const toolCall = agentTablesFor(this.dialect).agentToolCall;
+    return db.update(toolCall).set({
+      status: update.status,
+      ...update.output !== void 0 ? {
+        output: update.output
+      } : {},
+      ...update.error !== void 0 ? {
+        error: update.error
+      } : {},
+      ...update.executedByRef !== void 0 ? {
+        executedByRef: update.executedByRef
+      } : {},
+      ...update.decidedVia !== void 0 ? {
+        decidedVia: update.decidedVia
+      } : {},
+      ...update.remember !== void 0 ? {
+        remember: update.remember
+      } : {},
+      ...update.status === "executed" ? {
+        executedAt: new Date(this.clock())
+      } : {}
+    }).where((0, import_drizzle_orm3.and)((0, import_drizzle_orm3.eq)(toolCall.id, update.toolCallId), (0, import_drizzle_orm3.eq)(toolCall.proposalId, next.id)));
+  }
   async read(scope, id) {
     const [row] = await this.db.select().from(this.table).where((0, import_drizzle_orm3.and)((0, import_drizzle_orm3.eq)(this.table.id, physicalId(id)), (0, import_drizzle_orm3.eq)(this.table.scopeKey, scopeKey(scope)))).limit(1);
     return row && row.proposal.id === id && (0, import_nestjs_agent_core3.actionProposalScopeMatches)(row.proposal, scope) ? row : void 0;
@@ -1668,7 +1705,10 @@ var DrizzleActionProposalStore = class {
         ...discoveryMetadata(result.proposal),
         version: row2.version + 1
       }).where((0, import_drizzle_orm3.and)((0, import_drizzle_orm3.eq)(this.table.id, physicalId(id)), (0, import_drizzle_orm3.eq)(this.table.scopeKey, scopeKey(scope)), (0, import_drizzle_orm3.eq)(this.table.version, row2.version))), this.table.id);
-      if (changed === 1) return result;
+      if (changed === 1) {
+        await this.toolCallSettlement(this.db, row2.proposal, result.proposal);
+        return result;
+      }
     }
     const row = await this.read(scope, id);
     return row ? {
@@ -2668,6 +2708,7 @@ var import_drizzle_orm5 = require("drizzle-orm");
 var UNCLASSIFIED_ERROR_CODE = "unknown";
 var DEFAULT_AGENT_BUCKET = "(default)";
 var TOOL_CALL_STATUSES = [
+  "proposed",
   "auto_executed",
   "pending_approval",
   "executed",