fix (ai/rsc): "could not find InternalStreamableUIClient" bug (#2068)

vercel · Jun 24, 2024 · 74e2822 · 74e2822
1 parent 2b2960f
commit 74e2822
Show file tree

Hide file tree

Showing 24 changed files with 916 additions and 563 deletions.
diff --git a/.changeset/smart-trainers-build.md b/.changeset/smart-trainers-build.md
@@ -0,0 +1,5 @@
+---
+'ai': patch
+---
+
+fix (ai/rsc): "could not find InternalStreamableUIClient" bug
diff --git a/packages/core/rsc/__snapshots__/streamable.ui.test.tsx.snap b/packages/core/rsc/__snapshots__/streamable.ui.test.tsx.snap
@@ -0,0 +1,91 @@
+// Vitest Snapshot v1, https://vitest.dev/guide/snapshot.html
+
+exports[`rsc - render() > should emit React Nodes with async render function 1`] = `
+{
+  "children": {
+    "children": {},
+    "props": {
+      "c": undefined,
+      "n": {
+        "done": false,
+        "next": {
+          "done": true,
+          "value": <div>
+            Weather
+          </div>,
+        },
+        "value": <div>
+          Weather
+        </div>,
+      },
+    },
+    "type": "",
+  },
+  "props": {
+    "fallback": undefined,
+  },
+  "type": "Symbol(react.suspense)",
+}
+`;
+
+exports[`rsc - render() > should emit React Nodes with generator render function 1`] = `
+{
+  "children": {
+    "children": {},
+    "props": {
+      "c": undefined,
+      "n": {
+        "done": false,
+        "next": {
+          "done": false,
+          "next": {
+            "done": true,
+            "value": <div>
+              Weather
+            </div>,
+          },
+          "value": <div>
+            Weather
+          </div>,
+        },
+        "value": <div>
+          Loading...
+        </div>,
+      },
+    },
+    "type": "",
+  },
+  "props": {
+    "fallback": undefined,
+  },
+  "type": "Symbol(react.suspense)",
+}
+`;
+
+exports[`rsc - render() > should emit React Nodes with sync render function 1`] = `
+{
+  "children": {
+    "children": {},
+    "props": {
+      "c": undefined,
+      "n": {
+        "done": false,
+        "next": {
+          "done": true,
+          "value": <div>
+            Weather
+          </div>,
+        },
+        "value": <div>
+          Weather
+        </div>,
+      },
+    },
+    "type": "",
+  },
+  "props": {
+    "fallback": undefined,
+  },
+  "type": "Symbol(react.suspense)",
+}
+`;
diff --git a/packages/core/rsc/rsc-shared.mts b/packages/core/rsc/rsc-shared.mts
@@ -8,5 +8,4 @@ export {
   useActions,
   useSyncUIState,
   InternalAIProvider,
-  InternalStreamableUIClient,
 } from './shared-client';
diff --git a/packages/core/rsc/shared-client/client-wrapper.tsx b/packages/core/rsc/shared-client/client-wrapper.tsx
diff --git a/packages/core/rsc/shared-client/index.ts b/packages/core/rsc/shared-client/index.ts
@@ -8,4 +8,3 @@ export {
   useSyncUIState,
   InternalAIProvider,
 } from './context';
-export { InternalStreamableUIClient } from './client-wrapper';
diff --git a/packages/core/rsc/shared-client/streamable.tsx b/packages/core/rsc/shared-client/streamable.tsx
@@ -1,9 +1,4 @@
-import {
-  ReactElement,
-  startTransition,
-  useLayoutEffect,
-  useState,
-} from 'react';
+import { startTransition, useLayoutEffect, useState } from 'react';
 import { STREAMABLE_VALUE_TYPE } from '../constants';
 import type { StreamableValue } from '../types';
 
@@ -102,14 +97,6 @@ export function readStreamableValue<T = unknown>(
                     (curr as string) = curr + row.diff[1];
                   }
                   break;
-                case 1:
-                  (curr as ReactElement) = (
-                    <>
-                      {curr}
-                      {row.diff[1]}
-                    </>
-                  );
-                  break;
               }
             } else {
               curr = row.curr;