@langchain/langgraph-sdk 0.1.7 → 0.1.8

package/CHANGELOG.md

@@ -1,5 +1,11 @@
 # @langchain/langgraph-sdk
 
+## 0.1.8
+
+### Patch Changes
+
+- 90dcb8b: Add support for managing thread state manually outside of useStream hook via `experimental_thread` option
+
 ## 0.1.7
 
 ### Patch Changes

package/dist/react/index.d.ts

@@ -1,3 +1,3 @@
 export { useStream } from "./stream.js";
 export { FetchStreamTransport } from "./stream.custom.js";
-export type { MessageMetadata, UseStream, UseStreamOptions, UseStreamCustom, UseStreamCustomOptions, UseStreamTransport, } from "./types.js";
+export type { MessageMetadata, UseStream, UseStreamOptions, UseStreamCustom, UseStreamCustomOptions, UseStreamTransport, UseStreamThread, } from "./types.js";

package/dist/react/stream.lgp.cjs

@@ -39,6 +39,9 @@ function useThreadHistory(client, threadId, limit, options) {
     const onErrorRef = (0, react_1.useRef)(options?.onError);
     onErrorRef.current = options?.onError;
     const fetcher = (0, react_1.useCallback)((threadId, limit) => {
+        // If only passthrough is enabled, don't fetch history
+        if (options.passthrough)
+            return Promise.resolve([]);
         const client = clientRef.current;
         const key = getFetchHistoryKey(client, threadId, limit);
         if (threadId != null) {
@@ -66,7 +69,7 @@ function useThreadHistory(client, threadId, limit, options) {
         }
         setState({ key, data: undefined, error: undefined, isLoading: false });
         return Promise.resolve([]);
-    }, []);
+    }, [options.passthrough]);
     (0, react_1.useEffect)(() => {
         // Skip if a stream is already in progress, no need to fetch history
         if (options.submittingRef.current != null &&
@@ -166,10 +169,12 @@ function useStreamLGP(options) {
         options.fetchStateHistory != null
         ? options.fetchStateHistory.limit ?? false
         : options.fetchStateHistory ?? false;
-    const history = useThreadHistory(client, threadId, historyLimit, {
+    const builtInHistory = useThreadHistory(client, threadId, historyLimit, {
+        passthrough: options.experimental_thread != null,
         submittingRef: threadIdStreamingRef,
         onError: options.onError,
     });
+    const history = options.experimental_thread ?? builtInHistory;
     const getMessages = (value) => {
         const messagesKey = options.messagesKey ?? "messages";
         return Array.isArray(value[messagesKey]) ? value[messagesKey] : [];
@@ -179,7 +184,7 @@ function useStreamLGP(options) {
         return { ...current, [messagesKey]: messages };
     };
     const [branch, setBranch] = (0, react_1.useState)("");
-    const branchContext = (0, branching_js_1.getBranchContext)(branch, history.data);
+    const branchContext = (0, branching_js_1.getBranchContext)(branch, history.data ?? undefined);
     const historyValues = branchContext.threadHead?.values ??
         options.initialValues ??
         {};
@@ -344,7 +349,7 @@ function useStreamLGP(options) {
                     includeImplicitBranch;
                 if (shouldRefetch) {
                     const newHistory = await history.mutate(usableThreadId);
-                    const lastHead = newHistory.at(0);
+                    const lastHead = newHistory?.at(0);
                     if (lastHead) {
                         // We now have the latest update from /history
                         // Thus we can clear the local stream state
@@ -386,7 +391,7 @@ function useStreamLGP(options) {
             async onSuccess() {
                 runMetadataStorage?.removeItem(`lg:stream:${threadId}`);
                 const newHistory = await history.mutate(threadId);
-                const lastHead = newHistory.at(0);
+                const lastHead = newHistory?.at(0);
                 if (lastHead)
                     options.onFinish?.(lastHead, callbackMeta);
             },

package/dist/react/stream.lgp.js

@@ -35,6 +35,9 @@ function useThreadHistory(client, threadId, limit, options) {
     const onErrorRef = useRef(options?.onError);
     onErrorRef.current = options?.onError;
     const fetcher = useCallback((threadId, limit) => {
+        // If only passthrough is enabled, don't fetch history
+        if (options.passthrough)
+            return Promise.resolve([]);
         const client = clientRef.current;
         const key = getFetchHistoryKey(client, threadId, limit);
         if (threadId != null) {
@@ -62,7 +65,7 @@ function useThreadHistory(client, threadId, limit, options) {
         }
         setState({ key, data: undefined, error: undefined, isLoading: false });
         return Promise.resolve([]);
-    }, []);
+    }, [options.passthrough]);
     useEffect(() => {
         // Skip if a stream is already in progress, no need to fetch history
         if (options.submittingRef.current != null &&
@@ -162,10 +165,12 @@ export function useStreamLGP(options) {
         options.fetchStateHistory != null
         ? options.fetchStateHistory.limit ?? false
         : options.fetchStateHistory ?? false;
-    const history = useThreadHistory(client, threadId, historyLimit, {
+    const builtInHistory = useThreadHistory(client, threadId, historyLimit, {
+        passthrough: options.experimental_thread != null,
         submittingRef: threadIdStreamingRef,
         onError: options.onError,
     });
+    const history = options.experimental_thread ?? builtInHistory;
     const getMessages = (value) => {
         const messagesKey = options.messagesKey ?? "messages";
         return Array.isArray(value[messagesKey]) ? value[messagesKey] : [];
@@ -175,7 +180,7 @@ export function useStreamLGP(options) {
         return { ...current, [messagesKey]: messages };
     };
     const [branch, setBranch] = useState("");
-    const branchContext = getBranchContext(branch, history.data);
+    const branchContext = getBranchContext(branch, history.data ?? undefined);
     const historyValues = branchContext.threadHead?.values ??
         options.initialValues ??
         {};
@@ -340,7 +345,7 @@ export function useStreamLGP(options) {
                     includeImplicitBranch;
                 if (shouldRefetch) {
                     const newHistory = await history.mutate(usableThreadId);
-                    const lastHead = newHistory.at(0);
+                    const lastHead = newHistory?.at(0);
                     if (lastHead) {
                         // We now have the latest update from /history
                         // Thus we can clear the local stream state
@@ -382,7 +387,7 @@ export function useStreamLGP(options) {
             async onSuccess() {
                 runMetadataStorage?.removeItem(`lg:stream:${threadId}`);
                 const newHistory = await history.mutate(threadId);
-                const lastHead = newHistory.at(0);
+                const lastHead = newHistory?.at(0);
                 if (lastHead)
                     options.onFinish?.(lastHead, callbackMeta);
             },

package/dist/react/types.d.ts

@@ -50,6 +50,12 @@ export interface RunCallbackMeta {
     run_id: string;
     thread_id: string;
 }
+export interface UseStreamThread<StateType extends Record<string, unknown>> {
+    data: ThreadState<StateType>[] | null | undefined;
+    error: unknown;
+    isLoading: boolean;
+    mutate: (mutateId?: string) => Promise<ThreadState<StateType>[] | null | undefined>;
+}
 export interface UseStreamOptions<StateType extends Record<string, unknown> = Record<string, unknown>, Bag extends BagTemplate = BagTemplate> {
     /**
      * The ID of the assistant to use.
@@ -187,6 +193,11 @@ export interface UseStreamOptions<StateType extends Record<string, unknown> = Re
     fetchStateHistory?: boolean | {
         limit: number;
     };
+    /**
+     * Manage the thread state externally.
+     * @experimental
+     */
+    experimental_thread?: UseStreamThread<StateType>;
 }
 interface RunMetadataStorage {
     getItem(key: `lg:stream:${string}`): string | null;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@langchain/langgraph-sdk",
-  "version": "0.1.7",
+  "version": "0.1.8",
   "description": "Client library for interacting with the LangGraph API",
   "type": "module",
   "scripts": {