update history handler to correctly handle regenerating responses

Author: r-mahoney

src/History/HistoryHandler.ts

@@ -1,4 +1,4 @@
-import {  HistoryItem, Message } from "Types/types";
+import { HistoryItem, Message } from "Types/types";
 import LLMPlugin from "main";
 
 export class History {
@@ -19,17 +19,22 @@ export class History {
 		}
 	}
 
+	update(index: number, messages: Message[]) {
+		this.plugin.settings.promptHistory[index].messages = messages;
+		this.plugin.saveSettings();
+	}
+
 	reset() {
 		this.plugin.settings.promptHistory = [];
 		this.plugin.saveSettings();
 	}
 
-    //take in an index from the selected chat history
-    //overwrite history with new prompt/additional prompt
+	//take in an index from the selected chat history
+	//overwrite history with new prompt/additional prompt
 	overwriteHistory(messages: Message[], index: number) {
-		const historyItem = this.plugin.settings.promptHistory[index]
+		const historyItem = this.plugin.settings.promptHistory[index];
 		historyItem.messages = messages;
-		this.plugin.settings.promptHistory[index] = historyItem
-		this.plugin.saveSettings()
+		this.plugin.settings.promptHistory[index] = historyItem;
+		this.plugin.saveSettings();
 	}
 }

src/Plugin/Components/ChatContainer.ts

@@ -63,36 +63,44 @@ export class ChatContainer {
 		this.messageStore.subscribe(this.updateMessages.bind(this));
 	}
 
-private updateMessages(message: Message[]) {
-    const currentIndex = this.plugin.settings.currentIndex;
-    const fabIndex = this.plugin.settings.fabSettings.historyIndex;
-    const widgetIndex = this.plugin.settings.widgetSettings.historyIndex;
-    
-    if (currentIndex > -1) {
-        message = this.plugin.settings.promptHistory[currentIndex].messages;
-    }
-    
-    // Always update the current view
-    if (this.viewType === this.plugin.settings.currentView) {
-        this.resetChat();
-        this.generateIMLikeMessages(message);
-        return;
-    }
-    
-    // Update FAB view if it's showing the same history item
-    if (this.viewType === "floating-action-button" && fabIndex === currentIndex && currentIndex > -1) {
-        this.resetChat();
-        this.generateIMLikeMessages(message);
-        return;
-    }
-    
-    // Update Widget view if it's showing the same history item
-    if (this.viewType === "widget" && widgetIndex === currentIndex && currentIndex > -1) {
-        this.resetChat();
-        this.generateIMLikeMessages(message);
-        return;
-    }
-}
+	private updateMessages(message: Message[]) {
+		const currentIndex = this.plugin.settings.currentIndex;
+		const fabIndex = this.plugin.settings.fabSettings.historyIndex;
+		const widgetIndex = this.plugin.settings.widgetSettings.historyIndex;
+
+		if (currentIndex > -1) {
+			message = this.plugin.settings.promptHistory[currentIndex].messages;
+		}
+
+		// Always update the current view
+		if (this.viewType === this.plugin.settings.currentView) {
+			this.resetChat();
+			this.generateIMLikeMessages(message);
+			return;
+		}
+
+		// Update FAB view if it's showing the same history item
+		if (
+			this.viewType === "floating-action-button" &&
+			fabIndex === currentIndex &&
+			currentIndex > -1
+		) {
+			this.resetChat();
+			this.generateIMLikeMessages(message);
+			return;
+		}
+
+		// Update Widget view if it's showing the same history item
+		if (
+			this.viewType === "widget" &&
+			widgetIndex === currentIndex &&
+			currentIndex > -1
+		) {
+			this.resetChat();
+			this.generateIMLikeMessages(message);
+			return;
+		}
+	}
 
 	getMessages() {
 		return this.messageStore.getMessages();
@@ -792,6 +800,7 @@ private updateMessages(message: Message[]) {
 		messages.pop();
 		this.messageStore.setMessages(messages);
 		this.historyMessages.lastElementChild?.remove();
+		this.plugin.history.update(this.plugin.settings.currentIndex, messages);
 	}
 
 	removeMessage(header: Header, modelName: string) {