🎄 merry christmas: ai sdk v6 beta + tool approval (#1361)

components/chat.tsx

@@ -85,19 +85,54 @@ export function Chat({
     stop,
     regenerate,
     resumeStream,
+    addToolApprovalResponse,
   } = useChat<ChatMessage>({
     id,
     messages: initialMessages,
     experimental_throttle: 100,
     generateId: generateUUID,
+    // Auto-continue after tool approval (only for APPROVED tools)
+    // Denied tools don't need server continuation - state is saved on next user message
+    sendAutomaticallyWhen: ({ messages: currentMessages }) => {
+      const lastMessage = currentMessages.at(-1);
+      // Only continue if a tool was APPROVED (not denied)
+      const shouldContinue =
+        lastMessage?.parts?.some(
+          (part) =>
+            "state" in part &&
+            part.state === "approval-responded" &&
+            "approval" in part &&
+            (part.approval as { approved?: boolean })?.approved === true
+        ) ?? false;
+      return shouldContinue;
+    },
     transport: new DefaultChatTransport({
       api: "/api/chat",
       fetch: fetchWithErrorHandlers,
       prepareSendMessagesRequest(request) {
+        const lastMessage = request.messages.at(-1);
+
+        // Check if this is a tool approval continuation:
+        // - Last message is NOT a user message (meaning no new user input)
+        // - OR any message has tool parts that were responded to (approved or denied)
+        const isToolApprovalContinuation =
+          lastMessage?.role !== "user" ||
+          request.messages.some((msg) =>
+            msg.parts?.some((part) => {
+              const state = (part as { state?: string }).state;
+              return (
+                state === "approval-responded" || state === "output-denied"
+              );
+            })
+          );
+
         return {
           body: {
             id: request.id,
-            message: request.messages.at(-1),
+            // Send all messages for tool approval continuation, otherwise just the last user message
+            ...(isToolApprovalContinuation
+              ? { messages: request.messages }
+              : { message: lastMessage }),
             selectedChatModel: currentModelIdRef.current,
             selectedVisibilityType: visibilityType,
             ...request.body,
@@ -170,6 +205,7 @@ export function Chat({
         />
 
         <Messages
+          addToolApprovalResponse={addToolApprovalResponse}
           chatId={id}
           isArtifactVisible={isArtifactVisible}
           isReadonly={isReadonly}
@@ -203,6 +239,7 @@ export function Chat({
       </div>
 
       <Artifact
+        addToolApprovalResponse={addToolApprovalResponse}
         attachments={attachments}
         chatId={id}
         input={input}