Merge pull request #249 from eharris128/broken-history

Fixes chat history
Chat history was not being updated correctly when users were regenerating responses causing only the initial response to show when user refreshed UI. Updated History handler to remove assistant message from history to be replaced by regenerated message.

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) {