@aikirun/task 0.9.2 → 0.10.1

package/dist/index.d.ts

@@ -1,7 +1,7 @@
 import { Serializable } from '@aikirun/types/serializable';
 import { TaskName, TaskOptions } from '@aikirun/types/task';
-import { Schema } from '@aikirun/types/validator';
 import { WorkflowRunContext } from '@aikirun/workflow';
+import { StandardSchemaV1 } from '@standard-schema/spec';
 import { RequireAtLeastOneProp } from '@aikirun/types/utils';
 
 type NonEmptyArray<T> = [T, ...T[]];
@@ -25,7 +25,7 @@ type TypeOfValueAtPath<T extends object, Path extends PathFromObject<T>> = Path
 
 interface EventDefinition<Data> {
     _type: Data;
-    schema?: Schema<Data>;
+    schema?: StandardSchemaV1<Data>;
 }
 type EventsDefinition = Record<string, EventDefinition<unknown>>;
 
@@ -78,8 +78,8 @@ interface TaskParams<Input, Output> {
     handler: (input: Input) => Promise<Output>;
     opts?: TaskOptions;
     schema?: RequireAtLeastOneProp<{
-        input?: Schema<Input>;
-        output?: Schema<Output>;
+        input?: StandardSchemaV1<Input>;
+        output?: StandardSchemaV1<Output>;
     }>;
 }
 interface Task<Input, Output> {

package/dist/index.js

@@ -182,7 +182,7 @@ var TaskImpl = class _TaskImpl {
     const handle = run[INTERNAL].handle;
     handle[INTERNAL].assertExecutionAllowed();
     const inputRaw = isNonEmptyArray(args) ? args[0] : void 0;
-    const input = await this.parse(handle, this.params.schema?.input, inputRaw);
+    const input = await this.parse(handle, this.params.schema?.input, inputRaw, run.logger);
     const inputHash = await hashInput(input);
     const reference = this.params.opts?.reference;
     const path = getTaskPath(this.name, reference?.id ?? inputHash);
@@ -191,7 +191,7 @@ var TaskImpl = class _TaskImpl {
       await this.assertUniqueTaskReferenceId(handle, existingTaskInfo, inputHash, reference, run.logger);
     }
     if (existingTaskInfo?.state.status === "completed") {
-      return this.parse(handle, this.params.schema?.output, existingTaskInfo.state.output);
+      return this.parse(handle, this.params.schema?.output, existingTaskInfo.state.output, run.logger);
     }
     if (existingTaskInfo?.state.status === "failed") {
       const { state } = existingTaskInfo;
@@ -254,7 +254,7 @@ var TaskImpl = class _TaskImpl {
     while (true) {
       try {
         const outputRaw = await this.params.handler(input);
-        const output = await this.parse(handle, this.params.schema?.output, outputRaw);
+        const output = await this.parse(handle, this.params.schema?.output, outputRaw, logger);
         return { output, lastAttempt: attempts };
       } catch (error) {
         if (error instanceof WorkflowRunFailedError || error instanceof WorkflowRunSuspendedError || error instanceof WorkflowRunConflictError) {
@@ -332,20 +332,25 @@ var TaskImpl = class _TaskImpl {
       throw new TaskFailedError(taskId, attempts, "Task retry not allowed");
     }
   }
-  async parse(handle, schema, data) {
+  async parse(handle, schema, data, logger) {
     if (!schema) {
       return data;
     }
-    try {
-      return schema.parse(data);
-    } catch (error) {
-      await handle[INTERNAL].transitionState({
-        status: "failed",
-        cause: "self",
-        error: createSerializableError(error)
-      });
-      throw new WorkflowRunFailedError(handle.run.id, handle.run.attempts);
+    const schemaValidation = schema["~standard"].validate(data);
+    const schemaValidationResult = schemaValidation instanceof Promise ? await schemaValidation : schemaValidation;
+    if (!schemaValidationResult.issues) {
+      return schemaValidationResult.value;
     }
+    logger.error("Invalid task data", { "aiki.issues": schemaValidationResult.issues });
+    await handle[INTERNAL].transitionState({
+      status: "failed",
+      cause: "self",
+      error: {
+        name: "SchemaValidationError",
+        message: JSON.stringify(schemaValidationResult.issues)
+      }
+    });
+    throw new WorkflowRunFailedError(handle.run.id, handle.run.attempts);
   }
 };
 export {

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@aikirun/task",
-	"version": "0.9.2",
+	"version": "0.10.1",
 	"description": "Task SDK for Aiki - define reliable tasks with automatic retries, idempotency, and error handling",
 	"type": "module",
 	"main": "./dist/index.js",
@@ -18,8 +18,9 @@
 		"build": "tsup"
 	},
 	"dependencies": {
-		"@aikirun/types": "0.9.2",
-		"@aikirun/workflow": "0.9.2"
+		"@aikirun/types": "0.10.1",
+		"@aikirun/workflow": "0.10.1",
+		"@standard-schema/spec": "^1.1.0"
 	},
 	"publishConfig": {
 		"access": "public"