@exaudeus/workrail 1.5.2 → 1.5.4

package/dist/manifest.json

@@ -606,8 +606,8 @@
       "bytes": 259
     },
     "mcp/handlers/v2-checkpoint.js": {
-      "sha256": "24b7aaf920fd2c5e4ec801ab9c0f3f7d93aa8280c05b851f12e57343d352a4ec",
-      "bytes": 9196
+      "sha256": "48c20ca225f9595f845c2692083d581a3e1ebdb3f423b210ef8b37c69e727954",
+      "bytes": 9558
     },
     "mcp/handlers/v2-context-budget.d.ts": {
       "sha256": "cbad1741a183d52c9cbe558be2e09f776843d1f3ec8cd28d6d0d230668e4298c",

package/dist/mcp/handlers/v2-checkpoint.js

@@ -5,6 +5,7 @@ const neverthrow_1 = require("neverthrow");
 const types_js_1 = require("../types.js");
 const output_schemas_js_1 = require("../output-schemas.js");
 const v2_token_ops_js_1 = require("./v2-token-ops.js");
+const v2_execution_helpers_js_1 = require("./v2-execution-helpers.js");
 const index_js_1 = require("../../v2/durable-core/ids/index.js");
 const workflow_hash_ref_js_1 = require("../../v2/durable-core/ids/workflow-hash-ref.js");
 const index_js_2 = require("../../v2/durable-core/schemas/session/index.js");
@@ -58,25 +59,37 @@ function executeCheckpoint(input, ctx) {
     const nodeId = (0, index_js_1.asNodeId)(String(token.payload.nodeId));
     const attemptId = (0, index_js_1.asAttemptId)(String(token.payload.attemptId));
     const dedupeKey = `checkpoint:${String(sessionId)}:${String(runId)}:${String(nodeId)}:${String(attemptId)}`;
-    return gate.withHealthySessionLock(sessionId, (lock) => {
-        return sessionStore.load(sessionId)
-            .mapErr((cause) => ({ kind: 'store_failed', cause }))
-            .andThen((truth) => {
-            const originalNode = findNodeCreated(truth.events, nodeId);
-            if (!originalNode) {
-                return (0, neverthrow_1.errAsync)({ kind: 'missing_node_or_run' });
-            }
-            const alreadyRecorded = truth.events.some((e) => e.dedupeKey === dedupeKey);
-            if (alreadyRecorded) {
-                return replayCheckpoint(truth.events, dedupeKey, originalNode, sessionId, runId, nodeId, tokenCodecPorts);
+    return sessionStore.load(sessionId)
+        .mapErr((cause) => ({ kind: 'store_failed', cause }))
+        .andThen((truth) => {
+        const originalNode = findNodeCreated(truth.events, nodeId);
+        if (!originalNode) {
+            return (0, neverthrow_1.errAsync)({ kind: 'missing_node_or_run' });
+        }
+        const alreadyRecorded = truth.events.some((e) => e.dedupeKey === dedupeKey);
+        if (alreadyRecorded) {
+            return replayCheckpoint(truth.events, dedupeKey, originalNode, sessionId, runId, nodeId, tokenCodecPorts);
+        }
+        return gate.withHealthySessionLock(sessionId, (lock) => {
+            return sessionStore.load(sessionId)
+                .mapErr((cause) => ({ kind: 'store_failed', cause }))
+                .andThen((truthLocked) => {
+                const originalNodeLocked = findNodeCreated(truthLocked.events, nodeId);
+                if (!originalNodeLocked) {
+                    return (0, neverthrow_1.errAsync)({ kind: 'missing_node_or_run' });
+                }
+                const alreadyRecordedLocked = truthLocked.events.some((e) => e.dedupeKey === dedupeKey);
+                if (alreadyRecordedLocked) {
+                    return replayCheckpoint(truthLocked.events, dedupeKey, originalNodeLocked, sessionId, runId, nodeId, tokenCodecPorts);
+                }
+                return writeCheckpoint(truthLocked, dedupeKey, originalNodeLocked, sessionId, runId, nodeId, idFactory.mintNodeId(), () => idFactory.mintEventId(), lock, sessionStore, tokenCodecPorts);
+            });
+        }).mapErr((gateErr) => {
+            if (isGateError(gateErr)) {
+                return { kind: 'gate_failed', cause: gateErr };
             }
-            return writeCheckpoint(truth, dedupeKey, originalNode, sessionId, runId, nodeId, idFactory.mintNodeId(), () => idFactory.mintEventId(), lock, sessionStore, tokenCodecPorts);
+            return gateErr;
         });
-    }).mapErr((gateErr) => {
-        if (isGateError(gateErr)) {
-            return { kind: 'gate_failed', cause: gateErr };
-        }
-        return gateErr;
     });
 }
 function replayCheckpoint(events, dedupeKey, originalNode, sessionId, runId, nodeId, tokenCodecPorts) {
@@ -165,16 +178,8 @@ function mapCheckpointErrorToToolError(e) {
             return (0, types_js_1.errNotRetryable)('TOKEN_UNKNOWN_NODE', 'No durable node state found for this checkpointToken. Use a checkpointToken returned by WorkRail.');
         case 'event_schema_invalid':
             return (0, types_js_1.errNotRetryable)('INTERNAL_ERROR', `Checkpoint events failed schema validation: ${e.issues}`);
-        case 'gate_failed': {
-            const code = e.cause.code;
-            if (code === 'SESSION_LOCKED' || code === 'SESSION_LOCK_REENTRANT') {
-                return (0, types_js_1.errNotRetryable)('TOKEN_SESSION_LOCKED', `Session is locked: ${code}`);
-            }
-            if (code === 'SESSION_NOT_HEALTHY') {
-                return (0, types_js_1.errNotRetryable)('SESSION_NOT_HEALTHY', 'Session is not healthy.');
-            }
-            return (0, types_js_1.errNotRetryable)('INTERNAL_ERROR', `Session gate error: ${code}`);
-        }
+        case 'gate_failed':
+            return (0, v2_execution_helpers_js_1.mapExecutionSessionGateErrorToToolError)(e.cause);
         case 'store_failed':
             return (0, types_js_1.errNotRetryable)('INTERNAL_ERROR', `Session store error: ${e.cause.code}`);
     }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@exaudeus/workrail",
-  "version": "1.5.2",
+  "version": "1.5.4",
   "description": "Step-by-step workflow enforcement for AI agents via MCP",
   "license": "MIT",
   "repository": {