diff --git a/packages/opencode/src/cli/cmd/tui/routes/session/index.tsx b/packages/opencode/src/cli/cmd/tui/routes/session/index.tsx index c7d7537a2..780809bd6 100644 --- a/packages/opencode/src/cli/cmd/tui/routes/session/index.tsx +++ b/packages/opencode/src/cli/cmd/tui/routes/session/index.tsx @@ -597,7 +597,10 @@ export function Session() { keybind: "messages_copy", category: "Session", onSelect: (dialog) => { - const lastAssistantMessage = messages().findLast((msg) => msg.role === "assistant") + const revertID = session()?.revert?.messageID + const lastAssistantMessage = messages().findLast( + (msg) => msg.role === "assistant" && (!revertID || msg.id < revertID), + ) if (!lastAssistantMessage) { toast.show({ message: "No assistant messages found", variant: "error" }) dialog.clear()