@langchain/langgraph-sdk 0.0.103 → 0.0.105

package/CHANGELOG.md

@@ -1,5 +1,19 @@
 # @langchain/langgraph-sdk
 
+## 0.0.105
+
+### Patch Changes
+
+- 7054a6a: add context to assistantBase interface
+
+## 0.0.104
+
+### Patch Changes
+
+- af9ec5a: feat(sdk): add `isThreadLoading` option to `useStream`, handle thread error fetching
+- 8e1ec9e: feat(sdk): add Context API support for useStream
+- f43e48c: fix(sdk): handle subgraph custom events in stream processing of useStream
+
 ## 0.0.103
 
 ### Patch Changes

package/dist/react/stream.cjs

@@ -198,25 +198,38 @@ function fetchHistory(client, threadId, options) {
     const limit = typeof options?.limit === "number" ? options.limit : 1000;
     return client.threads.getHistory(threadId, { limit });
 }
-function useThreadHistory(threadId, client, limit, clearCallbackRef, submittingRef) {
-    const [history, setHistory] = (0, react_1.useState)([]);
+function useThreadHistory(threadId, client, limit, clearCallbackRef, submittingRef, onErrorRef) {
+    const [history, setHistory] = (0, react_1.useState)(undefined);
+    const [isLoading, setIsLoading] = (0, react_1.useState)(false);
+    const [error, setError] = (0, react_1.useState)(undefined);
     const clientHash = (0, client_js_1.getClientConfigHash)(client);
     const clientRef = (0, react_1.useRef)(client);
     clientRef.current = client;
     const fetcher = (0, react_1.useCallback)((threadId) => {
         if (threadId != null) {
             const client = clientRef.current;
+            setIsLoading(true);
             return fetchHistory(client, threadId, {
                 limit,
-            }).then((history) => {
+            })
+                .then((history) => {
                 setHistory(history);
                 return history;
+            }, (error) => {
+                setError(error);
+                onErrorRef.current?.(error);
+                return Promise.reject(error);
+            })
+                .finally(() => {
+                setIsLoading(false);
             });
         }
-        setHistory([]);
+        setHistory(undefined);
+        setError(undefined);
+        setIsLoading(false);
         clearCallbackRef.current?.();
         return Promise.resolve([]);
-    }, [clearCallbackRef, limit]);
+    }, [clearCallbackRef, onErrorRef, limit]);
     (0, react_1.useEffect)(() => {
         if (submittingRef.current)
             return;
@@ -224,6 +237,8 @@ function useThreadHistory(threadId, client, limit, clearCallbackRef, submittingR
     }, [fetcher, clientHash, limit, submittingRef, threadId]);
     return {
         data: history,
+        isLoading,
+        error,
         mutate: (mutateId) => fetcher(mutateId ?? threadId),
     };
 }
@@ -346,20 +361,22 @@ function useStream(options) {
         setStreamValues(null);
         messageManagerRef.current.clear();
     };
+    const onErrorRef = (0, react_1.useRef)(undefined);
+    onErrorRef.current = options.onError;
     const historyLimit = typeof fetchStateHistory === "object" && fetchStateHistory != null
         ? fetchStateHistory.limit ?? true
         : fetchStateHistory ?? true;
-    const history = useThreadHistory(threadId, client, historyLimit, clearCallbackRef, submittingRef);
+    const history = useThreadHistory(threadId, client, historyLimit, clearCallbackRef, submittingRef, onErrorRef);
     const getMessages = (0, react_1.useMemo)(() => {
         return (value) => Array.isArray(value[messagesKey])
             ? value[messagesKey]
             : [];
     }, [messagesKey]);
-    const { rootSequence, paths } = getBranchSequence(history.data);
+    const { rootSequence, paths } = getBranchSequence(history.data ?? []);
     const { history: flatHistory, branchByCheckpoint } = getBranchView(rootSequence, paths, branch);
     const threadHead = flatHistory.at(-1);
     const historyValues = threadHead?.values ?? options.initialValues ?? {};
-    const historyError = (() => {
+    const historyValueError = (() => {
         const error = threadHead?.tasks?.at(-1)?.error;
         if (error == null)
             return undefined;
@@ -379,10 +396,10 @@ function useStream(options) {
         const alreadyShown = new Set();
         return getMessages(historyValues).map((message, idx) => {
             const messageId = message.id ?? idx;
-            const firstSeenIdx = findLastIndex(history.data, (state) => getMessages(state.values)
+            const firstSeenIdx = findLastIndex(history.data ?? [], (state) => getMessages(state.values)
                 .map((m, idx) => m.id ?? idx)
                 .includes(messageId));
-            const firstSeen = history.data[firstSeenIdx];
+            const firstSeen = history.data?.[firstSeenIdx];
             const checkpointId = firstSeen?.checkpoint?.checkpoint_id;
             let branch = firstSeen && checkpointId != null
                 ? branchByCheckpoint[checkpointId]
@@ -433,7 +450,10 @@ function useStream(options) {
                 }
                 if (event === "updates")
                     options.onUpdateEvent?.(data);
-                if (event === "custom")
+                if (event === "custom" ||
+                    // if `streamSubgraphs: true`, then we also want
+                    // to also receive custom events from subgraphs
+                    event.startsWith("custom|"))
                     options.onCustomEvent?.(data, {
                         mutate: getMutateFn("stream", historyValues),
                     });
@@ -553,6 +573,8 @@ function useStream(options) {
                 onThreadId(thread.thread_id);
                 usableThreadId = thread.thread_id;
             }
+            if (!usableThreadId)
+                throw new Error("Failed to obtain valid thread ID.");
             const streamMode = unique([
                 ...(submitOptions?.streamMode ?? []),
                 ...trackStreamModeRef.current,
@@ -569,6 +591,7 @@ function useStream(options) {
             const stream = client.runs.stream(usableThreadId, assistantId, {
                 input: values,
                 config: submitOptions?.config,
+                context: submitOptions?.context,
                 command: submitOptions?.command,
                 interruptBefore: submitOptions?.interruptBefore,
                 interruptAfter: submitOptions?.interruptAfter,
@@ -631,7 +654,7 @@ function useStream(options) {
             void joinStreamRef.current?.(reconnectKey.runId);
         }
     }, [reconnectKey]);
-    const error = streamError ?? historyError;
+    const error = streamError ?? historyValueError ?? history.error;
     const values = streamValues ?? historyValues;
     return {
         get values() {
@@ -648,6 +671,7 @@ function useStream(options) {
         branch,
         setBranch,
         history: flatHistory,
+        isThreadLoading: history.isLoading && history.data == null,
         get experimental_branchTree() {
             if (historyLimit === false) {
                 throw new Error("`experimental_branchTree` is not available when `fetchStateHistory` is set to `false`");

package/dist/react/stream.d.ts

@@ -203,6 +203,10 @@ export interface UseStream<StateType extends Record<string, unknown> = Record<st
      * Whether the stream is currently running.
      */
     isLoading: boolean;
+    /**
+     * Whether the thread is currently being loaded.
+     */
+    isThreadLoading: boolean;
     /**
      * Stops the stream.
      */
@@ -264,8 +268,9 @@ export interface UseStream<StateType extends Record<string, unknown> = Record<st
 type ConfigWithConfigurable<ConfigurableType extends Record<string, unknown>> = Config & {
     configurable?: ConfigurableType;
 };
-interface SubmitOptions<StateType extends Record<string, unknown> = Record<string, unknown>, ConfigurableType extends Record<string, unknown> = Record<string, unknown>> {
-    config?: ConfigWithConfigurable<ConfigurableType>;
+interface SubmitOptions<StateType extends Record<string, unknown> = Record<string, unknown>, ContextType extends Record<string, unknown> = Record<string, unknown>> {
+    config?: ConfigWithConfigurable<ContextType>;
+    context?: ContextType;
     checkpoint?: Omit<Checkpoint, "thread_id"> | null;
     command?: Command;
     interruptBefore?: "*" | string[];

package/dist/react/stream.js

@@ -194,25 +194,38 @@ function fetchHistory(client, threadId, options) {
     const limit = typeof options?.limit === "number" ? options.limit : 1000;
     return client.threads.getHistory(threadId, { limit });
 }
-function useThreadHistory(threadId, client, limit, clearCallbackRef, submittingRef) {
-    const [history, setHistory] = useState([]);
+function useThreadHistory(threadId, client, limit, clearCallbackRef, submittingRef, onErrorRef) {
+    const [history, setHistory] = useState(undefined);
+    const [isLoading, setIsLoading] = useState(false);
+    const [error, setError] = useState(undefined);
     const clientHash = getClientConfigHash(client);
     const clientRef = useRef(client);
     clientRef.current = client;
     const fetcher = useCallback((threadId) => {
         if (threadId != null) {
             const client = clientRef.current;
+            setIsLoading(true);
             return fetchHistory(client, threadId, {
                 limit,
-            }).then((history) => {
+            })
+                .then((history) => {
                 setHistory(history);
                 return history;
+            }, (error) => {
+                setError(error);
+                onErrorRef.current?.(error);
+                return Promise.reject(error);
+            })
+                .finally(() => {
+                setIsLoading(false);
             });
         }
-        setHistory([]);
+        setHistory(undefined);
+        setError(undefined);
+        setIsLoading(false);
         clearCallbackRef.current?.();
         return Promise.resolve([]);
-    }, [clearCallbackRef, limit]);
+    }, [clearCallbackRef, onErrorRef, limit]);
     useEffect(() => {
         if (submittingRef.current)
             return;
@@ -220,6 +233,8 @@ function useThreadHistory(threadId, client, limit, clearCallbackRef, submittingR
     }, [fetcher, clientHash, limit, submittingRef, threadId]);
     return {
         data: history,
+        isLoading,
+        error,
         mutate: (mutateId) => fetcher(mutateId ?? threadId),
     };
 }
@@ -342,20 +357,22 @@ export function useStream(options) {
         setStreamValues(null);
         messageManagerRef.current.clear();
     };
+    const onErrorRef = useRef(undefined);
+    onErrorRef.current = options.onError;
     const historyLimit = typeof fetchStateHistory === "object" && fetchStateHistory != null
         ? fetchStateHistory.limit ?? true
         : fetchStateHistory ?? true;
-    const history = useThreadHistory(threadId, client, historyLimit, clearCallbackRef, submittingRef);
+    const history = useThreadHistory(threadId, client, historyLimit, clearCallbackRef, submittingRef, onErrorRef);
     const getMessages = useMemo(() => {
         return (value) => Array.isArray(value[messagesKey])
             ? value[messagesKey]
             : [];
     }, [messagesKey]);
-    const { rootSequence, paths } = getBranchSequence(history.data);
+    const { rootSequence, paths } = getBranchSequence(history.data ?? []);
     const { history: flatHistory, branchByCheckpoint } = getBranchView(rootSequence, paths, branch);
     const threadHead = flatHistory.at(-1);
     const historyValues = threadHead?.values ?? options.initialValues ?? {};
-    const historyError = (() => {
+    const historyValueError = (() => {
         const error = threadHead?.tasks?.at(-1)?.error;
         if (error == null)
             return undefined;
@@ -375,10 +392,10 @@ export function useStream(options) {
         const alreadyShown = new Set();
         return getMessages(historyValues).map((message, idx) => {
             const messageId = message.id ?? idx;
-            const firstSeenIdx = findLastIndex(history.data, (state) => getMessages(state.values)
+            const firstSeenIdx = findLastIndex(history.data ?? [], (state) => getMessages(state.values)
                 .map((m, idx) => m.id ?? idx)
                 .includes(messageId));
-            const firstSeen = history.data[firstSeenIdx];
+            const firstSeen = history.data?.[firstSeenIdx];
             const checkpointId = firstSeen?.checkpoint?.checkpoint_id;
             let branch = firstSeen && checkpointId != null
                 ? branchByCheckpoint[checkpointId]
@@ -429,7 +446,10 @@ export function useStream(options) {
                 }
                 if (event === "updates")
                     options.onUpdateEvent?.(data);
-                if (event === "custom")
+                if (event === "custom" ||
+                    // if `streamSubgraphs: true`, then we also want
+                    // to also receive custom events from subgraphs
+                    event.startsWith("custom|"))
                     options.onCustomEvent?.(data, {
                         mutate: getMutateFn("stream", historyValues),
                     });
@@ -549,6 +569,8 @@ export function useStream(options) {
                 onThreadId(thread.thread_id);
                 usableThreadId = thread.thread_id;
             }
+            if (!usableThreadId)
+                throw new Error("Failed to obtain valid thread ID.");
             const streamMode = unique([
                 ...(submitOptions?.streamMode ?? []),
                 ...trackStreamModeRef.current,
@@ -565,6 +587,7 @@ export function useStream(options) {
             const stream = client.runs.stream(usableThreadId, assistantId, {
                 input: values,
                 config: submitOptions?.config,
+                context: submitOptions?.context,
                 command: submitOptions?.command,
                 interruptBefore: submitOptions?.interruptBefore,
                 interruptAfter: submitOptions?.interruptAfter,
@@ -627,7 +650,7 @@ export function useStream(options) {
             void joinStreamRef.current?.(reconnectKey.runId);
         }
     }, [reconnectKey]);
-    const error = streamError ?? historyError;
+    const error = streamError ?? historyValueError ?? history.error;
     const values = streamValues ?? historyValues;
     return {
         get values() {
@@ -644,6 +667,7 @@ export function useStream(options) {
         branch,
         setBranch,
         history: flatHistory,
+        isThreadLoading: history.isLoading && history.data == null,
         get experimental_branchTree() {
             if (historyLimit === false) {
                 throw new Error("`experimental_branchTree` is not available when `fetchStateHistory` is set to `false`");

package/dist/schema.d.ts

@@ -76,6 +76,8 @@ export interface AssistantBase {
     graph_id: string;
     /** The assistant config. */
     config: Config;
+    /** The assistant context. */
+    context: unknown;
     /** The time the assistant was created. */
     created_at: string;
     /** The assistant metadata. */

package/package.json

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