packages/react: create separate server component bundle for Tokens (#240

)
vercel · Jun 29, 2023 · b7b93e5 · b7b93e5
1 parent 4198b95
commit b7b93e5
Show file tree

Hide file tree

Showing 7 changed files with 18 additions and 3 deletions.
diff --git a/.changeset/chilled-stingrays-change.md b/.changeset/chilled-stingrays-change.md
@@ -0,0 +1,5 @@
+---
+'ai': major
+---
+
+Add <Tokens> RSC to ai/react
diff --git a/packages/core/package.json b/packages/core/package.json
@@ -13,7 +13,7 @@
     "vue/dist/**/*"
   ],
   "scripts": {
-    "build": "tsup",
+    "build": "tsup && cat react/dist/index.server.d.ts >> react/dist/index.d.ts",
     "clean": "rm -rf dist && rm -rf react/dist && rm -rf svelte/dist && rm -rf vue/dist",
     "dev": "tsup --watch",
     "lint": "eslint \"./**/*.ts*\"",
@@ -31,6 +31,7 @@
     },
     "./react": {
       "types": "./react/dist/index.d.ts",
+      "react-server": "./react/dist/index.server.mjs",
       "import": "./react/dist/index.mjs",
       "module": "./react/dist/index.mjs",
       "require": "./react/dist/index.js"

diff --git a/packages/core/react/index.server.ts b/packages/core/react/index.server.ts
@@ -0,0 +1 @@
+export * from './tokens'
diff --git a/packages/core/react/index.ts b/packages/core/react/index.ts
@@ -1,3 +1,2 @@
 export * from './use-chat'
 export * from './use-completion'
-export * from './tokens'
diff --git a/packages/core/react/server.ts b/packages/core/react/server.ts
@@ -0,0 +1 @@
+export * from './tokens'
diff --git a/packages/core/react/tokens.tsx b/packages/core/react/tokens.tsx
@@ -9,6 +9,7 @@ type Props = {
 
 /**
  * A React Server Component that recursively renders a stream of tokens.
+ * Can only be used inside of server components.
  */
 export async function Tokens(props: Props) {
   const { stream } = props
@@ -24,7 +25,7 @@ export async function Tokens(props: Props) {
 
 type InternalProps = {
   reader: ReadableStreamDefaultReader
-} & Omit<Props, 'stream'>
+}
 
 async function RecursiveTokens({ reader }: InternalProps) {
   const { done, value } = await reader.read()

diff --git a/packages/core/tsup.config.ts b/packages/core/tsup.config.ts
@@ -19,6 +19,13 @@ export default defineConfig([
     external: ['react', 'svelte', 'vue'],
     dts: true
   },
+  {
+    entry: ['react/index.server.ts'],
+    outDir: 'react/dist',
+    format: ['cjs', 'esm'],
+    external: ['react', 'svelte', 'vue'],
+    dts: true
+  },
   // Svelte APIs
   {
     entry: ['svelte/index.ts'],