From 8832484272002b4f0f23333523b360ee967b6be9 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Fri, 4 Jul 2025 07:34:44 +0800
Subject: feat: enhence `.regenerate` (#1340)

---
 src/config/session.rs | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

(limited to 'src/config/session.rs')

diff --git a/src/config/session.rs b/src/config/session.rs
index f3a86de..3f24289 100644
--- a/src/config/session.rs
+++ b/src/config/session.rs
@@ -515,7 +515,13 @@ impl Session {
         if input.continue_output().is_some() {
             return messages;
         } else if input.regenerate() {
-            messages.pop();
+            while let Some(last) = messages.last() {
+                if !last.role.is_user() {
+                    messages.pop();
+                } else {
+                    break;
+                }
+            }
             return messages;
         }
         let mut need_add_msg = true;
-- 
cgit v1.2.3