fix(types): make sure Actions type is bound to prevent InferActions returning unknown (#1075)

Co-authored-by: Max Leiter <maxwell.leiter@gmail.com>

Author: juliusmarminge

.changeset/fresh-trees-listen.md

@@ -0,0 +1,5 @@
+---
+'ai': patch
+---
+
+fix `useActions` type inference

examples/next-ai-rsc/app/page.tsx

@@ -23,7 +23,7 @@ import { toast } from '@/components/ui/use-toast';
 
 export default function Page() {
   const [messages, setMessages] = useUIState<typeof AI>();
-  const { submitUserMessage } = useActions();
+  const { submitUserMessage } = useActions<typeof AI>();
   const [inputValue, setInputValue] = useState('');
   const { formRef, onKeyDown } = useEnterSubmit();
   const inputRef = useRef<HTMLTextAreaElement>(null);

packages/core/rsc/provider.tsx

@@ -77,11 +77,7 @@ export function createAI<
     ? wrapAction(onGetUIState, {})
     : undefined;
 
-  async function AI(props: {
-    children: React.ReactNode;
-    initialAIState?: AIState;
-    initialUIState?: UIState;
-  }) {
+  const AI: AIProvider<AIState, UIState, Actions> = async props => {
     if ('useState' in React) {
       // This file must be running on the React Server layer.
       // Ideally we should be using `import "server-only"` here but we can have a
@@ -114,7 +110,7 @@ export function createAI<
         {props.children}
       </InternalAIProvider>
     );
-  }
+  };
 
-  return AI as AIProvider<AIState, UIState, Actions>;
+  return AI;
 }

packages/core/rsc/types.ts

@@ -31,6 +31,8 @@ export type AIProviderProps<AIState = any, UIState = any, Actions = any> = {
   children: React.ReactNode;
   initialAIState?: AIState;
   initialUIState?: UIState;
+  /** $ActionTypes is only added for type inference and is never used at runtime **/
+  $ActionTypes?: Actions;
 };
 
 export type AIProvider<AIState = any, UIState = any, Actions = any> = (