fix history removal error

eastriverlee · Jan 27, 2024 · e27f19c · e27f19c
1 parent 94ca511
commit e27f19c
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/Sources/LLM/LLM.swift b/Sources/LLM/LLM.swift
@@ -178,7 +178,7 @@ open class LLM: ObservableObject {
                 output.yield("Input is too long.")
                 return false
             } else {
-                history.removeFirst(2)
+                history.removeFirst(min(2, history.count))
                 tokens = encode(preProcess(self.input, history))
                 initialCount = tokens.count
                 currentCount = Int32(initialCount)
@@ -282,8 +282,9 @@ open class LLM: ObservableObject {
         let response = getResponse(from: processedInput)
         let output = await makeOutputFrom(response)
         history += [(.user, input), (.bot, output)]
-        if historyLimit < history.count {
-            history.removeFirst(2)
+        let historyCount = history.count
+        if historyLimit < historyCount {
+            history.removeFirst(min(2, historyCount))
         }
         postProcess(output)
         isAvailable = true