From ad282ae74c29bb96a213f2d02412f080bc2ce938 Mon Sep 17 00:00:00 2001 From: sigoden Date: Thu, 9 Mar 2023 22:47:10 +0800 Subject: refactor: no need to confirm conversation when token used out --- src/config/mod.rs | 14 ++++++++------ 1 file changed, 8 insertions(+), 6 deletions(-) (limited to 'src/config/mod.rs') diff --git a/src/config/mod.rs b/src/config/mod.rs index 46ae84f..2c96161 100644 --- a/src/config/mod.rs +++ b/src/config/mod.rs @@ -308,12 +308,14 @@ impl Config { } pub fn start_conversation(&mut self) -> Result<()> { - if self.conversation.is_some() { - let ans = Confirm::new("Already in a conversation, start a new one?") - .with_default(true) - .prompt()?; - if !ans { - return Ok(()); + if let Some(conversation) = self.conversation.as_ref() { + if conversation.reamind_tokens() > 0 { + let ans = Confirm::new("Already in a conversation, start a new one?") + .with_default(true) + .prompt()?; + if !ans { + return Ok(()); + } } } self.conversation = Some(Conversation::new(self.role.clone())); -- cgit v1.2.3