@trigger.dev/react-hooks 4.5.13 → 4.5.15

package/dist/commonjs/hooks/useSessionStream.js

@@ -0,0 +1,236 @@
+"use strict";
+"use client";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.useSessionStream = useSessionStream;
+const react_1 = require("react");
+const throttle_js_1 = require("../utils/throttle.js");
+const trigger_swr_js_1 = require("../utils/trigger-swr.js");
+const useStableRequestCallback_js_1 = require("../utils/useStableRequestCallback.js");
+const useApiClient_js_1 = require("./useApiClient.js");
+/**
+ * Hook to read one channel of a Session's realtime stream.
+ *
+ * This hook subscribes to one of the session's channels (`out` by default, or `in`) and
+ * updates the `records` array as new records arrive. It is read-only: use `useSession` for
+ * two-way (read and write) communication. The subscription is automatically managed: it
+ * starts when the component mounts (or when `enabled` becomes `true`) and stops when the
+ * component unmounts or when `stop()` is called.
+ *
+ * Requires a Public Access Token with the `read:sessions:{id}` scope.
+ *
+ * @template TRecord - The type of each record on the channel
+ * @param sessionIdOrExternalId - The id or external id of the session to subscribe to
+ * @param options - Optional configuration for the subscription
+ * @returns An object containing:
+ *   - `records`: An array of all the records received so far (accumulates over time)
+ *   - `lastEventId`: The cursor of the last record seen, for resuming later
+ *   - `lastControl`: The last control record seen
+ *   - `error`: Any error that occurred during subscription
+ *   - `stop`: A function to manually stop the subscription
+ *
+ * @example
+ * ```tsx
+ * "use client";
+ * import { useSessionStream } from "@trigger.dev/react-hooks";
+ *
+ * function SessionViewer({ sessionId }: { sessionId: string }) {
+ *   const { records, error } = useSessionStream<string>(sessionId, {
+ *     accessToken: publicAccessToken,
+ *   });
+ *
+ *   if (error) return <div>Error: {error.message}</div>;
+ *
+ *   return <div>{records.join("")}</div>;
+ * }
+ * ```
+ *
+ * @example
+ * ```tsx
+ * // Read the input channel, resuming from a persisted cursor
+ * const { records, lastEventId, stop } = useSessionStream<MyRecord>(sessionId, {
+ *   accessToken: publicAccessToken,
+ *   io: "in",
+ *   lastEventId: persistedCursor,
+ *   onControl: (event) => {
+ *     if (event.subtype === "turn-complete") {
+ *       console.log("The turn is complete");
+ *     }
+ *   },
+ * });
+ * ```
+ */
+function useSessionStream(sessionIdOrExternalId, options) {
+    const hookId = (0, react_1.useId)();
+    const idKey = options?.id ?? hookId;
+    const io = options?.io ?? "out";
+    const [initialRecordsFallback] = (0, react_1.useState)([]);
+    const { data: records, mutate: mutateRecords } = (0, trigger_swr_js_1.useSWR)([idKey, sessionIdOrExternalId, io, "records"], null, {
+        fallbackData: initialRecordsFallback,
+    });
+    const recordsRef = (0, react_1.useRef)(records ?? []);
+    (0, react_1.useEffect)(() => {
+        recordsRef.current = records || [];
+    }, [records]);
+    const { data: lastEventId = undefined, mutate: setLastEventId } = (0, trigger_swr_js_1.useSWR)([idKey, sessionIdOrExternalId, io, "lastEventId"], null);
+    const lastEventIdRef = (0, react_1.useRef)(lastEventId);
+    const channelIdentityRef = (0, react_1.useRef)(`${idKey}:${sessionIdOrExternalId}:${io}`);
+    (0, react_1.useEffect)(() => {
+        const identity = `${idKey}:${sessionIdOrExternalId}:${io}`;
+        if (channelIdentityRef.current !== identity) {
+            channelIdentityRef.current = identity;
+            lastEventIdRef.current = lastEventId;
+        }
+    }, [idKey, sessionIdOrExternalId, io, lastEventId]);
+    const { data: lastControl = undefined, mutate: setLastControl } = (0, trigger_swr_js_1.useSWR)([idKey, sessionIdOrExternalId, io, "lastControl"], null);
+    const { data: _isComplete = false, mutate: setIsComplete } = (0, trigger_swr_js_1.useSWR)([idKey, sessionIdOrExternalId, io, "complete"], null);
+    const { data: error = undefined, mutate: setError } = (0, trigger_swr_js_1.useSWR)([idKey, sessionIdOrExternalId, io, "error"], null);
+    const abortControllerRef = (0, react_1.useRef)(null);
+    const stop = (0, react_1.useCallback)(() => {
+        if (abortControllerRef.current) {
+            abortControllerRef.current.abort();
+            abortControllerRef.current = null;
+        }
+    }, []);
+    const onRecordsCallback = options?.onRecords;
+    const onRecords = (0, react_1.useCallback)((recordsBatch) => {
+        if (onRecordsCallback) {
+            onRecordsCallback(recordsBatch);
+        }
+    }, [onRecordsCallback]);
+    const onControlCallback = options?.onControl;
+    const onControl = (0, react_1.useCallback)((event) => {
+        if (onControlCallback) {
+            onControlCallback(event);
+        }
+    }, [onControlCallback]);
+    const apiClient = (0, useApiClient_js_1.useApiClient)(options);
+    const timeoutInSeconds = options?.timeoutInSeconds;
+    const startEventId = options?.lastEventId;
+    const throttleInMs = options?.throttleInMs;
+    const from = options?.from;
+    const maxRecords = options?.maxRecords;
+    (0, react_1.useEffect)(() => {
+        if (maxRecords != null && maxRecords >= 0) {
+            const current = recordsRef.current;
+            if (current.length > maxRecords) {
+                mutateRecords(current.slice(current.length - maxRecords));
+            }
+        }
+    }, [maxRecords, mutateRecords]);
+    const triggerRequest = (0, react_1.useCallback)(async () => {
+        try {
+            if (!sessionIdOrExternalId || !apiClient) {
+                return;
+            }
+            const abortController = new AbortController();
+            abortControllerRef.current = abortController;
+            await processSessionStream(sessionIdOrExternalId, io, apiClient, mutateRecords, recordsRef, setLastEventId, setLastControl, setError, onRecords, onControl, abortControllerRef, timeoutInSeconds, startEventId !== undefined ? String(startEventId) : lastEventIdRef.current, throttleInMs ?? 16, from, maxRecords);
+        }
+        catch (err) {
+            if (err.name === "AbortError") {
+                abortControllerRef.current = null;
+                return;
+            }
+            setError(err);
+        }
+        finally {
+            if (abortControllerRef.current) {
+                abortControllerRef.current = null;
+            }
+            setIsComplete(true);
+        }
+    }, [
+        sessionIdOrExternalId,
+        io,
+        apiClient,
+        mutateRecords,
+        setLastEventId,
+        setLastControl,
+        setError,
+        setIsComplete,
+        onRecords,
+        onControl,
+        timeoutInSeconds,
+        startEventId,
+        throttleInMs,
+        from,
+        maxRecords,
+    ]);
+    const requestSubscription = (0, useStableRequestCallback_js_1.useStableRequestCallback)(triggerRequest);
+    (0, react_1.useEffect)(() => {
+        if (typeof options?.enabled === "boolean" && !options.enabled) {
+            return;
+        }
+        if (!sessionIdOrExternalId) {
+            return;
+        }
+        requestSubscription().finally(() => { });
+        return () => {
+            stop();
+        };
+    }, [sessionIdOrExternalId, io, stop, options?.enabled, requestSubscription]);
+    return { records: records ?? initialRecordsFallback, lastEventId, lastControl, error, stop };
+}
+async function processSessionStream(sessionIdOrExternalId, io, apiClient, mutateRecordsData, existingRecordsRef, setLastEventId, setLastControl, onError, onRecords, onControl, abortControllerRef, timeoutInSeconds, lastEventId, throttleInMs, from, maxRecords) {
+    let lastSeenEventId;
+    let publishedEventId;
+    let partsBatch = [];
+    const publishLastEventId = () => {
+        if (lastSeenEventId !== publishedEventId) {
+            publishedEventId = lastSeenEventId;
+            setLastEventId(lastSeenEventId);
+        }
+    };
+    const flushParts = () => {
+        if (partsBatch.length === 0)
+            return;
+        const batch = partsBatch;
+        partsBatch = [];
+        onRecords(batch);
+    };
+    try {
+        const stream = await apiClient.subscribeToSessionStream(sessionIdOrExternalId, io, {
+            signal: abortControllerRef.current?.signal,
+            timeoutInSeconds,
+            lastEventId,
+            from,
+            onPart: (part) => {
+                lastSeenEventId = part.id;
+                partsBatch.push(part);
+            },
+            onControl: (event) => {
+                setLastControl(event);
+                onControl(event);
+            },
+        });
+        const recordsQueue = (0, throttle_js_1.createThrottledQueue)(async (newRecords) => {
+            const combined = [...existingRecordsRef.current, ...newRecords];
+            const bounded = maxRecords != null && maxRecords >= 0 && combined.length > maxRecords
+                ? combined.slice(combined.length - maxRecords)
+                : combined;
+            existingRecordsRef.current = bounded;
+            mutateRecordsData(bounded);
+            publishLastEventId();
+            flushParts();
+        }, throttleInMs);
+        for await (const record of stream) {
+            recordsQueue.add(record);
+        }
+        await recordsQueue.flush();
+        publishLastEventId();
+        flushParts();
+    }
+    catch (err) {
+        if (err.name === "AbortError") {
+            return;
+        }
+        if (err instanceof Error) {
+            onError(err);
+        }
+        else {
+            onError(new Error(String(err)));
+        }
+        throw err;
+    }
+}
+//# sourceMappingURL=useSessionStream.js.map

package/dist/commonjs/hooks/useSessionStream.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"useSessionStream.js","sourceRoot":"","sources":["../../../src/hooks/useSessionStream.ts"],"names":[],"mappings":";AAAA,YAAY,CAAC;;;AAGb,iCAAwE;AACxE,sDAA4D;AAE5D,4DAAiD;AACjD,sFAAgF;AAEhF,uDAAiD;AAoFjD;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAmDG;AACH,0BACE,qBAA8B,EAC9B,OAA0C;IAE1C,MAAM,MAAM,GAAG,IAAA,aAAK,GAAE,CAAC;IACvB,MAAM,KAAK,GAAG,OAAO,EAAE,EAAE,IAAI,MAAM,CAAC;IACpC,MAAM,EAAE,GAAG,OAAO,EAAE,EAAE,IAAI,KAAK,CAAC;IAEhC,MAAM,CAAC,sBAAsB,CAAC,GAAG,IAAA,gBAAQ,EAAC,EAAoB,CAAC,CAAC;IAEhE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,EAAE,aAAa,EAAE,GAAG,IAAA,uBAAM,EACrD,CAAC,KAAK,EAAE,qBAAqB,EAAE,EAAE,EAAE,SAAS,CAAC,EAC7C,IAAI,EACJ;QACE,YAAY,EAAE,sBAAsB;KACrC,CACF,CAAC;IAEF,MAAM,UAAU,GAAG,IAAA,cAAM,EAAiB,OAAO,IAAK,EAAqB,CAAC,CAAC;IAC7E,IAAA,iBAAS,EAAC,GAAG,EAAE;QACb,UAAU,CAAC,OAAO,GAAG,OAAO,IAAK,EAAqB,CAAC;IACzD,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC;IAEd,MAAM,EAAE,IAAI,EAAE,WAAW,GAAG,SAAS,EAAE,MAAM,EAAE,cAAc,EAAE,GAAG,IAAA,uBAAM,EACtE,CAAC,KAAK,EAAE,qBAAqB,EAAE,EAAE,EAAE,aAAa,CAAC,EACjD,IAAI,CACL,CAAC;IACF,MAAM,cAAc,GAAG,IAAA,cAAM,EAAqB,WAAW,CAAC,CAAC;IAC/D,MAAM,kBAAkB,GAAG,IAAA,cAAM,EAAC,GAAG,KAAK,IAAI,qBAAqB,IAAI,EAAE,EAAE,CAAC,CAAC;IAC7E,IAAA,iBAAS,EAAC,GAAG,EAAE;QACb,MAAM,QAAQ,GAAG,GAAG,KAAK,IAAI,qBAAqB,IAAI,EAAE,EAAE,CAAC;QAC3D,IAAI,kBAAkB,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;YAC5C,kBAAkB,CAAC,OAAO,GAAG,QAAQ,CAAC;YACtC,cAAc,CAAC,OAAO,GAAG,WAAW,CAAC;QACvC,CAAC;IACH,CAAC,EAAE,CAAC,KAAK,EAAE,qBAAqB,EAAE,EAAE,EAAE,WAAW,CAAC,CAAC,CAAC;IAEpD,MAAM,EAAE,IAAI,EAAE,WAAW,GAAG,SAAS,EAAE,MAAM,EAAE,cAAc,EAAE,GAAG,IAAA,uBAAM,EAEtE,CAAC,KAAK,EAAE,qBAAqB,EAAE,EAAE,EAAE,aAAa,CAAC,EAAE,IAAI,CAAC,CAAC;IAE3D,MAAM,EAAE,IAAI,EAAE,WAAW,GAAG,KAAK,EAAE,MAAM,EAAE,aAAa,EAAE,GAAG,IAAA,uBAAM,EACjE,CAAC,KAAK,EAAE,qBAAqB,EAAE,EAAE,EAAE,UAAU,CAAC,EAC9C,IAAI,CACL,CAAC;IAEF,MAAM,EAAE,IAAI,EAAE,KAAK,GAAG,SAAS,EAAE,MAAM,EAAE,QAAQ,EAAE,GAAG,IAAA,uBAAM,EAC1D,CAAC,KAAK,EAAE,qBAAqB,EAAE,EAAE,EAAE,OAAO,CAAC,EAC3C,IAAI,CACL,CAAC;IAEF,MAAM,kBAAkB,GAAG,IAAA,cAAM,EAAyB,IAAI,CAAC,CAAC;IAEhE,MAAM,IAAI,GAAG,IAAA,mBAAW,EAAC,GAAG,EAAE;QAC5B,IAAI,kBAAkB,CAAC,OAAO,EAAE,CAAC;YAC/B,kBAAkB,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;YACnC,kBAAkB,CAAC,OAAO,GAAG,IAAI,CAAC;QACpC,CAAC;IACH,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,MAAM,iBAAiB,GAAG,OAAO,EAAE,SAAS,CAAC;IAC7C,MAAM,SAAS,GAAG,IAAA,mBAAW,EAC3B,CAAC,YAA2C,EAAE,EAAE;QAC9C,IAAI,iBAAiB,EAAE,CAAC;YACtB,iBAAiB,CAAC,YAAY,CAAC,CAAC;QAClC,CAAC;IACH,CAAC,EACD,CAAC,iBAAiB,CAAC,CACpB,CAAC;IAEF,MAAM,iBAAiB,GAAG,OAAO,EAAE,SAAS,CAAC;IAC7C,MAAM,SAAS,GAAG,IAAA,mBAAW,EAC3B,CAAC,KAAmB,EAAE,EAAE;QACtB,IAAI,iBAAiB,EAAE,CAAC;YACtB,iBAAiB,CAAC,KAAK,CAAC,CAAC;QAC3B,CAAC;IACH,CAAC,EACD,CAAC,iBAAiB,CAAC,CACpB,CAAC;IAEF,MAAM,SAAS,GAAG,IAAA,8BAAY,EAAC,OAAO,CAAC,CAAC;IACxC,MAAM,gBAAgB,GAAG,OAAO,EAAE,gBAAgB,CAAC;IACnD,MAAM,YAAY,GAAG,OAAO,EAAE,WAAW,CAAC;IAC1C,MAAM,YAAY,GAAG,OAAO,EAAE,YAAY,CAAC;IAC3C,MAAM,IAAI,GAAG,OAAO,EAAE,IAAI,CAAC;IAC3B,MAAM,UAAU,GAAG,OAAO,EAAE,UAAU,CAAC;IAEvC,IAAA,iBAAS,EAAC,GAAG,EAAE;QACb,IAAI,UAAU,IAAI,IAAI,IAAI,UAAU,IAAI,CAAC,EAAE,CAAC;YAC1C,MAAM,OAAO,GAAG,UAAU,CAAC,OAAO,CAAC;YACnC,IAAI,OAAO,CAAC,MAAM,GAAG,UAAU,EAAE,CAAC;gBAChC,aAAa,CAAC,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,GAAG,UAAU,CAAC,CAAC,CAAC;YAC5D,CAAC;QACH,CAAC;IACH,CAAC,EAAE,CAAC,UAAU,EAAE,aAAa,CAAC,CAAC,CAAC;IAEhC,MAAM,cAAc,GAAG,IAAA,mBAAW,EAAC,KAAK,IAAI,EAAE;QAC5C,IAAI,CAAC;YACH,IAAI,CAAC,qBAAqB,IAAI,CAAC,SAAS,EAAE,CAAC;gBACzC,OAAO;YACT,CAAC;YAED,MAAM,eAAe,GAAG,IAAI,eAAe,EAAE,CAAC;YAC9C,kBAAkB,CAAC,OAAO,GAAG,eAAe,CAAC;YAE7C,MAAM,oBAAoB,CACxB,qBAAqB,EACrB,EAAE,EACF,SAAS,EACT,aAAa,EACb,UAAU,EACV,cAAc,EACd,cAAc,EACd,QAAQ,EACR,SAAS,EACT,SAAS,EACT,kBAAkB,EAClB,gBAAgB,EAChB,YAAY,KAAK,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,OAAO,EAC1E,YAAY,IAAI,EAAE,EAClB,IAAI,EACJ,UAAU,CACX,CAAC;QACJ,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAK,GAAW,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;gBACvC,kBAAkB,CAAC,OAAO,GAAG,IAAI,CAAC;gBAClC,OAAO;YACT,CAAC;YAED,QAAQ,CAAC,GAAY,CAAC,CAAC;QACzB,CAAC;gBAAS,CAAC;YACT,IAAI,kBAAkB,CAAC,OAAO,EAAE,CAAC;gBAC/B,kBAAkB,CAAC,OAAO,GAAG,IAAI,CAAC;YACpC,CAAC;YAED,aAAa,CAAC,IAAI,CAAC,CAAC;QACtB,CAAC;IACH,CAAC,EAAE;QACD,qBAAqB;QACrB,EAAE;QACF,SAAS;QACT,aAAa;QACb,cAAc;QACd,cAAc;QACd,QAAQ;QACR,aAAa;QACb,SAAS;QACT,SAAS;QACT,gBAAgB;QAChB,YAAY;QACZ,YAAY;QACZ,IAAI;QACJ,UAAU;KACX,CAAC,CAAC;IACH,MAAM,mBAAmB,GAAG,IAAA,sDAAwB,EAAC,cAAc,CAAC,CAAC;IAErE,IAAA,iBAAS,EAAC,GAAG,EAAE;QACb,IAAI,OAAO,OAAO,EAAE,OAAO,KAAK,SAAS,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;YAC9D,OAAO;QACT,CAAC;QAED,IAAI,CAAC,qBAAqB,EAAE,CAAC;YAC3B,OAAO;QACT,CAAC;QAED,mBAAmB,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QAExC,OAAO,GAAG,EAAE;YACV,IAAI,EAAE,CAAC;QACT,CAAC,CAAC;IACJ,CAAC,EAAE,CAAC,qBAAqB,EAAE,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,mBAAmB,CAAC,CAAC,CAAC;IAE7E,OAAO,EAAE,OAAO,EAAE,OAAO,IAAI,sBAAsB,EAAE,WAAW,EAAE,WAAW,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;AAC/F,CAAC;AAED,KAAK,UAAU,oBAAoB,CACjC,qBAA6B,EAC7B,EAAgB,EAChB,SAAoB,EACpB,iBAA+C,EAC/C,kBAA0D,EAC1D,cAAgD,EAChD,cAAsD,EACtD,OAA2B,EAC3B,SAA2D,EAC3D,SAAwC,EACxC,kBAAkE,EAClE,gBAAyB,EACzB,WAAoB,EACpB,YAAqB,EACrB,IAA6B,EAC7B,UAAmB;IAEnB,IAAI,eAAmC,CAAC;IACxC,IAAI,gBAAoC,CAAC;IACzC,IAAI,UAAU,GAAkC,EAAE,CAAC;IAEnD,MAAM,kBAAkB,GAAG,GAAG,EAAE;QAC9B,IAAI,eAAe,KAAK,gBAAgB,EAAE,CAAC;YACzC,gBAAgB,GAAG,eAAe,CAAC;YACnC,cAAc,CAAC,eAAe,CAAC,CAAC;QAClC,CAAC;IACH,CAAC,CAAC;IAEF,MAAM,UAAU,GAAG,GAAG,EAAE;QACtB,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO;QACpC,MAAM,KAAK,GAAG,UAAU,CAAC;QACzB,UAAU,GAAG,EAAE,CAAC;QAChB,SAAS,CAAC,KAAK,CAAC,CAAC;IACnB,CAAC,CAAC;IAEF,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,wBAAwB,CAAU,qBAAqB,EAAE,EAAE,EAAE;YAC1F,MAAM,EAAE,kBAAkB,CAAC,OAAO,EAAE,MAAM;YAC1C,gBAAgB;YAChB,WAAW;YACX,IAAI;YACJ,MAAM,EAAE,CAAC,IAAI,EAAE,EAAE;gBACf,eAAe,GAAG,IAAI,CAAC,EAAE,CAAC;gBAC1B,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACxB,CAAC;YACD,SAAS,EAAE,CAAC,KAAK,EAAE,EAAE;gBACnB,cAAc,CAAC,KAAK,CAAC,CAAC;gBACtB,SAAS,CAAC,KAAK,CAAC,CAAC;YACnB,CAAC;SACF,CAAC,CAAC;QAEH,MAAM,YAAY,GAAG,IAAA,kCAAoB,EAAU,KAAK,EAAE,UAAU,EAAE,EAAE;YACtE,MAAM,QAAQ,GAAG,CAAC,GAAG,kBAAkB,CAAC,OAAO,EAAE,GAAG,UAAU,CAAC,CAAC;YAChE,MAAM,OAAO,GACX,UAAU,IAAI,IAAI,IAAI,UAAU,IAAI,CAAC,IAAI,QAAQ,CAAC,MAAM,GAAG,UAAU;gBACnE,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,GAAG,UAAU,CAAC;gBAC9C,CAAC,CAAC,QAAQ,CAAC;YACf,kBAAkB,CAAC,OAAO,GAAG,OAAO,CAAC;YACrC,iBAAiB,CAAC,OAAO,CAAC,CAAC;YAC3B,kBAAkB,EAAE,CAAC;YACrB,UAAU,EAAE,CAAC;QACf,CAAC,EAAE,YAAY,CAAC,CAAC;QAEjB,IAAI,KAAK,EAAE,MAAM,MAAM,IAAI,MAAM,EAAE,CAAC;YAClC,YAAY,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAC3B,CAAC;QAED,MAAM,YAAY,CAAC,KAAK,EAAE,CAAC;QAC3B,kBAAkB,EAAE,CAAC;QACrB,UAAU,EAAE,CAAC;IACf,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,IAAK,GAAW,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;YACvC,OAAO;QACT,CAAC;QAED,IAAI,GAAG,YAAY,KAAK,EAAE,CAAC;YACzB,OAAO,CAAC,GAAG,CAAC,CAAC;QACf,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;QAClC,CAAC;QAED,MAAM,GAAG,CAAC;IACZ,CAAC;AACH,CAAC"}

package/dist/commonjs/hooks/useSessionStreamChannel.d.ts

@@ -0,0 +1,86 @@
+import type { AnySessionChannel, ControlEvent, SessionChannelIn, SessionChannelName, SessionChannelOut, SSEStreamPart } from "@trigger.dev/core/v3";
+import type { UseApiClientOptions } from "./useApiClient.js";
+type ChannelRecord<TChannel extends AnySessionChannel, S extends "in" | "out"> = S extends "out" ? SessionChannelOut<TChannel> : SessionChannelIn<TChannel>;
+export type UseSessionStreamChannelInstance<TRecord> = {
+    /** The records received so far on the channel, in arrival order. */
+    records: Array<TRecord>;
+    /** The cursor of the last record seen; pass back as `lastEventId` to resume. */
+    lastEventId: string | undefined;
+    /** The last control record seen on the channel. */
+    lastControl: ControlEvent | undefined;
+    error: Error | undefined;
+    /** Abort the current request immediately, keep the records received so far. */
+    stop: () => void;
+};
+export type UseSessionStreamChannelOptions<TChannel extends AnySessionChannel, S extends "in" | "out"> = UseApiClientOptions & {
+    /**
+     * The id or external id of the session that owns the channel. May be
+     * undefined while it resolves; the subscription starts once it is set.
+     */
+    sessionId?: string;
+    id?: string;
+    enabled?: boolean;
+    /**
+     * Which side of the channel to read.
+     *
+     * @default "out"
+     */
+    io?: S;
+    /**
+     * The number of milliseconds to throttle the record updates.
+     *
+     * @default 16
+     */
+    throttleInMs?: number;
+    /**
+     * The number of seconds to wait for new data before the stream closes.
+     *
+     * @default 60 seconds
+     */
+    timeoutInSeconds?: number;
+    /** The cursor to resume from. If not provided, reads per `from`. */
+    lastEventId?: string | number;
+    /**
+     * Where a fresh subscription (no `lastEventId`) starts reading.
+     *
+     * - `"beginning"` (default): replay the full channel history, then live-tail.
+     * - `"latest"`: start at the current tail, for a last-value / live view.
+     *
+     * Ignored when `lastEventId` is set.
+     */
+    from?: "beginning" | "latest";
+    /**
+     * Cap the number of records kept in `records`. Use `maxRecords: 1` with
+     * `from: "latest"` for a bounded last-value view.
+     */
+    maxRecords?: number;
+    /** Invoked once per throttled flush with the batch of records (control records included). */
+    onRecords?: (records: Array<SSEStreamPart<ChannelRecord<TChannel, S>>>) => void;
+    /** Called when a control record is received on the channel. */
+    onControl?: (event: ControlEvent) => void;
+};
+/**
+ * Read one side of a named Session side channel, with record types inferred
+ * from a `defineSessionChannel` declaration passed as the type argument.
+ *
+ * The channel name is typesafe (`SessionChannelName<TChannel>`) and `records`
+ * is typed from the channel's `.out` / `.in` record type. Called without the
+ * type argument, the channel name is any string and `records` is `unknown`.
+ *
+ * Requires a Public Access Token scoped to the session (or to the channel).
+ *
+ * @example
+ * ```tsx
+ * import type { screenshotsChannel } from "./shared/channels";
+ *
+ * const { records } = useSessionStreamChannel<typeof screenshotsChannel>("screenshots", {
+ *   sessionId,
+ *   accessToken,
+ *   io: "out",
+ *   from: "latest",
+ *   maxRecords: 1,
+ * });
+ * ```
+ */
+export declare function useSessionStreamChannel<TChannel extends AnySessionChannel = AnySessionChannel, S extends "in" | "out" = "out">(channel: SessionChannelName<TChannel>, options: UseSessionStreamChannelOptions<TChannel, S>): UseSessionStreamChannelInstance<ChannelRecord<TChannel, S>>;
+export {};

package/dist/commonjs/hooks/useSessionStreamChannel.js

@@ -0,0 +1,209 @@
+"use strict";
+"use client";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.useSessionStreamChannel = useSessionStreamChannel;
+const react_1 = require("react");
+const throttle_js_1 = require("../utils/throttle.js");
+const trigger_swr_js_1 = require("../utils/trigger-swr.js");
+const useStableRequestCallback_js_1 = require("../utils/useStableRequestCallback.js");
+const useApiClient_js_1 = require("./useApiClient.js");
+/**
+ * Read one side of a named Session side channel, with record types inferred
+ * from a `defineSessionChannel` declaration passed as the type argument.
+ *
+ * The channel name is typesafe (`SessionChannelName<TChannel>`) and `records`
+ * is typed from the channel's `.out` / `.in` record type. Called without the
+ * type argument, the channel name is any string and `records` is `unknown`.
+ *
+ * Requires a Public Access Token scoped to the session (or to the channel).
+ *
+ * @example
+ * ```tsx
+ * import type { screenshotsChannel } from "./shared/channels";
+ *
+ * const { records } = useSessionStreamChannel<typeof screenshotsChannel>("screenshots", {
+ *   sessionId,
+ *   accessToken,
+ *   io: "out",
+ *   from: "latest",
+ *   maxRecords: 1,
+ * });
+ * ```
+ */
+function useSessionStreamChannel(channel, options) {
+    const hookId = (0, react_1.useId)();
+    const idKey = options.id ?? hookId;
+    const io = (options.io ?? "out");
+    const sessionId = options.sessionId;
+    const channelName = channel;
+    const [initialRecordsFallback] = (0, react_1.useState)([]);
+    const { data: records, mutate: mutateRecords } = (0, trigger_swr_js_1.useSWR)([idKey, sessionId, channelName, io, "records"], null, { fallbackData: initialRecordsFallback });
+    const recordsRef = (0, react_1.useRef)(records ?? []);
+    (0, react_1.useEffect)(() => {
+        recordsRef.current = records || [];
+    }, [records]);
+    const { data: lastEventId = undefined, mutate: setLastEventId } = (0, trigger_swr_js_1.useSWR)([idKey, sessionId, channelName, io, "lastEventId"], null);
+    const lastEventIdRef = (0, react_1.useRef)(lastEventId);
+    const channelIdentityRef = (0, react_1.useRef)(`${idKey}:${sessionId}:${channelName}:${io}`);
+    (0, react_1.useEffect)(() => {
+        const identity = `${idKey}:${sessionId}:${channelName}:${io}`;
+        if (channelIdentityRef.current !== identity) {
+            channelIdentityRef.current = identity;
+            lastEventIdRef.current = lastEventId;
+        }
+    }, [idKey, sessionId, channelName, io, lastEventId]);
+    const { data: lastControl = undefined, mutate: setLastControl } = (0, trigger_swr_js_1.useSWR)([idKey, sessionId, channelName, io, "lastControl"], null);
+    const { data: _isComplete = false, mutate: setIsComplete } = (0, trigger_swr_js_1.useSWR)([idKey, sessionId, channelName, io, "complete"], null);
+    const { data: error = undefined, mutate: setError } = (0, trigger_swr_js_1.useSWR)([idKey, sessionId, channelName, io, "error"], null);
+    const abortControllerRef = (0, react_1.useRef)(null);
+    const stop = (0, react_1.useCallback)(() => {
+        if (abortControllerRef.current) {
+            abortControllerRef.current.abort();
+            abortControllerRef.current = null;
+        }
+    }, []);
+    const onRecordsCallback = options.onRecords;
+    const onRecords = (0, react_1.useCallback)((recordsBatch) => {
+        if (onRecordsCallback) {
+            onRecordsCallback(recordsBatch);
+        }
+    }, [onRecordsCallback]);
+    const onControlCallback = options.onControl;
+    const onControl = (0, react_1.useCallback)((event) => {
+        if (onControlCallback) {
+            onControlCallback(event);
+        }
+    }, [onControlCallback]);
+    const apiClient = (0, useApiClient_js_1.useApiClient)(options);
+    const timeoutInSeconds = options.timeoutInSeconds;
+    const startEventId = options.lastEventId;
+    const throttleInMs = options.throttleInMs;
+    const from = options.from;
+    const maxRecords = options.maxRecords;
+    (0, react_1.useEffect)(() => {
+        if (maxRecords != null && maxRecords >= 0) {
+            const current = recordsRef.current;
+            if (current.length > maxRecords) {
+                mutateRecords(current.slice(current.length - maxRecords));
+            }
+        }
+    }, [maxRecords, mutateRecords]);
+    const triggerRequest = (0, react_1.useCallback)(async () => {
+        let abortController = null;
+        try {
+            if (!sessionId || !apiClient) {
+                return;
+            }
+            abortController = new AbortController();
+            abortControllerRef.current = abortController;
+            await processSessionChannelStream(sessionId, io, channelName, apiClient, mutateRecords, recordsRef, setLastEventId, setLastControl, setError, onRecords, onControl, abortControllerRef, timeoutInSeconds, startEventId !== undefined ? String(startEventId) : lastEventIdRef.current, throttleInMs ?? 16, from, maxRecords);
+        }
+        catch (err) {
+            if (err.name === "AbortError") {
+                return;
+            }
+            setError(err);
+        }
+        finally {
+            if (abortControllerRef.current === abortController) {
+                abortControllerRef.current = null;
+            }
+            setIsComplete(true);
+        }
+    }, [
+        sessionId,
+        io,
+        channelName,
+        apiClient,
+        mutateRecords,
+        setLastEventId,
+        setLastControl,
+        setError,
+        setIsComplete,
+        onRecords,
+        onControl,
+        timeoutInSeconds,
+        startEventId,
+        throttleInMs,
+        from,
+        maxRecords,
+    ]);
+    const requestSubscription = (0, useStableRequestCallback_js_1.useStableRequestCallback)(triggerRequest);
+    (0, react_1.useEffect)(() => {
+        if (typeof options.enabled === "boolean" && !options.enabled) {
+            return;
+        }
+        if (!sessionId) {
+            return;
+        }
+        requestSubscription().finally(() => { });
+        return () => {
+            stop();
+        };
+    }, [sessionId, channelName, io, stop, options.enabled, requestSubscription]);
+    return { records: records ?? initialRecordsFallback, lastEventId, lastControl, error, stop };
+}
+async function processSessionChannelStream(sessionIdOrExternalId, io, channel, apiClient, mutateRecordsData, existingRecordsRef, setLastEventId, setLastControl, onError, onRecords, onControl, abortControllerRef, timeoutInSeconds, lastEventId, throttleInMs, from, maxRecords) {
+    let lastSeenEventId;
+    let publishedEventId;
+    let partsBatch = [];
+    const publishLastEventId = () => {
+        if (lastSeenEventId !== publishedEventId) {
+            publishedEventId = lastSeenEventId;
+            setLastEventId(lastSeenEventId);
+        }
+    };
+    const flushParts = () => {
+        if (partsBatch.length === 0)
+            return;
+        const batch = partsBatch;
+        partsBatch = [];
+        onRecords(batch);
+    };
+    try {
+        const stream = await apiClient.subscribeToSessionStream(sessionIdOrExternalId, io, {
+            signal: abortControllerRef.current?.signal,
+            channel,
+            timeoutInSeconds,
+            lastEventId,
+            from,
+            onPart: (part) => {
+                lastSeenEventId = part.id;
+                partsBatch.push(part);
+            },
+            onControl: (event) => {
+                setLastControl(event);
+                onControl(event);
+            },
+        });
+        const recordsQueue = (0, throttle_js_1.createThrottledQueue)(async (newRecords) => {
+            const combined = [...existingRecordsRef.current, ...newRecords];
+            const bounded = maxRecords != null && maxRecords >= 0 && combined.length > maxRecords
+                ? combined.slice(combined.length - maxRecords)
+                : combined;
+            existingRecordsRef.current = bounded;
+            mutateRecordsData(bounded);
+            publishLastEventId();
+            flushParts();
+        }, throttleInMs);
+        for await (const record of stream) {
+            recordsQueue.add(record);
+        }
+        await recordsQueue.flush();
+        publishLastEventId();
+        flushParts();
+    }
+    catch (err) {
+        if (err.name === "AbortError") {
+            return;
+        }
+        if (err instanceof Error) {
+            onError(err);
+        }
+        else {
+            onError(new Error(String(err)));
+        }
+        throw err;
+    }
+}
+//# sourceMappingURL=useSessionStreamChannel.js.map

package/dist/commonjs/hooks/useSessionStreamChannel.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"useSessionStreamChannel.js","sourceRoot":"","sources":["../../../src/hooks/useSessionStreamChannel.ts"],"names":[],"mappings":";AAAA,YAAY,CAAC;;;AAWb,iCAAwE;AACxE,sDAA4D;AAE5D,4DAAiD;AACjD,sFAAgF;AAEhF,uDAAiD;AAqEjD;;;;;;;;;;;;;;;;;;;;;;GAsBG;AACH,iCAIE,OAAqC,EACrC,OAAoD;IAIpD,MAAM,MAAM,GAAG,IAAA,aAAK,GAAE,CAAC;IACvB,MAAM,KAAK,GAAG,OAAO,CAAC,EAAE,IAAI,MAAM,CAAC;IACnC,MAAM,EAAE,GAAG,CAAC,OAAO,CAAC,EAAE,IAAI,KAAK,CAAiB,CAAC;IACjD,MAAM,SAAS,GAAG,OAAO,CAAC,SAAS,CAAC;IACpC,MAAM,WAAW,GAAG,OAAiB,CAAC;IAEtC,MAAM,CAAC,sBAAsB,CAAC,GAAG,IAAA,gBAAQ,EAAC,EAAoB,CAAC,CAAC;IAEhE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,EAAE,aAAa,EAAE,GAAG,IAAA,uBAAM,EACrD,CAAC,KAAK,EAAE,SAAS,EAAE,WAAW,EAAE,EAAE,EAAE,SAAS,CAAC,EAC9C,IAAI,EACJ,EAAE,YAAY,EAAE,sBAAsB,EAAE,CACzC,CAAC;IAEF,MAAM,UAAU,GAAG,IAAA,cAAM,EAAiB,OAAO,IAAK,EAAqB,CAAC,CAAC;IAC7E,IAAA,iBAAS,EAAC,GAAG,EAAE;QACb,UAAU,CAAC,OAAO,GAAG,OAAO,IAAK,EAAqB,CAAC;IACzD,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC;IAEd,MAAM,EAAE,IAAI,EAAE,WAAW,GAAG,SAAS,EAAE,MAAM,EAAE,cAAc,EAAE,GAAG,IAAA,uBAAM,EACtE,CAAC,KAAK,EAAE,SAAS,EAAE,WAAW,EAAE,EAAE,EAAE,aAAa,CAAC,EAClD,IAAI,CACL,CAAC;IACF,MAAM,cAAc,GAAG,IAAA,cAAM,EAAqB,WAAW,CAAC,CAAC;IAC/D,MAAM,kBAAkB,GAAG,IAAA,cAAM,EAAC,GAAG,KAAK,IAAI,SAAS,IAAI,WAAW,IAAI,EAAE,EAAE,CAAC,CAAC;IAChF,IAAA,iBAAS,EAAC,GAAG,EAAE;QACb,MAAM,QAAQ,GAAG,GAAG,KAAK,IAAI,SAAS,IAAI,WAAW,IAAI,EAAE,EAAE,CAAC;QAC9D,IAAI,kBAAkB,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;YAC5C,kBAAkB,CAAC,OAAO,GAAG,QAAQ,CAAC;YACtC,cAAc,CAAC,OAAO,GAAG,WAAW,CAAC;QACvC,CAAC;IACH,CAAC,EAAE,CAAC,KAAK,EAAE,SAAS,EAAE,WAAW,EAAE,EAAE,EAAE,WAAW,CAAC,CAAC,CAAC;IAErD,MAAM,EAAE,IAAI,EAAE,WAAW,GAAG,SAAS,EAAE,MAAM,EAAE,cAAc,EAAE,GAAG,IAAA,uBAAM,EAEtE,CAAC,KAAK,EAAE,SAAS,EAAE,WAAW,EAAE,EAAE,EAAE,aAAa,CAAC,EAAE,IAAI,CAAC,CAAC;IAE5D,MAAM,EAAE,IAAI,EAAE,WAAW,GAAG,KAAK,EAAE,MAAM,EAAE,aAAa,EAAE,GAAG,IAAA,uBAAM,EACjE,CAAC,KAAK,EAAE,SAAS,EAAE,WAAW,EAAE,EAAE,EAAE,UAAU,CAAC,EAC/C,IAAI,CACL,CAAC;IAEF,MAAM,EAAE,IAAI,EAAE,KAAK,GAAG,SAAS,EAAE,MAAM,EAAE,QAAQ,EAAE,GAAG,IAAA,uBAAM,EAC1D,CAAC,KAAK,EAAE,SAAS,EAAE,WAAW,EAAE,EAAE,EAAE,OAAO,CAAC,EAC5C,IAAI,CACL,CAAC;IAEF,MAAM,kBAAkB,GAAG,IAAA,cAAM,EAAyB,IAAI,CAAC,CAAC;IAEhE,MAAM,IAAI,GAAG,IAAA,mBAAW,EAAC,GAAG,EAAE;QAC5B,IAAI,kBAAkB,CAAC,OAAO,EAAE,CAAC;YAC/B,kBAAkB,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;YACnC,kBAAkB,CAAC,OAAO,GAAG,IAAI,CAAC;QACpC,CAAC;IACH,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,MAAM,iBAAiB,GAAG,OAAO,CAAC,SAAS,CAAC;IAC5C,MAAM,SAAS,GAAG,IAAA,mBAAW,EAC3B,CAAC,YAA2C,EAAE,EAAE;QAC9C,IAAI,iBAAiB,EAAE,CAAC;YACtB,iBAAiB,CAAC,YAAY,CAAC,CAAC;QAClC,CAAC;IACH,CAAC,EACD,CAAC,iBAAiB,CAAC,CACpB,CAAC;IAEF,MAAM,iBAAiB,GAAG,OAAO,CAAC,SAAS,CAAC;IAC5C,MAAM,SAAS,GAAG,IAAA,mBAAW,EAC3B,CAAC,KAAmB,EAAE,EAAE;QACtB,IAAI,iBAAiB,EAAE,CAAC;YACtB,iBAAiB,CAAC,KAAK,CAAC,CAAC;QAC3B,CAAC;IACH,CAAC,EACD,CAAC,iBAAiB,CAAC,CACpB,CAAC;IAEF,MAAM,SAAS,GAAG,IAAA,8BAAY,EAAC,OAAO,CAAC,CAAC;IACxC,MAAM,gBAAgB,GAAG,OAAO,CAAC,gBAAgB,CAAC;IAClD,MAAM,YAAY,GAAG,OAAO,CAAC,WAAW,CAAC;IACzC,MAAM,YAAY,GAAG,OAAO,CAAC,YAAY,CAAC;IAC1C,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;IAC1B,MAAM,UAAU,GAAG,OAAO,CAAC,UAAU,CAAC;IAEtC,IAAA,iBAAS,EAAC,GAAG,EAAE;QACb,IAAI,UAAU,IAAI,IAAI,IAAI,UAAU,IAAI,CAAC,EAAE,CAAC;YAC1C,MAAM,OAAO,GAAG,UAAU,CAAC,OAAO,CAAC;YACnC,IAAI,OAAO,CAAC,MAAM,GAAG,UAAU,EAAE,CAAC;gBAChC,aAAa,CAAC,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,GAAG,UAAU,CAAC,CAAC,CAAC;YAC5D,CAAC;QACH,CAAC;IACH,CAAC,EAAE,CAAC,UAAU,EAAE,aAAa,CAAC,CAAC,CAAC;IAEhC,MAAM,cAAc,GAAG,IAAA,mBAAW,EAAC,KAAK,IAAI,EAAE;QAC5C,IAAI,eAAe,GAA2B,IAAI,CAAC;QACnD,IAAI,CAAC;YACH,IAAI,CAAC,SAAS,IAAI,CAAC,SAAS,EAAE,CAAC;gBAC7B,OAAO;YACT,CAAC;YAED,eAAe,GAAG,IAAI,eAAe,EAAE,CAAC;YACxC,kBAAkB,CAAC,OAAO,GAAG,eAAe,CAAC;YAE7C,MAAM,2BAA2B,CAC/B,SAAS,EACT,EAAE,EACF,WAAW,EACX,SAAS,EACT,aAAa,EACb,UAAU,EACV,cAAc,EACd,cAAc,EACd,QAAQ,EACR,SAAS,EACT,SAAS,EACT,kBAAkB,EAClB,gBAAgB,EAChB,YAAY,KAAK,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,OAAO,EAC1E,YAAY,IAAI,EAAE,EAClB,IAAI,EACJ,UAAU,CACX,CAAC;QACJ,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAK,GAAW,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;gBACvC,OAAO;YACT,CAAC;YAED,QAAQ,CAAC,GAAY,CAAC,CAAC;QACzB,CAAC;gBAAS,CAAC;YACT,IAAI,kBAAkB,CAAC,OAAO,KAAK,eAAe,EAAE,CAAC;gBACnD,kBAAkB,CAAC,OAAO,GAAG,IAAI,CAAC;YACpC,CAAC;YAED,aAAa,CAAC,IAAI,CAAC,CAAC;QACtB,CAAC;IACH,CAAC,EAAE;QACD,SAAS;QACT,EAAE;QACF,WAAW;QACX,SAAS;QACT,aAAa;QACb,cAAc;QACd,cAAc;QACd,QAAQ;QACR,aAAa;QACb,SAAS;QACT,SAAS;QACT,gBAAgB;QAChB,YAAY;QACZ,YAAY;QACZ,IAAI;QACJ,UAAU;KACX,CAAC,CAAC;IACH,MAAM,mBAAmB,GAAG,IAAA,sDAAwB,EAAC,cAAc,CAAC,CAAC;IAErE,IAAA,iBAAS,EAAC,GAAG,EAAE;QACb,IAAI,OAAO,OAAO,CAAC,OAAO,KAAK,SAAS,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;YAC7D,OAAO;QACT,CAAC;QAED,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,OAAO;QACT,CAAC;QAED,mBAAmB,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QAExC,OAAO,GAAG,EAAE;YACV,IAAI,EAAE,CAAC;QACT,CAAC,CAAC;IACJ,CAAC,EAAE,CAAC,SAAS,EAAE,WAAW,EAAE,EAAE,EAAE,IAAI,EAAE,OAAO,CAAC,OAAO,EAAE,mBAAmB,CAAC,CAAC,CAAC;IAE7E,OAAO,EAAE,OAAO,EAAE,OAAO,IAAI,sBAAsB,EAAE,WAAW,EAAE,WAAW,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;AAC/F,CAAC;AAED,KAAK,UAAU,2BAA2B,CACxC,qBAA6B,EAC7B,EAAgB,EAChB,OAAe,EACf,SAAoB,EACpB,iBAA+C,EAC/C,kBAA0D,EAC1D,cAAgD,EAChD,cAAsD,EACtD,OAA2B,EAC3B,SAA2D,EAC3D,SAAwC,EACxC,kBAAkE,EAClE,gBAAyB,EACzB,WAAoB,EACpB,YAAqB,EACrB,IAA6B,EAC7B,UAAmB;IAEnB,IAAI,eAAmC,CAAC;IACxC,IAAI,gBAAoC,CAAC;IACzC,IAAI,UAAU,GAAkC,EAAE,CAAC;IAEnD,MAAM,kBAAkB,GAAG,GAAG,EAAE;QAC9B,IAAI,eAAe,KAAK,gBAAgB,EAAE,CAAC;YACzC,gBAAgB,GAAG,eAAe,CAAC;YACnC,cAAc,CAAC,eAAe,CAAC,CAAC;QAClC,CAAC;IACH,CAAC,CAAC;IAEF,MAAM,UAAU,GAAG,GAAG,EAAE;QACtB,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO;QACpC,MAAM,KAAK,GAAG,UAAU,CAAC;QACzB,UAAU,GAAG,EAAE,CAAC;QAChB,SAAS,CAAC,KAAK,CAAC,CAAC;IACnB,CAAC,CAAC;IAEF,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,wBAAwB,CAAU,qBAAqB,EAAE,EAAE,EAAE;YAC1F,MAAM,EAAE,kBAAkB,CAAC,OAAO,EAAE,MAAM;YAC1C,OAAO;YACP,gBAAgB;YAChB,WAAW;YACX,IAAI;YACJ,MAAM,EAAE,CAAC,IAAI,EAAE,EAAE;gBACf,eAAe,GAAG,IAAI,CAAC,EAAE,CAAC;gBAC1B,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACxB,CAAC;YACD,SAAS,EAAE,CAAC,KAAK,EAAE,EAAE;gBACnB,cAAc,CAAC,KAAK,CAAC,CAAC;gBACtB,SAAS,CAAC,KAAK,CAAC,CAAC;YACnB,CAAC;SACF,CAAC,CAAC;QAEH,MAAM,YAAY,GAAG,IAAA,kCAAoB,EAAU,KAAK,EAAE,UAAU,EAAE,EAAE;YACtE,MAAM,QAAQ,GAAG,CAAC,GAAG,kBAAkB,CAAC,OAAO,EAAE,GAAG,UAAU,CAAC,CAAC;YAChE,MAAM,OAAO,GACX,UAAU,IAAI,IAAI,IAAI,UAAU,IAAI,CAAC,IAAI,QAAQ,CAAC,MAAM,GAAG,UAAU;gBACnE,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,GAAG,UAAU,CAAC;gBAC9C,CAAC,CAAC,QAAQ,CAAC;YACf,kBAAkB,CAAC,OAAO,GAAG,OAAO,CAAC;YACrC,iBAAiB,CAAC,OAAO,CAAC,CAAC;YAC3B,kBAAkB,EAAE,CAAC;YACrB,UAAU,EAAE,CAAC;QACf,CAAC,EAAE,YAAY,CAAC,CAAC;QAEjB,IAAI,KAAK,EAAE,MAAM,MAAM,IAAI,MAAM,EAAE,CAAC;YAClC,YAAY,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAC3B,CAAC;QAED,MAAM,YAAY,CAAC,KAAK,EAAE,CAAC;QAC3B,kBAAkB,EAAE,CAAC;QACrB,UAAU,EAAE,CAAC;IACf,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,IAAK,GAAW,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;YACvC,OAAO;QACT,CAAC;QAED,IAAI,GAAG,YAAY,KAAK,EAAE,CAAC;YACzB,OAAO,CAAC,GAAG,CAAC,CAAC;QACf,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;QAClC,CAAC;QAED,MAAM,GAAG,CAAC;IACZ,CAAC;AACH,CAAC"}

package/dist/commonjs/index.d.ts

@@ -5,3 +5,5 @@ export * from "./hooks/useRealtime.js";
 export * from "./hooks/useTaskTrigger.js";
 export * from "./hooks/useWaitToken.js";
 export * from "./hooks/useInputStreamSend.js";
+export * from "./hooks/useSessionStream.js";
+export * from "./hooks/useSessionStreamChannel.js";

package/dist/commonjs/index.js

@@ -21,4 +21,6 @@ __exportStar(require("./hooks/useRealtime.js"), exports);
 __exportStar(require("./hooks/useTaskTrigger.js"), exports);
 __exportStar(require("./hooks/useWaitToken.js"), exports);
 __exportStar(require("./hooks/useInputStreamSend.js"), exports);
+__exportStar(require("./hooks/useSessionStream.js"), exports);
+__exportStar(require("./hooks/useSessionStreamChannel.js"), exports);
 //# sourceMappingURL=index.js.map

package/dist/commonjs/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,gDAA8B;AAC9B,0DAAwC;AACxC,oDAAkC;AAClC,yDAAuC;AACvC,4DAA0C;AAC1C,0DAAwC;AACxC,gEAA8C"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,gDAA8B;AAC9B,0DAAwC;AACxC,oDAAkC;AAClC,yDAAuC;AACvC,4DAA0C;AAC1C,0DAAwC;AACxC,gEAA8C;AAC9C,8DAA4C;AAC5C,qEAAmD"}

package/dist/commonjs/utils/useStableRequestCallback.d.ts

@@ -0,0 +1,5 @@
+/**
+ * Keep subscription lifecycles controlled by their effects while using the
+ * latest request inputs.
+ */
+export declare function useStableRequestCallback(callback: () => Promise<void>): () => Promise<void>;

package/dist/commonjs/utils/useStableRequestCallback.js

@@ -0,0 +1,17 @@
+"use strict";
+"use client";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.useStableRequestCallback = useStableRequestCallback;
+const react_1 = require("react");
+/**
+ * Keep subscription lifecycles controlled by their effects while using the
+ * latest request inputs.
+ */
+function useStableRequestCallback(callback) {
+    const callbackRef = (0, react_1.useRef)(callback);
+    (0, react_1.useEffect)(() => {
+        callbackRef.current = callback;
+    }, [callback]);
+    return (0, react_1.useCallback)(() => callbackRef.current(), []);
+}
+//# sourceMappingURL=useStableRequestCallback.js.map

package/dist/commonjs/utils/useStableRequestCallback.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"useStableRequestCallback.js","sourceRoot":"","sources":["../../../src/utils/useStableRequestCallback.ts"],"names":[],"mappings":";AAAA,YAAY,CAAC;;;AAEb,iCAAuD;AAEvD;;;GAGG;AACH,kCAAyC,QAA6B;IACpE,MAAM,WAAW,GAAG,IAAA,cAAM,EAAC,QAAQ,CAAC,CAAC;IAErC,IAAA,iBAAS,EAAC,GAAG,EAAE;QACb,WAAW,CAAC,OAAO,GAAG,QAAQ,CAAC;IACjC,CAAC,EAAE,CAAC,QAAQ,CAAC,CAAC,CAAC;IAEf,OAAO,IAAA,mBAAW,EAAC,GAAG,EAAE,CAAC,WAAW,CAAC,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC;AACtD,CAAC"}

package/dist/esm/hooks/useApiClient.d.ts

@@ -12,6 +12,11 @@ export type UseApiClientOptions = {
     previewBranch?: string;
     /** Optional additional request configuration */
     requestOptions?: ApiRequestOptions;
+    /**
+     * Optional callback that mints a fresh access token. Used to reconnect a
+     * realtime stream that the server rejected because its token expired.
+     */
+    refreshAccessToken?: () => Promise<string>;
     /**
      * Enable or disable the API client instance.
      *

package/dist/esm/hooks/useApiClient.js

@@ -1,5 +1,6 @@
 "use client";
-import { ApiClient } from "@trigger.dev/core/v3";
+import { ApiClient, refreshAccessTokenOnce } from "@trigger.dev/core/v3";
+import { useCallback, useEffect, useRef } from "react";
 import { useTriggerAuthContextOptional } from "../contexts.js";
 /**
  * Hook to create an API client instance using authentication context or provided options.
@@ -26,6 +27,18 @@ export function useApiClient(options) {
     const baseUrl = options?.baseURL ?? auth?.baseURL ?? "https://api.trigger.dev";
     const accessToken = options?.accessToken ?? auth?.accessToken;
     const previewBranch = options?.previewBranch ?? auth?.previewBranch;
+    const refreshAccessToken = options?.refreshAccessToken ?? auth?.refreshAccessToken;
+    const refreshAccessTokenRef = useRef(refreshAccessToken);
+    useEffect(() => {
+        refreshAccessTokenRef.current = refreshAccessToken;
+    }, [refreshAccessToken]);
+    const stableRefreshAccessToken = useCallback(async () => {
+        const refresh = refreshAccessTokenRef.current;
+        if (!refresh) {
+            throw new Error("Missing refreshAccessToken in TriggerAuthContext or useApiClient options");
+        }
+        return refreshAccessTokenOnce(refresh);
+    }, []);
     if (!accessToken) {
         if (options?.enabled === false) {
             return undefined;
@@ -36,6 +49,6 @@ export function useApiClient(options) {
         ...auth?.requestOptions,
         ...options?.requestOptions,
     };
-    return new ApiClient(baseUrl, accessToken, previewBranch, requestOptions);
+    return new ApiClient(baseUrl, accessToken, previewBranch, requestOptions, undefined, refreshAccessToken ? stableRefreshAccessToken : undefined);
 }
 //# sourceMappingURL=useApiClient.js.map