use strings.TrimLeft to remove spaces

46b9953f · Arne Müller · 90c49bed · 46b9953f
Commit 46b9953f authored Oct 18, 2023 by Arne Müller
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 4 deletions

llm/llama.go llm/llama.go +2 -4

No files found.
--- a/llm/llama.go
+++ b/llm/llama.go
@@ -464,10 +464,8 @@ func (llm *llama) Predict(ctx context.Context, prevContext []int, prompt string,
 		return err
 	}
-	// Remove first leading space from prevConvo if present
+	// Remove leading spaces from prevConvo if present
-	if len(prevConvo) > 0 && prevConvo[0] == ' ' {
+	prevConvo = strings.TrimLeft(prevConvo, " ")
-		prevConvo = prevConvo[1:]
-	}
 	var nextContext strings.Builder
 	nextContext.WriteString(prevConvo)