fix: correct save of generation mode in local storage

Author: yaroslav8765

custom/ChatSurface.vue

@@ -241,6 +241,7 @@ onClickOutside(modeMenu, () => { isModeMenuOpen.value = false; });
 
 onMounted(async () => {
   agentStore.setAvailableModes(props.meta.modes, props.meta.defaultModeName);
+  agentStore.setCurrentGenerationModeFromLocalStorage();
   agentStore.regisrerTextInput(textInput.value);
   window.addEventListener('resize', updateHeight)
   textInput.value?.focus();

custom/composables/useAgentStore.ts

@@ -148,10 +148,6 @@ export const useAgentStore = defineStore('agent', () => {
     if (coreStore.isMobile) {
       setChatWidth(window.innerWidth);
     }
-    const ativeModeNameFromLocalStorage = getLocalStorageItem('activeModeName');
-    if (ativeModeNameFromLocalStorage) {
-      setActiveMode(ativeModeNameFromLocalStorage);
-    }
     appRoot.value = document.getElementById('app');
     header.value = document.getElementById('af-header-nav');
     if (appRoot.value && header.value) {
@@ -218,6 +214,13 @@ export const useAgentStore = defineStore('agent', () => {
       ?? null;
   }
 
+  function setCurrentGenerationModeFromLocalStorage() {
+    const activeModeNameFromLocalStorage = getLocalStorageItem('activeModeName');
+    if (activeModeNameFromLocalStorage) {
+      setActiveMode(activeModeNameFromLocalStorage);
+    }
+  }
+
   function setActiveMode(modeName: string) {
     if (!availableModes.value.some((mode: AgentMode) => mode.name === modeName)) {
       return;
@@ -300,12 +303,13 @@ export const useAgentStore = defineStore('agent', () => {
     availableModes,
     activeModeName,
     setAvailableModes,
+    setCurrentGenerationModeFromLocalStorage,
     setActiveMode,
     DEFAULT_CHAT_WIDTH,
     MAX_WIDTH,
     MIN_WIDTH,
     getLocalStorageItem,
     addDebugMessage,
-    abortCurrentChatRequest
+    abortCurrentChatRequest,
   }
 })