From f5cdd0f136d1d5ba9944135f714edf3979bb7d45 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Sat, 2 Nov 2024 07:13:35 +0800
Subject: refactor: forbid `.exit role` if the session is not empty (#963)

---
 src/config/mod.rs | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

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

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 => {
-- 
cgit v1.2.3