feat: Add option to restore last used model for previous chats

src/assets/locale/en/settings.json

@@ -23,6 +23,9 @@
             },
             "hideCurrentChatModelSettings": {
                 "label": "Hide the current Chat Model Settings"
+            },
+            "restoreLastChatModel": {
+                "label": "Restore last used model for previous chats"
             }
         },
         "webSearch": {

src/components/Option/Settings/general-settings.tsx

@@ -24,6 +24,11 @@ export const GeneralSettings = () => {
     false
   )
 
+  const [restoreLastChatModel, setRestoreLastChatModel] = useStorage(
+    "restoreLastChatModel",
+    false
+  )
+
   const [hideCurrentChatModelSettings, setHideCurrentChatModelSettings] =
     useStorage("hideCurrentChatModelSettings", false)
 
@@ -107,6 +112,18 @@ export const GeneralSettings = () => {
           onChange={(checked) => setHideCurrentChatModelSettings(checked)}
         />
       </div>
+      <div className="flex flex-row justify-between">
+        <div className="inline-flex items-center gap-2">
+          <span className="text-gray-700   dark:text-neutral-50">
+            {t("generalSettings.settings.restoreLastChatModel.label")}
+          </span>
+        </div>
+
+        <Switch
+          checked={restoreLastChatModel}
+          onChange={(checked) => setRestoreLastChatModel(checked)}
+        />
+      </div>
       <div className="flex flex-row justify-between">
         <span className="text-gray-700 dark:text-neutral-50 ">
           {t("generalSettings.settings.darkMode.label")}

src/components/Option/Sidebar.tsx

@@ -11,14 +11,24 @@ import { useMessageOption } from "~/hooks/useMessageOption"
 import { PencilIcon, Trash2 } from "lucide-react"
 import { useNavigate } from "react-router-dom"
 import { useTranslation } from "react-i18next"
+import {
+  getLastUsedChatModel,
+  lastUsedChatModelEnabled
+} from "@/services/model-settings"
 
 type Props = {
   onClose: () => void
 }
 
 export const Sidebar = ({ onClose }: Props) => {
-  const { setMessages, setHistory, setHistoryId, historyId, clearChat } =
+  const {
-    useMessageOption()
+    setMessages,
+    setHistory,
+    setHistoryId,
+    historyId,
+    clearChat,
+    setSelectedModel
+  } = useMessageOption()
   const { t } = useTranslation(["option", "common"])
   const client = useQueryClient()
   const navigate = useNavigate()
@@ -88,6 +98,13 @@ export const Sidebar = ({ onClose }: Props) => {
                   setHistoryId(chat.id)
                   setHistory(formatToChatHistory(history))
                   setMessages(formatToMessage(history))
+                  const isLastUsedChatModel = await lastUsedChatModelEnabled()
+                  if (isLastUsedChatModel) {
+                    const currentChatModel = await getLastUsedChatModel(chat.id)
+                    if (currentChatModel) {
+                      setSelectedModel(currentChatModel)
+                    }
+                  }
                   navigate("/")
                   onClose()
                 }}>

src/hooks/chat-helper/index.ts

@@ -1,4 +1,5 @@
 import { saveHistory, saveMessage } from "@/db"
+import { setLastUsedChatModel } from "@/services/model-settings"
 import { ChatHistory } from "@/store/option"
 
 export const saveMessageOnError = async ({
@@ -23,7 +24,7 @@ export const saveMessageOnError = async ({
   historyId: string | null
   selectedModel: string
   setHistoryId: (historyId: string) => void
-  isRegenerating: boolean,
+  isRegenerating: boolean
   message_source?: "copilot" | "web-ui"
 }) => {
   if (
@@ -66,6 +67,7 @@ export const saveMessageOnError = async ({
         [],
         2
       )
+      await setLastUsedChatModel(historyId, selectedModel)
     } else {
       const newHistoryId = await saveHistory(userMessage, false, message_source)
       if (!isRegenerating) {
@@ -89,6 +91,7 @@ export const saveMessageOnError = async ({
         2
       )
       setHistoryId(newHistoryId.id)
+      await setLastUsedChatModel(newHistoryId.id, selectedModel)
     }
 
     return true
@@ -115,7 +118,7 @@ export const saveMessageOnSuccess = async ({
   message: string
   image: string
   fullText: string
-  source: any[],
+  source: any[]
   message_source?: "copilot" | "web-ui"
 }) => {
   if (historyId) {
@@ -139,6 +142,7 @@ export const saveMessageOnSuccess = async ({
       source,
       2
     )
+    await setLastUsedChatModel(historyId, selectedModel!)
   } else {
     const newHistoryId = await saveHistory(message, false, message_source)
     await saveMessage(
@@ -160,5 +164,6 @@ export const saveMessageOnSuccess = async ({
       2
     )
     setHistoryId(newHistoryId.id)
+    await setLastUsedChatModel(newHistoryId.id, selectedModel!)
   }
 }

src/services/model-settings.ts

@@ -72,7 +72,6 @@ const getAllModelSettings = async () => {
       if (!value && key === "keepAlive") {
         settings[key] = "5m"
       }
-
     }
     return settings
   } catch (error) {
@@ -81,8 +80,10 @@ const getAllModelSettings = async () => {
   }
 }
 
-const setModelSetting = async (key: string,
+const setModelSetting = async (
-    value: string | number | boolean) => {
+  key: string,
+  value: string | number | boolean
+) => {
   await storage.set(key, value)
 }
 
@@ -98,4 +99,30 @@ export const getAllDefaultModelSettings = async (): Promise<ModelSettings> => {
   return settings
 }
 
+export const lastUsedChatModelEnabled = async (): Promise<boolean> => {
+  const isLastUsedChatModelEnabled = await storage.get<boolean | undefined>(
+    "restoreLastChatModel"
+  )
+  return isLastUsedChatModelEnabled ?? false
+}
+
+export const setLastUsedChatModelEnabled = async (
+  enabled: boolean
+): Promise<void> => {
+  await storage.set("restoreLastChatModel", enabled)
+}
+
+export const getLastUsedChatModel = async (
+  historyId: string
+): Promise<string | undefined> => {
+  return await storage.get<string | undefined>(`lastUsedChatModel-${historyId}`)
+}
+
+export const setLastUsedChatModel = async (
+  historyId: string,
+  model: string
+): Promise<void> => {
+  await storage.set(`lastUsedChatModel-${historyId}`, model)
+}
+
 export { getAllModelSettings, setModelSetting }