avoid removing when index is 0

pepperoni21 · Aug 26, 2024 · 0574b49 · 0574b49
1 parent 1900774
commit 0574b49
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/src/history.rs b/src/history.rs
@@ -34,12 +34,13 @@ impl MessagesHistory {
         // Replacing the oldest message if the limit is reached
         // The oldest message is the first one, unless it's a system message
         if messages.len() >= self.messages_number_limit as usize {
-            let index_to_remove = messages
-                .first()
-                .map(|m| if m.role == MessageRole::System { 1 } else { 0 })
-                .unwrap_or(0);
-
-            messages.remove(index_to_remove);
+            if let Some(index_to_remove) =
+                messages
+                    .first()
+                    .map(|m| if m.role == MessageRole::System { 1 } else { 0 })
+            {
+                messages.remove(index_to_remove);
+            }
         }
 
         if message.role == MessageRole::System {