1 files changed, 4 insertions, 0 deletions
diff --git a/src/config/session.rs b/src/config/session.rs
index e135c58..cbfe1c2 100644
--- a/src/config/session.rs
+++ b/src/config/session.rs
@@ -78,6 +78,10 @@ impl Session {
         self.model.total_tokens(&self.messages)
     }
 
+    pub fn user_messages_len(&self) -> usize {
+        self.messages.iter().filter(|v| v.role.is_user()).count()
+    }
+
     pub fn export(&self) -> Result<String> {
         self.guard_save()?;
         let (tokens, percent) = self.tokens_and_percent();