From d0a70a15a8875372c01129c2a3205a31749a5b75 Mon Sep 17 00:00:00 2001
From: Tat Dat Duong <david@duong.cz>
Date: Wed, 30 Apr 2025 00:34:22 +0200
Subject: [PATCH] fix(sdk-js): avoid sending run metadata in UI messages

---
 libs/langgraph/langgraph/graph/ui.py      |  3 +--
 libs/sdk-js/src/react-ui/server/server.ts | 11 +++--------
 libs/sdk-js/src/react-ui/types.ts         |  4 +++-
 3 files changed, 7 insertions(+), 11 deletions(-)

diff --git a/libs/langgraph/langgraph/graph/ui.py b/libs/langgraph/langgraph/graph/ui.py
index 70ebf95b1..aac9d2044 100644
--- a/libs/langgraph/langgraph/graph/ui.py
+++ b/libs/langgraph/langgraph/graph/ui.py
@@ -100,10 +100,9 @@ def push_ui_message(
         "name": name,
         "props": props,
         "metadata": {
-            **(config.get("metadata") or {}),
+            "run_id": config.get("run_id", None),
             "tags": config.get("tags", None),
             "name": config.get("run_name", None),
-            "run_id": config.get("run_id", None),
             **(metadata or {}),
             **({"message_id": message_id} if message_id else {}),
         },
diff --git a/libs/sdk-js/src/react-ui/server/server.ts b/libs/sdk-js/src/react-ui/server/server.ts
index 7bc7bab5d..708385694 100644
--- a/libs/sdk-js/src/react-ui/server/server.ts
+++ b/libs/sdk-js/src/react-ui/server/server.ts
@@ -38,13 +38,6 @@ export const typedUi = <Decl extends Record<string, ElementType>>(
   const runId = (config.metadata?.run_id as string | undefined) ?? config.runId;
   if (!runId) throw new Error("run_id is required");
 
-  const metadata = {
-    ...config.metadata,
-    tags: config.tags,
-    name: config.runName,
-    run_id: runId,
-  };
-
   const handlePush = <K extends keyof PropMap & string>(
     message: {
       id?: string;
@@ -60,7 +53,9 @@ export const typedUi = <Decl extends Record<string, ElementType>>(
       name: message?.name,
       props: message?.props,
       metadata: {
-        ...metadata,
+        run_id: runId,
+        tags: config.tags,
+        name: config.runName,
         ...message?.metadata,
         ...(options?.message ? { message_id: options.message.id } : null),
       },
diff --git a/libs/sdk-js/src/react-ui/types.ts b/libs/sdk-js/src/react-ui/types.ts
index c3f52b5b8..d60d81728 100644
--- a/libs/sdk-js/src/react-ui/types.ts
+++ b/libs/sdk-js/src/react-ui/types.ts
@@ -5,7 +5,9 @@ export interface UIMessage {
   name: string;
   props: Record<string, unknown>;
   metadata: {
-    run_id: string;
+    run_id?: string;
+    name?: string;
+    tags?: string[];
     message_id?: string;
     [key: string]: unknown;
   };