@trigger.dev/core 3.0.0-beta.14 → 3.0.0-beta.15

package/dist/v3/index.mjs

@@ -1,5 +1,5 @@
 import { z } from 'zod';
-import { SpanStatusCode, trace, propagation, context, DiagLogLevel, diag, DiagConsoleLogger, SpanKind } from '@opentelemetry/api';
+import { SpanStatusCode, propagation, context, trace, DiagLogLevel, diag, DiagConsoleLogger, SpanKind } from '@opentelemetry/api';
 import { fromZodError } from 'zod-validation-error';
 import { AsyncLocalStorage } from 'node:async_hooks';
 import { io } from 'socket.io-client';
@@ -125,7 +125,8 @@ var TaskRun = z.object({
   context: z.any(),
   tags: z.array(z.string()),
   isTest: z.boolean().default(false),
-  createdAt: z.coerce.date()
+  createdAt: z.coerce.date(),
+  idempotencyKey: z.string().optional()
 });
 var TaskRunExecutionTask = z.object({
   id: z.string(),
@@ -2061,7 +2062,8 @@ var SemanticInternalAttributes = {
   RETRY_AT: "retry.at",
   RETRY_DELAY: "retry.delay",
   RETRY_COUNT: "retry.count",
-  LINK_TITLE: "$link.title"
+  LINK_TITLE: "$link.title",
+  IDEMPOTENCY_KEY: "ctx.run.idempotencyKey"
 };
 
 // src/v3/tasks/taskContextManager.ts
@@ -2120,7 +2122,8 @@ var _TaskContextManager = class _TaskContextManager {
         [SemanticInternalAttributes.RUN_IS_TEST]: this.ctx.run.isTest,
         [SemanticInternalAttributes.ORGANIZATION_SLUG]: this.ctx.organization.slug,
         [SemanticInternalAttributes.ORGANIZATION_NAME]: this.ctx.organization.name,
-        [SemanticInternalAttributes.BATCH_ID]: this.ctx.batch?.id
+        [SemanticInternalAttributes.BATCH_ID]: this.ctx.batch?.id,
+        [SemanticInternalAttributes.IDEMPOTENCY_KEY]: this.ctx.run.idempotencyKey
       };
     }
     return {};
@@ -2211,6 +2214,27 @@ var _ApiClient = class _ApiClient {
     this.accessToken = accessToken;
     this.baseUrl = baseUrl.replace(/\/$/, "");
   }
+  async getRunResult(runId) {
+    try {
+      return await zodfetch(TaskRunExecutionResult, `${this.baseUrl}/api/v1/runs/${runId}/result`, {
+        method: "GET",
+        headers: __privateMethod(this, _getHeaders, getHeaders_fn).call(this, false)
+      }, zodFetchOptions);
+    } catch (error) {
+      if (error instanceof APIError) {
+        if (error.status === 404) {
+          return void 0;
+        }
+      }
+      throw error;
+    }
+  }
+  async getBatchResults(batchId) {
+    return await zodfetch(BatchTaskRunExecutionResult, `${this.baseUrl}/api/v1/batches/${batchId}/results`, {
+      method: "GET",
+      headers: __privateMethod(this, _getHeaders, getHeaders_fn).call(this, false)
+    }, zodFetchOptions);
+  }
   triggerTask(taskId, body, options) {
     return zodfetch(TriggerTaskResponse, `${this.baseUrl}/api/v1/tasks/${taskId}/trigger`, {
       method: "POST",
@@ -3456,12 +3480,23 @@ __name(_NoopTaskLogger, "NoopTaskLogger");
 var NoopTaskLogger = _NoopTaskLogger;
 function safeJsonProcess(value) {
   try {
-    return JSON.parse(JSON.stringify(value));
+    return JSON.parse(JSON.stringify(value, jsonErrorReplacer));
   } catch {
     return value;
   }
 }
 __name(safeJsonProcess, "safeJsonProcess");
+function jsonErrorReplacer(key, value) {
+  if (value instanceof Error) {
+    return {
+      name: value.name,
+      message: value.message,
+      stack: value.stack
+    };
+  }
+  return value;
+}
+__name(jsonErrorReplacer, "jsonErrorReplacer");
 
 // src/v3/logger/index.ts
 var API_NAME3 = "logger";
@@ -3734,10 +3769,9 @@ var _DevRuntimeManager = class _DevRuntimeManager {
       this._pendingCompletionNotifications.delete(params.id);
       return pendingCompletion;
     }
-    const promise = new Promise((resolve, reject) => {
+    const promise = new Promise((resolve) => {
       this._taskWaits.set(params.id, {
-        resolve,
-        reject
+        resolve
       });
     });
     return await promise;
@@ -3754,16 +3788,11 @@ var _DevRuntimeManager = class _DevRuntimeManager {
         const pendingCompletion = this._pendingCompletionNotifications.get(runId);
         if (pendingCompletion) {
           this._pendingCompletionNotifications.delete(runId);
-          if (pendingCompletion.ok) {
-            resolve(pendingCompletion);
-          } else {
-            reject(pendingCompletion);
-          }
+          resolve(pendingCompletion);
           return;
         }
         this._taskWaits.set(runId, {
-          resolve,
-          reject
+          resolve
         });
       });
     }));
@@ -3779,11 +3808,7 @@ var _DevRuntimeManager = class _DevRuntimeManager {
       this._pendingCompletionNotifications.set(execution.run.id, completion);
       return;
     }
-    if (completion.ok) {
-      wait.resolve(completion);
-    } else {
-      wait.reject(completion);
-    }
+    wait.resolve(completion);
     this._taskWaits.delete(execution.run.id);
   }
 };
@@ -3840,16 +3865,17 @@ var _ProdRuntimeManager = class _ProdRuntimeManager {
     return this.waitForDuration(date.getTime() - Date.now());
   }
   async waitForTask(params) {
-    const promise = new Promise((resolve, reject) => {
+    const promise = new Promise((resolve) => {
       this._taskWaits.set(params.id, {
-        resolve,
-        reject
+        resolve
       });
     });
     await this.ipc.send("WAIT_FOR_TASK", {
       friendlyId: params.id
     });
-    return await promise;
+    const result = await promise;
+    clock.reset();
+    return result;
   }
   async waitForBatch(params) {
     if (!params.runs.length) {
@@ -3861,8 +3887,7 @@ var _ProdRuntimeManager = class _ProdRuntimeManager {
     const promise = Promise.all(params.runs.map((runId) => {
       return new Promise((resolve, reject) => {
         this._taskWaits.set(runId, {
-          resolve,
-          reject
+          resolve
         });
       });
     }));
@@ -3871,6 +3896,7 @@ var _ProdRuntimeManager = class _ProdRuntimeManager {
       runFriendlyIds: params.runs
     });
     const results = await promise;
+    clock.reset();
     return {
       id: params.id,
       items: results
@@ -3881,11 +3907,7 @@ var _ProdRuntimeManager = class _ProdRuntimeManager {
     if (!wait) {
       return;
     }
-    if (completion.ok) {
-      wait.resolve(completion);
-    } else {
-      wait.reject(completion);
-    }
+    wait.resolve(completion);
     this._taskWaits.delete(execution.run.id);
   }
   get waitThresholdInMs() {
@@ -4774,7 +4796,7 @@ var _TaskExecutor = class _TaskExecutor {
               }
               return {
                 ok: true,
-                id: execution.attempt.id,
+                id: execution.run.id,
                 output: finalOutput.data,
                 outputType: finalOutput.dataType
               };
@@ -4782,7 +4804,7 @@ var _TaskExecutor = class _TaskExecutor {
               recordSpanException(span, stringifyError);
               return {
                 ok: false,
-                id: execution.attempt.id,
+                id: execution.run.id,
                 error: {
                   type: "INTERNAL_ERROR",
                   code: TaskRunErrorCodes.TASK_OUTPUT_ERROR,
@@ -4795,7 +4817,7 @@ var _TaskExecutor = class _TaskExecutor {
               const handleErrorResult = await __privateMethod(this, _handleError, handleError_fn).call(this, execution, runError, parsedPayload, ctx);
               recordSpanException(span, handleErrorResult.error ?? runError);
               return {
-                id: execution.attempt.id,
+                id: execution.run.id,
                 ok: false,
                 error: handleErrorResult.error ? parseError(handleErrorResult.error) : parseError(runError),
                 retry: handleErrorResult.status === "retry" ? handleErrorResult.retry : void 0,
@@ -4805,7 +4827,7 @@ var _TaskExecutor = class _TaskExecutor {
               recordSpanException(span, handleErrorError);
               return {
                 ok: false,
-                id: execution.attempt.id,
+                id: execution.run.id,
                 error: {
                   type: "INTERNAL_ERROR",
                   code: TaskRunErrorCodes.HANDLE_ERROR_ERROR,