fix

Author: tkawachi

aichat.go

@@ -134,8 +134,12 @@ func (aiChat *AIChat) fold(prompt *Prompt, input string) error {
 	if err != nil {
 		return err
 	}
+	idx := firstAllowedTokens
+	for idx > len(encoded) {
+		idx = len(encoded)
+	}
 
-	firstEncoded := encoded[:firstAllowedTokens]
+	firstEncoded := encoded[:idx]
 	firstInput := aiChat.encoder.Decode(firstEncoded)
 	temperature := firstNonZeroFloat32(aiChat.options.temperature, prompt.Temperature)
 	firstRequest := gogpt.ChatCompletionRequest{
@@ -154,15 +158,14 @@ func (aiChat *AIChat) fold(prompt *Prompt, input string) error {
 		return fmt.Errorf("no choices returned")
 	}
 	output := response.Choices[0].Message.Content
-	if firstAllowedTokens >= len(encoded) {
+	if idx >= len(encoded) {
 		fmt.Println(output)
 		return nil
 	}
 	if aiChat.options.verbose {
 		log.Printf("first output: %s", output)
 	}
 
-	idx := firstAllowedTokens
 	for idx < len(encoded) {
 		outputTokens, err := aiChat.encoder.Encode(output)
 		if err != nil {