@exaudeus/workrail 3.14.0 → 3.16.0

package/dist/mcp/handlers/v2-checkpoint.d.ts

@@ -2,7 +2,7 @@ import type { z } from 'zod';
 import type { ResultAsync as RA } from 'neverthrow';
 import type { ToolContext, ToolResult, V2ToolContext } from '../types.js';
 import type { V2CheckpointWorkflowInput } from '../v2/tools.js';
-import { V2CheckpointWorkflowOutputSchema } from '../output-schemas.js';
+import type { V2CheckpointWorkflowOutputSchema } from '../output-schemas.js';
 import { type ToolFailure } from './v2-execution-helpers.js';
 import type { ExecutionSessionGateErrorV2 } from '../../v2/usecases/execution-session-gate.js';
 import type { SessionEventLogStoreError } from '../../v2/ports/session-event-log-store.port.js';

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

@@ -4,7 +4,6 @@ exports.handleV2CheckpointWorkflow = handleV2CheckpointWorkflow;
 exports.executeCheckpoint = executeCheckpoint;
 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");
@@ -107,11 +106,11 @@ function replayCheckpoint(events, dedupeKey, originalNode, sessionId, runId, nod
         .andThen((resumeTokenValue) => (0, v2_token_ops_js_1.mintContinueAndCheckpointTokens)({
         entry: { sessionId: String(sessionId), runId: String(runId), nodeId: String(nodeId), attemptId: String(attemptId), workflowHashRef },
         ports: tokenCodecPorts, aliasStore, entropy,
-    }).andThen(({ continueToken }) => (0, neverthrow_1.okAsync)(output_schemas_js_1.V2CheckpointWorkflowOutputSchema.parse({
+    }).andThen(({ continueToken }) => (0, neverthrow_1.okAsync)({
         checkpointNodeId,
         resumeToken: resumeTokenValue,
         nextCall: { tool: 'continue_workflow', params: { continueToken } },
-    }))))
+    })))
         .mapErr((e) => ({ kind: 'store_failed', cause: e }));
 }
 function writeCheckpoint(truth, dedupeKey, originalNode, sessionId, runId, nodeId, attemptId, checkpointNodeId, mintEventId, lock, sessionStore, tokenCodecPorts, aliasStore, entropy) {
@@ -161,7 +160,7 @@ function writeCheckpoint(truth, dedupeKey, originalNode, sessionId, runId, nodeI
             eventIndex: truth.events.length,
             createdByEventId: nodeCreatedEventId,
         }];
-    return sessionStore.append(lock, { events: validated, snapshotPins })
+    return sessionStore.append(lock, { events: validated, snapshotPins }, truth)
         .mapErr((cause) => ({ kind: 'store_failed', cause }))
         .andThen(() => {
         const workflowHashRefRes = (0, workflow_hash_ref_js_1.deriveWorkflowHashRef)(originalNode.data.workflowHash);
@@ -170,11 +169,11 @@ function writeCheckpoint(truth, dedupeKey, originalNode, sessionId, runId, nodeI
             .andThen((resumeTokenValue) => (0, v2_token_ops_js_1.mintContinueAndCheckpointTokens)({
             entry: { sessionId: String(sessionId), runId: String(runId), nodeId: String(nodeId), attemptId: String(attemptId), workflowHashRef },
             ports: tokenCodecPorts, aliasStore, entropy,
-        }).andThen(({ continueToken }) => (0, neverthrow_1.okAsync)(output_schemas_js_1.V2CheckpointWorkflowOutputSchema.parse({
+        }).andThen(({ continueToken }) => (0, neverthrow_1.okAsync)({
             checkpointNodeId: String(checkpointNodeId),
             resumeToken: resumeTokenValue,
             nextCall: { tool: 'continue_workflow', params: { continueToken } },
-        }))))
+        })))
             .mapErr((e) => ({ kind: 'store_failed', cause: e }));
     });
 }

package/dist/mcp/handlers/v2-execution/advance.d.ts

@@ -1,5 +1,6 @@
 import type { V2ContinueWorkflowInput } from '../../v2/tools.js';
-import { createWorkflow } from '../../../types/workflow.js';
+import type { SessionIndex } from '../../../v2/durable-core/session-index.js';
+import type { Workflow } from '../../../types/workflow.js';
 import { type AttemptId } from '../../../v2/durable-core/tokens/index.js';
 import { type SessionId, type RunId, type NodeId, type WorkflowHash } from '../../../v2/durable-core/ids/index.js';
 import type { LoadedSessionTruthV2 } from '../../../v2/ports/session-event-log-store.port.js';
@@ -21,7 +22,7 @@ export declare function advanceAndRecord(args: {
     readonly inputContext: JsonValue | undefined;
     readonly inputOutput: V2ContinueWorkflowInput['output'];
     readonly lock: WithHealthySessionLock;
-    readonly pinnedWorkflow: ReturnType<typeof createWorkflow>;
+    readonly pinnedWorkflow: Workflow;
     readonly snapshotStore: import('../../../v2/ports/snapshot-store.port.js').SnapshotStorePortV2;
     readonly sessionStore: import('../../../v2/ports/session-event-log-store.port.js').SessionEventLogAppendStorePortV2 & import('../../../v2/ports/session-event-log-store.port.js').SessionEventLogReadonlyStorePortV2;
     readonly sha256: Sha256PortV2;
@@ -29,4 +30,5 @@ export declare function advanceAndRecord(args: {
         readonly mintNodeId: () => NodeId;
         readonly mintEventId: () => string;
     };
+    readonly lockedIndex: SessionIndex;
 }): RA<void, InternalError | SessionEventLogStoreError | SnapshotStoreError>;

package/dist/mcp/handlers/v2-execution/advance.js

@@ -3,18 +3,14 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.advanceAndRecord = advanceAndRecord;
 const neverthrow_1 = require("neverthrow");
 const v2_advance_core_js_1 = require("../v2-advance-core.js");
-const constants_js_1 = require("../../../v2/durable-core/constants.js");
 function advanceAndRecord(args) {
     const { truth, sessionId, runId, nodeId, attemptId, workflowHash, dedupeKey, inputContext, inputOutput, lock, pinnedWorkflow, snapshotStore, sessionStore, sha256, idFactory } = args;
-    const hasRun = truth.events.some((e) => e.kind === constants_js_1.EVENT_KIND.RUN_STARTED && e.scope?.runId === String(runId));
-    const hasNode = truth.events.some((e) => e.kind === constants_js_1.EVENT_KIND.NODE_CREATED && e.scope?.runId === String(runId) && e.scope?.nodeId === String(nodeId));
+    const hasRun = args.lockedIndex.runStartedByRunId.has(String(runId));
+    const nodeCreated = args.lockedIndex.nodeCreatedByNodeId.get(String(nodeId));
+    const hasNode = nodeCreated !== undefined;
     if (!hasRun || !hasNode) {
         return (0, neverthrow_1.errAsync)({ kind: 'missing_node_or_run' });
     }
-    const nodeCreated = truth.events.find((e) => e.kind === constants_js_1.EVENT_KIND.NODE_CREATED && e.scope?.nodeId === String(nodeId));
-    if (!nodeCreated) {
-        return (0, neverthrow_1.errAsync)({ kind: 'missing_node_or_run' });
-    }
     if (String(nodeCreated.data.workflowHash) !== String(workflowHash)) {
         return (0, neverthrow_1.errAsync)({ kind: 'workflow_hash_mismatch' });
     }
@@ -38,6 +34,7 @@ function advanceAndRecord(args) {
                 truth, sessionId, runId, attemptId, workflowHash, dedupeKey,
                 inputContext, inputOutput, lock, pinnedWorkflow,
                 ports: { snapshotStore, sessionStore, sha256, idFactory },
+                lockedIndex: args.lockedIndex,
             });
         }
         return (0, v2_advance_core_js_1.executeAdvanceCore)({
@@ -45,6 +42,7 @@ function advanceAndRecord(args) {
             truth, sessionId, runId, attemptId, workflowHash, dedupeKey,
             inputContext, inputOutput, lock, pinnedWorkflow,
             ports: { snapshotStore, sessionStore, sha256, idFactory },
+            lockedIndex: args.lockedIndex,
         });
     });
 }

package/dist/mcp/handlers/v2-execution/continue-advance.js

@@ -1,19 +1,27 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.handleAdvanceIntent = handleAdvanceIntent;
-const workflow_js_1 = require("../../../types/workflow.js");
+const workflow_object_cache_js_1 = require("./workflow-object-cache.js");
 const workflow_hash_ref_js_1 = require("../../../v2/durable-core/ids/workflow-hash-ref.js");
 const neverthrow_1 = require("neverthrow");
-const workflow_source_js_1 = require("../../../types/workflow-source.js");
 const workflow_definition_js_1 = require("../../../types/workflow-definition.js");
 const v2_error_mapping_js_1 = require("../v2-error-mapping.js");
-const constants_js_1 = require("../../../v2/durable-core/constants.js");
 const replay_js_1 = require("./replay.js");
 const advance_js_1 = require("./advance.js");
+const sorted_event_log_js_1 = require("../../../v2/durable-core/sorted-event-log.js");
+const session_index_js_1 = require("../../../v2/durable-core/session-index.js");
 function handleAdvanceIntent(args) {
     const { input, sessionId, runId, nodeId, attemptId, workflowHashRef, truth, gate, sessionStore, snapshotStore, pinnedStore, tokenCodecPorts, idFactory, sha256, aliasStore, entropy } = args;
     const dedupeKey = `advance_recorded:${sessionId}:${nodeId}:${attemptId}`;
-    const runStarted = truth.events.find((e) => e.kind === constants_js_1.EVENT_KIND.RUN_STARTED && e.scope.runId === String(runId));
+    const preLockSortedResult = (0, sorted_event_log_js_1.asSortedEventLog)(truth.events);
+    if (preLockSortedResult.isErr()) {
+        return (0, neverthrow_1.errAsync)({
+            kind: 'invariant_violation',
+            message: `Session events are not sorted: ${preLockSortedResult.error.message}`,
+        });
+    }
+    const preLockIndex = (0, session_index_js_1.buildSessionIndex)(preLockSortedResult.value);
+    const runStarted = preLockIndex.runStartedByRunId.get(String(runId));
     if (!runStarted) {
         return (0, neverthrow_1.errAsync)({
             kind: 'token_unknown_node',
@@ -37,7 +45,7 @@ function handleAdvanceIntent(args) {
             suggestion: 'Use the continueToken returned by WorkRail for this run.',
         });
     }
-    const nodeCreated = truth.events.find((e) => e.kind === constants_js_1.EVENT_KIND.NODE_CREATED && e.scope.nodeId === String(nodeId) && e.scope.runId === String(runId));
+    const nodeCreated = preLockIndex.nodeCreatedByNodeId.get(String(nodeId));
     if (!nodeCreated) {
         return (0, neverthrow_1.errAsync)({
             kind: 'token_unknown_node',
@@ -45,22 +53,24 @@ function handleAdvanceIntent(args) {
             suggestion: 'Use tokens returned by WorkRail for an existing node.',
         });
     }
-    const nodeRefRes = (0, workflow_hash_ref_js_1.deriveWorkflowHashRef)(nodeCreated.data.workflowHash);
-    if (nodeRefRes.isErr()) {
-        return (0, neverthrow_1.errAsync)({
-            kind: 'precondition_failed',
-            message: nodeRefRes.error.message,
-            suggestion: 'Re-pin the workflow via start_workflow.',
-        });
-    }
-    if (String(nodeRefRes.value) !== String(workflowHashRef)) {
-        return (0, neverthrow_1.errAsync)({
-            kind: 'precondition_failed',
-            message: 'workflowHash mismatch for this node.',
-            suggestion: 'Use the continueToken returned by WorkRail for this node.',
-        });
+    if (nodeCreated.data.workflowHash !== workflowHash) {
+        const nodeRefRes = (0, workflow_hash_ref_js_1.deriveWorkflowHashRef)(nodeCreated.data.workflowHash);
+        if (nodeRefRes.isErr()) {
+            return (0, neverthrow_1.errAsync)({
+                kind: 'precondition_failed',
+                message: nodeRefRes.error.message,
+                suggestion: 'Re-pin the workflow via start_workflow.',
+            });
+        }
+        if (String(nodeRefRes.value) !== String(workflowHashRef)) {
+            return (0, neverthrow_1.errAsync)({
+                kind: 'precondition_failed',
+                message: 'workflowHash mismatch for this node.',
+                suggestion: 'Use the continueToken returned by WorkRail for this node.',
+            });
+        }
     }
-    const existing = truth.events.find((e) => e.kind === constants_js_1.EVENT_KIND.ADVANCE_RECORDED && e.dedupeKey === dedupeKey);
+    const existing = preLockIndex.advanceRecordedByDedupeKey.get(dedupeKey);
     return pinnedStore.get(workflowHash)
         .mapErr((cause) => ({ kind: 'pinned_workflow_store_failed', cause }))
         .andThen((compiled) => {
@@ -75,7 +85,7 @@ function handleAdvanceIntent(args) {
                 suggestion: 'Re-pin the workflow via start_workflow.',
             });
         }
-        const pinnedWorkflow = (0, workflow_js_1.createWorkflow)(compiled.definition, (0, workflow_source_js_1.createBundledSource)());
+        const pinnedWorkflow = (0, workflow_object_cache_js_1.getCachedWorkflow)(workflowHash, compiled.definition);
         if (existing) {
             return (0, replay_js_1.replayFromRecordedAdvance)({
                 recordedEvent: existing,
@@ -95,9 +105,17 @@ function handleAdvanceIntent(args) {
         }
         return gate
             .withHealthySessionLock(sessionId, (lock) => sessionStore.load(sessionId).andThen((truthLocked) => {
-            const existingLocked = truthLocked.events.find((e) => e.kind === constants_js_1.EVENT_KIND.ADVANCE_RECORDED && e.dedupeKey === dedupeKey);
+            const lockedSortedResult = (0, sorted_event_log_js_1.asSortedEventLog)(truthLocked.events);
+            if (lockedSortedResult.isErr()) {
+                return (0, neverthrow_1.errAsync)({
+                    kind: 'invariant_violation',
+                    message: `Locked session events are not sorted: ${lockedSortedResult.error.message}`,
+                });
+            }
+            const lockedIndex = (0, session_index_js_1.buildSessionIndex)(lockedSortedResult.value);
+            const existingLocked = lockedIndex.advanceRecordedByDedupeKey.get(dedupeKey);
             if (existingLocked)
-                return (0, neverthrow_1.okAsync)({ kind: 'replay', truth: truthLocked, recordedEvent: existingLocked });
+                return (0, neverthrow_1.okAsync)({ kind: 'replay', truth: truthLocked, recordedEvent: existingLocked, precomputedIndex: lockedIndex });
             return (0, advance_js_1.advanceAndRecord)({
                 truth: truthLocked,
                 sessionId,
@@ -114,9 +132,21 @@ function handleAdvanceIntent(args) {
                 sessionStore,
                 sha256,
                 idFactory,
+                lockedIndex,
             }).andThen(() => sessionStore
                 .load(sessionId)
-                .map((truthAfter) => ({ kind: 'replay', truth: truthAfter, recordedEvent: null })));
+                .andThen((truthAfter) => {
+                const afterSortedResult = (0, sorted_event_log_js_1.asSortedEventLog)(truthAfter.events);
+                if (afterSortedResult.isErr()) {
+                    return (0, neverthrow_1.errAsync)({
+                        kind: 'invariant_violation',
+                        message: `Post-advance session events are not sorted: ${afterSortedResult.error.message}`,
+                    });
+                }
+                const index2 = (0, session_index_js_1.buildSessionIndex)(afterSortedResult.value);
+                const recordedEvent = index2.advanceRecordedByDedupeKey.get(dedupeKey) ?? null;
+                return (0, neverthrow_1.okAsync)({ kind: 'replay', truth: truthAfter, recordedEvent, precomputedIndex: index2 });
+            }));
         }))
             .mapErr((cause) => {
             if ((0, v2_error_mapping_js_1.isInternalError)(cause)) {
@@ -146,8 +176,7 @@ function handleAdvanceIntent(args) {
         })
             .andThen((res) => {
             const truth2 = res.truth;
-            const recordedEvent = res.recordedEvent ??
-                truth2.events.find((e) => e.kind === constants_js_1.EVENT_KIND.ADVANCE_RECORDED && e.dedupeKey === dedupeKey);
+            const recordedEvent = res.recordedEvent;
             if (!recordedEvent) {
                 return (0, neverthrow_1.errAsync)({
                     kind: 'invariant_violation',
@@ -157,6 +186,7 @@ function handleAdvanceIntent(args) {
             return (0, replay_js_1.replayFromRecordedAdvance)({
                 recordedEvent,
                 truth: truth2,
+                precomputedIndex: res.precomputedIndex,
                 sessionId,
                 runId,
                 nodeId,

package/dist/mcp/handlers/v2-execution/continue-rehydrate.d.ts

@@ -1,5 +1,5 @@
 import type { V2ContinueWorkflowInput } from '../../v2/tools.js';
-import { V2ContinueWorkflowOutputSchema } from '../../output-schemas.js';
+import type { V2ContinueWorkflowOutputSchema } from '../../output-schemas.js';
 import { type SessionId, type RunId, type NodeId } from '../../../v2/durable-core/ids/index.js';
 import type { LoadedSessionTruthV2 } from '../../../v2/ports/session-event-log-store.port.js';
 import type { TokenCodecPorts } from '../../../v2/durable-core/tokens/token-codec-ports.js';

package/dist/mcp/handlers/v2-execution/continue-rehydrate.js

@@ -6,11 +6,10 @@ const binding_drift_js_1 = require("../../../v2/durable-core/domain/binding-drif
 const binding_registry_js_1 = require("../../../application/services/compiler/binding-registry.js");
 const v2_workspace_resolution_js_1 = require("../v2-workspace-resolution.js");
 const snapshot_state_js_1 = require("../../../v2/durable-core/projections/snapshot-state.js");
-const workflow_js_1 = require("../../../types/workflow.js");
+const workflow_object_cache_js_1 = require("./workflow-object-cache.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 neverthrow_1 = require("neverthrow");
-const workflow_source_js_1 = require("../../../types/workflow-source.js");
 const workflow_definition_js_1 = require("../../../types/workflow-definition.js");
 const v2_execution_helpers_js_1 = require("../v2-execution-helpers.js");
 const prompt_renderer_js_1 = require("../../../v2/durable-core/domain/prompt-renderer.js");
@@ -19,6 +18,7 @@ const v2_state_conversion_js_1 = require("../v2-state-conversion.js");
 const constants_js_1 = require("../../../v2/durable-core/constants.js");
 const index_js_2 = require("./index.js");
 const step_content_envelope_js_1 = require("../../step-content-envelope.js");
+const assert_output_js_1 = require("../../assert-output.js");
 function handleRehydrateIntent(args) {
     const { input, sessionId, runId, nodeId, workflowHashRef, truth, tokenCodecPorts, pinnedStore, snapshotStore, idFactory, aliasStore, entropy, resolvedRootUris } = args;
     const runStarted = truth.events.find((e) => e.kind === constants_js_1.EVENT_KIND.RUN_STARTED && e.scope.runId === String(runId));
@@ -93,15 +93,15 @@ function handleRehydrateIntent(args) {
             if (!pending) {
                 const preferences = (0, v2_execution_helpers_js_1.derivePreferencesOrDefault)({ truth, runId, nodeId });
                 const nextIntent = (0, v2_state_conversion_js_1.deriveNextIntent)({ rehydrateOnly: true, isComplete, pending: null });
-                const parsed = output_schemas_js_1.V2ContinueWorkflowOutputSchema.parse({
+                const parsed = (0, assert_output_js_1.assertOutput)({
                     kind: 'ok',
                     isComplete,
                     pending: null,
                     preferences,
                     nextIntent,
                     nextCall: null,
-                    ...(driftWarnings.length > 0 ? { warnings: driftWarnings } : {}),
-                });
+                    ...(driftWarnings.length > 0 ? { warnings: [...driftWarnings] } : {}),
+                }, assert_output_js_1.assertContinueTokenPresence);
                 return (0, neverthrow_1.okAsync)({ response: parsed });
             }
             const attemptId = (0, v2_token_ops_js_1.newAttemptId)(idFactory);
@@ -115,7 +115,7 @@ function handleRehydrateIntent(args) {
             return (0, v2_token_ops_js_1.mintContinueAndCheckpointTokens)({ entry: entryBase, ports: tokenCodecPorts, aliasStore, entropy })
                 .mapErr((failure) => ({ kind: 'token_signing_failed', cause: failure }))
                 .andThen(({ continueToken: continueTokenValue, checkpointToken: checkpointTokenValue }) => {
-                const wf = (0, workflow_js_1.createWorkflow)(pinned.definition, (0, workflow_source_js_1.createBundledSource)());
+                const wf = (0, workflow_object_cache_js_1.getCachedWorkflow)(workflowHash, pinned.definition);
                 const metaRes = (0, prompt_renderer_js_1.renderPendingPrompt)({
                     workflow: wf,
                     stepId: String(pending.stepId),
@@ -138,7 +138,7 @@ function handleRehydrateIntent(args) {
                     meta,
                     references: pinned.resolvedReferences ?? buildPinnedReferencesFallback(pinned.definition.references ?? []),
                 });
-                const parsed = output_schemas_js_1.V2ContinueWorkflowOutputSchema.parse({
+                const parsed = (0, assert_output_js_1.assertOutput)({
                     kind: 'ok',
                     continueToken: continueTokenValue,
                     checkpointToken: checkpointTokenValue,
@@ -147,8 +147,8 @@ function handleRehydrateIntent(args) {
                     preferences,
                     nextIntent,
                     nextCall: (0, index_js_2.buildNextCall)({ continueToken: continueTokenValue, isComplete, pending: meta }),
-                    ...(driftWarnings.length > 0 ? { warnings: driftWarnings } : {}),
-                });
+                    ...(driftWarnings.length > 0 ? { warnings: [...driftWarnings] } : {}),
+                }, assert_output_js_1.assertContinueTokenPresence);
                 return (0, neverthrow_1.okAsync)({ response: parsed, contentEnvelope });
             });
         });

package/dist/mcp/handlers/v2-execution/replay.d.ts

@@ -1,7 +1,7 @@
-import { V2ContinueWorkflowOutputSchema } from '../../output-schemas.js';
+import type { V2ContinueWorkflowOutputSchema } from '../../output-schemas.js';
 import type { ExecutionSnapshotFileV1 } from '../../../v2/durable-core/schemas/execution-snapshot/index.js';
 import { type AttemptId } from '../../../v2/durable-core/tokens/index.js';
-import { createWorkflow } from '../../../types/workflow.js';
+import type { Workflow } from '../../../types/workflow.js';
 import type { DomainEventV1 } from '../../../v2/durable-core/schemas/session/index.js';
 import { type SessionId, type RunId, type NodeId, type WorkflowHash } from '../../../v2/durable-core/ids/index.js';
 import type { LoadedSessionTruthV2 } from '../../../v2/ports/session-event-log-store.port.js';
@@ -17,13 +17,14 @@ export declare function buildAdvancedReplayResponse(args: {
     readonly toNodeId: NodeId;
     readonly attemptId: AttemptId;
     readonly toSnapshot: ExecutionSnapshotFileV1;
-    readonly workflow: ReturnType<typeof createWorkflow>;
+    readonly workflow: Workflow;
     readonly truth: LoadedSessionTruthV2;
     readonly workflowHash: WorkflowHash;
     readonly ports: TokenCodecPorts;
     readonly sha256: Sha256PortV2;
     readonly aliasStore: import('../../../v2/ports/token-alias-store.port.js').TokenAliasStorePortV2;
     readonly entropy: import('../../../v2/ports/random-entropy.port.js').RandomEntropyPortV2;
+    readonly precomputedIndex?: import('../../../v2/durable-core/session-index.js').SessionIndex;
 }): RA<z.infer<typeof V2ContinueWorkflowOutputSchema>, ContinueWorkflowError>;
 export declare function replayFromRecordedAdvance(args: {
     readonly recordedEvent: Extract<DomainEventV1, {
@@ -35,10 +36,11 @@ export declare function replayFromRecordedAdvance(args: {
     readonly nodeId: NodeId;
     readonly workflowHash: WorkflowHash;
     readonly attemptId: AttemptId;
-    readonly pinnedWorkflow: ReturnType<typeof createWorkflow>;
+    readonly pinnedWorkflow: Workflow;
     readonly snapshotStore: import('../../../v2/ports/snapshot-store.port.js').SnapshotStorePortV2;
     readonly sha256: Sha256PortV2;
     readonly tokenCodecPorts: TokenCodecPorts;
     readonly aliasStore: import('../../../v2/ports/token-alias-store.port.js').TokenAliasStorePortV2;
     readonly entropy: import('../../../v2/ports/random-entropy.port.js').RandomEntropyPortV2;
+    readonly precomputedIndex?: import('../../../v2/durable-core/session-index.js').SessionIndex;
 }): RA<z.infer<typeof V2ContinueWorkflowOutputSchema>, ContinueWorkflowError>;

package/dist/mcp/handlers/v2-execution/replay.js

@@ -15,6 +15,8 @@ const v2_state_conversion_js_1 = require("../v2-state-conversion.js");
 const constants_js_1 = require("../../../v2/durable-core/constants.js");
 const index_js_2 = require("./index.js");
 const assessments_js_1 = require("../../../v2/projections/assessments.js");
+const assert_output_js_1 = require("../../assert-output.js");
+const sorted_event_log_js_1 = require("../../../v2/durable-core/sorted-event-log.js");
 function buildAdvancedReplayResponse(args) {
     const { sessionId, runId, fromNodeId, toNodeId, attemptId, toSnapshot, workflow, truth, workflowHash, ports, sha256, aliasStore, entropy } = args;
     const toNodeIdBranded = (0, index_js_1.asNodeId)(String(toNodeId));
@@ -75,29 +77,33 @@ function buildAdvancedReplayResponse(args) {
                 runId: (0, index_js_1.asRunId)(String(runId)),
                 nodeId: (0, index_js_1.asNodeId)(String(toNodeIdBranded)),
                 rehydrateOnly: false,
+                precomputedIndex: args.precomputedIndex,
             });
             if (result.isErr()) {
                 return (0, neverthrow_1.errAsync)({ kind: 'prompt_render_failed', message: result.error.message });
             }
             blockedMeta = result.value;
         }
-        const preferences = (0, v2_execution_helpers_js_1.derivePreferencesOrDefault)({ truth, runId, nodeId: toNodeIdBranded });
+        const preferences = (0, v2_execution_helpers_js_1.derivePreferencesOrDefault)({ truth, runId, nodeId: toNodeIdBranded, precomputedIndex: args.precomputedIndex });
         const nextIntent = (0, v2_state_conversion_js_1.deriveNextIntent)({ rehydrateOnly: false, isComplete, pending: blockedMeta });
-        return nextTokensMint.andThen((nextTokens) => retryContinueMint.andThen((retryContinueToken) => (0, neverthrow_1.okAsync)(output_schemas_js_1.V2ContinueWorkflowOutputSchema.parse({
-            kind: 'blocked',
-            continueToken: pending ? nextTokens.continueToken : undefined,
-            checkpointToken: pending ? nextTokens.checkpointToken : undefined,
-            isComplete,
-            pending: (0, output_schemas_js_1.toPendingStep)(blockedMeta),
-            preferences,
-            nextIntent,
-            nextCall: (0, index_js_2.buildNextCall)({ continueToken: pending ? nextTokens.continueToken : undefined, isComplete, pending: blockedMeta, retryContinueToken }),
-            blockers,
-            retryable,
-            retryContinueToken,
-            validation,
-            assessmentFollowup,
-        }))));
+        return nextTokensMint.andThen((nextTokens) => retryContinueMint.andThen((retryContinueToken) => {
+            const out = (0, assert_output_js_1.assertOutput)({
+                kind: 'blocked',
+                continueToken: pending ? nextTokens.continueToken : undefined,
+                checkpointToken: pending ? nextTokens.checkpointToken : undefined,
+                isComplete,
+                pending: (0, output_schemas_js_1.toPendingStep)(blockedMeta),
+                preferences,
+                nextIntent,
+                nextCall: (0, index_js_2.buildNextCall)({ continueToken: pending ? nextTokens.continueToken : undefined, isComplete, pending: blockedMeta, retryContinueToken }),
+                blockers,
+                retryable,
+                retryContinueToken,
+                validation,
+                assessmentFollowup,
+            }, assert_output_js_1.assertContinueTokenPresence);
+            return (0, neverthrow_1.okAsync)(out);
+        }));
     }
     let okMeta = null;
     if (pending) {
@@ -109,29 +115,38 @@ function buildAdvancedReplayResponse(args) {
             runId: (0, index_js_1.asRunId)(String(runId)),
             nodeId: (0, index_js_1.asNodeId)(String(toNodeIdBranded)),
             rehydrateOnly: false,
+            precomputedIndex: args.precomputedIndex,
         });
         if (result.isErr()) {
             return (0, neverthrow_1.errAsync)({ kind: 'prompt_render_failed', message: result.error.message });
         }
         okMeta = result.value;
     }
-    const preferences = (0, v2_execution_helpers_js_1.derivePreferencesOrDefault)({ truth, runId, nodeId: toNodeIdBranded });
+    const preferences = (0, v2_execution_helpers_js_1.derivePreferencesOrDefault)({ truth, runId, nodeId: toNodeIdBranded, precomputedIndex: args.precomputedIndex });
     const nextIntent = (0, v2_state_conversion_js_1.deriveNextIntent)({ rehydrateOnly: false, isComplete, pending: okMeta });
     const stepContext = buildStepContext(truth.events, fromNodeId);
-    return nextTokensMint.andThen((nextTokens) => (0, neverthrow_1.okAsync)(output_schemas_js_1.V2ContinueWorkflowOutputSchema.parse({
-        kind: 'ok',
-        continueToken: pending ? nextTokens.continueToken : undefined,
-        checkpointToken: pending ? nextTokens.checkpointToken : undefined,
-        isComplete,
-        pending: (0, output_schemas_js_1.toPendingStep)(okMeta),
-        preferences,
-        nextIntent,
-        nextCall: (0, index_js_2.buildNextCall)({ continueToken: pending ? nextTokens.continueToken : undefined, isComplete, pending: okMeta }),
-        stepContext,
-    })));
+    return nextTokensMint.andThen((nextTokens) => {
+        const out = (0, assert_output_js_1.assertOutput)({
+            kind: 'ok',
+            continueToken: pending ? nextTokens.continueToken : undefined,
+            checkpointToken: pending ? nextTokens.checkpointToken : undefined,
+            isComplete,
+            pending: (0, output_schemas_js_1.toPendingStep)(okMeta),
+            preferences,
+            nextIntent,
+            nextCall: (0, index_js_2.buildNextCall)({ continueToken: pending ? nextTokens.continueToken : undefined, isComplete, pending: okMeta }),
+            stepContext,
+        }, assert_output_js_1.assertContinueTokenPresence);
+        return (0, neverthrow_1.okAsync)(out);
+    });
 }
 function buildStepContext(events, completedNodeId) {
-    const projection = (0, assessments_js_1.projectAssessmentsV2)(events);
+    const sortedEventsRes = (0, sorted_event_log_js_1.asSortedEventLog)(events);
+    if (sortedEventsRes.isErr()) {
+        console.warn(`[workrail:replay] stepContext events unsorted for node '${String(completedNodeId)}' — stepContext will be absent: ${sortedEventsRes.error.message}`);
+        return undefined;
+    }
+    const projection = (0, assessments_js_1.projectAssessmentsV2)(sortedEventsRes.value);
     if (projection.isErr()) {
         console.warn(`[workrail:replay] stepContext projection failed for node '${String(completedNodeId)}' — stepContext will be absent: ${projection.error.message}`);
         return undefined;
@@ -160,7 +175,8 @@ function replayFromRecordedAdvance(args) {
         });
     }
     const toNodeId = (0, index_js_1.asNodeId)(String(recordedEvent.data.outcome.toNodeId));
-    const toNode = truth.events.find((e) => e.kind === constants_js_1.EVENT_KIND.NODE_CREATED && e.scope?.nodeId === String(toNodeId));
+    const toNode = args.precomputedIndex?.nodeCreatedByNodeId.get(String(toNodeId))
+        ?? truth.events.find((e) => e.kind === constants_js_1.EVENT_KIND.NODE_CREATED && e.scope?.nodeId === String(toNodeId));
     if (!toNode) {
         return (0, neverthrow_1.errAsync)({
             kind: 'invariant_violation',
@@ -191,6 +207,7 @@ function replayFromRecordedAdvance(args) {
             sha256,
             aliasStore,
             entropy,
+            precomputedIndex: args.precomputedIndex,
         });
     });
 }

package/dist/mcp/handlers/v2-execution/start.d.ts

@@ -1,6 +1,5 @@
 import type { V2ToolContext } from '../../types.js';
-import { V2StartWorkflowOutputSchema } from '../../output-schemas.js';
-import { createWorkflow } from '../../../types/workflow.js';
+import type { V2StartWorkflowOutputSchema } from '../../output-schemas.js';
 import type { DomainEventV1 } from '../../../v2/durable-core/schemas/session/index.js';
 import { type SessionId, type RunId, type NodeId, type WorkflowHash } from '../../../v2/durable-core/ids/index.js';
 import type { Sha256PortV2 } from '../../../v2/ports/sha256.port.js';
@@ -26,7 +25,7 @@ export declare function loadAndPinWorkflow(args: {
 }): RA<{
     readonly workflow: import('../../../types/workflow.js').Workflow;
     readonly workflowHash: WorkflowHash;
-    readonly pinnedWorkflow: ReturnType<typeof createWorkflow>;
+    readonly pinnedWorkflow: import('../../../types/workflow.js').Workflow;
     readonly firstStep: {
         readonly id: string;
     };
@@ -45,6 +44,7 @@ export declare function buildInitialEvents(args: {
     readonly idFactory: {
         readonly mintEventId: () => string;
     };
+    readonly goal: string;
 }): readonly DomainEventV1[];
 export declare function mintStartTokens(args: {
     readonly sessionId: SessionId;