diff --git a/web/containers/ModelDropdown/index.tsx b/web/containers/ModelDropdown/index.tsx index f5ad55ad5..b5abf89d7 100644 --- a/web/containers/ModelDropdown/index.tsx +++ b/web/containers/ModelDropdown/index.tsx @@ -183,10 +183,7 @@ const ModelDropdown = ({ if (!activeThread) return const modelId = activeAssistant?.model?.id - let model = downloadedModels.find((model) => model.id === modelId) - if (!model) { - model = undefined - } + const model = downloadedModels.find((model) => model.id === modelId) setSelectedModel(model) }, [ recommendedModel, diff --git a/web/hooks/useDeleteThread.ts b/web/hooks/useDeleteThread.ts index 2d14460ff..1ea6c7579 100644 --- a/web/hooks/useDeleteThread.ts +++ b/web/hooks/useDeleteThread.ts @@ -8,6 +8,8 @@ import { currentPromptAtom } from '@/containers/Providers/Jotai' import { toaster } from '@/containers/Toast' +import useSetActiveThread from './useSetActiveThread' + import { extensionManager } from '@/extension/ExtensionManager' import { deleteChatMessageAtom as deleteChatMessagesAtom } from '@/helpers/atoms/ChatMessage.atom' @@ -27,6 +29,7 @@ export default function useDeleteThread() { const deleteMessages = useSetAtom(deleteChatMessagesAtom) const deleteThreadState = useSetAtom(deleteThreadStateAtom) + const { setActiveThread } = useSetActiveThread() const cleanThread = useCallback( async (threadId: string) => { @@ -86,7 +89,7 @@ export default function useDeleteThread() { type: 'success', }) if (availableThreads.length > 0) { - setActiveThreadId(availableThreads[0].id) + setActiveThread(availableThreads[0]) } else { setActiveThreadId(undefined) } diff --git a/web/screens/Thread/ThreadRightPanel/Tools/index.tsx b/web/screens/Thread/ThreadRightPanel/Tools/index.tsx index e233bcd05..925eebe77 100644 --- a/web/screens/Thread/ThreadRightPanel/Tools/index.tsx +++ b/web/screens/Thread/ThreadRightPanel/Tools/index.tsx @@ -32,12 +32,9 @@ const Tools = () => { useEffect(() => { if (!activeThread) return - let model = downloadedModels.find( + const model = downloadedModels.find( (model) => model.id === activeAssistant?.model.id ) - if (!model) { - model = recommendedModel - } setSelectedModel(model) }, [ recommendedModel,