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/conversation.rs | 7 +++++++
 1 file changed, 7 insertions(+)

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

diff --git a/src/config/conversation.rs b/src/config/conversation.rs
index 7fe843d..93382a3 100644
--- a/src/config/conversation.rs
+++ b/src/config/conversation.rs
@@ -33,6 +33,13 @@ impl Conversation {
         Ok(())
     }
 
+    pub fn can_clear_role(&self) -> Result<()> {
+        if self.messages.is_empty() {
+            return Ok(());
+        }
+        bail!("Error: Cannot perform this action in the middle of conversation")
+    }
+
     pub fn update_tokens(&mut self) {
         self.tokens = num_tokens_from_messages(&self.build_emssages(""));
     }
-- 
cgit v1.2.3