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

Author: shuding

Diff for: .changeset/early-pants-count.md

@@ -0,0 +1,5 @@
+---
+'ai': patch
+---
+
+fix(ai/rsc): improve .done() argument type

Diff for: 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 for: 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) {