fix: #1594 - Model settings - change thread model - go back does not see according settings (#1595)

This commit is contained in:
Louis 2024-01-15 15:52:00 +07:00 committed by GitHub
parent 26912783c4
commit e3a06aad3e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -11,14 +11,17 @@ import { useAtomValue, useSetAtom } from 'jotai'
import { extensionManager } from '@/extension' import { extensionManager } from '@/extension'
import { setConvoMessagesAtom } from '@/helpers/atoms/ChatMessage.atom' import { setConvoMessagesAtom } from '@/helpers/atoms/ChatMessage.atom'
import { import {
ModelParams,
getActiveThreadIdAtom, getActiveThreadIdAtom,
setActiveThreadIdAtom, setActiveThreadIdAtom,
setThreadModelParamsAtom,
} from '@/helpers/atoms/Thread.atom' } from '@/helpers/atoms/Thread.atom'
export default function useSetActiveThread() { export default function useSetActiveThread() {
const activeThreadId = useAtomValue(getActiveThreadIdAtom) const activeThreadId = useAtomValue(getActiveThreadIdAtom)
const setActiveThreadId = useSetAtom(setActiveThreadIdAtom) const setActiveThreadId = useSetAtom(setActiveThreadIdAtom)
const setThreadMessage = useSetAtom(setConvoMessagesAtom) const setThreadMessage = useSetAtom(setConvoMessagesAtom)
const setThreadModelParams = useSetAtom(setThreadModelParamsAtom)
const setActiveThread = async (thread: Thread) => { const setActiveThread = async (thread: Thread) => {
if (activeThreadId === thread.id) { if (activeThreadId === thread.id) {
@ -35,6 +38,11 @@ export default function useSetActiveThread() {
setThreadMessage(thread.id, messages ?? []) setThreadMessage(thread.id, messages ?? [])
setActiveThreadId(thread.id) setActiveThreadId(thread.id)
const modelParams: ModelParams = {
...thread.assistants[0]?.model?.parameters,
...thread.assistants[0]?.model?.settings,
}
setThreadModelParams(thread.id, modelParams)
} }
return { activeThreadId, setActiveThread } return { activeThreadId, setActiveThread }