fix(ai/rsc): Improve streamable .done() types (#1132)

vercel · Mar 11, 2024 · 81ca3d6 · 81ca3d6
1 parent fce8b10
commit 81ca3d6
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 2 deletions.
diff --git a/.changeset/early-pants-count.md b/.changeset/early-pants-count.md
@@ -0,0 +1,5 @@
+---
+'ai': patch
+---
+
+fix(ai/rsc): improve .done() argument type
diff --git a/packages/core/rsc/index.ts b/packages/core/rsc/index.ts
@@ -6,10 +6,13 @@ export type {
   render,
   createAI,
 } from './rsc-server';
+
 export type {
   readStreamableValue,
   useUIState,
   useAIState,
   useActions,
   useSyncUIState,
 } from './rsc-client';
+
+export type { StreamableValue } from './types';
diff --git a/packages/core/rsc/streamable.tsx b/packages/core/rsc/streamable.tsx
@@ -91,7 +91,7 @@ export function createStreamableUI(initialValue?: React.ReactNode) {
       closed = true;
       reject(error);
     },
-    done(...args: any) {
+    done(...args: [] | [React.ReactNode]) {
       assertStream('.done()');
 
       if (warningTimeout) {
@@ -187,7 +187,7 @@ export function createStreamableValue<T = any, E = any>(initialValue?: T) {
 
       resolvable.resolve({ error });
     },
-    done(...args: any) {
+    done(...args: [] | [T]) {
       assertStream('.done()');
 
       if (warningTimeout) {