@trigger.dev/core 0.0.0-prerelease-20241023140846 → 0.0.0-prerelease-20241119135607

package/dist/esm/v3/apiClient/index.d.ts

@@ -1,28 +1,40 @@
 import { z } from "zod";
 import { AddTagsRequestBody, BatchTaskRunExecutionResult, BatchTriggerTaskRequestBody, CreateEnvironmentVariableRequestBody, CreateScheduleOptions, ListRunResponseItem, ListScheduleOptions, RescheduleRunRequestBody, TaskRunExecutionResult, TriggerTaskRequestBody, UpdateEnvironmentVariableRequestBody, UpdateMetadataRequestBody, UpdateScheduleOptions } from "../schemas/index.js";
 import { ApiRequestOptions, CursorPagePromise, ZodFetchOptions, isRequestOptions } from "./core.js";
-import { CreateEnvironmentVariableParams, ImportEnvironmentVariablesParams, ListProjectRunsQueryParams, ListRunsQueryParams, UpdateEnvironmentVariableParams } from "./types.js";
-export type { CreateEnvironmentVariableParams, ImportEnvironmentVariablesParams, UpdateEnvironmentVariableParams, };
+import { RunShape, AnyRunShape, RunStreamCallback, RunSubscription, TaskRunShape, RealtimeRun } from "./runStream.js";
+import { CreateEnvironmentVariableParams, ImportEnvironmentVariablesParams, ListProjectRunsQueryParams, ListRunsQueryParams, SubscribeToRunsQueryParams, UpdateEnvironmentVariableParams } from "./types.js";
+import { AnyRunTypes, TriggerJwtOptions } from "../types/tasks.js";
+export type { CreateEnvironmentVariableParams, ImportEnvironmentVariablesParams, UpdateEnvironmentVariableParams, SubscribeToRunsQueryParams, };
 export type TriggerOptions = {
     spanParentAsLink?: boolean;
 };
+export type TriggerRequestOptions = ZodFetchOptions & {
+    publicAccessToken?: TriggerJwtOptions;
+};
+export type TriggerApiRequestOptions = ApiRequestOptions & {
+    publicAccessToken?: TriggerJwtOptions;
+};
 export { isRequestOptions };
 export type { ApiRequestOptions };
+export type { RunShape, AnyRunShape, TaskRunShape, RealtimeRun, RunStreamCallback, RunSubscription, };
 /**
  * Trigger.dev v3 API client
  */
 export declare class ApiClient {
     #private;
-    private readonly accessToken;
-    private readonly baseUrl;
+    readonly baseUrl: string;
+    readonly accessToken: string;
     private readonly defaultRequestOptions;
     constructor(baseUrl: string, accessToken: string, requestOptions?: ApiRequestOptions);
+    get fetchClient(): typeof fetch;
     getRunResult(runId: string, requestOptions?: ZodFetchOptions): Promise<TaskRunExecutionResult | undefined>;
     getBatchResults(batchId: string, requestOptions?: ZodFetchOptions): Promise<BatchTaskRunExecutionResult | undefined>;
-    triggerTask(taskId: string, body: TriggerTaskRequestBody, options?: TriggerOptions, requestOptions?: ZodFetchOptions): import("./core.js").ApiPromise<{
+    triggerTask(taskId: string, body: TriggerTaskRequestBody, options?: TriggerOptions, requestOptions?: TriggerRequestOptions): Promise<{
+        publicAccessToken: string;
         id: string;
     }>;
-    batchTriggerTask(taskId: string, body: BatchTriggerTaskRequestBody, options?: TriggerOptions, requestOptions?: ZodFetchOptions): import("./core.js").ApiPromise<{
+    batchTriggerTask(taskId: string, body: BatchTriggerTaskRequestBody, options?: TriggerOptions, requestOptions?: TriggerRequestOptions): Promise<{
+        publicAccessToken: string;
         batchId: string;
         runs: string[];
     }>;
@@ -33,37 +45,28 @@ export declare class ApiClient {
         presignedUrl: string;
     }>;
     retrieveRun(runId: string, requestOptions?: ZodFetchOptions): import("./core.js").ApiPromise<{
+        status: "WAITING_FOR_DEPLOY" | "QUEUED" | "EXECUTING" | "REATTEMPTING" | "FROZEN" | "COMPLETED" | "CANCELED" | "FAILED" | "CRASHED" | "INTERRUPTED" | "SYSTEM_FAILURE" | "DELAYED" | "EXPIRED" | "TIMED_OUT";
         id: string;
-        status: "CANCELED" | "COMPLETED" | "FAILED" | "TIMED_OUT" | "WAITING_FOR_DEPLOY" | "QUEUED" | "EXECUTING" | "REATTEMPTING" | "FROZEN" | "CRASHED" | "INTERRUPTED" | "SYSTEM_FAILURE" | "DELAYED" | "EXPIRED";
         tags: string[];
         isTest: boolean;
         createdAt: Date;
         durationMs: number;
         costInCents: number;
         baseCostInCents: number;
-        depth: number;
-        triggerFunction: "triggerAndWait" | "trigger" | "batchTriggerAndWait" | "batchTrigger";
         taskIdentifier: string;
-        isQueued: boolean;
-        isExecuting: boolean;
-        isCompleted: boolean;
-        isSuccess: boolean;
-        isFailed: boolean;
-        isCancelled: boolean;
-        updatedAt: Date;
         relatedRuns: {
             root?: {
+                status: "WAITING_FOR_DEPLOY" | "QUEUED" | "EXECUTING" | "REATTEMPTING" | "FROZEN" | "COMPLETED" | "CANCELED" | "FAILED" | "CRASHED" | "INTERRUPTED" | "SYSTEM_FAILURE" | "DELAYED" | "EXPIRED" | "TIMED_OUT";
                 id: string;
-                status: "CANCELED" | "COMPLETED" | "FAILED" | "TIMED_OUT" | "WAITING_FOR_DEPLOY" | "QUEUED" | "EXECUTING" | "REATTEMPTING" | "FROZEN" | "CRASHED" | "INTERRUPTED" | "SYSTEM_FAILURE" | "DELAYED" | "EXPIRED";
                 tags: string[];
                 isTest: boolean;
                 createdAt: Date;
                 durationMs: number;
                 costInCents: number;
                 baseCostInCents: number;
+                taskIdentifier: string;
                 depth: number;
                 triggerFunction: "triggerAndWait" | "trigger" | "batchTriggerAndWait" | "batchTrigger";
-                taskIdentifier: string;
                 isQueued: boolean;
                 isExecuting: boolean;
                 isCompleted: boolean;
@@ -82,17 +85,17 @@ export declare class ApiClient {
                 metadata?: Record<string, any> | undefined;
             } | undefined;
             parent?: {
+                status: "WAITING_FOR_DEPLOY" | "QUEUED" | "EXECUTING" | "REATTEMPTING" | "FROZEN" | "COMPLETED" | "CANCELED" | "FAILED" | "CRASHED" | "INTERRUPTED" | "SYSTEM_FAILURE" | "DELAYED" | "EXPIRED" | "TIMED_OUT";
                 id: string;
-                status: "CANCELED" | "COMPLETED" | "FAILED" | "TIMED_OUT" | "WAITING_FOR_DEPLOY" | "QUEUED" | "EXECUTING" | "REATTEMPTING" | "FROZEN" | "CRASHED" | "INTERRUPTED" | "SYSTEM_FAILURE" | "DELAYED" | "EXPIRED";
                 tags: string[];
                 isTest: boolean;
                 createdAt: Date;
                 durationMs: number;
                 costInCents: number;
                 baseCostInCents: number;
+                taskIdentifier: string;
                 depth: number;
                 triggerFunction: "triggerAndWait" | "trigger" | "batchTriggerAndWait" | "batchTrigger";
-                taskIdentifier: string;
                 isQueued: boolean;
                 isExecuting: boolean;
                 isCompleted: boolean;
@@ -111,17 +114,17 @@ export declare class ApiClient {
                 metadata?: Record<string, any> | undefined;
             } | undefined;
             children?: {
+                status: "WAITING_FOR_DEPLOY" | "QUEUED" | "EXECUTING" | "REATTEMPTING" | "FROZEN" | "COMPLETED" | "CANCELED" | "FAILED" | "CRASHED" | "INTERRUPTED" | "SYSTEM_FAILURE" | "DELAYED" | "EXPIRED" | "TIMED_OUT";
                 id: string;
-                status: "CANCELED" | "COMPLETED" | "FAILED" | "TIMED_OUT" | "WAITING_FOR_DEPLOY" | "QUEUED" | "EXECUTING" | "REATTEMPTING" | "FROZEN" | "CRASHED" | "INTERRUPTED" | "SYSTEM_FAILURE" | "DELAYED" | "EXPIRED";
                 tags: string[];
                 isTest: boolean;
                 createdAt: Date;
                 durationMs: number;
                 costInCents: number;
                 baseCostInCents: number;
+                taskIdentifier: string;
                 depth: number;
                 triggerFunction: "triggerAndWait" | "trigger" | "batchTriggerAndWait" | "batchTrigger";
-                taskIdentifier: string;
                 isQueued: boolean;
                 isExecuting: boolean;
                 isCompleted: boolean;
@@ -140,9 +143,18 @@ export declare class ApiClient {
                 metadata?: Record<string, any> | undefined;
             }[] | undefined;
         };
+        depth: number;
+        triggerFunction: "triggerAndWait" | "trigger" | "batchTriggerAndWait" | "batchTrigger";
+        isQueued: boolean;
+        isExecuting: boolean;
+        isCompleted: boolean;
+        isSuccess: boolean;
+        isFailed: boolean;
+        isCancelled: boolean;
+        updatedAt: Date;
         attempts: ({
+            status: "EXECUTING" | "COMPLETED" | "CANCELED" | "FAILED" | "PENDING" | "PAUSED";
             id: string;
-            status: "PENDING" | "CANCELED" | "COMPLETED" | "FAILED" | "EXECUTING" | "PAUSED";
             createdAt: Date;
             updatedAt: Date;
             startedAt?: Date | undefined;
@@ -153,16 +165,22 @@ export declare class ApiClient {
                 stackTrace?: string | undefined;
             } | undefined;
         } | undefined)[];
+        attemptCount: number;
         payload?: any;
         payloadPresignedUrl?: string | undefined;
         output?: any;
         outputPresignedUrl?: string | undefined;
+        error?: {
+            message: string;
+            name?: string | undefined;
+            stackTrace?: string | undefined;
+        } | undefined;
         schedule?: {
             id: string;
             generator: {
+                description: string;
                 type: "CRON";
                 expression: string;
-                description: string;
             };
             externalId?: string | undefined;
             deduplicationKey?: string | undefined;
@@ -186,37 +204,28 @@ export declare class ApiClient {
         id: string;
     }>;
     rescheduleRun(runId: string, body: RescheduleRunRequestBody, requestOptions?: ZodFetchOptions): import("./core.js").ApiPromise<{
+        status: "WAITING_FOR_DEPLOY" | "QUEUED" | "EXECUTING" | "REATTEMPTING" | "FROZEN" | "COMPLETED" | "CANCELED" | "FAILED" | "CRASHED" | "INTERRUPTED" | "SYSTEM_FAILURE" | "DELAYED" | "EXPIRED" | "TIMED_OUT";
         id: string;
-        status: "CANCELED" | "COMPLETED" | "FAILED" | "TIMED_OUT" | "WAITING_FOR_DEPLOY" | "QUEUED" | "EXECUTING" | "REATTEMPTING" | "FROZEN" | "CRASHED" | "INTERRUPTED" | "SYSTEM_FAILURE" | "DELAYED" | "EXPIRED";
         tags: string[];
         isTest: boolean;
         createdAt: Date;
         durationMs: number;
         costInCents: number;
         baseCostInCents: number;
-        depth: number;
-        triggerFunction: "triggerAndWait" | "trigger" | "batchTriggerAndWait" | "batchTrigger";
         taskIdentifier: string;
-        isQueued: boolean;
-        isExecuting: boolean;
-        isCompleted: boolean;
-        isSuccess: boolean;
-        isFailed: boolean;
-        isCancelled: boolean;
-        updatedAt: Date;
         relatedRuns: {
             root?: {
+                status: "WAITING_FOR_DEPLOY" | "QUEUED" | "EXECUTING" | "REATTEMPTING" | "FROZEN" | "COMPLETED" | "CANCELED" | "FAILED" | "CRASHED" | "INTERRUPTED" | "SYSTEM_FAILURE" | "DELAYED" | "EXPIRED" | "TIMED_OUT";
                 id: string;
-                status: "CANCELED" | "COMPLETED" | "FAILED" | "TIMED_OUT" | "WAITING_FOR_DEPLOY" | "QUEUED" | "EXECUTING" | "REATTEMPTING" | "FROZEN" | "CRASHED" | "INTERRUPTED" | "SYSTEM_FAILURE" | "DELAYED" | "EXPIRED";
                 tags: string[];
                 isTest: boolean;
                 createdAt: Date;
                 durationMs: number;
                 costInCents: number;
                 baseCostInCents: number;
+                taskIdentifier: string;
                 depth: number;
                 triggerFunction: "triggerAndWait" | "trigger" | "batchTriggerAndWait" | "batchTrigger";
-                taskIdentifier: string;
                 isQueued: boolean;
                 isExecuting: boolean;
                 isCompleted: boolean;
@@ -235,17 +244,17 @@ export declare class ApiClient {
                 metadata?: Record<string, any> | undefined;
             } | undefined;
             parent?: {
+                status: "WAITING_FOR_DEPLOY" | "QUEUED" | "EXECUTING" | "REATTEMPTING" | "FROZEN" | "COMPLETED" | "CANCELED" | "FAILED" | "CRASHED" | "INTERRUPTED" | "SYSTEM_FAILURE" | "DELAYED" | "EXPIRED" | "TIMED_OUT";
                 id: string;
-                status: "CANCELED" | "COMPLETED" | "FAILED" | "TIMED_OUT" | "WAITING_FOR_DEPLOY" | "QUEUED" | "EXECUTING" | "REATTEMPTING" | "FROZEN" | "CRASHED" | "INTERRUPTED" | "SYSTEM_FAILURE" | "DELAYED" | "EXPIRED";
                 tags: string[];
                 isTest: boolean;
                 createdAt: Date;
                 durationMs: number;
                 costInCents: number;
                 baseCostInCents: number;
+                taskIdentifier: string;
                 depth: number;
                 triggerFunction: "triggerAndWait" | "trigger" | "batchTriggerAndWait" | "batchTrigger";
-                taskIdentifier: string;
                 isQueued: boolean;
                 isExecuting: boolean;
                 isCompleted: boolean;
@@ -264,17 +273,17 @@ export declare class ApiClient {
                 metadata?: Record<string, any> | undefined;
             } | undefined;
             children?: {
+                status: "WAITING_FOR_DEPLOY" | "QUEUED" | "EXECUTING" | "REATTEMPTING" | "FROZEN" | "COMPLETED" | "CANCELED" | "FAILED" | "CRASHED" | "INTERRUPTED" | "SYSTEM_FAILURE" | "DELAYED" | "EXPIRED" | "TIMED_OUT";
                 id: string;
-                status: "CANCELED" | "COMPLETED" | "FAILED" | "TIMED_OUT" | "WAITING_FOR_DEPLOY" | "QUEUED" | "EXECUTING" | "REATTEMPTING" | "FROZEN" | "CRASHED" | "INTERRUPTED" | "SYSTEM_FAILURE" | "DELAYED" | "EXPIRED";
                 tags: string[];
                 isTest: boolean;
                 createdAt: Date;
                 durationMs: number;
                 costInCents: number;
                 baseCostInCents: number;
+                taskIdentifier: string;
                 depth: number;
                 triggerFunction: "triggerAndWait" | "trigger" | "batchTriggerAndWait" | "batchTrigger";
-                taskIdentifier: string;
                 isQueued: boolean;
                 isExecuting: boolean;
                 isCompleted: boolean;
@@ -293,9 +302,18 @@ export declare class ApiClient {
                 metadata?: Record<string, any> | undefined;
             }[] | undefined;
         };
+        depth: number;
+        triggerFunction: "triggerAndWait" | "trigger" | "batchTriggerAndWait" | "batchTrigger";
+        isQueued: boolean;
+        isExecuting: boolean;
+        isCompleted: boolean;
+        isSuccess: boolean;
+        isFailed: boolean;
+        isCancelled: boolean;
+        updatedAt: Date;
         attempts: ({
+            status: "EXECUTING" | "COMPLETED" | "CANCELED" | "FAILED" | "PENDING" | "PAUSED";
             id: string;
-            status: "PENDING" | "CANCELED" | "COMPLETED" | "FAILED" | "EXECUTING" | "PAUSED";
             createdAt: Date;
             updatedAt: Date;
             startedAt?: Date | undefined;
@@ -306,16 +324,22 @@ export declare class ApiClient {
                 stackTrace?: string | undefined;
             } | undefined;
         } | undefined)[];
+        attemptCount: number;
         payload?: any;
         payloadPresignedUrl?: string | undefined;
         output?: any;
         outputPresignedUrl?: string | undefined;
+        error?: {
+            message: string;
+            name?: string | undefined;
+            stackTrace?: string | undefined;
+        } | undefined;
         schedule?: {
             id: string;
             generator: {
+                description: string;
                 type: "CRON";
                 expression: string;
-                description: string;
             };
             externalId?: string | undefined;
             deduplicationKey?: string | undefined;
@@ -334,19 +358,19 @@ export declare class ApiClient {
         message: string;
     }>;
     createSchedule(options: CreateScheduleOptions, requestOptions?: ZodFetchOptions): import("./core.js").ApiPromise<{
-        id: string;
         type: "DECLARATIVE" | "IMPERATIVE";
+        id: string;
         task: string;
         timezone: string;
-        active: boolean;
         generator: {
+            description: string;
             type: "CRON";
             expression: string;
-            description: string;
         };
+        active: boolean;
         environments: {
-            id: string;
             type: string;
+            id: string;
             userName?: string | null | undefined;
         }[];
         deduplicationKey?: string | null | undefined;
@@ -365,13 +389,13 @@ export declare class ApiClient {
             expression: z.ZodString;
             description: z.ZodString;
         }, "strip", z.ZodTypeAny, {
+            description: string;
             type: "CRON";
             expression: string;
-            description: string;
         }, {
+            description: string;
             type: "CRON";
             expression: string;
-            description: string;
         }>;
         timezone: z.ZodString;
         nextRun: z.ZodOptional<z.ZodNullable<z.ZodDate>>;
@@ -380,47 +404,47 @@ export declare class ApiClient {
             type: z.ZodString;
             userName: z.ZodOptional<z.ZodNullable<z.ZodString>>;
         }, "strip", z.ZodTypeAny, {
-            id: string;
             type: string;
+            id: string;
             userName?: string | null | undefined;
         }, {
-            id: string;
             type: string;
+            id: string;
             userName?: string | null | undefined;
         }>, "many">;
     }, "strip", z.ZodTypeAny, {
-        id: string;
         type: "DECLARATIVE" | "IMPERATIVE";
+        id: string;
         task: string;
         timezone: string;
-        active: boolean;
         generator: {
+            description: string;
             type: "CRON";
             expression: string;
-            description: string;
         };
+        active: boolean;
         environments: {
-            id: string;
             type: string;
+            id: string;
             userName?: string | null | undefined;
         }[];
         deduplicationKey?: string | null | undefined;
         externalId?: string | null | undefined;
         nextRun?: Date | null | undefined;
     }, {
-        id: string;
         type: "DECLARATIVE" | "IMPERATIVE";
+        id: string;
         task: string;
         timezone: string;
-        active: boolean;
         generator: {
+            description: string;
             type: "CRON";
             expression: string;
-            description: string;
         };
+        active: boolean;
         environments: {
-            id: string;
             type: string;
+            id: string;
             userName?: string | null | undefined;
         }[];
         deduplicationKey?: string | null | undefined;
@@ -428,19 +452,19 @@ export declare class ApiClient {
         nextRun?: Date | null | undefined;
     }>>;
     retrieveSchedule(scheduleId: string, requestOptions?: ZodFetchOptions): import("./core.js").ApiPromise<{
-        id: string;
         type: "DECLARATIVE" | "IMPERATIVE";
+        id: string;
         task: string;
         timezone: string;
-        active: boolean;
         generator: {
+            description: string;
             type: "CRON";
             expression: string;
-            description: string;
         };
+        active: boolean;
         environments: {
-            id: string;
             type: string;
+            id: string;
             userName?: string | null | undefined;
         }[];
         deduplicationKey?: string | null | undefined;
@@ -448,19 +472,19 @@ export declare class ApiClient {
         nextRun?: Date | null | undefined;
     }>;
     updateSchedule(scheduleId: string, options: UpdateScheduleOptions, requestOptions?: ZodFetchOptions): import("./core.js").ApiPromise<{
-        id: string;
         type: "DECLARATIVE" | "IMPERATIVE";
+        id: string;
         task: string;
         timezone: string;
-        active: boolean;
         generator: {
+            description: string;
             type: "CRON";
             expression: string;
-            description: string;
         };
+        active: boolean;
         environments: {
-            id: string;
             type: string;
+            id: string;
             userName?: string | null | undefined;
         }[];
         deduplicationKey?: string | null | undefined;
@@ -468,19 +492,19 @@ export declare class ApiClient {
         nextRun?: Date | null | undefined;
     }>;
     deactivateSchedule(scheduleId: string, requestOptions?: ZodFetchOptions): import("./core.js").ApiPromise<{
-        id: string;
         type: "DECLARATIVE" | "IMPERATIVE";
+        id: string;
         task: string;
         timezone: string;
-        active: boolean;
         generator: {
+            description: string;
             type: "CRON";
             expression: string;
-            description: string;
         };
+        active: boolean;
         environments: {
-            id: string;
             type: string;
+            id: string;
             userName?: string | null | undefined;
         }[];
         deduplicationKey?: string | null | undefined;
@@ -488,19 +512,19 @@ export declare class ApiClient {
         nextRun?: Date | null | undefined;
     }>;
     activateSchedule(scheduleId: string, requestOptions?: ZodFetchOptions): import("./core.js").ApiPromise<{
-        id: string;
         type: "DECLARATIVE" | "IMPERATIVE";
+        id: string;
         task: string;
         timezone: string;
-        active: boolean;
         generator: {
+            description: string;
             type: "CRON";
             expression: string;
-            description: string;
         };
+        active: boolean;
         environments: {
-            id: string;
             type: string;
+            id: string;
             userName?: string | null | undefined;
         }[];
         deduplicationKey?: string | null | undefined;
@@ -511,8 +535,8 @@ export declare class ApiClient {
         id: string;
     }>;
     listEnvVars(projectRef: string, slug: string, requestOptions?: ZodFetchOptions): import("./core.js").ApiPromise<{
-        value: string;
         name: string;
+        value: string;
     }[]>;
     importEnvVars(projectRef: string, slug: string, body: ImportEnvironmentVariablesParams, requestOptions?: ZodFetchOptions): import("./core.js").ApiPromise<{
         success: boolean;
@@ -532,5 +556,15 @@ export declare class ApiClient {
     updateRunMetadata(runId: string, body: UpdateMetadataRequestBody, requestOptions?: ZodFetchOptions): import("./core.js").ApiPromise<{
         metadata: Record<string, import("../../index.js").DeserializedJson>;
     }>;
+    subscribeToRun<TRunTypes extends AnyRunTypes>(runId: string, options?: {
+        signal?: AbortSignal;
+    }): RunSubscription<TRunTypes>;
+    subscribeToRunsWithTag<TRunTypes extends AnyRunTypes>(tag: string | string[], options?: {
+        signal?: AbortSignal;
+    }): RunSubscription<TRunTypes>;
+    subscribeToBatch<TRunTypes extends AnyRunTypes>(batchId: string, options?: {
+        signal?: AbortSignal;
+    }): RunSubscription<TRunTypes>;
+    generateJWTClaims(requestOptions?: ZodFetchOptions): Promise<Record<string, any>>;
 }
 export declare function mergeRequestOptions(defaultOptions: ZodFetchOptions, options?: ApiRequestOptions): ZodFetchOptions;

package/dist/esm/v3/apiClient/index.js

@@ -1,9 +1,11 @@
 import { z } from "zod";
+import { VERSION } from "../../version.js";
 import { BatchTaskRunExecutionResult, BatchTriggerTaskResponse, CanceledRunResponse, CreateUploadPayloadUrlResponseBody, DeletedScheduleObject, EnvironmentVariableResponseBody, EnvironmentVariableValue, EnvironmentVariables, ListRunResponseItem, ReplayRunResponse, RetrieveRunResponse, ScheduleObject, TaskRunExecutionResult, TriggerTaskResponse, UpdateMetadataResponseBody, } from "../schemas/index.js";
 import { taskContext } from "../task-context-api.js";
 import { isRequestOptions, zodfetch, zodfetchCursorPage, zodfetchOffsetLimitPage, } from "./core.js";
 import { ApiError } from "./errors.js";
-import { VERSION } from "../../version.js";
+import { runShapeStream, } from "./runStream.js";
+import { generateJWT } from "../jwt.js";
 const DEFAULT_ZOD_FETCH_OPTIONS = {
     retry: {
         maxAttempts: 3,
@@ -18,14 +20,28 @@ export { isRequestOptions };
  * Trigger.dev v3 API client
  */
 export class ApiClient {
-    accessToken;
     baseUrl;
+    accessToken;
     defaultRequestOptions;
     constructor(baseUrl, accessToken, requestOptions = {}) {
         this.accessToken = accessToken;
         this.baseUrl = baseUrl.replace(/\/$/, "");
         this.defaultRequestOptions = mergeRequestOptions(DEFAULT_ZOD_FETCH_OPTIONS, requestOptions);
     }
+    get fetchClient() {
+        const headers = this.#getHeaders(false);
+        const fetchClient = (input, requestInit) => {
+            const $requestInit = {
+                ...requestInit,
+                headers: {
+                    ...requestInit?.headers,
+                    ...headers,
+                },
+            };
+            return fetch(input, $requestInit);
+        };
+        return fetchClient;
+    }
     async getRunResult(runId, requestOptions) {
         try {
             return await zodfetch(TaskRunExecutionResult, `${this.baseUrl}/api/v1/runs/${runId}/result`, {
@@ -54,7 +70,31 @@ export class ApiClient {
             method: "POST",
             headers: this.#getHeaders(options?.spanParentAsLink ?? false),
             body: JSON.stringify(body),
-        }, mergeRequestOptions(this.defaultRequestOptions, requestOptions));
+        }, mergeRequestOptions(this.defaultRequestOptions, requestOptions))
+            .withResponse()
+            .then(async ({ response, data }) => {
+            const jwtHeader = response.headers.get("x-trigger-jwt");
+            if (typeof jwtHeader === "string") {
+                return {
+                    ...data,
+                    publicAccessToken: jwtHeader,
+                };
+            }
+            const claimsHeader = response.headers.get("x-trigger-jwt-claims");
+            const claims = claimsHeader ? JSON.parse(claimsHeader) : undefined;
+            const jwt = await generateJWT({
+                secretKey: this.accessToken,
+                payload: {
+                    ...claims,
+                    scopes: [`read:runs:${data.id}`].concat(body.options?.tags ? Array.from(body.options?.tags).map((t) => `read:tags:${t}`) : []),
+                },
+                expirationTime: requestOptions?.publicAccessToken?.expirationTime ?? "1h",
+            });
+            return {
+                ...data,
+                publicAccessToken: jwt,
+            };
+        });
     }
     batchTriggerTask(taskId, body, options, requestOptions) {
         const encodedTaskId = encodeURIComponent(taskId);
@@ -62,7 +102,24 @@ export class ApiClient {
             method: "POST",
             headers: this.#getHeaders(options?.spanParentAsLink ?? false),
             body: JSON.stringify(body),
-        }, mergeRequestOptions(this.defaultRequestOptions, requestOptions));
+        }, mergeRequestOptions(this.defaultRequestOptions, requestOptions))
+            .withResponse()
+            .then(async ({ response, data }) => {
+            const claimsHeader = response.headers.get("x-trigger-jwt-claims");
+            const claims = claimsHeader ? JSON.parse(claimsHeader) : undefined;
+            const jwt = await generateJWT({
+                secretKey: this.accessToken,
+                payload: {
+                    ...claims,
+                    scopes: [`read:batch:${data.batchId}`].concat(data.runs.map((r) => `read:runs:${r}`)),
+                },
+                expirationTime: requestOptions?.publicAccessToken?.expirationTime ?? "1h",
+            });
+            return {
+                ...data,
+                publicAccessToken: jwt,
+            };
+        });
     }
     createUploadPayloadUrl(filename, requestOptions) {
         return zodfetch(CreateUploadPayloadUrlResponseBody, `${this.baseUrl}/api/v1/packets/${filename}`, {
@@ -235,6 +292,39 @@ export class ApiClient {
             body: JSON.stringify(body),
         }, mergeRequestOptions(this.defaultRequestOptions, requestOptions));
     }
+    subscribeToRun(runId, options) {
+        return runShapeStream(`${this.baseUrl}/realtime/v1/runs/${runId}`, {
+            closeOnComplete: true,
+            headers: this.#getRealtimeHeaders(),
+            client: this,
+            signal: options?.signal,
+        });
+    }
+    subscribeToRunsWithTag(tag, options) {
+        const searchParams = createSearchQueryForSubscribeToRuns({
+            tags: tag,
+        });
+        return runShapeStream(`${this.baseUrl}/realtime/v1/runs${searchParams ? `?${searchParams}` : ""}`, {
+            closeOnComplete: false,
+            headers: this.#getRealtimeHeaders(),
+            client: this,
+            signal: options?.signal,
+        });
+    }
+    subscribeToBatch(batchId, options) {
+        return runShapeStream(`${this.baseUrl}/realtime/v1/batches/${batchId}`, {
+            closeOnComplete: false,
+            headers: this.#getRealtimeHeaders(),
+            client: this,
+            signal: options?.signal,
+        });
+    }
+    async generateJWTClaims(requestOptions) {
+        return zodfetch(z.record(z.any()), `${this.baseUrl}/api/v1/auth/jwt/claims`, {
+            method: "POST",
+            headers: this.#getHeaders(false),
+        }, mergeRequestOptions(this.defaultRequestOptions, requestOptions));
+    }
     #getHeaders(spanParentAsLink) {
         const headers = {
             "Content-Type": "application/json",
@@ -248,9 +338,31 @@ export class ApiClient {
                 headers["x-trigger-span-parent-as-link"] = "1";
             }
         }
+        if (typeof window !== "undefined" && typeof window.document !== "undefined") {
+            headers["x-trigger-client"] = "browser";
+        }
+        return headers;
+    }
+    #getRealtimeHeaders() {
+        const headers = {
+            Authorization: `Bearer ${this.accessToken}`,
+            "trigger-version": VERSION,
+        };
         return headers;
     }
 }
+function createSearchQueryForSubscribeToRuns(query) {
+    const searchParams = new URLSearchParams();
+    if (query) {
+        if (query.tasks) {
+            searchParams.append("tasks", Array.isArray(query.tasks) ? query.tasks.join(",") : query.tasks);
+        }
+        if (query.tags) {
+            searchParams.append("tags", Array.isArray(query.tags) ? query.tags.join(",") : query.tags);
+        }
+    }
+    return searchParams;
+}
 function createSearchQueryForListRuns(query) {
     const searchParams = new URLSearchParams();
     if (query) {