@langchain/langgraph-sdk 0.1.4 → 0.1.6

package/dist/react/stream.lgp.cjs

@@ -0,0 +1,485 @@
+"use strict";
+/* __LC_ALLOW_ENTRYPOINT_SIDE_EFFECTS__ */
+"use client";
+/* __LC_ALLOW_ENTRYPOINT_SIDE_EFFECTS__ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.useStreamLGP = useStreamLGP;
+const react_1 = require("react");
+const utils_js_1 = require("./utils.cjs");
+const errors_js_1 = require("./errors.cjs");
+const branching_js_1 = require("./branching.cjs");
+const manager_js_1 = require("./manager.cjs");
+const client_js_1 = require("../client.cjs");
+const messages_js_1 = require("./messages.cjs");
+const thread_js_1 = require("./thread.cjs");
+function getFetchHistoryKey(client, threadId, limit) {
+    return [(0, client_js_1.getClientConfigHash)(client), threadId, limit].join(":");
+}
+function fetchHistory(client, threadId, options) {
+    if (options?.limit === false) {
+        return client.threads.getState(threadId).then((state) => {
+            if (state.checkpoint == null)
+                return [];
+            return [state];
+        });
+    }
+    const limit = typeof options?.limit === "number" ? options.limit : 10;
+    return client.threads.getHistory(threadId, { limit });
+}
+function useThreadHistory(client, threadId, limit, options) {
+    const key = getFetchHistoryKey(client, threadId, limit);
+    const [state, setState] = (0, react_1.useState)(() => ({
+        data: undefined,
+        error: undefined,
+        isLoading: threadId != null,
+    }));
+    const clientRef = (0, react_1.useRef)(client);
+    clientRef.current = client;
+    const onErrorRef = (0, react_1.useRef)(options?.onError);
+    onErrorRef.current = options?.onError;
+    const fetcher = (0, react_1.useCallback)((threadId, limit) => {
+        if (threadId != null) {
+            const client = clientRef.current;
+            setState((state) => ({ ...state, isLoading: true }));
+            return fetchHistory(client, threadId, { limit }).then((data) => {
+                setState({ data, error: undefined, isLoading: false });
+                return data;
+            }, (error) => {
+                setState(({ data }) => ({ data, error, isLoading: false }));
+                onErrorRef.current?.(error);
+                return Promise.reject(error);
+            });
+        }
+        setState({ data: undefined, error: undefined, isLoading: false });
+        return Promise.resolve([]);
+    }, []);
+    (0, react_1.useEffect)(() => {
+        // Skip if a stream is already in progress, no need to fetch history
+        if (options.submittingRef.current != null &&
+            options.submittingRef.current === threadId) {
+            return;
+        }
+        void fetcher(threadId, limit);
+        // The `threadId` and `limit` arguments are already present in `key`
+        // Thus we don't need to include them in the dependency array
+        // eslint-disable-next-line react-hooks/exhaustive-deps
+    }, [fetcher, key]);
+    return {
+        data: state.data,
+        error: state.error,
+        isLoading: state.isLoading,
+        mutate: (mutateId) => fetcher(mutateId ?? threadId, limit),
+    };
+}
+function useStreamLGP(options) {
+    const reconnectOnMountRef = (0, react_1.useRef)(options.reconnectOnMount);
+    const runMetadataStorage = (0, react_1.useMemo)(() => {
+        if (typeof window === "undefined")
+            return null;
+        const storage = reconnectOnMountRef.current;
+        if (storage === true)
+            return window.sessionStorage;
+        if (typeof storage === "function")
+            return storage();
+        return null;
+    }, []);
+    const client = (0, react_1.useMemo)(() => options.client ??
+        new client_js_1.Client({
+            apiUrl: options.apiUrl,
+            apiKey: options.apiKey,
+            callerOptions: options.callerOptions,
+            defaultHeaders: options.defaultHeaders,
+        }), [
+        options.client,
+        options.apiKey,
+        options.apiUrl,
+        options.callerOptions,
+        options.defaultHeaders,
+    ]);
+    const [messageManager] = (0, react_1.useState)(() => new messages_js_1.MessageTupleManager());
+    const [stream] = (0, react_1.useState)(() => new manager_js_1.StreamManager(messageManager));
+    (0, react_1.useSyncExternalStore)(stream.subscribe, stream.getSnapshot, stream.getSnapshot);
+    const [threadId, onThreadId] = (0, thread_js_1.useControllableThreadId)(options);
+    const trackStreamModeRef = (0, react_1.useRef)([]);
+    const trackStreamMode = (0, react_1.useCallback)((...mode) => {
+        const ref = trackStreamModeRef.current;
+        for (const m of mode) {
+            if (!ref.includes(m))
+                ref.push(m);
+        }
+    }, []);
+    const hasUpdateListener = options.onUpdateEvent != null;
+    const hasCustomListener = options.onCustomEvent != null;
+    const hasLangChainListener = options.onLangChainEvent != null;
+    const hasDebugListener = options.onDebugEvent != null;
+    const hasCheckpointListener = options.onCheckpointEvent != null;
+    const hasTaskListener = options.onTaskEvent != null;
+    const callbackStreamMode = (0, react_1.useMemo)(() => {
+        const modes = [];
+        if (hasUpdateListener)
+            modes.push("updates");
+        if (hasCustomListener)
+            modes.push("custom");
+        if (hasLangChainListener)
+            modes.push("events");
+        if (hasDebugListener)
+            modes.push("debug");
+        if (hasCheckpointListener)
+            modes.push("checkpoints");
+        if (hasTaskListener)
+            modes.push("tasks");
+        return modes;
+    }, [
+        hasUpdateListener,
+        hasCustomListener,
+        hasLangChainListener,
+        hasDebugListener,
+        hasCheckpointListener,
+        hasTaskListener,
+    ]);
+    const clearCallbackRef = (0, react_1.useRef)(null);
+    clearCallbackRef.current = stream.clear;
+    const threadIdRef = (0, react_1.useRef)(threadId);
+    const threadIdStreamingRef = (0, react_1.useRef)(null);
+    // Cancel the stream if thread ID has changed
+    (0, react_1.useEffect)(() => {
+        if (threadIdRef.current !== threadId) {
+            threadIdRef.current = threadId;
+            stream.clear();
+        }
+    }, [threadId, stream]);
+    const historyLimit = typeof options.fetchStateHistory === "object" &&
+        options.fetchStateHistory != null
+        ? options.fetchStateHistory.limit ?? false
+        : options.fetchStateHistory ?? false;
+    const history = useThreadHistory(client, threadId, historyLimit, {
+        submittingRef: threadIdStreamingRef,
+        onError: options.onError,
+    });
+    const getMessages = (value) => {
+        const messagesKey = options.messagesKey ?? "messages";
+        return Array.isArray(value[messagesKey]) ? value[messagesKey] : [];
+    };
+    const setMessages = (current, messages) => {
+        const messagesKey = options.messagesKey ?? "messages";
+        return { ...current, [messagesKey]: messages };
+    };
+    const [branch, setBranch] = (0, react_1.useState)("");
+    const branchContext = (0, branching_js_1.getBranchContext)(branch, history.data);
+    const historyValues = branchContext.threadHead?.values ??
+        options.initialValues ??
+        {};
+    const historyError = (() => {
+        const error = branchContext.threadHead?.tasks?.at(-1)?.error;
+        if (error == null)
+            return undefined;
+        try {
+            const parsed = JSON.parse(error);
+            if (errors_js_1.StreamError.isStructuredError(parsed))
+                return new errors_js_1.StreamError(parsed);
+            return parsed;
+        }
+        catch {
+            // do nothing
+        }
+        return error;
+    })();
+    const messageMetadata = (() => {
+        const alreadyShown = new Set();
+        return getMessages(historyValues).map((message, idx) => {
+            const messageId = message.id ?? idx;
+            // Find the first checkpoint where the message was seen
+            const firstSeenState = (0, utils_js_1.findLast)(history.data ?? [], (state) => getMessages(state.values)
+                .map((m, idx) => m.id ?? idx)
+                .includes(messageId));
+            const checkpointId = firstSeenState?.checkpoint?.checkpoint_id;
+            let branch = checkpointId != null
+                ? branchContext.branchByCheckpoint[checkpointId]
+                : undefined;
+            if (!branch?.branch?.length)
+                branch = undefined;
+            // serialize branches
+            const optionsShown = branch?.branchOptions?.flat(2).join(",");
+            if (optionsShown) {
+                if (alreadyShown.has(optionsShown))
+                    branch = undefined;
+                alreadyShown.add(optionsShown);
+            }
+            return {
+                messageId: messageId.toString(),
+                firstSeenState,
+                branch: branch?.branch,
+                branchOptions: branch?.branchOptions,
+            };
+        });
+    })();
+    const stop = () => stream.stop(historyValues, {
+        onStop: (args) => {
+            if (runMetadataStorage && threadId) {
+                const runId = runMetadataStorage.getItem(`lg:stream:${threadId}`);
+                if (runId)
+                    void client.runs.cancel(threadId, runId);
+                runMetadataStorage.removeItem(`lg:stream:${threadId}`);
+            }
+            options.onStop?.(args);
+        },
+    });
+    // --- TRANSPORT ---
+    const submit = async (values, submitOptions) => {
+        // Unbranch things
+        const checkpointId = submitOptions?.checkpoint?.checkpoint_id;
+        setBranch(checkpointId != null
+            ? branchContext.branchByCheckpoint[checkpointId]?.branch ?? ""
+            : "");
+        stream.setStreamValues(() => {
+            if (submitOptions?.optimisticValues != null) {
+                return {
+                    ...historyValues,
+                    ...(typeof submitOptions.optimisticValues === "function"
+                        ? submitOptions.optimisticValues(historyValues)
+                        : submitOptions.optimisticValues),
+                };
+            }
+            return { ...historyValues };
+        });
+        // When `fetchStateHistory` is requested, thus we assume that branching
+        // is enabled. We then need to include the implicit branch.
+        const includeImplicitBranch = historyLimit === true || typeof historyLimit === "number";
+        let callbackMeta;
+        let rejoinKey;
+        let usableThreadId = threadId;
+        await stream.start(async (signal) => {
+            if (!usableThreadId) {
+                const thread = await client.threads.create({
+                    threadId: submitOptions?.threadId,
+                    metadata: submitOptions?.metadata,
+                });
+                usableThreadId = thread.thread_id;
+                // Pre-emptively update the thread ID before
+                // stream cancellation is kicked off and thread
+                // is being refetched
+                threadIdRef.current = usableThreadId;
+                threadIdStreamingRef.current = usableThreadId;
+                onThreadId(usableThreadId);
+            }
+            if (!usableThreadId) {
+                throw new Error("Failed to obtain valid thread ID.");
+            }
+            threadIdStreamingRef.current = usableThreadId;
+            const streamMode = (0, utils_js_1.unique)([
+                ...(submitOptions?.streamMode ?? []),
+                ...trackStreamModeRef.current,
+                ...callbackStreamMode,
+            ]);
+            let checkpoint = submitOptions?.checkpoint ??
+                (includeImplicitBranch
+                    ? branchContext.threadHead?.checkpoint
+                    : undefined) ??
+                undefined;
+            // Avoid specifying a checkpoint if user explicitly set it to null
+            if (submitOptions?.checkpoint === null)
+                checkpoint = undefined;
+            // eslint-disable-next-line @typescript-eslint/ban-ts-comment
+            // @ts-expect-error
+            if (checkpoint != null)
+                delete checkpoint.thread_id;
+            const streamResumable = submitOptions?.streamResumable ?? !!runMetadataStorage;
+            return client.runs.stream(usableThreadId, options.assistantId, {
+                input: values,
+                config: submitOptions?.config,
+                context: submitOptions?.context,
+                command: submitOptions?.command,
+                interruptBefore: submitOptions?.interruptBefore,
+                interruptAfter: submitOptions?.interruptAfter,
+                metadata: submitOptions?.metadata,
+                multitaskStrategy: submitOptions?.multitaskStrategy,
+                onCompletion: submitOptions?.onCompletion,
+                onDisconnect: submitOptions?.onDisconnect ??
+                    (streamResumable ? "continue" : "cancel"),
+                signal,
+                checkpoint,
+                streamMode,
+                streamSubgraphs: submitOptions?.streamSubgraphs,
+                streamResumable,
+                durability: submitOptions?.durability,
+                onRunCreated(params) {
+                    callbackMeta = {
+                        run_id: params.run_id,
+                        thread_id: params.thread_id ?? usableThreadId,
+                    };
+                    if (runMetadataStorage) {
+                        rejoinKey = `lg:stream:${usableThreadId}`;
+                        runMetadataStorage.setItem(rejoinKey, callbackMeta.run_id);
+                    }
+                    options.onCreated?.(callbackMeta);
+                },
+            });
+        }, {
+            getMessages,
+            setMessages,
+            initialValues: historyValues,
+            callbacks: options,
+            async onSuccess() {
+                if (rejoinKey)
+                    runMetadataStorage?.removeItem(rejoinKey);
+                const shouldRefetch = 
+                // We're expecting the whole thread state in onFinish
+                options.onFinish != null ||
+                    // We're fetching history, thus we need the latest checkpoint
+                    // to ensure we're not accidentally submitting to a wrong branch
+                    includeImplicitBranch;
+                if (shouldRefetch) {
+                    const newHistory = await history.mutate(usableThreadId);
+                    const lastHead = newHistory.at(0);
+                    if (lastHead) {
+                        // We now have the latest update from /history
+                        // Thus we can clear the local stream state
+                        options.onFinish?.(lastHead, callbackMeta);
+                        return null;
+                    }
+                }
+                return undefined;
+            },
+            onError(error) {
+                options.onError?.(error, callbackMeta);
+            },
+            onFinish() {
+                threadIdStreamingRef.current = null;
+            },
+        });
+    };
+    const joinStream = async (runId, lastEventId, joinOptions) => {
+        // eslint-disable-next-line no-param-reassign
+        lastEventId ??= "-1";
+        if (!threadId)
+            return;
+        const callbackMeta = {
+            thread_id: threadId,
+            run_id: runId,
+        };
+        await stream.start(async (signal) => {
+            threadIdStreamingRef.current = threadId;
+            return client.runs.joinStream(threadId, runId, {
+                signal,
+                lastEventId,
+                streamMode: joinOptions?.streamMode,
+            });
+        }, {
+            getMessages,
+            setMessages,
+            initialValues: historyValues,
+            callbacks: options,
+            async onSuccess() {
+                runMetadataStorage?.removeItem(`lg:stream:${threadId}`);
+                const newHistory = await history.mutate(threadId);
+                const lastHead = newHistory.at(0);
+                if (lastHead)
+                    options.onFinish?.(lastHead, callbackMeta);
+            },
+            onError(error) {
+                options.onError?.(error, callbackMeta);
+            },
+            onFinish() {
+                threadIdStreamingRef.current = null;
+            },
+        });
+    };
+    const reconnectKey = (0, react_1.useMemo)(() => {
+        if (!runMetadataStorage || stream.isLoading)
+            return undefined;
+        if (typeof window === "undefined")
+            return undefined;
+        const runId = runMetadataStorage?.getItem(`lg:stream:${threadId}`);
+        if (!runId)
+            return undefined;
+        return { runId, threadId };
+    }, [runMetadataStorage, stream.isLoading, threadId]);
+    const shouldReconnect = !!runMetadataStorage;
+    const reconnectRef = (0, react_1.useRef)({ threadId, shouldReconnect });
+    const joinStreamRef = (0, react_1.useRef)(joinStream);
+    joinStreamRef.current = joinStream;
+    (0, react_1.useEffect)(() => {
+        // reset shouldReconnect when switching threads
+        if (reconnectRef.current.threadId !== threadId) {
+            reconnectRef.current = { threadId, shouldReconnect };
+        }
+    }, [threadId, shouldReconnect]);
+    (0, react_1.useEffect)(() => {
+        if (reconnectKey && reconnectRef.current.shouldReconnect) {
+            reconnectRef.current.shouldReconnect = false;
+            void joinStreamRef.current?.(reconnectKey.runId);
+        }
+    }, [reconnectKey]);
+    const error = stream.error ?? historyError ?? history.error;
+    const values = stream.values ?? historyValues;
+    return {
+        get values() {
+            trackStreamMode("values");
+            return values;
+        },
+        client,
+        assistantId: options.assistantId,
+        error,
+        isLoading: stream.isLoading,
+        stop,
+        submit,
+        joinStream,
+        branch,
+        setBranch,
+        get history() {
+            if (historyLimit === false) {
+                throw new Error("`fetchStateHistory` must be set to `true` to use `history`");
+            }
+            return branchContext.flatHistory;
+        },
+        isThreadLoading: history.isLoading && history.data == null,
+        get experimental_branchTree() {
+            if (historyLimit === false) {
+                throw new Error("`fetchStateHistory` must be set to `true` to use `experimental_branchTree`");
+            }
+            return branchContext.branchTree;
+        },
+        get interrupt() {
+            if (values != null &&
+                "__interrupt__" in values &&
+                Array.isArray(values.__interrupt__)) {
+                const valueInterrupts = values.__interrupt__;
+                if (valueInterrupts.length === 0)
+                    return { when: "breakpoint" };
+                if (valueInterrupts.length === 1)
+                    return valueInterrupts[0];
+                // TODO: fix the typing of interrupts if multiple interrupts are returned
+                return valueInterrupts;
+            }
+            // If we're deferring to old interrupt detection logic, don't show the interrupt if the stream is loading
+            if (stream.isLoading)
+                return undefined;
+            const interrupts = branchContext.threadHead?.tasks?.at(-1)?.interrupts;
+            if (interrupts == null || interrupts.length === 0) {
+                // check if there's a next task present
+                const next = branchContext.threadHead?.next ?? [];
+                if (!next.length || error != null)
+                    return undefined;
+                return { when: "breakpoint" };
+            }
+            // Return only the current interrupt
+            return interrupts.at(-1);
+        },
+        get messages() {
+            trackStreamMode("messages-tuple", "values");
+            return getMessages(values);
+        },
+        getMessagesMetadata(message, index) {
+            trackStreamMode("values");
+            const streamMetadata = messageManager.get(message.id)?.metadata;
+            const historyMetadata = messageMetadata?.find((m) => m.messageId === (message.id ?? index));
+            if (streamMetadata != null || historyMetadata != null) {
+                return {
+                    ...historyMetadata,
+                    streamMetadata,
+                };
+            }
+            return undefined;
+        },
+    };
+}

package/dist/react/stream.lgp.d.ts

@@ -0,0 +1,7 @@
+import type { BagTemplate, UseStreamOptions, UseStream } from "./types.js";
+export declare function useStreamLGP<StateType extends Record<string, unknown> = Record<string, unknown>, Bag extends {
+    ConfigurableType?: Record<string, unknown>;
+    InterruptType?: unknown;
+    CustomEventType?: unknown;
+    UpdateType?: unknown;
+} = BagTemplate>(options: UseStreamOptions<StateType, Bag>): UseStream<StateType, Bag>;