@trigger.dev/core 0.0.0-v3-prerelease-20250108141813 → 0.0.0-v4-prerelease-20250916125920

package/dist/commonjs/v3/types/tasks.d.ts

@@ -1,21 +1,17 @@
 import { SerializableJson } from "../../schemas/json.js";
 import { TriggerApiRequestOptions } from "../apiClient/index.js";
+import { AnyOnCatchErrorHookFunction, OnCatchErrorHookFunction, OnCleanupHookFunction, OnCompleteHookFunction, OnFailureHookFunction, OnInitHookFunction, OnMiddlewareHookFunction, OnResumeHookFunction, OnStartHookFunction, OnSuccessHookFunction, OnWaitHookFunction, OnCancelHookFunction } from "../lifecycleHooks/types.js";
 import { RunTags } from "../schemas/api.js";
-import { MachineCpu, MachineMemory, RetryOptions, TaskMetadata, TaskRunContext } from "../schemas/index.js";
-import { QueueOptions } from "../schemas/schemas.js";
+import { MachineCpu, MachineMemory, MachinePresetName, RetryOptions, TaskMetadata, TaskRunContext } from "../schemas/index.js";
 import { IdempotencyKey } from "./idempotencyKeys.js";
+import { QueueOptions } from "./queues.js";
 import { AnySchemaParseFn, inferSchemaIn, inferSchemaOut, Schema } from "./schemas.js";
-import { Prettify } from "./utils.js";
 import { inferToolParameters, ToolTaskParameters } from "./tools.js";
-type RequireOne<T, K extends keyof T> = {
-    [X in Exclude<keyof T, K>]?: T[X];
-} & {
-    [P in K]-?: T[P];
-};
-export type Queue = RequireOne<QueueOptions, "name">;
+import { Prettify } from "./utils.js";
+import { JSONSchema } from "./jsonSchema.js";
+export type Queue = QueueOptions;
 export type TaskSchema = Schema;
 export type { inferSchemaIn } from "./schemas.js";
-type TaskRunConcurrencyOptions = Queue;
 export declare class SubtaskUnwrapError extends Error {
     readonly taskId: string;
     readonly runId: string;
@@ -33,24 +29,32 @@ export type RunFnParams<TInitOutput extends InitOutput> = Prettify<{
     ctx: Context;
     /** If you use the `init` function, this will be whatever you returned. */
     init?: TInitOutput;
-    /** Abort signal that is aborted when a task run exceeds it's maxDuration. Can be used to automatically cancel downstream requests */
-    signal?: AbortSignal;
+    /** Abort signal that is aborted when a task run exceeds it's maxDuration or if the task run is cancelled. Can be used to automatically cancel downstream requests */
+    signal: AbortSignal;
 }>;
 export type MiddlewareFnParams = Prettify<{
     ctx: Context;
     next: () => Promise<void>;
-    /** Abort signal that is aborted when a task run exceeds it's maxDuration. Can be used to automatically cancel downstream requests */
-    signal?: AbortSignal;
+    /** Abort signal that is aborted when a task run exceeds it's maxDuration or if the task run is cancelled. Can be used to automatically cancel downstream requests */
+    signal: AbortSignal;
 }>;
 export type InitFnParams = Prettify<{
     ctx: Context;
-    /** Abort signal that is aborted when a task run exceeds it's maxDuration. Can be used to automatically cancel downstream requests */
-    signal?: AbortSignal;
+    /** Abort signal that is aborted when a task run exceeds it's maxDuration or if the task run is cancelled. Can be used to automatically cancel downstream requests */
+    signal: AbortSignal;
 }>;
 export type StartFnParams = Prettify<{
     ctx: Context;
-    /** Abort signal that is aborted when a task run exceeds it's maxDuration. Can be used to automatically cancel downstream requests */
-    signal?: AbortSignal;
+    init?: InitOutput;
+    /** Abort signal that is aborted when a task run exceeds it's maxDuration or if the task run is cancelled. Can be used to automatically cancel downstream requests */
+    signal: AbortSignal;
+}>;
+export type CancelFnParams = Prettify<{
+    ctx: Context;
+    /** Abort signal that is aborted when a task run exceeds it's maxDuration or if the task run is cancelled. Can be used to automatically cancel downstream requests */
+    signal: AbortSignal;
+    runPromise: Promise<unknown>;
+    init?: InitOutput;
 }>;
 export type Context = TaskRunContext;
 export type SuccessFnParams<TInitOutput extends InitOutput> = RunFnParams<TInitOutput>;
@@ -70,14 +74,14 @@ export type HandleErrorModificationOptions = {
 export type HandleErrorResult = undefined | void | HandleErrorModificationOptions | Promise<undefined | void | HandleErrorModificationOptions>;
 export type HandleErrorArgs = {
     ctx: Context;
-    init: unknown;
+    init?: Record<string, unknown>;
     retry?: RetryOptions;
     retryAt?: Date;
     retryDelayInMs?: number;
     /** Abort signal that is aborted when a task run exceeds it's maxDuration. Can be used to automatically cancel downstream requests */
     signal?: AbortSignal;
 };
-export type HandleErrorFunction = (payload: any, error: unknown, params: HandleErrorArgs) => HandleErrorResult;
+export type HandleErrorFunction = AnyOnCatchErrorHookFunction;
 type CommonTaskOptions<TIdentifier extends string, TPayload = void, TOutput = unknown, TInitOutput extends InitOutput = any> = {
     /** An id for your task. This must be unique inside your project and not change between versions.  */
     id: TIdentifier;
@@ -122,18 +126,18 @@ type CommonTaskOptions<TIdentifier extends string, TPayload = void, TOutput = un
       });
      * ```
      */
-    queue?: QueueOptions;
-    /** Configure the spec of the machine you want your task to run on.
+    queue?: {
+        name?: string;
+        concurrencyLimit?: number;
+    };
+    /** Configure the spec of the [machine](https://trigger.dev/docs/machines) you want your task to run on.
      *
      * @example
      *
      * ```ts
      * export const heavyTask = task({
         id: "heavy-task",
-        machine: {
-          cpu: 2,
-          memory: 4,
-        },
+        machine: "medium-1x",
         run: async ({ payload, ctx }) => {
           //...
         },
@@ -165,8 +169,8 @@ type CommonTaskOptions<TIdentifier extends string, TPayload = void, TOutput = un
          */
         memory?: MachineMemory;
         /** Preset to use for the machine. Defaults to small-1x */
-        preset?: "micro" | "small-1x" | "small-2x" | "medium-1x" | "medium-2x" | "large-1x" | "large-2x";
-    };
+        preset?: MachinePresetName;
+    } | MachinePresetName;
     /**
      * The maximum duration in compute-time seconds that a task run is allowed to run. If the task run exceeds this duration, it will be stopped.
      *
@@ -181,16 +185,30 @@ type CommonTaskOptions<TIdentifier extends string, TPayload = void, TOutput = un
     run: (payload: TPayload, params: RunFnParams<TInitOutput>) => Promise<TOutput>;
     /**
      * init is called before the run function is called. It's useful for setting up any global state.
+     *
+     * @deprecated Use locals and middleware instead
      */
-    init?: (payload: TPayload, params: InitFnParams) => Promise<TInitOutput>;
+    init?: OnInitHookFunction<TPayload, TInitOutput>;
     /**
      * cleanup is called after the run function has completed.
+     *
+     * @deprecated Use middleware instead
      */
-    cleanup?: (payload: TPayload, params: RunFnParams<TInitOutput>) => Promise<void>;
+    cleanup?: OnCleanupHookFunction<TPayload, TInitOutput>;
     /**
      * handleError is called when the run function throws an error. It can be used to modify the error or return new retry options.
+     *
+     * @deprecated Use catchError instead
      */
-    handleError?: (payload: TPayload, error: unknown, params: HandleErrorFnParams<TInitOutput>) => HandleErrorResult;
+    handleError?: OnCatchErrorHookFunction<TPayload>;
+    /**
+     * catchError is called when the run function throws an error. It can be used to modify the error or return new retry options.
+     */
+    catchError?: OnCatchErrorHookFunction<TPayload>;
+    onResume?: OnResumeHookFunction<TPayload>;
+    onWait?: OnWaitHookFunction<TPayload>;
+    onComplete?: OnCompleteHookFunction<TPayload, TOutput>;
+    onCancel?: OnCancelHookFunction<TPayload, TOutput, TInitOutput>;
     /**
      * middleware allows you to run code "around" the run function. This can be useful for logging, metrics, or other cross-cutting concerns.
      *
@@ -208,21 +226,29 @@ type CommonTaskOptions<TIdentifier extends string, TPayload = void, TOutput = un
      * });
      * ```
      */
-    middleware?: (payload: TPayload, params: MiddlewareFnParams) => Promise<void>;
+    middleware?: OnMiddlewareHookFunction<TPayload>;
     /**
      * onStart is called the first time a task is executed in a run (not before every retry)
      */
-    onStart?: (payload: TPayload, params: StartFnParams) => Promise<void>;
+    onStart?: OnStartHookFunction<TPayload, TInitOutput>;
     /**
      * onSuccess is called after the run function has successfully completed.
      */
-    onSuccess?: (payload: TPayload, output: TOutput, params: SuccessFnParams<TInitOutput>) => Promise<void>;
+    onSuccess?: OnSuccessHookFunction<TPayload, TOutput, TInitOutput>;
     /**
      * onFailure is called after a task run has failed (meaning the run function threw an error and won't be retried anymore)
      */
-    onFailure?: (payload: TPayload, error: unknown, params: FailureFnParams<TInitOutput>) => Promise<void>;
+    onFailure?: OnFailureHookFunction<TPayload, TInitOutput>;
+    /**
+     * JSON Schema for the task payload. This will be synced to the server during indexing.
+     * Should be a valid JSON Schema Draft 7 object.
+     */
+    jsonSchema?: JSONSchema;
 };
 export type TaskOptions<TIdentifier extends string, TPayload = void, TOutput = unknown, TInitOutput extends InitOutput = any> = CommonTaskOptions<TIdentifier, TPayload, TOutput, TInitOutput>;
+export type TaskOptionsWithSchema<TIdentifier extends string, TOutput = unknown, TInitOutput extends InitOutput = any> = CommonTaskOptions<TIdentifier, any, TOutput, TInitOutput> & {
+    jsonSchema: JSONSchema;
+};
 export type TaskWithSchemaOptions<TIdentifier extends string, TSchema extends TaskSchema | undefined = undefined, TOutput = unknown, TInitOutput extends InitOutput = any> = CommonTaskOptions<TIdentifier, inferSchemaOut<TSchema>, TOutput, TInitOutput> & {
     schema?: TSchema;
 };
@@ -257,9 +283,7 @@ export type AnyBatchedRunHandle = BatchedRunHandle<string, any, any>;
  */
 export type BatchRunHandle<TTaskIdentifier extends string, TPayload, TOutput> = BrandedRun<{
     batchId: string;
-    isCached: boolean;
-    idempotencyKey?: string;
-    runs: Array<BatchedRunHandle<TTaskIdentifier, TPayload, TOutput>>;
+    runCount: number;
     publicAccessToken: string;
 }, TOutput, TPayload>;
 export type RunHandleOutput<TRunHandle> = TRunHandle extends RunHandle<string, any, infer TOutput> ? TOutput : never;
@@ -330,6 +354,7 @@ export interface Task<TIdentifier extends string, TInput = void, TOutput = any>
      */
     id: TIdentifier;
     description?: string;
+    jsonSchema?: JSONSchema;
     /**
      * Trigger a task with the given payload, and continue without waiting for the result. If you want to wait for the result, use `triggerAndWait`. Returns the id of the triggered task run.
      * @param payload
@@ -465,8 +490,20 @@ export type TriggerOptions = {
      * Specify a duration string like "1h", "10s", "30m", etc.
      */
     idempotencyKeyTTL?: string;
+    /**
+     * The maximum number of retry attempts for the task if it fails.
+     * If not specified, it will use the task or the default retry policy from your trigger.config file.
+     */
     maxAttempts?: number;
-    queue?: TaskRunConcurrencyOptions;
+    /**
+     * You can override the queue for the task. If a queue doesn't exist for the given name, the run will be in the PENDING_VERSION state until the queue is created..
+     */
+    queue?: string;
+    /**
+     * The `concurrencyKey` creates a copy of the queue for every unique value of the key.
+     * For example, if the queue (set when triggering or on the task) has a concurrency limit of 10,
+     * and you set the concurrency key to `userId`, then each user will have their own queue with a concurrency limit of 10.
+     */
     concurrencyKey?: string;
     /**
      * The delay before the task is executed. This can be a string like "1h" or a Date object.
@@ -495,6 +532,25 @@ export type TriggerOptions = {
      * **Note:** Runs in development have a default `ttl` of 10 minutes. You can override this by setting the `ttl` option.
      */
     ttl?: string | number;
+    /**
+     * If triggered at the same time, a higher priority run will be executed first.
+     *
+     The value is a time offset in seconds that determines the order of dequeuing.
+     * If you trigger two runs 9 seconds apart but the second one has `priority: 10`, it will be executed before the first one.
+     *
+     * @example
+     * ```ts
+     // no priority = 0
+     await myTask.trigger({ foo: "bar" });
+  
+     //... imagine 9s pass by
+  
+     // this run will start before the run above that was triggered 9s ago (with no priority)
+     await myTask.trigger({ foo: "bar" }, { priority: 10 });
+     ```
+     *
+     */
+    priority?: number;
     /**
      * Tags to attach to the run. Tags can be used to filter runs in the dashboard and using the SDK.
      *
@@ -510,7 +566,7 @@ export type TriggerOptions = {
      */
     tags?: RunTags;
     /**
-     * Metadata to attach to the run. Metadata can be used to store additional information about the run. Limited to 4KB.
+     * Metadata to attach to the run. Metadata can be used to store additional information about the run. Limited to 256KB.
      */
     metadata?: Record<string, SerializableJson>;
     /**
@@ -521,24 +577,48 @@ export type TriggerOptions = {
      * Minimum value is 5 seconds
      */
     maxDuration?: number;
-};
-export type TriggerAndWaitOptions = Omit<TriggerOptions, "idempotencyKey" | "idempotencyKeyTTL">;
-export type BatchTriggerOptions = {
-    idempotencyKey?: IdempotencyKey | string | string[];
-    idempotencyKeyTTL?: string;
     /**
-     * When true, triggers tasks sequentially in batch order. This ensures ordering but may be slower,
-     * especially for large batches.
+     * The machine preset to use for this run. This will override the task's machine preset and any defaults.
+     */
+    machine?: MachinePresetName;
+    /**
+     * Specify the version of the deployed task to run. By default the "current" version is used at the time of execution,
+     * but you can specify a specific version to run here. You can also set the TRIGGER_VERSION environment
+     * variables to run a specific version for all tasks.
      *
-     * When false (default), triggers tasks in parallel for better performance, but order is not guaranteed.
+     * @example
      *
-     * Note: This only affects the order of run creation, not the actual task execution.
+     * ```ts
+     * await myTask.trigger({ foo: "bar" }, { version: "20250208.1" });
+     * ```
      *
-     * @default false
+     * Note that this option is only available for `trigger` and NOT `triggerAndWait` (and their batch counterparts). The "wait" versions will always be locked
+     * to the same version as the parent task that is triggering the child tasks.
      */
-    triggerSequentially?: boolean;
+    version?: string;
+    /**
+     * Specify the region to run the task in. This overrides the default region set for your project in the dashboard.
+     *
+     * Check the Regions page in the dashboard for regions that are available to you.
+     *
+     * In DEV this won't do anything, so it's fine to set it in your code.
+     *
+     * @example
+     *
+     * ```ts
+     * await myTask.trigger({ foo: "bar" }, { region: "us-east-1" });
+     * ```
+     */
+    region?: string;
 };
-export type BatchTriggerAndWaitOptions = {
+export type TriggerAndWaitOptions = Omit<TriggerOptions, "version">;
+export type BatchTriggerOptions = {
+    /**
+     * If no idempotencyKey is set on an individual item in the batch, it will use this key on each item + the array index.
+     * This is useful to prevent work being done again if the task has to retry.
+     */
+    idempotencyKey?: IdempotencyKey | string | string[];
+    idempotencyKeyTTL?: string;
     /**
      * When true, triggers tasks sequentially in batch order. This ensures ordering but may be slower,
      * especially for large batches.
@@ -551,6 +631,7 @@ export type BatchTriggerAndWaitOptions = {
      */
     triggerSequentially?: boolean;
 };
+export type BatchTriggerAndWaitOptions = BatchTriggerOptions;
 export type TaskMetadataWithFunctions = TaskMetadata & {
     fns: {
         run: (payload: any, params: RunFnParams<any>) => Promise<any>;
@@ -563,6 +644,7 @@ export type TaskMetadataWithFunctions = TaskMetadata & {
         onStart?: (payload: any, params: StartFnParams) => Promise<void>;
         parsePayload?: AnySchemaParseFn;
     };
+    schema?: TaskSchema;
 };
 export type RunTypes<TTaskIdentifier extends string, TPayload, TOutput> = {
     output: TOutput;

package/dist/commonjs/v3/types/tasks.js.map

@@ -1 +1 @@
-{"version":3,"file":"tasks.js","sourceRoot":"","sources":["../../../../src/v3/types/tasks.ts"],"names":[],"mappings":";;;AA8BA,MAAa,kBAAmB,SAAQ,KAAK;IAC3B,MAAM,CAAS;IACf,KAAK,CAAS;IACd,KAAK,CAAW;IAEhC,YAAY,MAAc,EAAE,KAAa,EAAE,YAAqB;QAC9D,IAAI,YAAY,YAAY,KAAK,EAAE,CAAC;YAClC,KAAK,CAAC,YAAY,MAAM,KAAK,YAAY,CAAC,OAAO,EAAE,CAAC,CAAC;YACrD,IAAI,CAAC,KAAK,GAAG,YAAY,CAAC;YAC1B,IAAI,CAAC,IAAI,GAAG,oBAAoB,CAAC;QACnC,CAAC;aAAM,CAAC;YACN,KAAK,CAAC,YAAY,MAAM,EAAE,CAAC,CAAC;YAC5B,IAAI,CAAC,IAAI,GAAG,oBAAoB,CAAC;YACjC,IAAI,CAAC,KAAK,GAAG,YAAY,CAAC;QAC5B,CAAC;QAED,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;IACrB,CAAC;CACF;AAnBD,gDAmBC;AAED,MAAa,cAAoD,SAAQ,OAExE;IAUoB;IATnB,YACE,QAOS,EACQ,MAAmB;QAEpC,KAAK,CAAC,QAAQ,CAAC,CAAC;QAFC,WAAM,GAAN,MAAM,CAAa;IAGtC,CAAC;IAED,MAAM;QACJ,OAAO,IAAI,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE;YAC1B,IAAI,MAAM,CAAC,EAAE,EAAE,CAAC;gBACd,OAAO,MAAM,CAAC,MAAM,CAAC;YACvB,CAAC;iBAAM,CAAC;gBACN,MAAM,IAAI,kBAAkB,CAAC,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,EAAE,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC;YACrE,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC;CACF;AA1BD,wCA0BC"}
+{"version":3,"file":"tasks.js","sourceRoot":"","sources":["../../../../src/v3/types/tasks.ts"],"names":[],"mappings":";;;AAoCA,MAAa,kBAAmB,SAAQ,KAAK;IAC3B,MAAM,CAAS;IACf,KAAK,CAAS;IACd,KAAK,CAAW;IAEhC,YAAY,MAAc,EAAE,KAAa,EAAE,YAAqB;QAC9D,IAAI,YAAY,YAAY,KAAK,EAAE,CAAC;YAClC,KAAK,CAAC,YAAY,MAAM,KAAK,YAAY,CAAC,OAAO,EAAE,CAAC,CAAC;YACrD,IAAI,CAAC,KAAK,GAAG,YAAY,CAAC;YAC1B,IAAI,CAAC,IAAI,GAAG,oBAAoB,CAAC;QACnC,CAAC;aAAM,CAAC;YACN,KAAK,CAAC,YAAY,MAAM,EAAE,CAAC,CAAC;YAC5B,IAAI,CAAC,IAAI,GAAG,oBAAoB,CAAC;YACjC,IAAI,CAAC,KAAK,GAAG,YAAY,CAAC;QAC5B,CAAC;QAED,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;IACrB,CAAC;CACF;AAnBD,gDAmBC;AAED,MAAa,cAAoD,SAAQ,OAExE;IAUoB;IATnB,YACE,QAOS,EACQ,MAAmB;QAEpC,KAAK,CAAC,QAAQ,CAAC,CAAC;QAFC,WAAM,GAAN,MAAM,CAAa;IAGtC,CAAC;IAED,MAAM;QACJ,OAAO,IAAI,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE;YAC1B,IAAI,MAAM,CAAC,EAAE,EAAE,CAAC;gBACd,OAAO,MAAM,CAAC,MAAM,CAAC;YACvB,CAAC;iBAAM,CAAC;gBACN,MAAM,IAAI,kBAAkB,CAAC,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,EAAE,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC;YACrE,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC;CACF;AA1BD,wCA0BC"}

package/dist/commonjs/v3/usage/api.d.ts

@@ -1,4 +1,4 @@
-import type { UsageManager, UsageMeasurement, UsageSample } from "./types.js";
+import type { InitialUsageState, UsageManager, UsageMeasurement, UsageSample } from "./types.js";
 export declare class UsageAPI implements UsageManager {
     #private;
     private static _instance?;
@@ -10,4 +10,7 @@ export declare class UsageAPI implements UsageManager {
     stop(measurement: UsageMeasurement): UsageSample;
     pauseAsync<T>(cb: () => Promise<T>): Promise<T>;
     sample(): UsageSample | undefined;
+    flush(): Promise<void>;
+    reset(): void;
+    getInitialState(): InitialUsageState;
 }

package/dist/commonjs/v3/usage/api.js

@@ -33,6 +33,16 @@ class UsageAPI {
     sample() {
         return this.#getUsageManager().sample();
     }
+    flush() {
+        return this.#getUsageManager().flush();
+    }
+    reset() {
+        this.#getUsageManager().reset();
+        this.disable();
+    }
+    getInitialState() {
+        return this.#getUsageManager().getInitialState();
+    }
     #getUsageManager() {
         return (0, globals_js_1.getGlobal)(API_NAME) ?? NOOP_USAGE_MANAGER;
     }

package/dist/commonjs/v3/usage/api.js.map

@@ -1 +1 @@
-{"version":3,"file":"api.js","sourceRoot":"","sources":["../../../../src/v3/usage/api.ts"],"names":[],"mappings":";;;AAAA,MAAM,QAAQ,GAAG,OAAO,CAAC;AAEzB,oDAAkF;AAElF,+DAAyD;AAEzD,MAAM,kBAAkB,GAAG,IAAI,sCAAgB,EAAE,CAAC;AAElD,MAAa,QAAQ;IACX,MAAM,CAAC,SAAS,CAAY;IAEpC,gBAAuB,CAAC;IAEjB,MAAM,CAAC,WAAW;QACvB,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;YACpB,IAAI,CAAC,SAAS,GAAG,IAAI,QAAQ,EAAE,CAAC;QAClC,CAAC;QAED,OAAO,IAAI,CAAC,SAAS,CAAC;IACxB,CAAC;IAEM,qBAAqB,CAAC,OAAqB;QAChD,OAAO,IAAA,2BAAc,EAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;IAC3C,CAAC;IAEM,OAAO;QACZ,IAAI,CAAC,gBAAgB,EAAE,CAAC,OAAO,EAAE,CAAC;QAClC,IAAA,6BAAgB,EAAC,QAAQ,CAAC,CAAC;IAC7B,CAAC;IAEM,KAAK;QACV,OAAO,IAAI,CAAC,gBAAgB,EAAE,CAAC,KAAK,EAAE,CAAC;IACzC,CAAC;IAEM,IAAI,CAAC,WAA6B;QACvC,OAAO,IAAI,CAAC,gBAAgB,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IACnD,CAAC;IAEM,UAAU,CAAI,EAAoB;QACvC,OAAO,IAAI,CAAC,gBAAgB,EAAE,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;IAChD,CAAC;IAEM,MAAM;QACX,OAAO,IAAI,CAAC,gBAAgB,EAAE,CAAC,MAAM,EAAE,CAAC;IAC1C,CAAC;IAED,gBAAgB;QACd,OAAO,IAAA,sBAAS,EAAC,QAAQ,CAAC,IAAI,kBAAkB,CAAC;IACnD,CAAC;CACF;AAzCD,4BAyCC"}
+{"version":3,"file":"api.js","sourceRoot":"","sources":["../../../../src/v3/usage/api.ts"],"names":[],"mappings":";;;AAAA,MAAM,QAAQ,GAAG,OAAO,CAAC;AAEzB,oDAAkF;AAElF,+DAAyD;AAEzD,MAAM,kBAAkB,GAAG,IAAI,sCAAgB,EAAE,CAAC;AAElD,MAAa,QAAQ;IACX,MAAM,CAAC,SAAS,CAAY;IAEpC,gBAAuB,CAAC;IAEjB,MAAM,CAAC,WAAW;QACvB,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;YACpB,IAAI,CAAC,SAAS,GAAG,IAAI,QAAQ,EAAE,CAAC;QAClC,CAAC;QAED,OAAO,IAAI,CAAC,SAAS,CAAC;IACxB,CAAC;IAEM,qBAAqB,CAAC,OAAqB;QAChD,OAAO,IAAA,2BAAc,EAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;IAC3C,CAAC;IAEM,OAAO;QACZ,IAAI,CAAC,gBAAgB,EAAE,CAAC,OAAO,EAAE,CAAC;QAClC,IAAA,6BAAgB,EAAC,QAAQ,CAAC,CAAC;IAC7B,CAAC;IAEM,KAAK;QACV,OAAO,IAAI,CAAC,gBAAgB,EAAE,CAAC,KAAK,EAAE,CAAC;IACzC,CAAC;IAEM,IAAI,CAAC,WAA6B;QACvC,OAAO,IAAI,CAAC,gBAAgB,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IACnD,CAAC;IAEM,UAAU,CAAI,EAAoB;QACvC,OAAO,IAAI,CAAC,gBAAgB,EAAE,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;IAChD,CAAC;IAEM,MAAM;QACX,OAAO,IAAI,CAAC,gBAAgB,EAAE,CAAC,MAAM,EAAE,CAAC;IAC1C,CAAC;IAEM,KAAK;QACV,OAAO,IAAI,CAAC,gBAAgB,EAAE,CAAC,KAAK,EAAE,CAAC;IACzC,CAAC;IAEM,KAAK;QACV,IAAI,CAAC,gBAAgB,EAAE,CAAC,KAAK,EAAE,CAAC;QAChC,IAAI,CAAC,OAAO,EAAE,CAAC;IACjB,CAAC;IAEM,eAAe;QACpB,OAAO,IAAI,CAAC,gBAAgB,EAAE,CAAC,eAAe,EAAE,CAAC;IACnD,CAAC;IAED,gBAAgB;QACd,OAAO,IAAA,sBAAS,EAAC,QAAQ,CAAC,IAAI,kBAAkB,CAAC;IACnD,CAAC;CACF;AAtDD,4BAsDC"}

package/dist/commonjs/v3/usage/devUsageManager.d.ts

@@ -1,4 +1,4 @@
-import { UsageManager, UsageMeasurement, UsageSample } from "./types.js";
+import { InitialUsageState, UsageManager, UsageMeasurement, UsageSample } from "./types.js";
 import { ClockTime } from "../clock/clock.js";
 declare class DevUsageMeasurement implements UsageMeasurement {
     readonly id: string;
@@ -14,7 +14,12 @@ export declare class DevUsageManager implements UsageManager {
     private _firstMeasurement?;
     private _currentMeasurements;
     private _pauses;
+    private _initialState;
     disable(): void;
+    flush(): Promise<void>;
+    reset(): void;
+    setInitialState(state: InitialUsageState): void;
+    getInitialState(): InitialUsageState;
     sample(): UsageSample | undefined;
     start(): DevUsageMeasurement;
     stop(measurement: DevUsageMeasurement): UsageSample;

package/dist/commonjs/v3/usage/devUsageManager.js

@@ -38,7 +38,27 @@ class DevUsageManager {
     _firstMeasurement;
     _currentMeasurements = new Map();
     _pauses = new Map();
+    _initialState = {
+        cpuTime: 0,
+        costInCents: 0,
+    };
     disable() { }
+    async flush() { }
+    reset() {
+        this._firstMeasurement = undefined;
+        this._currentMeasurements.clear();
+        this._pauses.clear();
+        this._initialState = {
+            cpuTime: 0,
+            costInCents: 0,
+        };
+    }
+    setInitialState(state) {
+        this._initialState = state;
+    }
+    getInitialState() {
+        return this._initialState;
+    }
     sample() {
         return this._firstMeasurement?.sample();
     }
@@ -55,7 +75,9 @@ class DevUsageManager {
     stop(measurement) {
         measurement.stop();
         const sample = measurement.sample();
-        this._currentMeasurements.delete(measurement.id);
+        if (this._currentMeasurements.has(measurement.id)) {
+            this._currentMeasurements.delete(measurement.id);
+        }
         return sample;
     }
     async pauseAsync(cb) {

package/dist/commonjs/v3/usage/devUsageManager.js.map

@@ -1 +1 @@
-{"version":3,"file":"devUsageManager.js","sourceRoot":"","sources":["../../../../src/v3/usage/devUsageManager.ts"],"names":[],"mappings":";;;AACA,kDAAwC;AACxC,gDAAqE;AAErE,MAAM,mBAAmB;IAKL;IACR;IALF,OAAO,GAAuD,IAAI,GAAG,EAAE,CAAC;IACxE,QAAQ,CAAwB;IAExC,YACkB,EAAU,EAClB,YAAuB,oBAAK,CAAC,UAAU,EAAE;QADjC,OAAE,GAAF,EAAE,CAAQ;QAClB,cAAS,GAAT,SAAS,CAAgC;IAChD,CAAC;IAEJ,IAAI;QACF,IAAI,CAAC,QAAQ,GAAG,oBAAK,CAAC,UAAU,EAAE,CAAC;IACrC,CAAC;IAED,MAAM;QACJ,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,IAAI,oBAAK,CAAC,UAAU,EAAE,CAAC;QAEpD,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,IAAA,gCAAqB,EAAC,IAAI,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAErF,IAAI,QAAQ,KAAK,CAAC,EAAE,CAAC;YACnB,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC;QACrC,CAAC;QAED,MAAM,WAAW,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE;YAC5E,OAAO,KAAK,GAAG,IAAA,gCAAqB,EAAC,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,GAAG,IAAI,OAAO,CAAC,CAAC;QAC1E,CAAC,EAAE,CAAC,CAAC,CAAC;QAEN,MAAM,OAAO,GAAG,QAAQ,GAAG,WAAW,CAAC;QAEvC,OAAO;YACL,QAAQ;YACR,OAAO;SACR,CAAC;IACJ,CAAC;IAED,aAAa,CAAC,OAAe,EAAE,KAAgB,EAAE,GAAe;QAC9D,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC;IAC5C,CAAC;CACF;AAED,MAAa,eAAe;IAClB,iBAAiB,CAAuB;IACxC,oBAAoB,GAAqC,IAAI,GAAG,EAAE,CAAC;IACnE,OAAO,GAAuD,IAAI,GAAG,EAAE,CAAC;IAEhF,OAAO,KAAU,CAAC;IAElB,MAAM;QACJ,OAAO,IAAI,CAAC,iBAAiB,EAAE,MAAM,EAAE,CAAC;IAC1C,CAAC;IAED,KAAK;QACH,uBAAuB;QACvB,MAAM,EAAE,GAAG,oBAAoB,EAAE,CAAC;QAElC,MAAM,WAAW,GAAG,IAAI,mBAAmB,CAAC,EAAE,CAAC,CAAC;QAEhD,IAAI,CAAC,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAC5B,IAAI,CAAC,iBAAiB,GAAG,WAAW,CAAC;QACvC,CAAC;QAED,IAAI,CAAC,oBAAoB,CAAC,GAAG,CAAC,EAAE,EAAE,WAAW,CAAC,CAAC;QAE/C,OAAO,WAAW,CAAC;IACrB,CAAC;IAED,IAAI,CAAC,WAAgC;QACnC,WAAW,CAAC,IAAI,EAAE,CAAC;QAEnB,MAAM,MAAM,GAAG,WAAW,CAAC,MAAM,EAAE,CAAC;QAEpC,IAAI,CAAC,oBAAoB,CAAC,MAAM,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;QAEjD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,UAAU,CAAI,EAAoB;QACtC,MAAM,OAAO,GAAG,oBAAoB,EAAE,CAAC;QAEvC,MAAM,UAAU,GAAG,oBAAK,CAAC,UAAU,EAAE,CAAC;QAEtC,IAAI,CAAC;YACH,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,EAAE,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,CAAC;YAEjD,KAAK,MAAM,WAAW,IAAI,IAAI,CAAC,oBAAoB,CAAC,MAAM,EAAE,EAAE,CAAC;gBAC7D,WAAW,CAAC,aAAa,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;YACjD,CAAC;YAED,OAAO,MAAM,EAAE,EAAE,CAAC;QACpB,CAAC;gBAAS,CAAC;YACT,MAAM,QAAQ,GAAG,oBAAK,CAAC,UAAU,EAAE,CAAC;YAEpC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,EAAE,EAAE,KAAK,EAAE,UAAU,EAAE,GAAG,EAAE,QAAQ,EAAE,CAAC,CAAC;YAEhE,KAAK,MAAM,WAAW,IAAI,IAAI,CAAC,oBAAoB,CAAC,MAAM,EAAE,EAAE,CAAC;gBAC7D,WAAW,CAAC,aAAa,CAAC,OAAO,EAAE,UAAU,EAAE,QAAQ,CAAC,CAAC;YAC3D,CAAC;QACH,CAAC;IACH,CAAC;CACF;AA3DD,0CA2DC;AAED,SAAS,oBAAoB;IAC3B,IAAI,MAAM,GAAG,EAAE,CAAC;IAChB,MAAM,UAAU,GAAG,gEAAgE,CAAC;IACpF,MAAM,gBAAgB,GAAG,UAAU,CAAC,MAAM,CAAC;IAE3C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;QAC5B,MAAM,IAAI,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,gBAAgB,CAAC,CAAC,CAAC;IAC5E,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC"}
+{"version":3,"file":"devUsageManager.js","sourceRoot":"","sources":["../../../../src/v3/usage/devUsageManager.ts"],"names":[],"mappings":";;;AACA,kDAAwC;AACxC,gDAAqE;AAErE,MAAM,mBAAmB;IAKL;IACR;IALF,OAAO,GAAuD,IAAI,GAAG,EAAE,CAAC;IACxE,QAAQ,CAAwB;IAExC,YACkB,EAAU,EAClB,YAAuB,oBAAK,CAAC,UAAU,EAAE;QADjC,OAAE,GAAF,EAAE,CAAQ;QAClB,cAAS,GAAT,SAAS,CAAgC;IAChD,CAAC;IAEJ,IAAI;QACF,IAAI,CAAC,QAAQ,GAAG,oBAAK,CAAC,UAAU,EAAE,CAAC;IACrC,CAAC;IAED,MAAM;QACJ,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,IAAI,oBAAK,CAAC,UAAU,EAAE,CAAC;QAEpD,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,IAAA,gCAAqB,EAAC,IAAI,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAErF,IAAI,QAAQ,KAAK,CAAC,EAAE,CAAC;YACnB,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC;QACrC,CAAC;QAED,MAAM,WAAW,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE;YAC5E,OAAO,KAAK,GAAG,IAAA,gCAAqB,EAAC,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,GAAG,IAAI,OAAO,CAAC,CAAC;QAC1E,CAAC,EAAE,CAAC,CAAC,CAAC;QAEN,MAAM,OAAO,GAAG,QAAQ,GAAG,WAAW,CAAC;QAEvC,OAAO;YACL,QAAQ;YACR,OAAO;SACR,CAAC;IACJ,CAAC;IAED,aAAa,CAAC,OAAe,EAAE,KAAgB,EAAE,GAAe;QAC9D,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC;IAC5C,CAAC;CACF;AAED,MAAa,eAAe;IAClB,iBAAiB,CAAuB;IACxC,oBAAoB,GAAqC,IAAI,GAAG,EAAE,CAAC;IACnE,OAAO,GAAuD,IAAI,GAAG,EAAE,CAAC;IACxE,aAAa,GAAsB;QACzC,OAAO,EAAE,CAAC;QACV,WAAW,EAAE,CAAC;KACf,CAAC;IAEF,OAAO,KAAU,CAAC;IAElB,KAAK,CAAC,KAAK,KAAmB,CAAC;IAE/B,KAAK;QACH,IAAI,CAAC,iBAAiB,GAAG,SAAS,CAAC;QACnC,IAAI,CAAC,oBAAoB,CAAC,KAAK,EAAE,CAAC;QAClC,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;QACrB,IAAI,CAAC,aAAa,GAAG;YACnB,OAAO,EAAE,CAAC;YACV,WAAW,EAAE,CAAC;SACf,CAAC;IACJ,CAAC;IAED,eAAe,CAAC,KAAwB;QACtC,IAAI,CAAC,aAAa,GAAG,KAAK,CAAC;IAC7B,CAAC;IAED,eAAe;QACb,OAAO,IAAI,CAAC,aAAa,CAAC;IAC5B,CAAC;IAED,MAAM;QACJ,OAAO,IAAI,CAAC,iBAAiB,EAAE,MAAM,EAAE,CAAC;IAC1C,CAAC;IAED,KAAK;QACH,uBAAuB;QACvB,MAAM,EAAE,GAAG,oBAAoB,EAAE,CAAC;QAElC,MAAM,WAAW,GAAG,IAAI,mBAAmB,CAAC,EAAE,CAAC,CAAC;QAEhD,IAAI,CAAC,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAC5B,IAAI,CAAC,iBAAiB,GAAG,WAAW,CAAC;QACvC,CAAC;QAED,IAAI,CAAC,oBAAoB,CAAC,GAAG,CAAC,EAAE,EAAE,WAAW,CAAC,CAAC;QAE/C,OAAO,WAAW,CAAC;IACrB,CAAC;IAED,IAAI,CAAC,WAAgC;QACnC,WAAW,CAAC,IAAI,EAAE,CAAC;QAEnB,MAAM,MAAM,GAAG,WAAW,CAAC,MAAM,EAAE,CAAC;QAEpC,IAAI,IAAI,CAAC,oBAAoB,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE,CAAC,EAAE,CAAC;YAClD,IAAI,CAAC,oBAAoB,CAAC,MAAM,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;QACnD,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,UAAU,CAAI,EAAoB;QACtC,MAAM,OAAO,GAAG,oBAAoB,EAAE,CAAC;QAEvC,MAAM,UAAU,GAAG,oBAAK,CAAC,UAAU,EAAE,CAAC;QAEtC,IAAI,CAAC;YACH,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,EAAE,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,CAAC;YAEjD,KAAK,MAAM,WAAW,IAAI,IAAI,CAAC,oBAAoB,CAAC,MAAM,EAAE,EAAE,CAAC;gBAC7D,WAAW,CAAC,aAAa,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;YACjD,CAAC;YAED,OAAO,MAAM,EAAE,EAAE,CAAC;QACpB,CAAC;gBAAS,CAAC;YACT,MAAM,QAAQ,GAAG,oBAAK,CAAC,UAAU,EAAE,CAAC;YAEpC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,EAAE,EAAE,KAAK,EAAE,UAAU,EAAE,GAAG,EAAE,QAAQ,EAAE,CAAC,CAAC;YAEhE,KAAK,MAAM,WAAW,IAAI,IAAI,CAAC,oBAAoB,CAAC,MAAM,EAAE,EAAE,CAAC;gBAC7D,WAAW,CAAC,aAAa,CAAC,OAAO,EAAE,UAAU,EAAE,QAAQ,CAAC,CAAC;YAC3D,CAAC;QACH,CAAC;IACH,CAAC;CACF;AArFD,0CAqFC;AAED,SAAS,oBAAoB;IAC3B,IAAI,MAAM,GAAG,EAAE,CAAC;IAChB,MAAM,UAAU,GAAG,gEAAgE,CAAC;IACpF,MAAM,gBAAgB,GAAG,UAAU,CAAC,MAAM,CAAC;IAE3C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;QAC5B,MAAM,IAAI,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,gBAAgB,CAAC,CAAC,CAAC;IAC5E,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC"}

package/dist/commonjs/v3/usage/noopUsageManager.d.ts

@@ -1,8 +1,11 @@
-import { UsageManager, UsageMeasurement, UsageSample } from "./types.js";
+import { InitialUsageState, UsageManager, UsageMeasurement, UsageSample } from "./types.js";
 export declare class NoopUsageManager implements UsageManager {
     disable(): void;
+    flush(): Promise<void>;
     start(): UsageMeasurement;
     stop(measurement: UsageMeasurement): UsageSample;
     pauseAsync<T>(cb: () => Promise<T>): Promise<T>;
     sample(): UsageSample | undefined;
+    reset(): void;
+    getInitialState(): InitialUsageState;
 }

package/dist/commonjs/v3/usage/noopUsageManager.js

@@ -5,6 +5,9 @@ class NoopUsageManager {
     disable() {
         // Noop
     }
+    async flush() {
+        // Noop
+    }
     start() {
         return {
             sample: () => ({ cpuTime: 0, wallTime: 0 }),
@@ -19,6 +22,15 @@ class NoopUsageManager {
     sample() {
         return undefined;
     }
+    reset() {
+        // Noop
+    }
+    getInitialState() {
+        return {
+            cpuTime: 0,
+            costInCents: 0,
+        };
+    }
 }
 exports.NoopUsageManager = NoopUsageManager;
 //# sourceMappingURL=noopUsageManager.js.map

package/dist/commonjs/v3/usage/noopUsageManager.js.map

@@ -1 +1 @@
-{"version":3,"file":"noopUsageManager.js","sourceRoot":"","sources":["../../../../src/v3/usage/noopUsageManager.ts"],"names":[],"mappings":";;;AAEA,MAAa,gBAAgB;IAC3B,OAAO;QACL,OAAO;IACT,CAAC;IAED,KAAK;QACH,OAAO;YACL,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC;SAC5C,CAAC;IACJ,CAAC;IAED,IAAI,CAAC,WAA6B;QAChC,OAAO,WAAW,CAAC,MAAM,EAAE,CAAC;IAC9B,CAAC;IAED,UAAU,CAAI,EAAoB;QAChC,OAAO,EAAE,EAAE,CAAC;IACd,CAAC;IAED,MAAM;QACJ,OAAO,SAAS,CAAC;IACnB,CAAC;CACF;AAtBD,4CAsBC"}
+{"version":3,"file":"noopUsageManager.js","sourceRoot":"","sources":["../../../../src/v3/usage/noopUsageManager.ts"],"names":[],"mappings":";;;AAEA,MAAa,gBAAgB;IAC3B,OAAO;QACL,OAAO;IACT,CAAC;IAED,KAAK,CAAC,KAAK;QACT,OAAO;IACT,CAAC;IAED,KAAK;QACH,OAAO;YACL,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC;SAC5C,CAAC;IACJ,CAAC;IAED,IAAI,CAAC,WAA6B;QAChC,OAAO,WAAW,CAAC,MAAM,EAAE,CAAC;IAC9B,CAAC;IAED,UAAU,CAAI,EAAoB;QAChC,OAAO,EAAE,EAAE,CAAC;IACd,CAAC;IAED,MAAM;QACJ,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,KAAK;QACH,OAAO;IACT,CAAC;IAED,eAAe;QACb,OAAO;YACL,OAAO,EAAE,CAAC;YACV,WAAW,EAAE,CAAC;SACf,CAAC;IACJ,CAAC;CACF;AArCD,4CAqCC"}

package/dist/commonjs/v3/usage/prodUsageManager.d.ts

@@ -1,4 +1,4 @@
-import { UsageManager, UsageMeasurement, UsageSample } from "./types.js";
+import { InitialUsageState, UsageManager, UsageMeasurement, UsageSample } from "./types.js";
 export type ProdUsageManagerOptions = {
     heartbeatIntervalMs?: number;
     url?: string;
@@ -12,8 +12,12 @@ export declare class ProdUsageManager implements UsageManager {
     private _abortController;
     private _lastSample;
     private _usageClient;
+    private _initialState;
     constructor(delegageUsageManager: UsageManager, options: ProdUsageManagerOptions);
     get isReportingEnabled(): boolean;
+    setInitialState(state: InitialUsageState): void;
+    getInitialState(): InitialUsageState;
+    reset(): void;
     disable(): void;
     sample(): UsageSample | undefined;
     start(): UsageMeasurement;

package/dist/commonjs/v3/usage/prodUsageManager.js

@@ -10,6 +10,10 @@ class ProdUsageManager {
     _abortController;
     _lastSample;
     _usageClient;
+    _initialState = {
+        cpuTime: 0,
+        costInCents: 0,
+    };
     constructor(delegageUsageManager, options) {
         this.delegageUsageManager = delegageUsageManager;
         this.options = options;
@@ -20,6 +24,24 @@ class ProdUsageManager {
     get isReportingEnabled() {
         return typeof this._usageClient !== "undefined";
     }
+    setInitialState(state) {
+        this._initialState = state;
+    }
+    getInitialState() {
+        return this._initialState;
+    }
+    reset() {
+        this.delegageUsageManager.reset();
+        this._abortController?.abort();
+        this._abortController = new AbortController();
+        this._usageClient = undefined;
+        this._measurement = undefined;
+        this._lastSample = undefined;
+        this._initialState = {
+            cpuTime: 0,
+            costInCents: 0,
+        };
+    }
     disable() {
         this.delegageUsageManager.disable();
         this._abortController?.abort();
@@ -49,11 +71,18 @@ class ProdUsageManager {
             return;
         }
         this._abortController = new AbortController();
-        for await (const _ of (0, promises_1.setInterval)(this.options.heartbeatIntervalMs)) {
-            if (this._abortController.signal.aborted) {
-                break;
+        try {
+            for await (const _ of (0, promises_1.setInterval)(this.options.heartbeatIntervalMs, undefined, {
+                signal: this._abortController.signal,
+            })) {
+                await this.#reportUsage();
+            }
+        }
+        catch (error) {
+            if (error instanceof Error && error.name === "AbortError") {
+                return;
             }
-            await this.#reportUsage();
+            throw error;
         }
     }
     async flush() {

package/dist/commonjs/v3/usage/prodUsageManager.js.map

@@ -1 +1 @@
-{"version":3,"file":"prodUsageManager.js","sourceRoot":"","sources":["../../../../src/v3/usage/prodUsageManager.ts"],"names":[],"mappings":";;;AAAA,mDAAmD;AAEnD,qDAA+C;AAQ/C,MAAa,gBAAgB;IAOR;IACA;IAPX,YAAY,CAA+B;IAC3C,gBAAgB,CAA8B;IAC9C,WAAW,CAA0B;IACrC,YAAY,CAA0B;IAE9C,YACmB,oBAAkC,EAClC,OAAgC;QADhC,yBAAoB,GAApB,oBAAoB,CAAc;QAClC,YAAO,GAAP,OAAO,CAAyB;QAEjD,IAAI,IAAI,CAAC,OAAO,CAAC,GAAG,IAAI,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC;YACzC,IAAI,CAAC,YAAY,GAAG,IAAI,4BAAW,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QAC1E,CAAC;IACH,CAAC;IAED,IAAI,kBAAkB;QACpB,OAAO,OAAO,IAAI,CAAC,YAAY,KAAK,WAAW,CAAC;IAClD,CAAC;IAED,OAAO;QACL,IAAI,CAAC,oBAAoB,CAAC,OAAO,EAAE,CAAC;QACpC,IAAI,CAAC,gBAAgB,EAAE,KAAK,EAAE,CAAC;IACjC,CAAC;IAED,MAAM;QACJ,OAAO,IAAI,CAAC,YAAY,EAAE,MAAM,EAAE,CAAC;IACrC,CAAC;IAED,KAAK;QACH,IAAI,CAAC,IAAI,CAAC,kBAAkB,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,mBAAmB,EAAE,CAAC;YAClE,OAAO,IAAI,CAAC,oBAAoB,CAAC,KAAK,EAAE,CAAC;QAC3C,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC;YACvB,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,oBAAoB,CAAC,KAAK,EAAE,CAAC;YAEtD,IAAI,CAAC,wBAAwB,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;YAErD,OAAO,IAAI,CAAC,YAAY,CAAC;QAC3B,CAAC;QAED,OAAO,IAAI,CAAC,oBAAoB,CAAC,KAAK,EAAE,CAAC;IAC3C,CAAC;IAED,IAAI,CAAC,WAA6B;QAChC,OAAO,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IACrD,CAAC;IAED,KAAK,CAAC,UAAU,CAAI,EAAoB;QACtC,OAAO,IAAI,CAAC,oBAAoB,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;IAClD,CAAC;IAED,KAAK,CAAC,wBAAwB;QAC5B,IAAI,CAAC,IAAI,CAAC,YAAY,IAAI,CAAC,IAAI,CAAC,kBAAkB,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,mBAAmB,EAAE,CAAC;YACxF,OAAO;QACT,CAAC;QAED,IAAI,CAAC,gBAAgB,GAAG,IAAI,eAAe,EAAE,CAAC;QAE9C,IAAI,KAAK,EAAE,MAAM,CAAC,IAAI,IAAA,sBAAW,EAAC,IAAI,CAAC,OAAO,CAAC,mBAAmB,CAAC,EAAE,CAAC;YACpE,IAAI,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;gBACzC,MAAM;YACR,CAAC;YAED,MAAM,IAAI,CAAC,YAAY,EAAE,CAAC;QAC5B,CAAC;IACH,CAAC;IAED,KAAK,CAAC,KAAK;QACT,OAAO,MAAM,IAAI,CAAC,YAAY,EAAE,CAAC;IACnC,CAAC;IAED,KAAK,CAAC,YAAY;QAChB,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC;YACvB,OAAO;QACT,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,kBAAkB,EAAE,CAAC;YAC7B,OAAO;QACT,CAAC;QAED,MAAM,MAAM,GAAG,IAAI,CAAC,YAAY,CAAC;QAEjC,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,OAAO;QACT,CAAC;QAED,MAAM,MAAM,GAAG,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,CAAC;QAE1C,MAAM,sBAAsB,GAAG,IAAI,CAAC,WAAW;YAC7C,CAAC,CAAC,MAAM,CAAC,OAAO,GAAG,IAAI,CAAC,WAAW,CAAC,OAAO;YAC3C,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC;QAEnB,IAAI,CAAC,WAAW,GAAG,MAAM,CAAC;QAE1B,IAAI,sBAAsB,IAAI,CAAC,EAAE,CAAC;YAChC,OAAO;QACT,CAAC;QAED,MAAM,MAAM,CAAC,cAAc,CAAC,EAAE,UAAU,EAAE,sBAAsB,EAAE,CAAC,CAAC;IACtE,CAAC;CACF;AArGD,4CAqGC"}
+{"version":3,"file":"prodUsageManager.js","sourceRoot":"","sources":["../../../../src/v3/usage/prodUsageManager.ts"],"names":[],"mappings":";;;AAAA,mDAAmD;AAEnD,qDAA+C;AAQ/C,MAAa,gBAAgB;IAWR;IACA;IAXX,YAAY,CAA+B;IAC3C,gBAAgB,CAA8B;IAC9C,WAAW,CAA0B;IACrC,YAAY,CAA0B;IACtC,aAAa,GAAsB;QACzC,OAAO,EAAE,CAAC;QACV,WAAW,EAAE,CAAC;KACf,CAAC;IAEF,YACmB,oBAAkC,EAClC,OAAgC;QADhC,yBAAoB,GAApB,oBAAoB,CAAc;QAClC,YAAO,GAAP,OAAO,CAAyB;QAEjD,IAAI,IAAI,CAAC,OAAO,CAAC,GAAG,IAAI,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC;YACzC,IAAI,CAAC,YAAY,GAAG,IAAI,4BAAW,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QAC1E,CAAC;IACH,CAAC;IAED,IAAI,kBAAkB;QACpB,OAAO,OAAO,IAAI,CAAC,YAAY,KAAK,WAAW,CAAC;IAClD,CAAC;IAED,eAAe,CAAC,KAAwB;QACtC,IAAI,CAAC,aAAa,GAAG,KAAK,CAAC;IAC7B,CAAC;IAED,eAAe;QACb,OAAO,IAAI,CAAC,aAAa,CAAC;IAC5B,CAAC;IAED,KAAK;QACH,IAAI,CAAC,oBAAoB,CAAC,KAAK,EAAE,CAAC;QAClC,IAAI,CAAC,gBAAgB,EAAE,KAAK,EAAE,CAAC;QAC/B,IAAI,CAAC,gBAAgB,GAAG,IAAI,eAAe,EAAE,CAAC;QAC9C,IAAI,CAAC,YAAY,GAAG,SAAS,CAAC;QAC9B,IAAI,CAAC,YAAY,GAAG,SAAS,CAAC;QAC9B,IAAI,CAAC,WAAW,GAAG,SAAS,CAAC;QAC7B,IAAI,CAAC,aAAa,GAAG;YACnB,OAAO,EAAE,CAAC;YACV,WAAW,EAAE,CAAC;SACf,CAAC;IACJ,CAAC;IAED,OAAO;QACL,IAAI,CAAC,oBAAoB,CAAC,OAAO,EAAE,CAAC;QACpC,IAAI,CAAC,gBAAgB,EAAE,KAAK,EAAE,CAAC;IACjC,CAAC;IAED,MAAM;QACJ,OAAO,IAAI,CAAC,YAAY,EAAE,MAAM,EAAE,CAAC;IACrC,CAAC;IAED,KAAK;QACH,IAAI,CAAC,IAAI,CAAC,kBAAkB,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,mBAAmB,EAAE,CAAC;YAClE,OAAO,IAAI,CAAC,oBAAoB,CAAC,KAAK,EAAE,CAAC;QAC3C,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC;YACvB,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,oBAAoB,CAAC,KAAK,EAAE,CAAC;YAEtD,IAAI,CAAC,wBAAwB,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;YAErD,OAAO,IAAI,CAAC,YAAY,CAAC;QAC3B,CAAC;QAED,OAAO,IAAI,CAAC,oBAAoB,CAAC,KAAK,EAAE,CAAC;IAC3C,CAAC;IAED,IAAI,CAAC,WAA6B;QAChC,OAAO,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IACrD,CAAC;IAED,KAAK,CAAC,UAAU,CAAI,EAAoB;QACtC,OAAO,IAAI,CAAC,oBAAoB,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;IAClD,CAAC;IAED,KAAK,CAAC,wBAAwB;QAC5B,IAAI,CAAC,IAAI,CAAC,YAAY,IAAI,CAAC,IAAI,CAAC,kBAAkB,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,mBAAmB,EAAE,CAAC;YACxF,OAAO;QACT,CAAC;QAED,IAAI,CAAC,gBAAgB,GAAG,IAAI,eAAe,EAAE,CAAC;QAE9C,IAAI,CAAC;YACH,IAAI,KAAK,EAAE,MAAM,CAAC,IAAI,IAAA,sBAAW,EAAC,IAAI,CAAC,OAAO,CAAC,mBAAmB,EAAE,SAAS,EAAE;gBAC7E,MAAM,EAAE,IAAI,CAAC,gBAAgB,CAAC,MAAM;aACrC,CAAC,EAAE,CAAC;gBACH,MAAM,IAAI,CAAC,YAAY,EAAE,CAAC;YAC5B,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,KAAK,YAAY,KAAK,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;gBAC1D,OAAO;YACT,CAAC;YAED,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED,KAAK,CAAC,KAAK;QACT,OAAO,MAAM,IAAI,CAAC,YAAY,EAAE,CAAC;IACnC,CAAC;IAED,KAAK,CAAC,YAAY;QAChB,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC;YACvB,OAAO;QACT,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,kBAAkB,EAAE,CAAC;YAC7B,OAAO;QACT,CAAC;QAED,MAAM,MAAM,GAAG,IAAI,CAAC,YAAY,CAAC;QAEjC,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,OAAO;QACT,CAAC;QAED,MAAM,MAAM,GAAG,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,CAAC;QAE1C,MAAM,sBAAsB,GAAG,IAAI,CAAC,WAAW;YAC7C,CAAC,CAAC,MAAM,CAAC,OAAO,GAAG,IAAI,CAAC,WAAW,CAAC,OAAO;YAC3C,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC;QAEnB,IAAI,CAAC,WAAW,GAAG,MAAM,CAAC;QAE1B,IAAI,sBAAsB,IAAI,CAAC,EAAE,CAAC;YAChC,OAAO;QACT,CAAC;QAED,MAAM,MAAM,CAAC,cAAc,CAAC,EAAE,UAAU,EAAE,sBAAsB,EAAE,CAAC,CAAC;IACtE,CAAC;CACF;AApID,4CAoIC"}

package/dist/commonjs/v3/usage/types.d.ts

@@ -5,10 +5,17 @@ export type UsageSample = {
 export interface UsageMeasurement {
     sample(): UsageSample;
 }
+export type InitialUsageState = {
+    cpuTime: number;
+    costInCents: number;
+};
 export interface UsageManager {
     disable(): void;
+    getInitialState(): InitialUsageState;
     start(): UsageMeasurement;
     stop(measurement: UsageMeasurement): UsageSample;
     sample(): UsageSample | undefined;
     pauseAsync<T>(cb: () => Promise<T>): Promise<T>;
+    flush(): Promise<void>;
+    reset(): void;
 }

package/dist/commonjs/v3/usage-api.d.ts

@@ -1,3 +1,4 @@
 import { UsageAPI } from "./usage/api.js";
 /** Entrypoint for usage API */
 export declare const usage: UsageAPI;
+export type { UsageMeasurement, UsageSample } from "./usage/types.js";

package/dist/commonjs/v3/utils/crypto.d.ts

@@ -0,0 +1,2 @@
+export declare function randomUUID(): Promise<string>;
+export declare function digestSHA256(data: string): Promise<string>;