Remeber recently used model

src/hooks/useMessage.tsx

@@ -22,6 +22,7 @@ import { useTranslation } from "react-i18next"
 import { usePageAssist } from "@/context"
 import { formatDocs } from "@/chain/chat-with-x"
 import { OllamaEmbeddingsPageAssist } from "@/models/OllamaEmbedding"
+import { useStorage } from "@plasmohq/storage/hook"
 
 export const useMessage = () => {
   const {
@@ -33,6 +34,8 @@ export const useMessage = () => {
     setEmbeddingController
   } = usePageAssist()
   const { t } = useTranslation("option")
+  const [selectedModel, setSelectedModel] = useStorage("selectedModel")
+
   const {
     history,
     setHistory,
@@ -45,8 +48,6 @@ export const useMessage = () => {
     setIsLoading,
     isProcessing,
     setIsProcessing,
-    selectedModel,
-    setSelectedModel,
     chatMode,
     setChatMode,
     setIsEmbedding,

src/hooks/useMessageOption.tsx

@@ -30,6 +30,7 @@ import { PageAssistVectorStore } from "@/libs/PageAssistVectorStore"
 import { formatDocs } from "@/chain/chat-with-x"
 import { useWebUI } from "@/store/webui"
 import { isTTSEnabled } from "@/services/tts"
+import { useStorage } from "@plasmohq/storage/hook"
 
 export const useMessageOption = () => {
   const {
@@ -50,8 +51,6 @@ export const useMessageOption = () => {
     setIsLoading,
     isProcessing,
     setIsProcessing,
-    selectedModel,
-    setSelectedModel,
     chatMode,
     setChatMode,
     speechToTextLanguage,
@@ -67,8 +66,9 @@ export const useMessageOption = () => {
     selectedKnowledge,
     setSelectedKnowledge
   } = useStoreMessageOption()
+  const [selectedModel, setSelectedModel] = useStorage("selectedModel")
 
-  const { ttsEnabled, setTTSEnabled } = useWebUI()
+  const { ttsEnabled } = useWebUI()
 
   const { t } = useTranslation("option")