fix: #1594 - Model settings - change thread model - go back does not …

…see according settings (#1595)
janhq · Jan 15, 2024 · e3a06aa · e3a06aa
1 parent 2691278
commit e3a06aa
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/web/hooks/useSetActiveThread.ts b/web/hooks/useSetActiveThread.ts
@@ -11,14 +11,17 @@ import { useAtomValue, useSetAtom } from 'jotai'
 import { extensionManager } from '@/extension'
 import { setConvoMessagesAtom } from '@/helpers/atoms/ChatMessage.atom'
 import {
+  ModelParams,
   getActiveThreadIdAtom,
   setActiveThreadIdAtom,
+  setThreadModelParamsAtom,
 } from '@/helpers/atoms/Thread.atom'
 
 export default function useSetActiveThread() {
   const activeThreadId = useAtomValue(getActiveThreadIdAtom)
   const setActiveThreadId = useSetAtom(setActiveThreadIdAtom)
   const setThreadMessage = useSetAtom(setConvoMessagesAtom)
+  const setThreadModelParams = useSetAtom(setThreadModelParamsAtom)
 
   const setActiveThread = async (thread: Thread) => {
     if (activeThreadId === thread.id) {
@@ -35,6 +38,11 @@ export default function useSetActiveThread() {
     setThreadMessage(thread.id, messages ?? [])
 
     setActiveThreadId(thread.id)
+    const modelParams: ModelParams = {
+      ...thread.assistants[0]?.model?.parameters,
+      ...thread.assistants[0]?.model?.settings,
+    }
+    setThreadModelParams(thread.id, modelParams)
   }
 
   return { activeThreadId, setActiveThread }