@trigger.dev/sdk 3.0.0-beta.37 → 3.0.0-beta.39

package/dist/v3/index.mjs

@@ -1,8 +1,9 @@
-import { TriggerTracer, SemanticInternalAttributes, runtime, accessoryAttributes, apiClientManager, taskCatalog, taskContext, defaultRetryOptions, calculateNextRetryDelay, defaultFetchRetryOptions, calculateResetAt, eventFilterMatches, flattenAttributes, stringifyIO, logger, conditionallyImportPacket, parsePacket, createErrorTaskError } from '@trigger.dev/core/v3';
+import { TriggerTracer, SemanticInternalAttributes, runtime, accessoryAttributes, usage as usage$1, taskContext, apiClientManager, taskCatalog, TimezonesResult, defaultRetryOptions, calculateNextRetryDelay, defaultFetchRetryOptions, calculateResetAt, eventFilterMatches, flattenAttributes, stringifyIO, logger, conditionallyImportPacket, parsePacket, createErrorTaskError } from '@trigger.dev/core/v3';
 export { ApiError, AuthenticationError, BadRequestError, ConflictError, InternalServerError, NotFoundError, PermissionDeniedError, RateLimitError, UnprocessableEntityError, logger } from '@trigger.dev/core/v3';
 import { trace, context, SpanStatusCode, SpanKind } from '@opentelemetry/api';
 import { SEMATTRS_HTTP_STATUS_CODE, SEMATTRS_HTTP_METHOD, SEMATTRS_HTTP_URL, SEMATTRS_HTTP_HOST, SEMATTRS_HTTP_SCHEME, SEMATTRS_HTTP_RESPONSE_CONTENT_LENGTH, SEMATTRS_MESSAGING_OPERATION, SEMATTRS_MESSAGING_DESTINATION, SEMATTRS_MESSAGING_SYSTEM } from '@opentelemetry/semantic-conventions';
 import { AsyncLocalStorage } from 'node:async_hooks';
+import { zodfetch } from '@trigger.dev/core/v3/zodfetch';
 
 var __defProp = Object.defineProperty;
 var __defNormalProp = (obj, key, value) => key in obj ? __defProp(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
@@ -17,7 +18,7 @@ var __publicField = (obj, key, value) => {
 };
 
 // package.json
-var version = "3.0.0-beta.37";
+var version = "3.0.0-beta.39";
 
 // src/v3/tracer.ts
 var tracer = new TriggerTracer({
@@ -930,6 +931,116 @@ function calculateDurationInMs(options) {
   throw new Error("Invalid options");
 }
 __name(calculateDurationInMs, "calculateDurationInMs");
+var usage = {
+  /**
+  * Get the current running usage of this task run.
+  *
+  * @example
+  *
+  * ```typescript
+  * import { usage, task } from "@trigger.dev/sdk/v3";
+  *
+  * export const myTask = task({
+  *  id: "my-task",
+  *  run: async (payload, { ctx }) => {
+  *   // ... Do a bunch of work
+  *
+  *   const currentUsage = usage.getCurrent();
+  *
+  *   // You have access to the current compute cost and duration up to this point
+  *   console.log("Current attempt compute cost and duration", {
+  *     cost: currentUsage.compute.attempt.costInCents,
+  *     duration: currentUsage.compute.attempt.durationMs,
+  *   });
+  *
+  *   // You also can see the total compute cost and duration up to this point in the run, across all attempts
+  *   console.log("Current total compute cost and duration", {
+  *     cost: currentUsage.compute.total.costInCents,
+  *     duration: currentUsage.compute.total.durationMs,
+  *   });
+  *
+  *   // You can see the base cost of the run, which is the cost of the run before any compute costs
+  *   console.log("Total cost", {
+  *     cost: currentUsage.totalCostInCents,
+  *     baseCost: currentUsage.baseCostInCents,
+  *   });
+  *  },
+  * });
+  * ```
+  */
+  getCurrent: () => {
+    const sample = usage$1.sample();
+    const machine = taskContext.ctx?.machine;
+    const run = taskContext.ctx?.run;
+    if (!sample) {
+      return {
+        compute: {
+          attempt: {
+            costInCents: 0,
+            durationMs: 0
+          },
+          total: {
+            costInCents: run?.costInCents ?? 0,
+            durationMs: run?.durationMs ?? 0
+          }
+        },
+        baseCostInCents: run?.baseCostInCents ?? 0,
+        totalCostInCents: (run?.costInCents ?? 0) + (run?.baseCostInCents ?? 0)
+      };
+    }
+    const currentCostInCents = machine?.centsPerMs ? sample.cpuTime * machine.centsPerMs : 0;
+    return {
+      compute: {
+        attempt: {
+          costInCents: currentCostInCents,
+          durationMs: sample.cpuTime
+        },
+        total: {
+          costInCents: (run?.costInCents ?? 0) + currentCostInCents,
+          durationMs: (run?.durationMs ?? 0) + sample.cpuTime
+        }
+      },
+      baseCostInCents: run?.baseCostInCents ?? 0,
+      totalCostInCents: (run?.costInCents ?? 0) + currentCostInCents + (run?.baseCostInCents ?? 0)
+    };
+  },
+  /**
+  * Measure the cost and duration of a function.
+  *
+  * @example
+  *
+  * ```typescript
+  * import { usage } from "@trigger.dev/sdk/v3";
+  *
+  * export const myTask = task({
+  *  id: "my-task",
+  *  run: async (payload, { ctx }) => {
+  *    const { result, compute } = await usage.measure(async () => {
+  *      // Do some work
+  *      return "result";
+  *    });
+  *
+  *    console.log("Result", result);
+  *    console.log("Cost and duration", { cost: compute.costInCents, duration: compute.durationMs });
+  *  },
+  * });
+  * ```
+  */
+  measure: async (cb) => {
+    const measurement = usage$1.start();
+    const result = await cb();
+    const sample = usage$1.stop(measurement);
+    const machine = taskContext.ctx?.machine;
+    const costInCents = machine?.centsPerMs ? sample.cpuTime * machine.centsPerMs : 0;
+    return {
+      result,
+      compute: {
+        costInCents,
+        durationMs: sample.cpuTime
+      }
+    };
+  }
+};
 var runs = {
   replay: replayRun,
   cancel: cancelRun,
@@ -982,6 +1093,7 @@ __export(schedules_exports, {
   list: () => list,
   retrieve: () => retrieve,
   task: () => task2,
+  timezones: () => timezones,
   update: () => update
 });
 function task2(params) {
@@ -1048,6 +1160,19 @@ function list(options) {
   return apiClient.listSchedules(options);
 }
 __name(list, "list");
+function timezones(options) {
+  const baseUrl = apiClientManager.baseURL;
+  if (!baseUrl) {
+    throw apiClientMissingError();
+  }
+  return zodfetch(TimezonesResult, `${baseUrl}/api/v1/timezones${options?.excludeUtc === true ? "?excludeUtc=true" : ""}`, {
+    method: "GET",
+    headers: {
+      "Content-Type": "application/json"
+    }
+  });
+}
+__name(timezones, "timezones");
 
 // src/v3/envvars.ts
 var envvars_exports = {};
@@ -1252,6 +1377,6 @@ function configure(options) {
 }
 __name(configure, "configure");
 
-export { InMemoryCache, configure, createCache, envvars_exports as envvars, queue, retry, runs, schedules_exports as schedules, task, wait };
+export { InMemoryCache, configure, createCache, envvars_exports as envvars, queue, retry, runs, schedules_exports as schedules, task, usage, wait };
 //# sourceMappingURL=out.js.map
 //# sourceMappingURL=index.mjs.map