@langchain/langgraph-sdk 0.0.50 → 0.0.51

package/dist/react/stream.cjs

@@ -378,7 +378,17 @@ function useStream(options) {
                 if (event === "updates")
                     options.onUpdateEvent?.(data);
                 if (event === "custom")
-                    options.onCustomEvent?.(data);
+                    options.onCustomEvent?.(data, {
+                        mutate: (update) => setStreamValues((prev) => {
+                            // should not happen
+                            if (prev == null)
+                                return prev;
+                            return {
+                                ...prev,
+                                ...(typeof update === "function" ? update(prev) : update),
+                            };
+                        }),
+                    });
                 if (event === "metadata")
                     options.onMetadataEvent?.(data);
                 if (event === "values")

package/dist/react/stream.d.ts

@@ -96,7 +96,9 @@ interface UseStreamOptions<StateType extends Record<string, unknown> = Record<st
     /**
      * Callback that is called when a custom event is received.
      */
-    onCustomEvent?: (data: CustomStreamEvent<GetCustomEventType<Bag>>["data"]) => void;
+    onCustomEvent?: (data: CustomStreamEvent<GetCustomEventType<Bag>>["data"], options: {
+        mutate: (update: Partial<StateType> | ((prev: StateType) => Partial<StateType>)) => void;
+    }) => void;
     /**
      * Callback that is called when a metadata event is received.
      */

package/dist/react/stream.js

@@ -375,7 +375,17 @@ export function useStream(options) {
                 if (event === "updates")
                     options.onUpdateEvent?.(data);
                 if (event === "custom")
-                    options.onCustomEvent?.(data);
+                    options.onCustomEvent?.(data, {
+                        mutate: (update) => setStreamValues((prev) => {
+                            // should not happen
+                            if (prev == null)
+                                return prev;
+                            return {
+                                ...prev,
+                                ...(typeof update === "function" ? update(prev) : update),
+                            };
+                        }),
+                    });
                 if (event === "metadata")
                     options.onMetadataEvent?.(data);
                 if (event === "values")

package/dist/react-ui/index.cjs

@@ -1,8 +1,10 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.LoadExternalComponent = exports.useStreamContext = void 0;
+exports.uiMessageReducer = exports.LoadExternalComponent = exports.useStreamContext = void 0;
 const client_js_1 = require("./client.cjs");
 (0, client_js_1.bootstrapUiContext)();
 var client_js_2 = require("./client.cjs");
 Object.defineProperty(exports, "useStreamContext", { enumerable: true, get: function () { return client_js_2.useStreamContext; } });
 Object.defineProperty(exports, "LoadExternalComponent", { enumerable: true, get: function () { return client_js_2.LoadExternalComponent; } });
+var types_js_1 = require("./types.cjs");
+Object.defineProperty(exports, "uiMessageReducer", { enumerable: true, get: function () { return types_js_1.uiMessageReducer; } });

package/dist/react-ui/index.d.ts

@@ -1,2 +1,2 @@
 export { useStreamContext, LoadExternalComponent } from "./client.js";
-export type { UIMessage, RemoveUIMessage } from "./types.js";
+export { uiMessageReducer, type UIMessage, type RemoveUIMessage, } from "./types.js";

package/dist/react-ui/index.js

@@ -1,3 +1,4 @@
 import { bootstrapUiContext } from "./client.js";
 bootstrapUiContext();
 export { useStreamContext, LoadExternalComponent } from "./client.js";
+export { uiMessageReducer, } from "./types.js";

package/dist/react-ui/server/server.cjs

@@ -3,7 +3,7 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.typedUi = void 0;
 const uuid_1 = require("uuid");
 const typedUi = (config) => {
-    let collect = [];
+    let items = [];
     const runId = config.metadata?.run_id ?? config.runId;
     if (!runId)
         throw new Error("run_id is required");
@@ -13,23 +13,28 @@ const typedUi = (config) => {
         name: config.runName,
         run_id: runId,
     };
-    const create = (name, props) => ({
-        type: "ui",
-        id: (0, uuid_1.v4)(),
-        name,
-        content: props,
-        additional_kwargs: metadata,
-    });
-    const remove = (id) => ({ type: "remove-ui", id });
-    return {
-        create,
-        remove,
-        collect,
-        write: (name, props) => {
-            const evt = create(name, props);
-            collect.push(evt);
-            config.writer?.(evt);
-        },
+    const handlePush = (message, options) => {
+        const evt = {
+            type: "ui",
+            id: message?.id ?? (0, uuid_1.v4)(),
+            name: message?.name,
+            content: message?.content,
+            additional_kwargs: {
+                ...metadata,
+                ...message?.additional_kwargs,
+                ...(options?.message ? { message_id: options.message.id } : null),
+            },
+        };
+        items.push(evt);
+        config.writer?.(evt);
+        return evt;
     };
+    const handleDelete = (id) => {
+        const evt = { type: "remove-ui", id };
+        items.push(evt);
+        config.writer?.(evt);
+        return evt;
+    };
+    return { push: handlePush, delete: handleDelete, items };
 };
 exports.typedUi = typedUi;

package/dist/react-ui/server/server.d.ts

@@ -1,5 +1,8 @@
 import type { ComponentPropsWithoutRef, ElementType } from "react";
 import type { RemoveUIMessage, UIMessage } from "../types.js";
+interface MessageLike {
+    id?: string;
+}
 export declare const typedUi: <Decl extends Record<string, ElementType>>(config: {
     writer?: (chunk: unknown) => void;
     runId?: string;
@@ -7,8 +10,15 @@ export declare const typedUi: <Decl extends Record<string, ElementType>>(config:
     tags?: string[];
     runName?: string;
 }) => {
-    create: <K extends keyof Decl & string>(name: K, props: { [K_1 in keyof Decl]: ComponentPropsWithoutRef<Decl[K_1]>; }[K]) => UIMessage;
-    remove: (id: string) => RemoveUIMessage;
-    collect: (UIMessage | RemoveUIMessage)[];
-    write: <K_2 extends keyof Decl & string>(name: K_2, props: { [K_1 in keyof Decl]: ComponentPropsWithoutRef<Decl[K_1]>; }[K_2]) => void;
+    push: <K extends keyof Decl & string>(message: {
+        id?: string | undefined;
+        name: K;
+        content: { [K_1 in keyof Decl]: ComponentPropsWithoutRef<Decl[K_1]>; }[K];
+        additional_kwargs?: Record<string, unknown> | undefined;
+    }, options?: {
+        message?: MessageLike;
+    }) => UIMessage;
+    delete: (id: string) => RemoveUIMessage;
+    items: (UIMessage | RemoveUIMessage)[];
 };
+export {};

package/dist/react-ui/server/server.js

@@ -1,6 +1,6 @@
 import { v4 as uuidv4 } from "uuid";
 export const typedUi = (config) => {
-    let collect = [];
+    let items = [];
     const runId = config.metadata?.run_id ?? config.runId;
     if (!runId)
         throw new Error("run_id is required");
@@ -10,22 +10,27 @@ export const typedUi = (config) => {
         name: config.runName,
         run_id: runId,
     };
-    const create = (name, props) => ({
-        type: "ui",
-        id: uuidv4(),
-        name,
-        content: props,
-        additional_kwargs: metadata,
-    });
-    const remove = (id) => ({ type: "remove-ui", id });
-    return {
-        create,
-        remove,
-        collect,
-        write: (name, props) => {
-            const evt = create(name, props);
-            collect.push(evt);
-            config.writer?.(evt);
-        },
+    const handlePush = (message, options) => {
+        const evt = {
+            type: "ui",
+            id: message?.id ?? uuidv4(),
+            name: message?.name,
+            content: message?.content,
+            additional_kwargs: {
+                ...metadata,
+                ...message?.additional_kwargs,
+                ...(options?.message ? { message_id: options.message.id } : null),
+            },
+        };
+        items.push(evt);
+        config.writer?.(evt);
+        return evt;
     };
+    const handleDelete = (id) => {
+        const evt = { type: "remove-ui", id };
+        items.push(evt);
+        config.writer?.(evt);
+        return evt;
+    };
+    return { push: handlePush, delete: handleDelete, items };
 };

package/dist/react-ui/types.d.ts

@@ -5,6 +5,7 @@ export interface UIMessage {
     content: Record<string, unknown>;
     additional_kwargs: {
         run_id: string;
+        message_id?: string;
         [key: string]: unknown;
     };
 }

package/package.json

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