@aikirun/task 0.5.3 → 0.6.0

package/dist/index.d.ts

@@ -22,6 +22,15 @@ type PathFromObjectInternal<T, IncludeArrayKeys extends boolean> = And<[
 type ExtractObjectType<T> = T extends object ? T : never;
 type TypeOfValueAtPath<T extends object, Path extends PathFromObject<T>> = Path extends keyof T ? T[Path] : Path extends `${infer First}.${infer Rest}` ? First extends keyof T ? undefined extends T[First] ? Rest extends PathFromObject<ExtractObjectType<T[First]>> ? TypeOfValueAtPath<ExtractObjectType<T[First]>, Rest> | undefined : never : Rest extends PathFromObject<ExtractObjectType<T[First]>> ? TypeOfValueAtPath<ExtractObjectType<T[First]>, Rest> : never : never : never;
 
+interface Schema<Data> {
+    parse: (data: unknown) => Data;
+}
+interface EventDefinition<Data> {
+    _type: Data;
+    schema?: Schema<Data>;
+}
+type EventsDefinition = Record<string, EventDefinition<unknown>>;
+
 /**
  * Defines a durable task with deterministic execution and automatic retries.
  *
@@ -82,7 +91,7 @@ interface TaskBuilder<Input, Output> {
 interface Task<Input, Output> {
     id: TaskId;
     with(): TaskBuilder<Input, Output>;
-    start: <WorkflowInput, WorkflowOutput>(run: WorkflowRunContext<WorkflowInput, WorkflowOutput>, ...args: Input extends null ? [] : [Input]) => Promise<Output>;
+    start: <WorkflowInput, WorkflowOutput>(run: WorkflowRunContext<WorkflowInput, WorkflowOutput, EventsDefinition>, ...args: Input extends null ? [] : [Input]) => Promise<Output>;
 }
 
 export { type Task, type TaskParams, task };

package/dist/index.js

@@ -142,6 +142,7 @@ function getRetryParams(attempts, strategy) {
 // task.ts
 import { INTERNAL } from "@aikirun/types/symbols";
 import { TaskFailedError } from "@aikirun/types/task";
+import { WorkflowRunSuspendedError } from "@aikirun/types/workflow-run";
 function task(params) {
   return new TaskImpl(params);
 }
@@ -171,22 +172,25 @@ var TaskImpl = class _TaskImpl {
     if (taskState.status === "completed") {
       return taskState.output;
     }
+    if (taskState.status === "failed") {
+      throw new TaskFailedError(path, taskState.attempts, taskState.error.message);
+    }
     const logger = run.logger.child({
       "aiki.component": "task-execution",
       "aiki.taskPath": path
     });
     let attempts = 0;
     const retryStrategy = this.params.opts?.retry ?? { type: "never" };
-    if ("attempts" in taskState) {
+    if (taskState.status !== "none") {
       this.assertRetryAllowed(path, taskState.attempts, retryStrategy, logger);
-      logger.warn("Retrying task", {
+      logger.debug("Retrying task", {
         "aiki.attempts": taskState.attempts,
         "aiki.taskStatus": taskState.status
       });
       attempts = taskState.attempts;
     }
-    if (taskState.status === "failed") {
-      await this.delayIfNecessary(taskState);
+    if (taskState.status === "awaiting_retry" && handle.run.state.status === "running") {
+      throw new WorkflowRunSuspendedError(run.id);
     }
     attempts++;
     logger.info("Starting task", { "aiki.attempts": attempts });
@@ -199,37 +203,41 @@ var TaskImpl = class _TaskImpl {
   async tryExecuteTask(run, input, path, retryStrategy, currentAttempt, logger) {
     let attempts = currentAttempt;
     while (true) {
-      const attemptedAt = Date.now();
       try {
         const output = await this.params.handler(input);
         return { output, lastAttempt: attempts };
       } catch (error) {
         const serializableError = createSerializableError(error);
-        const taskFailedState = {
-          status: "failed",
-          reason: serializableError.message,
-          attempts,
-          attemptedAt,
-          error: serializableError
-        };
         const retryParams = getRetryParams(attempts, retryStrategy);
         if (!retryParams.retriesLeft) {
-          await run[INTERNAL].handle[INTERNAL].transitionTaskState(path, taskFailedState);
           logger.error("Task failed", {
             "aiki.attempts": attempts,
-            "aiki.reason": taskFailedState.reason
+            "aiki.reason": serializableError.message
+          });
+          await run[INTERNAL].handle[INTERNAL].transitionTaskState(path, {
+            status: "failed",
+            attempts,
+            error: serializableError
           });
-          throw new TaskFailedError(path, attempts, taskFailedState.reason);
+          throw new TaskFailedError(path, attempts, serializableError.message);
         }
-        const nextAttemptAt = Date.now() + retryParams.delayMs;
-        await run[INTERNAL].handle[INTERNAL].transitionTaskState(path, { ...taskFailedState, nextAttemptAt });
-        logger.debug("Task failed. Retrying", {
+        logger.debug("Task failed. It will be retried", {
           "aiki.attempts": attempts,
-          "aiki.nextAttemptAt": nextAttemptAt,
-          "aiki.reason": taskFailedState.reason
+          "aiki.nextAttemptInMs": retryParams.delayMs,
+          "aiki.reason": serializableError.message
         });
-        await delay(retryParams.delayMs);
-        attempts++;
+        if (retryParams.delayMs <= run[INTERNAL].options.spinThresholdMs) {
+          await delay(retryParams.delayMs);
+          attempts++;
+          continue;
+        }
+        await run[INTERNAL].handle[INTERNAL].transitionTaskState(path, {
+          status: "awaiting_retry",
+          attempts,
+          error: serializableError,
+          nextAttemptInMs: retryParams.delayMs
+        });
+        throw new WorkflowRunSuspendedError(run.id);
       }
     }
   }
@@ -242,15 +250,6 @@ var TaskImpl = class _TaskImpl {
       throw new TaskFailedError(path, attempts, "Task retry not allowed");
     }
   }
-  async delayIfNecessary(taskState) {
-    if (taskState.nextAttemptAt !== void 0) {
-      const now = Date.now();
-      const remainingDelay = Math.max(0, taskState.nextAttemptAt - now);
-      if (remainingDelay > 0) {
-        await delay(remainingDelay);
-      }
-    }
-  }
   async getPath(input) {
     const inputHash = await sha256(stableStringify(input));
     const taskPath = this.params.opts?.idempotencyKey ? `${this.id}/${inputHash}/${this.params.opts.idempotencyKey}` : `${this.id}/${inputHash}`;

package/package.json

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