@aikirun/workflow 0.8.0 → 0.9.0

package/dist/index.d.ts

@@ -281,6 +281,7 @@ declare class WorkflowVersionImpl<Input, Output, AppContext, TEventsDefinition e
     private handler;
     private tryExecuteWorkflow;
     private assertRetryAllowed;
+    private parse;
     private createFailedState;
     private createAwaitingRetryState;
 }

package/dist/index.js

@@ -878,19 +878,7 @@ var WorkflowVersionImpl = class _WorkflowVersionImpl {
     parentRunHandle[INTERNAL5].assertExecutionAllowed();
     const { client } = parentRunHandle[INTERNAL5];
     const inputRaw = isNonEmptyArray(args) ? args[0] : void 0;
-    let input = inputRaw;
-    if (this.params.schema?.input) {
-      try {
-        input = this.params.schema.input.parse(inputRaw);
-      } catch (error) {
-        await parentRunHandle[INTERNAL5].transitionState({
-          status: "failed",
-          cause: "self",
-          error: createSerializableError(error)
-        });
-        throw new WorkflowRunFailedError2(parentRun.id, parentRunHandle.run.attempts);
-      }
-    }
+    const input = await this.parse(parentRunHandle, this.params.schema?.input, inputRaw);
     const inputHash = await hashInput(input);
     const reference = this.params.opts?.reference;
     const path = getWorkflowRunPath(this.name, this.versionId, reference?.id ?? inputHash);
@@ -904,6 +892,9 @@ var WorkflowVersionImpl = class _WorkflowVersionImpl {
         parentRun.logger
       );
       const { run: existingRun } = await client.api.workflowRun.getByIdV1({ id: existingRunInfo.id });
+      if (existingRun.state.status === "completed") {
+        await this.parse(parentRunHandle, this.params.schema?.output, existingRun.state.output);
+      }
       const logger2 = parentRun.logger.child({
         "aiki.childWorkflowName": existingRun.name,
         "aiki.childWorkflowVersionId": existingRun.versionId,
@@ -985,29 +976,16 @@ var WorkflowVersionImpl = class _WorkflowVersionImpl {
     logger.info("Workflow complete");
   }
   async tryExecuteWorkflow(input, run, context, retryStrategy) {
+    const { handle } = run[INTERNAL5];
     while (true) {
       try {
         const outputRaw = await this.params.handler(run, input, context);
-        let output = outputRaw;
-        if (this.params.schema?.output) {
-          try {
-            output = this.params.schema.output.parse(outputRaw);
-          } catch (error) {
-            const { handle } = run[INTERNAL5];
-            await handle[INTERNAL5].transitionState({
-              status: "failed",
-              cause: "self",
-              error: createSerializableError(error)
-            });
-            throw new WorkflowRunFailedError2(run.id, handle.run.attempts);
-          }
-        }
+        const output = await this.parse(handle, this.params.schema?.output, outputRaw);
         return output;
       } catch (error) {
         if (error instanceof WorkflowRunSuspendedError4 || error instanceof WorkflowRunFailedError2 || error instanceof WorkflowRunConflictError5) {
           throw error;
         }
-        const { handle } = run[INTERNAL5];
         const attempts = handle.run.attempts;
         const retryParams = getRetryParams(attempts, retryStrategy);
         if (!retryParams.retriesLeft) {
@@ -1052,6 +1030,21 @@ var WorkflowVersionImpl = class _WorkflowVersionImpl {
       throw error;
     }
   }
+  async parse(handle, schema, data) {
+    if (!schema) {
+      return data;
+    }
+    try {
+      return schema.parse(data);
+    } catch (error) {
+      await handle[INTERNAL5].transitionState({
+        status: "failed",
+        cause: "self",
+        error: createSerializableError(error)
+      });
+      throw new WorkflowRunFailedError2(handle.run.id, handle.run.attempts);
+    }
+  }
   createFailedState(error) {
     if (error instanceof TaskFailedError) {
       return {

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@aikirun/workflow",
-	"version": "0.8.0",
+	"version": "0.9.0",
 	"description": "Workflow SDK for Aiki - define durable workflows with tasks, sleeps, waits, and event handling",
 	"type": "module",
 	"main": "./dist/index.js",
@@ -18,7 +18,7 @@
 		"build": "tsup"
 	},
 	"dependencies": {
-		"@aikirun/types": "0.8.0"
+		"@aikirun/types": "0.9.0"
 	},
 	"publishConfig": {
 		"access": "public"