From 88cc5cb1bcb5b69bfba7f6d61866cf6ebce9dd81 Mon Sep 17 00:00:00 2001
From: Aiden Cline <63023139+rekram1-node@users.noreply.github.com>
Date: Sun, 12 Jul 2026 22:08:57 +0000
Subject: [PATCH] fix(tui): restore reverted prompt optimistically

---
 .../tui/src/routes/session/dialog-message.tsx     | 15 ++++-----------
 1 file changed, 4 insertions(+), 11 deletions(-)

diff --git a/packages/tui/src/routes/session/dialog-message.tsx b/packages/tui/src/routes/session/dialog-message.tsx
index 9cb689c0b3..1ddb629952 100644
--- a/packages/tui/src/routes/session/dialog-message.tsx
+++ b/packages/tui/src/routes/session/dialog-message.tsx
@@ -25,19 +25,12 @@ export function DialogMessage(props: { messageID: string; sessionID: string }) {
           title: "Revert",
           value: "session.revert",
           description: "undo messages and file changes",
-          onSelect: async (dialog) => {
-            const error = await sdk.api.session.revert
-              .stage({ sessionID: props.sessionID, messageID: props.messageID })
-              .then(
-                () => undefined,
-                (error) => error,
-              )
-            if (error) {
-              toast.show({ message: errorMessage(error), variant: "error", duration: 5000 })
-              return
-            }
+          onSelect: (dialog) => {
             const value = message()
             if (value?.type === "user") promptRef.current?.set(revertedPrompt(value))
+            void sdk.api.session.revert
+              .stage({ sessionID: props.sessionID, messageID: props.messageID })
+              .catch((error) => toast.show({ message: errorMessage(error), variant: "error", duration: 5000 }))
             dialog.clear()
           },
         },