@langchain/langgraph-sdk 0.0.78 → 0.0.79

package/dist/react/stream.cjs

@@ -346,12 +346,14 @@ function useStream(options) {
         }
     };
     async function consumeStream(action) {
+        let getCallbackMeta;
         try {
             setIsLoading(true);
             setStreamError(undefined);
             submittingRef.current = true;
             abortRef.current = new AbortController();
             const run = await action(abortRef.current.signal);
+            getCallbackMeta = run.getCallbackMeta;
             let streamError;
             for await (const { event, data } of run.stream) {
                 if (event === "error") {
@@ -406,14 +408,14 @@ function useStream(options) {
                 throw streamError;
             const lastHead = result.at(0);
             if (lastHead)
-                onFinish?.(lastHead);
+                onFinish?.(lastHead, getCallbackMeta?.());
         }
         catch (error) {
             if (!(error instanceof Error &&
                 (error.name === "AbortError" || error.name === "TimeoutError"))) {
                 console.error(error);
                 setStreamError(error);
-                onError?.(error);
+                onError?.(error, getCallbackMeta?.());
             }
         }
         finally {
@@ -439,6 +441,7 @@ function useStream(options) {
                     return history.mutate(threadId);
                 },
                 stream,
+                getCallbackMeta: () => ({ thread_id: threadId, run_id: runId }),
             };
         });
     };
@@ -480,6 +483,8 @@ function useStream(options) {
             if (checkpoint != null)
                 delete checkpoint.thread_id;
             let rejoinKey;
+            let callbackMeta;
+            const streamResumable = submitOptions?.streamResumable ?? !!runMetadataStorage;
             const stream = client.runs.stream(usableThreadId, assistantId, {
                 input: values,
                 config: submitOptions?.config,
@@ -490,26 +495,27 @@ function useStream(options) {
                 multitaskStrategy: submitOptions?.multitaskStrategy,
                 onCompletion: submitOptions?.onCompletion,
                 onDisconnect: submitOptions?.onDisconnect ??
-                    (runMetadataStorage ? "continue" : "cancel"),
+                    (streamResumable ? "continue" : "cancel"),
                 signal,
                 checkpoint,
                 streamMode,
                 streamSubgraphs: submitOptions?.streamSubgraphs,
-                streamResumable: submitOptions?.streamResumable ?? !!runMetadataStorage,
+                streamResumable,
                 onRunCreated(params) {
-                    const runParams = {
+                    callbackMeta = {
                         run_id: params.run_id,
                         thread_id: params.thread_id ?? usableThreadId,
                     };
                     if (runMetadataStorage) {
-                        rejoinKey = `lg:stream:${runParams.thread_id}`;
-                        runMetadataStorage.setItem(rejoinKey, runParams.run_id);
+                        rejoinKey = `lg:stream:${callbackMeta.thread_id}`;
+                        runMetadataStorage.setItem(rejoinKey, callbackMeta.run_id);
                     }
-                    onCreated?.(runParams);
+                    onCreated?.(callbackMeta);
                 },
             });
             return {
                 stream,
+                getCallbackMeta: () => callbackMeta,
                 onSuccess: () => {
                     if (rejoinKey)
                         runMetadataStorage?.removeItem(rejoinKey);

package/dist/react/stream.d.ts

@@ -53,6 +53,10 @@ type GetInterruptType<Bag extends BagTemplate> = Bag extends {
 type GetCustomEventType<Bag extends BagTemplate> = Bag extends {
     CustomEventType: unknown;
 } ? Bag["CustomEventType"] : unknown;
+interface RunCallbackMeta {
+    run_id: string;
+    thread_id: string;
+}
 export interface UseStreamOptions<StateType extends Record<string, unknown> = Record<string, unknown>, Bag extends BagTemplate = BagTemplate> {
     /**
      * The ID of the assistant to use.
@@ -88,18 +92,15 @@ export interface UseStreamOptions<StateType extends Record<string, unknown> = Re
     /**
      * Callback that is called when an error occurs.
      */
-    onError?: (error: unknown) => void;
+    onError?: (error: unknown, run: RunCallbackMeta | undefined) => void;
     /**
      * Callback that is called when the stream is finished.
      */
-    onFinish?: (state: ThreadState<StateType>) => void;
+    onFinish?: (state: ThreadState<StateType>, run: RunCallbackMeta | undefined) => void;
     /**
      * Callback that is called when a new stream is created.
      */
-    onCreated?: (run: {
-        run_id: string;
-        thread_id: string;
-    }) => void;
+    onCreated?: (run: RunCallbackMeta) => void;
     /**
      * Callback that is called when an update event is received.
      */

package/dist/react/stream.js

@@ -343,12 +343,14 @@ export function useStream(options) {
         }
     };
     async function consumeStream(action) {
+        let getCallbackMeta;
         try {
             setIsLoading(true);
             setStreamError(undefined);
             submittingRef.current = true;
             abortRef.current = new AbortController();
             const run = await action(abortRef.current.signal);
+            getCallbackMeta = run.getCallbackMeta;
             let streamError;
             for await (const { event, data } of run.stream) {
                 if (event === "error") {
@@ -403,14 +405,14 @@ export function useStream(options) {
                 throw streamError;
             const lastHead = result.at(0);
             if (lastHead)
-                onFinish?.(lastHead);
+                onFinish?.(lastHead, getCallbackMeta?.());
         }
         catch (error) {
             if (!(error instanceof Error &&
                 (error.name === "AbortError" || error.name === "TimeoutError"))) {
                 console.error(error);
                 setStreamError(error);
-                onError?.(error);
+                onError?.(error, getCallbackMeta?.());
             }
         }
         finally {
@@ -436,6 +438,7 @@ export function useStream(options) {
                     return history.mutate(threadId);
                 },
                 stream,
+                getCallbackMeta: () => ({ thread_id: threadId, run_id: runId }),
             };
         });
     };
@@ -477,6 +480,8 @@ export function useStream(options) {
             if (checkpoint != null)
                 delete checkpoint.thread_id;
             let rejoinKey;
+            let callbackMeta;
+            const streamResumable = submitOptions?.streamResumable ?? !!runMetadataStorage;
             const stream = client.runs.stream(usableThreadId, assistantId, {
                 input: values,
                 config: submitOptions?.config,
@@ -487,26 +492,27 @@ export function useStream(options) {
                 multitaskStrategy: submitOptions?.multitaskStrategy,
                 onCompletion: submitOptions?.onCompletion,
                 onDisconnect: submitOptions?.onDisconnect ??
-                    (runMetadataStorage ? "continue" : "cancel"),
+                    (streamResumable ? "continue" : "cancel"),
                 signal,
                 checkpoint,
                 streamMode,
                 streamSubgraphs: submitOptions?.streamSubgraphs,
-                streamResumable: submitOptions?.streamResumable ?? !!runMetadataStorage,
+                streamResumable,
                 onRunCreated(params) {
-                    const runParams = {
+                    callbackMeta = {
                         run_id: params.run_id,
                         thread_id: params.thread_id ?? usableThreadId,
                     };
                     if (runMetadataStorage) {
-                        rejoinKey = `lg:stream:${runParams.thread_id}`;
-                        runMetadataStorage.setItem(rejoinKey, runParams.run_id);
+                        rejoinKey = `lg:stream:${callbackMeta.thread_id}`;
+                        runMetadataStorage.setItem(rejoinKey, callbackMeta.run_id);
                     }
-                    onCreated?.(runParams);
+                    onCreated?.(callbackMeta);
                 },
             });
             return {
                 stream,
+                getCallbackMeta: () => callbackMeta,
                 onSuccess: () => {
                     if (rejoinKey)
                         runMetadataStorage?.removeItem(rejoinKey);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@langchain/langgraph-sdk",
-  "version": "0.0.78",
+  "version": "0.0.79",
   "description": "Client library for interacting with the LangGraph API",
   "type": "module",
   "packageManager": "yarn@1.22.19",