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

1 files changed, 3 insertions, 2 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 => {