From cfcf718769536dcf511cb427b6b493dcd2e687e2 Mon Sep 17 00:00:00 2001
From: n4ze3m <n4ze3m@gmail.com>
Date: Tue, 16 Apr 2024 23:04:25 +0530
Subject: [PATCH] Remeber recently used model

---
 src/components/Layouts/Layout.tsx         | 5 ++++-
 src/components/Option/Knowledge/index.tsx | 2 +-
 src/components/Sidepanel/Chat/empty.tsx   | 1 +
 src/hooks/useMessage.tsx                  | 5 +++--
 src/hooks/useMessageOption.tsx            | 6 +++---
 src/libs/process-knowledge.ts             | 2 +-
 src/services/ollama.ts                    | 1 +
 7 files changed, 14 insertions(+), 8 deletions(-)

diff --git a/src/components/Layouts/Layout.tsx b/src/components/Layouts/Layout.tsx
index 51f0117..b1ec2ed 100644
--- a/src/components/Layouts/Layout.tsx
+++ b/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) =>
diff --git a/src/components/Option/Knowledge/index.tsx b/src/components/Option/Knowledge/index.tsx
index 12585d3..8d5f28a 100644
--- a/src/components/Option/Knowledge/index.tsx
+++ b/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)
                             }
                           }
diff --git a/src/components/Sidepanel/Chat/empty.tsx b/src/components/Sidepanel/Chat/empty.tsx
index 5bde702..41305b8 100644
--- a/src/components/Sidepanel/Chat/empty.tsx
+++ b/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"
diff --git a/src/hooks/useMessage.tsx b/src/hooks/useMessage.tsx
index 07fe767..04d3e0f 100644
--- a/src/hooks/useMessage.tsx
+++ b/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,
diff --git a/src/hooks/useMessageOption.tsx b/src/hooks/useMessageOption.tsx
index 04070a8..495e12c 100644
--- a/src/hooks/useMessageOption.tsx
+++ b/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")
 
diff --git a/src/libs/process-knowledge.ts b/src/libs/process-knowledge.ts
index eeff415..4e3f362 100644
--- a/src/libs/process-knowledge.ts
+++ b/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
diff --git a/src/services/ollama.ts b/src/services/ollama.ts
index 2b3986a..ccca697 100644
--- a/src/services/ollama.ts
+++ b/src/services/ollama.ts
@@ -314,3 +314,4 @@ export const getPageShareUrl = async () => {
 export const setPageShareUrl = async (pageShareUrl: string) => {
   await storage.set("pageShareUrl", pageShareUrl)
 }
+