From 899c4af9ea4dedd295a909f6ddc968b20157a505 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Fri, 10 Mar 2023 07:13:22 +0800
Subject: fix: forbid .clear role in middle of conversation

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

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

diff --git a/src/config/mod.rs b/src/config/mod.rs
index ebbd909..ff7acae 100644
--- a/src/config/mod.rs
+++ b/src/config/mod.rs
@@ -186,6 +186,14 @@ impl Config {
         }
     }
 
+    pub fn clear_role(&mut self) -> Result<()> {
+        if let Some(conversation) = self.conversation.as_ref() {
+            conversation.can_clear_role()?;
+        }
+        self.role = None;
+        Ok(())
+    }
+
     pub fn create_temp_role(&mut self, prompt: &str) -> Result<()> {
         let role = Role::new(prompt, self.temperature);
         if let Some(conversation) = self.conversation.as_mut() {
-- 
cgit v1.2.3