diff --git a/libs/sdk-js/.gitignore b/libs/sdk-js/.gitignore
index 16d769abe..128c6dc98 100644
--- a/libs/sdk-js/.gitignore
+++ b/libs/sdk-js/.gitignore
@@ -18,10 +18,6 @@ react-ui/server.cjs
 react-ui/server.js
 react-ui/server.d.ts
 react-ui/server.d.cts
-react-ui/types.cjs
-react-ui/types.js
-react-ui/types.d.ts
-react-ui/types.d.cts
 node_modules
 dist
 .yarn
diff --git a/libs/sdk-js/langchain.config.js b/libs/sdk-js/langchain.config.js
index d2853bfa5..21d796c87 100644
--- a/libs/sdk-js/langchain.config.js
+++ b/libs/sdk-js/langchain.config.js
@@ -16,8 +16,7 @@ export const config = {
     client: "client",
     react: "react/index",
     "react-ui": "react-ui/index",
-    "react-ui/server": "react-ui/server",
-    "react-ui/types": "react-ui/types",
+    "react-ui/server": "react-ui/server/index",
   },
   tsConfigPath: resolve("./tsconfig.json"),
   cjsSource: "./dist-cjs",
diff --git a/libs/sdk-js/package.json b/libs/sdk-js/package.json
index ef016ee43..65f7cc70e 100644
--- a/libs/sdk-js/package.json
+++ b/libs/sdk-js/package.json
@@ -99,15 +99,6 @@
       "import": "./react-ui/server.js",
       "require": "./react-ui/server.cjs"
     },
-    "./react-ui/types": {
-      "types": {
-        "import": "./react-ui/types.d.ts",
-        "require": "./react-ui/types.d.cts",
-        "default": "./react-ui/types.d.ts"
-      },
-      "import": "./react-ui/types.js",
-      "require": "./react-ui/types.cjs"
-    },
     "./package.json": "./package.json"
   },
   "files": [
@@ -131,10 +122,6 @@
     "react-ui/server.cjs",
     "react-ui/server.js",
     "react-ui/server.d.ts",
-    "react-ui/server.d.cts",
-    "react-ui/types.cjs",
-    "react-ui/types.js",
-    "react-ui/types.d.ts",
-    "react-ui/types.d.cts"
+    "react-ui/server.d.cts"
   ]
 }
diff --git a/libs/sdk-js/src/react-ui/server/index.ts b/libs/sdk-js/src/react-ui/server/index.ts
new file mode 100644
index 000000000..45c2c1e78
--- /dev/null
+++ b/libs/sdk-js/src/react-ui/server/index.ts
@@ -0,0 +1,6 @@
+export { typedUi } from "./server.js";
+export {
+  uiMessageReducer,
+  type UIMessage,
+  type RemoveUIMessage,
+} from "../types.js";
diff --git a/libs/sdk-js/src/react-ui/server.ts b/libs/sdk-js/src/react-ui/server/server.ts
similarity index 95%
rename from libs/sdk-js/src/react-ui/server.ts
rename to libs/sdk-js/src/react-ui/server/server.ts
index 759098c66..e62281484 100644
--- a/libs/sdk-js/src/react-ui/server.ts
+++ b/libs/sdk-js/src/react-ui/server/server.ts
@@ -1,6 +1,6 @@
 import { v4 as uuidv4 } from "uuid";
 import type { ComponentPropsWithoutRef, ElementType } from "react";
-import type { RemoveUIMessage, UIMessage } from "./types.js";
+import type { RemoveUIMessage, UIMessage } from "../types.js";
 
 export const typedUi = <Decl extends Record<string, ElementType>>(config: {
   writer?: (chunk: unknown) => void;