diff --git a/client/src/components/Input/NewConversationMenu/index.jsx b/client/src/components/Input/NewConversationMenu/index.jsx index 6392a4e08c..e1b6b635dd 100644 --- a/client/src/components/Input/NewConversationMenu/index.jsx +++ b/client/src/components/Input/NewConversationMenu/index.jsx @@ -61,7 +61,11 @@ export default function NewConversationMenu() { // save selected model to localstoreage useEffect(() => { - if (endpoint) localStorage.setItem('lastConversationSetup', JSON.stringify(conversation)); + if (endpoint) { + const lastSelectedModel = JSON.parse(localStorage.getItem('lastSelectedModel')) || {}; + localStorage.setItem('lastConversationSetup', JSON.stringify(conversation)); + localStorage.setItem('lastSelectedModel', JSON.stringify({ ...lastSelectedModel, [endpoint] : conversation.model })); + } }, [conversation]); // set the current model @@ -113,7 +117,7 @@ export default function NewConversationMenu() {