Fixed

command/markov.go

@@ -24,23 +24,45 @@ func MarkovCommand(s *discordgo.Session, i *discordgo.InteractionCreate) {
 			if numMessages <= 0 {
 				numMessages = 100
 			} else if numMessages > 1000 {
-                numMessages = 1000
+				numMessages = 1000 // Limit to 1000 messages max
 			}
 		}
 	}
-    log.Printf("Fetching %d messages", numMessages)
+	log.Printf("Fetching up to %d messages", numMessages)
 
-    // Fetch the specified number of messages from the channel
+	// Fetch messages in batches
-    messages, err := s.ChannelMessages(channelID, numMessages, "", "", "")
+	var allMessages []*discordgo.Message
+	var lastMessageID string
+
+	for len(allMessages) < numMessages {
+		batchSize := 100
+		if numMessages-len(allMessages) < 100 {
+			batchSize = numMessages - len(allMessages)
+		}
+
+		batch, err := s.ChannelMessages(channelID, batchSize, lastMessageID, "", "")
 		if err != nil {
 			log.Printf("Error fetching messages: %v", err)
 			respondWithError(s, i, "Failed to fetch messages")
 			return
 		}
-    log.Printf("Fetched %d messages", len(messages))
+
+		if len(batch) == 0 {
+			break // No more messages to fetch
+		}
+
+		allMessages = append(allMessages, batch...)
+		lastMessageID = batch[len(batch)-1].ID
+
+		if len(batch) < 100 {
+			break // Less than 100 messages returned, we've reached the end
+		}
+	}
+
+	log.Printf("Fetched %d messages", len(allMessages))
 
 	// Build the Markov chain from the fetched messages
-	chain := buildMarkovChain(messages)
+	chain := buildMarkovChain(allMessages)
 	log.Printf("Built Markov chain with %d entries", len(chain))
 
 	// Generate a new message using the Markov chain
@@ -53,7 +75,7 @@ func MarkovCommand(s *discordgo.Session, i *discordgo.InteractionCreate) {
 	}
 
 	// Respond to the interaction with the generated message
-	err = s.InteractionRespond(i.Interaction, &discordgo.InteractionResponse{
+	err := s.InteractionRespond(i.Interaction, &discordgo.InteractionResponse{
 		Type: discordgo.InteractionResponseChannelMessageWithSource,
 		Data: &discordgo.InteractionResponseData{
 			Content: newMessage,