@lostgradient/weft 0.24.0 → 0.24.1

package/dist/worker/registry.js

@@ -132,7 +132,7 @@ export class WorkerRegistry {
       return pickFairShare(eligible, this.#fairShareCounts, fairShareKey);
     return pickLeastLoaded(eligible);
   }
-  assignTask(workerId, operationId, visibilityTimeout, fairShareKey, attemptToken) {
+  assignTask(workerId, operationId, visibilityTimeout, fairShareKey, attemptToken, workflowRevision) {
     const deadline = Date.now() + visibilityTimeout, task = {
       operationId,
       workerId,
@@ -144,6 +144,8 @@ export class WorkerRegistry {
       task.fairShareKey = fairShareKey;
       this.#fairShareCounts.increment(workerId, fairShareKey);
     }
+    if (workflowRevision !== void 0)
+      task.workflowRevision = workflowRevision;
     this.#inFlightTasks.set(operationId, task);
     this.taskAssigned(workerId);
   }

package/dist/workers/workflow-runner.d.ts

@@ -42,6 +42,14 @@ export interface WorkflowRunnerContext {
     generators: Map<string, AsyncGenerator>;
     abortControllers: Map<string, AbortController>;
     replayStates: Map<string, WorkerReplayState>;
+    /**
+     * The revision (WFT-20) captured from the `run` message that started this
+     * workflow, keyed by workflow id — a `resume` inbound message never
+     * re-sends it, so `workflow-worker-entry.ts`'s `attachWorkerProtocol` reads
+     * it from here to stamp every outbound message for the workflow's
+     * remaining turns.
+     */
+    workflowRevisions: Map<string, string>;
 }
 export declare function createWorkflowRunnerContext(): WorkflowRunnerContext;
 export declare function handleRunMessage(context: WorkflowRunnerContext, message: {
@@ -53,6 +61,7 @@ export declare function handleRunMessage(context: WorkflowRunnerContext, message
     executionStateOwnerId?: string;
     deadline?: number;
     headers?: [string, string][];
+    workflowRevision?: string;
 }, getWorkflowHandler: (type: string) => ((ctx: WorkerWorkflowContext, input: unknown) => AsyncGenerator) | undefined, postLog?: WorkerLogPoster): Promise<WorkerOutboundMessage>;
 export declare function handleResumeMessage(context: WorkflowRunnerContext, message: {
     workflowId: string;

package/dist/workers/workflow-runner.js

@@ -77,10 +77,15 @@ export function createWorkflowRunnerContext() {
   return {
     generators: new Map,
     abortControllers: new Map,
-    replayStates: new Map
+    replayStates: new Map,
+    workflowRevisions: new Map
   };
 }
 export async function handleRunMessage(context, message, getWorkflowHandler, postLog) {
+  if (message.workflowRevision !== void 0)
+    context.workflowRevisions.set(message.workflowId, message.workflowRevision);
+  else
+    context.workflowRevisions.delete(message.workflowId);
   const handler = getWorkflowHandler(message.workflowType);
   if (!handler)
     return {
@@ -152,8 +157,10 @@ export async function handleCancelMessage(context, message) {
     try {
       await capturedGenerator.return(void 0);
     } catch {}
-  if (context.generators.get(message.workflowId) === capturedGenerator)
+  if (context.generators.get(message.workflowId) === capturedGenerator) {
     context.generators.delete(message.workflowId);
+    context.workflowRevisions.delete(message.workflowId);
+  }
   if (context.abortControllers.get(message.workflowId) === capturedController)
     context.abortControllers.delete(message.workflowId);
   if (context.replayStates.get(message.workflowId) === capturedReplayState)

package/dist/workers/workflow-worker-entry.js

@@ -49,7 +49,9 @@ export function initializeWorkerMessageLoop(workflows) {
   });
 }
 function postOutboundMessage(runnerContext, message, inboundMessage) {
-  const outboundMessage = attachWorkerProtocol(message, inboundMessage);
+  const activeTurnRevision = runnerContext.workflowRevisions.get(inboundMessage.workflowId), outboundMessage = attachWorkerProtocol(runnerContext, message, inboundMessage);
+  if (outboundMessage.type === "completed" || outboundMessage.type === "failed")
+    runnerContext.workflowRevisions.delete(inboundMessage.workflowId);
   try {
     assertWorkerProtocolMessageWithinLimit(outboundMessage, inboundMessage.maxProtocolMessageBytes);
     if (outboundMessage.type === "checkpoint")
@@ -62,7 +64,8 @@ function postOutboundMessage(runnerContext, message, inboundMessage) {
       workflowId: inboundMessage.workflowId,
       error: error instanceof Error ? error.message : String(error),
       failureCategory: "resource",
-      ...inboundMessage.turnId === void 0 ? {} : { turnId: inboundMessage.turnId }
+      ...inboundMessage.turnId === void 0 ? {} : { turnId: inboundMessage.turnId },
+      ...activeTurnRevision === void 0 ? {} : { workflowRevision: activeTurnRevision }
     });
     try {
       workerPostMessage(failedMessage);
@@ -71,10 +74,12 @@ function postOutboundMessage(runnerContext, message, inboundMessage) {
     }
   }
 }
-function attachWorkerProtocol(message, inboundMessage) {
+function attachWorkerProtocol(runnerContext, message, inboundMessage) {
+  const workflowRevision = runnerContext.workflowRevisions.get(inboundMessage.workflowId);
   return {
     ...message,
     protocolVersion: WORKER_PROTOCOL_VERSION,
-    ...inboundMessage.turnId === void 0 ? {} : { turnId: inboundMessage.turnId }
+    ...inboundMessage.turnId === void 0 ? {} : { turnId: inboundMessage.turnId },
+    ...workflowRevision === void 0 ? {} : { workflowRevision }
   };
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lostgradient/weft",
-  "version": "0.24.0",
+  "version": "0.24.1",
   "description": "A Bun-native durable execution engine with pluggable key-value storage.",
   "keywords": [
     "bun",
@@ -221,7 +221,7 @@
     "dev": "bun --watch run src/index.ts",
     "format": "prettier --write \"**/*.{ts,tsx,js,jsx,json,md}\"",
     "format:check": "prettier --check \"**/*.{ts,tsx,js,jsx,json,md}\"",
-    "lint": "oxlint --type-aware --tsconfig ./tsconfig.json --report-unused-disable-directives-severity=error && bun run scripts/check-lint-disables.ts && bun run scripts/check-implementation-file-sizes.ts && bun run scripts/check-import-cycles.ts && bun run scripts/check-internal-imports.ts && bun run scripts/check-engine-internals-field-access.ts && bun run scripts/check-definition-vocabulary.ts && bun run scripts/check-catalog-completeness.ts && bun run scripts/check-catalog-drift.ts && bun run scripts/check-type-ergonomics.ts",
+    "lint": "oxlint --type-aware --tsconfig ./tsconfig.json --report-unused-disable-directives-severity=error && bun run scripts/check-lint-disables.ts && bun run scripts/check-implementation-file-sizes.ts && bun run scripts/check-import-cycles.ts && bun run scripts/check-internal-imports.ts && bun run scripts/check-engine-internals-field-access.ts && bun run scripts/check-revision-keyed-lookups.ts && bun run scripts/check-definition-vocabulary.ts && bun run scripts/check-catalog-completeness.ts && bun run scripts/check-catalog-drift.ts && bun run scripts/check-type-ergonomics.ts",
     "lint:fix": "oxlint --fix --type-aware --tsconfig ./tsconfig.json",
     "markdown-doctests:inventory": "bun run scripts/extract-markdown-doctests.ts inventory",
     "prepack": "bun run scripts/run-gates.ts prepack",