@dyyz1993/pi-coding-agent 0.74.36 → 0.74.39

package/dist/core/agent-session.js

@@ -2528,18 +2528,20 @@ export class AgentSession {
                 // Non-user message: leaf = selected node
                 newLeafId = targetId;
             }
-            // Safety guard: reject navigation that would eliminate ALL user messages.
-            // The new leaf path is root → ... → newLeafId (or null for root reset).
-            // If a summary is provided, it will be a child of newLeafId, so the effective
-            // leaf path includes newLeafId's ancestors. Check both cases.
-            const effectiveLeafId = summaryText ? newLeafId : newLeafId;
-            const userMsgCount = this.sessionManager.countUserMessagesOnPath(effectiveLeafId);
-            if (userMsgCount === 0) {
-                return {
-                    cancelled: true,
-                    editorText: undefined,
-                    reason: `Navigation to "${targetId}" would remove all user messages from the active path. This likely means the target is before the first user message. Use file-level rollback (snapshot.rollback) instead.`,
-                };
+            // Safety guard: reject navigation that would eliminate ALL user messages
+            // ONLY when file rollback is involved (skipFiles !== true).
+            // When skipFiles=true (message-only rollback), going back to empty is valid —
+            // the user wants to undo their mistaken message.
+            if (options.skipFiles !== true) {
+                const effectiveLeafId = summaryText ? newLeafId : newLeafId;
+                const userMsgCount = this.sessionManager.countUserMessagesOnPath(effectiveLeafId);
+                if (userMsgCount === 0) {
+                    return {
+                        cancelled: true,
+                        editorText: undefined,
+                        reason: `Navigation to "${targetId}" would remove all user messages and restore files to their pre-session state. Use message-only rollback (skipFiles: true) to undo without file changes.`,
+                    };
+                }
             }
             // Switch leaf (with or without summary)
             // Summary is attached at the navigation target position (newLeafId), not the old branch