From 6bb06b870258de0cf45ee33d6993c5220cb2f313 Mon Sep 17 00:00:00 2001
From: Tat Dat Duong <david@duong.cz>
Date: Fri, 18 Apr 2025 01:07:13 +0200
Subject: [PATCH] fix(sdk-js): make sure to wrap client component in
 UseStreamContext

---
 libs/sdk-js/package.json            | 2 +-
 libs/sdk-js/src/react-ui/client.tsx | 6 +++++-
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/libs/sdk-js/package.json b/libs/sdk-js/package.json
index fa8e5150e..928fe654f 100644
--- a/libs/sdk-js/package.json
+++ b/libs/sdk-js/package.json
@@ -1,6 +1,6 @@
 {
   "name": "@langchain/langgraph-sdk",
-  "version": "0.0.67",
+  "version": "0.0.68",
   "description": "Client library for interacting with the LangGraph API",
   "type": "module",
   "packageManager": "yarn@1.22.19",
diff --git a/libs/sdk-js/src/react-ui/client.tsx b/libs/sdk-js/src/react-ui/client.tsx
index ab5d4abe0..e7fa9703b 100644
--- a/libs/sdk-js/src/react-ui/client.tsx
+++ b/libs/sdk-js/src/react-ui/client.tsx
@@ -175,7 +175,11 @@ export function LoadExternalComponent({
   }, [uiClient, uiNamespace, message.name, shadowRootId, hasClientComponent]);
 
   if (hasClientComponent) {
-    return React.createElement(clientComponent, message.props);
+    return (
+      <UseStreamContext.Provider value={{ stream, meta }}>
+        {React.createElement(clientComponent, message.props)}
+      </UseStreamContext.Provider>
+    );
   }
 
   return (