Remeber recently used model

src/components/Layouts/Layout.tsx

@@ -107,7 +107,10 @@ export default function OptionLayout({
               <div>
                 <Select
                   value={selectedModel}
-                  onChange={setSelectedModel}
+                  onChange={(e) =>  {
+                    setSelectedModel(e)
+                    localStorage.setItem("selectedModel", e)
+                  }}
                   size="large"
                   loading={isModelsLoading || isModelsFetching}
                   filterOption={(input, option) =>

src/components/Option/Knowledge/index.tsx

@@ -96,7 +96,7 @@ export const KnowledgeSettings = () => {
                         onClick={() => {
                           if (window.confirm(t("confirm.delete"))) {
                             deleteKnowledgeMutation(record.id)
-                            if (selectedKnowledge.id === record.id) {
+                            if (selectedKnowledge?.id === record?.id) {
                               setSelectedKnowledge(null)
                             }
                           }

src/components/Sidepanel/Chat/empty.tsx

@@ -97,6 +97,7 @@ export const EmptySidePanel = () => {
             <Select
               onChange={(e) => {
                 setSelectedModel(e)
+                localStorage.setItem("selectedModel", e)
               }}
               value={selectedModel}
               size="large"

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

src/libs/process-knowledge.ts

@@ -58,7 +58,7 @@ export const processKnowledge = async (msg: any, id: string): Promise<void> => {
           knownledge_id: knowledge.id,
           file_id: doc.source_id
         })
-      } else if (doc.type === "txt" || doc.type === "text/plain") {
+      } else {
         const loader = new PageAssisTXTUrlLoader({
           name: doc.filename,
           url: doc.content

src/services/ollama.ts

@@ -314,3 +314,4 @@ export const getPageShareUrl = async () => {
 export const setPageShareUrl = async (pageShareUrl: string) => {
   await storage.set("pageShareUrl", pageShareUrl)
 }
+