refactor: forbid `.exit role` if the session is not empty (#963)

2 files changed, 4 insertions, 3 deletions

diff --git a/src/config/mod.rs b/src/config/mod.rs
index b1a1d2e..8f3d96c 100644
--- a/src/config/mod.rs
+++ b/src/config/mod.rs
@@ -833,7 +833,7 @@ impl Config {
 
     pub fn use_role_obj(&mut self, role: Role) -> Result<()> {
         if self.agent.is_some() {
-            bail!("Cannot perform this action because you are using a agent")
+            bail!("Cannot perform this operation because you are using a agent")
         }
         if let Some(session) = self.session.as_mut() {
             session.guard_empty()?;
@@ -861,6 +861,7 @@ impl Config {
 
     pub fn exit_role(&mut self) -> Result<()> {
         if let Some(session) = self.session.as_mut() {
+            session.guard_empty()?;
             session.clear_role();
         } else if self.role.is_some() {
             self.role = None;
@@ -1193,7 +1194,7 @@ impl Config {
         abort_signal: AbortSignal,
     ) -> Result<()> {
         if config.read().agent.is_some() {
-            bail!("Cannot perform this action because you are using a agent")
+            bail!("Cannot perform this operation because you are using a agent")
         }
         let rag = match rag {
             None => {
diff --git a/src/config/session.rs b/src/config/session.rs
index db75ba7..7335128 100644
--- a/src/config/session.rs
+++ b/src/config/session.rs
@@ -367,7 +367,7 @@ impl Session {
 
     pub fn guard_empty(&self) -> Result<()> {
         if !self.is_empty() {
-            bail!("This action cannot be performed in a session with messages.")
+            bail!("Cannot perform this operation because the session has messages, please `.empty session` first.");
         }
         Ok(())
     }