@exaudeus/workrail 3.15.0 → 3.17.0

package/dist/mcp/handlers/v2-advance-core/input-validation.js

@@ -5,14 +5,28 @@ const neverthrow_1 = require("neverthrow");
 const workflow_js_1 = require("../../../types/workflow.js");
 const run_context_js_1 = require("../../../v2/projections/run-context.js");
 const preferences_js_1 = require("../../../v2/projections/preferences.js");
+const sorted_event_log_js_1 = require("../../../v2/durable-core/sorted-event-log.js");
 const context_merge_js_1 = require("../../../v2/durable-core/domain/context-merge.js");
 const constants_js_1 = require("../../../v2/durable-core/constants.js");
 const assessment_validation_js_1 = require("./assessment-validation.js");
 const assessment_consequences_js_1 = require("./assessment-consequences.js");
 function validateAdvanceInputs(args) {
     const { truth, runId, currentNodeId, inputContext, inputOutput, pinnedWorkflow, pendingStep } = args;
-    const storedContextRes = (0, run_context_js_1.projectRunContextV2)(truth.events);
-    const storedContext = storedContextRes.isOk() ? storedContextRes.value.byRunId[String(runId)]?.context : undefined;
+    let sortedEvents;
+    let storedContext;
+    if (args.precomputedIndex) {
+        sortedEvents = args.precomputedIndex.sortedEvents;
+        storedContext = args.precomputedIndex.runContextByRunId.get(String(runId));
+    }
+    else {
+        const sortedEventsRes = (0, sorted_event_log_js_1.asSortedEventLog)(truth.events);
+        if (sortedEventsRes.isErr()) {
+            return (0, neverthrow_1.err)({ kind: 'invariant_violation', message: sortedEventsRes.error.message });
+        }
+        sortedEvents = sortedEventsRes.value;
+        const storedContextRes = (0, run_context_js_1.projectRunContextV2)(sortedEvents);
+        storedContext = storedContextRes.isOk() ? storedContextRes.value.byRunId[String(runId)]?.context : undefined;
+    }
     const inputContextObj = inputContext && typeof inputContext === 'object' && inputContext !== null && !Array.isArray(inputContext)
         ? inputContext
         : undefined;
@@ -39,14 +53,23 @@ function validateAdvanceInputs(args) {
     const notesOptional = outputContract !== undefined ||
         (step !== null && step !== undefined && 'notesOptional' in step && step.notesOptional === true);
     const parentByNodeId = {};
-    for (const e of truth.events) {
-        if (e.kind !== constants_js_1.EVENT_KIND.NODE_CREATED)
-            continue;
-        if (e.scope?.runId !== String(runId))
-            continue;
-        parentByNodeId[String(e.scope.nodeId)] = e.data.parentNodeId;
+    if (args.precomputedIndex) {
+        for (const [nodeId, evt] of args.precomputedIndex.nodeCreatedByNodeId) {
+            if (evt.scope.runId === String(runId)) {
+                parentByNodeId[nodeId] = evt.data.parentNodeId ?? null;
+            }
+        }
+    }
+    else {
+        for (const e of truth.events) {
+            if (e.kind !== constants_js_1.EVENT_KIND.NODE_CREATED)
+                continue;
+            if (e.scope?.runId !== String(runId))
+                continue;
+            parentByNodeId[String(e.scope.nodeId)] = e.data.parentNodeId;
+        }
     }
-    const prefs = (0, preferences_js_1.projectPreferencesV2)(truth.events, parentByNodeId);
+    const prefs = (0, preferences_js_1.projectPreferencesV2)(sortedEvents, parentByNodeId);
     const effectivePrefs = prefs.isOk() ? prefs.value.byNodeId[String(currentNodeId)]?.effective : undefined;
     const rawAutonomy = effectivePrefs?.autonomy ?? 'guided';
     const rawRiskPolicy = effectivePrefs?.riskPolicy ?? 'conservative';

package/dist/mcp/handlers/v2-advance-core/outcome-blocked.d.ts

@@ -1,4 +1,5 @@
 import { ResultAsync as RA } from 'neverthrow';
+import type { SessionIndex } from '../../../v2/durable-core/session-index.js';
 import type { ExecutionSnapshotFileV1 } from '../../../v2/durable-core/schemas/execution-snapshot/index.js';
 import type { SessionEventLogStoreError } from '../../../v2/ports/session-event-log-store.port.js';
 import type { SnapshotStoreError } from '../../../v2/ports/snapshot-store.port.js';
@@ -15,4 +16,5 @@ export declare function buildBlockedOutcome(args: {
     readonly v: ValidatedAdvanceInputs;
     readonly lock: WithHealthySessionLock;
     readonly ports: AdvanceCorePorts;
+    readonly lockedIndex: SessionIndex;
 }): RA<void, InternalError | SessionEventLogStoreError | SnapshotStoreError>;

package/dist/mcp/handlers/v2-advance-core/outcome-blocked.js

@@ -102,7 +102,7 @@ function buildBlockedOutcome(args) {
     return snapshotStore.putExecutionSnapshotV1(blockedSnapshotRes.value).andThen((blockedSnapshotRef) => {
         return (0, event_builders_js_1.buildAndAppendPlan)({
             kind: 'blocked',
-            truth, sessionId, runId, currentNodeId, attemptId, workflowHash,
+            truth, lockedIndex: args.lockedIndex, sessionId, runId, currentNodeId, attemptId, workflowHash,
             extraEventsToAppend, blockers: blockersRes.value, snapshotRef: blockedSnapshotRef,
             outputsToAppend,
             sessionStore, idFactory, lock,

package/dist/mcp/handlers/v2-advance-core/outcome-success.d.ts

@@ -1,4 +1,5 @@
 import { ResultAsync as RA } from 'neverthrow';
+import type { SessionIndex } from '../../../v2/durable-core/session-index.js';
 import type { SessionEventLogStoreError } from '../../../v2/ports/session-event-log-store.port.js';
 import type { SnapshotStoreError } from '../../../v2/ports/snapshot-store.port.js';
 import type { WithHealthySessionLock } from '../../../v2/durable-core/ids/with-healthy-session-lock.js';
@@ -12,4 +13,5 @@ export declare function buildSuccessOutcome(args: {
     readonly v: ValidatedAdvanceInputs;
     readonly lock: WithHealthySessionLock;
     readonly ports: AdvanceCorePorts;
+    readonly lockedIndex: SessionIndex;
 }): RA<void, InternalError | SessionEventLogStoreError | SnapshotStoreError>;

package/dist/mcp/handlers/v2-advance-core/outcome-success.js

@@ -149,7 +149,7 @@ function buildSuccessOutcome(args) {
         const outputsToAppend = [...notesOutputs, ...artifactOutputsRes.value];
         return (0, event_builders_js_1.buildAndAppendPlan)({
             kind: 'advanced',
-            truth, sessionId, runId, currentNodeId, attemptId, workflowHash,
+            truth, lockedIndex: args.lockedIndex, sessionId, runId, currentNodeId, attemptId, workflowHash,
             extraEventsToAppend, toNodeKind: successNodeKind(mode),
             snapshotRef: newSnapshotRef, outputsToAppend,
             sessionStore, idFactory, lock,

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.d.ts

@@ -28,4 +28,5 @@ export declare function handleAdvanceIntent(args: {
     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 cleanResponseFormat?: boolean;
 }): RA<z.infer<typeof V2ContinueWorkflowOutputSchema>, ContinueWorkflowError>;

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 { input, sessionId, runId, nodeId, attemptId, workflowHashRef, truth, gate, sessionStore, snapshotStore, pinnedStore, tokenCodecPorts, idFactory, sha256, aliasStore, entropy, cleanResponseFormat } = 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,
@@ -91,13 +101,22 @@ function handleAdvanceIntent(args) {
                 tokenCodecPorts,
                 aliasStore,
                 entropy,
+                cleanResponseFormat,
             });
         }
         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 +133,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 +177,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 +187,7 @@ function handleAdvanceIntent(args) {
             return (0, replay_js_1.replayFromRecordedAdvance)({
                 recordedEvent,
                 truth: truth2,
+                precomputedIndex: res.precomputedIndex,
                 sessionId,
                 runId,
                 nodeId,
@@ -168,6 +199,7 @@ function handleAdvanceIntent(args) {
                 tokenCodecPorts,
                 aliasStore,
                 entropy,
+                cleanResponseFormat,
             });
         });
     });

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';
@@ -27,4 +27,5 @@ export declare function handleRehydrateIntent(args: {
     readonly aliasStore: import('../../../v2/ports/token-alias-store.port.js').TokenAliasStorePortV2;
     readonly entropy: import('../../../v2/ports/random-entropy.port.js').RandomEntropyPortV2;
     readonly resolvedRootUris?: readonly string[];
+    readonly cleanResponseFormat?: boolean;
 }): RA<RehydrateResult, ContinueWorkflowError>;

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,8 +18,9 @@ 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 { input, sessionId, runId, nodeId, workflowHashRef, truth, tokenCodecPorts, pinnedStore, snapshotStore, idFactory, aliasStore, entropy, resolvedRootUris, cleanResponseFormat } = args;
     const runStarted = truth.events.find((e) => e.kind === constants_js_1.EVENT_KIND.RUN_STARTED && e.scope.runId === String(runId));
     const workflowId = runStarted?.data.workflowId;
     if (!runStarted || typeof workflowId !== 'string' || workflowId.trim() === '') {
@@ -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),
@@ -124,6 +124,7 @@ function handleRehydrateIntent(args) {
                     runId: (0, index_js_1.asRunId)(String(runId)),
                     nodeId: (0, index_js_1.asNodeId)(String(nodeId)),
                     rehydrateOnly: true,
+                    cleanResponseFormat,
                 });
                 if (metaRes.isErr()) {
                     return (0, neverthrow_1.errAsync)({
@@ -138,7 +139,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 +148,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/index.js

@@ -75,6 +75,7 @@ function loadAndRehydrate(args) {
         aliasStore: tokenAliasStore,
         entropy,
         resolvedRootUris: args.ctx.v2.resolvedRootUris,
+        cleanResponseFormat: args.ctx.featureFlags?.isEnabled('cleanResponseFormat') ?? false,
     }));
 }
 function executeContinueWorkflow(input, ctx) {
@@ -140,6 +141,7 @@ function executeContinueWorkflow(input, ctx) {
                     sha256,
                     aliasStore: tokenAliasStore,
                     entropy,
+                    cleanResponseFormat: ctx.featureFlags?.isEnabled('cleanResponseFormat') ?? false,
                 }))
                     .map((response) => ({ response }));
             });

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,15 @@ 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;
+    readonly cleanResponseFormat?: boolean;
 }): RA<z.infer<typeof V2ContinueWorkflowOutputSchema>, ContinueWorkflowError>;
 export declare function replayFromRecordedAdvance(args: {
     readonly recordedEvent: Extract<DomainEventV1, {
@@ -35,10 +37,12 @@ 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;
+    readonly cleanResponseFormat?: boolean;
 }): RA<z.infer<typeof V2ContinueWorkflowOutputSchema>, ContinueWorkflowError>;